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