mitkUSImageSource.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 mitkUSImageSource_h
#define mitkUSImageSource_h
 
#include <mutex>
 
// ITK
#include <itkProcessObject.h>
 
// MITK
#include <MitkUSExports.h>
#include <mitkCommon.h>
#include "mitkBasicCombinationOpenCVImageFilter.h"
#include "mitkOpenCVToMitkImageFilter.h"
#include "mitkImageToOpenCVImageFilter.h"
 
namespace mitk {
  class MITKUS_EXPORT USImageSource : public itk::Object
  {
  public:
    static const char* IMAGE_PROPERTY_IDENTIFIER;
 
    mitkClassMacroItkParent(USImageSource, itk::Object);
 
    itkGetMacro(ImageFilter, mitk::BasicCombinationOpenCVImageFilter::Pointer);
 
    void PushFilter(AbstractOpenCVImageFilter::Pointer filter);
    bool RemoveFilter(AbstractOpenCVImageFilter::Pointer filter);
    bool GetIsFilterInThePipeline(AbstractOpenCVImageFilter::Pointer filter);
 
    std::vector<mitk::Image::Pointer> GetNextImage();
 
  protected:
    USImageSource();
    ~USImageSource() override;
    virtual void GetNextRawImage(std::vector<cv::Mat>&);
 
    virtual void GetNextRawImage(std::vector<mitk::Image::Pointer>&) = 0;
 
    mitk::OpenCVToMitkImageFilter::Pointer m_OpenCVToMitkFilter;
    mitk::ImageToOpenCVImageFilter::Pointer m_MitkToOpenCVFilter;
 
  private:
    BasicCombinationOpenCVImageFilter::Pointer m_ImageFilter;
 
    int                                        m_CurrentImageId;
 
    std::mutex m_ImageFilterMutex;
  };
} // namespace mitk
#endif