Subversion Repositories Kolibri OS

/**************************************************************************

 *

 * Copyright 2009 VMware, Inc.

 * All Rights Reserved.

 *

 * Permission is hereby granted, free of charge, to any person obtaining a

 * copy of this software and associated documentation files (the

 * "Software"), to deal in the Software without restriction, including

 * without limitation the rights to use, copy, modify, merge, publish,

 * distribute, sub license, and/or sell copies of the Software, and to

 * permit persons to whom the Software is furnished to do so, subject to

 * the following conditions:

 *

 * The above copyright notice and this permission notice (including the

 * next paragraph) shall be included in all copies or substantial portions

 * of the Software.

 *

 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS

 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF

 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.

 * IN NO EVENT SHALL VMWARE AND/OR ITS SUPPLIERS BE LIABLE FOR

 * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,

 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE

 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

 *

 **************************************************************************/

/* Helper utility for uploading user buffers & other data, and

 * coalescing small buffers into larger ones.

 */

#include "pipe/p_defines.h"

#include "util/u_inlines.h"

#include "pipe/p_context.h"

#include "util/u_memory.h"

#include "util/u_math.h"

#include "u_upload_mgr.h"

struct u_upload_mgr {

   struct pipe_context *pipe;

   unsigned default_size;  /* Minimum size of the upload buffer, in bytes. */

   unsigned alignment;     /* Alignment of each sub-allocation. */

   unsigned bind;          /* Bitmask of PIPE_BIND_* flags. */

   struct pipe_resource *buffer;   /* Upload buffer. */

   struct pipe_transfer *transfer; /* Transfer object for the upload buffer. */

   uint8_t *map;    /* Pointer to the mapped upload buffer. */

   unsigned size;   /* Actual size of the upload buffer. */

   unsigned offset; /* Aligned offset to the upload buffer, pointing

                     * at the first unused byte. */

};

struct u_upload_mgr *u_upload_create( struct pipe_context *pipe,

                                      unsigned default_size,

                                      unsigned alignment,

                                      unsigned bind )

{

   struct u_upload_mgr *upload = CALLOC_STRUCT( u_upload_mgr );

   if (!upload)

      return NULL;

   upload->pipe = pipe;

   upload->default_size = default_size;

   upload->alignment = alignment;

   upload->bind = bind;

   upload->buffer = NULL;

   return upload;

}

void u_upload_unmap( struct u_upload_mgr *upload )

{

   if (upload->transfer) {

      struct pipe_box *box = &upload->transfer->box;

      if ((int) upload->offset > box->x) {

         pipe_buffer_flush_mapped_range(upload->pipe, upload->transfer,

                                        box->x, upload->offset - box->x);

      }

      pipe_transfer_unmap(upload->pipe, upload->transfer);

      upload->transfer = NULL;

      upload->map = NULL;

   }

}

/* Release old buffer.

 *

 * This must usually be called prior to firing the command stream

 * which references the upload buffer, as many memory managers will

 * cause subsequent maps of a fired buffer to wait.

 *

 * Can improve this with a change to pipe_buffer_write to use the

 * DONT_WAIT bit, but for now, it's easiest just to grab a new buffer.

 */

void u_upload_flush( struct u_upload_mgr *upload )

{

   /* Unmap and unreference the upload buffer. */

   u_upload_unmap(upload);

   pipe_resource_reference( &upload->buffer, NULL );

   upload->size = 0;

}

void u_upload_destroy( struct u_upload_mgr *upload )

{

   u_upload_flush( upload );

   FREE( upload );

}

static enum pipe_error

u_upload_alloc_buffer( struct u_upload_mgr *upload,

                       unsigned min_size )

{

   unsigned size;

   /* Release the old buffer, if present:

    */

   u_upload_flush( upload );

   /* Allocate a new one:

    */

   size = align(MAX2(upload->default_size, min_size), 4096);

   upload->buffer = pipe_buffer_create( upload->pipe->screen,

                                        upload->bind,

                                        PIPE_USAGE_STREAM,

                                        size );

   if (upload->buffer == NULL) {

      return PIPE_ERROR_OUT_OF_MEMORY;

   }

   /* Map the new buffer. */

   upload->map = pipe_buffer_map_range(upload->pipe, upload->buffer,

                                       0, size,

                                       PIPE_TRANSFER_WRITE |

                                       PIPE_TRANSFER_FLUSH_EXPLICIT,

                                       &upload->transfer);

   if (upload->map == NULL) {

      upload->transfer = NULL;

      upload->size = 0;

      pipe_resource_reference(&upload->buffer, NULL);

      return PIPE_ERROR_OUT_OF_MEMORY;

   }

   upload->size = size;

   upload->offset = 0;

   return PIPE_OK;

}

enum pipe_error u_upload_alloc( struct u_upload_mgr *upload,

                                unsigned min_out_offset,

                                unsigned size,

                                unsigned *out_offset,

                                struct pipe_resource **outbuf,

                                void **ptr )

{

   unsigned alloc_size = align( size, upload->alignment );

   unsigned alloc_offset = align(min_out_offset, upload->alignment);

   unsigned offset;

   /* Init these return values here in case we fail below to make

    * sure the caller doesn't get garbage values.

    */

   *out_offset = ~0;

   pipe_resource_reference(outbuf, NULL);

   *ptr = NULL;

   /* Make sure we have enough space in the upload buffer

    * for the sub-allocation. */

   if (MAX2(upload->offset, alloc_offset) + alloc_size > upload->size) {

      enum pipe_error ret = u_upload_alloc_buffer(upload,

                                                  alloc_offset + alloc_size);

      if (ret != PIPE_OK)

         return ret;

   }

   offset = MAX2(upload->offset, alloc_offset);

   if (!upload->map) {

      upload->map = pipe_buffer_map_range(upload->pipe, upload->buffer,

					  offset, upload->size - offset,

					  PIPE_TRANSFER_WRITE |

					  PIPE_TRANSFER_FLUSH_EXPLICIT |

					  PIPE_TRANSFER_UNSYNCHRONIZED,

					  &upload->transfer);

      if (!upload->map) {

         upload->transfer = NULL;

         return PIPE_ERROR_OUT_OF_MEMORY;

      }

      upload->map -= offset;

   }

   assert(offset < upload->buffer->width0);

   assert(offset + size <= upload->buffer->width0);

   assert(size);

   /* Emit the return values: */

   *ptr = upload->map + offset;

   pipe_resource_reference( outbuf, upload->buffer );

   *out_offset = offset;

   upload->offset = offset + alloc_size;

   return PIPE_OK;

}

enum pipe_error u_upload_data( struct u_upload_mgr *upload,

                               unsigned min_out_offset,

                               unsigned size,

                               const void *data,

                               unsigned *out_offset,

                               struct pipe_resource **outbuf)

{

   uint8_t *ptr;

   enum pipe_error ret = u_upload_alloc(upload, min_out_offset, size,

                                        out_offset, outbuf,

                                        (void**)&ptr);

   if (ret != PIPE_OK)

      return ret;

   memcpy(ptr, data, size);

   return PIPE_OK;

}

/* As above, but upload the full contents of a buffer.  Useful for

 * uploading user buffers, avoids generating an explosion of GPU

 * buffers if you have an app that does lots of small vertex buffer

 * renders or DrawElements calls.

 */

enum pipe_error u_upload_buffer( struct u_upload_mgr *upload,

                                 unsigned min_out_offset,

                                 unsigned offset,

                                 unsigned size,

                                 struct pipe_resource *inbuf,

                                 unsigned *out_offset,

                                 struct pipe_resource **outbuf)

{

   enum pipe_error ret = PIPE_OK;

   struct pipe_transfer *transfer = NULL;

   const char *map = NULL;

   map = (const char *)pipe_buffer_map_range(upload->pipe,

                                             inbuf,

                                             offset, size,

                                             PIPE_TRANSFER_READ,

                                             &transfer);

   if (map == NULL) {

      return PIPE_ERROR_OUT_OF_MEMORY;

   }

   if (0)

      debug_printf("upload ptr %p ofs %d sz %d\n", map, offset, size);

   ret = u_upload_data( upload,

                        min_out_offset,

                        size,

                        map,

                        out_offset,

                        outbuf);

   pipe_buffer_unmap( upload->pipe, transfer );

   return ret;

}