mitkToFImageRecorderFilter.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 mitkToFImageRecorderFilter_h
#define mitkToFImageRecorderFilter_h
 
#include <mitkImageToImageFilter.h>
#include <MitkToFHardwareExports.h>
 
#include <mitkToFImageWriter.h>
 
namespace mitk
{
  class MITKTOFHARDWARE_EXPORT ToFImageRecorderFilter : public ImageToImageFilter
  {
  public:
    mitkClassMacro(ToFImageRecorderFilter, ImageToImageFilter);
    itkFactorylessNewMacro(Self);
    itkCloneMacro(Self);
 
    void SetFileName(std::string fileName);
    void SetImageType(ToFImageWriter::ToFImageType tofImageType);
    ToFImageWriter::Pointer GetToFImageWriter();
    void SetToFImageWriter(ToFImageWriter::Pointer tofImageWriter);
    void StartRecording();
    void StopRecording();
 
    using itk::ProcessObject::SetInput;
 
    void SetInput( const InputImageType* input) override;
    void SetInput(unsigned int idx,  const InputImageType* input) override;
     Image* GetInput();
     Image* GetInput(unsigned int idx);
 
  protected:
    ToFImageRecorderFilter();
    ~ToFImageRecorderFilter() override;
    void GenerateData() override;
    void CreateOutputsForAllInputs();
 
    bool m_RecordingStarted; 
    ToFImageWriter::Pointer m_ToFImageWriter; 
    std::string m_FileExtension;
  };
} //END mitk namespace
#endif