mitkThreadedToFRawDataReconstruction.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 __mitkThreadedToFRawDataReconstruction_h
#define __mitkThreadedToFRawDataReconstruction_h

// mitk includes
#include "mitkImageSource.h"
#include <MitkPMDExports.h>

// itk includes
#include <itkBarrier.h>
#include <itkFastMutexLock.h>

// vtk includes
#include "vtkShortArray.h"

namespace mitk
{
  struct ThreadDataStruct
  {
    std::vector<std::vector<short> > m_InputData;
    std::vector<float*> m_OutputData;
    unsigned int m_DataSize;
    unsigned int m_LineWidth;
    unsigned int m_FrameHeight;
    unsigned int m_ModulationFrequency;
    itk::Barrier::Pointer m_Barrier;              
    itk::FastMutexLock::Pointer m_ImageDataMutex; 
    itk::FastMutexLock::Pointer m_ThreadDataMutex; 
  };

  class MITKPMD_EXPORT ThreadedToFRawDataReconstruction : public itk::ProcessObject
  {
   public:

    mitkClassMacroItkParent( ThreadedToFRawDataReconstruction ,  itk::ProcessObject );
    itkFactorylessNewMacro(Self);
    itkCloneMacro(Self);
    itkGetMacro(Init, bool);

    void SetChannelData(vtkShortArray* sourceData);

    void Initialize(int width, int height, int modulationFrequency, int sourceDataSize );

    void GetDistances(float* dist);
    void GetAmplitudes(float* ampl);
    void GetIntensities(float* inten);
    void GetAllData(float* dist, float* ampl, float* inten);

    void Update();

   protected:
    ThreadedToFRawDataReconstruction();
    ~ThreadedToFRawDataReconstruction();

    virtual void GenerateData();

    virtual void BeforeThreadedGenerateData();

    static ITK_THREAD_RETURN_TYPE ThreadedGenerateDataCallbackFunction(void* data);
    void ResizeOutputImage(float* in, float* out);

    // member variables
    int m_StackSize;                
    int m_Width;
    int m_Height;
    int m_ImageSize;
    int m_SourceDataSize;
    vtkShortArray* m_SourceData;
    bool m_Init;

    float* m_CISDist;               
    float* m_CISAmpl;               
    float* m_CISInten;              
    float* m_ThreadedCISDist;
    float* m_ThreadedCISAmpl;
    float* m_ThreadedCISInten;

    itk::MultiThreader::Pointer m_Threader;
    ThreadDataStruct* m_ThreadData;

  };
} //end mitk namespace

#endif // __mitkThreadedToFRawDataReconstruction_h