#include <functional>
#include <algorithm>
-#define DO_GINAC_ASSERT
-#include "assertion.h"
-
#include "symmetry.h"
#include "lst.h"
#include "numeric.h" // for factorial()
GINAC_IMPLEMENT_REGISTERED_CLASS(symmetry, basic)
+/*
+ Some notes about the structure of a symmetry tree:
+ - The leaf nodes of the tree are of type "none", have one index, and no
+ children (of course). They are constructed by the symmetry(unsigned)
+ constructor.
+ - Leaf nodes are the only nodes that only have one index.
+ - Container nodes contain two or more children. The "indices" set member
+ is the set union of the index sets of all children, and the "children"
+ vector stores the children themselves.
+ - The index set of each child of a "symm", "anti" or "cycl" node must
+ have the same size. It follows that the children of such a node are
+ either all leaf nodes, or all container nodes with two or more indices.
+*/
+
//////////
// default constructor, destructor, copy constructor assignment operator and helpers
//////////
while (true) {
ex e;
if (n.find_ex("child", e, sym_lst, i))
- add(ex_to_symmetry(e));
+ add(ex_to<symmetry>(e));
else
break;
i++;
DEFAULT_UNARCHIVE(symmetry)
//////////
-// functions overriding virtual functions from bases classes
+// functions overriding virtual functions from base classes
//////////
int symmetry::compare_same_type(const basic & other) const
{
GINAC_ASSERT(is_of_type(other, symmetry));
- const symmetry &o = static_cast<const symmetry &>(other);
// All symmetry trees are equal. They are not supposed to appear in
// ordinary expressions anyway...
return 0;
}
-void symmetry::print(const print_context & c, unsigned level = 0) const
+void symmetry::print(const print_context & c, unsigned level) const
{
debugmsg("symmetry print", LOGLEVEL_PRINT);
- if (children.empty()) {
- if (indices.size() > 0)
- c.s << *(indices.begin());
- } else {
+ if (is_of_type(c, print_tree)) {
+
+ c.s << std::string(level, ' ') << class_name()
+ << std::hex << ", hash=0x" << hashvalue << ", flags=0x" << flags << std::dec
+ << ", type=";
+
switch (type) {
- case none: c.s << '!'; break;
- case symmetric: c.s << '+'; break;
- case antisymmetric: c.s << '-'; break;
- case cyclic: c.s << '@'; break;
- default: c.s << '?'; break;
+ case none: c.s << "none"; break;
+ case symmetric: c.s << "symm"; break;
+ case antisymmetric: c.s << "anti"; break;
+ case cyclic: c.s << "cycl"; break;
+ default: c.s << "<unknown>"; break;
}
- c.s << '(';
- for (unsigned i=0; i<children.size(); i++) {
- children[i].print(c);
- if (i != children.size() - 1)
- c.s << ",";
+
+ c.s << ", indices=(";
+ if (!indices.empty()) {
+ std::set<unsigned>::const_iterator i = indices.begin(), end = indices.end();
+ --end;
+ while (i != end)
+ c.s << *i++ << ",";
+ c.s << *i;
+ }
+ c.s << ")\n";
+
+ unsigned delta_indent = static_cast<const print_tree &>(c).delta_indent;
+ exvector::const_iterator i = children.begin(), end = children.end();
+ while (i != end) {
+ i->print(c, level + delta_indent);
+ ++i;
+ }
+
+ } else {
+
+ if (children.empty()) {
+ if (indices.size() > 0)
+ c.s << *(indices.begin());
+ else
+ c.s << "none";
+ } else {
+ switch (type) {
+ case none: c.s << '!'; break;
+ case symmetric: c.s << '+'; break;
+ case antisymmetric: c.s << '-'; break;
+ case cyclic: c.s << '@'; break;
+ default: c.s << '?'; break;
+ }
+ c.s << '(';
+ unsigned num = children.size();
+ for (unsigned i=0; i<num; i++) {
+ children[i].print(c);
+ if (i != num - 1)
+ c.s << ",";
+ }
+ c.s << ')';
}
- c.s << ')';
}
}
// All children must have the same number of indices
if (type != none && !children.empty()) {
GINAC_ASSERT(is_ex_exactly_of_type(children[0], symmetry));
- if (ex_to_symmetry(children[0]).indices.size() != c.indices.size())
+ if (ex_to<symmetry>(children[0]).indices.size() != c.indices.size())
throw (std::logic_error("symmetry:add(): children must have same number of indices"));
}
{
GINAC_ASSERT(is_ex_exactly_of_type(lh, symmetry));
GINAC_ASSERT(is_ex_exactly_of_type(rh, symmetry));
- GINAC_ASSERT(ex_to_symmetry(lh).indices.size() == ex_to_symmetry(rh).indices.size());
- std::set<unsigned>::const_iterator ait = ex_to_symmetry(lh).indices.begin(), aitend = ex_to_symmetry(lh).indices.end(), bit = ex_to_symmetry(rh).indices.begin();
+ GINAC_ASSERT(ex_to<symmetry>(lh).indices.size() == ex_to<symmetry>(rh).indices.size());
+ std::set<unsigned>::const_iterator ait = ex_to<symmetry>(lh).indices.begin(), aitend = ex_to<symmetry>(lh).indices.end(), bit = ex_to<symmetry>(rh).indices.begin();
while (ait != aitend) {
int cmpval = v[*ait].compare(v[*bit]);
if (cmpval < 0)
{
GINAC_ASSERT(is_ex_exactly_of_type(lh, symmetry));
GINAC_ASSERT(is_ex_exactly_of_type(rh, symmetry));
- GINAC_ASSERT(ex_to_symmetry(lh).indices.size() == ex_to_symmetry(rh).indices.size());
- std::set<unsigned>::const_iterator ait = ex_to_symmetry(lh).indices.begin(), aitend = ex_to_symmetry(lh).indices.end(), bit = ex_to_symmetry(rh).indices.begin();
+ GINAC_ASSERT(ex_to<symmetry>(lh).indices.size() == ex_to<symmetry>(rh).indices.size());
+ std::set<unsigned>::const_iterator ait = ex_to<symmetry>(lh).indices.begin(), aitend = ex_to<symmetry>(lh).indices.end(), bit = ex_to<symmetry>(rh).indices.begin();
while (ait != aitend) {
v[*ait].swap(v[*bit]);
++ait; ++bit;
int canonicalize(exvector::iterator v, const symmetry &symm)
{
- // No children? Then do nothing
- if (symm.children.empty())
+ // Less than two indices? Then do nothing
+ if (symm.indices.size() < 2)
return INT_MAX;
// Canonicalize children first
exvector::const_iterator first = symm.children.begin(), last = symm.children.end();
while (first != last) {
GINAC_ASSERT(is_ex_exactly_of_type(*first, symmetry));
- int child_sign = canonicalize(v, ex_to_symmetry(*first));
+ int child_sign = canonicalize(v, ex_to<symmetry>(*first));
if (child_sign == 0)
return 0;
if (child_sign != INT_MAX) {
first = symm.children.begin();
switch (symm.type) {
case symmetry::symmetric:
+ // Sort the children in ascending order
shaker_sort(first, last, sy_is_less(v), sy_swap(v, something_changed));
break;
case symmetry::antisymmetric:
+ // Sort the children in ascending order, keeping track of the signum
sign *= permutation_sign(first, last, sy_is_less(v), sy_swap(v, something_changed));
break;
case symmetry::cyclic:
+ // Permute the smallest child to the front
cyclic_permutation(first, last, min_element(first, last, sy_is_less(v)), sy_swap(v, something_changed));
break;
default:
static ex symm(const ex & e, exvector::const_iterator first, exvector::const_iterator last, bool asymmetric)
{
// Need at least 2 objects for this operation
- int num = last - first;
+ unsigned num = last - first;
if (num < 2)
return e;
ex symmetrize_cyclic(const ex & e, exvector::const_iterator first, exvector::const_iterator last)
{
// Need at least 2 objects for this operation
- int num = last - first;
+ unsigned num = last - first;
if (num < 2)
return e;