a8806c91edd4d40ed23f63b6bb3153c2a92d9595
[mesa.git] / src / mesa / state_tracker / st_cb_drawtex.c
1 /**************************************************************************
2 *
3 * Copyright 2008 Tungsten Graphics, Inc., Cedar Park, Texas.
4 * All Rights Reserved.
5 *
6 **************************************************************************/
7
8
9 /**
10 * Implementation of glDrawTex() for GL_OES_draw_tex
11 */
12
13
14
15 #include "main/imports.h"
16 #include "main/image.h"
17 #include "main/macros.h"
18 #include "main/mfeatures.h"
19 #include "program/program.h"
20 #include "program/prog_print.h"
21
22 #include "st_context.h"
23 #include "st_atom.h"
24 #include "st_cb_drawtex.h"
25
26 #include "pipe/p_context.h"
27 #include "pipe/p_defines.h"
28 #include "util/u_inlines.h"
29 #include "pipe/p_shader_tokens.h"
30 #include "util/u_draw_quad.h"
31 #include "util/u_simple_shaders.h"
32 #include "util/u_upload_mgr.h"
33
34 #include "cso_cache/cso_context.h"
35
36
37 struct cached_shader
38 {
39 void *handle;
40
41 uint num_attribs;
42 uint semantic_names[2 + MAX_TEXTURE_UNITS];
43 uint semantic_indexes[2 + MAX_TEXTURE_UNITS];
44 };
45
46 #define MAX_SHADERS (2 * MAX_TEXTURE_UNITS)
47
48 /**
49 * Simple linear list cache.
50 * Most of the time there'll only be one cached shader.
51 */
52 static struct cached_shader CachedShaders[MAX_SHADERS];
53 static GLuint NumCachedShaders = 0;
54
55
56 static void *
57 lookup_shader(struct pipe_context *pipe,
58 uint num_attribs,
59 const uint *semantic_names,
60 const uint *semantic_indexes)
61 {
62 GLuint i, j;
63
64 /* look for existing shader with same attributes */
65 for (i = 0; i < NumCachedShaders; i++) {
66 if (CachedShaders[i].num_attribs == num_attribs) {
67 GLboolean match = GL_TRUE;
68 for (j = 0; j < num_attribs; j++) {
69 if (semantic_names[j] != CachedShaders[i].semantic_names[j] ||
70 semantic_indexes[j] != CachedShaders[i].semantic_indexes[j]) {
71 match = GL_FALSE;
72 break;
73 }
74 }
75 if (match)
76 return CachedShaders[i].handle;
77 }
78 }
79
80 /* not found - create new one now */
81 if (NumCachedShaders >= MAX_SHADERS) {
82 return NULL;
83 }
84
85 CachedShaders[i].num_attribs = num_attribs;
86 for (j = 0; j < num_attribs; j++) {
87 CachedShaders[i].semantic_names[j] = semantic_names[j];
88 CachedShaders[i].semantic_indexes[j] = semantic_indexes[j];
89 }
90
91 CachedShaders[i].handle =
92 util_make_vertex_passthrough_shader(pipe,
93 num_attribs,
94 semantic_names,
95 semantic_indexes);
96 NumCachedShaders++;
97
98 return CachedShaders[i].handle;
99 }
100
101 static void
102 st_DrawTex(struct gl_context *ctx, GLfloat x, GLfloat y, GLfloat z,
103 GLfloat width, GLfloat height)
104 {
105 struct st_context *st = ctx->st;
106 struct pipe_context *pipe = st->pipe;
107 struct cso_context *cso = ctx->st->cso_context;
108 struct pipe_resource *vbuffer = NULL;
109 GLuint i, numTexCoords, numAttribs;
110 GLboolean emitColor;
111 uint semantic_names[2 + MAX_TEXTURE_UNITS];
112 uint semantic_indexes[2 + MAX_TEXTURE_UNITS];
113 struct pipe_vertex_element velements[2 + MAX_TEXTURE_UNITS];
114 unsigned offset;
115
116 st_validate_state(st);
117
118 /* determine if we need vertex color */
119 if (ctx->FragmentProgram._Current->Base.InputsRead & VARYING_BIT_COL0)
120 emitColor = GL_TRUE;
121 else
122 emitColor = GL_FALSE;
123
124 /* determine how many enabled sets of texcoords */
125 numTexCoords = 0;
126 for (i = 0; i < ctx->Const.MaxTextureUnits; i++) {
127 if (ctx->Texture.Unit[i]._ReallyEnabled & TEXTURE_2D_BIT) {
128 numTexCoords++;
129 }
130 }
131
132 /* total number of attributes per vertex */
133 numAttribs = 1 + emitColor + numTexCoords;
134
135 /* load vertex buffer */
136 {
137 #define SET_ATTRIB(VERT, ATTR, X, Y, Z, W) \
138 do { \
139 GLuint k = (((VERT) * numAttribs + (ATTR)) * 4); \
140 assert(k < 4 * 4 * numAttribs); \
141 vbuf[k + 0] = X; \
142 vbuf[k + 1] = Y; \
143 vbuf[k + 2] = Z; \
144 vbuf[k + 3] = W; \
145 } while (0)
146
147 const GLfloat x0 = x, y0 = y, x1 = x + width, y1 = y + height;
148 GLfloat *vbuf = NULL;
149 GLuint attr;
150
151 if (u_upload_alloc(st->uploader, 0,
152 numAttribs * 4 * 4 * sizeof(GLfloat),
153 &offset, &vbuffer, (void **) &vbuf) != PIPE_OK) {
154 return;
155 }
156
157 z = CLAMP(z, 0.0f, 1.0f);
158
159 /* positions (in clip coords) */
160 {
161 const struct gl_framebuffer *fb = st->ctx->DrawBuffer;
162 const GLfloat fb_width = (GLfloat)fb->Width;
163 const GLfloat fb_height = (GLfloat)fb->Height;
164
165 const GLfloat clip_x0 = (GLfloat)(x0 / fb_width * 2.0 - 1.0);
166 const GLfloat clip_y0 = (GLfloat)(y0 / fb_height * 2.0 - 1.0);
167 const GLfloat clip_x1 = (GLfloat)(x1 / fb_width * 2.0 - 1.0);
168 const GLfloat clip_y1 = (GLfloat)(y1 / fb_height * 2.0 - 1.0);
169
170 SET_ATTRIB(0, 0, clip_x0, clip_y0, z, 1.0f); /* lower left */
171 SET_ATTRIB(1, 0, clip_x1, clip_y0, z, 1.0f); /* lower right */
172 SET_ATTRIB(2, 0, clip_x1, clip_y1, z, 1.0f); /* upper right */
173 SET_ATTRIB(3, 0, clip_x0, clip_y1, z, 1.0f); /* upper left */
174
175 semantic_names[0] = TGSI_SEMANTIC_POSITION;
176 semantic_indexes[0] = 0;
177 }
178
179 /* colors */
180 if (emitColor) {
181 const GLfloat *c = ctx->Current.Attrib[VERT_ATTRIB_COLOR0];
182 SET_ATTRIB(0, 1, c[0], c[1], c[2], c[3]);
183 SET_ATTRIB(1, 1, c[0], c[1], c[2], c[3]);
184 SET_ATTRIB(2, 1, c[0], c[1], c[2], c[3]);
185 SET_ATTRIB(3, 1, c[0], c[1], c[2], c[3]);
186 semantic_names[1] = TGSI_SEMANTIC_COLOR;
187 semantic_indexes[1] = 0;
188 attr = 2;
189 }
190 else {
191 attr = 1;
192 }
193
194 /* texcoords */
195 for (i = 0; i < ctx->Const.MaxTextureUnits; i++) {
196 if (ctx->Texture.Unit[i]._ReallyEnabled & TEXTURE_2D_BIT) {
197 struct gl_texture_object *obj = ctx->Texture.Unit[i]._Current;
198 struct gl_texture_image *img = obj->Image[0][obj->BaseLevel];
199 const GLfloat wt = (GLfloat) img->Width;
200 const GLfloat ht = (GLfloat) img->Height;
201 const GLfloat s0 = obj->CropRect[0] / wt;
202 const GLfloat t0 = obj->CropRect[1] / ht;
203 const GLfloat s1 = (obj->CropRect[0] + obj->CropRect[2]) / wt;
204 const GLfloat t1 = (obj->CropRect[1] + obj->CropRect[3]) / ht;
205
206 /*printf("crop texcoords: %g, %g .. %g, %g\n", s0, t0, s1, t1);*/
207 SET_ATTRIB(0, attr, s0, t0, 0.0f, 1.0f); /* lower left */
208 SET_ATTRIB(1, attr, s1, t0, 0.0f, 1.0f); /* lower right */
209 SET_ATTRIB(2, attr, s1, t1, 0.0f, 1.0f); /* upper right */
210 SET_ATTRIB(3, attr, s0, t1, 0.0f, 1.0f); /* upper left */
211
212 semantic_names[attr] = TGSI_SEMANTIC_GENERIC;
213 semantic_indexes[attr] = 0;
214
215 attr++;
216 }
217 }
218
219 u_upload_unmap(st->uploader);
220
221 #undef SET_ATTRIB
222 }
223
224
225 cso_save_viewport(cso);
226 cso_save_stream_outputs(cso);
227 cso_save_vertex_shader(cso);
228 cso_save_geometry_shader(cso);
229 cso_save_vertex_elements(cso);
230 cso_save_aux_vertex_buffer_slot(cso);
231
232 {
233 void *vs = lookup_shader(pipe, numAttribs,
234 semantic_names, semantic_indexes);
235 cso_set_vertex_shader_handle(cso, vs);
236 }
237 cso_set_geometry_shader_handle(cso, NULL);
238
239 for (i = 0; i < numAttribs; i++) {
240 velements[i].src_offset = i * 4 * sizeof(float);
241 velements[i].instance_divisor = 0;
242 velements[i].vertex_buffer_index = 0;
243 velements[i].src_format = PIPE_FORMAT_R32G32B32A32_FLOAT;
244 }
245 cso_set_vertex_elements(cso, numAttribs, velements);
246 cso_set_stream_outputs(st->cso_context, 0, NULL, 0);
247
248 /* viewport state: viewport matching window dims */
249 {
250 const struct gl_framebuffer *fb = st->ctx->DrawBuffer;
251 const GLboolean invert = (st_fb_orientation(fb) == Y_0_TOP);
252 const GLfloat width = (GLfloat)fb->Width;
253 const GLfloat height = (GLfloat)fb->Height;
254 struct pipe_viewport_state vp;
255 vp.scale[0] = 0.5f * width;
256 vp.scale[1] = height * (invert ? -0.5f : 0.5f);
257 vp.scale[2] = 1.0f;
258 vp.scale[3] = 1.0f;
259 vp.translate[0] = 0.5f * width;
260 vp.translate[1] = 0.5f * height;
261 vp.translate[2] = 0.0f;
262 vp.translate[3] = 0.0f;
263 cso_set_viewport(cso, &vp);
264 }
265
266
267 util_draw_vertex_buffer(pipe, cso, vbuffer,
268 cso_get_aux_vertex_buffer_slot(cso),
269 offset, /* offset */
270 PIPE_PRIM_TRIANGLE_FAN,
271 4, /* verts */
272 numAttribs); /* attribs/vert */
273
274
275 pipe_resource_reference(&vbuffer, NULL);
276
277 /* restore state */
278 cso_restore_viewport(cso);
279 cso_restore_vertex_shader(cso);
280 cso_restore_geometry_shader(cso);
281 cso_restore_vertex_elements(cso);
282 cso_restore_aux_vertex_buffer_slot(cso);
283 cso_restore_stream_outputs(cso);
284 }
285
286
287 void
288 st_init_drawtex_functions(struct dd_function_table *functions)
289 {
290 functions->DrawTex = st_DrawTex;
291 }
292
293
294 /**
295 * Free any cached shaders
296 */
297 void
298 st_destroy_drawtex(struct st_context *st)
299 {
300 GLuint i;
301 for (i = 0; i < NumCachedShaders; i++) {
302 cso_delete_vertex_shader(st->cso_context, CachedShaders[i].handle);
303 }
304 NumCachedShaders = 0;
305 }