New repo, containing the basic idea
This commit is contained in:
Johan B.W. de Vries
commit
547576de00
8
Makefile
Normal file
8
Makefile
Normal file
@ -0,0 +1,8 @@
|
|||||||
|
%.wat: %.py compile.py
|
||||||
|
python3.8 compile.py $< > $@
|
||||||
|
|
||||||
|
%.wasm: %.wat
|
||||||
|
wat2wasm $^ -o $@
|
||||||
|
|
||||||
|
server:
|
||||||
|
python3.8 -m http.server
|
||||||
133
compile.py
Normal file
133
compile.py
Normal file
@ -0,0 +1,133 @@
|
|||||||
|
import _ast
|
||||||
|
import ast
|
||||||
|
import sys
|
||||||
|
|
||||||
|
class Import:
|
||||||
|
def __init__(self, module, name, intname):
|
||||||
|
self.module = module
|
||||||
|
self.name = name
|
||||||
|
self.intname = intname
|
||||||
|
self.params = None
|
||||||
|
|
||||||
|
def generate(self):
|
||||||
|
return '(import "{}" "{}" (func ${}{}))'.format(
|
||||||
|
self.module,
|
||||||
|
self.name,
|
||||||
|
self.intname,
|
||||||
|
''.join(' (param {})'.format(x) for x in self.params)
|
||||||
|
)
|
||||||
|
|
||||||
|
class Statement:
|
||||||
|
def __init__(self, name, *args):
|
||||||
|
self.name = name
|
||||||
|
self.args = args
|
||||||
|
|
||||||
|
def generate(self):
|
||||||
|
return '{} {}'.format(self.name, ' '.join(self.args))
|
||||||
|
|
||||||
|
class Function:
|
||||||
|
def __init__(self, name, exported=True):
|
||||||
|
self.name = name
|
||||||
|
self.exported = exported # TODO: Use __all__!
|
||||||
|
self.statements = []
|
||||||
|
|
||||||
|
def generate(self):
|
||||||
|
return '(func {}\n {})'.format(
|
||||||
|
('(export "{}")' if self.exported else '${}').format(self.name),
|
||||||
|
'\n '.join(x.generate() for x in self.statements),
|
||||||
|
)
|
||||||
|
|
||||||
|
class Visitor(ast.NodeVisitor):
|
||||||
|
def __init__(self):
|
||||||
|
self._stack = []
|
||||||
|
self.imports = []
|
||||||
|
self.functions = []
|
||||||
|
|
||||||
|
def visit_ImportFrom(self, node):
|
||||||
|
for alias in node.names:
|
||||||
|
self.imports.append(Import(
|
||||||
|
node.module,
|
||||||
|
alias.name,
|
||||||
|
alias.asname,
|
||||||
|
))
|
||||||
|
|
||||||
|
def visit_FunctionDef(self, node):
|
||||||
|
func = Function(
|
||||||
|
node.name,
|
||||||
|
)
|
||||||
|
|
||||||
|
self._stack.append(func)
|
||||||
|
self.generic_visit(node)
|
||||||
|
self._stack.pop()
|
||||||
|
|
||||||
|
self.functions.append(func)
|
||||||
|
|
||||||
|
def visit_Expr(self, node):
|
||||||
|
self.generic_visit(node)
|
||||||
|
|
||||||
|
def visit_Call(self, node):
|
||||||
|
self.generic_visit(node)
|
||||||
|
|
||||||
|
func = self._stack[-1]
|
||||||
|
func.statements.append(
|
||||||
|
Statement('call', '$' + node.func.id)
|
||||||
|
)
|
||||||
|
|
||||||
|
def visit_BinOp(self, node):
|
||||||
|
self.generic_visit(node)
|
||||||
|
|
||||||
|
func = self._stack[-1]
|
||||||
|
|
||||||
|
if 'Add' == node.op.__class__.__name__:
|
||||||
|
func.statements.append(
|
||||||
|
Statement('i32.add')
|
||||||
|
)
|
||||||
|
elif 'Mult' == node.op.__class__.__name__:
|
||||||
|
func.statements.append(
|
||||||
|
Statement('i32.mul')
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
err(node.op)
|
||||||
|
|
||||||
|
def visit_Constant(self, node):
|
||||||
|
if not self._stack:
|
||||||
|
# Constant outside of any function
|
||||||
|
imp = self.imports[-1]
|
||||||
|
prefix = imp.name + '('
|
||||||
|
val = node.value.strip()
|
||||||
|
|
||||||
|
if val.startswith(prefix) and val.endswith(')'):
|
||||||
|
imp.params = val[len(prefix):-1].split(',')
|
||||||
|
else:
|
||||||
|
func = self._stack[-1]
|
||||||
|
if isinstance(node.value, int):
|
||||||
|
func.statements.append(
|
||||||
|
Statement('i32.const', str(node.value))
|
||||||
|
)
|
||||||
|
|
||||||
|
self.generic_visit(node)
|
||||||
|
|
||||||
|
def generate(self):
|
||||||
|
return '(module\n {}\n {})'.format(
|
||||||
|
'\n '.join(x.generate() for x in self.imports),
|
||||||
|
'\n '.join(x.generate() for x in self.functions),
|
||||||
|
)
|
||||||
|
|
||||||
|
def err(msg: str) -> None:
|
||||||
|
sys.stderr.write('{}\n'.format(msg))
|
||||||
|
|
||||||
|
def main(source: str) -> int:
|
||||||
|
with open(source, 'r') as fil:
|
||||||
|
code = fil.read()
|
||||||
|
|
||||||
|
res = ast.parse(code, source)
|
||||||
|
|
||||||
|
visitor = Visitor()
|
||||||
|
visitor.visit(res)
|
||||||
|
|
||||||
|
print(visitor.generate())
|
||||||
|
|
||||||
|
return 0
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
sys.exit(main(*sys.argv[1:]))
|
||||||
20
func.html
Normal file
20
func.html
Normal file
@ -0,0 +1,20 @@
|
|||||||
|
<!doctype html>
|
||||||
|
|
||||||
|
<html>
|
||||||
|
|
||||||
|
<head>
|
||||||
|
<meta charset="utf-8">
|
||||||
|
<title>Simple add example</title>
|
||||||
|
</head>
|
||||||
|
|
||||||
|
<body>
|
||||||
|
<script>
|
||||||
|
|
||||||
|
WebAssembly.instantiateStreaming(fetch('func.wasm'))
|
||||||
|
.then(obj => {
|
||||||
|
console.log(obj.instance.exports.add(1, 2)); // "3"
|
||||||
|
});
|
||||||
|
</script>
|
||||||
|
</body>
|
||||||
|
|
||||||
|
</html>
|
||||||
29
log.html
Normal file
29
log.html
Normal file
@ -0,0 +1,29 @@
|
|||||||
|
<!doctype html>
|
||||||
|
|
||||||
|
<html>
|
||||||
|
|
||||||
|
<head>
|
||||||
|
<meta charset="utf-8">
|
||||||
|
<title>Simple log example</title>
|
||||||
|
</head>
|
||||||
|
|
||||||
|
<body>
|
||||||
|
<script>
|
||||||
|
|
||||||
|
var importObject = {
|
||||||
|
console: {
|
||||||
|
log: function(arg) {
|
||||||
|
console.log(arg);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
WebAssembly.instantiateStreaming(fetch('log.wasm'), importObject)
|
||||||
|
.then(obj => {
|
||||||
|
obj.instance.exports.logIt();
|
||||||
|
});
|
||||||
|
|
||||||
|
</script>
|
||||||
|
</body>
|
||||||
|
|
||||||
|
</html>
|
||||||
Loading…
x
Reference in New Issue
Block a user