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_linked_shader *shader,
  45               unsigned shader_type)
  46 {
  47    unsigned i;
  48    struct pipe_shader_buffer buffers[MAX_SHADER_STORAGE_BUFFERS];
  49    struct gl_program_constants *c;
  50
  51    if (!shader || !st->pipe->set_shader_buffers)
  52       return;
  53
  54    c = &st->ctx->Const.Program[shader->Stage];
  55
  56    for (i = 0; i < shader->NumShaderStorageBlocks; i++) {
  57       struct gl_shader_storage_buffer_binding *binding;
  58       struct st_buffer_object *st_obj;
  59       struct pipe_shader_buffer *sb = &buffers[i];
  60
  61       binding = &st->ctx->ShaderStorageBufferBindings[
  62             shader->ShaderStorageBlocks[i]->Binding];
  63       st_obj = st_buffer_object(binding->BufferObject);
  64
  65       sb->buffer = st_obj->buffer;
  66
  67       if (sb->buffer) {
  68          sb->buffer_offset = binding->Offset;
  69          sb->buffer_size = sb->buffer->width0 - binding->Offset;
  70
  71          /* AutomaticSize is FALSE if the buffer was set with BindBufferRange.
  72           * Take the minimum just to be sure.
  73           */
  74          if (!binding->AutomaticSize)
  75             sb->buffer_size = MIN2(sb->buffer_size, (unsigned) binding->Size);
  76       }
  77       else {
  78          sb->buffer_offset = 0;
  79          sb->buffer_size = 0;
  80       }
  81    }
  82    st->pipe->set_shader_buffers(st->pipe, shader_type, c->MaxAtomicBuffers,
  83                                 shader->NumShaderStorageBlocks, buffers);
  84    /* clear out any stale shader buffers */
  85    if (shader->NumShaderStorageBlocks < c->MaxShaderStorageBlocks)
  86       st->pipe->set_shader_buffers(
  87             st->pipe, shader_type,
  88             c->MaxAtomicBuffers + shader->NumShaderStorageBlocks,
  89             c->MaxShaderStorageBlocks - shader->NumShaderStorageBlocks,
  90             NULL);
  91 }
  92
  93 static void bind_vs_ssbos(struct st_context *st)
  94 {
  95    struct gl_shader_program *prog =
  96       st->ctx->_Shader->CurrentProgram[MESA_SHADER_VERTEX];
  97
  98    if (!prog)
  99       return;
 100
 101    st_bind_ssbos(st, prog->_LinkedShaders[MESA_SHADER_VERTEX],
 102                  PIPE_SHADER_VERTEX);
 103 }
 104
 105 const struct st_tracked_state st_bind_vs_ssbos = {
 106    "st_bind_vs_ssbos",
 107    {
 108       0,
 109       ST_NEW_VERTEX_PROGRAM | ST_NEW_STORAGE_BUFFER,
 110    },
 111    bind_vs_ssbos
 112 };
 113
 114 static void bind_fs_ssbos(struct st_context *st)
 115 {
 116    struct gl_shader_program *prog =
 117       st->ctx->_Shader->CurrentProgram[MESA_SHADER_FRAGMENT];
 118
 119    if (!prog)
 120       return;
 121
 122    st_bind_ssbos(st, prog->_LinkedShaders[MESA_SHADER_FRAGMENT],
 123                  PIPE_SHADER_FRAGMENT);
 124 }
 125
 126 const struct st_tracked_state st_bind_fs_ssbos = {
 127    "st_bind_fs_ssbos",
 128    {
 129       0,
 130       ST_NEW_FRAGMENT_PROGRAM | ST_NEW_STORAGE_BUFFER,
 131    },
 132    bind_fs_ssbos
 133 };
 134
 135 static void bind_gs_ssbos(struct st_context *st)
 136 {
 137    struct gl_shader_program *prog =
 138       st->ctx->_Shader->CurrentProgram[MESA_SHADER_GEOMETRY];
 139
 140    if (!prog)
 141       return;
 142
 143    st_bind_ssbos(st, prog->_LinkedShaders[MESA_SHADER_GEOMETRY],
 144                  PIPE_SHADER_GEOMETRY);
 145 }
 146
 147 const struct st_tracked_state st_bind_gs_ssbos = {
 148    "st_bind_gs_ssbos",
 149    {
 150       0,
 151       ST_NEW_GEOMETRY_PROGRAM | ST_NEW_STORAGE_BUFFER,
 152    },
 153    bind_gs_ssbos
 154 };
 155
 156 static void bind_tcs_ssbos(struct st_context *st)
 157 {
 158    struct gl_shader_program *prog =
 159       st->ctx->_Shader->CurrentProgram[MESA_SHADER_TESS_CTRL];
 160
 161    if (!prog)
 162       return;
 163
 164    st_bind_ssbos(st, prog->_LinkedShaders[MESA_SHADER_TESS_CTRL],
 165                  PIPE_SHADER_TESS_CTRL);
 166 }
 167
 168 const struct st_tracked_state st_bind_tcs_ssbos = {
 169    "st_bind_tcs_ssbos",
 170    {
 171       0,
 172       ST_NEW_TESSCTRL_PROGRAM | ST_NEW_STORAGE_BUFFER,
 173    },
 174    bind_tcs_ssbos
 175 };
 176
 177 static void bind_tes_ssbos(struct st_context *st)
 178 {
 179    struct gl_shader_program *prog =
 180       st->ctx->_Shader->CurrentProgram[MESA_SHADER_TESS_EVAL];
 181
 182    if (!prog)
 183       return;
 184
 185    st_bind_ssbos(st, prog->_LinkedShaders[MESA_SHADER_TESS_EVAL],
 186                  PIPE_SHADER_TESS_EVAL);
 187 }
 188
 189 const struct st_tracked_state st_bind_tes_ssbos = {
 190    "st_bind_tes_ssbos",
 191    {
 192       0,
 193       ST_NEW_TESSEVAL_PROGRAM | ST_NEW_STORAGE_BUFFER,
 194    },
 195    bind_tes_ssbos
 196 };
 197
 198 static void bind_cs_ssbos(struct st_context *st)
 199 {
 200    struct gl_shader_program *prog =
 201       st->ctx->_Shader->CurrentProgram[MESA_SHADER_COMPUTE];
 202
 203    if (!prog)
 204       return;
 205
 206    st_bind_ssbos(st, prog->_LinkedShaders[MESA_SHADER_COMPUTE],
 207                  PIPE_SHADER_COMPUTE);
 208 }
 209
 210 const struct st_tracked_state st_bind_cs_ssbos = {
 211    "st_bind_cs_ssbos",
 212    {
 213       0,
 214       ST_NEW_COMPUTE_PROGRAM | ST_NEW_STORAGE_BUFFER,
 215    },
 216    bind_cs_ssbos
 217 };