- prepared for 1.0.13 release
[ginac.git] / ginac / symmetry.h
1 /** @file symmetry.h
2  *
3  *  Interface to GiNaC's symmetry definitions. */
4
5 /*
6  *  GiNaC Copyright (C) 1999-2003 Johannes Gutenberg University Mainz, Germany
7  *
8  *  This program is free software; you can redistribute it and/or modify
9  *  it under the terms of the GNU General Public License as published by
10  *  the Free Software Foundation; either version 2 of the License, or
11  *  (at your option) any later version.
12  *
13  *  This program is distributed in the hope that it will be useful,
14  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
15  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  *  GNU General Public License for more details.
17  *
18  *  You should have received a copy of the GNU General Public License
19  *  along with this program; if not, write to the Free Software
20  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
21  */
22
23 #ifndef __GINAC_SYMMETRY_H__
24 #define __GINAC_SYMMETRY_H__
25
26 #include <set>
27
28 #include "ex.h"
29
30 namespace GiNaC {
31
32
33 class sy_is_less;
34 class sy_swap;
35
36 /** This class describes the symmetry of a group of indices. These objects
37  *  can be grouped into a tree to form complex mixed symmetries. */
38 class symmetry : public basic
39 {
40         friend class sy_is_less;
41         friend class sy_swap;
42         friend int canonicalize(exvector::iterator v, const symmetry &symm);
43
44         GINAC_DECLARE_REGISTERED_CLASS(symmetry, basic)
45
46         // types
47 public:
48         /** Type of symmetry */
49         typedef enum {
50                 none,          /**< no symmetry properties */
51                 symmetric,     /**< totally symmetric */
52                 antisymmetric, /**< totally antisymmetric */
53                 cyclic         /**< cyclic symmetry */
54         } symmetry_type;
55
56         // other constructors
57 public:
58         /** Create leaf node that represents one index. */
59         symmetry(unsigned i);
60
61         /** Create node with two children. */
62         symmetry(symmetry_type t, const symmetry &c1, const symmetry &c2);
63
64         // functions overriding virtual functions from base classes
65 public:
66         void print(const print_context & c, unsigned level = 0) const;
67
68         // non-virtual functions in this class
69 public:
70         /** Get symmetry type. */
71         symmetry_type get_type() const {return type;}
72
73         /** Set symmetry type. */
74         void set_type(symmetry_type t) {type = t;}
75
76         /** Add child node, check index sets for consistency. */
77         symmetry &add(const symmetry &c);
78
79         /** Verify that all indices of this node are in the range [0..n-1].
80          *  This function throws an exception if the verification fails.
81          *  If the top node has a type != none and no children, add all indices
82          *  in the range [0..n-1] as children. */
83         void validate(unsigned n);
84
85         /** Check whether this node actually represents any kind of symmetry. */
86         bool has_symmetry(void) const {return type != none || !children.empty(); }
87
88         // member variables
89 private:
90         /** Type of symmetry described by this node. */
91         symmetry_type type;
92
93         /** Sorted union set of all indices handled by this node. */
94         std::set<unsigned> indices;
95
96         /** Vector of child nodes. */
97         exvector children;
98 };
99
100
101 // global functions
102
103 inline symmetry sy_none(void) { return symmetry(); }
104 inline symmetry sy_none(const symmetry &c1, const symmetry &c2) { return symmetry(symmetry::none, c1, c2); }
105 inline symmetry sy_none(const symmetry &c1, const symmetry &c2, const symmetry &c3) { return symmetry(symmetry::none, c1, c2).add(c3); }
106 inline symmetry sy_none(const symmetry &c1, const symmetry &c2, const symmetry &c3, const symmetry &c4) { return symmetry(symmetry::none, c1, c2).add(c3).add(c4); }
107
108 inline symmetry sy_symm(void) { symmetry s; s.set_type(symmetry::symmetric); return s; }
109 inline symmetry sy_symm(const symmetry &c1, const symmetry &c2) { return symmetry(symmetry::symmetric, c1, c2); }
110 inline symmetry sy_symm(const symmetry &c1, const symmetry &c2, const symmetry &c3) { return symmetry(symmetry::symmetric, c1, c2).add(c3); }
111 inline symmetry sy_symm(const symmetry &c1, const symmetry &c2, const symmetry &c3, const symmetry &c4) { return symmetry(symmetry::symmetric, c1, c2).add(c3).add(c4); }
112
113 inline symmetry sy_anti(void) { symmetry s; s.set_type(symmetry::antisymmetric); return s; }
114 inline symmetry sy_anti(const symmetry &c1, const symmetry &c2) { return symmetry(symmetry::antisymmetric, c1, c2); }
115 inline symmetry sy_anti(const symmetry &c1, const symmetry &c2, const symmetry &c3) { return symmetry(symmetry::antisymmetric, c1, c2).add(c3); }
116 inline symmetry sy_anti(const symmetry &c1, const symmetry &c2, const symmetry &c3, const symmetry &c4) { return symmetry(symmetry::antisymmetric, c1, c2).add(c3).add(c4); }
117
118 inline symmetry sy_cycl(void) { symmetry s; s.set_type(symmetry::cyclic); return s; }
119 inline symmetry sy_cycl(const symmetry &c1, const symmetry &c2) { return symmetry(symmetry::cyclic, c1, c2); }
120 inline symmetry sy_cycl(const symmetry &c1, const symmetry &c2, const symmetry &c3) { return symmetry(symmetry::cyclic, c1, c2).add(c3); }
121 inline symmetry sy_cycl(const symmetry &c1, const symmetry &c2, const symmetry &c3, const symmetry &c4) { return symmetry(symmetry::cyclic, c1, c2).add(c3).add(c4); }
122
123 /** Canonicalize the order of elements of an expression vector, according to
124  *  the symmetry properties defined in a symmetry tree.
125  *
126  *  @param v Start of expression vector
127  *  @param symm Root node of symmetry tree
128  *  @return the overall sign introduced by the reordering (+1, -1 or 0)
129  *          or INT_MAX if nothing changed */
130 extern int canonicalize(exvector::iterator v, const symmetry &symm);
131
132 /** Symmetrize expression over a set of objects (symbols, indices). */
133 ex symmetrize(const ex & e, exvector::const_iterator first, exvector::const_iterator last);
134
135 /** Symmetrize expression over a set of objects (symbols, indices). */
136 inline ex symmetrize(const ex & e, const exvector & v)
137 {
138         return symmetrize(e, v.begin(), v.end());
139 }
140
141 /** Antisymmetrize expression over a set of objects (symbols, indices). */
142 ex antisymmetrize(const ex & e, exvector::const_iterator first, exvector::const_iterator last);
143
144 /** Antisymmetrize expression over a set of objects (symbols, indices). */
145 inline ex antisymmetrize(const ex & e, const exvector & v)
146 {
147         return antisymmetrize(e, v.begin(), v.end());
148 }
149
150 /** Symmetrize expression by cyclic permuation over a set of objects
151  *  (symbols, indices). */
152 ex symmetrize_cyclic(const ex & e, exvector::const_iterator first, exvector::const_iterator last);
153
154 /** Symmetrize expression by cyclic permutation over a set of objects
155  *  (symbols, indices). */
156 inline ex symmetrize_cyclic(const ex & e, const exvector & v)
157 {
158         return symmetrize(e, v.begin(), v.end());
159 }
160
161 // utility functions
162
163 /** Specialization of is_exactly_a<symmetry>(obj) for symmetry objects. */
164 template<> inline bool is_exactly_a<symmetry>(const basic & obj)
165 {
166         return obj.tinfo()==TINFO_symmetry;
167 }
168
169 } // namespace GiNaC
170
171 #endif // ndef __GINAC_SYMMETRY_H__