mitkPickPointsOnStereoVideo.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 mitkPickPointsOnStereoVideo_h
#define mitkPickPointsOnStereoVideo_h

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

namespace mitk {

class NIFTKOPENCV_EXPORT PickPointsOnStereoVideo : public itk::Object
{

public:

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

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

  itkSetMacro ( TrackerIndex, int);
  itkSetMacro ( ReferenceIndex, int);
  itkSetMacro ( AllowableTimingError, long long);
  itkSetMacro ( OrderedPoints, bool);
  itkSetMacro ( PickingLine, bool);
  itkSetMacro ( AskOverWrite, bool);
  itkSetMacro ( HaltOnVideoReadFail, bool);
  itkSetMacro ( WriteAnnotatedImages, bool);
  itkSetMacro ( HalfImageWidth, bool);
  itkSetMacro ( Frequency, unsigned int);

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

  PickPointsOnStereoVideo();
  virtual ~PickPointsOnStereoVideo();

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

private:
  std::string                   m_VideoIn; //the video in file
  std::string                   m_Directory; //the directory containing the data
  std::string                   m_OutDirectory; //the directory to save to

  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_OrderedPoints; //picked points can be ordered or unordered
  bool                          m_PickingLine; //if true we are picking a line defined by a vector of points
  bool                          m_AskOverWrite; //if true, we will ask if you want to overwrite existing results
  bool                          m_HaltOnVideoReadFail; //halt if video read fail
  bool                          m_WriteAnnotatedImages; //halt if video read fail
  bool                          m_HalfImageWidth; //halves the image width used for point picking, useful for small screens

  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