mitkCUDASegmentedVRScene.h

/*=========================================================================

  Program:   3DMed
  Date:      $Date: 2014-02-25 18:30:00 +0800 $
  Version:   $Version: 4.6.0 $
  Copyright: MIPG, Institute of Automation, Chinese Academy of Sciences

=========================================================================*/

#ifndef __mitkCUDASegmentedVRScene_h
#define __mitkCUDASegmentedVRScene_h

#include "mitkRCPtr.h"
#include "mitkVis2Scene3D.h"
#include "mitkCUDAVolumeRenderingIncludes.h"
#include "modules/mitkCUDASegmentedVR.h"
#include "mitkICVolume.h"
#include "mitkRGBATransferFunction1D.h"
#include "mitkPerformanceCounter.h"

class mitkSegmentedVolumeLut;

class MITK_CUDA_VOLUME_RENDERING_API mitkCUDASegmentedVRScene : public mitkVis2Scene3D
{
public:
    MITK_TYPE(mitkCUDASegmentedVRScene,mitkVis2Scene3D)
    mitkCUDASegmentedVRScene();

    virtual void RenderGL();

    void SetData(mitkICVolume *data);

    void SetMaskVolume(mitkICVolume *data);

    mitkICVolume* GetMaskVolume();

    void SetDataModified();

    void SetMaskModified();

    void SetBackGroundColor(float r,float g, float b);

    void GetBackGroundColor(float &r,float &g, float &b);

    mitkRGBATransferFunction1D* GetTransferFunction(int index);

    void SetTransferFunction(int index, mitkRGBATransferFunction1D* tf);

    void SetStdSampleDistance(float stdSampleDistance);

    float GetStdSampleDistance();

    void SetShade(bool shade);
    
    bool GetShade();

    void ShadeOn();
    
    void ShadeOff();

    void SetLightDirection(float x,float y,float z);
    
    void GetLightDirection(float &x,float &y,float &z);

    void SetLightIntensity(float intensity);
    
    void SetLightColor(float r,float g,float b);

    void SetAmbient(float value);
    
    float GetAmbient();

    void SetDiffuse(float value);
    
    float GetDiffuse();

    void SetSpecular(float value);
    
    float GetSpecular();

    void SetSpecularPower(float value);
    
    float GetSpecularPower();

    void RotateLightDirection(bool turnOn);

    virtual void OnMouseDown(int mouseButton, bool ctrlDown, bool shiftDown, int xPos, int yPos);
    virtual void OnMouseUp(int mouseButton, bool ctrlDown, bool shiftDown, int xPos, int yPos);
    virtual void OnMouseMove(bool ctrlDown, bool shiftDown, int xPos, int yPos);

    void SetIllustrativeShading(int illustrative);

    bool GetIllustrativeShading();

    void SetCroppingBounds(const float CroppingBounds[6]);

    void GetCroppingBounds(float CroppingBounds[6]);

    mitkCUDASegmentedVR& GetKernelModule() { return m_module; }
    

protected:
    virtual ~mitkCUDASegmentedVRScene();
    virtual void GetSize3D(float size[3]);

private:
    mitkCUDASegmentedVRScene(const mitkCUDASegmentedVRScene&);
    void operator = (const mitkCUDASegmentedVRScene&);

private:
    mitkRCPtr<mitkICVolume> m_Data;
    mitkRCPtr<mitkICVolume> m_Mask;
    mitkRCPtr<mitkRGBATransferFunction1D>* m_TFs;
    mitkLocalVersion* m_TFVersions;

    int m_SampleNumber;

    mitkSegmentedVolumeLut *m_CPU_LUT;

    mitkCUDASegmentedVR m_module;
    
    bool m_glewInit;
    bool m_dataModified;
    bool m_maskModified;

    bool m_RotateLightDirection;
    bool m_RotatingLightDirection;

    mitkVis2Vector4 m_OldLightDirection;

    TimeAverager m_TimeAverager1;
};

#endif

---