/*
**	Yab2Cpp
**
**	Transpiler by Samuel D. Crow
**
**	Based on Yab
**
*/
#include "yab2cpp.h"

class label;

/* function definitions */
shared_ptr<fn>fn::getCurrentSub()
{
	return callStack.back();
}

void fn::generateOnNSub(shared_ptr<expression>e, unsigned int skip)
{
	shared_ptr<label>r=shared_ptr<label>(new label());
	shared_ptr<fn> self=shared_ptr<fn>(new fn(r));
	fn::callStack.push_back(self);
	label::generateOnNTo(e, skip);
	r->generate();
}

void fn::generateGosub(shared_ptr<label> sub)
{
	shared_ptr<label>r=shared_ptr<label>(new label());
	shared_ptr<fn> self=shared_ptr<fn>(new fn(r));
	fn::callStack.push_back(self);
	sub->generateJumpTo();
	r->generate();
}

fn::fn(shared_ptr<label>gosub):codeType(T_GOSUB)
{
	this->ret=gosub;
}

shared_ptr<fn> fn::getSub(string &name)
{
	auto iter=fn::functions.find(name);
	if(iter==fn::functions.end()) return NULL;
	return iter->second;
}

shared_ptr<operands>fn::generateCall(string &name,
	list<shared_ptr<operands> >paramList)
{
	auto v=params.begin();
	shared_ptr<operands>current;
	shared_ptr<fn>g=fn::getSub(name);
	if (g==NULL)
	{
		error(E_SUBROUTINE_NOT_FOUND);
	}
	if (paramList.size()>params.size())
	{
		error(E_TOO_MANY_PARAMETERS);
	}
	/* TODO CHECK THIS */
	output_cpp << "struct f" << g->getID()
		<< "*sub" << this->getID()
		<< "= new struct f" << g->getID()
		<< "();\n";

	/* TODO Make parameter processing a separate function */
	while(paramList.size()>0)
	{
		current= paramList.front();
		paramList.pop_front();
		if(current->getSimpleVarType()!=(*v)->getType())
		{
			error(E_TYPE_MISMATCH);
		}
		(*v)->assignment(shared_ptr<expression>(
			new expression(current.get())));
		++v;
	}
	/* pad remaining unassigned variables with empty values */
	while (v!=params.end())
	{
		switch ((*v)->getType())
		{
			case T_FLOATVAR:
				(*v)->assignment(shared_ptr<expression>(
					new expression(
						new constOp("0.0", T_FLOAT)
					)));
				break;
			case T_INTVAR:
				(*v)->assignment(shared_ptr<expression>(
					new expression(
						new constOp("0", T_INT)
					)));
				break;
			case T_STRINGVAR:
				(*v)->assignment(shared_ptr<expression>(
					new expression(
						new constOp("", T_STRING)
					)));
			default:
				error(E_INTERNAL);
		}
		++v;
	}
	g->startAddr->generateJumpTo();
	g->ret->generate();
	return g->rc;
}

void fn::generateReturn()
{
	shared_ptr<fn>c=getCurrentSub();
	switch(c->getType())
	{
		case T_UNKNOWNFUNC:
			/* set return type to NONE */
			this->kind=T_NONE;
			/*fallthrough*/
		case T_GOSUB:
			c->ret->generateJumpTo();
			fn::callStack.pop_back();
			break;
		default:
			error(E_TYPE_MISMATCH);
	}
}

void fn::generateReturn(shared_ptr<expression>expr)
{
	this->rc=expr->evaluate();
	this->kind=rc->getSimpleVarType();
	this->ret->generateJumpTo();
	fn::callStack.pop_back();
	switch (this->getType())
	{
		case T_UNKNOWNFUNC:
			return;
		case T_STRINGFUNC:
			if (kind!=T_STRINGVAR) error(E_TYPE_MISMATCH);
			return;
		case T_INTFUNC:
			if (kind!=T_INTVAR&&kind!=T_FLOATVAR) error(E_TYPE_MISMATCH);
			return;
		case T_FLOATFUNC:
			if(kind!=T_FLOATVAR) error(E_TYPE_MISMATCH);
			return;
		case T_GOSUB:
			error(E_GOSUB_CANNOT_RETURN_VALUE);
		default:
			error(E_BAD_SYNTAX);
	}
}

/* not allowed in function definitions directly */
void fn::generateBreak()
{
	error(E_BAD_SYNTAX);
}

void fn::close()
{
	/* check if no returns and no return type */
	enum CODES t=this->getType();
	if (this->kind==T_UNKNOWN&&
		(t==T_UNKNOWNFUNC||t==T_VOIDFUNC))
	{
		/* generate a typeless return */
		this->generateReturn();
	}
	funcs_h << "};\n";
	fn::locals.clear();
	fn::statics.clear();
	this->params.clear();
	scopeGlobal=true;
}

fn::fn(string &s, enum CODES t, shared_ptr<operands>returnCode):codeType(t)
{
	/*check for nesting error */
	if (!scopeGlobal) error(E_END_FUNCTION);
	/*check if this function name is already used*/
	if (fn::functions.find(s)!=fn::functions.end()) error(E_DUPLICATE_SYMBOL);
	this->id= ++nextID;
	/*define storage for locals*/
	funcs_h << "struct f" << this->id <<"\n{\n";
	/*define label space for return*/
	this->ret=shared_ptr<label>(new label());
	/*allocate function name*/
	fn::functions[s]=shared_ptr<fn>(this);
	/* initiate local scope */
	scopeGlobal=false;
	/*keep track of where the return code will be sent to*/
	this->rc=returnCode;
	/*allocate and generate start address label*/
	this->startAddr=shared_ptr<label>(new label());
	startAddr->generate();
}