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