mitkToFImageWriter.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 __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();

    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=0){};

  protected:

    void CheckForFileExtension(std::string& fileName);

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

    int m_ToFCaptureWidth; 
    int m_ToFCaptureHeight; 
    int m_ToFPixelNumber; 
    int m_ToFImageSizeInBytes; 
    int m_RGBCaptureWidth; 
    int m_RGBCaptureHeight; 
    int m_RGBPixelNumber; 
    int m_RGBImageSizeInBytes; 
    int m_NumOfFrames; 
    ToFImageWriter::ToFImageType m_ToFImageType; 

    bool m_DistanceImageSelected; 
    bool m_AmplitudeImageSelected; 
    bool m_IntensityImageSelected; 
    bool m_RGBImageSelected; 

  private:

  };
} //END mitk namespace
#endif // __mitkToFImageWriter_h