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 _mitk_OpenCVVideo_Source_h_
#define _mitk_OpenCVVideo_Source_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 // Header