mitkFDKSpeedUsingGpu.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 __mitkFDKSpeedUsingGpu_h
#define __mitkFDKSpeedUsingGpu_h

#include <math.h>
#include <vector>

#include <GL/glew.h>
#include <GL/glut.h>
#include <Cg/cgGL.h>
#include "mitkProjectDataToVolumeFilter.h"
#include "mitkConeBeamProjectData.h"
#include "mitkVolume.h"
#include "mitkICVolume.h"
#include "mitkOoCVolume.h"
#include "mitkReconstructionCGIncludes.h"

#include "mitkSIMD.h"
#include <math.h>

#ifdef USE_SIMD
#pragma pack(push,16)    /* Must ensure class & union 16-B aligned */
#endif

#define PI 3.14159265358979
#define TEXTURECHANNEL 8
#define COLORATTACHMENTNUM 2
#define WINDOWSIZE 8
// using GPU for acceleration, and NVIDIA card is required,  and 8800series or up is better.
// ************************************************************************************************
// This implementation is not our fasted version, if you want to get the fasted one, please mail to
// tian@ieee.org


class MITK_RECONSTRUCTION_CG_API mitkFDKSpeedUsingGpu : public mitkProjectDataToVolumeFilter  
{
public:

    template<class Ty>  friend int  t_ExecuteFDKSpeedUsingGpu(mitkConeBeamProjectData *input,mitkVolume *outdata,mitkFDKSpeedUsingGpu *self, Ty *a);

    MITK_TYPE(mitkFDKSpeedUsingGpu , mitkProjectDataToVolumeFilter)
    
    virtual void PrintSelf(ostream& os);

    mitkFDKSpeedUsingGpu();

    void SetFov(float m[6]);

    void SetVolumeBlockNum(int m) { volumeBlockNum = m;}

    int GetVolumeBlockNum(){return volumeBlockNum;}

protected:
    virtual bool Execute();

    virtual ~mitkFDKSpeedUsingGpu();

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

    void initWindow();

    void initFBO();

    void Bind();
    
    bool CheckFBOStatus();
    
    void initCG();

    void CheckForCgError(const char *situation);

    bool CreateTexture();

    void AttachFBO(int layer);

    void InputPrjectData(float *data);

    void ReadData(int i,float *fresult);

    void SetTexMatrix(float angle);

    void EnableFshader();

    void EnableVshader();

    void DrawQuad(int h);
    
    void DestroyALL();

    void SetVolumeBlockSize(int m) {volumeBlockSize = m; volumeBlockSizeAddOne = m+1;};

    void SetFrustum(float offsetX, float offsetZ, float D, float cdetectorLengthx, float cdetectorLengthz);

    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;}

private:
    int xrayNumx;
    int xrayNumz;
    int angleNum;

    int width;
    int height;
    int imgslice;


    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 v[6];
    float hvz;
    float sz ;
    float cz ;
    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,
        layerParam,WParam,HParam,flagParam,
        texMatrixParam[TEXTURECHANNEL];
    
    GLuint projectDataID, ResultID;
    GLuint fbotex;
    GLuint fb;
    GLint m_preFboId;

    int volumeBlockNum;
    int volumeBlockSize;
    int volumeBlockSizeAddOne;

    int head;
    int flag;

    void _bindCurrentFBO();
    void _unBind();
    mitkFDKSpeedUsingGpu(const mitkFDKSpeedUsingGpu &);
    void operator = (const mitkFDKSpeedUsingGpu&);

};

//#define DEFINED_mitkFDKSpeedUsingGpu

#endif 

---