mitkToFNrrdImageWriter.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 mitkToFNrrdImageWriter_h
#define mitkToFNrrdImageWriter_h
 
#include <MitkToFHardwareExports.h>
#include "mitkToFImageWriter.h"
 
#include <fstream>
 
namespace mitk
{
  class MITKTOFHARDWARE_EXPORT ToFNrrdImageWriter : public ToFImageWriter
  {
  public:
    mitkClassMacro( ToFNrrdImageWriter , ToFImageWriter );
    itkFactorylessNewMacro(Self);
    itkCloneMacro(Self);
 
    void Open() override;
    void Close() override;
    void Add(float* distanceFloatData, float* amplitudeFloatData, float* intensityFloatData, unsigned char* rgbData=nullptr) override;
 
  protected:
 
    std::ofstream m_DistanceOutfile; 
    std::ofstream m_AmplitudeOutfile; 
    std::ofstream m_IntensityOutfile; 
    std::ofstream m_RGBOutfile; 
 
  private:
 
    ToFNrrdImageWriter();
    ~ToFNrrdImageWriter() override;
 
    void OpenStreamFile(std::ofstream &outfile, std::string outfileName);
    void CloseStreamFile(std::ofstream &outfile, std::string fileName);
    void ConvertStreamToNrrdFormat( std::string fileName );
  };
} //END mitk namespace
#endif