mitkToFImageWriter.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 __mitkToFImageWriter_h
#define __mitkToFImageWriter_h

#include <MitkToFHardwareExports.h>
#include "mitkCommon.h"
#include "mitkToFImageGrabber.h"

#include "itkObject.h"
#include "itkObjectFactory.h"

namespace mitk
{
  class MITKTOFHARDWARE_EXPORT ToFImageWriter : public itk::Object
  {
  public:

    ToFImageWriter();

    ~ToFImageWriter() override;

    mitkClassMacroItkParent( ToFImageWriter , itk::Object );

    itkFactorylessNewMacro(Self);

    itkCloneMacro(Self);

    itkGetMacro( DistanceImageFileName, std::string );
    itkGetMacro( AmplitudeImageFileName, std::string );
    itkGetMacro( IntensityImageFileName, std::string );
    itkGetMacro( RGBImageFileName, std::string );
    itkGetMacro( Extension, std::string );
    itkGetMacro( ToFCaptureWidth, int );
    itkGetMacro( ToFCaptureHeight, int );
    itkGetMacro( RGBCaptureWidth, int );
    itkGetMacro( RGBCaptureHeight, int );
    itkGetMacro( DistanceImageSelected, bool );
    itkGetMacro( AmplitudeImageSelected, bool );
    itkGetMacro( IntensityImageSelected, bool );
    itkGetMacro( RGBImageSelected, bool );

    itkSetMacro( DistanceImageFileName, std::string );
    itkSetMacro( AmplitudeImageFileName, std::string );
    itkSetMacro( IntensityImageFileName, std::string );
    itkSetMacro( RGBImageFileName, std::string );
    itkSetMacro( Extension, std::string );
    itkSetMacro( ToFCaptureWidth, int );
    itkSetMacro( ToFCaptureHeight, int );
    itkSetMacro( RGBCaptureWidth, int );
    itkSetMacro( RGBCaptureHeight, int );
    itkSetMacro( DistanceImageSelected, bool );
    itkSetMacro( AmplitudeImageSelected, bool );
    itkSetMacro( IntensityImageSelected, bool );
    itkSetMacro( RGBImageSelected, bool );

    enum ToFImageType{ ToFImageType3D, ToFImageType2DPlusT };
    ToFImageWriter::ToFImageType GetToFImageType();
    void SetToFImageType(ToFImageWriter::ToFImageType toFImageType);
    virtual void Open(){};
    virtual void Close(){};
    virtual void Add(float* /*distanceFloatData*/, float* /*amplitudeFloatData*/, float* /*intensityFloatData*/, unsigned char* /*rgbData*/=nullptr){};

  protected:

    void CheckForFileExtension(std::string& fileName);

    // member variables
    std::string m_Extension; 
    std::string m_DistanceImageFileName; 
    std::string m_AmplitudeImageFileName; 
    std::string m_IntensityImageFileName; 
    std::string m_RGBImageFileName; 

    int m_NumOfFrames; 
    bool m_DistanceImageSelected; 
    bool m_AmplitudeImageSelected; 
    bool m_IntensityImageSelected; 
    bool m_RGBImageSelected; 
    int m_ToFCaptureWidth; 
    int m_ToFCaptureHeight; 
    int m_RGBCaptureWidth; 
    int m_RGBCaptureHeight; 
    int m_ToFPixelNumber; 
    int m_ToFImageSizeInBytes; 
    int m_RGBPixelNumber; 
    int m_RGBImageSizeInBytes; 
    ToFImageWriter::ToFImageType m_ToFImageType; 
  };
} //END mitk namespace
#endif // __mitkToFImageWriter_h