/* $XFree86$ */
/**************************************************************************

Copyright 1999, 2000 ATI Technologies Inc. and Precision Insight, Inc.,
                                               Cedar Park, Texas. 
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
on 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
ATI, PRECISION INSIGHT AND/OR THEIR 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.

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

/*
 * Authors:
 *   Kevin E. Martin <kevin@precisioninsight.com>
 *
 */

#include "r128_init.h"
#include "r128_mesa.h"
#include "r128_xmesa.h"
#include "r128_context.h"
#include "r128_lock.h"
#include "r128_reg.h"
#include "r128_cce.h"
#include "r128_state.h"
#include "r128_vb.h"

#include "stages.h"

#define TEX0                                                            \
do {                                                                    \
    v->v.tu0 = tc0[i][0];                                               \
    v->v.tv0 = tc0[i][1];                                               \
} while (0)

#define TEX1                                                            \
do {                                                                    \
    v->v.tu1 = tc1[i][0];                                               \
    v->v.tv1 = tc1[i][1];                                               \
} while (0)

#define SPC                                                             \
do {                                                                    \
    GLubyte *spec = &(VB->Spec[0][i][0]);                               \
    v->v.spec_frgb.r = spec[0];                                         \
    v->v.spec_frgb.g = spec[1];                                         \
    v->v.spec_frgb.b = spec[2];                                         \
} while (0)


#define COL                                                             \
do {                                                                    \
    GLubyte *col = &(VB->Color[0]->data[i][0]);                         \
    v->v.dif_argb.a = col[3];                                           \
    v->v.dif_argb.r = col[0];                                           \
    v->v.dif_argb.g = col[1];                                           \
    v->v.dif_argb.b = col[2];                                           \
} while (0)

#if 1
/* FIXME: These are handled by the Rage 128 */
#define TEX0_4
#define TEX1_4
#else
#define TEX0_4                                                          \
do {                                                                    \
    if (VB->TexCoordPtr[0]->size == 4) {                                \
	GLfloat (*tc)[4] = VB->TexCoordPtr[0]->data;                    \
	v = &(R128_DRIVER_DATA(VB)->verts[start]);                      \
	for (i = start; i < end; i++, v++)   {                          \
	    float oow = 1.0 / tc[i][3];                                 \
	    v->v.rhw *= tc[i][3];                                       \
	    v->v.tu0 *= oow;                                            \
	    v->v.tv0 *= oow;                                            \
	}                                                               \
    }                                                                   \
} while (0)

#if USE_RHW2
#define TEX1_4                                                          \
do {                                                                    \
    if (VB->TexCoordPtr[1]->size == 4) {                                \
	GLfloat (*tc)[4] = VB->TexCoordPtr[1]->data;                    \
	v = &(R128_DRIVER_DATA(VB)->verts[start]);                      \
	for (i = start; i < end; i++, v++)   {                          \
	    float oow = 1.0 / tc[i][3];                                 \
	    v->v.rhw2 *= tc[i][3];                                      \
	    v->v.tu1  *= oow;                                           \
	    v->v.tv1  *= oow;                                           \
	}                                                               \
    }                                                                   \
} while (0)
#else
#define TEX1_4
#endif
#endif

#if USE_RHW2
#define COORD                                                           \
do {                                                                    \
    GLfloat *win = VB->Win.data[i];                                     \
    v->v.x =                win[0];                                     \
    v->v.y =   r128height - win[1];                                     \
    v->v.z =        scale * win[2];                                     \
    v->v.rhw = v->v.rhw2 =  win[3];                                     \
} while (0)
#else
#define COORD                                                           \
do {                                                                    \
    GLfloat *win = VB->Win.data[i];                                     \
    v->v.x =                win[0];                                     \
    v->v.y =   r128height - win[1];                                     \
    v->v.z =        scale * win[2];                                     \
    v->v.rhw =              win[3];                                     \
} while (0)
#endif

#define NOP

/* Setup the r128 vertex buffer entries */
#define SETUPFUNC(name,win,col,tex0,tex1,tex0_4,tex1_4,spec)            \
static void name(struct vertex_buffer *VB, GLuint start, GLuint end)    \
{                                                                       \
    r128ContextPtr        r128ctx    = R128_CONTEXT(VB->ctx);           \
    __DRIdrawablePrivate *dPriv      = r128ctx->driDrawable;            \
    r128VertexPtr         v;                                            \
    GLfloat               (*tc0)[4];                                    \
    GLfloat               (*tc1)[4];                                    \
    GLfloat               r128height = dPriv->h;                        \
    GLfloat               scale;                                        \
    int                   i;                                            \
                                                                        \
    (void) r128height; (void) r128ctx;                                  \
                                                                        \
    gl_import_client_data(VB, VB->ctx->RenderFlags,                     \
			  (VB->ClipOrMask                               \
			   ? VEC_WRITABLE | VEC_GOOD_STRIDE             \
			   : VEC_GOOD_STRIDE));                         \
                                                                        \
    switch (VB->ctx->Visual->DepthBits) {                               \
    case 16: scale = 1.0 / 65536.0;      break;                         \
    case 24: scale = 1.0 / 16777216.0;   break;                         \
    case 32: scale = 1.0 / 4294967296.0; break;                         \
    default: scale = 1.0 / 65536.0;      break;                         \
    }                                                                   \
                                                                        \
    tc0 = VB->TexCoordPtr[0]->data;                                     \
    tc1 = VB->TexCoordPtr[1]->data;                                     \
                                                                        \
    v = &(R128_DRIVER_DATA(VB)->verts[start]);                          \
                                                                        \
    if (VB->ClipOrMask == 0)                                            \
	for (i = start; i < end; i++, v++) {                            \
	    win;                                                        \
	    col;                                                        \
	    spec;                                                       \
	    tex0;                                                       \
	    tex1;                                                       \
	}                                                               \
    else                                                                \
	for (i = start; i < end; i++, v++) {                            \
	    if (VB->ClipMask[i] == 0) {                                 \
		win;                                                    \
		spec;                                                   \
		tex0;                                                   \
		tex1;                                                   \
	    }                                                           \
	    col;                                                        \
	}                                                               \
    tex0_4;                                                             \
    tex1_4;                                                             \
}


SETUPFUNC(rs_wt0,       COORD, NOP, TEX0, NOP,  TEX0_4, NOP,    NOP)
SETUPFUNC(rs_wt1,       COORD, NOP, NOP,  TEX1, NOP,    TEX1_4, NOP)
SETUPFUNC(rs_wt0t1,     COORD, NOP, TEX0, TEX1, TEX0_4, TEX1_4, NOP)
SETUPFUNC(rs_wg,        COORD, COL, NOP,  NOP,  NOP,    NOP,    NOP)
SETUPFUNC(rs_wgs,       COORD, COL, NOP,  NOP,  NOP,    NOP,    SPC)
SETUPFUNC(rs_wgt0,      COORD, COL, TEX0, NOP,  TEX0_4, NOP,    NOP)
SETUPFUNC(rs_wgt1,      COORD, COL, NOP,  TEX1, NOP,    TEX1_4, NOP)
SETUPFUNC(rs_wgt0t1,    COORD, COL, TEX0, TEX1, TEX0_4, TEX1_4, NOP)
SETUPFUNC(rs_wgst0,     COORD, COL, TEX0, NOP,  TEX0_4, NOP,    SPC)
SETUPFUNC(rs_wgst1,     COORD, COL, NOP,  TEX1, NOP,    TEX1_4, SPC)
SETUPFUNC(rs_wgst0t1,   COORD, COL, TEX0, TEX1, TEX0_4, TEX1_4, SPC)

SETUPFUNC(rs_t0,        NOP,   NOP, TEX0, NOP,  TEX0_4, NOP,    NOP)
SETUPFUNC(rs_t1,        NOP,   NOP, NOP,  TEX1, NOP,    TEX1_4, NOP)
SETUPFUNC(rs_t0t1,      NOP,   NOP, TEX0, TEX1, TEX0_4, TEX1_4, NOP)
SETUPFUNC(rs_g,         NOP,   COL, NOP,  NOP,  NOP,    NOP,    NOP)
SETUPFUNC(rs_gs,        NOP,   COL, NOP,  NOP,  NOP,    NOP,    SPC)
SETUPFUNC(rs_gt0,       NOP,   COL, TEX0, NOP,  TEX0_4, NOP,    NOP)
SETUPFUNC(rs_gt1,       NOP,   COL, NOP,  TEX1, NOP,    TEX1_4, NOP)
SETUPFUNC(rs_gt0t1,     NOP,   COL, TEX0, TEX1, TEX0_4, TEX1_4, NOP)
SETUPFUNC(rs_gst0,      NOP,   COL, TEX0, NOP,  TEX0_4, NOP,    SPC)
SETUPFUNC(rs_gst1,      NOP,   COL, NOP,  TEX1, NOP,    TEX1_4, SPC)
SETUPFUNC(rs_gst0t1,    NOP,   COL, TEX0, TEX1, TEX0_4, TEX1_4, SPC)

static void rs_invalid(struct vertex_buffer *VB, GLuint start, GLuint end)
{
    fprintf(stderr, "r128RasterSetup(): invalid setup function\n");
}

typedef void (*setupFunc)(struct vertex_buffer *, GLuint, GLuint);
static setupFunc setup_func[0x20];

/* Initialize the table of vertex buffer setup functions */
void r128SetupInit(void)
{
    int i;

    for (i = 0; i < 0x20; i++) setup_func[i] = rs_invalid;

    /* Funcs to build vertices from scratch */
    setup_func[R128_WIN_BIT|R128_TEX0_BIT] = rs_wt0;
    setup_func[R128_WIN_BIT|R128_TEX1_BIT] = rs_wt1;
    setup_func[R128_WIN_BIT|R128_TEX0_BIT|R128_TEX1_BIT] = rs_wt0t1;

    setup_func[R128_WIN_BIT|R128_RGBA_BIT] = rs_wg;
    setup_func[R128_WIN_BIT|R128_RGBA_BIT|R128_SPEC_BIT] = rs_wgs;
    setup_func[R128_WIN_BIT|R128_RGBA_BIT|R128_TEX0_BIT] = rs_wgt0;
    setup_func[R128_WIN_BIT|R128_RGBA_BIT|R128_TEX1_BIT] = rs_wgt1;
    setup_func[R128_WIN_BIT|R128_RGBA_BIT|R128_TEX0_BIT|R128_TEX1_BIT] = rs_wgt0t1;
    setup_func[R128_WIN_BIT|R128_RGBA_BIT|R128_SPEC_BIT|R128_TEX0_BIT] = rs_wgst0;
    setup_func[R128_WIN_BIT|R128_RGBA_BIT|R128_SPEC_BIT|R128_TEX1_BIT] = rs_wgst1;
    setup_func[R128_WIN_BIT|R128_RGBA_BIT|R128_SPEC_BIT|R128_TEX0_BIT|R128_TEX1_BIT] = rs_wgst0t1;


    /* Funcs to repair vertices */
    setup_func[R128_TEX0_BIT] = rs_t0;
    setup_func[R128_TEX1_BIT] = rs_t1;
    setup_func[R128_TEX0_BIT|R128_TEX1_BIT] = rs_t0t1;
    setup_func[R128_RGBA_BIT] = rs_g;
    setup_func[R128_RGBA_BIT|R128_SPEC_BIT] = rs_gs;
    setup_func[R128_RGBA_BIT|R128_TEX0_BIT] = rs_gt0;
    setup_func[R128_RGBA_BIT|R128_TEX1_BIT] = rs_gt1;
    setup_func[R128_RGBA_BIT|R128_TEX0_BIT|R128_TEX1_BIT] = rs_gt0t1;
    setup_func[R128_RGBA_BIT|R128_SPEC_BIT|R128_TEX0_BIT] = rs_gst0;
    setup_func[R128_RGBA_BIT|R128_SPEC_BIT|R128_TEX1_BIT] = rs_gst1;
    setup_func[R128_RGBA_BIT|R128_SPEC_BIT|R128_TEX0_BIT|R128_TEX1_BIT] = rs_gst0t1;
}

/* Initialize the vertex buffer setup functions based on the current
   rendering state */
void r128ChooseRasterSetupFunc(GLcontext *ctx)
{
    r128ContextPtr r128ctx   = R128_CONTEXT(ctx);
    int            funcIndex = R128_WIN_BIT | R128_RGBA_BIT;

    if (ctx->Texture.Enabled & 0xf) {
	if (ctx->Texture.Unit[0].EnvMode == GL_REPLACE)
	    funcIndex &= ~R128_RGBA_BIT;
	funcIndex |= R128_TEX0_BIT;
    }

    if (ctx->Texture.Enabled & 0xf0)
	funcIndex |= R128_TEX1_BIT;

    /* FIXME: Verify this works properly */
    if (ctx->Light.Model.ColorControl == GL_SEPARATE_SPECULAR_COLOR)
	funcIndex |= R128_SPEC_BIT;

    r128ctx->SetupIndex = funcIndex;
    ctx->Driver.RasterSetup = setup_func[funcIndex];
}

/* Check to see if any updates of the vertex buffer entries are needed */
void r128CheckPartialRasterSetup(GLcontext *ctx,
				 struct gl_pipeline_stage *s)
{
    r128ContextPtr r128ctx = R128_CONTEXT(ctx);
    int            tmp     = r128ctx->SetupDone;

    s->type = 0;
    r128ctx->SetupDone = GL_FALSE;

    if ((ctx->Array.Summary & VERT_OBJ_ANY) == 0) return;
    if (ctx->IndirectTriangles) return;

    r128ctx->SetupDone = tmp;
}


/* Update the vertex buffer entries, if necessary */
void r128PartialRasterSetup(struct vertex_buffer *VB)
{
    r128ContextPtr r128ctx   = R128_CONTEXT(VB->ctx);
    int            new       = VB->pipeline->new_outputs;
    int            available = VB->pipeline->outputs;
    int            index     = 0;

    if (new & VERT_WIN) {
	new = available;
	index |= R128_WIN_BIT;
    }

    if (new & VERT_RGBA)      index |= R128_RGBA_BIT | R128_SPEC_BIT;
    if (new & VERT_TEX0_ANY)  index |= R128_TEX0_BIT;
    if (new & VERT_TEX1_ANY)  index |= R128_TEX1_BIT;

    r128ctx->SetupDone &= ~index;
    index              &= r128ctx->SetupIndex;
    r128ctx->SetupDone |= index;

    if (index) setup_func[index](VB, VB->Start, VB->Count);
}

/* Perform the raster setup for the fast path, if using CVA */
void r128DoRasterSetup(struct vertex_buffer *VB)
{
    GLcontext *ctx = VB->ctx;

    if (VB->Type == VB_CVA_PRECALC)   r128PartialRasterSetup(VB);
    else if (ctx->Driver.RasterSetup) ctx->Driver.RasterSetup(VB,
							      VB->CopyStart,
							      VB->Count);
}

/* Resize an existing vertex buffer */
void r128ResizeVB(struct vertex_buffer *VB, GLuint size)
{
    r128VertexBufferPtr r128vb = R128_DRIVER_DATA(VB);

    while (r128vb->size < size)
	r128vb->size *= 2;

    free(r128vb->vert_store);
    r128vb->vert_store = malloc(sizeof(r128Vertex) * r128vb->size + 31);
    if (!r128vb->vert_store) {
	fprintf(stderr, "Cannot allocate vertex store!  Exiting...\n");
	exit(1);
    }

    r128vb->verts = (r128VertexPtr)(((CARD32)r128vb->vert_store + 31) & ~31);

    gl_vector1ui_free(&r128vb->clipped_elements);
    gl_vector1ui_alloc(&r128vb->clipped_elements,
		       VEC_WRITABLE, r128vb->size, 32);
    if (!r128vb->clipped_elements.start) {
	fprintf(stderr, "Cannot allocate clipped elements!  Exiting...\n");
	exit(1);
    }

    free(VB->ClipMask);
    VB->ClipMask = (GLubyte *)malloc(sizeof(GLubyte) * r128vb->size);
    if (!VB->ClipMask) {
	fprintf(stderr, "Cannot allocate clipmask!  Exiting...\n");
	exit(1);
    }
}

/* Create a new device-dependent vertex buffer */
void r128DDRegisterVB(struct vertex_buffer *VB)
{
    r128VertexBufferPtr r128vb;

    r128vb = (r128VertexBufferPtr)calloc(1, sizeof(*r128vb));

    r128vb->size = VB->Size * 2;
    r128vb->vert_store = malloc(sizeof(r128Vertex) * r128vb->size + 31);
    if (!r128vb->vert_store) {
	fprintf(stderr, "Cannot allocate vertex store!  Exiting...\n");
	exit(1);
    }

    r128vb->verts = (r128VertexPtr)(((CARD32)r128vb->vert_store + 31) & ~31);

    gl_vector1ui_alloc(&r128vb->clipped_elements,
		       VEC_WRITABLE, r128vb->size, 32);
    if (!r128vb->clipped_elements.start) {
	fprintf(stderr, "Cannot allocate clipped elements!  Exiting...\n");
	exit(1);
    }

    free(VB->ClipMask);
    VB->ClipMask = (GLubyte *)malloc(sizeof(GLubyte) * r128vb->size);
    if (!VB->ClipMask) {
	fprintf(stderr, "Cannot allocate clipmask!  Exiting...\n");
	exit(1);
    }

    VB->driver_data = r128vb;
}

/* Destroy a device-dependent vertex buffer */
void r128DDUnregisterVB(struct vertex_buffer *VB)
{
    r128VertexBufferPtr r128vb = R128_DRIVER_DATA(VB);

    if (r128vb) {
	if (r128vb->vert_store)     free(r128vb->vert_store);
	gl_vector1ui_free(&r128vb->clipped_elements);
	free(r128vb);
	VB->driver_data = 0;
    }
}