Lot's of work on a CGenerator, AST and language improvements

2013-11-01 02:52:18 -04:00
parent ededb069c1
commit 77f2b0a3e5
14 changed files with 232 additions and 28 deletions
--- a/src/CGenerator.cpp
+++ b/src/CGenerator.cpp
@@ -0,0 +1,125 @@
+#include "CGenerator.h"
+
+CGenerator::CGenerator() {
+	tabLevel = 0;
+}
+CGenerator::~CGenerator() {
+
+}
+
+std::string CGenerator::tabs() {
+	std::string returnTabs;
+	for (int i = 0; i < tabLevel; i++)
+		returnTabs += "\t";
+	return returnTabs;
+}
+
+std::string CGenerator::generate(NodeTree<ASTData>* from) {
+	ASTData data = from->getData();
+	std::vector<NodeTree<ASTData>*> children = from->getChildren();
+		std::string output = "";
+	switch (data.type) {
+		case translation_unit:
+			//Do nothing
+			break;
+		case interpreter_directive:
+			//Do nothing
+			break;
+		case import:
+			return "#include \"" + data.symbol.getName() + "\"\n";
+			break;
+		case identifier:
+			return data.symbol.getName();
+			break;
+		case function:
+			output += "\n" + ValueTypeToCType(data.valueType) + " " + data.symbol.getName() + "(";
+			for (int i = 0; i < children.size()-1; i++) {
+				if (i > 0)
+					output += ", ";
+				output += ASTData::ValueTypeToString(children[i]->getData().valueType) + " " + generate(children[i]);
+			}
+			output+= ")\n" + generate(children[children.size()-1]);
+			return output;
+			break;
+		case code_block:
+			output += tabs() + "{\n";
+			tabLevel++;
+			for (int i = 0; i < children.size(); i++)
+				output += generate(children[i]);
+			tabLevel--;
+			output += tabs() + "}";
+			return output;
+			break;
+		case expression:
+			output += " " + data.symbol.getName() + ", ";
+			break;
+		case boolean_expression:
+			output += " " + data.symbol.getName() + " ";
+			break;
+		case statement:
+			return tabs() + generate(children[0]) + ";\n";
+			break;
+		case if_statement:
+			output += "if (" + generate(children[0]) + ") \n" + generate(children[1]);
+			if (children.size() > 2)
+				output += " else " + generate(children[2]);
+			return output;
+			break;
+		case return_statement:
+			return "return " + generate(children[0]);
+		case assignment_statement:
+			return generate(children[0]) + " = " + generate(children[1]);
+		case declaration_statement:
+			return ASTData::ValueTypeToString(children[0]->getData().valueType) + " " + generate(children[0]) + " = " + generate(children[1]);
+		case function_call:
+		{
+			//Handle operators specially for now. Will later replace with
+			//Inlined functions in the standard library
+			std::string name = data.symbol.getName();
+			if (name == "+" || name == "-" || name == "*" || name == "/") {
+				return "((" + generate(children[0]) + ")" + name + "(" + generate(children[1]) + "))";
+			}
+			output += data.symbol.getName() + "(";
+			for (int i = 0; i < children.size(); i++)
+				if (i < children.size()-1)
+					output += generate(children[i]) + ", ";
+				else output += generate(children[i]);
+			output += ") ";
+			return output;
+		}
+		case value:
+			return data.symbol.getName();
+
+		default:
+			std::cout << "Nothing!" << std::endl;
+	}
+	for (int i = 0; i < children.size(); i++)
+		output += generate(children[i]);
+
+	return output;
+}
+
+std::string CGenerator::ValueTypeToCType(ValueType type) {
+	switch (type) {
+		case none:
+			return "none";
+			break;
+		case boolean:
+			return "bool";
+			break;
+		case integer:
+			return "int";
+			break;
+		case floating:
+			return "float";
+			break;
+		case double_percision:
+			return "double";
+			break;
+		case char_string:
+			return "char*";
+			break;
+		default:
+			return "unknown_ValueType";
+	}
+}