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
  33 #include "cso_cache/cso_context.h"
  34
  35
  36 #if FEATURE_OES_draw_texture
  37
  38
  39 struct cached_shader
  40 {
  41    void *handle;
  42
  43    uint num_attribs;
  44    uint semantic_names[2 + MAX_TEXTURE_UNITS];
  45    uint semantic_indexes[2 + MAX_TEXTURE_UNITS];
  46 };
  47
  48 #define MAX_SHADERS (2 * MAX_TEXTURE_UNITS)
  49
  50 /**
  51  * Simple linear list cache.
  52  * Most of the time there'll only be one cached shader.
  53  */
  54 static struct cached_shader CachedShaders[MAX_SHADERS];
  55 static GLuint NumCachedShaders = 0;
  56
  57
  58 static void *
  59 lookup_shader(struct pipe_context *pipe,
  60               uint num_attribs,
  61               const uint *semantic_names,
  62               const uint *semantic_indexes)
  63 {
  64    GLuint i, j;
  65
  66    /* look for existing shader with same attributes */
  67    for (i = 0; i < NumCachedShaders; i++) {
  68       if (CachedShaders[i].num_attribs == num_attribs) {
  69          GLboolean match = GL_TRUE;
  70          for (j = 0; j < num_attribs; j++) {
  71             if (semantic_names[j] != CachedShaders[i].semantic_names[j] ||
  72                 semantic_indexes[j] != CachedShaders[i].semantic_indexes[j]) {
  73                match = GL_FALSE;
  74                break;
  75             }
  76          }
  77          if (match)
  78             return CachedShaders[i].handle;
  79       }
  80    }
  81
  82    /* not found - create new one now */
  83    if (NumCachedShaders >= MAX_SHADERS) {
  84       return NULL;
  85    }
  86
  87    CachedShaders[i].num_attribs = num_attribs;
  88    for (j = 0; j < num_attribs; j++) {
  89       CachedShaders[i].semantic_names[j] = semantic_names[j];
  90       CachedShaders[i].semantic_indexes[j] = semantic_indexes[j];
  91    }
  92
  93    CachedShaders[i].handle =
  94       util_make_vertex_passthrough_shader(pipe,
  95                                           num_attribs,
  96                                           semantic_names,
  97                                           semantic_indexes);
  98    NumCachedShaders++;
  99
 100    return CachedShaders[i].handle;
 101 }
 102
 103 static void
 104 st_DrawTex(struct gl_context *ctx, GLfloat x, GLfloat y, GLfloat z,
 105            GLfloat width, GLfloat height)
 106 {
 107    struct st_context *st = ctx->st;
 108    struct pipe_context *pipe = st->pipe;
 109    struct cso_context *cso = ctx->st->cso_context;
 110    struct pipe_resource *vbuffer;
 111    struct pipe_transfer *vbuffer_transfer;
 112    GLuint i, numTexCoords, numAttribs;
 113    GLboolean emitColor;
 114    uint semantic_names[2 + MAX_TEXTURE_UNITS];
 115    uint semantic_indexes[2 + MAX_TEXTURE_UNITS];
 116    struct pipe_vertex_element velements[2 + MAX_TEXTURE_UNITS];
 117
 118    st_validate_state(st);
 119
 120    /* determine if we need vertex color */
 121    if (ctx->FragmentProgram._Current->Base.InputsRead & FRAG_BIT_COL0)
 122       emitColor = GL_TRUE;
 123    else
 124       emitColor = GL_FALSE;
 125
 126    /* determine how many enabled sets of texcoords */
 127    numTexCoords = 0;
 128    for (i = 0; i < ctx->Const.MaxTextureUnits; i++) {
 129       if (ctx->Texture.Unit[i]._ReallyEnabled & TEXTURE_2D_BIT) {
 130          numTexCoords++;
 131       }
 132    }
 133
 134    /* total number of attributes per vertex */
 135    numAttribs = 1 + emitColor + numTexCoords;
 136
 137
 138    /* create the vertex buffer */
 139    vbuffer = pipe_buffer_create(pipe->screen, PIPE_BIND_VERTEX_BUFFER,
 140                                 PIPE_USAGE_STREAM,
 141                                 numAttribs * 4 * 4 * sizeof(GLfloat));
 142
 143    /* load vertex buffer */
 144    {
 145 #define SET_ATTRIB(VERT, ATTR, X, Y, Z, W)                              \
 146       do {                                                              \
 147          GLuint k = (((VERT) * numAttribs + (ATTR)) * 4);               \
 148          assert(k < 4 * 4 * numAttribs);                                \
 149          vbuf[k + 0] = X;                                               \
 150          vbuf[k + 1] = Y;                                               \
 151          vbuf[k + 2] = Z;                                               \
 152          vbuf[k + 3] = W;                                               \
 153       } while (0)
 154
 155       const GLfloat x0 = x, y0 = y, x1 = x + width, y1 = y + height;
 156       GLfloat *vbuf = (GLfloat *) pipe_buffer_map(pipe, vbuffer,
 157                                                   PIPE_TRANSFER_WRITE,
 158                                                   &vbuffer_transfer);
 159       GLuint attr;
 160
 161       z = CLAMP(z, 0.0f, 1.0f);
 162
 163       /* positions (in clip coords) */
 164       {
 165          const struct gl_framebuffer *fb = st->ctx->DrawBuffer;
 166          const GLfloat fb_width = (GLfloat)fb->Width;
 167          const GLfloat fb_height = (GLfloat)fb->Height;
 168
 169          const GLfloat clip_x0 = (GLfloat)(x0 / fb_width * 2.0 - 1.0);
 170          const GLfloat clip_y0 = (GLfloat)(y0 / fb_height * 2.0 - 1.0);
 171          const GLfloat clip_x1 = (GLfloat)(x1 / fb_width * 2.0 - 1.0);
 172          const GLfloat clip_y1 = (GLfloat)(y1 / fb_height * 2.0 - 1.0);
 173
 174          SET_ATTRIB(0, 0, clip_x0, clip_y0, z, 1.0f);   /* lower left */
 175          SET_ATTRIB(1, 0, clip_x1, clip_y0, z, 1.0f);   /* lower right */
 176          SET_ATTRIB(2, 0, clip_x1, clip_y1, z, 1.0f);   /* upper right */
 177          SET_ATTRIB(3, 0, clip_x0, clip_y1, z, 1.0f);   /* upper left */
 178
 179          semantic_names[0] = TGSI_SEMANTIC_POSITION;
 180          semantic_indexes[0] = 0;
 181       }
 182
 183       /* colors */
 184       if (emitColor) {
 185          const GLfloat *c = ctx->Current.Attrib[VERT_ATTRIB_COLOR0];
 186          SET_ATTRIB(0, 1, c[0], c[1], c[2], c[3]);
 187          SET_ATTRIB(1, 1, c[0], c[1], c[2], c[3]);
 188          SET_ATTRIB(2, 1, c[0], c[1], c[2], c[3]);
 189          SET_ATTRIB(3, 1, c[0], c[1], c[2], c[3]);
 190          semantic_names[1] = TGSI_SEMANTIC_COLOR;
 191          semantic_indexes[1] = 0;
 192          attr = 2;
 193       }
 194       else {
 195          attr = 1;
 196       }
 197
 198       /* texcoords */
 199       for (i = 0; i < ctx->Const.MaxTextureUnits; i++) {
 200          if (ctx->Texture.Unit[i]._ReallyEnabled & TEXTURE_2D_BIT) {
 201             struct gl_texture_object *obj = ctx->Texture.Unit[i]._Current;
 202             struct gl_texture_image *img = obj->Image[0][obj->BaseLevel];
 203             const GLfloat wt = (GLfloat) img->Width;
 204             const GLfloat ht = (GLfloat) img->Height;
 205             const GLfloat s0 = obj->CropRect[0] / wt;
 206             const GLfloat t0 = obj->CropRect[1] / ht;
 207             const GLfloat s1 = (obj->CropRect[0] + obj->CropRect[2]) / wt;
 208             const GLfloat t1 = (obj->CropRect[1] + obj->CropRect[3]) / ht;
 209
 210             /*printf("crop texcoords: %g, %g .. %g, %g\n", s0, t0, s1, t1);*/
 211             SET_ATTRIB(0, attr, s0, t0, 0.0f, 1.0f);  /* lower left */
 212             SET_ATTRIB(1, attr, s1, t0, 0.0f, 1.0f);  /* lower right */
 213             SET_ATTRIB(2, attr, s1, t1, 0.0f, 1.0f);  /* upper right */
 214             SET_ATTRIB(3, attr, s0, t1, 0.0f, 1.0f);  /* upper left */
 215
 216             semantic_names[attr] = TGSI_SEMANTIC_GENERIC;
 217             semantic_indexes[attr] = 0;
 218
 219             attr++;
 220          }
 221       }
 222
 223       pipe_buffer_unmap(pipe, vbuffer_transfer);
 224
 225 #undef SET_ATTRIB
 226    }
 227
 228
 229    cso_save_viewport(cso);
 230    cso_save_vertex_shader(cso);
 231    cso_save_geometry_shader(cso);
 232    cso_save_vertex_elements(cso);
 233    cso_save_vertex_buffers(cso);
 234
 235    {
 236       void *vs = lookup_shader(pipe, numAttribs,
 237                                semantic_names, semantic_indexes);
 238       cso_set_vertex_shader_handle(cso, vs);
 239    }
 240    cso_set_geometry_shader_handle(cso, NULL);
 241
 242    for (i = 0; i < numAttribs; i++) {
 243       velements[i].src_offset = i * 4 * sizeof(float);
 244       velements[i].instance_divisor = 0;
 245       velements[i].vertex_buffer_index = 0;
 246       velements[i].src_format = PIPE_FORMAT_R32G32B32A32_FLOAT;
 247    }
 248    cso_set_vertex_elements(cso, numAttribs, velements);
 249
 250    /* viewport state: viewport matching window dims */
 251    {
 252       const struct gl_framebuffer *fb = st->ctx->DrawBuffer;
 253       const GLboolean invert = (st_fb_orientation(fb) == Y_0_TOP);
 254       const GLfloat width = (GLfloat)fb->Width;
 255       const GLfloat height = (GLfloat)fb->Height;
 256       struct pipe_viewport_state vp;
 257       vp.scale[0] =  0.5f * width;
 258       vp.scale[1] = height * (invert ? -0.5f : 0.5f);
 259       vp.scale[2] = 1.0f;
 260       vp.scale[3] = 1.0f;
 261       vp.translate[0] = 0.5f * width;
 262       vp.translate[1] = 0.5f * height;
 263       vp.translate[2] = 0.0f;
 264       vp.translate[3] = 0.0f;
 265       cso_set_viewport(cso, &vp);
 266    }
 267
 268
 269    util_draw_vertex_buffer(pipe, cso, vbuffer,
 270                            0,  /* offset */
 271                            PIPE_PRIM_TRIANGLE_FAN,
 272                            4,  /* verts */
 273                            numAttribs); /* attribs/vert */
 274
 275
 276    pipe_resource_reference(&vbuffer, NULL);
 277
 278    /* restore state */
 279    cso_restore_viewport(cso);
 280    cso_restore_vertex_shader(cso);
 281    cso_restore_geometry_shader(cso);
 282    cso_restore_vertex_elements(cso);
 283    cso_restore_vertex_buffers(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 }
 306
 307
 308 #endif /* FEATURE_OES_draw_texture */