#include <stdio.h>
#include <stdint.h>
#include "nouveau_class.h"
#include <nouveau_pushbuf.h>

#include "screen.h"
#include "object.h"
#include "fifo.h"

static void tcl_fp_tx0(void)
{
	uint32_t *map;
	struct nouveau_bo *bo;
	struct nouveau_grobj *subch3D = grobj[NvSub3D];

	bo = screen_allocmem(0x1000);
	if (!bo) {
		return;
	}
	map = bo->map;

	printf("fragprog mapped at %p\n", map);

/*
#	0x18009e82 0x1c9dc901 0x0001c900 0x0001c900
#		INST 0: TXPR R1 (TR0.xyzw), attrib.texcoord[0], abs(texture[0])
#	0x02803e81 0x1c9dc904 0x0001c901 0x0001c900
#		INST 1: MULX R0 (TR0.xyzw), R1, attrib.color + END
*/
	map[0] = 0x18009e82;
	map[1] = 0x1c9dc901;
	map[2] = 0x0001c900;
	map[3] = 0x0001c900;

	map[4] = 0x02803e81;
	map[5] = 0x1c9dc904;
	map[6] = 0x0001c901;
	map[7] = 0x0001c900;

	nouveau_bo_unmap(bo);

	BEGIN_RING(chan, subch3D, NV34TCL_FP_ACTIVE_PROGRAM, 1);
	OUT_RELOCd(chan, bo, 0, NOUVEAU_BO_VRAM | NOUVEAU_BO_GART |
				NOUVEAU_BO_RD | NOUVEAU_BO_LOW,
				NV34TCL_FP_ACTIVE_PROGRAM_DMA0, 0);
	BEGIN_RING(chan, subch3D, NV34TCL_FP_CONTROL, 1);
	OUT_RING  (chan, 0);
	BEGIN_RING(chan, subch3D, NV34TCL_FP_REG_CONTROL, 1);
	OUT_RING  (chan, (1<<16)|4);

	nouveau_bo_ref(NULL, &bo);
}

void tcl_triangle_fixed(void)
{
	struct nouveau_grobj *subch3D = grobj[NvSub3D];

	printf("-- TCL triangle, fixed pipe\n");

	/*
		3 for pos
		4 for color0
	*/
	BEGIN_RING(chan, subch3D, NV34TCL_VTXFMT(0), 4);
	OUT_RING  (chan, (3<<NV34TCL_VTXFMT_SIZE_SHIFT)|NV34TCL_VTXFMT_TYPE_FLOAT);
        OUT_RING  (chan, NV34TCL_VTXFMT_TYPE_FLOAT);
        OUT_RING  (chan, NV34TCL_VTXFMT_TYPE_FLOAT);
        OUT_RING  (chan, (4<<NV34TCL_VTXFMT_SIZE_SHIFT)|NV34TCL_VTXFMT_TYPE_FLOAT);

	BEGIN_RING(chan, subch3D, NV34TCL_VERTEX_BEGIN_END, 1);
	OUT_RING  (chan, NV34TCL_VERTEX_BEGIN_END_TRIANGLES);

	BEGIN_RING(chan, subch3D, NV34TCL_VERTEX_DATA|METHOD_CONSTANT, 3*(4+3));
	OUT_RINGf (chan, 1.0); OUT_RINGf(chan, 1.0); OUT_RINGf(chan, 0.0); OUT_RINGf(chan, 0.0);
	OUT_RINGf (chan, 1.0); OUT_RINGf(chan, 0.0); OUT_RINGf(chan, 0.0);
	OUT_RINGf (chan, 1.0); OUT_RINGf(chan, 1.0); OUT_RINGf(chan, 1.0); OUT_RINGf(chan, 0.0);
	OUT_RINGf (chan, 0.0); OUT_RINGf(chan, 1.0); OUT_RINGf(chan, 0.0);
	OUT_RINGf (chan, 1.0); OUT_RINGf(chan, 0.0); OUT_RINGf(chan, 0.0); OUT_RINGf(chan, 1.0);
	OUT_RINGf (chan, 0.0); OUT_RINGf(chan, 0.0); OUT_RINGf(chan, 1.0);

	BEGIN_RING(chan, subch3D, NV34TCL_VERTEX_BEGIN_END, 1);
	OUT_RING  (chan, NV34TCL_VERTEX_BEGIN_END_STOP);

	FIRE_RING(chan);
}

void tcl_triangle_vtxattr(void)
{
	struct nouveau_grobj *subch3D = grobj[NvSub3D];

	printf("-- TCL triangle, vertex attribs\n");

	/* set color, then pos */
	BEGIN_RING(chan, subch3D, NV34TCL_VERTEX_BEGIN_END, 1);
	OUT_RING  (chan, NV34TCL_VERTEX_BEGIN_END_TRIANGLES);

	BEGIN_RING(chan, subch3D, NV34TCL_VTX_ATTR_3F_X(3), 3);
	OUT_RINGf (chan, 1.0);
	OUT_RINGf (chan, 0.0);
	OUT_RINGf( chan, 0.0);
	BEGIN_RING(chan, subch3D, NV34TCL_VTX_ATTR_3F_X(0), 3);
	OUT_RINGf (chan, 1.0);
	OUT_RINGf (chan, 0.0);
	OUT_RINGf (chan, 0.0);
	BEGIN_RING(chan, subch3D, NV34TCL_VTX_ATTR_3F_X(3), 3);
	OUT_RINGf (chan, 0.0);
	OUT_RINGf (chan, 1.0);
	OUT_RINGf (chan, 0.0);
	BEGIN_RING(chan, subch3D, NV34TCL_VTX_ATTR_3F_X(0), 3);
	OUT_RINGf (chan, 0.0);
	OUT_RINGf (chan, 1.0);
	OUT_RINGf (chan, 0.0);
	BEGIN_RING(chan, subch3D, NV34TCL_VTX_ATTR_3F_X(3), 3);
	OUT_RINGf (chan, 0.0);
	OUT_RINGf (chan, 0.0);
	OUT_RINGf (chan, 1.0);
	BEGIN_RING(chan, subch3D, NV34TCL_VTX_ATTR_3F_X(0), 3);
	OUT_RINGf (chan, 0.0);
	OUT_RINGf (chan, 0.0);
	OUT_RINGf (chan, 1.0);

	BEGIN_RING(chan, subch3D, NV34TCL_VERTEX_BEGIN_END, 1);
	OUT_RING  (chan, NV34TCL_VERTEX_BEGIN_END_STOP);

	FIRE_RING(chan);
}

void tcl_triangle_tx0(void)
{
	struct nouveau_grobj *subch3D = grobj[NvSub3D];

	printf("-- TCL triangle, textured\n");

	BEGIN_RING(chan, subch3D, NV34TCL_TX_UNITS_ENABLE, 1);
	OUT_RING  (chan, NV34TCL_TX_UNITS_ENABLE_TX0);

	tcl_fp_tx0();

	BEGIN_RING(chan, subch3D, NV34TCL_TX_OFFSET(0), 7);
	OUT_RING  (chan, screen_offset);
	OUT_RING  (chan, NV34TCL_TX_FORMAT_DMA0
		| NV34TCL_TX_FORMAT_NO_BORDER
		| NV34TCL_TX_FORMAT_DIMS_2D
		| NV34TCL_TX_FORMAT_FORMAT_A8R8G8B8
		| (8<<NV34TCL_TX_FORMAT_BASE_SIZE_U_SHIFT)
		| (8<<NV34TCL_TX_FORMAT_BASE_SIZE_V_SHIFT)
		| 0x10000
	);
	OUT_RING  (chan, NV34TCL_TX_WRAP_S_REPEAT
		| NV34TCL_TX_WRAP_T_REPEAT
		| NV34TCL_TX_WRAP_R_REPEAT
	);
	OUT_RING  (chan, NV34TCL_TX_ENABLE_ENABLE
		| 0x800
	);
	OUT_RING  (chan, (screen_pitch<<16)|0xaae4);	/* swizzle */
	OUT_RING  (chan, NV34TCL_TX_FILTER_MINIFY_NEAREST
		| NV34TCL_TX_FILTER_MAGNIFY_NEAREST
		| 0x2000
	);
	OUT_RING  (chan, (256<<16)|256);	/* rect size */

	BEGIN_RING(chan, subch3D, NV34TCL_TX_MATRIX_ENABLE(0), 1);
	OUT_RING  (chan, 1);
	BEGIN_RING(chan, subch3D, NV34TCL_TX0_MATRIX(0), 16);
	OUT_RINGf (chan, 1.0);
	OUT_RINGf (chan, 0.0);
	OUT_RINGf (chan, 0.0);
	OUT_RINGf (chan, 0.0);

	OUT_RINGf (chan, 0.0);
	OUT_RINGf (chan, 1.0);
	OUT_RINGf (chan, 0.0);
	OUT_RINGf (chan, 0.0);

	OUT_RINGf (chan, 0.0);
	OUT_RINGf (chan, 0.0);
	OUT_RINGf (chan, 1.0);
	OUT_RINGf (chan, 0.0);

	OUT_RINGf (chan, 0.0);
	OUT_RINGf (chan, 0.0);
	OUT_RINGf (chan, 0.0);
	OUT_RINGf (chan, 1.0);

	/* set color, then pos */
	BEGIN_RING(chan, subch3D, NV34TCL_VERTEX_BEGIN_END, 1);
	OUT_RING  (chan, NV34TCL_VERTEX_BEGIN_END_TRIANGLES);

	BEGIN_RING(chan, subch3D, NV34TCL_VTX_ATTR_2F_X(8), 2);
	OUT_RINGf (chan, 1.0);
	OUT_RINGf (chan, 0.0);
	BEGIN_RING(chan, subch3D, NV34TCL_VTX_ATTR_3F_X(0), 3);
	OUT_RINGf (chan, 1.0);
	OUT_RINGf (chan, 0.0);
	OUT_RINGf (chan, 0.0);
	BEGIN_RING(chan, subch3D, NV34TCL_VTX_ATTR_2F_X(8), 2);
	OUT_RINGf (chan, 0.0);
	OUT_RINGf (chan, 1.0);
	BEGIN_RING(chan, subch3D, NV34TCL_VTX_ATTR_3F_X(0), 3);
	OUT_RINGf (chan, 0.0);
	OUT_RINGf (chan, 1.0);
	OUT_RINGf (chan, 0.0);
	BEGIN_RING(chan, subch3D, NV34TCL_VTX_ATTR_2F_X(8), 2);
	OUT_RINGf (chan, 0.0);
	OUT_RINGf (chan, 0.0);
	BEGIN_RING(chan, subch3D, NV34TCL_VTX_ATTR_3F_X(0), 3);
	OUT_RINGf (chan, 0.0);
	OUT_RINGf (chan, 0.0);
	OUT_RINGf (chan, 1.0);

	BEGIN_RING(chan, subch3D, NV34TCL_VERTEX_BEGIN_END, 1);
	OUT_RING  (chan, NV34TCL_VERTEX_BEGIN_END_STOP);

	BEGIN_RING(chan, subch3D, NV34TCL_TX_UNITS_ENABLE, 1);
	OUT_RING  (chan, 0);
	BEGIN_RING(chan, subch3D, NV34TCL_TX_MATRIX_ENABLE(0), 1);
	OUT_RING  (chan, 0);
	BEGIN_RING(chan, subch3D, NV34TCL_TX_ENABLE(0), 1);
	OUT_RING  (chan, 0);

	FIRE_RING(chan);
}