automatic commit
[folded-ctf.git] / pose.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   This the pose of a target. In the case of cats, we have a bounding
24   box, the head location and scale, and the body location. All are
25   given in the image coordinate system.
26
27  */
28
29 #ifndef POSE_H
30 #define POSE_H
31
32 #include "rgb_image.h"
33 #include "image.h"
34
35 class Pose {
36 public:
37   scalar_t _bounding_box_xmin, _bounding_box_ymin;
38   scalar_t _bounding_box_xmax, _bounding_box_ymax;
39   scalar_t _head_xc, _head_yc, _head_radius;
40   scalar_t _belly_xc, _belly_yc;
41
42   Pose();
43
44   bool hit(int level, Pose *pose);
45   bool collide(int level, Pose *pose);
46
47   void horizontal_flip(scalar_t scene_width);
48
49   void draw(int thickness, int r, int g, int b, int level, RGBImage *image);
50
51   void print(ostream *out);
52
53   void write(ostream *out);
54   void read(istream *in);
55 };
56
57 #endif