r300g: simplify the code for buffer uploads
[mesa.git] / src / gallium / drivers / r300 / r300_screen_buffer.c
1 /*
2 * Copyright 2010 Red Hat Inc.
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: Dave Airlie
24 */
25
26 #include <stdio.h>
27
28 #include "util/u_inlines.h"
29 #include "util/u_memory.h"
30 #include "util/u_upload_mgr.h"
31 #include "util/u_math.h"
32
33 #include "r300_screen_buffer.h"
34 #include "r300_winsys.h"
35
36 unsigned r300_buffer_is_referenced(struct pipe_context *context,
37 struct pipe_resource *buf,
38 enum r300_reference_domain domain)
39 {
40 struct r300_context *r300 = r300_context(context);
41 struct r300_buffer *rbuf = r300_buffer(buf);
42
43 if (r300_buffer_is_user_buffer(buf))
44 return PIPE_UNREFERENCED;
45
46 if (r300->rws->cs_is_buffer_referenced(r300->cs, rbuf->cs_buf, domain))
47 return PIPE_REFERENCED_FOR_READ | PIPE_REFERENCED_FOR_WRITE;
48
49 return PIPE_UNREFERENCED;
50 }
51
52 static unsigned r300_buffer_is_referenced_by_cs(struct pipe_context *context,
53 struct pipe_resource *buf,
54 unsigned level, int layer)
55 {
56 return r300_buffer_is_referenced(context, buf, R300_REF_CS);
57 }
58
59 void r300_upload_index_buffer(struct r300_context *r300,
60 struct pipe_resource **index_buffer,
61 unsigned index_size, unsigned *start,
62 unsigned count)
63 {
64 unsigned index_offset;
65 uint8_t *ptr = r300_buffer(*index_buffer)->user_buffer;
66
67 *index_buffer = NULL;
68
69 u_upload_data(r300->upload_ib,
70 count * index_size,
71 ptr + (*start * index_size),
72 &index_offset,
73 index_buffer);
74
75 *start = index_offset / index_size;
76 }
77
78 void r300_upload_user_buffers(struct r300_context *r300)
79 {
80 int i, nr = r300->velems->count;
81
82 for (i = 0; i < nr; i++) {
83 struct pipe_vertex_buffer *vb =
84 &r300->vertex_buffer[r300->velems->velem[i].vertex_buffer_index];
85
86 if (r300_buffer_is_user_buffer(vb->buffer)) {
87 u_upload_data(r300->upload_vb,
88 vb->buffer->width0,
89 r300_buffer(vb->buffer)->user_buffer,
90 &vb->buffer_offset, &vb->buffer);
91
92 r300->validate_buffers = TRUE;
93 }
94 }
95 }
96
97 static void r300_buffer_destroy(struct pipe_screen *screen,
98 struct pipe_resource *buf)
99 {
100 struct r300_screen *r300screen = r300_screen(screen);
101 struct r300_buffer *rbuf = r300_buffer(buf);
102 struct r300_winsys_screen *rws = r300screen->rws;
103
104 if (rbuf->constant_buffer)
105 FREE(rbuf->constant_buffer);
106
107 if (rbuf->buf)
108 rws->buffer_reference(rws, &rbuf->buf, NULL);
109
110 util_slab_free(&r300screen->pool_buffers, rbuf);
111 }
112
113 static struct pipe_transfer*
114 r300_buffer_get_transfer(struct pipe_context *context,
115 struct pipe_resource *resource,
116 unsigned level,
117 unsigned usage,
118 const struct pipe_box *box)
119 {
120 struct r300_context *r300 = r300_context(context);
121 struct pipe_transfer *transfer =
122 util_slab_alloc(&r300->pool_transfers);
123
124 transfer->resource = resource;
125 transfer->level = level;
126 transfer->usage = usage;
127 transfer->box = *box;
128 transfer->stride = 0;
129 transfer->layer_stride = 0;
130 transfer->data = NULL;
131
132 /* Note strides are zero, this is ok for buffers, but not for
133 * textures 2d & higher at least.
134 */
135 return transfer;
136 }
137
138 static void r300_buffer_transfer_destroy(struct pipe_context *pipe,
139 struct pipe_transfer *transfer)
140 {
141 struct r300_context *r300 = r300_context(pipe);
142 util_slab_free(&r300->pool_transfers, transfer);
143 }
144
145 static void *
146 r300_buffer_transfer_map( struct pipe_context *pipe,
147 struct pipe_transfer *transfer )
148 {
149 struct r300_context *r300 = r300_context(pipe);
150 struct r300_screen *r300screen = r300_screen(pipe->screen);
151 struct r300_winsys_screen *rws = r300screen->rws;
152 struct r300_buffer *rbuf = r300_buffer(transfer->resource);
153 uint8_t *map;
154 boolean flush = FALSE;
155 unsigned i;
156
157 if (rbuf->user_buffer)
158 return (uint8_t *) rbuf->user_buffer + transfer->box.x;
159 if (rbuf->constant_buffer)
160 return (uint8_t *) rbuf->constant_buffer + transfer->box.x;
161
162 /* check if the mapping is to a range we already flushed */
163 if (transfer->usage & PIPE_TRANSFER_DISCARD) {
164 for (i = 0; i < rbuf->num_ranges; i++) {
165 if ((transfer->box.x >= rbuf->ranges[i].start) &&
166 (transfer->box.x < rbuf->ranges[i].end))
167 flush = TRUE;
168
169 if (flush) {
170 /* unreference this hw buffer and allocate a new one */
171 rws->buffer_reference(rws, &rbuf->buf, NULL);
172
173 rbuf->num_ranges = 0;
174 rbuf->buf =
175 r300screen->rws->buffer_create(r300screen->rws,
176 rbuf->b.b.width0, 16,
177 rbuf->b.b.bind,
178 rbuf->b.b.usage,
179 rbuf->domain);
180 rbuf->cs_buf =
181 r300screen->rws->buffer_get_cs_handle(r300screen->rws,
182 rbuf->buf);
183 break;
184 }
185 }
186 }
187
188 map = rws->buffer_map(rws, rbuf->buf, r300->cs, transfer->usage);
189
190 if (map == NULL)
191 return NULL;
192
193 /* map_buffer() returned a pointer to the beginning of the buffer,
194 * but transfers are expected to return a pointer to just the
195 * region specified in the box.
196 */
197 return map + transfer->box.x;
198 }
199
200 static void r300_buffer_transfer_flush_region( struct pipe_context *pipe,
201 struct pipe_transfer *transfer,
202 const struct pipe_box *box)
203 {
204 struct r300_buffer *rbuf = r300_buffer(transfer->resource);
205 unsigned i;
206 unsigned offset = transfer->box.x + box->x;
207 unsigned length = box->width;
208
209 assert(box->x + box->width <= transfer->box.width);
210
211 if (rbuf->user_buffer)
212 return;
213 if (rbuf->constant_buffer)
214 return;
215
216 /* mark the range as used */
217 for(i = 0; i < rbuf->num_ranges; ++i) {
218 if(offset <= rbuf->ranges[i].end && rbuf->ranges[i].start <= (offset+box->width)) {
219 rbuf->ranges[i].start = MIN2(rbuf->ranges[i].start, offset);
220 rbuf->ranges[i].end = MAX2(rbuf->ranges[i].end, (offset+length));
221 return;
222 }
223 }
224
225 rbuf->ranges[rbuf->num_ranges].start = offset;
226 rbuf->ranges[rbuf->num_ranges].end = offset+length;
227 rbuf->num_ranges++;
228 }
229
230 static void r300_buffer_transfer_unmap( struct pipe_context *pipe,
231 struct pipe_transfer *transfer )
232 {
233 struct r300_screen *r300screen = r300_screen(pipe->screen);
234 struct r300_winsys_screen *rws = r300screen->rws;
235 struct r300_buffer *rbuf = r300_buffer(transfer->resource);
236
237 if (rbuf->buf) {
238 rws->buffer_unmap(rws, rbuf->buf);
239 }
240 }
241
242 static void r300_buffer_transfer_inline_write(struct pipe_context *pipe,
243 struct pipe_resource *resource,
244 unsigned level,
245 unsigned usage,
246 const struct pipe_box *box,
247 const void *data,
248 unsigned stride,
249 unsigned layer_stride)
250 {
251 struct r300_buffer *rbuf = r300_buffer(resource);
252 struct pipe_transfer *transfer = NULL;
253 uint8_t *map = NULL;
254
255 if (rbuf->constant_buffer) {
256 memcpy(rbuf->constant_buffer + box->x, data, box->width);
257 return;
258 }
259
260 transfer = r300_buffer_get_transfer(pipe, resource, 0,
261 PIPE_TRANSFER_WRITE | PIPE_TRANSFER_DISCARD, box);
262 map = r300_buffer_transfer_map(pipe, transfer);
263
264 memcpy(map, data, box->width);
265
266 r300_buffer_transfer_unmap(pipe, transfer);
267 r300_buffer_transfer_destroy(pipe, transfer);
268 }
269
270 struct u_resource_vtbl r300_buffer_vtbl =
271 {
272 u_default_resource_get_handle, /* get_handle */
273 r300_buffer_destroy, /* resource_destroy */
274 r300_buffer_is_referenced_by_cs, /* is_buffer_referenced */
275 r300_buffer_get_transfer, /* get_transfer */
276 r300_buffer_transfer_destroy, /* transfer_destroy */
277 r300_buffer_transfer_map, /* transfer_map */
278 r300_buffer_transfer_flush_region, /* transfer_flush_region */
279 r300_buffer_transfer_unmap, /* transfer_unmap */
280 r300_buffer_transfer_inline_write /* transfer_inline_write */
281 };
282
283 struct pipe_resource *r300_buffer_create(struct pipe_screen *screen,
284 const struct pipe_resource *templ)
285 {
286 struct r300_screen *r300screen = r300_screen(screen);
287 struct r300_buffer *rbuf;
288 unsigned alignment = 16;
289
290 rbuf = util_slab_alloc(&r300screen->pool_buffers);
291
292 rbuf->magic = R300_BUFFER_MAGIC;
293
294 rbuf->b.b = *templ;
295 rbuf->b.vtbl = &r300_buffer_vtbl;
296 pipe_reference_init(&rbuf->b.b.reference, 1);
297 rbuf->b.b.screen = screen;
298 rbuf->domain = R300_DOMAIN_GTT;
299 rbuf->num_ranges = 0;
300 rbuf->buf = NULL;
301 rbuf->constant_buffer = NULL;
302 rbuf->user_buffer = NULL;
303
304 /* Alloc constant buffers in RAM. */
305 if (templ->bind & PIPE_BIND_CONSTANT_BUFFER) {
306 rbuf->constant_buffer = MALLOC(templ->width0);
307 return &rbuf->b.b;
308 }
309
310 rbuf->buf =
311 r300screen->rws->buffer_create(r300screen->rws,
312 rbuf->b.b.width0, alignment,
313 rbuf->b.b.bind, rbuf->b.b.usage,
314 rbuf->domain);
315 rbuf->cs_buf =
316 r300screen->rws->buffer_get_cs_handle(r300screen->rws, rbuf->buf);
317
318 if (!rbuf->buf) {
319 util_slab_free(&r300screen->pool_buffers, rbuf);
320 return NULL;
321 }
322
323 return &rbuf->b.b;
324 }
325
326 struct pipe_resource *r300_user_buffer_create(struct pipe_screen *screen,
327 void *ptr,
328 unsigned bytes,
329 unsigned bind)
330 {
331 struct r300_screen *r300screen = r300_screen(screen);
332 struct r300_buffer *rbuf;
333
334 rbuf = util_slab_alloc(&r300screen->pool_buffers);
335
336 rbuf->magic = R300_BUFFER_MAGIC;
337
338 pipe_reference_init(&rbuf->b.b.reference, 1);
339 rbuf->b.vtbl = &r300_buffer_vtbl;
340 rbuf->b.b.screen = screen;
341 rbuf->b.b.target = PIPE_BUFFER;
342 rbuf->b.b.format = PIPE_FORMAT_R8_UNORM;
343 rbuf->b.b.usage = PIPE_USAGE_IMMUTABLE;
344 rbuf->b.b.bind = bind;
345 rbuf->b.b.width0 = bytes;
346 rbuf->b.b.height0 = 1;
347 rbuf->b.b.depth0 = 1;
348 rbuf->b.b.array_size = 1;
349 rbuf->b.b.flags = 0;
350 rbuf->domain = R300_DOMAIN_GTT;
351 rbuf->num_ranges = 0;
352 rbuf->buf = NULL;
353 rbuf->constant_buffer = NULL;
354 rbuf->user_buffer = ptr;
355 return &rbuf->b.b;
356 }