mitkIGTLMessageToUSImageFilter.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 mitkIGTLMessageToUSImageFilter_h
#define mitkIGTLMessageToUSImageFilter_h
 
#include <mitkCommon.h>
#include <MitkUSExports.h>
#include <mitkUSImageSource.h>
#include <mitkIGTLMessageSource.h>
#include <igtlImageMessage.h>
 
namespace mitk
{
  class MITKUS_EXPORT IGTLMessageToUSImageFilter : public USImageSource
  {
  public:
    mitkClassMacro(IGTLMessageToUSImageFilter, USImageSource);
    itkFactorylessNewMacro(Self);
    itkCloneMacro(Self);
 
    void SetNumberOfExpectedOutputs(unsigned int numOutputs);
 
    void ConnectTo(mitk::IGTLMessageSource* UpstreamFilter);
 
  protected:
    IGTLMessageToUSImageFilter();
 
    using Superclass::GetNextRawImage;
 
    void GetNextRawImage(std::vector<mitk::Image::Pointer>& imgVector) override;
 
  private:
    mitk::IGTLMessageSource* m_upstream;
    mitk::Image::Pointer m_previousImage;
    template <typename TPixel>
    void Initiate(mitk::Image::Pointer& img, igtl::ImageMessage* msg, bool big_endian);
  };
}  // namespace mitk
 
#endif