r300g: upload only vertex buffers referenced by vertex elements
[mesa.git] / src / gallium / drivers / r300 / r300_screen_buffer.c
index b97d0d76a4a1f652c63061d735a989a327c85550..fe53f30b3080d29ecd15bc9c2b30f465aa1b07cf 100644 (file)
  *
  * Authors: Dave Airlie
  */
+
 #include <stdio.h>
 
 #include "util/u_inlines.h"
-#include "util/u_format.h"
 #include "util/u_memory.h"
 #include "util/u_upload_mgr.h"
 #include "util/u_math.h"
 
 #include "r300_screen_buffer.h"
-
 #include "r300_winsys.h"
 
-boolean r300_buffer_is_referenced(struct r300_context *r300,
-                                 struct pipe_buffer *buf)
+unsigned r300_buffer_is_referenced(struct pipe_context *context,
+                                  struct pipe_resource *buf,
+                                   enum r300_reference_domain domain)
 {
+    struct r300_context *r300 = r300_context(context);
     struct r300_buffer *rbuf = r300_buffer(buf);
+
     if (r300_buffer_is_user_buffer(buf))
-       return FALSE;
+       return PIPE_UNREFERENCED;
+
+    if (r300->rws->is_buffer_referenced(r300->rws, rbuf->buf, domain))
+        return PIPE_REFERENCED_FOR_READ | PIPE_REFERENCED_FOR_WRITE;
 
-    return r300->rws->is_buffer_referenced(r300->rws, rbuf->buf);
-    
+    return PIPE_UNREFERENCED;
+}
+
+static unsigned r300_buffer_is_referenced_by_cs(struct pipe_context *context,
+                                                struct pipe_resource *buf,
+                                                unsigned face, unsigned level)
+{
+    return r300_buffer_is_referenced(context, buf, R300_REF_CS);
 }
+
+/* External helper, not required to implent u_resource_vtbl:
+ */
 int r300_upload_index_buffer(struct r300_context *r300,
-                            struct pipe_buffer **index_buffer,
+                            struct pipe_resource **index_buffer,
                             unsigned index_size,
                             unsigned start,
                             unsigned count)
 {
-   struct pipe_buffer *upload_buffer = NULL;
+   struct pipe_resource *upload_buffer = NULL;
    unsigned index_offset = start * index_size;
    int ret = 0;
 
@@ -68,52 +82,43 @@ int r300_upload_index_buffer(struct r300_context *r300,
     }
  done:
     //    if (upload_buffer)
-    // pipe_buffer_reference(&upload_buffer, NULL);
+    // pipe_resource_reference(&upload_buffer, NULL);
     return ret;
 }
 
+/* External helper, not required to implement u_resource_vtbl:
+ */
 int r300_upload_user_buffers(struct r300_context *r300)
 {
     enum pipe_error ret = PIPE_OK;
     int i, nr;
 
-    nr = r300->vertex_buffer_count;
+    nr = r300->velems->count;
 
     for (i = 0; i < nr; i++) {
-
-       if (r300_buffer_is_user_buffer(r300->vertex_buffer[i].buffer)) {
-           struct pipe_buffer *upload_buffer = NULL;
-           unsigned offset = 0; /*r300->vertex_buffer[i].buffer_offset * 4;*/
-           unsigned size = r300->vertex_buffer[i].buffer->size;
-           unsigned upload_offset;
-           ret = u_upload_buffer(r300->upload_vb,
-                                 offset, size,
-                                 r300->vertex_buffer[i].buffer,
-                                 &upload_offset, &upload_buffer);
-           if (ret)
-               return ret;
-
-           pipe_buffer_reference(&r300->vertex_buffer[i].buffer, NULL);
-           r300->vertex_buffer[i].buffer = upload_buffer;
-           r300->vertex_buffer[i].buffer_offset = upload_offset;
-       }
+        struct pipe_vertex_buffer *vb =
+            &r300->vertex_buffer[r300->velems->velem[i].vertex_buffer_index];
+
+        if (r300_buffer_is_user_buffer(vb->buffer)) {
+            struct pipe_resource *upload_buffer = NULL;
+            unsigned offset = 0; /*vb->buffer_offset * 4;*/
+            unsigned size = vb->buffer->width0;
+            unsigned upload_offset;
+            ret = u_upload_buffer(r300->upload_vb,
+                                  offset, size,
+                                  vb->buffer,
+                                  &upload_offset, &upload_buffer);
+            if (ret)
+                return ret;
+
+            pipe_resource_reference(&vb->buffer, NULL);
+            vb->buffer = upload_buffer;
+            vb->buffer_offset = upload_offset;
+        }
     }
     return ret;
 }
 
-static struct r300_winsys_buffer *
-r300_winsys_buffer_create(struct r300_screen *r300screen,
-                         unsigned alignment,
-                         unsigned usage,
-                         unsigned size)
-{
-    struct r300_winsys_screen *rws = r300screen->rws;
-    struct r300_winsys_buffer *buf;
-
-    buf = rws->buffer_create(rws, alignment, usage, size);
-    return buf;
-}
-
 static void r300_winsys_buffer_destroy(struct r300_screen *r300screen,
                                       struct r300_buffer *rbuf)
 {
@@ -125,70 +130,10 @@ static void r300_winsys_buffer_destroy(struct r300_screen *r300screen,
     }
 }
 
-static struct pipe_buffer *r300_buffer_create(struct pipe_screen *screen,
-                                             unsigned alignment,
-                                             unsigned usage,
-                                             unsigned size)
+static void r300_buffer_destroy(struct pipe_screen *screen,
+                               struct pipe_resource *buf)
 {
     struct r300_screen *r300screen = r300_screen(screen);
-    struct r300_buffer *rbuf;
-
-    rbuf = CALLOC_STRUCT(r300_buffer);
-    if (!rbuf)
-       goto error1;
-
-    rbuf->magic = R300_BUFFER_MAGIC;
-
-    pipe_reference_init(&rbuf->base.reference, 1);
-    rbuf->base.screen = screen;
-    rbuf->base.alignment = alignment;
-    rbuf->base.usage = usage;
-    rbuf->base.size = size;
-
-    rbuf->buf = r300_winsys_buffer_create(r300screen,
-                                         alignment,
-                                         usage,
-                                         size);
-
-    if (!rbuf->buf)
-       goto error2;
-
-    return &rbuf->base;
-error2:
-    FREE(rbuf);
-error1:
-    return NULL;
-}
-
-
-static struct pipe_buffer *r300_user_buffer_create(struct pipe_screen *screen,
-                                                  void *ptr,
-                                                  unsigned bytes)
-{
-    struct r300_buffer *rbuf;
-
-    rbuf = CALLOC_STRUCT(r300_buffer);
-    if (!rbuf)
-       goto no_rbuf;
-
-    rbuf->magic = R300_BUFFER_MAGIC;
-
-    pipe_reference_init(&rbuf->base.reference, 1);
-    rbuf->base.screen = screen;
-    rbuf->base.alignment = 1;
-    rbuf->base.usage = 0;
-    rbuf->base.size = bytes;
-
-    rbuf->user_buffer = ptr;
-    return &rbuf->base;
-
-no_rbuf:
-    return NULL;
-}
-
-static void r300_buffer_destroy(struct pipe_buffer *buf)
-{
-    struct r300_screen *r300screen = r300_screen(buf->screen);
     struct r300_buffer *rbuf = r300_buffer(buf);
 
     r300_winsys_buffer_destroy(r300screen, rbuf);
@@ -196,70 +141,76 @@ static void r300_buffer_destroy(struct pipe_buffer *buf)
 }
 
 static void *
-r300_buffer_map_range(struct pipe_screen *screen,
-                     struct pipe_buffer *buf,
-                     unsigned offset, unsigned length,
-                     unsigned usage )
+r300_buffer_transfer_map( struct pipe_context *pipe,
+                         struct pipe_transfer *transfer )
 {
-    struct r300_screen *r300screen = r300_screen(screen);
+    struct r300_screen *r300screen = r300_screen(pipe->screen);
     struct r300_winsys_screen *rws = r300screen->rws;
-    struct r300_buffer *rbuf = r300_buffer(buf);
-    void *map;
-    int flush = 0;
-    int i;
+    struct r300_buffer *rbuf = r300_buffer(transfer->resource);
+    uint8_t *map;
+    boolean flush = FALSE;
+    unsigned i;
 
     if (rbuf->user_buffer)
-       return rbuf->user_buffer;
+        return (uint8_t *) rbuf->user_buffer + transfer->box.x;
 
-    if (rbuf->base.usage & PIPE_BUFFER_USAGE_CONSTANT)
+    if (rbuf->b.b.bind & PIPE_BIND_CONSTANT_BUFFER) {
        goto just_map;
+    }
 
     /* check if the mapping is to a range we already flushed */
-    if (usage & PIPE_BUFFER_USAGE_DISCARD) {
+    if (transfer->usage & PIPE_TRANSFER_DISCARD) {
        for (i = 0; i < rbuf->num_ranges; i++) {
+           if ((transfer->box.x >= rbuf->ranges[i].start) &&
+               (transfer->box.x < rbuf->ranges[i].end))
+               flush = TRUE;
 
-           if ((offset >= rbuf->ranges[i].start) &&
-               (offset < rbuf->ranges[i].end))
-               flush = 1;
-           
            if (flush) {
                /* unreference this hw buffer and allocate a new one */
                rws->buffer_reference(rws, &rbuf->buf, NULL);
 
                rbuf->num_ranges = 0;
-               rbuf->map = NULL;
-               rbuf->buf = r300_winsys_buffer_create(r300screen,
-                                                     rbuf->base.alignment,
-                                                     rbuf->base.usage,
-                                                     rbuf->base.size);
+               rbuf->buf = r300screen->rws->buffer_create(r300screen->rws, 16,
+                                                     rbuf->b.b.bind,
+                                                      rbuf->domain,
+                                                     rbuf->b.b.width0);
                break;
            }
        }
     }
 just_map:
-    map = rws->buffer_map(rws, rbuf->buf, usage | R300_USAGE_FLAG_DONT_SYNC);
-   
-    return map;
+    map = rws->buffer_map(rws, rbuf->buf, transfer->usage);
+
+    if (map == NULL)
+        return NULL;
+
+    /* map_buffer() returned a pointer to the beginning of the buffer,
+     * but transfers are expected to return a pointer to just the
+     * region specified in the box.
+     */
+    return map + transfer->box.x;
 }
 
-static void 
-r300_buffer_flush_mapped_range( struct pipe_screen *screen,
-                               struct pipe_buffer *buf,
-                               unsigned offset,
-                               unsigned length )
+static void r300_buffer_transfer_flush_region( struct pipe_context *pipe,
+                                              struct pipe_transfer *transfer,
+                                              const struct pipe_box *box)
 {
-    struct r300_buffer *rbuf = r300_buffer(buf);
-    int i;
+    struct r300_buffer *rbuf = r300_buffer(transfer->resource);
+    unsigned i;
+    unsigned offset = transfer->box.x + box->x;
+    unsigned length = box->width;
+
+    assert(box->x + box->width <= transfer->box.width);
 
     if (rbuf->user_buffer)
        return;
 
-    if (rbuf->base.usage & PIPE_BUFFER_USAGE_CONSTANT)
+    if (rbuf->b.b.bind & PIPE_BIND_CONSTANT_BUFFER)
        return;
 
     /* mark the range as used */
     for(i = 0; i < rbuf->num_ranges; ++i) {
-       if(offset <= rbuf->ranges[i].end && rbuf->ranges[i].start <= (offset+length)) {
+       if(offset <= rbuf->ranges[i].end && rbuf->ranges[i].start <= (offset+box->width)) {
            rbuf->ranges[i].start = MIN2(rbuf->ranges[i].start, offset);
            rbuf->ranges[i].end   = MAX2(rbuf->ranges[i].end, (offset+length));
            return;
@@ -271,44 +222,97 @@ r300_buffer_flush_mapped_range( struct pipe_screen *screen,
     rbuf->num_ranges++;
 }
 
-static void *
-r300_buffer_map(struct pipe_screen *screen,
-               struct pipe_buffer *buf,
-               unsigned usage)
+static void r300_buffer_transfer_unmap( struct pipe_context *pipe,
+                           struct pipe_transfer *transfer )
 {
-    struct r300_screen *r300screen = r300_screen(screen);
+    struct r300_screen *r300screen = r300_screen(pipe->screen);
     struct r300_winsys_screen *rws = r300screen->rws;
-    struct r300_buffer *rbuf = r300_buffer(buf);
-    void *map;
+    struct r300_buffer *rbuf = r300_buffer(transfer->resource);
 
-   if (rbuf->user_buffer)
-      return rbuf->user_buffer;
-
-    map = rws->buffer_map(rws, rbuf->buf, usage);
-
-    return map;
+    if (rbuf->buf) {
+        rws->buffer_unmap(rws, rbuf->buf);
+    }
 }
 
-static void
-r300_buffer_unmap(struct pipe_screen *screen,
-                 struct pipe_buffer *buf)
+struct u_resource_vtbl r300_buffer_vtbl = 
+{
+   u_default_resource_get_handle,      /* get_handle */
+   r300_buffer_destroy,                     /* resource_destroy */
+   r300_buffer_is_referenced_by_cs,  /* is_buffer_referenced */
+   u_default_get_transfer,          /* get_transfer */
+   u_default_transfer_destroy,      /* transfer_destroy */
+   r300_buffer_transfer_map,        /* transfer_map */
+   r300_buffer_transfer_flush_region,  /* transfer_flush_region */
+   r300_buffer_transfer_unmap,      /* transfer_unmap */
+   u_default_transfer_inline_write   /* transfer_inline_write */
+};
+
+struct pipe_resource *r300_buffer_create(struct pipe_screen *screen,
+                                        const struct pipe_resource *templ)
 {
     struct r300_screen *r300screen = r300_screen(screen);
-    struct r300_winsys_screen *rws = r300screen->rws;
-    struct r300_buffer *rbuf = r300_buffer(buf);
+    struct r300_buffer *rbuf;
+    unsigned alignment = 16;
 
-    if (rbuf->buf) {
-        rws->buffer_unmap(rws, rbuf->buf);
-    }
+    rbuf = CALLOC_STRUCT(r300_buffer);
+    if (!rbuf)
+       goto error1;
+
+    rbuf->magic = R300_BUFFER_MAGIC;
+
+    rbuf->b.b = *templ;
+    rbuf->b.vtbl = &r300_buffer_vtbl;
+    pipe_reference_init(&rbuf->b.b.reference, 1);
+    rbuf->b.b.screen = screen;
+
+    if (rbuf->b.b.bind & R300_BIND_OQBO)
+        alignment = 4096;
+
+    rbuf->domain = R300_DOMAIN_GTT;
+
+    rbuf->buf = r300screen->rws->buffer_create(r300screen->rws,
+                                         alignment,
+                                         rbuf->b.b.bind,
+                                          rbuf->domain,
+                                         rbuf->b.b.width0);
+
+    if (!rbuf->buf)
+       goto error2;
+
+    return &rbuf->b.b;
+error2:
+    FREE(rbuf);
+error1:
+    return NULL;
 }
 
-void r300_screen_init_buffer_functions(struct r300_screen *r300screen)
+struct pipe_resource *r300_user_buffer_create(struct pipe_screen *screen,
+                                             void *ptr,
+                                             unsigned bytes,
+                                             unsigned bind)
 {
-    r300screen->screen.buffer_create = r300_buffer_create;
-    r300screen->screen.user_buffer_create = r300_user_buffer_create;
-    r300screen->screen.buffer_map = r300_buffer_map;
-    r300screen->screen.buffer_map_range = r300_buffer_map_range;
-    r300screen->screen.buffer_flush_mapped_range = r300_buffer_flush_mapped_range;
-    r300screen->screen.buffer_unmap = r300_buffer_unmap;
-    r300screen->screen.buffer_destroy = r300_buffer_destroy;
+    struct r300_buffer *rbuf;
+
+    rbuf = CALLOC_STRUCT(r300_buffer);
+    if (!rbuf)
+       goto no_rbuf;
+
+    rbuf->magic = R300_BUFFER_MAGIC;
+
+    pipe_reference_init(&rbuf->b.b.reference, 1);
+    rbuf->b.vtbl = &r300_buffer_vtbl;
+    rbuf->b.b.screen = screen;
+    rbuf->b.b.format = PIPE_FORMAT_R8_UNORM;
+    rbuf->b.b.usage = PIPE_USAGE_IMMUTABLE;
+    rbuf->b.b.bind = bind;
+    rbuf->b.b.width0 = bytes;
+    rbuf->b.b.height0 = 1;
+    rbuf->b.b.depth0 = 1;
+    rbuf->domain = R300_DOMAIN_GTT;
+
+    rbuf->user_buffer = ptr;
+    return &rbuf->b.b;
+
+no_rbuf:
+    return NULL;
 }