vtkPlusOpenCVCaptureVideoSource.h

Go to the documentation of this file.

/*=Plus=header=begin======================================================
Program: Plus
Copyright (c) Laboratory for Percutaneous Surgery. All rights reserved.
See License.txt for details.
=========================================================Plus=header=end*/

#ifndef __vtkPlusOpenCVCaptureVideoSource_h
#define __vtkPlusOpenCVCaptureVideoSource_h

#include "vtkPlusDataCollectionExport.h"
#include "vtkPlusDevice.h"

// OpenCV includes
#include <opencv2/videoio.hpp>

namespace cv
{
  class VideoCapture;
  class Mat;
}

class vtkPlusDataCollectionExport vtkPlusOpenCVCaptureVideoSource : public vtkPlusDevice
{
public:
  static vtkPlusOpenCVCaptureVideoSource* New();
  vtkTypeMacro(vtkPlusOpenCVCaptureVideoSource, vtkPlusDevice);
  virtual void PrintSelf(ostream& os, vtkIndent indent) VTK_OVERRIDE;

  virtual PlusStatus ReadConfiguration(vtkXMLDataElement* config);
  virtual PlusStatus WriteConfiguration(vtkXMLDataElement* config);

  PlusStatus FreezeDevice(bool freeze);

  virtual bool IsTracker() const { return false; }

  virtual PlusStatus InternalUpdate();

  virtual PlusStatus NotifyConfigured();

  vtkGetStdStringMacro(VideoURL);
  vtkSetStdStringMacro(VideoURL);

  vtkGetMacro(DeviceIndex, int);
  vtkSetMacro(DeviceIndex, int);

  vtkGetMacro(AutofocusEnabled, bool);
  vtkSetMacro(AutofocusEnabled, bool);

  vtkGetMacro(AutoexposureEnabled, bool);
  vtkSetMacro(AutoexposureEnabled, bool);

  vtkGetMacro(FourCC, std::string);
  vtkSetMacro(FourCC, std::string);

  static cv::VideoCaptureAPIs CaptureAPIFromString(const std::string& apiString);
  static std::string StringFromCaptureAPI(cv::VideoCaptureAPIs api);

protected:
  vtkPlusOpenCVCaptureVideoSource();
  ~vtkPlusOpenCVCaptureVideoSource();

  virtual PlusStatus InternalConnect();
  virtual PlusStatus InternalDisconnect();

protected:
  std::string                       VideoURL;
  int                               DeviceIndex;
  std::shared_ptr<cv::VideoCapture> Capture;
  std::shared_ptr<cv::Mat>          Frame;
  std::shared_ptr<cv::Mat>          UndistortedFrame;
  cv::VideoCaptureAPIs              RequestedCaptureAPI;
  bool                              AutofocusEnabled;
  bool                              AutoexposureEnabled;
  std::string                       FourCC;

  FrameSizeType                     FrameSize;

  std::shared_ptr<cv::Mat>          CameraMatrix;
  std::shared_ptr<cv::Mat>          DistortionCoefficients;
};

#endif // __vtkPlusOpenCVCaptureVideoSource_h