root / rgbdslam / external / siftgpu / src / SiftGPU / ProgramCL.h @ 9240aaa3
History | View | Annotate | Download (5.97 KB)
1 | 9240aaa3 | Alex | ////////////////////////////////////////////////////////////////////////////
|
---|---|---|---|
2 | // File: ProgramCL.h
|
||
3 | // Author: Changchang Wu
|
||
4 | // Description : interface for the ProgramCL classes.
|
||
5 | // ProgramCL: Cg programs
|
||
6 | // ShaderBagCG: All Cg shaders for Sift in a bag
|
||
7 | // FilterCL: Cg Gaussian Filters
|
||
8 | //
|
||
9 | // Copyright (c) 2007 University of North Carolina at Chapel Hill
|
||
10 | // All Rights Reserved
|
||
11 | //
|
||
12 | // Permission to use, copy, modify and distribute this software and its
|
||
13 | // documentation for educational, research and non-profit purposes, without
|
||
14 | // fee, and without a written agreement is hereby granted, provided that the
|
||
15 | // above copyright notice and the following paragraph appear in all copies.
|
||
16 | //
|
||
17 | // The University of North Carolina at Chapel Hill make no representations
|
||
18 | // about the suitability of this software for any purpose. It is provided
|
||
19 | // 'as is' without express or implied warranty.
|
||
20 | //
|
||
21 | // Please send BUG REPORTS to ccwu@cs.unc.edu
|
||
22 | //
|
||
23 | ////////////////////////////////////////////////////////////////////////////
|
||
24 | |||
25 | |||
26 | #if defined(CL_SIFTGPU_ENABLED)
|
||
27 | |||
28 | #ifndef _PROGRAM_CL_H
|
||
29 | #define _PROGRAM_CL_H
|
||
30 | |||
31 | #include "ProgramGPU.h" |
||
32 | |||
33 | class ProgramCL: public ProgramGPU |
||
34 | { |
||
35 | cl_program _program; |
||
36 | cl_kernel _kernel; |
||
37 | int _valid;
|
||
38 | public:
|
||
39 | int IsValidProgram(){return _program && _valid;} |
||
40 | ProgramCL(const char* name, const char * code, cl_context contex, cl_device_id device); |
||
41 | ProgramCL(); |
||
42 | void PrintBuildLog(cl_device_id device, int all); |
||
43 | virtual ~ProgramCL(); |
||
44 | virtual int UseProgram(){return 1;} |
||
45 | virtual void * GetProgramID() {return _kernel;} |
||
46 | friend class ProgramBagCL; |
||
47 | friend class ProgramBagCLN; |
||
48 | }; |
||
49 | |||
50 | class CLTexImage; |
||
51 | class FilterCL |
||
52 | { |
||
53 | public:
|
||
54 | ProgramCL* s_shader_h; |
||
55 | ProgramCL* s_shader_v; |
||
56 | int _size;
|
||
57 | int _id;
|
||
58 | CLTexImage * _weight; |
||
59 | public:
|
||
60 | FilterCL() : s_shader_h(NULL), s_shader_v(NULL), _size(0), _id(0), _weight(NULL) {} |
||
61 | ~FilterCL() {if(s_shader_h) delete s_shader_h; if(s_shader_v) delete s_shader_v; if(_weight) delete _weight; } |
||
62 | }; |
||
63 | |||
64 | class SiftParam; |
||
65 | |||
66 | class ProgramBagCL |
||
67 | { |
||
68 | protected:
|
||
69 | cl_platform_id _platform; |
||
70 | cl_device_id _device; |
||
71 | cl_context _context; |
||
72 | cl_command_queue _queue; |
||
73 | protected:
|
||
74 | ProgramCL * s_gray; |
||
75 | ProgramCL * s_sampling; |
||
76 | ProgramCL * s_sampling_k; |
||
77 | ProgramCL * s_sampling_u; |
||
78 | ProgramCL * s_zero_pass; |
||
79 | ProgramCL * s_packup; |
||
80 | ProgramCL * s_unpack; |
||
81 | ProgramCL * s_unpack_dog; |
||
82 | ProgramCL * s_unpack_grd; |
||
83 | ProgramCL * s_unpack_key; |
||
84 | ProgramCL * s_dog_pass; |
||
85 | ProgramCL * s_grad_pass; |
||
86 | ProgramCL * s_grad_pass2; |
||
87 | ProgramCL * s_gray_pack; |
||
88 | ProgramCL * s_keypoint; |
||
89 | public:
|
||
90 | FilterCL * f_gaussian_skip0; |
||
91 | vector<FilterCL*> f_gaussian_skip0_v; |
||
92 | FilterCL * f_gaussian_skip1; |
||
93 | FilterCL ** f_gaussian_step; |
||
94 | int _gaussian_step_num;
|
||
95 | public:
|
||
96 | ProgramBagCL(); |
||
97 | bool InitializeContext();
|
||
98 | virtual ~ProgramBagCL(); |
||
99 | void FinishCL();
|
||
100 | cl_context GetContextCL() {return _context;}
|
||
101 | cl_command_queue GetCommandQueue() {return _queue;}
|
||
102 | static const char* GetErrorString(cl_int error); |
||
103 | static bool CheckErrorCL(cl_int error, const char* location = NULL); |
||
104 | public:
|
||
105 | FilterCL * CreateGaussianFilter(float sigma);
|
||
106 | void CreateGaussianFilters(SiftParam¶m);
|
||
107 | void SelectInitialSmoothingFilter(int octave_min, SiftParam¶m); |
||
108 | void FilterInitialImage(CLTexImage* tex, CLTexImage* buf);
|
||
109 | void FilterSampledImage(CLTexImage* tex, CLTexImage* buf);
|
||
110 | void UnpackImage(CLTexImage*src, CLTexImage* dst);
|
||
111 | void UnpackImageDOG(CLTexImage*src, CLTexImage* dst);
|
||
112 | void UnpackImageGRD(CLTexImage*src, CLTexImage* dst);
|
||
113 | void UnpackImageKEY(CLTexImage*src, CLTexImage* dog, CLTexImage* dst);
|
||
114 | void ComputeDOG(CLTexImage*tex, CLTexImage* texp, CLTexImage* dog, CLTexImage* grad, CLTexImage* rot);
|
||
115 | void ComputeKEY(CLTexImage*dog, CLTexImage* key, float Tdog, float Tedge); |
||
116 | public:
|
||
117 | virtual void SampleImageU(CLTexImage *dst, CLTexImage *src, int log_scale); |
||
118 | virtual void SampleImageD(CLTexImage *dst, CLTexImage *src, int log_scale = 1); |
||
119 | virtual void FilterImage(FilterCL* filter, CLTexImage *dst, CLTexImage *src, CLTexImage*tmp);
|
||
120 | virtual ProgramCL* CreateFilterH(float kernel[], int width); |
||
121 | virtual ProgramCL* CreateFilterV(float kernel[], int width); |
||
122 | virtual FilterCL* CreateFilter(float kernel[], int width); |
||
123 | public:
|
||
124 | virtual void InitProgramBag(SiftParam¶m);
|
||
125 | virtual void LoadDescriptorShader();
|
||
126 | virtual void LoadDescriptorShaderF2();
|
||
127 | virtual void LoadOrientationShader();
|
||
128 | virtual void LoadGenListShader(int ndoglev, int nlev); |
||
129 | virtual void UnloadProgram() ;
|
||
130 | virtual void LoadKeypointShader();
|
||
131 | virtual void LoadFixedShaders();
|
||
132 | virtual void LoadDisplayShaders();
|
||
133 | virtual void LoadDynamicShaders(SiftParam& param);
|
||
134 | public:
|
||
135 | //parameters
|
||
136 | virtual void SetGradPassParam(int texP); |
||
137 | virtual void SetGenListEndParam(int ktex); |
||
138 | virtual void SetGenListStartParam(float width, int tex0); |
||
139 | virtual void SetGenListInitParam(int w, int h); |
||
140 | virtual void SetMarginCopyParam(int xmax, int ymax); |
||
141 | virtual void SetDogTexParam(int texU, int texD); |
||
142 | virtual void SetGenListStepParam(int tex, int tex0); |
||
143 | virtual void SetGenVBOParam( float width, float fwidth, float size); |
||
144 | virtual void SetFeatureDescirptorParam(int gtex, int otex, float dwidth, float fwidth, float width, float height, float sigma); |
||
145 | virtual void SetFeatureOrientationParam(int gtex, int width, int height, float sigma, int stex, float step); |
||
146 | virtual void SetSimpleOrientationInput(int oTex, float sigma, float sigma_step); |
||
147 | |||
148 | }; |
||
149 | |||
150 | class CLTexImage ; |
||
151 | class ProgramBagCLN: public ProgramBagCL |
||
152 | { |
||
153 | public:
|
||
154 | virtual void SampleImageD(CLTexImage *dst, CLTexImage *src, int log_scale = 1); |
||
155 | virtual FilterCL* CreateFilter(float kernel[], int width); |
||
156 | virtual ProgramCL* CreateFilterH(float kernel[], int width); |
||
157 | virtual ProgramCL* CreateFilterV(float kernel[], int width); |
||
158 | virtual void FilterImage(FilterCL* filter, CLTexImage *dst, CLTexImage *src, CLTexImage*tmp);
|
||
159 | virtual void LoadFixedShaders();
|
||
160 | virtual void LoadDisplayShaders();
|
||
161 | }; |
||
162 | #endif
|
||
163 | #endif
|