mitkToFNrrdImageWriter.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 __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=0) override;

  protected:

    std::ofstream m_DistanceOutfile; 
    std::ofstream m_AmplitudeOutfile; 
    std::ofstream m_IntensityOutfile; 
    std::ofstream m_RGBOutfile; 

  private:

    ToFNrrdImageWriter();
    ~ToFNrrdImageWriter();

    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 // __mitkToFNrrdImageWriter_h