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
43static void
44st_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                                prog->sh.ShaderStorageBlocksWriteAccess);
87   /* clear out any stale shader buffers */
88   if (prog->info.num_ssbos < c->MaxShaderStorageBlocks)
89      st->pipe->set_shader_buffers(
90            st->pipe, shader_type,
91            buffer_base + prog->info.num_ssbos,
92            c->MaxShaderStorageBlocks - prog->info.num_ssbos,
93            NULL, 0);
94}
95
96void st_bind_vs_ssbos(struct st_context *st)
97{
98   struct gl_program *prog =
99      st->ctx->_Shader->CurrentProgram[MESA_SHADER_VERTEX];
100
101   st_bind_ssbos(st, prog, PIPE_SHADER_VERTEX);
102}
103
104void st_bind_fs_ssbos(struct st_context *st)
105{
106   struct gl_program *prog =
107      st->ctx->_Shader->CurrentProgram[MESA_SHADER_FRAGMENT];
108
109   st_bind_ssbos(st, prog, PIPE_SHADER_FRAGMENT);
110}
111
112void st_bind_gs_ssbos(struct st_context *st)
113{
114   struct gl_program *prog =
115      st->ctx->_Shader->CurrentProgram[MESA_SHADER_GEOMETRY];
116
117   st_bind_ssbos(st, prog, PIPE_SHADER_GEOMETRY);
118}
119
120void st_bind_tcs_ssbos(struct st_context *st)
121{
122   struct gl_program *prog =
123      st->ctx->_Shader->CurrentProgram[MESA_SHADER_TESS_CTRL];
124
125   st_bind_ssbos(st, prog, PIPE_SHADER_TESS_CTRL);
126}
127
128void st_bind_tes_ssbos(struct st_context *st)
129{
130   struct gl_program *prog =
131      st->ctx->_Shader->CurrentProgram[MESA_SHADER_TESS_EVAL];
132
133   st_bind_ssbos(st, prog, PIPE_SHADER_TESS_EVAL);
134}
135
136void st_bind_cs_ssbos(struct st_context *st)
137{
138   struct gl_program *prog =
139      st->ctx->_Shader->CurrentProgram[MESA_SHADER_COMPUTE];
140
141   st_bind_ssbos(st, prog, PIPE_SHADER_COMPUTE);
142}
143