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. //
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. //
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/>. //
15 // Written by Francois Fleuret, (C) IDIAP //
16 // Contact <francois.fleuret@idiap.ch> for comments & bug reports //
17 ///////////////////////////////////////////////////////////////////////////
21 This class factorizes the information from a PoseCell needed by a
22 PiFeature to be evaluated on an image. Since there can be in there
23 costly operations such as trigonometric mappings, it provides a
24 substantial cost optimization.
28 #ifndef PI_REFERENTIAL_H
29 #define PI_REFERENTIAL_H
31 #include "rectangle.h"
32 #include "pose_cell.h"
33 #include "rgb_image.h"
36 // The best scale so that the head size is
37 // ~global.reference_head_size. This is an integer scale as defined
41 scalar_t _horizontal_polarity;
43 // The head frame in _common_scale. The vectors are of length the _RADIUS_ of the head
44 scalar_t _head_xc, _head_yc, _head_radius;
45 scalar_t _head_window_scaling;
46 scalar_t _head_ux, _head_uy, _head_vx, _head_vy;
47 scalar_t _head_ux_nopolarity, _head_uy_nopolarity, _head_vx_nopolarity, _head_vy_nopolarity;
49 // The body frame in that _common_scale. The vectors are of length the radii of the ellipse
50 scalar_t _body_xc, _body_yc;
51 scalar_t _body_radius_1, _body_radius_2, _body_tilt;
52 scalar_t _body_ux, _body_uy, _body_vx, _body_vy;
53 // int _body_edge_shift;
55 // The belly frame is defined by the body location and head scale
56 scalar_t _belly_xc, _belly_yc;
57 scalar_t _belly_ux, _belly_uy, _belly_vx, _belly_vy;
58 scalar_t _belly_ux_nopolarity, _belly_uy_nopolarity, _belly_vx_nopolarity, _belly_vy_nopolarity;
59 scalar_t _belly_window_scaling;
61 // The head-belly frame is defined by the head location and the body
63 scalar_t _head_belly_xc, _head_belly_yc;
64 scalar_t _head_belly_ux, _head_belly_uy, _head_belly_vx, _head_belly_vy;
65 int _head_belly_edge_shift;
67 void draw_frame(RGBImage *image,
68 int registration_mode,
75 PiReferential(PoseCell *cell);
78 // A frame centered on the head, of size four times the head radius
79 // and, flipped verically if the body center is on the left of the
83 // Same as above, without the flipping
85 // A frame centered on the body center, of size size times the
86 // head rardius, flipped vertically if the body center is on the
87 // left of the head center
90 // Same as above, without the flipping
93 // A frame centered on the middle point between the head center
94 // and the body center, of size twice the distance head center -
95 // body center in the head-body direction, and of four times the
96 // head radius in the other
99 // Same as above with rotation of the edges
105 // The rectangle coordinates are in the reference frames. For the
106 // head for instance , [-1,1] x [-1,1] corresponds to the head
109 void register_rectangle(int registration_mode,
113 int register_edge(int registration_type, int edge_type);
115 void draw(RGBImage *image, int level);
117 void draw_window(RGBImage *image,
118 int registration_mode, Rectangle *window,
121 void draw_edge_and_scale(RGBImage *image,
122 int registration_mode, Rectangle *window,
123 int _tag, int _edge_scale);
125 static void print_registration_mode(ostream *out, int registration_mode);