root / rgbdslam / external / siftgpu / src / SiftGPU / ShaderMan.cpp @ 9240aaa3
History | View | Annotate | Download (7.28 KB)
1 |
////////////////////////////////////////////////////////////////////////////
|
---|---|
2 |
// File: ShaderMan.cpp
|
3 |
// Author: Changchang Wu
|
4 |
// Description : implementation of the ShaderMan class.
|
5 |
// A Shader Manager that calls different implementation of shaders
|
6 |
//
|
7 |
//
|
8 |
// Copyright (c) 2007 University of North Carolina at Chapel Hill
|
9 |
// All Rights Reserved
|
10 |
//
|
11 |
// Permission to use, copy, modify and distribute this software and its
|
12 |
// documentation for educational, research and non-profit purposes, without
|
13 |
// fee, and without a written agreement is hereby granted, provided that the
|
14 |
// above copyright notice and the following paragraph appear in all copies.
|
15 |
//
|
16 |
// The University of North Carolina at Chapel Hill make no representations
|
17 |
// about the suitability of this software for any purpose. It is provided
|
18 |
// 'as is' without express or implied warranty.
|
19 |
//
|
20 |
// Please send BUG REPORTS to ccwu@cs.unc.edu
|
21 |
//
|
22 |
////////////////////////////////////////////////////////////////////////////
|
23 |
|
24 |
|
25 |
#include "GL/glew.h" |
26 |
#include <vector> |
27 |
#include <iostream> |
28 |
#include <stdlib.h> |
29 |
#include <math.h> |
30 |
using std::vector;
|
31 |
using std::ostream;
|
32 |
using std::endl;
|
33 |
|
34 |
|
35 |
#include "ProgramGLSL.h" |
36 |
#include "GlobalUtil.h" |
37 |
#include "GLTexImage.h" |
38 |
#include "SiftGPU.h" |
39 |
#include "ShaderMan.h" |
40 |
|
41 |
///
|
42 |
ShaderBag * ShaderMan::s_bag = NULL;
|
43 |
|
44 |
//////////////////////////////////////////////////////////////////////
|
45 |
// Construction/Destruction
|
46 |
//////////////////////////////////////////////////////////////////////
|
47 |
|
48 |
void ShaderMan::InitShaderMan(SiftParam¶m)
|
49 |
{ |
50 |
if(s_bag) return; |
51 |
|
52 |
if(GlobalUtil::_usePackedTex ) s_bag = new ShaderBagPKSL; |
53 |
else s_bag =new ShaderBagGLSL; |
54 |
|
55 |
GlobalUtil::StartTimer("Load Programs");
|
56 |
s_bag->LoadFixedShaders(); |
57 |
s_bag->LoadDynamicShaders(param); |
58 |
if(GlobalUtil::_UseSiftGPUEX) s_bag->LoadDisplayShaders();
|
59 |
GlobalUtil::StopTimer(); |
60 |
|
61 |
GlobalUtil::CheckErrorsGL("InitShaderMan");
|
62 |
} |
63 |
|
64 |
|
65 |
void ShaderMan::DestroyShaders()
|
66 |
{ |
67 |
if(s_bag) delete s_bag; |
68 |
s_bag = NULL;
|
69 |
} |
70 |
|
71 |
void ShaderMan::UnloadProgram()
|
72 |
{ |
73 |
if(s_bag) s_bag->UnloadProgram();
|
74 |
} |
75 |
|
76 |
void ShaderMan::FilterImage(FilterProgram* filter, GLTexImage *dst, GLTexImage *src, GLTexImage*tmp)
|
77 |
{ |
78 |
if(filter == NULL) return; |
79 |
|
80 |
//////////////////////////////
|
81 |
src->FillMargin(filter->_size, 0);
|
82 |
|
83 |
//output parameter
|
84 |
if(tmp) tmp->AttachToFBO(0); |
85 |
else dst->AttachToFBO(0); |
86 |
|
87 |
|
88 |
//input parameter
|
89 |
src->BindTex(); |
90 |
dst->FitTexViewPort(); |
91 |
|
92 |
//horizontal filter
|
93 |
filter->s_shader_h->UseProgram(); |
94 |
dst->DrawQuad(); |
95 |
|
96 |
//parameters
|
97 |
if(tmp)
|
98 |
{ |
99 |
// fill margin for out-of-boundary lookup
|
100 |
tmp->DetachFBO(0);
|
101 |
tmp->AttachToFBO(0);
|
102 |
tmp->FillMargin(0, filter->_size);
|
103 |
tmp->DetachFBO(0);
|
104 |
dst->AttachToFBO(0);
|
105 |
tmp->BindTex(); |
106 |
} |
107 |
else
|
108 |
{ |
109 |
glFinish(); |
110 |
// fill margin for out-of-boundary lookup
|
111 |
dst->FillMargin(0, filter->_size);
|
112 |
dst->BindTex(); |
113 |
} |
114 |
|
115 |
//vertical filter
|
116 |
filter->s_shader_v->UseProgram(); |
117 |
dst->DrawQuad(); |
118 |
|
119 |
|
120 |
//clean up
|
121 |
dst->UnbindTex(); |
122 |
dst->DetachFBO(0);
|
123 |
|
124 |
//
|
125 |
ShaderMan::UnloadProgram(); |
126 |
} |
127 |
|
128 |
|
129 |
void ShaderMan::FilterInitialImage(GLTexImage* tex, GLTexImage* buf)
|
130 |
{ |
131 |
if(s_bag->f_gaussian_skip0) FilterImage(s_bag->f_gaussian_skip0, tex, tex, buf);
|
132 |
} |
133 |
|
134 |
void ShaderMan::FilterSampledImage(GLTexImage* tex, GLTexImage* buf)
|
135 |
{ |
136 |
if(s_bag->f_gaussian_skip1) FilterImage(s_bag->f_gaussian_skip1, tex, tex, buf);
|
137 |
} |
138 |
|
139 |
void ShaderMan::TextureCopy(GLTexImage*dst, GLTexImage*src)
|
140 |
{ |
141 |
|
142 |
dst->AttachToFBO(0);
|
143 |
|
144 |
src->BindTex(); |
145 |
|
146 |
dst->FitTexViewPort(); |
147 |
|
148 |
dst->DrawQuad(); |
149 |
|
150 |
dst->UnbindTex(); |
151 |
// ShaderMan::UnloadProgram();
|
152 |
dst->DetachFBO(0);
|
153 |
return;
|
154 |
} |
155 |
void ShaderMan::TextureDownSample(GLTexImage *dst, GLTexImage *src, int scale) |
156 |
{ |
157 |
//output parameter
|
158 |
|
159 |
dst->AttachToFBO(0);
|
160 |
|
161 |
//input parameter
|
162 |
src->BindTex(); |
163 |
|
164 |
//
|
165 |
dst->FitTexViewPort(); |
166 |
|
167 |
s_bag->s_sampling->UseProgram(); |
168 |
|
169 |
dst->DrawQuadDS(scale); |
170 |
src->UnbindTex(); |
171 |
|
172 |
UnloadProgram(); |
173 |
|
174 |
dst->DetachFBO(0);
|
175 |
} |
176 |
|
177 |
void ShaderMan::TextureUpSample(GLTexImage *dst, GLTexImage *src, int scale) |
178 |
{ |
179 |
|
180 |
//output parameter
|
181 |
dst->AttachToFBO(0);
|
182 |
//input parameter
|
183 |
src->BindTex(); |
184 |
|
185 |
dst->FitTexViewPort(); |
186 |
|
187 |
GlobalUtil::SetTextureParameterUS(); |
188 |
|
189 |
if(GlobalUtil::_usePackedTex)
|
190 |
{ |
191 |
s_bag->s_sampling->UseProgram(); |
192 |
} |
193 |
|
194 |
dst->DrawQuadUS(scale); |
195 |
src->UnbindTex(); |
196 |
|
197 |
UnloadProgram(); |
198 |
|
199 |
dst->DetachFBO(0);
|
200 |
|
201 |
GlobalUtil::SetTextureParameter(); |
202 |
} |
203 |
|
204 |
|
205 |
|
206 |
void ShaderMan::UseShaderDisplayGaussian()
|
207 |
{ |
208 |
if(s_bag && s_bag->s_display_gaussian) s_bag->s_display_gaussian->UseProgram();
|
209 |
} |
210 |
|
211 |
void ShaderMan::UseShaderDisplayDOG()
|
212 |
{ |
213 |
if(s_bag && s_bag->s_display_dog) s_bag->s_display_dog->UseProgram();
|
214 |
} |
215 |
|
216 |
|
217 |
|
218 |
void ShaderMan::UseShaderRGB2Gray()
|
219 |
{ |
220 |
if(s_bag && s_bag->s_gray)s_bag->s_gray->UseProgram();
|
221 |
} |
222 |
|
223 |
|
224 |
void ShaderMan::UseShaderDisplayGrad()
|
225 |
{ |
226 |
if(s_bag && s_bag->s_display_grad) s_bag->s_display_grad->UseProgram();
|
227 |
} |
228 |
|
229 |
|
230 |
void ShaderMan::UseShaderDisplayKeypoints()
|
231 |
{ |
232 |
if(s_bag && s_bag->s_display_keys) s_bag->s_display_keys->UseProgram();
|
233 |
} |
234 |
|
235 |
|
236 |
|
237 |
|
238 |
|
239 |
void ShaderMan::UseShaderGradientPass(int texP) |
240 |
{ |
241 |
s_bag->s_grad_pass->UseProgram(); |
242 |
s_bag->SetGradPassParam(texP); |
243 |
} |
244 |
|
245 |
|
246 |
void ShaderMan::UseShaderKeypoint(int texU, int texD) |
247 |
{ |
248 |
s_bag->s_keypoint->UseProgram(); |
249 |
s_bag->SetDogTexParam(texU, texD); |
250 |
} |
251 |
|
252 |
|
253 |
|
254 |
void ShaderMan::UseShaderGenListInit(int w, int h, int tight) |
255 |
{ |
256 |
if(tight)
|
257 |
{ |
258 |
s_bag->s_genlist_init_tight->UseProgram(); |
259 |
}else
|
260 |
{ |
261 |
s_bag->s_genlist_init_ex->UseProgram(); |
262 |
s_bag->SetGenListInitParam(w, h); |
263 |
} |
264 |
} |
265 |
|
266 |
void ShaderMan::UseShaderGenListHisto()
|
267 |
{ |
268 |
s_bag->s_genlist_histo->UseProgram(); |
269 |
|
270 |
} |
271 |
|
272 |
|
273 |
|
274 |
|
275 |
void ShaderMan::UseShaderGenListStart(float fw, int tex0) |
276 |
{ |
277 |
s_bag->s_genlist_start->UseProgram(); |
278 |
s_bag->SetGenListStartParam(fw, tex0); |
279 |
} |
280 |
|
281 |
void ShaderMan::UseShaderGenListStep(int tex, int tex0) |
282 |
{ |
283 |
s_bag->s_genlist_step->UseProgram(); |
284 |
s_bag->SetGenListStepParam( tex, tex0); |
285 |
} |
286 |
|
287 |
void ShaderMan::UseShaderGenListEnd(int ktex) |
288 |
{ |
289 |
s_bag->s_genlist_end->UseProgram(); |
290 |
s_bag->SetGenListEndParam(ktex); |
291 |
} |
292 |
|
293 |
void ShaderMan::UseShaderDebug()
|
294 |
{ |
295 |
if(s_bag->s_debug) s_bag->s_debug->UseProgram();
|
296 |
} |
297 |
|
298 |
void ShaderMan::UseShaderZeroPass()
|
299 |
{ |
300 |
if(s_bag->s_zero_pass) s_bag->s_zero_pass->UseProgram();
|
301 |
} |
302 |
|
303 |
void ShaderMan::UseShaderGenVBO( float width, float fwidth, float size) |
304 |
{ |
305 |
s_bag->s_vertex_list->UseProgram(); |
306 |
s_bag->SetGenVBOParam(width, fwidth, size); |
307 |
} |
308 |
void ShaderMan::UseShaderMarginCopy(int xmax, int ymax) |
309 |
{ |
310 |
s_bag->s_margin_copy->UseProgram(); |
311 |
s_bag->SetMarginCopyParam(xmax, ymax); |
312 |
|
313 |
} |
314 |
void ShaderMan::UseShaderCopyKeypoint()
|
315 |
{ |
316 |
s_bag->s_copy_key->UseProgram(); |
317 |
} |
318 |
|
319 |
void ShaderMan::UseShaderSimpleOrientation(int oTex, float sigma, float sigma_step) |
320 |
{ |
321 |
s_bag->s_orientation->UseProgram(); |
322 |
s_bag->SetSimpleOrientationInput(oTex, sigma, sigma_step); |
323 |
} |
324 |
|
325 |
|
326 |
|
327 |
void ShaderMan::UseShaderOrientation(int gtex, int width, int height, float sigma, int auxtex, float step, int keypoint_list) |
328 |
{ |
329 |
s_bag->s_orientation->UseProgram(); |
330 |
|
331 |
//changes in v345.
|
332 |
//set sigma to 0 to identify keypoit list mode
|
333 |
//set sigma to negative to identify fixed_orientation
|
334 |
if(keypoint_list) sigma = 0.0f; |
335 |
else if(GlobalUtil::_FixedOrientation) sigma = - sigma; |
336 |
|
337 |
s_bag->SetFeatureOrientationParam(gtex, width, height, sigma, auxtex, step); |
338 |
} |
339 |
|
340 |
void ShaderMan::UseShaderDescriptor(int gtex, int otex, int dwidth, int fwidth, int width, int height, float sigma) |
341 |
{ |
342 |
s_bag->s_descriptor_fp->UseProgram(); |
343 |
s_bag->SetFeatureDescirptorParam(gtex, otex, (float)dwidth, (float)fwidth, (float)width, (float)height, sigma); |
344 |
} |
345 |
|
346 |
void ShaderMan::SelectInitialSmoothingFilter(int octave_min, SiftParam¶m) |
347 |
{ |
348 |
s_bag->SelectInitialSmoothingFilter(octave_min, param); |
349 |
} |