mitk3DBackProjectionUsingGPU.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 __mitk3DBackProjectionUsingGPU_h
#define __mitk3DBackProjectionUsingGPU_h

#include "mitkBackProjectionUsingGPU.h"
#include <GL/glew.h>
#include <GL/glut.h>
#include <Cg/cgGL.h>
#include <math.h>
#include <vector>

#define TEXTURECHANNEL 8
#define COLORATTACHMENTNUM 2
#define WINDOWSIZE 8





class MITK_RECONSTRUCTION_CG_API mitk3DBackProjectionUsingGPU : public mitkBackProjectionUsingGPU
{
public:
    MITK_TYPE(mitk3DBackProjectionUsingGPU,mitkBackProjectionUsingGPU)

    virtual void PrintSelf(ostream& os);

    mitk3DBackProjectionUsingGPU();  

    virtual void InputProjectionData(float *data);

    virtual bool CreateTexture();

    virtual void Draw (int mBottom);

    virtual void ReadResult(int mSubSliceNum,float * result);

    virtual void Rotate(float angle);

    virtual void SetParameter(int xrayNumx,int xrayNumz,int angleNum,double begenAngle,double endAngle,double SOD,double SDD,double detectorLengthX,double detectorLengthZ,float detectorMoveX,float detectorMoveZ,double *fov,int interpolationModel,int volumeBlockNum,int width,int height,int imgslice);

private:

    virtual bool _createTexture();

    void SetDtheta(float m) {dthetaDegree = m;}

    void initWindow();

    void initFBO();

    void Bind();

    bool CheckFBOStatus();

    void initCG();

    void CheckForCgError(const char *situation);

    void AttachFBO(int layer);

    void EnableFshader();

    void EnableVshader();

    void DestroyALL();

    void SetVolumeBlockSize(int m) {volumeBlockSize = m; };

    void SetFrustum(float zn,float zf,float zc, float detectorLengthx,float detectorLengthz);
        
    // for temp use
    void SetFov(double *m)
    {
        fov[0] = m[0]; 
        fov[1] = m[1]; 
        fov[2] = m[2];
        fov[3] = m[3]; 
        fov[4] = m[4]; 
        fov[5] = m[5];
    }
    void SetFov(double xmin,double xmax,double ymin,double ymax,double zmin,double zmax)
    {
        fov[0] = xmin;
        fov[1] = xmax; 
        fov[2] = ymin;
        fov[3] = ymax; 
        fov[4] = zmin; 
        fov[5] = zmax;
    }
    void SetXrayNumx(int m){xrayNumx = m;}
    void SetXrayNumz(int m){xrayNumz = m;}
    void SetAngleNum(int m){angleNum = m;}
    void SetWidth(int m){width = m;}
    void SetHeight(int m){height = m;}
    void SetImgslice(int m){imgslice = m;}

    int xrayNumx;
    int xrayNumz;
    int angleNum;
    int width;
    int height;
    int imgslice;
    double fov[6];
    int volumeBlockSize;


    struct TexParam 
    {
        const char *name;
        int unit;
        int target;
        int tex;
    };
    std::vector<TexParam> m_texParams;

    void _scanTexParams(CGprogram m_prog);
    void _setTexture(char *name, GLuint tex);
    void _bindTextures();

    float dthetaDegree;
    float zNear;
    float zFar;     //L
    float zCenter;  //D
    float nearDetectorLengthX;
    float nearDetectorLengthZ;
    GLfloat frustum[16];

    float zCenter2;
    GLuint windowHandle;    
    CGcontext cgContext;
    CGprofile fProfile ,vProfile;
    CGprogram fProgram ,vProgram;
    CGparameter mvpMatrixParam,
        preResultTexParam,
        texMatrixParam[TEXTURECHANNEL];

    GLuint *projectDataID, *ResultID;
    GLuint fbotex;
    GLuint fb;
    GLint m_preFboId;
    int head;
    int flag;
    void _bindCurrentFBO();
    void _unBind();
    
 protected:  
     virtual ~mitk3DBackProjectionUsingGPU(); 

private:

    mitk3DBackProjectionUsingGPU(const mitk3DBackProjectionUsingGPU&);
    void operator=(const mitk3DBackProjectionUsingGPU&);
};


//#define DEFINED_mitk3DBackProjectionUsingGPU
#endif

---