opencv/apps/traincascade/_imagestorage.h

   1 #ifndef _IMAGESTORAGE_H\r
   2 #define _IMAGESTORAGE_H\r
   3 \r
   4 #include <highgui.h>\r
   5 \r
   6 //------------------- Background reading ---------------------\r
   7 struct CvBackgroundReader
   8 {
   9     CvBackgroundReader();
  10     virtual ~CvBackgroundReader();
  11
  12     CvMat   src;
  13     CvMat   img;
  14     CvPoint offset;
  15     float   scale;
  16     float   scaleFactor;
  17     float   stepFactor;
  18     CvPoint point;
  19 };\r
  20 \r
  21 struct CvBackgroundData
  22 {
  23     CvBackgroundData();
  24     CvBackgroundData( const char* fileName, CvSize _winSize );
  25     virtual ~CvBackgroundData();
  26     bool getImage( CvMat* img, bool reset );
  27     bool getNext( bool reset);
  28
  29     CvBackgroundReader* bgReader;
  30     int    count;
  31     char** fileName;
  32     int    last;
  33     int    round;
  34     CvSize winSize;
  35 };
  36 \r
  37 //--------------------- VecFile reading ------------------------\r
  38 struct CvVecFile
  39 {
  40     CvVecFile();
  41     CvVecFile( const char* _vecFileName );
  42     virtual ~CvVecFile();
  43
  44     FILE*  input;
  45     int    count;
  46     int    vecSize;
  47     int    last;
  48     short* vector;
  49     int base;
  50 };\r
  51 \r
  52 //-----------------------  CvImageReader  -----------------------\r
  53 class CvImageReader\r
  54 {\r
  55 public:\r
  56     CvImageReader( const char* _vecFileName, const char* _bgfileName, CvSize _winSize );\r
  57     virtual ~CvImageReader();\r
  58 \r
  59     bool getNegImage(CvMat* img, bool reset = false);\r
  60     bool getPosImage(CvMat* img, bool reset = false);\r
  61 private:\r
  62     CvBackgroundData* bgData;\r
  63     CvVecFile* vecFile;\r
  64 };\r
  65 \r
  66 #endif