ProgramCG.h

Go to the documentation of this file.

//      File:           ProgramCG.h
//      Author:         Changchang Wu
//      Description :   interface for the ProgramCG classes.
//              ProgramCG:              Cg programs
//              ShaderBagCG:    All Cg shaders for Sift in a bag
//              FilterGLCG:             Cg Gaussian Filters
//
//      Copyright (c) 2007 University of North Carolina at Chapel Hill
//      All Rights Reserved
//
//      Permission to use, copy, modify and distribute this software and its
//      documentation for educational, research and non-profit purposes, without
//      fee, and without a written agreement is hereby granted, provided that the
//      above copyright notice and the following paragraph appear in all copies.
//      
//      The University of North Carolina at Chapel Hill make no representations
//      about the suitability of this software for any purpose. It is provided
//      'as is' without express or implied warranty. 
//
//      Please send BUG REPORTS to ccwu@cs.unc.edu
//


#if defined(CG_SIFTGPU_ENABLED) 

#ifndef _PROGRAM_CG_H
#define _PROGRAM_CG_H

#include "ProgramGPU.h"
class FragmentProgram;
#include "Cg/cgGL.h"

class ProgramCG:public ProgramGPU  
{
        CGprogram                 _programID; 
        CGprofile                 _profile;
        int                               _valid;
public:
        static CGcontext _Context;
        static CGprofile _FProfile;     
public:
        operator CGprogram (){return _programID;}
        CGprogram GetProgramID(){return _programID;}
        int UseProgram();
        int IsValidProgram(){return _programID && _valid;}
        static void  ErrorCallback();
        static void InitContext();
        static void DestroyContext();
        ProgramCG(const char * code, const char** cg_compile_args= NULL, CGprofile profile = ProgramCG::_FProfile);
        ProgramCG();
        virtual ~ProgramCG();

};

class ShaderBagCG:public ShaderBag
{
        CGparameter _param_dog_texu;
        CGparameter     _param_dog_texd;
        CGparameter _param_genlist_start_tex0;
        CGparameter _param_ftex_width;
        CGparameter _param_genlist_step_tex;
        CGparameter _param_genlist_step_tex0;
        CGparameter _param_genvbo_size;
        CGparameter _param_orientation_gtex;
        CGparameter _param_orientation_stex;
        CGparameter _param_orientation_size;
        CGparameter _param_descriptor_gtex;
        CGparameter _param_descriptor_size;
        CGparameter _param_descriptor_dsize;
        CGparameter _param_margin_copy_truncate;
        CGparameter _param_genlist_init_bbox;
public:
        virtual void LoadDescriptorShader();
        void    LoadDescriptorShaderF2();
        static void  WriteOrientationCodeToStream(ostream& out);
        virtual void SetGenListInitParam(int w, int h);
        virtual void SetMarginCopyParam(int xmax, int ymax);
        virtual void SetFeatureOrientationParam(int gtex, int width, int height, float sigma, int stex = 0, float step = 1.0f);
        virtual void SetFeatureDescirptorParam(int gtex, int otex, float dwidth, float fwidth, float width, float height, float sigma);
        virtual void SetSimpleOrientationInput(int oTex, float sigma, float sigma_step);
        void LoadOrientationShader();
        virtual void SetGenListStartParam(float width, int tex0);
        static ProgramCG* LoadGenListStepShader(int start, int step);
        static ProgramCG* LoadGenListStepShaderV2(int start, int step);
        void LoadGenListShader(int ndoglev,  int nlev);
        virtual void UnloadProgram();
        virtual void SetDogTexParam(int texU, int texD);
        virtual void SetGenListStepParam(int tex, int tex0);
        virtual void SetGenVBOParam( float width, float fwidth,  float size);
        virtual void LoadFixedShaders();
        virtual void LoadDisplayShaders();
        virtual void LoadKeypointShader(float threshold, float edgeThreshold);
        virtual int  LoadKeypointShaderMR(float threshold, float edgeThreshold);
        ShaderBagCG();
        virtual ~ShaderBagCG(){}
};


class FilterGLCG : public FilterProgram
{
private:
        ProgramGPU* CreateFilterH(float kernel[], float offset[], int width);
        ProgramGPU* CreateFilterV(float kernel[], float offset[], int height);
        //packed version 
        ProgramGPU* CreateFilterHPK(float kernel[], float offset[], int width);
        ProgramGPU* CreateFilterVPK(float kernel[], float offset[], int height);
};

class ShaderBagPKCG:public ShaderBag
{
private:
        CGparameter _param_dog_texu;
        CGparameter     _param_dog_texd;
        CGparameter _param_margin_copy_truncate;
        CGparameter _param_grad_pass_texp;
        CGparameter _param_genlist_init_bbox;
        CGparameter _param_genlist_start_tex0;
        CGparameter _param_ftex_width;
        CGparameter _param_genlist_step_tex;
        CGparameter _param_genlist_step_tex0;
        CGparameter _param_genlist_end_ktex;
        CGparameter _param_genvbo_size;
        CGparameter _param_orientation_gtex;
        CGparameter _param_orientation_otex;
        CGparameter _param_orientation_size;
        CGparameter     _param_descriptor_gtex; 
        CGparameter     _param_descriptor_otex;
        CGparameter     _param_descriptor_size; 
        CGparameter     _param_descriptor_dsize;

public:
        ShaderBagPKCG();
        virtual ~ShaderBagPKCG(){}
        virtual void LoadDescriptorShader();
        virtual void LoadDescriptorShaderF2();
        virtual void LoadOrientationShader();
        virtual void LoadGenListShader(int ndoglev, int nlev);
        virtual void LoadGenListShaderV2(int ndoglev, int nlev);
        virtual void UnloadProgram() ;
        virtual void LoadKeypointShader(float threshold, float edgeTrheshold);
        virtual void LoadFixedShaders();
        virtual void LoadDisplayShaders();
        virtual void SetGradPassParam(int texP);
        virtual void SetGenListEndParam(int ktex);
public:
        //parameters
        virtual void SetGenListStartParam(float width, int tex0);
        virtual void SetGenListInitParam(int w, int h);
        virtual void SetMarginCopyParam(int xmax, int ymax);
        virtual void SetDogTexParam(int texU, int texD);
        virtual void SetGenListStepParam(int tex, int tex0);
        virtual void SetGenVBOParam( float width, float fwidth, float size);
        virtual void SetFeatureDescirptorParam(int gtex, int otex, float dwidth, float fwidth, float width, float height, float sigma);
        virtual void SetFeatureOrientationParam(int gtex, int width, int height, float sigma, int stex, float step);
        virtual void SetSimpleOrientationInput(int oTex, float sigma, float sigma_step);
};
#endif
#endif