/*
 * Copyright (C) 2009 Francisco Jerez.
 * 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, sublicense, 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 NONINFRINGEMENT.
 * IN NO EVENT SHALL THE COPYRIGHT OWNER(S) 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.
 *
 */

#include "nouveau_driver.h"
#include "nouveau_context.h"
#include "nouveau_bufferobj.h"
#include "nouveau_fbo.h"

#include "main/dd.h"
#include "main/framebuffer.h"
#include "main/light.h"
#include "main/state.h"
#include "drivers/common/meta.h"
#include "drivers/common/driverfuncs.h"
#include "swrast/swrast.h"
#include "vbo/vbo.h"
#include "tnl/tnl.h"

#define need_GL_EXT_framebuffer_object
#define need_GL_EXT_fog_coord

#include "main/remap_helper.h"

static const struct dri_extension nouveau_extensions[] = {
	{ "GL_EXT_framebuffer_object",  GL_EXT_framebuffer_object_functions },
	{ "GL_ARB_multitexture",	NULL },
	{ "GL_EXT_texture_lod_bias",	NULL },
	{ "GL_SGIS_generate_mipmap",	NULL },
	{ "GL_ARB_texture_env_combine",	NULL },
	{ "GL_ARB_texture_env_dot3",	NULL },
	{ "GL_ARB_texture_env_add",	NULL },
	{ "GL_EXT_fog_coord",		GL_EXT_fog_coord_functions },
	{ NULL,				NULL }
};

GLboolean
nouveau_context_create(const __GLcontextModes *visual, __DRIcontext *dri_ctx,
		       void *share_ctx)
{
	__DRIscreen *dri_screen = dri_ctx->driScreenPriv;
	struct nouveau_screen *screen = dri_screen->private;
	struct nouveau_context *nctx;
	GLcontext *ctx;

	ctx = screen->driver->context_create(screen, visual, share_ctx);
	if (!ctx)
		return GL_FALSE;

	nctx = to_nouveau_context(ctx);
	nctx->dri_context = dri_ctx;
	dri_ctx->driverPrivate = ctx;

	return GL_TRUE;
}

GLboolean
nouveau_context_init(GLcontext *ctx, struct nouveau_screen *screen,
		     const GLvisual *visual, GLcontext *share_ctx)
{
	struct nouveau_context *nctx = to_nouveau_context(ctx);
	struct dd_function_table functions;

	nctx->screen = screen;
	nctx->fallback = HWTNL;

	/* Initialize the function pointers */
	_mesa_init_driver_functions(&functions);
	nouveau_driver_functions_init(&functions);
	nouveau_bufferobj_functions_init(&functions);
	nouveau_texture_functions_init(&functions);
	nouveau_fbo_functions_init(&functions);

	/* Initialize the mesa context */
	_mesa_initialize_context(ctx, visual, share_ctx, &functions, NULL);

	nouveau_state_init(ctx);
	nouveau_bo_state_init(ctx);
	_mesa_meta_init(ctx);
	_swrast_CreateContext(ctx);
	_vbo_CreateContext(ctx);
	_tnl_CreateContext(ctx);
	nouveau_span_functions_init(ctx);
	_mesa_allow_light_in_model(ctx, GL_FALSE);

	/* Enable any supported extensions */
	driInitExtensions(ctx, nouveau_extensions, GL_TRUE);

	return GL_TRUE;
}

void
nouveau_context_destroy(__DRIcontext *dri_ctx)
{
	struct nouveau_context *nctx = dri_ctx->driverPrivate;
	GLcontext *ctx = &nctx->base;

	if (nctx->screen->context == nctx)
		nctx->screen->context = NULL;

	_tnl_DestroyContext(ctx);
	_vbo_DestroyContext(ctx);
	_swrast_DestroyContext(ctx);
	_mesa_meta_free(ctx);
	nouveau_bo_state_destroy(ctx);
	context_drv(ctx)->context_destroy(ctx);
}

static void
nouveau_update_renderbuffers(__DRIcontext *context, __DRIdrawable *drawable,
			     unsigned int *stamp)
{
	struct nouveau_context *nctx = context->driverPrivate;
	GLcontext *ctx = &nctx->base;
	__DRIscreen *screen = context->driScreenPriv;
	struct gl_framebuffer *fb = drawable->driverPrivate;
	unsigned int attachments[10];
	__DRIbuffer *buffers = NULL;
	int i = 0, count, ret;

	attachments[i++] = __DRI_BUFFER_FRONT_LEFT;
	if (fb->Visual.doubleBufferMode)
		attachments[i++] = __DRI_BUFFER_BACK_LEFT;
	if (fb->Visual.haveDepthBuffer && fb->Visual.haveStencilBuffer)
		attachments[i++] = __DRI_BUFFER_DEPTH_STENCIL;
	else if (fb->Visual.haveDepthBuffer)
		attachments[i++] = __DRI_BUFFER_DEPTH;
	else if (fb->Visual.haveStencilBuffer)
		attachments[i++] = __DRI_BUFFER_STENCIL;

	buffers = (*screen->dri2.loader->getBuffers)(drawable,
						     &drawable->w, &drawable->h,
						     attachments, i, &count,
						     drawable->loaderPrivate);
	if (buffers == NULL)
		return;

	for (i = 0; i < count; i++) {
		struct gl_renderbuffer *rb;
		struct nouveau_surface *s;
		uint32_t old_handle;
		int index;

		switch (buffers[i].attachment) {
		case __DRI_BUFFER_FRONT_LEFT:
		case __DRI_BUFFER_FAKE_FRONT_LEFT:
			index = BUFFER_FRONT_LEFT;
			break;
		case __DRI_BUFFER_BACK_LEFT:
			index = BUFFER_BACK_LEFT;
			break;
		case __DRI_BUFFER_DEPTH:
		case __DRI_BUFFER_DEPTH_STENCIL:
			index = BUFFER_DEPTH;
			break;
		case __DRI_BUFFER_STENCIL:
			index = BUFFER_STENCIL;
			break;
		default:
			assert(0);
		}

		rb = fb->Attachment[index].Renderbuffer;
		s = &to_nouveau_renderbuffer(rb)->surface;

		s->width = drawable->w;
		s->height = drawable->h;
		s->pitch = buffers[i].pitch;
		s->cpp = buffers[i].cpp;

		/* Don't bother to reopen the bo if it happens to be
		 * the same. */
		if (s->bo) {
			ret = nouveau_bo_handle_get(s->bo, &old_handle);
			assert(!ret);
		}

		if (!s->bo || old_handle != buffers[i].name) {
			nouveau_bo_ref(NULL, &s->bo);
			ret = nouveau_bo_handle_ref(context_dev(ctx),
						    buffers[i].name, &s->bo);
			assert(!ret);

			context_dirty(ctx, FRAMEBUFFER);
		}
	}

	_mesa_resize_framebuffer(ctx, fb, drawable->w, drawable->h);
}

GLboolean
nouveau_context_make_current(__DRIcontext *dri_ctx, __DRIdrawable *dri_draw,
			     __DRIdrawable *dri_read)
{
	if (dri_ctx) {
		struct nouveau_context *nctx = dri_ctx->driverPrivate;
		GLcontext *ctx = &nctx->base;

		if (nctx->screen->context != nctx) {
			nctx->screen->context = nctx;
			BITSET_ONES(nctx->dirty);
		}

		/* Ask the X server for new renderbuffers. */
		nouveau_update_renderbuffers(dri_ctx, dri_draw,
					     &nctx->drawable.d_stamp);
		if (dri_draw != dri_read)
			nouveau_update_renderbuffers(dri_ctx, dri_read,
						     &nctx->drawable.r_stamp);

		/* Pass it down to mesa. */
		_mesa_make_current(ctx, dri_draw->driverPrivate,
				   dri_read->driverPrivate);
		_mesa_update_state(ctx);

		FIRE_RING(context_chan(ctx));

	} else {
		_mesa_make_current(NULL, NULL, NULL);
	}

	return GL_TRUE;
}

GLboolean
nouveau_context_unbind(__DRIcontext *dri_ctx)
{
	return GL_TRUE;
}

void
nouveau_fallback(GLcontext *ctx, enum nouveau_fallback mode)
{
	struct nouveau_context *nctx = to_nouveau_context(ctx);

	nctx->fallback = MAX2(HWTNL, mode);

	if (mode < SWRAST)
		nouveau_state_emit(ctx);
	else
		FIRE_RING(context_chan(ctx));
}

void
nouveau_validate_framebuffer(GLcontext *ctx)
{
	struct nouveau_context *nctx = to_nouveau_context(ctx);

	/* Someone's planning to draw something really soon. */
	nctx->drawable.dirty = GL_TRUE;
}