artvertiser/calibmodel.h

   1 #ifndef CALIBMODEL_H
   2 #define CALIBMODEL_H
   3
   4 #include <cv.h>
   5 #include <highgui.h>
   6 #include <garfeild.h>
   7
   8 class CalibModel {
   9 public:
  10         CvPoint corners[4];
  11         CvPoint artvert_corners[4];
  12
  13         LightMap map;
  14         CamAugmentation augm;
  15
  16         CalibModel();
  17         ~CalibModel();
  18
  19         void useModelFile( const char *modelfile );
  20
  21         bool buildCached(int nbcam, CvCapture *capture, bool cache, planar_object_recognizer &detector, bool dont_try_to_train );
  22
  23     int getImageWidth() { return image_width; }
  24     int getImageHeight() { return image_height; }
  25 private:
  26         IplImage *image;
  27
  28         int image_width, image_height;
  29
  30         const char *win;
  31         const char *modelfile;
  32
  33         enum State { TAKE_SHOT, CORNERS, ARTVERT_CORNERS };
  34         State state;
  35         int grab;
  36         static void onMouseStatic(int event, int x, int y, int flags, void* param);
  37         void onMouse(int event, int x, int y, int flags);
  38         bool interactiveSetup(CvCapture *capture);
  39 };
  40
  41
  42 #endif