mitkUSImageVideoSource.h

Go to the documentation of this file.

/*===================================================================

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center,
Division of Medical and Biological Informatics.
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 or http://www.mitk.org for details.

===================================================================*/

#ifndef MITKUSImageVideoSource_H_HEADER_INCLUDED_
#define MITKUSImageVideoSource_H_HEADER_INCLUDED_

// ITK
#include <itkProcessObject.h>

// MITK
#include "mitkUSImageSource.h"
#include "mitkConvertGrayscaleOpenCVImageFilter.h"
#include "mitkCropOpenCVImageFilter.h"
#include "mitkBasicCombinationOpenCVImageFilter.h"

// OpenCV
#include <highgui.h>

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();
    virtual ~USImageVideoSource();

    virtual void GetNextRawImage( cv::Mat& image ) override;

    virtual void GetNextRawImage( 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 /* MITKUSImageVideoSource_H_HEADER_INCLUDED_ */