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