mitkUSImageVideoSource.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 mitkUSImageVideoSource_h
#define mitkUSImageVideoSource_h
 
// ITK
#include <itkProcessObject.h>
 
// MITK
#include "mitkUSImageSource.h"
#include "mitkConvertGrayscaleOpenCVImageFilter.h"
#include "mitkCropOpenCVImageFilter.h"
#include "mitkBasicCombinationOpenCVImageFilter.h"
 
// OpenCV
#include <opencv2/videoio.hpp>
 
namespace mitk {
  class MITKUS_EXPORT USImageVideoSource : public mitk::USImageSource
  {
  public:
    mitkClassMacroItkParent(USImageVideoSource, itk::ProcessObject);
    itkFactorylessNewMacro(Self);
    itkCloneMacro(Self);
 
    struct USImageRoi
    {
      int topLeftX;
      int topLeftY;
      int bottomRightX;
      int bottomRightY;
 
      USImageRoi()
        : topLeftX(0), topLeftY(0), bottomRightX(0), bottomRightY(0) { };
      USImageRoi(unsigned int topLeftX, unsigned int topLeftY, unsigned int bottomRightX, unsigned int bottomRightY)
        : topLeftX(topLeftX), topLeftY(topLeftY), bottomRightX(bottomRightX), bottomRightY(bottomRightY) { };
    };
 
    struct USImageCropping
    {
      unsigned int top;
      unsigned int bottom;
      unsigned int left;
      unsigned int right;
 
      USImageCropping()
        : top(0), bottom(0), left(0), right(0) { };
      USImageCropping(unsigned int top, unsigned int bottom, unsigned int left, unsigned int right)
        : top(top), bottom(bottom), left(left), right(right) { };
    };
 
    void SetVideoFileInput(std::string path);
 
    void SetCameraInput(int deviceID);
 
    void ReleaseInput();
 
    void SetColorOutput(bool isColor);
 
    void SetRegionOfInterest(int topLeftX, int topLeftY, int bottomRightX, int bottomRightY);
 
    void SetRegionOfInterest(USImageRoi regionOfInterest);
 
    void SetCropping(USImageCropping cropping);
 
    void RemoveRegionOfInterest();
 
    void OverrideResolution(int width, int height);
 
    // Getter & Setter
    itkGetMacro(IsVideoReady, bool);
    itkGetMacro(ResolutionOverride, bool);
    itkSetMacro(ResolutionOverride, bool);
    itkGetMacro(IsGreyscale,bool);
    itkGetMacro(ResolutionOverrideWidth,int);
    itkGetMacro(ResolutionOverrideHeight,int);
    int GetImageHeight();
    int GetImageWidth();
    USImageCropping GetCropping();
    USImageRoi GetRegionOfInterest();
 
    bool GetIsReady();
 
  protected:
    USImageVideoSource();
    ~USImageVideoSource() override;
 
    void GetNextRawImage( std::vector<cv::Mat>& image ) override;
 
    void GetNextRawImage( std::vector<mitk::Image::Pointer>& image ) override;
 
    cv::VideoCapture* m_VideoCapture;
 
    bool m_IsVideoReady;
 
    bool m_IsGreyscale;
 
    bool m_IsCropped;
 
    int  m_ResolutionOverrideWidth;
    int  m_ResolutionOverrideHeight;
    bool m_ResolutionOverride;
 
    ConvertGrayscaleOpenCVImageFilter::Pointer  m_GrayscaleFilter;
    CropOpenCVImageFilter::Pointer              m_CropFilter;
  };
} // namespace mitk
#endif