gradient.h
00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023 #ifndef __FIREVISION_CLASSIFIERS_GRADIENT_H_
00024 #define __FIREVISION_CLASSIFIERS_GRADIENT_H_
00025
00026 #include <classifiers/classifier.h>
00027 #include <classifiers/qualifiers.h>
00028
00029 #include <models/scanlines/grid.h>
00030
00031 namespace firevision {
00032 #if 0
00033 }
00034 #endif
00035
00036 class GradientClassifier: public Classifier
00037 {
00038 public:
00039 GradientClassifier(std::list<ScanlineGrid* >* scanlines, Qualifier* q,
00040 unsigned int threshold, unsigned int max_size = 0,
00041 bool use_rising_flank = true,
00042 bool use_falling_flank = true);
00043 virtual ~GradientClassifier();
00044
00045 virtual std::list< ROI > * classify();
00046 virtual void set_src_buffer(unsigned char *yuv422_planar,
00047 unsigned int width, unsigned int height);
00048
00049 virtual void set_threshold(unsigned int threshold, unsigned int max_size = 0);
00050 virtual void set_edges(bool use_rising_edge, bool use_falling_edge);
00051
00052 private:
00053 int _last_val;
00054 fawkes::point_t _last_pos;
00055
00056 unsigned int _threshold;
00057 unsigned int _max_size;
00058
00059 std::list<ScanlineGrid* >* _scanlines;
00060 Qualifier* _q;
00061
00062 bool _use_falling_edge;
00063 bool _use_rising_edge;
00064 };
00065
00066 }
00067
00068 #endif // __FIREVISION_CLASSIFIERS_GRADIENT_H_