/*
 * Copyright 2009 Nicolai Hähnle <nhaehnle@gmail.com>
 *
 * 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
 * THE AUTHOR(S) 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. */

#include "radeon_program.h"

#include <stdio.h>

static const char * textarget_to_string(rc_texture_target target)
{
	switch(target) {
	case RC_TEXTURE_2D_ARRAY: return "2D_ARRAY";
	case RC_TEXTURE_1D_ARRAY: return "1D_ARRAY";
	case RC_TEXTURE_CUBE: return "CUBE";
	case RC_TEXTURE_3D: return "3D";
	case RC_TEXTURE_RECT: return "RECT";
	case RC_TEXTURE_2D: return "2D";
	case RC_TEXTURE_1D: return "1D";
	default: return "BAD_TEXTURE_TARGET";
	}
}

static void rc_print_comparefunc(FILE * f, const char * lhs, rc_compare_func func, const char * rhs)
{
	if (func == RC_COMPARE_FUNC_NEVER) {
		fprintf(f, "false");
	} else if (func == RC_COMPARE_FUNC_ALWAYS) {
		fprintf(f, "true");
	} else {
		const char * op;
		switch(func) {
		case RC_COMPARE_FUNC_LESS: op = "<"; break;
		case RC_COMPARE_FUNC_EQUAL: op = "=="; break;
		case RC_COMPARE_FUNC_LEQUAL: op = "<="; break;
		case RC_COMPARE_FUNC_GREATER: op = ">"; break;
		case RC_COMPARE_FUNC_NOTEQUAL: op = "!="; break;
		case RC_COMPARE_FUNC_GEQUAL: op = ">="; break;
		default: op = "???"; break;
		}
		fprintf(f, "%s %s %s", lhs, op, rhs);
	}
}

static void rc_print_register(FILE * f, rc_register_file file, int index, unsigned int reladdr)
{
	if (file == RC_FILE_NONE) {
		fprintf(f, "none");
	} else if (file == RC_FILE_SPECIAL) {
		switch(index) {
		case RC_SPECIAL_ALU_RESULT: fprintf(f, "aluresult"); break;
		default: fprintf(f, "special[%i]", index); break;
		}
	} else {
		const char * filename;
		switch(file) {
		case RC_FILE_TEMPORARY: filename = "temp"; break;
		case RC_FILE_INPUT: filename = "input"; break;
		case RC_FILE_OUTPUT: filename = "output"; break;
		case RC_FILE_ADDRESS: filename = "addr"; break;
		case RC_FILE_CONSTANT: filename = "const"; break;
		default: filename = "BAD FILE"; break;
		}
		fprintf(f, "%s[%i%s]", filename, index, reladdr ? " + addr[0]" : "");
	}
}

static void rc_print_mask(FILE * f, unsigned int mask)
{
	if (mask & RC_MASK_X) fprintf(f, "x");
	if (mask & RC_MASK_Y) fprintf(f, "y");
	if (mask & RC_MASK_Z) fprintf(f, "z");
	if (mask & RC_MASK_W) fprintf(f, "w");
}

static void rc_print_dst_register(FILE * f, struct rc_dst_register dst)
{
	rc_print_register(f, dst.File, dst.Index, dst.RelAddr);
	if (dst.WriteMask != RC_MASK_XYZW) {
		fprintf(f, ".");
		rc_print_mask(f, dst.WriteMask);
	}
}

static char rc_swizzle_char(unsigned int swz)
{
	switch(swz) {
	case RC_SWIZZLE_X: return 'x';
	case RC_SWIZZLE_Y: return 'y';
	case RC_SWIZZLE_Z: return 'z';
	case RC_SWIZZLE_W: return 'w';
	case RC_SWIZZLE_ZERO: return '0';
	case RC_SWIZZLE_ONE: return '1';
	case RC_SWIZZLE_HALF: return 'H';
	case RC_SWIZZLE_UNUSED: return '_';
	}
	return '?';
}

static void rc_print_swizzle(FILE * f, unsigned int swizzle, unsigned int negate)
{
	unsigned int comp;
	for(comp = 0; comp < 4; ++comp) {
		rc_swizzle swz = GET_SWZ(swizzle, comp);
		if (GET_BIT(negate, comp))
			fprintf(f, "-");
		fprintf(f, "%c", rc_swizzle_char(swz));
	}
}

static void rc_print_src_register(FILE * f, struct rc_src_register src)
{
	int trivial_negate = (src.Negate == RC_MASK_NONE || src.Negate == RC_MASK_XYZW);

	if (src.Negate == RC_MASK_XYZW)
		fprintf(f, "-");
	if (src.Abs)
		fprintf(f, "|");

	rc_print_register(f, src.File, src.Index, src.RelAddr);

	if (src.Abs && !trivial_negate)
		fprintf(f, "|");

	if (src.Swizzle != RC_SWIZZLE_XYZW || !trivial_negate) {
		fprintf(f, ".");
		rc_print_swizzle(f, src.Swizzle, trivial_negate ? 0 : src.Negate);
	}

	if (src.Abs && trivial_negate)
		fprintf(f, "|");
}

static void rc_print_normal_instruction(FILE * f, struct rc_instruction * inst)
{
	const struct rc_opcode_info * opcode = rc_get_opcode_info(inst->U.I.Opcode);
	unsigned int reg;

	fprintf(f, "%s", opcode->Name);

	switch(inst->U.I.SaturateMode) {
	case RC_SATURATE_NONE: break;
	case RC_SATURATE_ZERO_ONE: fprintf(f, "_SAT"); break;
	case RC_SATURATE_MINUS_PLUS_ONE: fprintf(f, "_SAT2"); break;
	default: fprintf(f, "_BAD_SAT"); break;
	}

	if (opcode->HasDstReg) {
		fprintf(f, " ");
		rc_print_dst_register(f, inst->U.I.DstReg);
		if (opcode->NumSrcRegs)
			fprintf(f, ",");
	}

	for(reg = 0; reg < opcode->NumSrcRegs; ++reg) {
		if (reg > 0)
			fprintf(f, ",");
		fprintf(f, " ");
		rc_print_src_register(f, inst->U.I.SrcReg[reg]);
	}

	if (opcode->HasTexture) {
		fprintf(f, ", %s%s[%u]",
			textarget_to_string(inst->U.I.TexSrcTarget),
			inst->U.I.TexShadow ? "SHADOW" : "",
			inst->U.I.TexSrcUnit);
	}

	fprintf(f, ";");

	if (inst->U.I.WriteALUResult) {
		fprintf(f, " [aluresult = (");
		rc_print_comparefunc(f,
			(inst->U.I.WriteALUResult == RC_ALURESULT_X) ? "x" : "w",
			inst->U.I.ALUResultCompare, "0");
		fprintf(f, ")]");
	}

	fprintf(f, "\n");
}

static void rc_print_pair_instruction(FILE * f, struct rc_instruction * fullinst)
{
	struct rc_pair_instruction * inst = &fullinst->U.P;
	int printedsrc = 0;

	for(unsigned int src = 0; src < 3; ++src) {
		if (inst->RGB.Src[src].Used) {
			if (printedsrc)
				fprintf(f, ", ");
			fprintf(f, "src%i.xyz = ", src);
			rc_print_register(f, inst->RGB.Src[src].File, inst->RGB.Src[src].Index, 0);
			printedsrc = 1;
		}
		if (inst->Alpha.Src[src].Used) {
			if (printedsrc)
				fprintf(f, ", ");
			fprintf(f, "src%i.w = ", src);
			rc_print_register(f, inst->Alpha.Src[src].File, inst->Alpha.Src[src].Index, 0);
			printedsrc = 1;
		}
	}
	fprintf(f, "\n");

	if (inst->RGB.Opcode != RC_OPCODE_NOP) {
		const struct rc_opcode_info * opcode = rc_get_opcode_info(inst->RGB.Opcode);

		fprintf(f, "     %s%s", opcode->Name, inst->RGB.Saturate ? "_SAT" : "");
		if (inst->RGB.WriteMask)
			fprintf(f, " temp[%i].%s%s%s", inst->RGB.DestIndex,
				(inst->RGB.WriteMask & 1) ? "x" : "",
				(inst->RGB.WriteMask & 2) ? "y" : "",
				(inst->RGB.WriteMask & 4) ? "z" : "");
		if (inst->RGB.OutputWriteMask)
			fprintf(f, " color.%s%s%s",
				(inst->RGB.OutputWriteMask & 1) ? "x" : "",
				(inst->RGB.OutputWriteMask & 2) ? "y" : "",
				(inst->RGB.OutputWriteMask & 4) ? "z" : "");
		if (inst->WriteALUResult == RC_ALURESULT_X)
			fprintf(f, " aluresult");

		for(unsigned int arg = 0; arg < opcode->NumSrcRegs; ++arg) {
			const char* abs = inst->RGB.Arg[arg].Abs ? "|" : "";
			const char* neg = inst->RGB.Arg[arg].Negate ? "-" : "";
			fprintf(f, ", %s%ssrc%i.%c%c%c%s", neg, abs, inst->RGB.Arg[arg].Source,
				rc_swizzle_char(GET_SWZ(inst->RGB.Arg[arg].Swizzle, 0)),
				rc_swizzle_char(GET_SWZ(inst->RGB.Arg[arg].Swizzle, 1)),
				rc_swizzle_char(GET_SWZ(inst->RGB.Arg[arg].Swizzle, 2)),
				abs);
		}
		fprintf(f, "\n");
	}

	if (inst->Alpha.Opcode != RC_OPCODE_NOP) {
		const struct rc_opcode_info * opcode = rc_get_opcode_info(inst->Alpha.Opcode);

		fprintf(f, "     %s%s", opcode->Name, inst->Alpha.Saturate ? "_SAT" : "");
		if (inst->Alpha.WriteMask)
			fprintf(f, " temp[%i].w", inst->Alpha.DestIndex);
		if (inst->Alpha.OutputWriteMask)
			fprintf(f, " color.w");
		if (inst->Alpha.DepthWriteMask)
			fprintf(f, " depth.w");
		if (inst->WriteALUResult == RC_ALURESULT_W)
			fprintf(f, " aluresult");

		for(unsigned int arg = 0; arg < opcode->NumSrcRegs; ++arg) {
			const char* abs = inst->Alpha.Arg[arg].Abs ? "|" : "";
			const char* neg = inst->Alpha.Arg[arg].Negate ? "-" : "";
			fprintf(f, ", %s%ssrc%i.%c%s", neg, abs, inst->Alpha.Arg[arg].Source,
				rc_swizzle_char(inst->Alpha.Arg[arg].Swizzle), abs);
		}
		fprintf(f, "\n");
	}

	if (inst->WriteALUResult) {
		fprintf(f, "      [aluresult = (");
		rc_print_comparefunc(f, "result", inst->ALUResultCompare, "0");
		fprintf(f, ")]\n");
	}
}

/**
 * Print program to stderr, default options.
 */
void rc_print_program(const struct rc_program *prog)
{
	unsigned int linenum = 0;
	struct rc_instruction *inst;

	fprintf(stderr, "# Radeon Compiler Program\n");

	for(inst = prog->Instructions.Next; inst != &prog->Instructions; inst = inst->Next) {
		fprintf(stderr, "%3d: ", linenum);

		if (inst->Type == RC_INSTRUCTION_PAIR)
			rc_print_pair_instruction(stderr, inst);
		else
			rc_print_normal_instruction(stderr, inst);

		linenum++;
	}
}