some minor improvements.

[stickletrack.git] / stickletrack.cpp

diff --git a/stickletrack.cpp b/stickletrack.cpp
index 0db07cee8d72762f4d85bd64bdcb75ba055befbd..85deb6d42de2cae5a1375ff0295aaf3bdd109fe1 100755 (executable)
--- a/stickletrack.cpp
+++ b/stickletrack.cpp
@@ -19,8 +19,6 @@
 #include "stickletrack.h"
 #include "tracking.h"
 
 #include "stickletrack.h"
 #include "tracking.h"
 

-#define BACKSECONDS 30
-

 double frametime = 0;
 int framenum = 0;
 
 double frametime = 0;
 int framenum = 0;
 


@@ -45,10 +43,17 @@ static vector<tag> **tagintime;
 static int tagintimeidx=0;
 static int moresleep=0;
 static int gotoframe=0;
 static int tagintimeidx=0;
 static int moresleep=0;
 static int gotoframe=0;

-static int maxoutputfps;
-static double rescalingfactor;

 static ofstream tanjaLogFile;
 
 static ofstream tanjaLogFile;
 

+struct comargs {
+  int backBufferSize;
+  char *videoFileName;
+  int maxOutputFPS;
+  double rescalingFactor;
+};
+
+static comargs comArgs;
+

 void genBaseDir() { 
 #if defined(_WIN32)
   Props.basedir = getenv("HOMEDRIVE");
 void genBaseDir() { 
 #if defined(_WIN32)
   Props.basedir = getenv("HOMEDRIVE");


@@ -62,6 +67,13 @@ void genBaseDir() {
   cout << "Using " << Props.basedir << " for data output." << endl;
 }
 
   cout << "Using " << Props.basedir << " for data output." << endl;
 }
 

+bool isWindowClosed (const char* name) {
+  if ( cvGetWindowHandle(name) == NULL )
+    return true;
+  else
+    return false;
+}
+

 void beforeExit() {
   cout << "Exitting ..." << endl;
 
 void beforeExit() {
   cout << "Exitting ..." << endl;
 


@@ -116,19 +128,19 @@ void drawTimes(Mat& mContours) {
   alles << framenum << " : " << fixed << (framenum)/(double)Props.fps << "s : +" << (double)moresleep/Props.fps << "ms";
 
   string text = alles.str();
   alles << framenum << " : " << fixed << (framenum)/(double)Props.fps << "s : +" << (double)moresleep/Props.fps << "ms";
 
   string text = alles.str();

-  int fontFace = FONT_HERSHEY_SCRIPT_SIMPLEX;
+  int fontFace = FONT_HERSHEY_DUPLEX;

 
 

-  double fontScale = Props.diagonal/1000.0;
-  int thickness = 3;
+  double fontScale = Props.width / 600.0;
+  int thickness = Props.width / 250.0;

 
   int baseline=0;
   Size textSize = getTextSize(text, fontFace,
                              fontScale, thickness, &baseline);
 
 
   int baseline=0;
   Size textSize = getTextSize(text, fontFace,
                              fontScale, thickness, &baseline);
 

-  Point textOrg(10, 10 + textSize.height);
+  Point textOrg(Props.width*0.01, Props.width*0.01 + textSize.height);

 
   rectangle(mContours, Point(0, 0),
 
   rectangle(mContours, Point(0, 0),

-           Point(Props.width, 20+textSize.height),
+           Point(Props.width, Props.width*0.02 + textSize.height),

            Scalar(0,0,255), -1);
 
   putText(mContours, text, textOrg, fontFace, fontScale,
            Scalar(0,0,255), -1);
 
   putText(mContours, text, textOrg, fontFace, fontScale,


@@ -194,7 +206,7 @@ int timeWarp(int x, int y) {
   int wannago = startClickFrame + 10*horiMove*Props.fps/Props.width;
 
   int maxgo = min(ilatetagintime, Props.totframes);
   int wannago = startClickFrame + 10*horiMove*Props.fps/Props.width;
 
   int maxgo = min(ilatetagintime, Props.totframes);

-  int mingo = max(1, ilatetagintime - BACKSECONDS*Props.fps + 2);
+  int mingo = max(1, ilatetagintime - comArgs.backBufferSize*Props.fps + 2);

 
   if (wannago > maxgo)
     return maxgo;
 
   if (wannago > maxgo)
     return maxgo;


@@ -206,7 +218,7 @@ int timeWarp(int x, int y) {
 
 int rotatingTime( int whereyouwant ) {
   if ( whereyouwant < 0 )
 
 int rotatingTime( int whereyouwant ) {
   if ( whereyouwant < 0 )

-    return BACKSECONDS*Props.fps + whereyouwant;
+    return comArgs.backBufferSize*Props.fps + whereyouwant;

   else
     return whereyouwant;
 }
   else
     return whereyouwant;
 }


@@ -305,7 +317,7 @@ void mouseTracking(int evt, int x, int y, int flags, void* param){
 void writeTanjaLog (int writeframenum, vector<tag> *tagstowrite) {
   stringstream outline;
 
 void writeTanjaLog (int writeframenum, vector<tag> *tagstowrite) {
   stringstream outline;
 

-  if ( writeframenum%(Props.fps/maxoutputfps) == 0 ) {
+  if ( writeframenum%(Props.fps/comArgs.maxOutputFPS) == 0 ) {

     if ( tagstowrite->size() >= Prefs.manyfish ) {
       outline << (double)writeframenum/Props.fps;
       for ( int itag = 0; itag < Prefs.manyfish; itag++) {
     if ( tagstowrite->size() >= Prefs.manyfish ) {
       outline << (double)writeframenum/Props.fps;
       for ( int itag = 0; itag < Prefs.manyfish; itag++) {


@@ -356,21 +368,30 @@ int process(VideoCapture& capture) {
   char filename[200];
   bool pleaseExit = false;
 
   char filename[200];
   bool pleaseExit = false;
 

-  namedWindow("original", CV_WINDOW_KEEPRATIO);
+  namedWindow("stickletrack_original", CV_WINDOW_KEEPRATIO);

     
   Mat frame, origframe, combinedmask;
 
   Props.fps = capture.get(CV_CAP_PROP_FPS);
     
   Mat frame, origframe, combinedmask;
 
   Props.fps = capture.get(CV_CAP_PROP_FPS);

-  Props.width = capture.get(CV_CAP_PROP_FRAME_WIDTH)*rescalingfactor;
-  Props.height = capture.get(CV_CAP_PROP_FRAME_HEIGHT)*rescalingfactor;
+  Props.width = capture.get(CV_CAP_PROP_FRAME_WIDTH)*comArgs.rescalingFactor;
+  Props.height = capture.get(CV_CAP_PROP_FRAME_HEIGHT)*comArgs.rescalingFactor;

   Props.totframes = capture.get(CV_CAP_PROP_FRAME_COUNT);
   Props.diagonal = sqrt( pow(Props.width, 2) + pow(Props.height, 2) );
 
   Props.totframes = capture.get(CV_CAP_PROP_FRAME_COUNT);
   Props.diagonal = sqrt( pow(Props.width, 2) + pow(Props.height, 2) );
 

-  Mat frameintime[BACKSECONDS*Props.fps];
+  if ( Props.width == 0 || Props.height == 0 || Props.fps == 0 || Props.totframes == 0 ) {
+    cerr << "Something got wrong while reading video-file info!" << endl;
+    cerr << "Width: " << Props.width << endl;
+    cerr << "Height: " << Props.height << endl;
+    cerr << "FPS: " << Props.fps << endl;
+    cerr << "Total frames: " << Props.totframes << endl;
+    exit(1);
+  }

 
 

-  tagintime = new vector<tag>*[BACKSECONDS*Props.fps];
+  Mat frameintime[comArgs.backBufferSize*Props.fps];

 
 

-  for (int iback = 0; iback < BACKSECONDS*Props.fps; iback++)
+  tagintime = new vector<tag>*[comArgs.backBufferSize*Props.fps];
+
+  for (int iback = 0; iback < comArgs.backBufferSize*Props.fps; iback++)

     tagintime[iback] = new vector<tag>;
 
   capture >> frame;
     tagintime[iback] = new vector<tag>;
 
   capture >> frame;


@@ -390,9 +411,7 @@ int process(VideoCapture& capture) {
   openTanjaLog();
 
   masking_init();
   openTanjaLog();
 
   masking_init();

-  tracking_init();
-
-  cvSetMouseCallback("contours_picture", mouseTracking, 0);
+  tracking_init(&mouseTracking);

 
   capture.set(CV_CAP_PROP_POS_FRAMES, 0);
 
 
   capture.set(CV_CAP_PROP_POS_FRAMES, 0);
 


@@ -421,7 +440,7 @@ int process(VideoCapture& capture) {
          break;
 
        frame.convertTo(origframe, CV_32FC3);
          break;
 
        frame.convertTo(origframe, CV_32FC3);

-       resize(origframe, origframe, Size(0,0), rescalingfactor, rescalingfactor);
+       resize(origframe, origframe, Size(0,0), comArgs.rescalingFactor, comArgs.rescalingFactor);

 
        masking_getCombinedMask(origframe, combinedmask);
       
 
        masking_getCombinedMask(origframe, combinedmask);
       


@@ -434,15 +453,16 @@ int process(VideoCapture& capture) {
        frameintime[tagintimeidx] = origframe;
        
        ilatetagintime = framenum;
        frameintime[tagintimeidx] = origframe;
        
        ilatetagintime = framenum;

-       tagintimeidx = (tagintimeidx+1)%(BACKSECONDS*Props.fps);
+       tagintimeidx = (tagintimeidx+1)%(comArgs.backBufferSize*Props.fps);

 
 

-       if ( ilatetagintime >= BACKSECONDS*Props.fps-1 )
-         writeTanjaLog( framenum - (BACKSECONDS*Props.fps-1), tagintime[ tagintimeidx ] );
+       if ( ilatetagintime >= comArgs.backBufferSize*Props.fps-1 )
+         writeTanjaLog( framenum - (comArgs.backBufferSize*Props.fps-1), tagintime[ tagintimeidx ] );

       }
 
       gotoframe = framenum + 1;
  
       }
 
       gotoframe = framenum + 1;
  

-      imshow("original", origframe/255.0);
+      if ( ! isWindowClosed("stickletrack_original") )
+       imshow("stickletrack_original", origframe/255.0);

     }
     
     drawTimes(tracking_getFrame());
     }
     
     drawTimes(tracking_getFrame());


@@ -451,7 +471,8 @@ int process(VideoCapture& capture) {
       circle( tracking_getFrame(), Point2f(tags[nearestTags[0]].x, tags[nearestTags[0]].y), Props.diagonal / 100.0, Scalar(0,0,255), -1, 8 );
     }
    
       circle( tracking_getFrame(), Point2f(tags[nearestTags[0]].x, tags[nearestTags[0]].y), Props.diagonal / 100.0, Scalar(0,0,255), -1, 8 );
     }
    

-    tracking_showFrame();
+    if ( tracking_showFrame() )
+      pleaseExit = true;

 
     char key;
 
 
     char key;
 


@@ -470,11 +491,11 @@ int process(VideoCapture& capture) {
 
   }
   
 
   }
   

-  if ( ilatetagintime >= BACKSECONDS*Props.fps-1 ) {
-    int timeidx = (tagintimeidx+1)%(BACKSECONDS*Props.fps);
-    for ( int itime = 0; itime < BACKSECONDS*Props.fps-1; itime++ ) {
-      writeTanjaLog( framenum+1+itime - (BACKSECONDS*Props.fps-1), tagintime[ timeidx ] );
-      timeidx = (timeidx+1)%(BACKSECONDS*Props.fps);
+  if ( ilatetagintime >= comArgs.backBufferSize*Props.fps-1 ) {
+    int timeidx = (tagintimeidx+1)%(comArgs.backBufferSize*Props.fps);
+    for ( int itime = 0; itime < comArgs.backBufferSize*Props.fps-1; itime++ ) {
+      writeTanjaLog( framenum+1+itime - (comArgs.backBufferSize*Props.fps-1), tagintime[ timeidx ] );
+      timeidx = (timeidx+1)%(comArgs.backBufferSize*Props.fps);

     }
   }
   else {
     }
   }
   else {


@@ -491,25 +512,53 @@ int process(VideoCapture& capture) {
   return 0;
 }
 
   return 0;
 }
 

-int main(int ac, char** av) {
-  if ( ac != 4 ) {
-    cout << "Usage: stickletrack <maxoutputfps> <rescalingfactor> <videofilename>" << endl;
+void showUsage() {
+  cout << "Usage: stickletrack [OPTIONS]... <path to videofile>" << endl;
+  cout << endl << "Available options:" << endl;
+  cout << "-x <rescalingfactor>" << "\t" << "rescale video by a factor <rescalingfactor>" << endl;
+  cout << "-o <maxfps>" << "\t" << "write to log maximally <maxfps> times per second" << endl;
+  cout << "-b <bsize>" << "\t" << "buffer <bsize> seconds for going back in time" << endl;
+}
+
+bool parseComArgs (int ac, char** av) {
+  if ( ac < 2 || ac%2 ) {
+    showUsage();

     exit(0);
   }
 
     exit(0);
   }
 

-  maxoutputfps = atoi( av[1] );
-  rescalingfactor = atof( av[2] );
+  comArgs.videoFileName = NULL;
+  comArgs.rescalingFactor = 1.0;
+  comArgs.maxOutputFPS = 25;
+  comArgs.backBufferSize = 10;
+
+  for (int i = 1; i < ac-1; i++)
+    if (i + 1 != ac) 
+      if (av[i] == "-x") {
+       comArgs.rescalingFactor = atof(av[i + 1]);
+      } else if (av[i] == "-o") {
+       comArgs.maxOutputFPS = atoi(av[i + 1]);
+      } else if (av[i] == "-b") {
+       comArgs.backBufferSize = atoi(av[i + 1]);
+      } else {
+       cout << "Unknown command line parameters!\n";
+       showUsage();
+       exit(0);
+      }
+
+  comArgs.videoFileName = av[ac-1];
+}

 
 

-  capture.open( av[3] ); 
+int main(int ac, char** av) {
+  parseComArgs(ac, av);
+
+  capture.open( comArgs.videoFileName );

 
   genBaseDir();
   
 
   genBaseDir();
   

-   if (!capture.isOpened()) {
+  if (!capture.isOpened()) {

     cerr << "Failed to open a video device or video file!\n" << endl;
     return 1;
   }
     cerr << "Failed to open a video device or video file!\n" << endl;
     return 1;
   }

-
-   cout << "Exit with q if you want to save your data!!!" << endl;

   
   return process(capture);
 }
   
   return process(capture);
 }