CB2C.PY

#!/usr/bin/python
import sys
import os
#stack for keeping track of IF statements - holds line numbers of IF
ifStatementStack = []
#stack for keeping track of LOOPWHILE statements - holds like numbers of LOOPWHILE
loopStatementStack = []
#giant buffer holding output
output = []
#list of names of variables

#format: VARNAME SIZE SIGN
variables = []

varnames = []
vartypes = ["BINARY", "HEX", "CHAR"]
varsizes = []
varstofree = []

# This if only runs if this file is called as a script - if it is included, it doesn't
if __name__ == "__main__":
	if len(sys.argv) < 2:
		print 'No file specified'
		quit()
	inputf = open(sys.argv[1])
	linenum = 1
	# Flags for program START and END - one per program
	hasstart = 0
	hasend = 0

	# Header for all CUTE BASIC compilations
	output.append("#include \"CBSTDLIB.H\"")
	
	# Comment for finding the struct declarations
	output.append("/*Structs go here*/")
	output.append("")
	output.append("")
	
	# Basic nibble type
	output.append("typedef struct _nib")
	output.append("{")
	output.append("uint8_t data : 4;")
	output.append("} nib;")
	
	output.append("typedef struct _snib")
	output.append("{")
	output.append("int8_t data : 4;")
	output.append("} snib;")
	
	output.append("/*Function declarations go here*/")
	output.append("/*Function declarations go here*/")
	output.append("")
	output.append("")

	multicomment = 0
	
	for line in inputf.readlines():
		
		if len(line) > 1 and (line.find("#/") != -1):
			multicomment = 0
			continue
	
		elif multicomment == 1:
			continue	
		
		elif len(line.strip()) > 1 and line.strip()[0] == "/" and line.strip()[1] == "#":
			multicomment = 1
			continue

		elif len(line.strip()) > 0 and line.strip()[0] == "#":
			continue

		# Check for the start
		if line.strip() == "START":
			if hasstart == 1:
				print "START already declared"
			else:
				hasstart = 1
				output.append("")
				output.append("")
				output.append("int main()")
				output.append("{")
				# Some fun
				output.append("puts(HEADER);")
				output.append("puts(\"\");")
		elif line.strip() == "END":
			if hasend == 1:
				print "END already declared"
			else:
				hasend = 1
				output.append("END:")
				output.append("return 0;")
				output.append("}")
		splitline = line.split()
		# Variable declaration and initalisation
		# LET - specifically declaration
		if len(splitline) == 0:
			continue
		elif splitline[0] == "ASM" or splitline[0] == "PASSTHROC":
			del splitline[0]
			newline = ""
			for i in splitline:
				newline += i
				newline += " "
			output.append(newline)
			continue
		
		
		elif splitline[0] == "LET":
			for i in varnames:
				if len(splitline) > 1 and i == splitline[1]:
					print "Variable name " + splitline[1] + " already declared!"
					quit(1)
			# The variable name will be the second whitespace delimited field
			varnames.append(splitline[1])

				
			
			if len(splitline) > 2 and splitline[2] == "AS":
				# Firstly, check if we're using SIGNED or UNSIGNED - if neither, UNSIGNED.
				if len(splitline) <= 6 or splitline[6] == "UNSIGNED":
					output.append("nib *" + splitline[1] + " = malloc(sizeof(nib) * " + splitline[3] + ");")
					#output.append(splitline[1] + ".extdata = malloc(" + str(int(splitline[3]) * 8) + ");")
					varstofree.append(splitline[1])
				elif len(splitline) > 6 and splitline[6] == "SIGNED":
					output.append("snib *" + splitline[1] + " = malloc(sizeof(snib) * " + splitline[3] + ");")
					#output.append(splitline[1] + ".extdata = malloc(" + str(int(splitline[3]) * 8) + ");")
					varstofree.append(splitline[1])
				if len(splitline) > 4 and splitline[4] == "BE":
					output.append(splitline[1] + ".data = " + splitline[5] + ";")
			else:
				output.append("nib *" + splitline[1] + " = malloc(sizeof(nib));")
				if splitline[2] == "BE":
					output.append(splitline[1] + "[0].data = " + splitline[3] + ";")
					
					
		#Math, with assignment as the first thing on the line
		elif ((splitline[0] in varnames) and (splitline[1] == "BE" or splitline[3] == "BE")): 
			vartypes = ["BINARY", "HEX", "CHAR"]
			vartype = ""
			indexright = 0
			indexleft = 0
			if splitline[1] == "AT":
				indexright = int(splitline[2])
				del splitline[1]
				del splitline[1]
			if len(splitline) > 3 and splitline[3] == "AT":
				indexleft = int(splitline[4])
				del splitline[3]
				del splitline[3]
			if(splitline[2] in ["COMPLEMENT", "NEGATE", "INCREMENT", "DECREMENT", "NOT", "ADDRESSOF", "CONTENTOF"]):
				#unary operations
				
				if splitline[3] in vartypes:
					vartype = splitline[3]
					splitline.remove(splitline[3])
					if vartype == "BINARY":
						splitline[3] = "0b" + splitline[3]
					elif vartype == "HEX":
						splitline[3] = "0x" + splitline[3]
					elif vartype[3] == "CHAR":
						splitline[3] = "\'" + splitline[3] + "\'"
				elif splitline[3] in varnames:
					splitline[3] = splitline[3] + "[" + str(indexleft) + "].data"
				splitline[0] += "[" + str(indexright) + "]"
				#unary math/bit/logic operators
				if splitline[2] == "COMPLEMENT":
					output.append(splitline[0] + ".data = ~" + splitline[3] + ";")
				elif splitline[2] == "NEGATE":
					output.append(splitline[0] + ".data = -" + splitline[3] + ";")
				elif splitline[2] == "INCREMENT":
					output.append(splitline[0] + ".data = " + splitline[3] + " + 1;")
				elif splitline[2] == "DECREMENT":
					output.append(splitline[0] + ".data = " + splitline[3] + " - 1;")
				elif splitline[2] == "NOT":
					output.append(splitline[0] + ".data = !" + splitline[3] + ";")
				elif splitline[2] == "ADDRESSOF":
					output.append(splitline[0] + ".data = &(" + splitline[3] + ");")
				elif splitline[2] == "CONTENTOF":
					output.append(splitline[0] + ".data = *(" + splitline[3] + ");")
			
			binopindexright = 0			
			binopindexleft = 0			

			if splitline[2] in vartypes:
				vartype = splitline[2]
				splitline.remove(splitline[2])
				if vartype == "BINARY":
					splitline[2] = "0b" + splitline[2]
				elif vartype == "HEX":
					splitline[2] = "0x" + splitline[2]
				elif vartype[2] == "CHAR":
					splitline[2] = "\'" + splitline[2] + "\'"
			else:
				if len(splitline) > 3 and splitline[3] == "AT":
					binopindexright = int(splitline[3])
					del splitline[3:4]
				splitline[2] = splitline[2] + "[" + str(binopindexright) + "].data"
				
			if len(splitline) > 4:
				if splitline[4] in vartypes:
					vartype = splitline[4]
					splitline.remove(splitline[4])
					if vartype == "BINARY":
						splitline[4] = "0b" + splitline[4]
					elif vartype == "HEX":
						splitline[4] = "0x" + splitline[4]
					elif vartype[4] == "CHAR":
						splitline[4] = "\'" + splitline[4] + "\'"
				else:
					if len(splitline) > 5 and splitline[5] == "AT":
						binopindexright = int(splitline[6])
						del splitline[5:6]
					splitline[4] = splitline[4] + "[" + str(binopindexleft) + "].data"

			print splitline
				
			if len(splitline) > 3:	
				#binary math operators
				if splitline[3] == "ADD":
					output.append(splitline[0] + ".data = " + splitline[2] + " + " + splitline[4] + " ;")
				elif splitline[3] == "SUBTRACT":
					output.append(splitline[0] + ".data = " + splitline[2] + " - " + splitline[4] + " ;")
				elif splitline[3] == "MULTIPLY":
					output.append(splitline[0] + ".data = " + splitline[2] + " * " + splitline[4] + " ;")
				elif splitline[3] == "DIVIDE":
					output.append(splitline[0] + ".data = " + splitline[2] + " / " + splitline[4] + " ;")
				elif splitline[3] == "MODULUS":
					output.append(splitline[0] + ".data = " + splitline[2] + " % " + splitline[4] + " ;")
			
				#binary bitwise operators
				elif splitline[3] == "AND":
					output.append(splitline[0] + ".data = " + splitline[2] + " & " + splitline[4] + " ;")
				elif splitline[3] == "NAND":
					output.append(splitline[0] + ".data = ~(" + splitline[2] + " & " + splitline[4] + ") ;")
				elif splitline[3] == "OR":
					output.append(splitline[0] + ".data = " + splitline[2] + " | " + splitline[4] + " ;")
				elif splitline[3] == "NOR":
					output.append(splitline[0] + ".data = ~(" + splitline[2] + " | " + splitline[4] + ") ;")
				elif splitline[3] == "XOR":
					output.append(splitline[0] + ".data = " + splitline[2] + " ^ " + splitline[4] + ";")
				elif splitline[3] == "XNOR":
					output.append(splitline[0] + ".data = ~(" + splitline[2] + " ^ " + splitline[4] + ") ;")
				
				#binary bitshifting
				elif (splitline[3] == "RROTATE" or splitline[3] == "RSHIFT"):
					output.append(splitline[0] + ".data = " + splitline[2] + " >> " + splitline[4] + ";")
				elif (splitline[3] == "LROTATE" or splitline[3] == "LSHIFT"):
					output.append(splitline[0] + ".data = " + splitline[2] + " << " + splitline[4] + ";")
				
				#binary logical operators
				elif splitline[3] == "ALSO":
					output.append(splitline[0] + ".data = " + splitline[2] + " && " + splitline[4] + ";")
				elif splitline[3] == "EITHER":
					output.append(splitline[0] + ".data = " + splitline[2] + " || " + splitline[4] + ";")
				
				#Logical tests
				elif splitline[3] == "EQUALS":
					output.append(splitline[0] + ".data = " + splitline[2] + " == " + splitline[4] + ";")
				elif splitline[3] == "NOTEQUALS":
					output.append(splitline[0] + ".data = " + splitline[2] + " != " + splitline[4] + ";")
				elif splitline[3] == "GREATER":
					output.append(splitline[0] + ".data = " + splitline[2] + " > " + splitline[4] + ";")
				elif splitline[3] == "GREATEREQUALS":
					output.append(splitline[0] + ".data = " + splitline[2] + " >= " + splitline[4] + ";")
				elif splitline[3] == "LESS":
					output.append(splitline[0] + ".data = " + splitline[2] + " < " + splitline[4] + ";")
				elif splitline[3] == "LESSEQUALS":
					output.append(splitline[0] + ".data = " + splitline[2] + " <= " + splitline[4] + ";")

		#Conditionals + GOTO
		elif(splitline[0] in ["IF", "ELSEIF", "ELSE", "ENDIF", "LOOPWHILE", "ENDLOOP", "LOOPAGAIN", "EXITLOOP", "EXITIF", "EARLYEND", "LABEL", "GOTO"]):
			vartypes = ["BINARY", "HEX", "CHAR"]
			vartype = ""
			boolindex = 0
			if len(splitline) > 1 and splitline[1] in vartypes:
				vartype = splitline[1]
				splitline.remove(splitline[1])
				if vartype == "BINARY":
					splitline[1] = "0b" + splitline[1]
				elif vartype == "HEX":
					splitline[1] = "0x" + splitline[1]
				elif vartype[1] == "CHAR":
					splitline[1] = "\'" + splitline[1] + "\'"
			elif len(splitline) > 1 and splitline[2] == "AT":
				boolindex = int(splitline[3])
			if len(splitline) > 1:
				splitline[1] = splitline[1] + "[" + str(boolindex) + "].data"
	
			#If/else/elseif/endif
			if splitline[0] == "IF":
				output.append("if (" + splitline[1] + ") {")
				ifStatementStack.append(linenum)
			elif splitline[0] == "ELSEIF":
				output.append("}")
				output.append("else if (" + splitline[1] + ") {")
			elif splitline[0] == "ELSE":
				output.append("}")
				output.append("else {")
			elif splitline[0]== "ENDIF":
				output.append("}")
				output.append("exitif"  + str(ifStatementStack.pop()) + ":")
				output.append(";")
				
			
			#Loopwhile/Endloop
			elif splitline[0] == "LOOPWHILE":
				output.append("while (" + splitline[1] + ") {")
				loopStatementStack.append(linenum)
			elif splitline[0] == "ENDLOOP":
				output.append("}")
				output.append("exitloop" + str(loopStatementStack.pop()) + ":")
				output.append(";")
			
			#GOTO derivatives
			elif splitline[0] == "LOOPAGAIN":
				output.append("continue;")
			elif splitline[0] == "EXITLOOP":
				output.append("goto exitloop" + str(loopStatementStack[-1]) + ";")
			elif splitline[0] == "EXITIF":
				output.append("goto exitif"+ str(ifStatementStack[-1]) + ";")
		
			#GOTO/Labels
			elif splitline[0] == "EARLYEND":
				output.append("goto END;")
			elif splitline[0] == "LABEL":
				output.append(splitline[1] + ":")
			elif splitline[0] == "GOTO":
				output.append("goto " + splitline[1] + ";")
				
		#function definition
		#if splitline[0] == "DEFINE":
		if splitline[0] == "FUNCTION":
			#must match declaration sig
			nextparam = 0
			temp = ""
			if len(splitline) > 3 and splitline[2] == "RETURNS":
				temp += "nibx" + splitline[3] + " " + splitline[1] + "("
				nextparam = 4
			else:
				temp += "void " + splitline[1] + "("
				nextparam = 2
			while len(splitline) >= nextparam + 2:
				if splitline[nextparam] != "TAKES" and splitline[nextparam] != "ASWELL":
					nextparam = nextparam + 1
					continue
				temp += "nibx" + splitline[nextparam + 1] + " " + splitline[nextparam + 2]
				nextparam += 3
				if len(splitline) >= nextparam + 2:
					temp += ","
			temp2 = temp
			temp2 += ");"
			functionindex = output.index("/*Function declarations go here*/")
			output.insert(functionindex + 1, temp2)
			temp += "){"
			output.append(temp)
		elif splitline[0] == "ENDFUNCTION":
			output.append("}")
		elif splitline[0] == "RETURN":
			output.append("return " + splitline[1] +";")
		elif splitline[0] == "CALL":
			if splitline[1] == "GETCHAR":
				
				#assume we're starting at index 0, unless told otherwise
				startindex = 0
				readlocation = "stdin"
				
				#this is how they can tell us otherwise
				if len(splitline) > 3 and splitline[3] == "AT":
					startindex = splitline[4]
					splitline.remove(splitline[3])
					splitline.remove(splitline[3])
				if len(splitline) > 3:
					readlocation = splitline[3]
					
				#output.append(splitline[1] + ".extdata[0] = getchar();")
				if readlocation != "stdin":
					output.append("FILE *file" + str(linenum) + "= fopen(\"" + readlocation + "\", \"r\");")
					readlocation = "file" + str(linenum)
				output.append("char tempc" + str(linenum) + " = getc(" + readlocation + ");")
				output.append(splitline[2] + "[" + str(startindex) + "].data = tempc" + str(linenum) + ";")
				output.append(splitline[2] + "[" + str(int(startindex) + 1) + "].data = tempc" + str(linenum) + " >> 4;")
				if readlocation != "stdin":
					output.append("fclose(file" + str(linenum) + ");")			
			elif splitline[1] == "PUTCHAR":	
				
				#assume we're starting at index 0, unless told otherwise
				startindex = 0
				writelocation = "stdout"
				
				#this is how they can tell us otherwise
				if len(splitline) > 3 and splitline[3] == "AT":
					startindex = splitline[4]
					splitline.remove(splitline[3])
					splitline.remove(splitline[3])
				if len(splitline) > 3:
					writelocation = splitline[3]
				
				output.append("char ch" + str(linenum) + " = 0;")
				output.append("ch" + str(linenum) + " = (ch" + str(linenum) + " & 0xF0) | (" + str(splitline[2]) + "[" + str(startindex) + "].data & 0xF);")
				output.append("ch" + str(linenum) + " = (ch" + str(linenum) + " & 0x0F) | ((" + str(splitline[2]) + "[" + str(int(startindex) + 1) + "].data & 0xF) << 4);")
				if writelocation != "stdout":
					output.append("FILE *file" + str(linenum) + "= fopen(\"" + writelocation + "\", \"a\");")
					writelocation = "file" + str(linenum)
				output.append("putc(ch" + str(linenum) + ", " + writelocation + ");")
				if writelocation != "stdout":				
					output.append("fclose(file" + str(linenum) + ");")
			else:
				nextparam = 2
				temp = ""
				temp += splitline[1] + "( "
				while len(splitline) > nextparam:
					temp += splitline[nextparam]
					if len(splitline) >= nextparam + 2:
						temp += ", "
					nextparam += 2
				temp += ");"
				output.append(temp)
		
		if splitline[0] == "NOHEADER":
			output.remove("puts(HEADER);")
			output.remove("puts(\"\");")
		
			
		linenum = linenum + 1
	inputf.close()
	if hasstart == 0:
		print "No START declared!"
		quit(1)
	if hasend == 0:
		print "No END declared!"
		quit(1)
	if hasend == 1 and hasstart == 1:
		for i in output:
			print i
	quit(0)