/**********************************************************/
/*  apimagic: cparser-based API normalization utility     */
/*  Copyright (C) 2015--2016  Z. Gilboa                   */
/*  Released under GPLv2 and GPLv3; see COPYING.APIMAGIC. */
/**********************************************************/

#include <stdio.h>

#include <cparser/ast/ast_t.h>
#include <cparser/ast/entity_t.h>
#include <cparser/ast/symbol_t.h>

#include <apimagic/apimagic.h>
#include "apimagic_driver_impl.h"

static int output_enum(
	const char *			symbol,
	const struct amgc_entity	enumvals[],
	const struct amgc_layout *	layout,
	FILE *				fout)
{
	const struct amgc_entity *	enumval;
	struct amgc_layout		elayout;
	size_t				len;

	if (!layout || !layout->symwidth) {
		if (!layout)
			memset(&elayout,0,sizeof(elayout));
		else
			memcpy(&elayout,layout,sizeof(elayout));

		if (!elayout.tabwidth)
			elayout.tabwidth = AMGC_TAB_WIDTH;

		layout  = &elayout;
		enumval = enumvals;

		for (; enumval->entity || enumval->altname; enumval++) {
			len = strlen(enumval->altname
				? enumval->altname
				: enumval->entity->base.symbol->string);

			if (len > (unsigned)elayout.symwidth)
				elayout.symwidth = len;
		}
	}

	if (layout->header && (fputs(layout->header,fout) < 0))
		return -1;

	if (fprintf(fout,"enum %s {\n",symbol) < 0)
		return -1;

	for (enumval=enumvals; enumval->entity || enumval->altname; enumval++) {
		symbol = enumval->altname
			? enumval->altname
			: enumval->entity->base.symbol->string;

		if (fprintf(fout,"\t%s",symbol) < 0)
			return -1;

		if (amgc_output_pad_symbol(symbol,layout,fout) < 0)
			return -1;

		if ((enumval->enumval < 0) && (enumval->enumval > -128))
			fprintf(fout,"= (%d)",enumval->enumval);
		else if ((enumval->enumval >= 0) && (enumval->enumval < 2048))
			fprintf(fout,"= %d",enumval->enumval);
		else
			fprintf(fout,"= 0x%08x",(unsigned)enumval->enumval);

		if (fputs(",\n",fout) < 0)
			return -1;
	}

	if (fputs("};\n",fout) < 0)
		return -1;

	if (layout->footer && (fputs(layout->footer,fout) < 0))
		return -1;

	return 0;
}

int amgc_output_unit_enum(
	const struct amgc_unit_ctx *	uctx,
	const union entity_t *		entity,
	const struct amgc_layout *	layout,
	FILE *				fout)
{
	struct amgc_entity *	enumvals;
	const char *		symbol;
	int			ret;

	if (entity->base.kind != ENTITY_ENUM)
		return -1;
	else if (amgc_get_enum_members(uctx,entity,&enumvals))
		return -1;

	if (entity->base.symbol)
		symbol = entity->base.symbol->string;
	else
		symbol = "";

	ret = output_enum(symbol,enumvals,layout,fout);
	amgc_free_enum_members(enumvals);

	return ret;
}

int amgc_output_custom_enum(
	const struct amgc_entity *	penum,
	const struct amgc_entity	enumvals[],
	const struct amgc_layout *	layout,
	FILE *				fout)
{
	const struct amgc_entity *	aentity;
	const char *			symbol;

	if (penum->entity && penum->entity->base.kind != ENTITY_ENUM)
		return -1;
	else if (!penum->entity && !penum->altname)
		return -1;

	for (aentity=enumvals; aentity->entity || aentity->altname; aentity++)
		if (aentity->entity && aentity->entity->base.kind != ENTITY_ENUM_VALUE)
			return -1;

	if (!penum->entity)
		symbol = penum->altname;
	else if (penum->entity->base.symbol)
		symbol = penum->entity->base.symbol->string;
	else
		symbol = "";

	return output_enum(symbol,enumvals,layout,fout);
}

int  amgc_output_unit_enums(
	const struct amgc_unit_ctx *	uctx,
	const struct amgc_layout *	layout,
	FILE *				fout)
{
	const struct amgc_entity * aentity;

	for (aentity=uctx->entities->enums; aentity->entity; aentity++)
		if (amgc_output_unit_enum(uctx,aentity->entity,layout,fout))
			return -1;

	return 0;
}

int  amgc_list_unit_enums(
	const struct amgc_unit_ctx *	uctx,
	const struct amgc_layout *	layout,
	FILE *				fout)
{
	const struct amgc_entity * aentity;

	if (layout && layout->header && (fputs(layout->header,fout) < 0))
		return -1;

	for (aentity=uctx->entities->enums; aentity->entity; aentity++)
		if ((fprintf(fout,"enum %s;\n",
				aentity->entity && aentity->entity->base.symbol
				? aentity->entity->base.symbol->string
				: aentity->altname) < 0))
			return -1;

	if (layout && layout->footer && (fputs(layout->footer,fout) < 0))
		return -1;

	return 0;
}