mitkToFCameraMITKPlayerController.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 __mitkToFCameraMITKPlayerController_h
#define __mitkToFCameraMITKPlayerController_h

#include <MitkToFHardwareExports.h>
#include "mitkCommon.h"
#include "mitkFileReader.h"
#include "mitkImage.h"

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


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

    mitkClassMacroItkParent( ToFCameraMITKPlayerController , itk::Object );

    itkFactorylessNewMacro(Self)
    itkCloneMacro(Self)

    
    virtual bool OpenCameraConnection();
    virtual bool CloseCameraConnection();
    virtual void GetAmplitudes(float* amplitudeArray);
    virtual void GetIntensities(float* intensityArray);
    virtual void GetDistances(float* distanceArray);
    virtual void GetRgb(unsigned char* rgbArray);
    virtual void UpdateCamera();

    virtual void SetInputFileName(std::string inputFileName);

    itkGetMacro(PixelNumber, int);
    itkGetMacro(RGBPixelNumber, int);
    itkGetMacro(CaptureWidth, int);
    itkGetMacro(CaptureHeight, int);
    itkGetMacro(RGBCaptureWidth, int);
    itkGetMacro(RGBCaptureHeight, int);
    itkGetMacro( DistanceImageFileName, std::string );
    itkGetMacro( AmplitudeImageFileName, std::string );
    itkGetMacro( IntensityImageFileName, std::string );
    itkGetMacro( RGBImageFileName, std::string );

    itkSetMacro( DistanceImageFileName, std::string );
    itkSetMacro( AmplitudeImageFileName, std::string );
    itkSetMacro( IntensityImageFileName, std::string );
    itkSetMacro( RGBImageFileName, std::string );
    enum ToFImageType{ ToFImageType3D, ToFImageType2DPlusT };

  protected:

    ToFCameraMITKPlayerController();

    ~ToFCameraMITKPlayerController();

    int m_PixelNumber; 
    int m_RGBPixelNumber; 
    int m_NumberOfBytes; 
    int m_NumberOfRGBBytes; 
    int m_CaptureWidth; 
    int m_CaptureHeight; 
    int m_RGBCaptureWidth; 
    int m_RGBCaptureHeight; 
    bool m_ConnectionCheck; 

    std::string m_InputFileName;
    ToFImageType m_ToFImageType; 

    Image::Pointer m_DistanceImage;
    Image::Pointer m_AmplitudeImage;
    Image::Pointer m_IntensityImage;
    Image::Pointer m_RGBImage;
    std::vector<bool> m_ImageStatus;

    FILE* m_DistanceInfile; 
    FILE* m_AmplitudeInfile; 
    FILE* m_IntensityInfile; 
    FILE* m_RGBInfile; 

    float* m_IntensityArray; 
    float* m_DistanceArray; 
    float* m_AmplitudeArray; 
    unsigned char* m_RGBArray; 

    std::string m_DistanceImageFileName; 
    std::string m_AmplitudeImageFileName; 
    std::string m_IntensityImageFileName; 
    std::string m_RGBImageFileName; 

    long m_PixelStartInFile;
    int m_CurrentFrame;
    int m_NumOfFrames;

  private:

    void AccessData(int frame, Image::Pointer image, float* &data);
    void CleanUp();
  };
} //END mitk namespace
#endif