Add simplex_order::get_complex()
[libstick.git] / include / libstick-0.1 / simplicialcomplex.h
1 #ifndef simplicialcomplex_h_nealaezeojeeChuh
2 #define simplicialcomplex_h_nealaezeojeeChuh
3
4 #include <stdint.h>
5 #include <cstdlib>
6 #include <cstring>
7 #include <algorithm>
8 #include <vector>
9 #include <limits>
10
11 #include <iostream>
12
13 #include "booleanmatrix.h"
14
15
16 namespace libstick {
17
18 /** A simplicial complex is a std::vector of simplices such that each face is
19 * also part of the complex. Every simplex has dimension at most MAXDIM. The
20 * indices of simplices resp. their faces are of type IT. To each simplex a
21 * value is assigend, which is of type VT. When a simplicial_complex is
22 * instantiated, a single (-1) dimensional simplex is automatically created.
23 * Each 0-dimensional simplex automatically has this simplex as its face.
24 * Consequently, the innner class simplex_order gives the extended boundary
25 * matrix. */
26 template<int MAXDIM, class IT, class VT>
27 class simplicial_complex {
28
29 public:
30 /** The type of this class. */
31 typedef simplicial_complex<MAXDIM, IT, VT> simplcompltype;
32 /** Type of indices of simplices. */
33 typedef IT index_type;
34 /** To every simplex a function value is assigned according to which a
35 * filtration is considered. This is the value type of the function. */
36 typedef VT value_type;
37
38 /** A simplex of the complex. */
39 struct simplex {
40 /** Dimension of the simplex. */
41 int dim;
42 /** The indices of the faces of the simplex. */
43 index_type faces[MAXDIM+1];
44 /** The value of the simplex. */
45 value_type value;
46
47 /** Create a new simplex with dimension 'dim', (dim+1)-faces and
48 * its value. If simpley is 0-dimensional, its face is
49 * automatically set to one (-1)-dimensional simplex. */
50 static simplex create(int dim, index_type* faces, value_type value) {
51 assert(0 <= dim && dim <= MAXDIM);
52
53 simplex s;
54 s.dim = dim;
55 s.value = value;
56
57 if (dim > 0)
58 memcpy(s.faces, faces, face_count_bydim(dim)*sizeof(index_type));
59 else
60 s.faces[0] = 0;
61
62 return s;
63 }
64
65 /** Create a (-1)-dimensional simplex. It has the lowest possible value. */
66 static simplex create_minusonedim_simplex() {
67 simplex s;
68
69 s.dim = -1;
70 s.faces[0] = 0;
71 s.value = std::numeric_limits<VT>::has_infinity
72 ? -std::numeric_limits<VT>::infinity()
73 : std::numeric_limits<VT>::min();
74
75 return s;
76 }
77
78 /** Get number of faces. */
79 size_t face_count() const {
80 return face_count_bydim(dim);
81 }
82
83 /** Get number of faces of a dim-dimensional simplex. */
84 static size_t face_count_bydim(int dim) {
85 assert(-1 <= dim && dim <= MAXDIM);
86 return dim + 1;
87 }
88 };
89
90 /** An order of the simplices of complex c. An order can be interpreted
91 * as a permuation of the complex's std::vector of simplices. */
92 class simplex_order {
93
94 public:
95 typedef boolean_colmatrix<IT> boundary_matrix;
96
97 /** Create a standard order of the complex c, i.e., the identity permutation. */
98 simplex_order(const simplcompltype &c) :
99 c(c)
100 {
101 reset();
102 }
103
104 /** Reset order to the identity permutation of the complex's simplices. */
105 void reset() {
106 order.clear();
107 for (unsigned i=0; i < c.size(); ++i)
108 order.push_back(i);
109 revorder = order;
110 }
111
112 /** Return number of simplices. */
113 size_t size() const {
114 assert(order.size() == revorder.size());
115 return order.size();
116 }
117
118 /** Get i-th simplex in the simplex order. */
119 const simplex& get_simplex(index_type i) const {
120 assert(i < size());
121 return c.simplices[order.at(i)];
122 }
123
124 const simplcompltype& get_complex() const {
125 return c;
126 }
127
128 /** Returns true iff the faces of simplex i are before i in this order. */
129 bool is_filtration() const {
130 assert(size() == c.size());
131
132 for (unsigned i=0; i < size(); ++i)
133 for (unsigned f=0; f < get_simplex(i).face_count(); ++f)
134 if (revorder[get_simplex(i).faces[f]] >= i)
135 return false;
136
137 return true;
138 }
139
140 /** Returns true iff is_filtration() gives true and values of simplices
141 * are monotone w.r.t. this order of simplices. */
142 bool is_monotone() const {
143 assert(size() == c.size());
144
145 for (unsigned i=1; i < size(); ++i)
146 if (get_simplex(i-1).value > get_simplex(i).value)
147 return false;
148
149 return is_filtration();
150 }
151
152 /** Sort simplices such that is_monotone() gives true. This
153 * requires that the complex's is_monotone() gave true
154 * beforehand.*/
155 void make_monotone_filtration() {
156 assert(c.is_monotone());
157
158 sort(order.begin(), order.end(), cmp_monotone_filtration(c));
159 restore_revorder_from_order();
160
161 assert(c.is_monotone());
162 assert(is_filtration());
163 assert(is_monotone());
164 }
165
166 /** Get the boundary matrix of the complex according to this order. */
167 boundary_matrix get_boundary_matrix() const {
168 boundary_matrix mat(size());
169
170 for (unsigned c=0; c < size(); ++c)
171 for(unsigned r=0; r < get_simplex(c).face_count(); ++r)
172 mat.set(revorder[get_simplex(c).faces[r]], c, true);
173
174 return mat;
175 }
176
177 private:
178 /** Reconstruct 'revorder' by inverting the permutation given by 'order'. */
179 void restore_revorder_from_order() {
180 // Make revorder * order the identity permutation
181 for (unsigned i=0; i < size(); ++i)
182 revorder[order[i]] = i;
183 }
184
185 /** The complex of which we consider a simplex order. */
186 const simplcompltype &c;
187
188 /** The i-th simplex in order is the order[i]-th simplex of the
189 * complex. 'order' can be seen as a permutation of the
190 * simplices saved in 'c'. */
191 std::vector<index_type> order;
192
193 /** The i-th simplex in the complex is the revorder[i]-th
194 * simplex in order. 'revorder' can be seen as the inverse
195 * permutation saved in 'order'. */
196 std::vector<index_type> revorder;
197 };
198
199 public:
200 simplicial_complex() {
201 // Add the one minus-one dimensional simplex
202 add_simplex(simplex::create_minusonedim_simplex());
203 }
204
205 /** Remove all simplices except the dummy simplex */
206 void clear() {
207 simplices.resize(1);
208 }
209
210 /** Return number of simplices. */
211 size_t size() const {
212 return simplices.size();
213 }
214
215 /** Add a simplex to the complex. The dimension of the faces must be
216 * dim-1, and they must already be part of the complex. Returns the
217 * index of the added simplex. */
218 index_type add_simplex(int dim, index_type* faces, value_type value) {
219 return add_simplex(simplex::create(dim, faces, value));
220 }
221
222 /** Add a simplex to the complex of at least dimension 1. The dimension
223 * of the faces must be dim-1, and they must already be part of the
224 * complex. The value of the simplex is set to the maximum value of its
225 * faces. Returns the index of the added simplex. */
226 index_type add_simplex(int dim, index_type* faces) {
227 assert(dim >= 1);
228
229 // Get max value of its faces
230 VT value = simplices[faces[0]].value;
231 for (size_t i=0; i < simplex::face_count_bydim(dim); ++i)
232 value = std::max(value, simplices[faces[i]].value);
233
234 return add_simplex(dim, faces, value);
235 }
236
237 /** Add a simplex to the complex. The dimension of the faces must be
238 * dim-1, and they must already be part of the complex. Returns the
239 * index of the added simplex. */
240 index_type add_simplex(simplex s) {
241 // Check requirements for faces
242 for (unsigned i=0; i < s.face_count(); ++i) {
243 // Faces are already in complex.
244 assert(s.faces[i] < size());
245 // Faces have dimension dim-1
246 assert(simplices[s.faces[i]].dim == s.dim-1);
247 }
248
249 // index_type must be large enough
250 assert(simplices.size() < std::numeric_limits<IT>::max());
251
252 index_type idx = simplices.size();
253 simplices.push_back(s);
254 return idx;
255 }
256
257 /** Add an array of simplices */
258 void add_simplices(simplex* sarray, size_t count) {
259 for (unsigned i=0; i < count; ++i)
260 add_simplex(sarray[i]);
261 }
262
263 /** Return true iff for each simplex i with dimension dim it holds that
264 * the faces of i are contained in the complex and have dimension dim-1. */
265 bool is_complex() const {
266 for (unsigned i=0; i < size(); ++i) {
267
268 const simplex &s = simplices[i];
269 for (unsigned f=0; f < s.face_count(); ++f) {
270
271 if (s.faces[f] >= size())
272 return false;
273
274 const simplex &face = simplices[s.faces[f]];
275 if (face.dim != s.dim-1)
276 return false;
277 }
278 }
279 return true;
280 }
281
282 /** Returns true iff simplex's values are monotone w.r.t.
283 * face-inclusion, i.e., for each simplex its value is not smaller than
284 * the values of its faces. Requires that is_complex() gives true. */
285 bool is_monotone() const {
286 assert(is_complex());
287
288 typename std::vector<simplex>::const_iterator it = ++simplices.begin();
289 for (; it != simplices.end(); ++it)
290 for (unsigned f=0; f < it->face_count(); ++f)
291 if (simplices[it->faces[f]].value > it->value)
292 return false;
293
294 return true;
295 }
296
297 private:
298 /** Compares (operator<) two simplices (i.e. indices) in a
299 * simplex_order w.r.t. lexicographical order on (value,
300 * dimension)-tuples. */
301 struct cmp_monotone_filtration {
302 const simplicial_complex &c;
303
304 cmp_monotone_filtration(const simplicial_complex &c) :
305 c(c){
306 }
307
308 bool operator()(index_type i, index_type j) {
309 const simplex& si = c.simplices[i];
310 const simplex& sj = c.simplices[j];
311
312 if (si.value < sj.value)
313 return true;
314 else if (si.value == sj.value)
315 return si.dim < sj.dim;
316 else
317 return false;
318 }
319 };
320
321 public:
322 /** A list of simplices */
323 std::vector<simplex> simplices;
324 };
325
326 }
327
328
329 #endif