src/mesa/state_tracker/st_atom_storagebuf.c

   1 /**************************************************************************
   2  *
   3  * Copyright 2014 Ilia Mirkin. All Rights Reserved.
   4  *
   5  * Permission is hereby granted, free of charge, to any person obtaining a
   6  * copy of this software and associated documentation files (the
   7  * "Software"), to deal in the Software without restriction, including
   8  * without limitation the rights to use, copy, modify, merge, publish,
   9  * distribute, sub license, and/or sell copies of the Software, and to
  10  * permit persons to whom the Software is furnished to do so, subject to
  11  * the following conditions:
  12  *
  13  * The above copyright notice and this permission notice (including the
  14  * next paragraph) shall be included in all copies or substantial portions
  15  * of the Software.
  16  *
  17  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  18  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  19  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
  20  * IN NO EVENT SHALL VMWARE AND/OR ITS SUPPLIERS BE LIABLE FOR
  21  * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
  22  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
  23  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  24  *
  25  **************************************************************************/
  26
  27 #include "main/imports.h"
  28 #include "program/prog_parameter.h"
  29 #include "program/prog_print.h"
  30 #include "compiler/glsl/ir_uniform.h"
  31
  32 #include "pipe/p_context.h"
  33 #include "pipe/p_defines.h"
  34 #include "util/u_inlines.h"
  35 #include "util/u_surface.h"
  36
  37 #include "st_debug.h"
  38 #include "st_cb_bufferobjects.h"
  39 #include "st_context.h"
  40 #include "st_atom.h"
  41 #include "st_program.h"
  42
  43 static void
  44 st_bind_ssbos(struct st_context *st, struct gl_program *prog,
  45               enum pipe_shader_type shader_type)
  46 {
  47    unsigned i;
  48    struct pipe_shader_buffer buffers[MAX_SHADER_STORAGE_BUFFERS];
  49    struct gl_program_constants *c;
  50    int buffer_base;
  51    if (!prog || !st->pipe->set_shader_buffers)
  52       return;
  53
  54    c = &st->ctx->Const.Program[prog->info.stage];
  55
  56    buffer_base = st->has_hw_atomics ? 0 : c->MaxAtomicBuffers;
  57
  58    for (i = 0; i < prog->info.num_ssbos; i++) {
  59       struct gl_buffer_binding *binding;
  60       struct st_buffer_object *st_obj;
  61       struct pipe_shader_buffer *sb = &buffers[i];
  62
  63       binding = &st->ctx->ShaderStorageBufferBindings[
  64             prog->sh.ShaderStorageBlocks[i]->Binding];
  65       st_obj = st_buffer_object(binding->BufferObject);
  66
  67       sb->buffer = st_obj->buffer;
  68
  69       if (sb->buffer) {
  70          sb->buffer_offset = binding->Offset;
  71          sb->buffer_size = sb->buffer->width0 - binding->Offset;
  72
  73          /* AutomaticSize is FALSE if the buffer was set with BindBufferRange.
  74           * Take the minimum just to be sure.
  75           */
  76          if (!binding->AutomaticSize)
  77             sb->buffer_size = MIN2(sb->buffer_size, (unsigned) binding->Size);
  78       }
  79       else {
  80          sb->buffer_offset = 0;
  81          sb->buffer_size = 0;
  82       }
  83    }
  84    st->pipe->set_shader_buffers(st->pipe, shader_type, buffer_base,
  85                                 prog->info.num_ssbos, buffers);
  86    /* clear out any stale shader buffers */
  87    if (prog->info.num_ssbos < c->MaxShaderStorageBlocks)
  88       st->pipe->set_shader_buffers(
  89             st->pipe, shader_type,
  90             buffer_base + prog->info.num_ssbos,
  91             c->MaxShaderStorageBlocks - prog->info.num_ssbos,
  92             NULL);
  93 }
  94
  95 void st_bind_vs_ssbos(struct st_context *st)
  96 {
  97    struct gl_program *prog =
  98       st->ctx->_Shader->CurrentProgram[MESA_SHADER_VERTEX];
  99
 100    st_bind_ssbos(st, prog, PIPE_SHADER_VERTEX);
 101 }
 102
 103 void st_bind_fs_ssbos(struct st_context *st)
 104 {
 105    struct gl_program *prog =
 106       st->ctx->_Shader->CurrentProgram[MESA_SHADER_FRAGMENT];
 107
 108    st_bind_ssbos(st, prog, PIPE_SHADER_FRAGMENT);
 109 }
 110
 111 void st_bind_gs_ssbos(struct st_context *st)
 112 {
 113    struct gl_program *prog =
 114       st->ctx->_Shader->CurrentProgram[MESA_SHADER_GEOMETRY];
 115
 116    st_bind_ssbos(st, prog, PIPE_SHADER_GEOMETRY);
 117 }
 118
 119 void st_bind_tcs_ssbos(struct st_context *st)
 120 {
 121    struct gl_program *prog =
 122       st->ctx->_Shader->CurrentProgram[MESA_SHADER_TESS_CTRL];
 123
 124    st_bind_ssbos(st, prog, PIPE_SHADER_TESS_CTRL);
 125 }
 126
 127 void st_bind_tes_ssbos(struct st_context *st)
 128 {
 129    struct gl_program *prog =
 130       st->ctx->_Shader->CurrentProgram[MESA_SHADER_TESS_EVAL];
 131
 132    st_bind_ssbos(st, prog, PIPE_SHADER_TESS_EVAL);
 133 }
 134
 135 void st_bind_cs_ssbos(struct st_context *st)
 136 {
 137    struct gl_program *prog =
 138       st->ctx->_Shader->CurrentProgram[MESA_SHADER_COMPUTE];
 139
 140    st_bind_ssbos(st, prog, PIPE_SHADER_COMPUTE);
 141 }