def main(args):
contextf = open(args[0], 'r')
exprfile = open(args[1], 'r')
result = args[2]
try:
p = parser.parseProgram(' '.join(contextf.readlines())).optimize()
contextf.close()
except:
print 'Parsing of context failed.'
exprfile.close()
contextf.close()
sys.exit(1)
expr = exprfile.readline().strip().split()
exprfile.close()
try:
r = apply(p[expr[0]], map(int, expr[1:]))
except:
print 'Parsing succeeded, evaluation failed.'
sys.exit(2)
if result == ` r `:
print 'Evaluation result confirmed.'
sys.exit(0)
else:
print 'Evaluation succeeded with unexpected result:', r
sys.exit(3)
def main(args):
contextf = open (args[0],'r')
exprfile = open (args[1],'r')
result = args[2]
try:
p=parser.parseProgram(' '.join(contextf.readlines())).optimize()
contextf.close()
except:
print 'Parsing of context failed.'
exprfile.close()
contextf.close()
sys.exit(1)
expr = exprfile.readline().strip().split()
exprfile.close()
try:
r = apply(p[expr[0]],map(int,expr[1:]))
except:
print 'Parsing succeeded, evaluation failed.'
sys.exit(2)
if result==`r`:
print 'Evaluation result confirmed.'
sys.exit(0)
else:
print 'Evaluation succeeded with unexpected result:',r
sys.exit(3)
class TestPyFL(unittest.TestCase):
p = parser.parseProgram("""
mult n m = if n == 0 then 0 else m+(mult (n-1) m)
fac n = if n == 0 then 1 else mult n (fac (n-1))
test n = if n == 1 then 1 else mult (n+1) (fac (n-1)) (if n then 0 else m)
drei n = if n==n then 1+1+1 else 0+7-0
sn n = n-1+n-n+1+n+5-n+2-n""")
def testPrettyPrinter(self):
self.assertEqual(str(self.p['mult']), "mult n m = if(n == 0) then 0 else m + (mult (n - 1) m)")
self.assertEqual(str(self.p['fac']), "fac n = if(n == 0) then 1 else (mult n (fac (n - 1)))")
def testEvaluator(self):
self.assertEqual(self.p.eval("fac", (6,)), 720)
self.assertEqual(self.p.optimize().eval("fac", (6,)), 720)
self.assertEqual(parser.parseApply("fac 6").eval(funcs=self.p), 720)
def testEvaluatorCached(self):
import caching_evaluator
self.testEvaluator()
def testOptimizer(self):
self.assertEqual(str(self.p.optimize()['sn']), "sn n = 7")
self.assertEqual(str(self.p.optimize()['fac']), "fac n = if n then (mult n (fac (-1 + n))) else 1")
self.assertEqual(self.p.optimize()['fac'](6), 720)
def testParser(self):
self.assertRaises(pyparsing.ParseException, parser.parseProgram, "n")
self.assertRaises(pyparsing.ParseException, parser.parseProgram, "fac n = if n == 0 then 1 else mult n (fac (n-1))\nfoo")
def main(args):
term = open(args[0], 'r')
resu = open(args[1], 'w')
try:
resu.write(str(parser.parseProgram(' '.join(term.readlines()))) + '\n')
term.close()
resu.close()
print 'Parsing succeeded.'
sys.exit(0)
except StandardError:
term.close()
resu.close()
print 'Parsing failed.'
sys.exit(1)
def main(args):
term = open (args[0],'r')
resu = open (args[1],'w')
try:
resu.write(str(parser.parseProgram(' '.join(term.readlines())))+'\n')
term.close()
resu.close()
print 'Parsing succeeded.'
sys.exit(0)
except StandardError:
term.close()
resu.close()
print 'Parsing failed.'
sys.exit(1)
def generateCode(inputFile, debug=False, outputFile='pro'):
# Read the inputFile and parse the program
program = open(inputFile).read()
ST, TAC, debug = parseProgram(program)
# Log the data for inspection
if debug:
TAC.printCode('TAC')
debug.log(ST.symbolTable, 'symbols')
debug.log(ST.functionList, 'functionList')
debug.log(ST.addressDescriptor, 'addressDescriptor')
# We create a new object which will store the code
RTC = RuntimeCode.RuntimeCode(ST, TAC)
counter = 0
# Fix the labesl of the code, replace numbers with labels
RTC.fixLabels()
# We now traverse over the entire code to generate data
for function in TAC.code:
RTC.addFunction(function)
# Different stuff for main
if (function == 'main'):
#allocate space for the registers by updating stack pointer
RTC.addLine(['sub', '$sp', '$sp', '200'])
#set fame pointer of the callee
RTC.addLine(['la', '$fp', '200($sp)', ''])
RTC.addLine(['la', '$s5', '__display__', ''])
RTC.addLine(['lw', '$s7', '0($s5)', ''])
#set display[level]
RTC.addLine([
'la', '$v0',
'-' + str(ST.getAttributeFromFunctionList(function, 'width')) +
'($sp)', ''
])
RTC.addLine(['sw', '$v0', '0($s5)', ''])
RTC.addLine([
'li', '$v0',
ST.getAttributeFromFunctionList(function, 'width'), ''
])
RTC.addLine(['sub', '$sp', '$sp', '$v0'])
else:
#allocate space for the registers by updating stack pointer
RTC.addLine(['sub', '$sp', '$sp', '72'])
#store return address of the caller
RTC.addLine(['sw', '$ra', '0($sp)', ''])
#sstore the frame pointer of the caller
RTC.addLine(['sw', '$fp', '4($sp)', ''])
#set fame pointer of the callee
RTC.addLine(['la', '$fp', '72($sp)', ''])
#storing display[level]
RTC.addLine([
'li', '$v0',
ST.getAttributeFromFunctionList(function, 'level'), ''
])
RTC.addLine(['la', '$s5', '__display__', ''])
RTC.addLine(['add', '$v0', '$v0', '$v0'])
RTC.addLine(['add', '$v0', '$v0', '$v0'])
RTC.addLine(['add', '$s6', '$v0', '$s5'])
RTC.addLine(['lw', '$s7', '0($s6)', ''])
RTC.addLine(['sw', '$s7', '8($sp)', ''])
#set display[level]
RTC.addLine([
'la', '$v0',
'-' + str(ST.getAttributeFromFunctionList(function, 'width')) +
'($sp)', ''
])
RTC.addLine(['sw', '$v0', '0($s6)', ''])
#store remaining registers
RTC.addLine(['sw', '$t0', '12($sp)', ''])
RTC.addLine(['sw', '$t1', '16($sp)', ''])
RTC.addLine(['sw', '$t2', '20($sp)', ''])
RTC.addLine(['sw', '$t3', '24($sp)', ''])
RTC.addLine(['sw', '$t4', '28($sp)', ''])
RTC.addLine(['sw', '$t5', '32($sp)', ''])
RTC.addLine(['sw', '$t6', '36($sp)', ''])
RTC.addLine(['sw', '$t7', '40($sp)', ''])
RTC.addLine(['sw', '$t8', '44($sp)', ''])
RTC.addLine(['sw', '$t9', '48($sp)', ''])
RTC.addLine(['sw', '$s0', '52($sp)', ''])
RTC.addLine(['sw', '$s1', '56($sp)', ''])
RTC.addLine(['sw', '$s2', '60($sp)', ''])
RTC.addLine(['sw', '$s3', '64($sp)', ''])
RTC.addLine(['sw', '$s4', '68($sp)', ''])
# Create space for local data
RTC.addLine([
'li', '$v0',
ST.getAttributeFromFunctionList(function, 'width'), ''
])
RTC.addLine(['sub', '$sp', '$sp', '$v0'])
# Copy the parameters
for x in range(
ST.getAttributeFromFunctionList(function, 'numParam')):
RTC.addLine(['sw', '$a' + str(x), str(4 * x) + '($sp)', ''])
for line in TAC.code[function]:
if line[3] == 'JUMPLABEL':
counter = 0
reg = RTC.nextReg(line[2])
RTC.addLine(['jal', reg, '', ''])
RTC.reloadParents(
ST.getAttributeFromFunctionList(function, 'level'),
function)
elif line[3] == 'JUMPBACK':
RTC.addLine(['b', function + 'end', '', ''])
elif line[3] == 'PARAM':
reg = RTC.nextReg(line[0])
RTC.addLine(['move', '$a' + str(counter), reg, ''])
counter = counter + 1
elif line[3] == '=':
reg1 = RTC.nextReg(line[0])
reg2 = RTC.nextReg(line[1])
RTC.addLine(['move', reg1, reg2, ''])
RTC.flushTemporary(line[0])
elif line[3] == '=i':
reg = RTC.nextReg(line[0])
RTC.addLine(['li', reg, line[1], ''])
RTC.flushTemporary(line[0])
elif line[3] == '=REF':
reg = RTC.nextReg(line[0])
RTC.addLine(['la', reg, line[1], ''])
RTC.flushTemporary(line[0])
elif line[3] == 'uni-':
reg1 = RTC.nextReg(line[0])
reg2 = RTC.nextReg(line[1])
RTC.addLine(['neg', reg1, reg2, ''])
RTC.flushTemporary(line[0])
elif line[3] == '+':
reg1 = RTC.nextReg(line[0])
reg2 = RTC.nextReg(line[1])
reg3 = RTC.nextReg(line[2])
RTC.addLine(['add', reg1, reg2, reg3])
RTC.flushTemporary(line[0])
elif line[3] == '-':
reg1 = RTC.nextReg(line[0])
reg2 = RTC.nextReg(line[1])
reg3 = RTC.nextReg(line[2])
RTC.addLine(['sub', reg1, reg2, reg3])
RTC.flushTemporary(line[0])
elif line[3] == '*':
reg1 = RTC.nextReg(line[1])
reg2 = RTC.nextReg(line[2])
reg3 = RTC.nextReg(line[0])
RTC.addLine(['mult', reg1, reg2, ''])
RTC.addLine(['mflo', reg3, '', ''])
RTC.flushTemporary(line[0])
elif line[3] == '/':
reg1 = RTC.nextReg(line[1])
reg2 = RTC.nextReg(line[2])
reg3 = RTC.nextReg(line[0])
RTC.addLine(['div', reg1, reg2, ''])
RTC.addLine(['mflo', reg3, '', ''])
RTC.flushTemporary(line[0])
elif line[3] == '%':
reg1 = RTC.nextReg(line[1])
reg2 = RTC.nextReg(line[2])
reg3 = RTC.nextReg(line[0])
RTC.addLine(['div', reg1, reg2, ''])
RTC.addLine(['mfhi', reg3, '', ''])
RTC.flushTemporary(line[0])
elif line[3] == '<':
reg1 = RTC.nextReg(line[0])
reg2 = RTC.nextReg(line[1])
reg3 = RTC.nextReg(line[2])
RTC.addLine(['slt', reg1, reg2, reg3])
RTC.flushTemporary(line[0])
elif line[3] == '>':
reg1 = RTC.nextReg(line[0])
reg2 = RTC.nextReg(line[1])
reg3 = RTC.nextReg(line[2])
RTC.addLine(['sgt', reg1, reg2, reg3])
RTC.flushTemporary(line[0])
elif line[3] == '<=':
reg1 = RTC.nextReg(line[0])
reg2 = RTC.nextReg(line[1])
reg3 = RTC.nextReg(line[2])
RTC.addLine(['sle', reg1, reg2, reg3])
RTC.flushTemporary(line[0])
elif line[3] == '>=':
reg1 = RTC.nextReg(line[0])
reg2 = RTC.nextReg(line[1])
reg3 = RTC.nextReg(line[2])
RTC.addLine(['sge', reg1, reg2, reg3])
RTC.flushTemporary(line[0])
elif line[3] == '==':
reg1 = RTC.nextReg(line[0])
reg2 = RTC.nextReg(line[1])
reg3 = RTC.nextReg(line[2])
RTC.addLine(['seq', reg1, reg2, reg3])
RTC.flushTemporary(line[0])
elif line[3] == '!=':
reg1 = RTC.nextReg(line[0])
reg2 = RTC.nextReg(line[1])
reg3 = RTC.nextReg(line[2])
RTC.addLine(['sne', reg1, reg2, reg3])
RTC.flushTemporary(line[0])
elif line[3] == 'COND_GOTO_Z':
reg1 = RTC.nextReg(line[0])
RTC.addLine(['beq', reg1, '$0', line[2]])
elif line[3] == 'GOTO':
RTC.addLine(['b', line[2], '', ''])
elif line[3] == 'FUNCTION_RETURN':
reg1 = RTC.nextReg(line[0])
RTC.addLine(['move', reg1, '$v0', ''])
elif line[3] == 'RETURN':
reg1 = RTC.nextReg(line[0])
RTC.addLine(['move', '$v0', reg1, ''])
RTC.addLine(['b', function + 'end', '', ''])
elif line[3] == 'HALT':
RTC.addLine(['jal', 'exit', '', ''])
elif line[3] == 'PRINT' and line[0] == '':
RTC.addLine(['jal', 'print_newline', '', ''])
elif line[3] == 'PRINT' and line[2] == 'UNDEFINED':
RTC.addLine(['jal', 'print_undefined', '', ''])
elif line[3] == 'PRINT':
reg = RTC.nextReg(line[0])
RTC.addLine(['move', '$a0', reg, ''])
if line[2] == 'NUMBER':
RTC.addLine(['jal', 'print_integer', '', ''])
elif line[2] == 'STRING':
RTC.addLine(['jal', 'print_string', '', ''])
else:
RTC.addLine(['jal', 'print_boolean', '', ''])
else:
RTC.addLine(line)
# Data unloading happens for all function save main
if function != 'main':
# Add a label to point to the end of the function
RTC.addLine(['LABEL', function + 'end', '', ''])
# Remove the local data
RTC.addLine([
'addi', '$sp', '$sp',
ST.getAttributeFromFunctionList(function, 'width')
])
# Get enviornment pointers
RTC.addLine(['lw', '$ra', '0($sp)', ''])
RTC.addLine(['lw', '$fp', '4($sp)', ''])
RTC.addLine(['lw', '$a0', '8($sp)', ''])
# diplay level
RTC.addLine([
'li', '$a1',
ST.getAttributeFromFunctionList(function, 'level'), ''
])
RTC.addLine(['la', '$s5', '__display__', ''])
RTC.addLine(['add', '$a1', '$a1', '$a1'])
RTC.addLine(['add', '$a1', '$a1', '$a1'])
RTC.addLine(['add', '$s6', '$a1', '$s5'])
RTC.addLine(['sw', '$a0', '0($s6)', ''])
# Registers
RTC.addLine(['lw', '$t0', '12($sp)', ''])
RTC.addLine(['lw', '$t1', '16($sp)', ''])
RTC.addLine(['lw', '$t2', '20($sp)', ''])
RTC.addLine(['lw', '$t3', '24($sp)', ''])
RTC.addLine(['lw', '$t4', '28($sp)', ''])
RTC.addLine(['lw', '$t5', '32($sp)', ''])
RTC.addLine(['lw', '$t6', '36($sp)', ''])
RTC.addLine(['lw', '$t7', '40($sp)', ''])
RTC.addLine(['lw', '$t8', '44($sp)', ''])
RTC.addLine(['lw', '$t9', '48($sp)', ''])
RTC.addLine(['lw', '$s0', '52($sp)', ''])
RTC.addLine(['lw', '$s1', '56($sp)', ''])
RTC.addLine(['lw', '$s2', '60($sp)', ''])
RTC.addLine(['lw', '$s3', '64($sp)', ''])
RTC.addLine(['lw', '$s4', '68($sp)', ''])
RTC.addLine(['addi', '$sp', '$sp', '72'])
# Jump to the calling procedure
RTC.addLine(['jr', '$ra', '', ''])
# Print the generated code
RTC.printCode(outputFile)
#!/usr/bin/env python
import parser, evaluator, prettyprinter, optimizer
p = parser.parseProgram("""
mult n m = if n == 0 then 0 else m+(mult (n-1) m)
fac n = if n == 0 then 1 else mult n (fac (n-1))
test n = if n == 1 then 1 else mult (n+1) (fac (n-1)) (if n then 0 else m)
drei n = if n==n then 1+1+1 else 0+7-0
sn n = n-1+n-n+1+n+5-n+2-n""")
def main():
print p
print
optimized = p.optimize()
print p.optimize()
print
i = 1
try:
while True:
print "fac %d ="%i,
print optimized['fac'](i)
i *= 2
except RuntimeError, e:
print
print e
print
import caching_evaluator
print "Retrying with cache:"
def generateCode(inputFile, debug=False, outputFile='pro'):
# Read the inputFile and parse the program
program = open(inputFile).read()
ST, TAC, debug = parseProgram(program)
# Log the data for inspection
if debug:
TAC.printCode('TAC')
debug.log(ST.symbolTable, 'symbols')
debug.log(ST.functionList, 'functionList')
debug.log(ST.addressDescriptor, 'addressDescriptor')
# We create a new object which will store the code
RTC = RuntimeCode.RuntimeCode(ST, TAC)
counter = 0;
# Fix the labesl of the code, replace numbers with labels
RTC.fixLabels()
# We now traverse over the entire code to generate data
for function in TAC.code:
RTC.addFunction(function)
# Different stuff for main
if (function == 'main'):
#allocate space for the registers by updating stack pointer
RTC.addLine(['sub', '$sp', '$sp', '200'])
#set fame pointer of the callee
RTC.addLine(['la', '$fp', '200($sp)', ''])
RTC.addLine(['la', '$s5', '__display__', ''])
RTC.addLine(['lw', '$s7', '0($s5)', ''])
#set display[level]
RTC.addLine(['la', '$v0', '-' + str(ST.getAttributeFromFunctionList(function, 'width')) + '($sp)', ''])
RTC.addLine(['sw','$v0', '0($s5)', ''])
RTC.addLine(['li', '$v0', ST.getAttributeFromFunctionList(function, 'width'), ''])
RTC.addLine(['sub', '$sp', '$sp', '$v0'])
else:
#allocate space for the registers by updating stack pointer
RTC.addLine(['sub', '$sp','$sp','72'])
#store return address of the caller
RTC.addLine(['sw','$ra','0($sp)',''])
#sstore the frame pointer of the caller
RTC.addLine(['sw','$fp','4($sp)',''])
#set fame pointer of the callee
RTC.addLine(['la','$fp','72($sp)',''])
#storing display[level]
RTC.addLine(['li','$v0',ST.getAttributeFromFunctionList(function, 'level'),''])
RTC.addLine(['la', '$s5', '__display__', ''])
RTC.addLine(['add', '$v0', '$v0', '$v0'])
RTC.addLine(['add', '$v0', '$v0', '$v0'])
RTC.addLine(['add', '$s6', '$v0', '$s5'])
RTC.addLine(['lw','$s7','0($s6)',''])
RTC.addLine(['sw','$s7','8($sp)',''])
#set display[level]
RTC.addLine(['la', '$v0', '-' + str(ST.getAttributeFromFunctionList(function, 'width'))+'($sp)' , ''])
RTC.addLine(['sw','$v0','0($s6)',''])
#store remaining registers
RTC.addLine(['sw','$t0','12($sp)',''])
RTC.addLine(['sw','$t1','16($sp)',''])
RTC.addLine(['sw','$t2','20($sp)',''])
RTC.addLine(['sw','$t3','24($sp)',''])
RTC.addLine(['sw','$t4','28($sp)',''])
RTC.addLine(['sw','$t5','32($sp)',''])
RTC.addLine(['sw','$t6','36($sp)',''])
RTC.addLine(['sw','$t7','40($sp)',''])
RTC.addLine(['sw','$t8','44($sp)',''])
RTC.addLine(['sw','$t9','48($sp)',''])
RTC.addLine(['sw','$s0','52($sp)',''])
RTC.addLine(['sw','$s1','56($sp)',''])
RTC.addLine(['sw','$s2','60($sp)',''])
RTC.addLine(['sw','$s3','64($sp)',''])
RTC.addLine(['sw','$s4','68($sp)',''])
# Create space for local data
RTC.addLine(['li','$v0',ST.getAttributeFromFunctionList(function, 'width'),''])
RTC.addLine(['sub','$sp','$sp','$v0'])
# Copy the parameters
for x in range(ST.getAttributeFromFunctionList(function, 'numParam')):
RTC.addLine(['sw','$a' + str(x), str(4*x) + '($sp)', ''])
for line in TAC.code[function]:
if line[3] == 'JUMPLABEL':
counter = 0 ;
reg = RTC.nextReg(line[2])
RTC.addLine(['jal', reg, '', ''])
RTC.reloadParents(ST.getAttributeFromFunctionList(function, 'level'), function)
elif line[3] == 'JUMPBACK':
RTC.addLine(['b', function + 'end', '', ''])
elif line[3] == 'PARAM':
reg = RTC.nextReg(line[0])
RTC.addLine(['move', '$a'+str(counter), reg,''])
counter = counter +1 ;
elif line[3] == '=':
reg1 = RTC.nextReg(line[0])
reg2 = RTC.nextReg(line[1])
RTC.addLine(['move', reg1, reg2, ''])
RTC.flushTemporary(line[0])
elif line[3] == '=i':
reg = RTC.nextReg(line[0])
RTC.addLine(['li', reg, line[1], ''])
RTC.flushTemporary(line[0])
elif line[3] == '=REF':
reg = RTC.nextReg(line[0])
RTC.addLine(['la', reg, line[1], ''])
RTC.flushTemporary(line[0])
elif line[3] == 'uni-':
reg1 = RTC.nextReg(line[0])
reg2 = RTC.nextReg(line[1])
RTC.addLine(['neg', reg1, reg2, ''])
RTC.flushTemporary(line[0])
elif line[3] == '+':
reg1 = RTC.nextReg(line[0])
reg2 = RTC.nextReg(line[1])
reg3 = RTC.nextReg(line[2])
RTC.addLine(['add', reg1, reg2, reg3])
RTC.flushTemporary(line[0])
elif line[3] == '-':
reg1 = RTC.nextReg(line[0])
reg2 = RTC.nextReg(line[1])
reg3 = RTC.nextReg(line[2])
RTC.addLine(['sub', reg1, reg2, reg3])
RTC.flushTemporary(line[0])
elif line[3] == '*':
reg1 = RTC.nextReg(line[1])
reg2 = RTC.nextReg(line[2])
reg3 = RTC.nextReg(line[0])
RTC.addLine(['mult', reg1, reg2,''])
RTC.addLine(['mflo', reg3,'',''])
RTC.flushTemporary(line[0])
elif line[3] == '/':
reg1 = RTC.nextReg(line[1])
reg2 = RTC.nextReg(line[2])
reg3 = RTC.nextReg(line[0])
RTC.addLine(['div', reg1, reg2, ''])
RTC.addLine(['mflo', reg3, '', ''])
RTC.flushTemporary(line[0])
elif line[3] == '%':
reg1 = RTC.nextReg(line[1])
reg2 = RTC.nextReg(line[2])
reg3 = RTC.nextReg(line[0])
RTC.addLine(['div', reg1, reg2, ''])
RTC.addLine(['mfhi', reg3, '', ''])
RTC.flushTemporary(line[0])
elif line[3] == '<':
reg1 = RTC.nextReg(line[0])
reg2 = RTC.nextReg(line[1])
reg3 = RTC.nextReg(line[2])
RTC.addLine(['slt', reg1, reg2, reg3])
RTC.flushTemporary(line[0])
elif line[3] == '>':
reg1 = RTC.nextReg(line[0])
reg2 = RTC.nextReg(line[1])
reg3 = RTC.nextReg(line[2])
RTC.addLine(['sgt', reg1, reg2, reg3])
RTC.flushTemporary(line[0])
elif line[3] == '<=':
reg1 = RTC.nextReg(line[0])
reg2 = RTC.nextReg(line[1])
reg3 = RTC.nextReg(line[2])
RTC.addLine(['sle', reg1, reg2, reg3])
RTC.flushTemporary(line[0])
elif line[3] == '>=':
reg1 = RTC.nextReg(line[0])
reg2 = RTC.nextReg(line[1])
reg3 = RTC.nextReg(line[2])
RTC.addLine(['sge', reg1, reg2, reg3])
RTC.flushTemporary(line[0])
elif line[3] == '==':
reg1 = RTC.nextReg(line[0])
reg2 = RTC.nextReg(line[1])
reg3 = RTC.nextReg(line[2])
RTC.addLine(['seq', reg1, reg2, reg3])
RTC.flushTemporary(line[0])
elif line[3] == '!=':
reg1 = RTC.nextReg(line[0])
reg2 = RTC.nextReg(line[1])
reg3 = RTC.nextReg(line[2])
RTC.addLine(['sne', reg1, reg2, reg3])
RTC.flushTemporary(line[0])
elif line[3] == 'COND_GOTO_Z':
reg1 = RTC.nextReg(line[0])
RTC.addLine(['beq', reg1, '$0', line[2]])
elif line[3] == 'GOTO':
RTC.addLine(['b', line[2], '', ''])
elif line[3] == 'FUNCTION_RETURN':
reg1 = RTC.nextReg(line[0])
RTC.addLine(['move', reg1, '$v0', ''])
elif line[3] == 'RETURN':
reg1 = RTC.nextReg(line[0])
RTC.addLine(['move', '$v0', reg1, ''])
RTC.addLine(['b', function + 'end', '', ''])
elif line[3] == 'HALT':
RTC.addLine(['jal', 'exit', '', ''])
elif line[3] == 'PRINT' and line[0] == '':
RTC.addLine(['jal', 'print_newline', '', ''])
elif line[3] == 'PRINT' and line[2] == 'UNDEFINED':
RTC.addLine(['jal', 'print_undefined', '', ''])
elif line[3] == 'PRINT':
reg = RTC.nextReg(line[0])
RTC.addLine(['move', '$a0', reg, ''])
if line[2] == 'NUMBER':
RTC.addLine(['jal', 'print_integer', '', ''])
elif line[2] == 'STRING':
RTC.addLine(['jal', 'print_string', '', ''])
else:
RTC.addLine(['jal', 'print_boolean', '', ''])
else:
RTC.addLine(line)
# Data unloading happens for all function save main
if function != 'main':
# Add a label to point to the end of the function
RTC.addLine(['LABEL', function + 'end', '', ''])
# Remove the local data
RTC.addLine(['addi','$sp','$sp',ST.getAttributeFromFunctionList(function,'width')])
# Get enviornment pointers
RTC.addLine(['lw','$ra','0($sp)',''])
RTC.addLine(['lw','$fp','4($sp)',''])
RTC.addLine(['lw','$a0','8($sp)',''])
# diplay level
RTC.addLine(['li','$a1',ST.getAttributeFromFunctionList(function, 'level'),''])
RTC.addLine(['la', '$s5', '__display__', ''])
RTC.addLine(['add', '$a1', '$a1', '$a1'])
RTC.addLine(['add', '$a1', '$a1', '$a1'])
RTC.addLine(['add', '$s6', '$a1', '$s5'])
RTC.addLine(['sw','$a0','0($s6)',''])
# Registers
RTC.addLine(['lw','$t0','12($sp)',''])
RTC.addLine(['lw','$t1','16($sp)',''])
RTC.addLine(['lw','$t2','20($sp)',''])
RTC.addLine(['lw','$t3','24($sp)',''])
RTC.addLine(['lw','$t4','28($sp)',''])
RTC.addLine(['lw','$t5','32($sp)',''])
RTC.addLine(['lw','$t6','36($sp)',''])
RTC.addLine(['lw','$t7','40($sp)',''])
RTC.addLine(['lw','$t8','44($sp)',''])
RTC.addLine(['lw','$t9','48($sp)',''])
RTC.addLine(['lw','$s0','52($sp)',''])
RTC.addLine(['lw','$s1','56($sp)',''])
RTC.addLine(['lw','$s2','60($sp)',''])
RTC.addLine(['lw','$s3','64($sp)',''])
RTC.addLine(['lw','$s4','68($sp)',''])
RTC.addLine(['addi','$sp','$sp','72'])
# Jump to the calling procedure
RTC.addLine(['jr','$ra','',''])
# Print the generated code
RTC.printCode(outputFile)
#!/usr/bin/env python
import parser, evaluator, prettyprinter, optimizer
p = parser.parseProgram("""
mult n m = if n == 0 then 0 else m+(mult (n-1) m)
fac n = if n == 0 then 1 else mult n (fac (n-1))
test n = if n == 1 then 1 else mult (n+1) (fac (n-1)) (if n then 0 else m)
drei n = if n==n then 1+1+1 else 0+7-0
sn n = n-1+n-n+1+n+5-n+2-n""")
def main():
print p
print
optimized = p.optimize()
print p.optimize()
print
i = 1
try:
while True:
print "fac %d =" % i,
print optimized['fac'](i)
i *= 2
except RuntimeError, e:
print
print e
print
import caching_evaluator
outputMain = []
outputData = []
outputText = []
exitFound = False
regDesc = []
varList = []
initList = []
tacString = ""
if __name__ == '__main__':
input_ = open(argv[1]).read()
TAC, ST = parser.parseProgram(input_)
outputData.append(".data")
mips.code.append(".text")
# printsym()
# ST.printSymbolTable()
# ST.printFunctionList()
# mips.code.append("main:")
# print str(ST.getAttributeFromFunctionList('main','width'))
mips.code.append('sub $sp, $sp, '+str(ST.getAttributeFromFunctionList('main','width')))
mips.code.append('b '+ 'main')
# mips.code.append('add $sp, $sp, '+str(ST.getAttributeFromFunctionList('main','width')))
mips
lineNum = 0
# print ST.string
# print ST.string
