r300g: more solid is_buffer_referenced implementation
[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_format.h"
30 #include "util/u_memory.h"
31 #include "util/u_upload_mgr.h"
32 #include "util/u_math.h"
33
34 #include "r300_screen_buffer.h"
35 #include "r300_winsys.h"
36
37 unsigned r300_buffer_is_referenced(struct pipe_context *context,
38 struct pipe_resource *buf,
39 enum r300_reference_domain domain)
40 {
41 struct r300_context *r300 = r300_context(context);
42 struct r300_buffer *rbuf = r300_buffer(buf);
43
44 if (r300_buffer_is_user_buffer(buf))
45 return PIPE_UNREFERENCED;
46
47 if (r300->rws->is_buffer_referenced(r300->rws, rbuf->buf, domain))
48 return PIPE_REFERENCED_FOR_READ | PIPE_REFERENCED_FOR_WRITE;
49
50 return PIPE_UNREFERENCED;
51 }
52
53 static unsigned r300_buffer_is_referenced_by_cs(struct pipe_context *context,
54 struct pipe_resource *buf,
55 unsigned face, unsigned level)
56 {
57 return r300_buffer_is_referenced(context, buf, R300_REF_CS);
58 }
59
60 /* External helper, not required to implent u_resource_vtbl:
61 */
62 int r300_upload_index_buffer(struct r300_context *r300,
63 struct pipe_resource **index_buffer,
64 unsigned index_size,
65 unsigned start,
66 unsigned count)
67 {
68 struct pipe_resource *upload_buffer = NULL;
69 unsigned index_offset = start * index_size;
70 int ret = 0;
71
72 if (r300_buffer_is_user_buffer(*index_buffer)) {
73 ret = u_upload_buffer(r300->upload_ib,
74 index_offset,
75 count * index_size,
76 *index_buffer,
77 &index_offset,
78 &upload_buffer);
79 if (ret) {
80 goto done;
81 }
82 *index_buffer = upload_buffer;
83 }
84 done:
85 // if (upload_buffer)
86 // pipe_resource_reference(&upload_buffer, NULL);
87 return ret;
88 }
89
90 /* External helper, not required to implement u_resource_vtbl:
91 */
92 int r300_upload_user_buffers(struct r300_context *r300)
93 {
94 enum pipe_error ret = PIPE_OK;
95 int i, nr;
96
97 nr = r300->vertex_buffer_count;
98
99 for (i = 0; i < nr; i++) {
100 if (r300_buffer_is_user_buffer(r300->vertex_buffer[i].buffer)) {
101 struct pipe_resource *upload_buffer = NULL;
102 unsigned offset = 0; /*r300->vertex_buffer[i].buffer_offset * 4;*/
103 unsigned size = r300->vertex_buffer[i].buffer->width0;
104 unsigned upload_offset;
105 ret = u_upload_buffer(r300->upload_vb,
106 offset, size,
107 r300->vertex_buffer[i].buffer,
108 &upload_offset, &upload_buffer);
109 if (ret)
110 return ret;
111
112 pipe_resource_reference(&r300->vertex_buffer[i].buffer, NULL);
113 r300->vertex_buffer[i].buffer = upload_buffer;
114 r300->vertex_buffer[i].buffer_offset = upload_offset;
115 }
116 }
117 return ret;
118 }
119
120 static struct r300_winsys_buffer *
121 r300_winsys_buffer_create(struct r300_screen *r300screen,
122 unsigned alignment,
123 unsigned usage,
124 unsigned size)
125 {
126 struct r300_winsys_screen *rws = r300screen->rws;
127 struct r300_winsys_buffer *buf;
128
129 buf = rws->buffer_create(rws, alignment, usage, size);
130 return buf;
131 }
132
133 static void r300_winsys_buffer_destroy(struct r300_screen *r300screen,
134 struct r300_buffer *rbuf)
135 {
136 struct r300_winsys_screen *rws = r300screen->rws;
137
138 if (rbuf->buf) {
139 rws->buffer_reference(rws, &rbuf->buf, NULL);
140 rbuf->buf = NULL;
141 }
142 }
143
144 static void r300_buffer_destroy(struct pipe_screen *screen,
145 struct pipe_resource *buf)
146 {
147 struct r300_screen *r300screen = r300_screen(screen);
148 struct r300_buffer *rbuf = r300_buffer(buf);
149
150 r300_winsys_buffer_destroy(r300screen, rbuf);
151 FREE(rbuf);
152 }
153
154 static void *
155 r300_buffer_transfer_map( struct pipe_context *pipe,
156 struct pipe_transfer *transfer )
157 {
158 struct r300_screen *r300screen = r300_screen(pipe->screen);
159 struct r300_winsys_screen *rws = r300screen->rws;
160 struct r300_buffer *rbuf = r300_buffer(transfer->resource);
161 uint8_t *map;
162 boolean flush = FALSE;
163 unsigned i;
164
165 if (rbuf->user_buffer)
166 return (uint8_t *) rbuf->user_buffer + transfer->box.x;
167
168 if (rbuf->b.b.bind & PIPE_BIND_CONSTANT_BUFFER) {
169 goto just_map;
170 }
171
172 /* check if the mapping is to a range we already flushed */
173 if (transfer->usage & PIPE_TRANSFER_DISCARD) {
174 for (i = 0; i < rbuf->num_ranges; i++) {
175 if ((transfer->box.x >= rbuf->ranges[i].start) &&
176 (transfer->box.x < rbuf->ranges[i].end))
177 flush = TRUE;
178
179 if (flush) {
180 /* unreference this hw buffer and allocate a new one */
181 rws->buffer_reference(rws, &rbuf->buf, NULL);
182
183 rbuf->num_ranges = 0;
184 rbuf->buf = r300_winsys_buffer_create(r300screen,
185 16,
186 rbuf->b.b.bind, /* XXX */
187 rbuf->b.b.width0);
188 break;
189 }
190 }
191 }
192 just_map:
193 map = rws->buffer_map(rws, rbuf->buf, transfer->usage);
194
195 if (map == NULL)
196 return NULL;
197
198 /* map_buffer() returned a pointer to the beginning of the buffer,
199 * but transfers are expected to return a pointer to just the
200 * region specified in the box.
201 */
202 return map + transfer->box.x;
203 }
204
205 static void r300_buffer_transfer_flush_region( struct pipe_context *pipe,
206 struct pipe_transfer *transfer,
207 const struct pipe_box *box)
208 {
209 struct r300_buffer *rbuf = r300_buffer(transfer->resource);
210 unsigned i;
211 unsigned offset = transfer->box.x + box->x;
212 unsigned length = box->width;
213
214 assert(box->x + box->width <= transfer->box.width);
215
216 if (rbuf->user_buffer)
217 return;
218
219 if (rbuf->b.b.bind & PIPE_BIND_CONSTANT_BUFFER)
220 return;
221
222 /* mark the range as used */
223 for(i = 0; i < rbuf->num_ranges; ++i) {
224 if(offset <= rbuf->ranges[i].end && rbuf->ranges[i].start <= (offset+box->width)) {
225 rbuf->ranges[i].start = MIN2(rbuf->ranges[i].start, offset);
226 rbuf->ranges[i].end = MAX2(rbuf->ranges[i].end, (offset+length));
227 return;
228 }
229 }
230
231 rbuf->ranges[rbuf->num_ranges].start = offset;
232 rbuf->ranges[rbuf->num_ranges].end = offset+length;
233 rbuf->num_ranges++;
234 }
235
236 static void r300_buffer_transfer_unmap( struct pipe_context *pipe,
237 struct pipe_transfer *transfer )
238 {
239 struct r300_screen *r300screen = r300_screen(pipe->screen);
240 struct r300_winsys_screen *rws = r300screen->rws;
241 struct r300_buffer *rbuf = r300_buffer(transfer->resource);
242
243 if (rbuf->buf) {
244 rws->buffer_unmap(rws, rbuf->buf);
245 }
246 }
247
248 struct u_resource_vtbl r300_buffer_vtbl =
249 {
250 u_default_resource_get_handle, /* get_handle */
251 r300_buffer_destroy, /* resource_destroy */
252 r300_buffer_is_referenced_by_cs, /* is_buffer_referenced */
253 u_default_get_transfer, /* get_transfer */
254 u_default_transfer_destroy, /* transfer_destroy */
255 r300_buffer_transfer_map, /* transfer_map */
256 r300_buffer_transfer_flush_region, /* transfer_flush_region */
257 r300_buffer_transfer_unmap, /* transfer_unmap */
258 u_default_transfer_inline_write /* transfer_inline_write */
259 };
260
261 struct pipe_resource *r300_buffer_create(struct pipe_screen *screen,
262 const struct pipe_resource *templ)
263 {
264 struct r300_screen *r300screen = r300_screen(screen);
265 struct r300_buffer *rbuf;
266 unsigned alignment = 16;
267
268 rbuf = CALLOC_STRUCT(r300_buffer);
269 if (!rbuf)
270 goto error1;
271
272 rbuf->magic = R300_BUFFER_MAGIC;
273
274 rbuf->b.b = *templ;
275 rbuf->b.vtbl = &r300_buffer_vtbl;
276 pipe_reference_init(&rbuf->b.b.reference, 1);
277 rbuf->b.b.screen = screen;
278
279 if (rbuf->b.b.bind & R300_BIND_OQBO)
280 alignment = 4096;
281
282 rbuf->buf = r300_winsys_buffer_create(r300screen,
283 alignment,
284 rbuf->b.b.bind,
285 rbuf->b.b.width0);
286
287 if (!rbuf->buf)
288 goto error2;
289
290 return &rbuf->b.b;
291 error2:
292 FREE(rbuf);
293 error1:
294 return NULL;
295 }
296
297 struct pipe_resource *r300_user_buffer_create(struct pipe_screen *screen,
298 void *ptr,
299 unsigned bytes,
300 unsigned bind)
301 {
302 struct r300_buffer *rbuf;
303
304 rbuf = CALLOC_STRUCT(r300_buffer);
305 if (!rbuf)
306 goto no_rbuf;
307
308 rbuf->magic = R300_BUFFER_MAGIC;
309
310 pipe_reference_init(&rbuf->b.b.reference, 1);
311 rbuf->b.vtbl = &r300_buffer_vtbl;
312 rbuf->b.b.screen = screen;
313 rbuf->b.b.format = PIPE_FORMAT_R8_UNORM;
314 rbuf->b.b._usage = PIPE_USAGE_IMMUTABLE;
315 rbuf->b.b.bind = bind;
316 rbuf->b.b.width0 = bytes;
317 rbuf->b.b.height0 = 1;
318 rbuf->b.b.depth0 = 1;
319
320 rbuf->user_buffer = ptr;
321 return &rbuf->b.b;
322
323 no_rbuf:
324 return NULL;
325 }