/*  Copyright (C) 1996-1997  Id Software, Inc.

    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program; if not, write to the Free Software
    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA

    See file, 'COPYING', for details.
*/
static const char rcsid[] =
	"$Id$";

#ifdef HAVE_CONFIG_H
# include "config.h"
#endif
#ifdef HAVE_STRING_H
# include <string.h>
#endif
#ifdef HAVE_STRINGS_H
# include <strings.h>
#endif
#include <stdlib.h>

#include <QF/hash.h>

#include "qfcc.h"
#include "def.h"
#include "opcodes.h"
#include "options.h"
#include "type.h"

hashtab_t  *opcode_type_table_ab;
hashtab_t  *opcode_type_table_abc;

opcode_t   *op_done;
opcode_t   *op_return;
opcode_t   *op_if;
opcode_t   *op_ifnot;
opcode_t   *op_ifbe;
opcode_t   *op_ifb;
opcode_t   *op_ifae;
opcode_t   *op_ifa;
opcode_t   *op_state;
opcode_t   *op_goto;
opcode_t   *op_jump;
opcode_t   *op_jumpb;

statref_t *
PR_NewStatref (int ofs, int field)
{
	statref_t  *ref = calloc (1, sizeof (statref_t));

	ref->ofs = ofs;
	ref->field = field;
	return ref;
}

void
PR_AddStatementRef (def_t *def, dstatement_t *st, int field)
{
	if (def) {
		statref_t  *ref = PR_NewStatref (st - pr.statements, field);

		ref->next = def->refs;
		def->refs = ref;

		def->users--;
		def->used = 1;
	}
}

#define ROTL(x,n) (((x)<<(n))|(x)>>(32-n))

static unsigned long
get_hash (void *_op, void *_tab)
{
	opcode_t   *op = (opcode_t *) _op;
	hashtab_t **tab = (hashtab_t **) _tab;
	unsigned long hash;

	if (tab == &opcode_type_table_ab) {
		hash = ROTL (~op->type_a, 8) + ROTL (~op->type_b, 16);
	} else if (tab == &opcode_type_table_abc) {
		hash = ROTL (~op->type_a, 8) + ROTL (~op->type_b, 16)
			+ ROTL (~op->type_c, 24);
	} else {
		abort ();
	}
	return hash + Hash_String (op->name);;
}

static int
compare (void *_opa, void *_opb, void *_tab)
{
	opcode_t   *opa = (opcode_t *) _opa;
	opcode_t   *opb = (opcode_t *) _opb;
	hashtab_t **tab = (hashtab_t **) _tab;
	int         cmp;

	if (tab == &opcode_type_table_ab) {
		cmp = (opa->type_a == opb->type_a)
			  && (opa->type_b == opb->type_b);
	} else if (tab == &opcode_type_table_abc) {
		cmp = (opa->type_a == opb->type_a)
			  && (opa->type_b == opb->type_b)
			  && (opa->type_c == opb->type_c);
	} else {
		abort ();
	}
	return cmp && !strcmp (opa->name, opb->name);
}

opcode_t *
PR_Opcode_Find (const char *name, def_t *var_a, def_t *var_b, def_t *var_c)
{
	opcode_t    op;
	hashtab_t **tab;

	op.name = name;
	if (var_a && var_b && var_c) {
		op.type_a = var_a->type->type;
		op.type_b = var_b->type->type;
		op.type_c = var_c->type->type;
		tab = &opcode_type_table_abc;
	} else if (var_a && var_b) {
		op.type_a = var_a->type->type;
		op.type_b = var_b->type->type;
		tab = &opcode_type_table_ab;
	} else {
		tab = 0;
	}
	return Hash_FindElement (*tab, &op);
}

void
PR_Opcode_Init_Tables (void)
{
	opcode_t   *op;

	PR_Opcode_Init ();
	opcode_type_table_ab = Hash_NewTable (1021, 0, 0, &opcode_type_table_ab);
	opcode_type_table_abc = Hash_NewTable (1021, 0, 0, &opcode_type_table_abc);
	Hash_SetHashCompare (opcode_type_table_ab, get_hash, compare);
	Hash_SetHashCompare (opcode_type_table_abc, get_hash, compare);
	for (op = pr_opcodes; op->name; op++) {
		if (op->min_version > options.code.progsversion)
			continue;
		if (options.code.progsversion == PROG_ID_VERSION
			&& op->type_c == ev_integer)
			op->type_c = ev_float;
		Hash_AddElement (opcode_type_table_ab, op);
		Hash_AddElement (opcode_type_table_abc, op);
		if (!strcmp (op->name, "<DONE>")) {
			op_done = op;
		} else if (!strcmp (op->name, "<RETURN>")) {
			op_return = op;
		} else if (!strcmp (op->name, "<IF>")) {
			op_if = op;
		} else if (!strcmp (op->name, "<IFNOT>")) {
			op_ifnot = op;
		} else if (!strcmp (op->name, "<IFBE>")) {
			op_ifbe = op;
		} else if (!strcmp (op->name, "<IFB>")) {
			op_ifb = op;
		} else if (!strcmp (op->name, "<IFAE>")) {
			op_ifae = op;
		} else if (!strcmp (op->name, "<IFA>")) {
			op_ifa = op;
		} else if (!strcmp (op->name, "<STATE>")) {
			op_state = op;
		} else if (!strcmp (op->name, "<GOTO>")) {
			op_goto = op;
		} else if (!strcmp (op->name, "<JUMP>")) {
			op_jump = op;
		} else if (!strcmp (op->name, "<JUMPB>")) {
			op_jumpb = op;
		}
	}
}