mitkScene.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 __mitkScene_h
#define __mitkScene_h

#include "mitkSceneBase.h"
#include "mitkRCPtr.h"
#include "mitkManipulator.h"
#include "mitkQuaternion.h"
#include "mitkCamera.h"
#include "mitkLight.h"

class mitkList;
class mitkModel;

class MITK_VISUALIZATION_API mitkScene : public mitkSceneBase
{
public:
    MITK_TYPE(mitkScene,mitkSceneBase)
    mitkScene();
    mitkScene(mitkSceneContainer* container);
protected:
    virtual ~mitkScene();

public:
// Scene Functions
    const int * GetViewPort() { return m_ViewPort; }
    void GetViewPort(int vp[4])
    {
        vp[0] = m_ViewPort[0];
        vp[1] = m_ViewPort[1];
        vp[2] = m_ViewPort[2];
        vp[3] = m_ViewPort[3];
    }
    virtual void InitGL();
    virtual void DestroyGL();
    virtual void ResizeGL(int newX, int newY);
    virtual void RenderGL();
    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);
    virtual void OnMouseWheel(bool ctrlDown, bool shiftDown, int xPos, int yPos, int delta);
    void SetBackColor(unsigned char rColor, unsigned char gColor, unsigned char bColor);
    void SetBackColor(float rColor, float gColor, float bColor);
    void SetBackColor(int rColor, int gColor, int bColor);
    void GetBackColor(float &rColor, float &gColor, float &bColor);
    void GetBackColor(unsigned char &rColor, unsigned char &gColor, unsigned char &bColor);
    void GetBackColor(int &rColor, int &gColor, int &bColor);
    virtual void ResetScene();
    virtual void Translate(float deltX, float deltY, float deltZ);
    virtual void Rotate(float deltX, float deltY, float deltZ);
    virtual void Rotate(float ax, float ay, float az, float angle);
    virtual void Rotate(const mitkQuaternion &q);
    virtual void SetRotation(float x, float y, float z);
    virtual void SetRotation(float ax, float ay, float az, float angle);
    virtual void SetRotation(const mitkQuaternion &q);
    virtual void Scale(float delt);
    virtual mitkManipulator* CreateManipulator();
    void SetManipulator(mitkManipulator *mani);
    void AddModel(mitkModel *model);    
    void RemoveModel(mitkModel *model);
    void RemoveModel(int i);
    void RemoveAllModels();
    void RemoveAllModel() { this->RemoveAllModels(); }
    mitkModel* GetModel(int i);
    mitkList*  GetModels();
    int GetModelCount();
    int GetNumberOfPropsRendered() {return m_NumberOfPropsRendered;}
    void SetCamera(mitkCamera *cam);
    mitkCamera* GetCamera();
    void SetDefaultLight(mitkLight *lit);
    mitkLight* GetDefaultLight();
    mitkLight* GetLight() { return this->GetDefaultLight(); }
    void AddAdditionalLight(mitkLight *lit);
    void RemoveAdditionalLight(int idx);
    void RemoveAdditionalLight(mitkLight *lit);
    void RemoveAllAdditionalLights();
    int GetAdditionalLightCount();
    mitkLight* GetAdditionalLight(int idx);
    void AlignScene();
    float* GetZbufferData(int x1, int y1, int x2, int y2);
    void GetZbufferData(int x1, int y1, int x2, int y2, float *buffer);
    unsigned char* GetPixelData(int xOffset, int yOffset, int width, int height);
    void GetPixelData(int xOffset, int yOffset, int width, int height, unsigned char *buffer, int packAlignment = 4, bool swapRGB = false)
    { this->GetPixelData24(xOffset, yOffset, width, height, buffer, packAlignment, swapRGB); }
    void GetPixelData24(int xOffset, int yOffset, int width, int height, unsigned char *buffer, int packAlignment = 4, bool swapRGB = false);
    void GetPixelData32(int xOffset, int yOffset, int width, int height, unsigned char *buffer, bool swapRGB = false);
    void  SetSelected(bool isSelected) {if(isSelected) m_Selected = 1; else m_Selected = 0;}
    bool  GetSelected() {return (m_Selected) ? true : false;}
    float GetTranslateSpeed() { return m_TranslateSpeed; }
    float GetTranslationX()    { return m_TransX; }
    float GetTranslationY()    { return m_TransY; }
    float GetTranslationZ()    { return m_TransZ; }
    void   GetTranslation(float trans[3]) {trans[0] = m_TransX; trans[1] = m_TransY; trans[2] = m_TransZ;}
    const mitkQuaternion& GetRotation() { return m_Rot; }
    float GetScale() { return m_Scale; }
    void EnableLoD();
    void DisableLoD();
    bool IsLoDEnabled();
    bool IsModelsModified() 
    {
        return m_ModelsModified!=0;
    }
    bool IsCameraModified()
    {
        return m_CameraModified!=0;
    }
    bool IsAdditionalLightModified()
    {
        return m_AdditionalLightModified!=0;
    }   

protected:
    mitkScene(const mitkScene&);
    void operator = (const mitkScene&);

    float m_RotX, m_RotY, m_RotZ, m_RotateSpeed;
    float m_RotDeltX, m_RotDeltY, m_RotDeltZ;
    mitkQuaternion m_Rot;
    float m_TransX, m_TransY, m_TransZ, m_TranslateSpeed;
    float m_Scale, m_ScaleSpeed;
    
    float m_BackColor[4];
    int   m_ViewPort[4];

    
    mitkRCPtr<mitkManipulator> m_Mani;
    mitkRCPtr<mitkCamera> m_Camera; 
    mitkRCPtr<mitkLight> m_Light;
    mitkList *m_AdditionalLights;
    mitkList *m_Models;
    mitkModel **m_VisibleModels;
    int m_NumberOfPropsRendered;
    int m_RotAccuCount;
    unsigned char m_VisibleModelCount;  
    unsigned char m_CameraModified;
    unsigned char m_AdditionalLightModified;
    unsigned char m_Selected;
    unsigned char m_ModelsModified;


    void _init();
    void _drawScene();
    void _drawBorder();
    void _drawLights();
    bool _isAdditionalLightsModified();
    void _rotateModels();

};

#endif

---