src/mesa/state_tracker/st_atom_constbuf.c

   1 /**************************************************************************
   2  *
   3  * Copyright 2007 VMware, Inc.
   4  * All Rights Reserved.
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a
   7  * copy of this software and associated documentation files (the
   8  * "Software"), to deal in the Software without restriction, including
   9  * without limitation the rights to use, copy, modify, merge, publish,
  10  * distribute, sub license, and/or sell copies of the Software, and to
  11  * permit persons to whom the Software is furnished to do so, subject to
  12  * the following conditions:
  13  *
  14  * The above copyright notice and this permission notice (including the
  15  * next paragraph) shall be included in all copies or substantial portions
  16  * of the Software.
  17  *
  18  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  19  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  20  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
  21  * IN NO EVENT SHALL VMWARE AND/OR ITS SUPPLIERS BE LIABLE FOR
  22  * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
  23  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
  24  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  25  *
  26  **************************************************************************/
  27
  28 /*
  29  * Authors:
  30  *   Keith Whitwell <keithw@vmware.com>
  31  *   Brian Paul
  32  */
  33
  34 #include "main/imports.h"
  35 #include "program/prog_parameter.h"
  36 #include "program/prog_print.h"
  37 #include "main/shaderapi.h"
  38 #include "pipe/p_context.h"
  39 #include "pipe/p_defines.h"
  40 #include "util/u_inlines.h"
  41 #include "util/u_upload_mgr.h"
  42 #include "cso_cache/cso_context.h"
  43
  44 #include "st_debug.h"
  45 #include "st_context.h"
  46 #include "st_atom.h"
  47 #include "st_atom_constbuf.h"
  48 #include "st_program.h"
  49 #include "st_cb_bufferobjects.h"
  50
  51 /**
  52  * Pass the given program parameters to the graphics pipe as a
  53  * constant buffer.
  54  */
  55 void st_upload_constants(struct st_context *st, struct gl_program *prog)
  56 {
  57    gl_shader_stage stage = prog->info.stage;
  58    struct gl_program_parameter_list *params = prog->Parameters;
  59    enum pipe_shader_type shader_type = pipe_shader_type_from_mesa(stage);
  60
  61    assert(shader_type == PIPE_SHADER_VERTEX ||
  62           shader_type == PIPE_SHADER_FRAGMENT ||
  63           shader_type == PIPE_SHADER_GEOMETRY ||
  64           shader_type == PIPE_SHADER_TESS_CTRL ||
  65           shader_type == PIPE_SHADER_TESS_EVAL ||
  66           shader_type == PIPE_SHADER_COMPUTE);
  67
  68    /* update the ATI constants before rendering */
  69    if (shader_type == PIPE_SHADER_FRAGMENT && st->fp->ati_fs) {
  70       struct ati_fragment_shader *ati_fs = st->fp->ati_fs;
  71       unsigned c;
  72
  73       for (c = 0; c < MAX_NUM_FRAGMENT_CONSTANTS_ATI; c++) {
  74          if (ati_fs->LocalConstDef & (1 << c))
  75             memcpy(params->ParameterValues[c],
  76                    ati_fs->Constants[c], sizeof(GLfloat) * 4);
  77          else
  78             memcpy(params->ParameterValues[c],
  79                    st->ctx->ATIFragmentShader.GlobalConstants[c], sizeof(GLfloat) * 4);
  80       }
  81    }
  82
  83    /* Make all bindless samplers/images bound texture/image units resident in
  84     * the context.
  85     */
  86    st_make_bound_samplers_resident(st, prog);
  87    st_make_bound_images_resident(st, prog);
  88
  89    /* update constants */
  90    if (params && params->NumParameters) {
  91       struct pipe_constant_buffer cb;
  92       const uint paramBytes = params->NumParameters * sizeof(GLfloat) * 4;
  93
  94       /* Update the constants which come from fixed-function state, such as
  95        * transformation matrices, fog factors, etc.  The rest of the values in
  96        * the parameters list are explicitly set by the user with glUniform,
  97        * glProgramParameter(), etc.
  98        */
  99       if (params->StateFlags)
 100          _mesa_load_state_parameters(st->ctx, params);
 101
 102       _mesa_shader_write_subroutine_indices(st->ctx, stage);
 103
 104       /* We always need to get a new buffer, to keep the drivers simple and
 105        * avoid gratuitous rendering synchronization.
 106        * Let's use a user buffer to avoid an unnecessary copy.
 107        */
 108       if (!st->has_user_constbuf) {
 109          cb.buffer = NULL;
 110          cb.user_buffer = NULL;
 111          u_upload_data(st->pipe->const_uploader, 0, paramBytes,
 112                        st->ctx->Const.UniformBufferOffsetAlignment,
 113                        params->ParameterValues, &cb.buffer_offset, &cb.buffer);
 114          u_upload_unmap(st->pipe->const_uploader);
 115       } else {
 116          cb.buffer = NULL;
 117          cb.user_buffer = params->ParameterValues;
 118          cb.buffer_offset = 0;
 119       }
 120       cb.buffer_size = paramBytes;
 121
 122       if (ST_DEBUG & DEBUG_CONSTANTS) {
 123          debug_printf("%s(shader=%d, numParams=%d, stateFlags=0x%x)\n",
 124                       __func__, shader_type, params->NumParameters,
 125                       params->StateFlags);
 126          _mesa_print_parameter_list(params);
 127       }
 128
 129       cso_set_constant_buffer(st->cso_context, shader_type, 0, &cb);
 130       pipe_resource_reference(&cb.buffer, NULL);
 131
 132       st->state.constants[shader_type].ptr = params->ParameterValues;
 133       st->state.constants[shader_type].size = paramBytes;
 134    }
 135    else if (st->state.constants[shader_type].ptr) {
 136       /* Unbind. */
 137       st->state.constants[shader_type].ptr = NULL;
 138       st->state.constants[shader_type].size = 0;
 139       cso_set_constant_buffer(st->cso_context, shader_type, 0, NULL);
 140    }
 141 }
 142
 143
 144 /**
 145  * Vertex shader:
 146  */
 147 void st_update_vs_constants(struct st_context *st )
 148 {
 149    st_upload_constants(st, &st->vp->Base);
 150 }
 151
 152 /**
 153  * Fragment shader:
 154  */
 155 void st_update_fs_constants(struct st_context *st )
 156 {
 157    st_upload_constants(st, &st->fp->Base);
 158 }
 159
 160
 161 /* Geometry shader:
 162  */
 163 void st_update_gs_constants(struct st_context *st )
 164 {
 165    struct st_common_program *gp = st->gp;
 166
 167    if (gp)
 168       st_upload_constants(st, &gp->Base);
 169 }
 170
 171 /* Tessellation control shader:
 172  */
 173 void st_update_tcs_constants(struct st_context *st )
 174 {
 175    struct st_common_program *tcp = st->tcp;
 176
 177    if (tcp)
 178       st_upload_constants(st, &tcp->Base);
 179 }
 180
 181 /* Tessellation evaluation shader:
 182  */
 183 void st_update_tes_constants(struct st_context *st )
 184 {
 185    struct st_common_program *tep = st->tep;
 186
 187    if (tep)
 188       st_upload_constants(st, &tep->Base);
 189 }
 190
 191 /* Compute shader:
 192  */
 193 void st_update_cs_constants(struct st_context *st )
 194 {
 195    struct st_compute_program *cp = st->cp;
 196
 197    if (cp)
 198       st_upload_constants(st, &cp->Base);
 199 }
 200
 201 static void st_bind_ubos(struct st_context *st, struct gl_program *prog,
 202                          unsigned shader_type)
 203 {
 204    unsigned i;
 205    struct pipe_constant_buffer cb = { 0 };
 206
 207    if (!prog)
 208       return;
 209
 210    for (i = 0; i < prog->info.num_ubos; i++) {
 211       struct gl_buffer_binding *binding;
 212       struct st_buffer_object *st_obj;
 213
 214       binding =
 215          &st->ctx->UniformBufferBindings[prog->sh.UniformBlocks[i]->Binding];
 216       st_obj = st_buffer_object(binding->BufferObject);
 217
 218       cb.buffer = st_obj->buffer;
 219
 220       if (cb.buffer) {
 221          cb.buffer_offset = binding->Offset;
 222          cb.buffer_size = cb.buffer->width0 - binding->Offset;
 223
 224          /* AutomaticSize is FALSE if the buffer was set with BindBufferRange.
 225           * Take the minimum just to be sure.
 226           */
 227          if (!binding->AutomaticSize)
 228             cb.buffer_size = MIN2(cb.buffer_size, (unsigned) binding->Size);
 229       }
 230       else {
 231          cb.buffer_offset = 0;
 232          cb.buffer_size = 0;
 233       }
 234
 235       cso_set_constant_buffer(st->cso_context, shader_type, 1 + i, &cb);
 236    }
 237 }
 238
 239 void st_bind_vs_ubos(struct st_context *st)
 240 {
 241    struct gl_program *prog =
 242       st->ctx->_Shader->CurrentProgram[MESA_SHADER_VERTEX];
 243
 244    st_bind_ubos(st, prog, PIPE_SHADER_VERTEX);
 245 }
 246
 247 void st_bind_fs_ubos(struct st_context *st)
 248 {
 249    struct gl_program *prog =
 250       st->ctx->_Shader->CurrentProgram[MESA_SHADER_FRAGMENT];
 251
 252    st_bind_ubos(st, prog, PIPE_SHADER_FRAGMENT);
 253 }
 254
 255 void st_bind_gs_ubos(struct st_context *st)
 256 {
 257    struct gl_program *prog =
 258       st->ctx->_Shader->CurrentProgram[MESA_SHADER_GEOMETRY];
 259
 260    st_bind_ubos(st, prog, PIPE_SHADER_GEOMETRY);
 261 }
 262
 263 void st_bind_tcs_ubos(struct st_context *st)
 264 {
 265    struct gl_program *prog =
 266       st->ctx->_Shader->CurrentProgram[MESA_SHADER_TESS_CTRL];
 267
 268    st_bind_ubos(st, prog, PIPE_SHADER_TESS_CTRL);
 269 }
 270
 271 void st_bind_tes_ubos(struct st_context *st)
 272 {
 273    struct gl_program *prog =
 274       st->ctx->_Shader->CurrentProgram[MESA_SHADER_TESS_EVAL];
 275
 276    st_bind_ubos(st, prog, PIPE_SHADER_TESS_EVAL);
 277 }
 278
 279 void st_bind_cs_ubos(struct st_context *st)
 280 {
 281    struct gl_program *prog =
 282       st->ctx->_Shader->CurrentProgram[MESA_SHADER_COMPUTE];
 283
 284    st_bind_ubos(st, prog, PIPE_SHADER_COMPUTE);
 285 }