24 #ifndef __FIREVISION_MODELS_COLOR_BAYES_BAYES_GENERATOR_
25 #define __FIREVISION_MODELS_COLOR_BAYES_BAYES_GENERATOR_
27 #include <fvutils/colormap/generator.h>
32 namespace firevision {
38 class BayesHistosToLut;
45 hint_t fg_object = H_UNKNOWN,
46 unsigned int lut_width = 256,
unsigned int lut_height = 256);
49 virtual void set_fg_object(hint_t
object);
50 virtual void set_buffer(
unsigned char *buffer,
51 unsigned int width,
unsigned int height);
53 virtual void consider();
57 virtual void reset_undo();
59 virtual void set_selection(std::vector< fawkes::rectangle_t > region);
61 virtual bool has_histograms();
62 virtual std::map< hint_t, Histogram * > * get_histograms();
64 virtual void load_histograms(
const char *filename);
65 virtual void save_histograms(
const char *filename);
67 void set_min_probability(
float min_prob);
70 bool is_in_region(
unsigned int x,
unsigned int y);
71 void normalize_histos();
73 typedef std::map< hint_t, Histogram * > HistogramMap;
74 HistogramMap fg_histos;
75 HistogramMap bg_histos;
77 HistogramMap::iterator histo_it;
84 unsigned int lut_width;
85 unsigned int lut_height;
86 unsigned int lut_depth;
88 unsigned int image_width;
89 unsigned int image_height;
91 unsigned int norm_size;
93 unsigned char *buffer;
94 std::vector< fawkes::rectangle_t > region;
95 std::vector< fawkes::rectangle_t >::iterator rit;