automatic commit
[folded-ctf.git] / pose_cell_set.h
1
2 ///////////////////////////////////////////////////////////////////////////
3 // This program is free software: you can redistribute it and/or modify  //
4 // it under the terms of the version 3 of the GNU General Public License //
5 // as published by the Free Software Foundation.                         //
6 //                                                                       //
7 // This program is distributed in the hope that it will be useful, but   //
8 // WITHOUT ANY WARRANTY; without even the implied warranty of            //
9 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU      //
10 // General Public License for more details.                              //
11 //                                                                       //
12 // You should have received a copy of the GNU General Public License     //
13 // along with this program. If not, see <http://www.gnu.org/licenses/>.  //
14 //                                                                       //
15 // Written by Francois Fleuret                                           //
16 // (C) Idiap Research Institute                                          //
17 //                                                                       //
18 // Contact <francois.fleuret@idiap.ch> for comments & bug reports        //
19 ///////////////////////////////////////////////////////////////////////////
20
21 /*
22
23   A set of PoseCell.
24
25  */
26
27 #ifndef POSE_CELL_SET_H
28 #define POSE_CELL_SET_H
29
30 #include "pose_cell.h"
31 #include "labelled_image.h"
32
33 class PoseCellSet {
34 protected:
35   // This is a bit ugly. Notice that all the code is written in such a
36   // way that we never have more than one such PoseCellSet in memory
37   // at any moment
38   static const int max_nb_cells = 500000;
39
40   int _max_nb;
41   int _nb_added;
42   PoseCell *_cells;
43
44 public:
45
46   PoseCellSet();
47   ~PoseCellSet();
48
49   inline int nb_cells() { return _nb_added; }
50
51   inline PoseCell *get_cell(int k) {
52     ASSERT(k >= 0 && k < _nb_added);
53     return _cells + k;
54   }
55
56   PoseCell *get_containing_cell(Pose *p);
57
58   void erase_content();
59   void add_cell(PoseCell *cell);
60 };
61
62 #endif