basicOpticalFlow.h

Go to the documentation of this file.

// ----------------------------------------------------------------------------
//
// $Id$
//
// Copyright 2008, 2009, 2010, 2011, 2012  Antonio Franchi and Paolo Stegagno    
//
// This file is part of MIP.
//
// MIP is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// MIP is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with MIP. If not, see <http://www.gnu.org/licenses/>.
//
// Contact info: antonio.franchi@tuebingen.mpg.de stegagno@diag.uniroma1.it
//
// ----------------------------------------------------------------------------

#ifndef BASIC_OPTICAL_FLOW_H
#define BASIC_OPTICAL_FLOW_H

#include <SpacesArma.h>
#include <opencv2/opencv.hpp>
#include <opencv2/highgui/highgui.hpp>

namespace MipAlgorithms{


 /* @{ */

 class basicOpticalFlowPar {
  public:
   bool visualize;
   bool dense;
   Decimal pyrScale; // used to set pyramid scaling factor (default = 0.5)
   Decimal pyrLevels; // used to set pyramid levels (default = 3)
   Decimal polySigma; // Standard deviation to smooth derivatives (linked to polyN, default = 1.1)
   int iterations;  // number of iterations (default = 5)
   int polyN;  // Size of the pixel neighborhood (typical 5 or 7, default = 5)
   int winSize;   // Size of search window
   int maxCorners;
   Decimal qualityLevel;
   Decimal minDistance;
   
   basicOpticalFlowPar();
 };
 
 class basicOpticalFlow {
  private:
   basicOpticalFlowPar* _par;
   cv::Mat prevImage, grayim, cflow, currFlow;
   
   std::vector<cv::Point2f> prevFeatures;
   std::vector<cv::Point2f> currFeatures;
   std::vector<uchar>  status;
   std::vector<float> errors;
   int featuresNumber;
   
   void drawOptFlowMap(const cv::Mat& flow, cv::Mat& cflowmap, int step, double, const cv::Scalar& color);
   
  public:
   basicOpticalFlow(cv::Mat & initImage);
   ~basicOpticalFlow();
   
   void visualizeOutput();
   
   void getFlow(cv::Mat & newImage, cv::Mat & flow);
 };

 /* @} */
};// end namespace MipAlgorithms


#endif /* __LUCAKANADE_H */