#include "stdio.h"
#include "codeGenerator.h"

void generateH(Simulation& simulation, string fileName, string agentSufix, string modelSufix, string messageBoardSufix)
{
  int i, j;
  FILE* hFile;
  ModelList& modelList = simulation.getModelList();
  AgentList& agentList = simulation.getAgentList();

  string simulationName = simulation.getName();
  string straux;

  if((hFile = fopen((char*)fileName.c_str(), "w"))==NULL)
  {
	printf("Error opening file %s for writing\n", (char*)fileName.c_str());
	return;
  }

  straux = "\n#ifndef SIMULATION_" + simulationName + "\n";
  straux += "#define SIMULATION_" + simulationName + "\n\n";
  fputs((char*)straux.c_str(), hFile);

  straux = simulation.getHeaderCode() + "\n\n";
  fputs((char*)straux.c_str(), hFile);

  straux = "#include <string>\n#include <map>\n\n";
  straux += "using namespace std;\n\n";
  fputs((char*)straux.c_str(), hFile);

  straux = "class " + simulationName + messageBoardSufix + "\n{\npublic:\n";
  straux += simulationName + messageBoardSufix + "(){};\n\n";
  straux += "  void clear()\n  {\n    messageBoard_.clear();\n  };\n\n";
  straux += "  string getMessage(string index)\n  {\n    return messageBoard_[index];\n  };\n\n";
  straux += "  void postMessage(string index, string msg)\n  {\n    messageBoard_[index] = msg;\n  };\n\n";
  straux += "  operator=(" + simulationName + messageBoardSufix + "& msgBrdNew)\n  {\n    messageBoard_ = msgBrdNew.messageBoard_;\n  };\n\n";
  straux += "protected:\n  map<string, string> messageBoard_;\n};\n\n";
  fputs((char*)straux.c_str(), hFile);

  straux = "class " + simulationName + agentSufix +"\n{\npublic:\n";
  straux += simulationName + agentSufix + "()\n  {\n    messageBoard_ = NULL;\n  };\n\n";
  straux += "  void setMessageBoard(" + simulationName + messageBoardSufix + "* messageBoard)\n";
  straux += "  {\n    messageBoard_ = messageBoard;\n  };\n\n";
  straux += "  void setMessageBoardNew(" + simulationName + messageBoardSufix + "* messageBoardNew)\n";
  straux += "  {\n    messageBoardNew_ = messageBoardNew;\n  };\n\n";
  straux += "protected:\n  " + simulationName + messageBoardSufix + "* messageBoard_;\n";
  straux += "  " + simulationName + messageBoardSufix + "* messageBoardNew_;\n};\n\n";
  fputs((char*)straux.c_str(), hFile);

  string modelName;
  for(i=0; i<modelList.size(); i++)
  {
	modelName = modelList[i].getName();
	AttributeList& attrList = modelList[i].getAttributes();
	straux = "class " + modelName + modelSufix + ": public " + simulationName + agentSufix + "\n{\npublic:\n";
	straux += "  " + modelName + modelSufix + "();\n";
	straux += "  virtual void Tick()";
	if(modelList[i].getTickCode() == "") straux += " = 0";
	straux += ";\n\nprotected:\n";
	for(j=0; j<attrList.size(); j++)
	  straux += "  " + attrList[j].getType() + " " + attrList[j].getName() + ";\n";
	straux += "};\n\n";
	fputs((char*)straux.c_str(), hFile);
  }

  string agentName;
  string agentModel;
  for(i=0; i<agentList.size(); i++)
  {
	agentName = agentList[i].getName();
	agentModel = agentList[i].getModel();
	AttributeList& attrList = agentList[i].getAttributes();
	straux = "class " + agentName + agentSufix + ": public ";
	if(agentModel == "") straux += simulationName + agentSufix;
	else straux += agentModel + modelSufix;
	straux += "\n{\npublic:\n";
	straux += "  " + agentName + agentSufix + "();\n";
	straux += "  void Tick();\n";
	straux += "\nprotected:\n";
	for(j=0; j<attrList.size(); j++)
	{
	  if(attrList[j].getType() != "")
		straux += "  " + attrList[j].getType() + " " + attrList[j].getName() + ";\n";
	}
	straux += "};\n\n";
	fputs((char*)straux.c_str(), hFile);
  }

  straux = "class " + simulationName + "\n{\npublic:\n";
  straux += "  " + simulationName + "();\n\n";
  straux += "  void clear();\n  void Tick();\n";
  straux += "  " + simulationName + messageBoardSufix + "& getMessageBoard();\n\n";
  straux += "protected:\n";
  straux += "  " + simulationName + messageBoardSufix + " messageBoard_;\n";
  straux += "  " + simulationName + messageBoardSufix + " messageBoardNew_;\n\n";
  fputs((char*)straux.c_str(), hFile);

  for(i=0; i<agentList.size(); i++)
  {
	agentName = agentList[i].getName();

	straux = "  " + agentName + agentSufix + " " + agentName + ";\n";
	fputs((char*)straux.c_str(), hFile);
  }

  straux = "};\n\n";
  straux += "#endif\n";
  fputs((char*)straux.c_str(), hFile);

  fclose(hFile);
}

void generateCPP(Simulation& simulation, string fileName, string hFileName, string agentSufix, string modelSufix, string messageBoardSufix)
{
  int i, j;
  FILE* cppFile;
  ModelList& modelList = simulation.getModelList();
  AgentList& agentList = simulation.getAgentList();

  string simulationName = simulation.getName();
  string straux;

  if((cppFile = fopen((char*)fileName.c_str(), "w"))==NULL)
  {
	printf("Error opening file %s for writing\n", (char*)fileName.c_str());
	return;
  }

  straux = "#include \"" + hFileName + "\"\n\n";
  fputs((char*)straux.c_str(), cppFile);

  string modelName;
  for(i=0; i<modelList.size(); i++)
  {
	modelName = modelList[i].getName();
	AttributeList& attrList = modelList[i].getAttributes();
	straux = modelName + modelSufix + "::" + modelName + modelSufix + "()\n{\n";
	for(j=0; j<attrList.size(); j++)
	{
	  if(attrList[j].getInitVal() != "")
		straux += "  " + attrList[j].getName() + " = " + attrList[j].getInitVal() + ";\n";
	}
	straux += "}\n\n";
	fputs((char*)straux.c_str(), cppFile);

	straux = "void " + modelName + modelSufix + "::Tick()\n{\n";
	straux += modelList[i].getTickCode();
	straux += "\n}\n\n";
	fputs((char*)straux.c_str(), cppFile);
  }

  string agentName;
  string agentModel;
  for(i=0; i<agentList.size(); i++)
  {
	agentName = agentList[i].getName();
	agentModel = agentList[i].getModel();
	AttributeList& attrList = agentList[i].getAttributes();

	straux = agentName + agentSufix + "::" + agentName + agentSufix + "()\n{\n";
	for(j=0; j<attrList.size(); j++)
	{
	  if(attrList[j].getInitVal() != "")
		straux += "  " + attrList[j].getName() + " = " + attrList[j].getInitVal() + ";\n";
	}
	straux += "}\n\n";
	fputs((char*)straux.c_str(), cppFile);

	straux = "void " + agentName + agentSufix + "::Tick()\n{\n";
	if(agentModel != "")
	  straux += agentModel + modelSufix + "::Tick();\n\n";
	straux += agentList[i].getTickCode();
	straux += "\n}\n\n";
	fputs((char*)straux.c_str(), cppFile);
  }

  straux = simulationName + "::" + simulationName + "()\n{\n";
  fputs((char*)straux.c_str(), cppFile);

  for(i=0; i<agentList.size(); i++)
  {
	agentName = agentList[i].getName();
	straux = "  " + agentName + ".setMessageBoard(&messageBoard_);\n";
	straux += "  " + agentName + ".setMessageBoardNew(&messageBoardNew_);\n";
	fputs((char*)straux.c_str(), cppFile);
  }

  straux = "}\n\n";
  fputs((char*)straux.c_str(), cppFile);

  straux = "void " + simulationName + "::clear()\n{\n";
  straux += "  messageBoard_.clear();\n  messageBoardNew_.clear();\n}\n\n";
  fputs((char*)straux.c_str(), cppFile);

  straux = simulationName + messageBoardSufix + "& " + simulationName + "::getMessageBoard()\n{\n";
  straux += "  return messageBoard_;\n}\n\n";
  fputs((char*)straux.c_str(), cppFile);

  straux = "void " + simulationName + "::Tick()\n{\n";
  fputs((char*)straux.c_str(), cppFile);

  for(i=0; i<agentList.size(); i++)
  {
	agentName = agentList[i].getName();
	straux = "  " + agentName + ".Tick();\n";
	fputs((char*)straux.c_str(), cppFile);
  }

  straux = "  messageBoard_ = messageBoardNew_;\n}\n";
  fputs((char*)straux.c_str(), cppFile);

  fclose(cppFile);
}

void generateCode(Simulation& simulation)
{
  string simulationName = simulation.getName();
  if(simulationName == "") return;

  string agentSufix = "__AGENT__";
  string modelSufix = "__MODEL__";
  string messageBoardSufix = "__MESSAGEBOARD__";
  string hFileName = simulationName + ".h";
  string cppFileName = simulationName + ".cpp";

  generateH(simulation, hFileName, agentSufix, modelSufix, messageBoardSufix);
  generateCPP(simulation, cppFileName, hFileName, agentSufix, modelSufix, messageBoardSufix);
}