mitkIGTLMessageToUSImageFilter.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 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;

    virtual void GetNextRawImage(mitk::Image::Pointer& img);

  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_