gallium: change set_constant_buffer to be UBO-friendly
[mesa.git] / src / gallium / drivers / radeonsi / r600_buffer.c
1 /*
2 * Copyright 2010 Jerome Glisse <glisse@freedesktop.org>
3 *
4 * Permission is hereby granted, free of charge, to any person obtaining a
5 * copy of this software and associated documentation files (the "Software"),
6 * to deal in the Software without restriction, including without limitation
7 * on the rights to use, copy, modify, merge, publish, distribute, sub
8 * license, and/or sell copies of the Software, and to permit persons to whom
9 * the Software is furnished to do so, subject to the following conditions:
10 *
11 * The above copyright notice and this permission notice (including the next
12 * paragraph) shall be included in all copies or substantial portions of the
13 * Software.
14 *
15 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17 * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL
18 * THE AUTHOR(S) AND/OR THEIR SUPPLIERS BE LIABLE FOR ANY CLAIM,
19 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
20 * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
21 * USE OR OTHER DEALINGS IN THE SOFTWARE.
22 *
23 * Authors:
24 * Jerome Glisse
25 * Corbin Simpson <MostAwesomeDude@gmail.com>
26 */
27 #include <byteswap.h>
28
29 #include "pipe/p_screen.h"
30 #include "util/u_format.h"
31 #include "util/u_math.h"
32 #include "util/u_inlines.h"
33 #include "util/u_memory.h"
34 #include "util/u_upload_mgr.h"
35
36 #include "r600.h"
37 #include "radeonsi_pipe.h"
38
39 static void r600_buffer_destroy(struct pipe_screen *screen,
40 struct pipe_resource *buf)
41 {
42 struct r600_screen *rscreen = (struct r600_screen*)screen;
43 struct r600_resource *rbuffer = r600_resource(buf);
44
45 pb_reference(&rbuffer->buf, NULL);
46 util_slab_free(&rscreen->pool_buffers, rbuffer);
47 }
48
49 static struct pipe_transfer *r600_get_transfer(struct pipe_context *ctx,
50 struct pipe_resource *resource,
51 unsigned level,
52 unsigned usage,
53 const struct pipe_box *box)
54 {
55 struct r600_context *rctx = (struct r600_context*)ctx;
56 struct pipe_transfer *transfer = util_slab_alloc(&rctx->pool_transfers);
57
58 transfer->resource = resource;
59 transfer->level = level;
60 transfer->usage = usage;
61 transfer->box = *box;
62 transfer->stride = 0;
63 transfer->layer_stride = 0;
64 transfer->data = NULL;
65
66 /* Note strides are zero, this is ok for buffers, but not for
67 * textures 2d & higher at least.
68 */
69 return transfer;
70 }
71
72 static void *r600_buffer_transfer_map(struct pipe_context *pipe,
73 struct pipe_transfer *transfer)
74 {
75 struct r600_resource *rbuffer = r600_resource(transfer->resource);
76 struct r600_context *rctx = (struct r600_context*)pipe;
77 uint8_t *data;
78
79 if (rbuffer->b.b.user_ptr)
80 return (uint8_t*)rbuffer->b.b.user_ptr + transfer->box.x;
81
82 data = rctx->ws->buffer_map(rbuffer->cs_buf, rctx->cs, transfer->usage);
83 if (!data)
84 return NULL;
85
86 return (uint8_t*)data + transfer->box.x;
87 }
88
89 static void r600_buffer_transfer_unmap(struct pipe_context *pipe,
90 struct pipe_transfer *transfer)
91 {
92 /* no-op */
93 }
94
95 static void r600_buffer_transfer_flush_region(struct pipe_context *pipe,
96 struct pipe_transfer *transfer,
97 const struct pipe_box *box)
98 {
99 }
100
101 static void r600_transfer_destroy(struct pipe_context *ctx,
102 struct pipe_transfer *transfer)
103 {
104 struct r600_context *rctx = (struct r600_context*)ctx;
105 util_slab_free(&rctx->pool_transfers, transfer);
106 }
107
108 static const struct u_resource_vtbl r600_buffer_vtbl =
109 {
110 u_default_resource_get_handle, /* get_handle */
111 r600_buffer_destroy, /* resource_destroy */
112 r600_get_transfer, /* get_transfer */
113 r600_transfer_destroy, /* transfer_destroy */
114 r600_buffer_transfer_map, /* transfer_map */
115 r600_buffer_transfer_flush_region, /* transfer_flush_region */
116 r600_buffer_transfer_unmap, /* transfer_unmap */
117 NULL /* transfer_inline_write */
118 };
119
120 bool r600_init_resource(struct r600_screen *rscreen,
121 struct r600_resource *res,
122 unsigned size, unsigned alignment,
123 unsigned bind, unsigned usage)
124 {
125 uint32_t initial_domain, domains;
126
127 /* Staging resources particpate in transfers and blits only
128 * and are used for uploads and downloads from regular
129 * resources. We generate them internally for some transfers.
130 */
131 if (usage == PIPE_USAGE_STAGING) {
132 domains = RADEON_DOMAIN_GTT;
133 initial_domain = RADEON_DOMAIN_GTT;
134 } else {
135 domains = RADEON_DOMAIN_GTT | RADEON_DOMAIN_VRAM;
136
137 switch(usage) {
138 case PIPE_USAGE_DYNAMIC:
139 case PIPE_USAGE_STREAM:
140 case PIPE_USAGE_STAGING:
141 initial_domain = RADEON_DOMAIN_GTT;
142 break;
143 case PIPE_USAGE_DEFAULT:
144 case PIPE_USAGE_STATIC:
145 case PIPE_USAGE_IMMUTABLE:
146 default:
147 initial_domain = RADEON_DOMAIN_VRAM;
148 break;
149 }
150 }
151
152 res->buf = rscreen->ws->buffer_create(rscreen->ws, size, alignment, bind, initial_domain);
153 if (!res->buf) {
154 return false;
155 }
156
157 res->cs_buf = rscreen->ws->buffer_get_cs_handle(res->buf);
158 res->domains = domains;
159 return true;
160 }
161
162 struct pipe_resource *r600_buffer_create(struct pipe_screen *screen,
163 const struct pipe_resource *templ)
164 {
165 struct r600_screen *rscreen = (struct r600_screen*)screen;
166 struct r600_resource *rbuffer;
167 /* XXX We probably want a different alignment for buffers and textures. */
168 unsigned alignment = 4096;
169
170 rbuffer = util_slab_alloc(&rscreen->pool_buffers);
171
172 rbuffer->b.b = *templ;
173 pipe_reference_init(&rbuffer->b.b.reference, 1);
174 rbuffer->b.b.screen = screen;
175 rbuffer->b.vtbl = &r600_buffer_vtbl;
176 rbuffer->b.b.user_ptr = NULL;
177
178 if (!r600_init_resource(rscreen, rbuffer, templ->width0, alignment, templ->bind, templ->usage)) {
179 util_slab_free(&rscreen->pool_buffers, rbuffer);
180 return NULL;
181 }
182 return &rbuffer->b.b;
183 }
184
185 struct pipe_resource *r600_user_buffer_create(struct pipe_screen *screen,
186 void *ptr, unsigned bytes,
187 unsigned bind)
188 {
189 struct r600_screen *rscreen = (struct r600_screen*)screen;
190 struct r600_resource *rbuffer;
191
192 rbuffer = util_slab_alloc(&rscreen->pool_buffers);
193
194 pipe_reference_init(&rbuffer->b.b.reference, 1);
195 rbuffer->b.vtbl = &r600_buffer_vtbl;
196 rbuffer->b.b.screen = screen;
197 rbuffer->b.b.target = PIPE_BUFFER;
198 rbuffer->b.b.format = PIPE_FORMAT_R8_UNORM;
199 rbuffer->b.b.usage = PIPE_USAGE_IMMUTABLE;
200 rbuffer->b.b.bind = bind;
201 rbuffer->b.b.width0 = bytes;
202 rbuffer->b.b.height0 = 1;
203 rbuffer->b.b.depth0 = 1;
204 rbuffer->b.b.array_size = 1;
205 rbuffer->b.b.flags = 0;
206 rbuffer->b.b.user_ptr = ptr;
207 rbuffer->buf = NULL;
208 return &rbuffer->b.b;
209 }
210
211 void r600_upload_index_buffer(struct r600_context *rctx,
212 struct pipe_index_buffer *ib, unsigned count)
213 {
214 struct r600_resource *rbuffer = r600_resource(ib->buffer);
215
216 u_upload_data(rctx->uploader, 0, count * ib->index_size,
217 rbuffer->b.b.user_ptr, &ib->offset, &ib->buffer);
218 }
219
220 void r600_upload_const_buffer(struct r600_context *rctx, struct r600_resource **rbuffer,
221 uint32_t *const_offset)
222 {
223 if ((*rbuffer)->b.b.user_ptr) {
224 uint8_t *ptr = (*rbuffer)->b.b.user_ptr;
225 unsigned size = (*rbuffer)->b.b.width0;
226
227 *rbuffer = NULL;
228
229 if (R600_BIG_ENDIAN) {
230 uint32_t *tmpPtr;
231 unsigned i;
232
233 if (!(tmpPtr = malloc(size))) {
234 R600_ERR("Failed to allocate BE swap buffer.\n");
235 return;
236 }
237
238 for (i = 0; i < size / 4; ++i) {
239 tmpPtr[i] = bswap_32(((uint32_t *)ptr)[i]);
240 }
241
242 u_upload_data(rctx->uploader, 0, size, tmpPtr, const_offset,
243 (struct pipe_resource**)rbuffer);
244
245 free(tmpPtr);
246 } else {
247 u_upload_data(rctx->uploader, 0, size, ptr, const_offset,
248 (struct pipe_resource**)rbuffer);
249 }
250 } else {
251 *const_offset = 0;
252 }
253 }