-
-
Notifications
You must be signed in to change notification settings - Fork 0
/
compile_go.py
120 lines (87 loc) · 3.17 KB
/
compile_go.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
# Written by Folkert van Heusden
# Released in the public domain
# www.vanheusden.com
import sys
from compile_base import CompileBase
class CompileToGo(CompileBase):
loopNr = 0
lnrs = []
def header(self):
print('Brainfuck to Go compiler.', file=sys.stderr)
def get_name():
return ('go', None)
def genindent(self, level):
return ' ' * (level * 3)
def invokeFunction(self, funcNr):
print('%sf%d()' % (self.genindent(self.lindentlevel), funcNr))
def addToDataPtr(self, n, dot, position):
ind = self.genindent(self.lindentlevel)
if n == 1:
print('%sdata_ptr++' % ind)
else:
print('%sdata_ptr += %d' % (ind, n))
def subFromDataPtr(self, n, dot, position):
ind = self.genindent(self.lindentlevel)
if n == 1:
print('%sdata_ptr--' % ind)
else:
print('%sdata_ptr -= %d' % (ind, n))
def addToData(self, n, dot, position):
ind = self.genindent(self.lindentlevel)
if n == 1:
print('%sdata_mem[data_ptr]++' % ind)
else:
print('%sdata_mem[data_ptr] += %d' % (ind, n))
def subFromData(self, n, dot, position):
ind = self.genindent(self.lindentlevel)
if n == 1:
print('%sdata_mem[data_ptr]--' % ind)
else:
print('%sdata_mem[data_ptr] -= %d' % (ind, n))
def emitCharacter(self, n, dot):
for i in range(0, n):
print('%sfmt.Printf("%%c", data_mem[data_ptr])' % self.genindent(self.lindentlevel))
def startLoop(self, n, position):
for j in range(0, n):
print('%sfor data_mem[data_ptr] > 0 {' % self.genindent(self.lindentlevel))
self.lindentlevel += 1
def finishLoop(self, n, dot, position):
for j in range(0, n):
self.lindentlevel -= 1
print('%s}' % self.genindent(self.lindentlevel))
def addComment(self, s):
print('/* %s */' % s)
def multilineCommentStart(self):
print('/*')
def multilineCommentLine(self, s):
print(' * %s' % s)
def multilineCommentEnd(self):
print(' */')
def emitProgramBootstrap(self, file):
print('package main')
print('')
print(f'// This is a translation of "{file}".')
for i in self.copyrightNotice:
print('// %s' % i)
print('')
print('import (')
print(' "fmt"')
print(')')
print('')
print('var data_ptr uint64')
print('var data_mem [32768]uint8')
print('')
def emitFunctions(self):
for blkLoop in range(0, len(self.blocks)):
print('%sfunc f%d() {' % (self.genindent(self.lindentlevel), blkLoop))
self.lindentlevel += 1
self.translate(self.blocks[blkLoop][0], self.blocks[blkLoop][1])
self.lindentlevel -= 1
print('%s}' % self.genindent(self.lindentlevel))
def emitMainFunction(self):
print('func main() {')
self.lindentlevel += 1
print('%sdata_ptr = 0' % self.genindent(self.lindentlevel))
self.translate(0, len(self.allCode))
self.lindentlevel -= 1
print('}')