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

#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  // MITKIGTLMessageToUSImageFilter_H_HEADER_INCLUDED_