#include "pipe/p_context.h"
#include "util/u_staging.h"
#include "nvfx_resource.h"
#include "nvfx_transfer.h"
#include "nouveau/nouveau_screen.h"

static unsigned int
nvfx_resource_is_referenced(struct pipe_context *pipe,
			    struct pipe_resource *pr,
			    unsigned face, unsigned level)
{
	return !!nouveau_reference_flags(nvfx_resource(pr)->bo);
}

static struct pipe_resource *
nvfx_resource_create(struct pipe_screen *screen,
		     const struct pipe_resource *template)
{
	if (template->target == PIPE_BUFFER)
		return nvfx_buffer_create(screen, template);
	else
		return nvfx_miptree_create(screen, template);
}

static void
nvfx_resource_destroy(struct pipe_screen *screen, struct pipe_resource *pr)
{
	if (pr->target == PIPE_BUFFER)
		return nvfx_buffer_destroy(screen, pr);
	else
		return nvfx_miptree_destroy(screen, pr);
}

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

static boolean
nvfx_resource_get_handle(struct pipe_screen *pscreen,
                        struct pipe_resource *pr,
                        struct winsys_handle *whandle)
{
	struct nvfx_resource* res = (struct nvfx_resource*)pr;

	if (!res || !res->bo)
		return FALSE;

	return nouveau_screen_bo_get_handle(pscreen, res->bo, nvfx_subresource_pitch(pr, 0), whandle);
}

void
nvfx_init_resource_functions(struct pipe_context *pipe)
{
	pipe->get_transfer = nvfx_transfer_new;
	pipe->transfer_map = nvfx_transfer_map;
	pipe->transfer_flush_region = u_default_transfer_flush_region;
	pipe->transfer_unmap = nvfx_transfer_unmap;
	pipe->transfer_destroy = util_staging_transfer_destroy;
	pipe->transfer_inline_write = u_default_transfer_inline_write;
	pipe->is_resource_referenced = nvfx_resource_is_referenced;
}

void
nvfx_screen_init_resource_functions(struct pipe_screen *pscreen)
{
	pscreen->resource_create = nvfx_resource_create;
	pscreen->resource_from_handle = nvfx_resource_from_handle;
	pscreen->resource_get_handle = nvfx_resource_get_handle;
	pscreen->resource_destroy = nvfx_resource_destroy;
	pscreen->user_buffer_create = nvfx_user_buffer_create;

	pscreen->get_tex_surface = nvfx_miptree_surface_new;
	pscreen->tex_surface_destroy = nvfx_miptree_surface_del;
}