#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <ctype.h>
#include "translator.h"

void freeasmlns(struct asmln** lns, int count) {
	for(int i = 0; i < count; i++) {
		free(lns[i]->instr);
		free(lns[i]);
	}
	free(lns);
}

char* heapstr(const char* input) {
	char* newstr = (char*)malloc(sizeof(char)*(strlen(input)+1));
	strcpy(newstr, input);
	return newstr;
}

char* switchseg(struct line* ln) {
	char* seg = ln->tokens[1];
	if(strcmp(seg, "local") == 0)
		return heapstr("@LCL");
	if(strcmp(seg, "argument") == 0)
		return heapstr("@ARG");
	if(strcmp(seg, "this") == 0)
		return heapstr("@THIS");
	if(strcmp(seg, "that") == 0)
		return heapstr("@THAT");
	fprintf(stderr, "Unrecognized segment '%s'; line %i\n", seg, ln->truen);
	exit(1);
}

// produce comment as follows:
// pop/push segment i
char* mkcom(struct line* ln, int indlen) {
	int comlen = sizeof(char) * (strlen(ln->tokens[0]) + strlen(ln->tokens[1]) + indlen + 6);
	char* comment = (char*)malloc(comlen);
	snprintf(comment, comlen, "// %s %s %s", ln->tokens[0], ln->tokens[1], ln->tokens[2]);
	return comment;
}

void checkind(struct line* ln, int indsz) {
	for(int i = 0; i < indsz; i++) {
		if(!isdigit(ln->tokens[2][i])) {
			fprintf(stderr, "Invalid index '%s'; line %i\n", ln->tokens[2], ln->truen);
			exit(1);
		}
	}
}

char* mkind(char* ind, int indsz) {
	int newsz = sizeof(char) * (indsz + 2);
	char* newind = (char*)malloc(newsz);
	snprintf(newind, newsz, "@%s", ind);
	return newind;
}

char* mkstatind(char* ind, int indsz, char* fname) {
	int fnamelen = strlen(fname);
	int newsz = sizeof(char) * (fnamelen + indsz + 3);
	char* newind = (char*)malloc(newsz);
	snprintf(newind, newsz, "@%s.%s", fname, ind);
	return newind;
}

char* mktempind(char* ind, int indsz) {
	int intind = atoi(ind);
	int newsz = sizeof(char) * (indsz + 3);
	char* newind = (char*)malloc(newsz);
	snprintf(newind, newsz, "@%i", intind+5);
	return newind;
}

void checkasmsize(int* size, struct asmln*** lns, int targ) {
	if(targ >= (*size)) {
		(*size)=targ*2;
		(*lns) = (struct asmln**)realloc((*lns), (*size));
	}
}

struct asmln* mkasmln(struct line* ln, char* content) {
	struct asmln* newln = (struct asmln*)malloc(sizeof(struct asmln));
	newln->truen = ln->truen;
	newln->instr = content;
	return newln;
}

void checkopamnt(int amnt, struct line* ln) {
	if(ln->tokenscount < 2) {
		fprintf(stderr, "Missing memory segment; line %i", ln->truen);
		exit(1);
	}
	if(amnt > 2)
		if(ln->tokenscount < 3) {
			fprintf(stderr, "Missing operation index; line %i", ln->truen);
			exit(1);
		}
}

void startpoppush(struct line* ln, struct asmln*** asmlns, int* asmind) {
	checkopamnt(3, ln);
	int asmi = *asmind;

	// // operation segment i
	int indlen = strlen(ln->tokens[2]);
	(*asmlns)[asmi] = mkasmln(ln, mkcom(ln, indlen));
	asmi++;
	
	// @segment
	(*asmlns)[asmi] = mkasmln(ln, switchseg(ln));
	asmi++;
	
	// D=M
	(*asmlns)[asmi] = mkasmln(ln, heapstr("D=M"));
	asmi++;

	// @i
	checkind(ln, indlen);
	(*asmlns)[asmi] = mkasmln(ln, mkind(ln->tokens[2], indlen));
	asmi++;

	(*asmind) = asmi;
}

void pushcons(struct line* ln, struct asmln*** asmlns, int* asmind, int* asmsize) {
	checkopamnt(3, ln);
	int asmi = *asmind;
	int totalinstrs = 8;
	checkasmsize(asmsize, asmlns, sizeof(struct asmln*)*(asmi+totalinstrs));

	// // push constant i
	int indlen = strlen(ln->tokens[2]);
	(*asmlns)[asmi] = mkasmln(ln, mkcom(ln, indlen));
	asmi++;

	// @i
	checkind(ln, indlen);
	(*asmlns)[asmi] = mkasmln(ln, mkind(ln->tokens[2], indlen));
	asmi++;

	const int instcount = 6;
	const char* instrs[] = { 
		"D=A",
		"@SP",
		"A=M",
		"M=D",
		"@SP",
		"M=M+1"
	};

	for(int i = 0; i < instcount; i++) {
		(*asmlns)[asmi] = mkasmln(ln, heapstr(instrs[i]));
		asmi++;
	}
	
	(*asmind) = asmi;
}

void pushstat(struct line* ln, struct asmln*** asmlns, int* asmind, int* asmsize, char* fname) {
	checkopamnt(3, ln);
	int asmi = *asmind;
	int totalinstrs = 8;
	checkasmsize(asmsize, asmlns, sizeof(struct asmln*)*(asmi+totalinstrs));

	// // push static i
	int indlen = strlen(ln->tokens[2]);
	(*asmlns)[asmi] = mkasmln(ln, mkcom(ln, indlen));
	asmi++;

	// @fname.i
	checkind(ln, indlen);
	(*asmlns)[asmi] = mkasmln(ln, mkstatind(ln->tokens[2], indlen, fname));
	asmi++;

	const int instcount = 6;
	const char* instrs[] = { 
		"D=M",
		"@SP",
		"A=M",
		"M=D",
		"@SP",
		"M=M+1"
	};

	for(int i = 0; i < instcount; i++) {
		(*asmlns)[asmi] = mkasmln(ln, heapstr(instrs[i]));
		asmi++;
	}
	
	(*asmind) = asmi;
}

void popstat(struct line* ln, struct asmln*** asmlns, int* asmind, int* asmsize, char* fname) {
	checkopamnt(3, ln);
	int asmi = *asmind;
	int totalinstrs = 8;
	checkasmsize(asmsize, asmlns, sizeof(struct asmln*)*(asmi+totalinstrs));

	// // pop static i
	int indlen = strlen(ln->tokens[2]);
	(*asmlns)[asmi] = mkasmln(ln, mkcom(ln, indlen));
	asmi++;

	const int instcount = 5;
	const char* instrs[] = { 
		"@SP",
		"D=M",
		"M=M-1",
		"A=D",
		"D=M"
	};

	for(int i = 0; i < instcount; i++) {
		(*asmlns)[asmi] = mkasmln(ln, heapstr(instrs[i]));
		asmi++;
	}

	// @fname.i
	checkind(ln, indlen);
	(*asmlns)[asmi] = mkasmln(ln, mkstatind(ln->tokens[2], indlen, fname));
	asmi++;

	// M=D
	(*asmlns)[asmi] = mkasmln(ln, heapstr("M=D"));
	asmi++;
	
	(*asmind) = asmi;
}

void pushtemp(struct line* ln, struct asmln*** asmlns, int* asmind, int* asmsize) {
	int asmi = *asmind;
	int totalinstrs = 8;
	checkasmsize(asmsize, asmlns, sizeof(struct asmln*)*(asmi+totalinstrs));

	// // pop static i
	int indlen = strlen(ln->tokens[2]);
	(*asmlns)[asmi] = mkasmln(ln, mkcom(ln, indlen));
	asmi++;
	
	// @5+i
	checkind(ln, indlen);
	(*asmlns)[asmi] = mkasmln(ln, mktempind(ln->tokens[2], indlen));
	asmi++;

	const int instcount = 6;
	const char* instrs[] = {
		"D=M",
		"@SP",
		"A=M",
		"M=D",
		"@SP",
		"M=M+1"
	};

	for(int i = 0; i < instcount; i++) {
		(*asmlns)[asmi] = mkasmln(ln, heapstr(instrs[i]));
		asmi++;
	}

	(*asmind) = asmi;
}

void mkpush(struct line* ln, struct asmln*** asmlns, int* asmind, int* asmsize) {
	int asmi = *asmind;
	int totalinstrs = 11;
	checkasmsize(asmsize, asmlns, sizeof(struct asmln*)*(asmi+totalinstrs));

	startpoppush(ln, asmlns, &asmi);

	const int instcount = 7;
	const char* instrs[] = {
		"A=D+A",
		"D=M",
		"@SP",
		"A=M",
		"M=D",
		"@SP",
		"M=M+1"
	};

	for(int i = 0; i < instcount; i++) {
		(*asmlns)[asmi] = mkasmln(ln, heapstr(instrs[i]));
		asmi++;
	}

	(*asmind) = asmi;
}

void poptemp(struct line* ln, struct asmln*** asmlns, int* asmind, int* asmsize) {
	int asmi = *asmind;
	int totalinstrs = 7;
	checkasmsize(asmsize, asmlns, sizeof(struct asmln*)*(asmi+totalinstrs));

	// // pop static i
	int indlen = strlen(ln->tokens[2]);
	(*asmlns)[asmi] = mkasmln(ln, mkcom(ln, indlen));
	asmi++;

	const int instcount = 4;
	const char* instrs[] = {
		"@SP",
		"M=M-1",
		"A=M",
		"D=M"
	};

	for(int i = 0; i < instcount; i++) {
		(*asmlns)[asmi] = mkasmln(ln, heapstr(instrs[i]));
		asmi++;
	}
	
	// @5+i
	checkind(ln, indlen);
	(*asmlns)[asmi] = mkasmln(ln, mktempind(ln->tokens[2], indlen));
	asmi++;

	// M=D
	(*asmlns)[asmi] = mkasmln(ln, heapstr("M=D"));
	asmi++;

	(*asmind) = asmi;
}

void mkpop(struct line* ln, struct asmln*** asmlns, int* asmind, int* asmsize) {
	int asmi = *asmind;
	int totalinstrs = 14;
	checkasmsize(asmsize, asmlns, sizeof(struct asmln*)*(asmi+totalinstrs));

	startpoppush(ln, asmlns, &asmi);

	const int instcount = 10;
	const char* instrs[] = {
		"D=D+A",
		"@R13",
		"M=D",
		"@SP",
		"M=M-1",
		"A=M",
		"D=M",
		"@R13",
		"A=M",
		"M=D"
	};

	for(int i = 0; i < instcount; i++) {
		(*asmlns)[asmi] = mkasmln(ln, heapstr(instrs[i]));
		asmi++;
	}

	(*asmind) = asmi;
}

void switchop(struct line* ln, struct asmln*** asmlns, int* asmind, int* asmsize, char* fname) {
	if(strcmp(ln->tokens[0], "push") == 0)
		if(strcmp(ln->tokens[1], "constant") == 0)
			pushcons(ln, asmlns, asmind, asmsize);
		else if(strcmp(ln->tokens[1], "static") == 0)
			pushstat(ln, asmlns, asmind, asmsize, fname);
		else if(strcmp(ln->tokens[1], "temp") == 0)
			pushtemp(ln, asmlns, asmind, asmsize);
		else
			mkpush(ln, asmlns, asmind, asmsize);
	else if(strcmp(ln->tokens[0], "pop") == 0)
		if(strcmp(ln->tokens[1], "static") == 0)
			popstat(ln, asmlns, asmind, asmsize, fname);
		else if(strcmp(ln->tokens[1], "temp") == 0)
			poptemp(ln, asmlns, asmind, asmsize);
		else
			mkpop(ln, asmlns, asmind, asmsize);
}

struct asmln** translate(struct line** lns, int lnscount, int* asmcount, char* fname) {
	int sizebet = sizeof(struct asmln*)*(lnscount * 15);
	int asmi = (*asmcount);
	struct asmln** asmlns = (struct asmln**)malloc(sizebet);
	
	for(int i = 0; i < lnscount; i++) {
		struct line* ln = lns[i];
		switchop(ln, &asmlns, &asmi, &sizebet, fname);
	}

	(*asmcount) = asmi;
	return asmlns;
}