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