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_HEADER_INCLUDED_
#define MITKUSImageSource_H_HEADER_INCLUDED_

// ITK
#include <itkProcessObject.h>
#include <itkFastMutexLock.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;

    itk::FastMutexLock::Pointer m_ImageFilterMutex;
  };
} // namespace mitk
#endif /* MITKUSImageSource_H_HEADER_INCLUDED_ */