/*
 *	aegis - project change supervisor
 *	Copyright (C) 1991-1994, 1997-1999, 2001 Peter Miller;
 *	All rights reserved.
 *
 *	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, USA.
 *
 * MANIFEST: enumerated types functions generation
 */

#include <indent.h>
#include <mem.h>
#include <parse.h>
#include <type/enumeration.h>


typedef struct type_enum_ty type_enum_ty;
struct type_enum_ty
{
	/* inherited */
	TYPE_TY

	/* instance variables */
	size_t		nelements;
	string_ty	**element;
};


static void constructor _((type_ty *));

static void
constructor(type)
	type_ty		*type;
{
	type_enum_ty	*this;

	this = (type_enum_ty *)type;
	this->nelements = 0;
	this->element = 0;
}


static void destructor _((type_ty *));

static void
destructor(type)
	type_ty		*type;
{
	type_enum_ty	*this;
	size_t		j;

	this = (type_enum_ty *)type;
	for (j = 0; j < this->nelements; ++j)
		str_free(this->element[j]);
	if (this->element)
		mem_free(this->element);
}


static void gen_include _((type_ty *));

static void
gen_include(type)
	type_ty		*type;
{
	type_enum_ty	*this;
	size_t		j;

	this = (type_enum_ty *)type;
	indent_putchar('\n');
	indent_printf("#ifndef %s_DEF\n", this->name->str_text);
	indent_printf("#define %s_DEF\n", this->name->str_text);
	indent_printf("enum %s_ty\n", this->name->str_text);
	indent_printf("{\n"/*}*/);
	for (j = 0; j < this->nelements; ++j)
	{
		indent_printf
		(
			"%s_%s",
			this->name->str_text,
			this->element[j]->str_text
		);
		if (j < this->nelements - 1)
			indent_putchar(',');
		indent_putchar('\n');
	}
	indent_printf(/*{*/"};\n");
	indent_printf("#ifdef CONF_enum_is_int\n");
	indent_printf
	(
		"typedef enum %s_ty %s_ty;\n",
		this->name->str_text,
		this->name->str_text
	);
	indent_printf("#else\n");
	indent_printf("typedef int %s_ty;\n", this->name->str_text);
	indent_printf("#endif\n");
	indent_printf("#endif /* %s_DEF */\n", this->name->str_text);

	indent_putchar('\n');
	indent_printf("extern type_ty %s_type;\n", this->name->str_text);

	indent_putchar('\n');
	indent_printf
	(
		"void %s_write _((struct output_ty *, const char *, %s_ty));\n",
		this->name->str_text,
		this->name->str_text
	);
	indent_printf
	(
		"char *%s_ename _((%s_ty));\n",
		this->name->str_text,
		this->name->str_text
	);
}


static void gen_include_declarator _((type_ty *, string_ty *, int));

static void
gen_include_declarator(type, variable_name, is_a_list)
	type_ty		*type;
	string_ty	*variable_name;
	int		is_a_list;
{
	indent_printf
	(
		"%s_ty\1%s%s;\n",
		type->name->str_text,
		(is_a_list ? "*" : ""),
		variable_name->str_text
	);
}


static void gen_code _((type_ty *));

static void
gen_code(type)
	type_ty		*type;
{
	type_enum_ty	*this;
	size_t		j;
	string_ty	*s;

	this = (type_enum_ty *)type;
	indent_putchar('\n');
	indent_printf("static char *%s_s[] =\n", this->name->str_text);
	indent_printf("{\n"/*}*/);
	for (j = 0; j < this->nelements; ++j)
		indent_printf("\"%s\",\n", this->element[j]->str_text);
	indent_printf(/*{*/"};\n");
	indent_printf
	(
		"%s\1*%s_f[SIZEOF(%s_s)];\n",
		"static string_ty",
		this->name->str_text,
		this->name->str_text
	);

	indent_putchar('\n');
	indent_printf("char *\n");
	indent_printf("%s_ename(this)\n", this->name->str_text);
	indent_more();
	indent_printf("%s_ty\1this;\n", this->name->str_text);
	indent_less();
	indent_printf("{\n"/*}*/);
	indent_printf("static char\1buffer[20];\n\n");
	indent_printf("if ((int)this >= 0 && (int)this < %d)\n", this->nelements);
	indent_more();
	indent_printf("return %s_s[this];\n", this->name->str_text);
	indent_less();
	indent_printf("sprintf(buffer, \"%%d\", (int)this);\n");
	indent_printf("return buffer;\n");
	indent_printf(/*{*/"}\n");

	indent_putchar('\n');
	indent_printf("void\n");
	indent_printf("%s_write(fp, name, this)\n", this->name->str_text);
	indent_more();
	indent_printf("%s\1*fp;\n", "output_ty");
	indent_printf("%s\1*name;\n", "const char");
	indent_printf("%s_ty\1this;\n", this->name->str_text);
	indent_less();
	indent_printf("{\n"/*}*/);
	indent_printf("if (name)\n");
	indent_printf("{\n"/*}*/);
	indent_printf("if (this == 0 && type_enum_option_query())\n");
	indent_more();
	indent_printf("return;\n");
	indent_less();
	indent_printf("output_fputs(fp, name);\n");
	indent_printf("output_fputs(fp, \" = \");\n");
	indent_printf(/*{*/"}\n");
	indent_printf
	(
		"output_fputs(fp, %s_s[this]);\n",
		this->name->str_text
	);
	indent_printf("if (name)\n");
	indent_more();
	indent_printf("output_fputs(fp, \";\\n\");\n");
	indent_less();
	indent_printf(/*{*/"}\n");

	indent_putchar('\n');
	indent_printf
	(
		"static int %s_parse _((string_ty *));\n",
		this->name->str_text
	);

	indent_putchar('\n');
	indent_printf("static int\n");
	indent_printf("%s_parse(name)\n", this->name->str_text);
	indent_more();
	indent_printf("%s\1*name;\n", "string_ty");
	indent_less();
	indent_printf("{\n"/*}*/);
	indent_printf("%s\1j;\n", "int");
	indent_putchar('\n');
	indent_printf
	(
		"slow_to_fast(%s_s, %s_f, SIZEOF(%s_s));\n",
		this->name->str_text,
		this->name->str_text,
		this->name->str_text
	);
	indent_printf
	(
		"for (j = 0; j < SIZEOF(%s_f); ++j)\n",
		this->name->str_text
	);
	indent_printf("{\n");
	indent_printf("if (str_equal(name, %s_f[j]))\n", this->name->str_text);
	indent_more();
	indent_printf("return j;\n");
	indent_less();
	indent_printf("}\n");
	indent_printf("return -1;\n");
	indent_printf(/*{*/"}\n");

	indent_putchar('\n');
	indent_printf
	(
		"static string_ty *%s_fuzzy _((string_ty *));\n",
		this->name->str_text
	);

	indent_putchar('\n');
	indent_printf("static string_ty *\n");
	indent_printf("%s_fuzzy(name)\n", this->name->str_text);
	indent_more();
	indent_printf("%s\1*name;\n", "string_ty");
	indent_less();
	indent_printf("{\n"/*}*/);
	indent_printf("return\n");
	indent_more();
	indent_printf("generic_enum_fuzzy\n");
	indent_printf("(\n"/*)*/);
	indent_printf("name,\n");
	indent_printf("%s_f,\n", this->name->str_text);
	indent_printf("SIZEOF(%s_f)\n", this->name->str_text);
	indent_printf(/*(*/");\n");
	indent_less();
	indent_printf(/*{*/"}\n");

	s = str_format
	(
		"generic_enum__init(%s_s, SIZEOF(%s_s));",
		this->name->str_text,
		this->name->str_text
	);
	generate_code__init(s);
	str_free(s);

	indent_putchar('\n');
	indent_printf
	(
		"static struct rpt_value_ty *%s_convert _((void *));\n",
		this->name->str_text
	);

	indent_putchar('\n');
	indent_printf("static struct rpt_value_ty *\n");
	indent_printf("%s_convert(this)\n", this->name->str_text);
	indent_more();
	indent_printf("%s\1*this;\n", "void");
	indent_less();
	indent_printf("{\n"/*}*/);
	indent_printf("if (!%s_f[0])\n", this->name->str_text);
	indent_more();
	indent_printf
	(
		"slow_to_fast(%s_s, %s_f, SIZEOF(%s_s));\n",
		this->name->str_text,
		this->name->str_text,
		this->name->str_text
	);
	indent_less();
	indent_printf("return\n");
	indent_more();
	indent_printf("generic_enum_convert\n");
	indent_printf("(\n"/*)*/);
	indent_printf("this,\n");
	indent_printf("%s_f,\n", this->name->str_text);
	indent_printf("SIZEOF(%s_f)\n", this->name->str_text);
	indent_printf(/*(*/");\n");
	indent_less();
	indent_printf(/*{*/"}\n");

	indent_putchar('\n');
	indent_printf("type_ty %s_type =\n", this->name->str_text);
	indent_printf("{\n"/*}*/);
	indent_printf("\"%s\",\n", this->name->str_text);
	indent_printf("0, /* alloc */\n");
	indent_printf("0, /* free */\n");
	indent_printf("%s_parse,\n", this->name->str_text);
	indent_printf("0, /* list_parse */\n");
	indent_printf("0, /* struct_parse */\n");
	indent_printf("%s_fuzzy,\n", this->name->str_text);
	indent_printf("%s_convert,\n", this->name->str_text);
	indent_printf("generic_enum_is_set,\n");
	indent_printf(/*{*/"};\n");
}


static void gen_code_declarator _((type_ty *, string_ty *, int));

static void
gen_code_declarator(type, variable_name, is_a_list)
	type_ty		*type;
	string_ty	*variable_name;
	int		is_a_list;
{
	indent_printf("%s_write(fp, "/*)*/, type->name->str_text);
	if (is_a_list)
		indent_printf("(char *)0");
	else
		indent_printf("\"%s\"", variable_name->str_text);
	indent_printf(/*(*/", this->%s);\n", variable_name->str_text);
}


static void gen_free_declarator _((type_ty *, string_ty *, int));

static void
gen_free_declarator(type, variable_name, is_a_list)
	type_ty		*type;
	string_ty	*variable_name;
	int		is_a_list;
{
	if (is_a_list)
		indent_printf(";\n");
}


static void member_add _((type_ty *, string_ty *, type_ty *));

static void
member_add(type, member_name, member_type)
	type_ty		*type;
	string_ty	*member_name;
	type_ty		*member_type;
{
	type_enum_ty	*this;
	size_t		nbytes;

	this = (type_enum_ty *)type;
	nbytes = (this->nelements + 1) * sizeof(this->element[0]);
	this->element = mem_change_size(this->element, nbytes);
	this->element[this->nelements++] = str_copy(member_name);
}


type_method_ty type_enumeration =
{
	sizeof(type_enum_ty),
	"enumeration",
	1, /* has a mask */
	constructor,
	destructor,
	gen_include,
	gen_include_declarator,
	gen_code,
	gen_code_declarator,
	gen_free_declarator,
	member_add,
	0, /* in_include_file */
};