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
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_binding_to_sb(struct gl_buffer_binding *binding,
45                 struct pipe_shader_buffer *sb)
46{
47   struct st_buffer_object *st_obj =
48      st_buffer_object(binding->BufferObject);
49
50   if (st_obj && st_obj->buffer) {
51     sb->buffer = st_obj->buffer;
52     sb->buffer_offset = binding->Offset;
53     sb->buffer_size = st_obj->buffer->width0 - binding->Offset;
54
55     /* AutomaticSize is FALSE if the buffer was set with BindBufferRange.
56      * Take the minimum just to be sure.
57      */
58     if (!binding->AutomaticSize)
59       sb->buffer_size = MIN2(sb->buffer_size, (unsigned) binding->Size);
60   } else {
61     sb->buffer = NULL;
62     sb->buffer_offset = 0;
63     sb->buffer_size = 0;
64   }
65}
66
67static void
68st_bind_atomics(struct st_context *st, struct gl_program *prog,
69                gl_shader_stage stage)
70{
71   unsigned i;
72   enum pipe_shader_type shader_type = pipe_shader_type_from_mesa(stage);
73
74   if (!prog || !st->pipe->set_shader_buffers || st->has_hw_atomics)
75      return;
76
77   /* For !has_hw_atomics, the atomic counters have been rewritten to be above
78    * the SSBOs used by the program.
79    */
80   unsigned buffer_base = prog->info.num_ssbos;
81   unsigned used_bindings = 0;
82   for (i = 0; i < prog->sh.data->NumAtomicBuffers; i++) {
83      struct gl_active_atomic_buffer *atomic =
84         &prog->sh.data->AtomicBuffers[i];
85      struct pipe_shader_buffer sb;
86
87      st_binding_to_sb(&st->ctx->AtomicBufferBindings[atomic->Binding], &sb);
88
89      st->pipe->set_shader_buffers(st->pipe, shader_type,
90                                   buffer_base + atomic->Binding, 1, &sb, 0x1);
91      used_bindings = MAX2(atomic->Binding + 1, used_bindings);
92   }
93   st->last_used_atomic_bindings[shader_type] = used_bindings;
94}
95
96void
97st_bind_vs_atomics(struct st_context *st)
98{
99   struct gl_program *prog =
100      st->ctx->_Shader->CurrentProgram[MESA_SHADER_VERTEX];
101
102   st_bind_atomics(st, prog, MESA_SHADER_VERTEX);
103}
104
105void
106st_bind_fs_atomics(struct st_context *st)
107{
108   struct gl_program *prog =
109      st->ctx->_Shader->CurrentProgram[MESA_SHADER_FRAGMENT];
110
111   st_bind_atomics(st, prog, MESA_SHADER_FRAGMENT);
112}
113
114void
115st_bind_gs_atomics(struct st_context *st)
116{
117   struct gl_program *prog =
118      st->ctx->_Shader->CurrentProgram[MESA_SHADER_GEOMETRY];
119
120   st_bind_atomics(st, prog, MESA_SHADER_GEOMETRY);
121}
122
123void
124st_bind_tcs_atomics(struct st_context *st)
125{
126   struct gl_program *prog =
127      st->ctx->_Shader->CurrentProgram[MESA_SHADER_TESS_CTRL];
128
129   st_bind_atomics(st, prog, MESA_SHADER_TESS_CTRL);
130}
131
132void
133st_bind_tes_atomics(struct st_context *st)
134{
135   struct gl_program *prog =
136      st->ctx->_Shader->CurrentProgram[MESA_SHADER_TESS_EVAL];
137
138   st_bind_atomics(st, prog, MESA_SHADER_TESS_EVAL);
139}
140
141void
142st_bind_cs_atomics(struct st_context *st)
143{
144   if (st->has_hw_atomics) {
145      st_bind_hw_atomic_buffers(st);
146      return;
147   }
148   struct gl_program *prog =
149      st->ctx->_Shader->CurrentProgram[MESA_SHADER_COMPUTE];
150
151   st_bind_atomics(st, prog, MESA_SHADER_COMPUTE);
152}
153
154void
155st_bind_hw_atomic_buffers(struct st_context *st)
156{
157   struct pipe_shader_buffer buffers[PIPE_MAX_HW_ATOMIC_BUFFERS];
158   int i;
159
160   if (!st->has_hw_atomics)
161      return;
162
163   for (i = 0; i < st->ctx->Const.MaxAtomicBufferBindings; i++)
164      st_binding_to_sb(&st->ctx->AtomicBufferBindings[i], &buffers[i]);
165
166   st->pipe->set_hw_atomic_buffers(st->pipe, 0, st->ctx->Const.MaxAtomicBufferBindings, buffers);
167}
168