-
Notifications
You must be signed in to change notification settings - Fork 6
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
1 changed file
with
64 additions
and
59 deletions.
There are no files selected for viewing
123 changes: 64 additions & 59 deletions
123
examples/tensor_network_graphs/tensor_network_graphs.cpp
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,88 +1,93 @@ | ||
#include <SeQuant/core/runtime.hpp> | ||
#include <SeQuant/core/bliss.hpp> | ||
#include <SeQuant/core/parse_expr.hpp> | ||
#include <SeQuant/core/runtime.hpp> | ||
#include <SeQuant/core/tensor_network.hpp> | ||
#include <SeQuant/domain/mbpt/convention.hpp> | ||
|
||
#include <string> | ||
#include <cassert> | ||
#include <codecvt> | ||
#include <locale> | ||
#include <iostream> | ||
#include <locale> | ||
#include <optional> | ||
#include <string> | ||
|
||
using namespace sequant; | ||
using namespace sequant; | ||
|
||
std::wstring from_utf8(std::string_view str) { | ||
std::wstring_convert<std::codecvt_utf8_utf16<wchar_t>> converter; | ||
return converter.from_bytes(std::string(str)); | ||
std::wstring_convert<std::codecvt_utf8_utf16<wchar_t>> converter; | ||
return converter.from_bytes(std::string(str)); | ||
} | ||
|
||
std::optional<TensorNetwork> to_network(const ExprPtr &expr) { | ||
if (expr.is<Tensor>()) { | ||
return TensorNetwork({expr}); | ||
} else if (expr.is<Product>()) { | ||
for (const ExprPtr &factor : expr.as<Product>().factors()) { | ||
if (!factor.is<Tensor>()) { | ||
return {}; | ||
} | ||
} | ||
if (expr.is<Tensor>()) { | ||
return TensorNetwork({expr}); | ||
} else if (expr.is<Product>()) { | ||
for (const ExprPtr &factor : expr.as<Product>().factors()) { | ||
if (!factor.is<Tensor>()) { | ||
return {}; | ||
} | ||
} | ||
|
||
return TensorNetwork(expr.as<Product>().factors()); | ||
} else { | ||
return {}; | ||
} | ||
return TensorNetwork(expr.as<Product>().factors()); | ||
} else { | ||
return {}; | ||
} | ||
} | ||
|
||
void print_help() { | ||
std::wcout << "Helper to generate dot (GraphViz) representations of tensor network graphs.\n"; | ||
std::wcout << "Usage:\n"; | ||
std::wcout << " <exe> [options] <network 1> [<network 2> [... [<network N>] ] ]\n"; | ||
std::wcout << "Options:\n"; | ||
std::wcout << " --help Shows this help message\n"; | ||
std::wcout << " --no-named Treat all indices as unnamed (even if they are external)\n"; | ||
std::wcout << "Helper to generate dot (GraphViz) representations of tensor " | ||
"network graphs.\n"; | ||
std::wcout << "Usage:\n"; | ||
std::wcout | ||
<< " <exe> [options] <network 1> [<network 2> [... [<network N>] ] ]\n"; | ||
std::wcout << "Options:\n"; | ||
std::wcout << " --help Shows this help message\n"; | ||
std::wcout << " --no-named Treat all indices as unnamed (even if they are " | ||
"external)\n"; | ||
} | ||
|
||
int main(int argc, char **argv) { | ||
set_locale(); | ||
mbpt::set_default_convention(); | ||
|
||
set_locale(); | ||
mbpt::set_default_convention(); | ||
|
||
bool use_named_indices = true; | ||
const TensorNetwork::named_indices_t empty_named_indices; | ||
bool use_named_indices = true; | ||
const TensorNetwork::named_indices_t empty_named_indices; | ||
|
||
if (argc <= 1) { | ||
print_help(); | ||
return 1; | ||
} | ||
if (argc <= 1) { | ||
print_help(); | ||
return 1; | ||
} | ||
|
||
for (std::size_t i = 1; i < static_cast<std::size_t>(argc); ++i) { | ||
std::wstring current = from_utf8(argv[i]); | ||
if (current == L"--help") { | ||
print_help(); | ||
return 0; | ||
} else if (current == L"--no-named") { | ||
use_named_indices = false; | ||
continue; | ||
} | ||
for (std::size_t i = 1; i < static_cast<std::size_t>(argc); ++i) { | ||
std::wstring current = from_utf8(argv[i]); | ||
if (current == L"--help") { | ||
print_help(); | ||
return 0; | ||
} else if (current == L"--no-named") { | ||
use_named_indices = false; | ||
continue; | ||
} | ||
|
||
ExprPtr expr; | ||
try { | ||
expr = parse_expr(current); | ||
} catch (const ParseError &e) { | ||
std::wcout << "Failed to parse expression '" << current << "': " << e.what() << std::endl; | ||
return 1; | ||
} | ||
assert(expr); | ||
ExprPtr expr; | ||
try { | ||
expr = parse_expr(current); | ||
} catch (const ParseError &e) { | ||
std::wcout << "Failed to parse expression '" << current | ||
<< "': " << e.what() << std::endl; | ||
return 1; | ||
} | ||
assert(expr); | ||
|
||
std::optional<TensorNetwork> network = to_network(expr); | ||
if (!network.has_value()) { | ||
std::wcout << "Failed to construct tensor network for input '" << current << "'" << std::endl; | ||
return 2; | ||
} | ||
std::optional<TensorNetwork> network = to_network(expr); | ||
if (!network.has_value()) { | ||
std::wcout << "Failed to construct tensor network for input '" << current | ||
<< "'" << std::endl; | ||
return 2; | ||
} | ||
|
||
TensorNetwork::Graph graph = network->create_graph(use_named_indices ? nullptr : &empty_named_indices); | ||
std::wcout << "Graph for '" << current << "'\n"; | ||
graph.bliss_graph->write_dot(std::wcout, graph.vertex_labels); | ||
} | ||
TensorNetwork::Graph graph = network->create_graph( | ||
use_named_indices ? nullptr : &empty_named_indices); | ||
std::wcout << "Graph for '" << current << "'\n"; | ||
graph.bliss_graph->write_dot(std::wcout, graph.vertex_labels); | ||
} | ||
} |