Subversion Repositories Kolibri OS

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

 *

 * Copyright 2006 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.

 *

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

/**

 * \file

 * Buffer manager using the old texture memory manager.

 *

 * \author Jose Fonseca 

 */

#include "pipe/p_defines.h"

#include "util/u_debug.h"

#include "os/os_thread.h"

#include "util/u_memory.h"

#include "util/list.h"

#include "util/u_mm.h"

#include "pb_buffer.h"

#include "pb_bufmgr.h"

/**

 * Convenience macro (type safe).

 */

#define SUPER(__derived) (&(__derived)->base)

struct mm_pb_manager

{

   struct pb_manager base;

   pipe_mutex mutex;

   pb_size size;

   struct mem_block *heap;

   pb_size align2;

   struct pb_buffer *buffer;

   void *map;

};

static INLINE struct mm_pb_manager *

mm_pb_manager(struct pb_manager *mgr)

{

   assert(mgr);

   return (struct mm_pb_manager *)mgr;

}

struct mm_buffer

{

   struct pb_buffer base;

   struct mm_pb_manager *mgr;

   struct mem_block *block;

};

static INLINE struct mm_buffer *

mm_buffer(struct pb_buffer *buf)

{

   assert(buf);

   return (struct mm_buffer *)buf;

}

static void

mm_buffer_destroy(struct pb_buffer *buf)

{

   struct mm_buffer *mm_buf = mm_buffer(buf);

   struct mm_pb_manager *mm = mm_buf->mgr;

   assert(!pipe_is_referenced(&mm_buf->base.reference));

   pipe_mutex_lock(mm->mutex);

   u_mmFreeMem(mm_buf->block);

   FREE(mm_buf);

   pipe_mutex_unlock(mm->mutex);

}

static void *

mm_buffer_map(struct pb_buffer *buf,

              unsigned flags,

              void *flush_ctx)

{

   struct mm_buffer *mm_buf = mm_buffer(buf);

   struct mm_pb_manager *mm = mm_buf->mgr;

   /* XXX: it will be necessary to remap here to propagate flush_ctx */

   return (unsigned char *) mm->map + mm_buf->block->ofs;

}

static void

mm_buffer_unmap(struct pb_buffer *buf)

{

   /* No-op */

}

static enum pipe_error

mm_buffer_validate(struct pb_buffer *buf,

                   struct pb_validate *vl,

                   unsigned flags)

{

   struct mm_buffer *mm_buf = mm_buffer(buf);

   struct mm_pb_manager *mm = mm_buf->mgr;

   return pb_validate(mm->buffer, vl, flags);

}

static void

mm_buffer_fence(struct pb_buffer *buf,

                struct pipe_fence_handle *fence)

{

   struct mm_buffer *mm_buf = mm_buffer(buf);

   struct mm_pb_manager *mm = mm_buf->mgr;

   pb_fence(mm->buffer, fence);

}

static void

mm_buffer_get_base_buffer(struct pb_buffer *buf,

                          struct pb_buffer **base_buf,

                          pb_size *offset)

{

   struct mm_buffer *mm_buf = mm_buffer(buf);

   struct mm_pb_manager *mm = mm_buf->mgr;

   pb_get_base_buffer(mm->buffer, base_buf, offset);

   *offset += mm_buf->block->ofs;

}

static const struct pb_vtbl

mm_buffer_vtbl = {

      mm_buffer_destroy,

      mm_buffer_map,

      mm_buffer_unmap,

      mm_buffer_validate,

      mm_buffer_fence,

      mm_buffer_get_base_buffer

};

static struct pb_buffer *

mm_bufmgr_create_buffer(struct pb_manager *mgr,

                        pb_size size,

                        const struct pb_desc *desc)

{

   struct mm_pb_manager *mm = mm_pb_manager(mgr);

   struct mm_buffer *mm_buf;

   /* We don't handle alignments larger then the one initially setup */

   assert(pb_check_alignment(desc->alignment, (pb_size)1 << mm->align2));

   if(!pb_check_alignment(desc->alignment, (pb_size)1 << mm->align2))

      return NULL;

   pipe_mutex_lock(mm->mutex);

   mm_buf = CALLOC_STRUCT(mm_buffer);

   if (!mm_buf) {

      pipe_mutex_unlock(mm->mutex);

      return NULL;

   }

   pipe_reference_init(&mm_buf->base.reference, 1);

   mm_buf->base.alignment = desc->alignment;

   mm_buf->base.usage = desc->usage;

   mm_buf->base.size = size;

   mm_buf->base.vtbl = &mm_buffer_vtbl;

   mm_buf->mgr = mm;

   mm_buf->block = u_mmAllocMem(mm->heap, (int)size, (int)mm->align2, 0);

   if(!mm_buf->block) {

#if 0

      debug_printf("warning: heap full\n");

      mmDumpMemInfo(mm->heap);

#endif

      FREE(mm_buf);

      pipe_mutex_unlock(mm->mutex);

      return NULL;

   }

   /* Some sanity checks */

   assert(0 <= (pb_size)mm_buf->block->ofs && (pb_size)mm_buf->block->ofs < mm->size);

   assert(size <= (pb_size)mm_buf->block->size && (pb_size)mm_buf->block->ofs + (pb_size)mm_buf->block->size <= mm->size);

   pipe_mutex_unlock(mm->mutex);

   return SUPER(mm_buf);

}

static void

mm_bufmgr_flush(struct pb_manager *mgr)

{

   /* No-op */

}

static void

mm_bufmgr_destroy(struct pb_manager *mgr)

{

   struct mm_pb_manager *mm = mm_pb_manager(mgr);

   pipe_mutex_lock(mm->mutex);

   u_mmDestroy(mm->heap);

   pb_unmap(mm->buffer);

   pb_reference(&mm->buffer, NULL);

   pipe_mutex_unlock(mm->mutex);

   FREE(mgr);

}

struct pb_manager *

mm_bufmgr_create_from_buffer(struct pb_buffer *buffer,

                             pb_size size, pb_size align2)

{

   struct mm_pb_manager *mm;

   if(!buffer)

      return NULL;

   mm = CALLOC_STRUCT(mm_pb_manager);

   if (!mm)

      return NULL;

   mm->base.destroy = mm_bufmgr_destroy;

   mm->base.create_buffer = mm_bufmgr_create_buffer;

   mm->base.flush = mm_bufmgr_flush;

   mm->size = size;

   mm->align2 = align2; /* 64-byte alignment */

   pipe_mutex_init(mm->mutex);

   mm->buffer = buffer;

   mm->map = pb_map(mm->buffer,

		    PB_USAGE_CPU_READ |

		    PB_USAGE_CPU_WRITE, NULL);

   if(!mm->map)

      goto failure;

   mm->heap = u_mmInit(0, (int)size);

   if (!mm->heap)

      goto failure;

   return SUPER(mm);

failure:

if(mm->heap)

   u_mmDestroy(mm->heap);

   if(mm->map)

      pb_unmap(mm->buffer);

   FREE(mm);

   return NULL;

}

struct pb_manager *

mm_bufmgr_create(struct pb_manager *provider,

                 pb_size size, pb_size align2)

{

   struct pb_buffer *buffer;

   struct pb_manager *mgr;

   struct pb_desc desc;

   if(!provider)

      return NULL;

   memset(&desc, 0, sizeof(desc));

   desc.alignment = 1 << align2;

   buffer = provider->create_buffer(provider, size, &desc);

   if (!buffer)

      return NULL;

   mgr = mm_bufmgr_create_from_buffer(buffer, size, align2);

   if (!mgr) {

      pb_reference(&buffer, NULL);

      return NULL;

   }

  return mgr;

}