#include "pipe/p_context.h"
#include "nvc0_resource.h"
#include "nouveau/nouveau_screen.h"

static unsigned
nvc0_resource_is_referenced(struct pipe_context *pipe,
                            struct pipe_resource *resource,
                            unsigned face, unsigned level)
{
   struct nvc0_resource *res = nvc0_resource(resource);
   unsigned flags = 0;

#ifdef NOUVEAU_USERSPACE_MM
   flags = res->status;
#else
   unsigned bo_flags = nouveau_bo_pending(res->bo);
   if (bo_flags & NOUVEAU_BO_RD)
      flags = PIPE_REFERENCED_FOR_READ;
   if (bo_flags & NOUVEAU_BO_WR)
      flags |= PIPE_REFERENCED_FOR_WRITE;
#endif
   return flags;
}

static struct pipe_resource *
nvc0_resource_create(struct pipe_screen *screen,
                     const struct pipe_resource *templ)
{
   switch (templ->target) {
   case PIPE_BUFFER:
      return nvc0_buffer_create(screen, templ);
   default:
      return nvc0_miptree_create(screen, templ);
   }
}

static struct pipe_resource *
nvc0_resource_from_handle(struct pipe_screen * screen,
                          const struct pipe_resource *templ,
                          struct winsys_handle *whandle)
{
   if (templ->target == PIPE_BUFFER)
      return NULL;
   else
      return nvc0_miptree_from_handle(screen, templ, whandle);
}

void
nvc0_init_resource_functions(struct pipe_context *pcontext)
{
   pcontext->get_transfer = u_get_transfer_vtbl;
   pcontext->transfer_map = u_transfer_map_vtbl;
   pcontext->transfer_flush_region = u_transfer_flush_region_vtbl;
   pcontext->transfer_unmap = u_transfer_unmap_vtbl;
   pcontext->transfer_destroy = u_transfer_destroy_vtbl;
   pcontext->transfer_inline_write = u_transfer_inline_write_vtbl;
   pcontext->is_resource_referenced = nvc0_resource_is_referenced;
}

void
nvc0_screen_init_resource_functions(struct pipe_screen *pscreen)
{
   pscreen->resource_create = nvc0_resource_create;
   pscreen->resource_from_handle = nvc0_resource_from_handle;
   pscreen->resource_get_handle = u_resource_get_handle_vtbl;
   pscreen->resource_destroy = u_resource_destroy_vtbl;
   pscreen->user_buffer_create = nvc0_user_buffer_create;
   pscreen->get_tex_surface = nvc0_miptree_surface_new;
   pscreen->tex_surface_destroy = nvc0_miptree_surface_del;
}