mitkMakeMaskImagesFromStereoVideo.h

Go to the documentation of this file.

/*=============================================================================

  NifTK: A software platform for medical image computing.

  Copyright (c) University College London (UCL). All rights reserved.

  This software is distributed WITHOUT ANY WARRANTY; without even
  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  PURPOSE.

  See LICENSE.txt in the top level directory for details.

=============================================================================*/

#ifndef mitkMakeMaskImagesFromStereoVideo_h
#define mitkMakeMaskImagesFromStereoVideo_h

#include "niftkOpenCVExports.h"
#include <string>
#include <itkObject.h>
#include <itkObjectFactory.h>
#include <mitkCommon.h>
#include <cv.h>
#include <highgui.h>
#include "mitkVideoTrackerMatching.h"
#include <set>

namespace mitk {

class NIFTKOPENCV_EXPORT MakeMaskImagesFromStereoVideo : public itk::Object
{

public:

  mitkClassMacroItkParent(MakeMaskImagesFromStereoVideo, itk::Object)
  itkNewMacro(MakeMaskImagesFromStereoVideo)

  
  void Initialise (std::string directory);
  void  Project(mitk::VideoTrackerMatching::Pointer matcher);
  
  void  SetMatcherCameraToTracker(mitk::VideoTrackerMatching::Pointer matcher);

  itkSetStringMacro ( TimeStampsFile);
  itkSetMacro ( TrackerIndex, int);
  itkSetMacro ( ReferenceIndex, int);
  itkSetMacro ( AllowableTimingError, long long);
  itkSetMacro ( AskOverWrite, bool);
  itkSetMacro ( HaltOnVideoReadFail, bool);
  itkSetMacro ( Frequency, unsigned int);

  itkGetMacro ( InitOK, bool);
  itkGetMacro ( ProjectOK, bool);
  itkGetMacro ( WorldToLeftCameraMatrices, std::vector < cv::Mat > );
 
protected:

  MakeMaskImagesFromStereoVideo();
  virtual ~MakeMaskImagesFromStereoVideo();

  MakeMaskImagesFromStereoVideo(const MakeMaskImagesFromStereoVideo&); // Purposefully not implemented.
  MakeMaskImagesFromStereoVideo& operator=(const MakeMaskImagesFromStereoVideo&); // Purposefully not implemented.

private:
  std::string                   m_VideoIn; //the video in file
  std::string                   m_Directory; //the directory containing the data
  std::string                   m_TimeStampsFile;
  std::set<unsigned long long>  m_TimeStampsSet;
  int                           m_TrackerIndex; //the tracker index to use for frame matching
  int                           m_ReferenceIndex; //the reference index to use for frame matching, not used by default
 
  bool                          m_InitOK;
  bool                          m_ProjectOK;
  bool                          m_AskOverWrite; //if true, we will ask if you want to overwrite existing results
  bool                          m_HaltOnVideoReadFail; //halt if video read fail

  unsigned int                  m_StartFrame; //you can exclude some frames at the start
  unsigned int                  m_EndFrame; // and at the end
  unsigned int                  m_Frequency; // the sample rate (process every m_Frequency frame)

  //the dimensions of the video screen in pixels
  double   m_VideoWidth;
  double   m_VideoHeight;

  std::vector < cv::Mat >       m_WorldToLeftCameraMatrices;    // the saved camera positions

  cv::VideoCapture*             m_Capture;
  
  long long                     m_AllowableTimingError; // the maximum permisable timing error when setting points or calculating projection errors;
  
}; // end class

} // end namespace

#endif