mitkOpenCVVideoSource.h

Go to the documentation of this file.

/*============================================================================
 
The Medical Imaging Interaction Toolkit (MITK)
 
Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.
 
Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.
 
============================================================================*/
 
#ifndef mitkOpenCVVideoSource_h
#define mitkOpenCVVideoSource_h
 
#include "mitkConfig.h"
 
#include "mitkVideoSource.h"
#include "mitkUndistortCameraImage.h"
 
// HighGui camera interface: a convenient way for grabbing from a video capture (on windows VfW is used)
#include <opencv2/highgui.hpp>
 
// For Providing ITK Image Interface
#include "itkRGBPixel.h"
#include "itkImage.h"
#include "itkImageRegionIterator.h"
#include "mitkOpenCVImageSource.h"
 
namespace mitk
{
  class MITKOPENCVVIDEOSUPPORT_EXPORT OpenCVVideoSource :
      virtual public VideoSource, virtual public OpenCVImageSource
  {
  public:
    typedef itk::RGBPixel< unsigned char >    CharPixelType;
    typedef itk::FixedArray <float,3>         HSVPixelType;
    typedef itk::Image< CharPixelType , 2 >   RGBPixelImageType;
    typedef itk::Image<HSVPixelType,2> HSVPixelImageType;
    typedef itk::ImageRegionIterator< RGBPixelImageType > RGBConstIteratorType;
    typedef itk::ImageRegionIterator< HSVPixelImageType > HSVConstIteratorType;
 
    mitkClassMacro( OpenCVVideoSource, VideoSource );
    itkFactorylessNewMacro(Self);
    itkCloneMacro(Self);
 
    virtual void SetVideoFileInput(const char * filename, bool repeatVideo, bool useCVCAMLib = false);
    virtual void SetVideoCameraInput(int cameraindex, bool useCVCAMLib = false);
    virtual double GetVideoCaptureProperty(int property_id);
    virtual int SetVideoCaptureProperty(int property_id, double value);
 
    virtual void GetCurrentFrameAsOpenCVImage(IplImage * image);
    cv::Mat GetImage() override;
    virtual const IplImage * GetCurrentFrame();
    virtual void GetCurrentFrameAsItkHSVPixelImage(HSVPixelImageType::Pointer &Image);
    void FetchFrame() override;
    unsigned char * GetVideoTexture() override;
    void StartCapturing() override;
    void StopCapturing() override;
    virtual IplImage* FlipImage(IplImage* input);
    virtual void EnableOnlineImageUndistortion(mitk::Point3D focal, mitk::Point3D principal, mitk::Point4D distortion);
    virtual void DisableOnlineImageUndistortion();
    virtual bool OnlineImageUndistortionEnabled() const;
 
    void PauseCapturing() override;
    itkGetConstMacro( VideoFileName, std::string );
 
    virtual void SetEnableXAxisFlip(bool enable);
    virtual void SetEnableYAxisFlip(bool enable);
 
    itkGetConstMacro( GrabbingDeviceNumber, short );
 
    itkGetMacro( RepeatVideo, bool );
    itkSetMacro( RepeatVideo, bool );
 
 
  protected:
    OpenCVVideoSource();
    ~OpenCVVideoSource() override;
 
    void Reset();
 
    void UpdateVideoTexture();
 
    // Helper functions
    void sleep(unsigned int ms);
    void RGBtoHSV(float r, float g, float b, float &h, float &s, float &v);
 
    // HighGUI Library capture device
    CvCapture    * m_VideoCapture;
 
    // current Video image
    IplImage     * m_CurrentImage;
    unsigned char* m_CurrentVideoTexture;
 
    IplImage     * m_PauseImage;
 
    std::string m_VideoFileName;
 
    short m_GrabbingDeviceNumber;
 
    // Repeat a video file
    bool m_RepeatVideo;
 
    // Switch between CVCAM Lib and HighGui Lib
    bool m_UseCVCAMLib;
 
    // On-the-fly undistortion of the captured video image
    bool m_UndistortImage;
    mitk::UndistortCameraImage::Pointer m_UndistortCameraImage;
 
    bool m_FlipXAxisEnabled;
 
    bool m_FlipYAxisEnabled;
  };
}
#endif