Started on our own AST
This will help with optimizing code and generating WASM code
This commit is contained in:
Johan B.W. de Vries
parent
efba4e0daa
commit
e7b72b6a6b
713
py2wasm/ourlang.py
Normal file
713
py2wasm/ourlang.py
Normal file
@ -0,0 +1,713 @@
|
|||||||
|
"""
|
||||||
|
Contains the syntax tree for ourlang
|
||||||
|
"""
|
||||||
|
from typing import Any, Dict, List, NoReturn, Union, Tuple
|
||||||
|
|
||||||
|
import ast
|
||||||
|
|
||||||
|
class OurType:
|
||||||
|
"""
|
||||||
|
Type base class
|
||||||
|
"""
|
||||||
|
__slots__ = ()
|
||||||
|
|
||||||
|
def render(self) -> str:
|
||||||
|
"""
|
||||||
|
Renders the type back to source code format
|
||||||
|
|
||||||
|
This'll look like Python code.
|
||||||
|
"""
|
||||||
|
raise NotImplementedError(self, 'render')
|
||||||
|
|
||||||
|
def alloc_size(self) -> int:
|
||||||
|
"""
|
||||||
|
When allocating this type in memory, how many bytes do we need to reserve?
|
||||||
|
"""
|
||||||
|
raise NotImplementedError(self, 'alloc_size')
|
||||||
|
|
||||||
|
class OurTypeNone(OurType):
|
||||||
|
"""
|
||||||
|
The None (or Void) type
|
||||||
|
"""
|
||||||
|
__slots__ = ()
|
||||||
|
|
||||||
|
def render(self) -> str:
|
||||||
|
return 'None'
|
||||||
|
|
||||||
|
class OurTypeInt32(OurType):
|
||||||
|
"""
|
||||||
|
The Integer type, signed and 32 bits wide
|
||||||
|
"""
|
||||||
|
__slots__ = ()
|
||||||
|
|
||||||
|
def render(self) -> str:
|
||||||
|
return 'i32'
|
||||||
|
|
||||||
|
def alloc_size(self) -> int:
|
||||||
|
return 4
|
||||||
|
|
||||||
|
class OurTypeInt64(OurType):
|
||||||
|
"""
|
||||||
|
The Integer type, signed and 64 bits wide
|
||||||
|
"""
|
||||||
|
__slots__ = ()
|
||||||
|
|
||||||
|
def render(self) -> str:
|
||||||
|
return 'i64'
|
||||||
|
|
||||||
|
class OurTypeFloat32(OurType):
|
||||||
|
"""
|
||||||
|
The Float type, 32 bits wide
|
||||||
|
"""
|
||||||
|
__slots__ = ()
|
||||||
|
|
||||||
|
def render(self) -> str:
|
||||||
|
return 'f32'
|
||||||
|
|
||||||
|
class OurTypeFloat64(OurType):
|
||||||
|
"""
|
||||||
|
The Float type, 64 bits wide
|
||||||
|
"""
|
||||||
|
__slots__ = ()
|
||||||
|
|
||||||
|
def render(self) -> str:
|
||||||
|
return 'f64'
|
||||||
|
|
||||||
|
class Expression:
|
||||||
|
"""
|
||||||
|
An expression within a statement
|
||||||
|
"""
|
||||||
|
__slots__ = ()
|
||||||
|
|
||||||
|
def render(self) -> str:
|
||||||
|
"""
|
||||||
|
Renders the expression back to source code format
|
||||||
|
|
||||||
|
This'll look like Python code.
|
||||||
|
"""
|
||||||
|
raise NotImplementedError(self, 'render')
|
||||||
|
|
||||||
|
class Constant(Expression):
|
||||||
|
"""
|
||||||
|
An constant value expression within a statement
|
||||||
|
"""
|
||||||
|
__slots__ = ('type', )
|
||||||
|
|
||||||
|
type: OurType
|
||||||
|
|
||||||
|
def __init__(self, type_: OurType) -> None:
|
||||||
|
self.type = type_
|
||||||
|
|
||||||
|
class ConstantInt32(Constant):
|
||||||
|
"""
|
||||||
|
An Int32 constant value expression within a statement
|
||||||
|
"""
|
||||||
|
__slots__ = ('value', )
|
||||||
|
|
||||||
|
value: int
|
||||||
|
|
||||||
|
def __init__(self, type_: OurTypeInt32, value: int) -> None:
|
||||||
|
super().__init__(type_)
|
||||||
|
self.value = value
|
||||||
|
|
||||||
|
def render(self) -> str:
|
||||||
|
return str(self.value)
|
||||||
|
|
||||||
|
class ConstantInt64(Constant):
|
||||||
|
"""
|
||||||
|
An Int64 constant value expression within a statement
|
||||||
|
"""
|
||||||
|
__slots__ = ('value', )
|
||||||
|
|
||||||
|
value: int
|
||||||
|
|
||||||
|
def __init__(self, type_: OurTypeInt64, value: int) -> None:
|
||||||
|
super().__init__(type_)
|
||||||
|
self.value = value
|
||||||
|
|
||||||
|
def render(self) -> str:
|
||||||
|
return str(self.value)
|
||||||
|
|
||||||
|
class ConstantFloat32(Constant):
|
||||||
|
"""
|
||||||
|
An Float32 constant value expression within a statement
|
||||||
|
"""
|
||||||
|
__slots__ = ('value', )
|
||||||
|
|
||||||
|
value: float
|
||||||
|
|
||||||
|
def __init__(self, type_: OurTypeFloat32, value: float) -> None:
|
||||||
|
super().__init__(type_)
|
||||||
|
self.value = value
|
||||||
|
|
||||||
|
def render(self) -> str:
|
||||||
|
return str(self.value)
|
||||||
|
|
||||||
|
class ConstantFloat64(Constant):
|
||||||
|
"""
|
||||||
|
An Float64 constant value expression within a statement
|
||||||
|
"""
|
||||||
|
__slots__ = ('value', )
|
||||||
|
|
||||||
|
value: float
|
||||||
|
|
||||||
|
def __init__(self, type_: OurTypeFloat64, value: float) -> None:
|
||||||
|
super().__init__(type_)
|
||||||
|
self.value = value
|
||||||
|
|
||||||
|
def render(self) -> str:
|
||||||
|
return str(self.value)
|
||||||
|
|
||||||
|
class VariableReference(Expression):
|
||||||
|
"""
|
||||||
|
An variable reference expression within a statement
|
||||||
|
"""
|
||||||
|
__slots__ = ('type', 'name', )
|
||||||
|
|
||||||
|
type: OurType
|
||||||
|
name: str
|
||||||
|
|
||||||
|
def __init__(self, type_: OurType, name: str) -> None:
|
||||||
|
self.type = type_
|
||||||
|
self.name = name
|
||||||
|
|
||||||
|
def render(self) -> str:
|
||||||
|
return str(self.name)
|
||||||
|
|
||||||
|
class BinaryOp(Expression):
|
||||||
|
"""
|
||||||
|
A binary operator expression within a statement
|
||||||
|
"""
|
||||||
|
__slots__ = ('operator', 'left', 'right', )
|
||||||
|
|
||||||
|
operator: str
|
||||||
|
left: Expression
|
||||||
|
right: Expression
|
||||||
|
|
||||||
|
def __init__(self, operator: str, left: Expression, right: Expression) -> None:
|
||||||
|
self.operator = operator
|
||||||
|
self.left = left
|
||||||
|
self.right = right
|
||||||
|
|
||||||
|
def render(self) -> str:
|
||||||
|
return f'{self.left.render()} {self.operator} {self.right.render()}'
|
||||||
|
|
||||||
|
class UnaryOp(Expression):
|
||||||
|
"""
|
||||||
|
A unary operator expression within a statement
|
||||||
|
"""
|
||||||
|
__slots__ = ('operator', 'right', )
|
||||||
|
|
||||||
|
operator: str
|
||||||
|
right: Expression
|
||||||
|
|
||||||
|
def __init__(self, operator: str, right: Expression) -> None:
|
||||||
|
self.operator = operator
|
||||||
|
self.right = right
|
||||||
|
|
||||||
|
def render(self) -> str:
|
||||||
|
return f'{self.operator}{self.right.render()}'
|
||||||
|
|
||||||
|
class FunctionCall(Expression):
|
||||||
|
"""
|
||||||
|
A function call expression within a statement
|
||||||
|
"""
|
||||||
|
__slots__ = ('function', 'arguments', )
|
||||||
|
|
||||||
|
function: 'Function'
|
||||||
|
arguments: List[Expression]
|
||||||
|
|
||||||
|
def __init__(self, function: 'Function') -> None:
|
||||||
|
self.function = function
|
||||||
|
self.arguments = []
|
||||||
|
|
||||||
|
def render(self) -> str:
|
||||||
|
args = ', '.join(
|
||||||
|
arg.render()
|
||||||
|
for arg in self.arguments
|
||||||
|
)
|
||||||
|
|
||||||
|
return f'{self.function.name}({args})'
|
||||||
|
|
||||||
|
class Statement:
|
||||||
|
"""
|
||||||
|
A statement within a function
|
||||||
|
"""
|
||||||
|
__slots__ = ()
|
||||||
|
|
||||||
|
def render(self) -> List[str]:
|
||||||
|
"""
|
||||||
|
Renders the type back to source code format
|
||||||
|
|
||||||
|
This'll look like Python code.
|
||||||
|
"""
|
||||||
|
raise NotImplementedError(self, 'render')
|
||||||
|
|
||||||
|
class StatementReturn(Statement):
|
||||||
|
"""
|
||||||
|
A return statement within a function
|
||||||
|
"""
|
||||||
|
__slots__ = ('value', )
|
||||||
|
|
||||||
|
def __init__(self, value: Expression) -> None:
|
||||||
|
self.value = value
|
||||||
|
|
||||||
|
def render(self) -> List[str]:
|
||||||
|
"""
|
||||||
|
Renders the type back to source code format
|
||||||
|
|
||||||
|
This'll look like Python code.
|
||||||
|
"""
|
||||||
|
return [f'return {self.value.render()}']
|
||||||
|
|
||||||
|
class StatementIf(Statement):
|
||||||
|
"""
|
||||||
|
An if statement within a function
|
||||||
|
"""
|
||||||
|
__slots__ = ('test', 'statements', 'else_statements', )
|
||||||
|
|
||||||
|
test: Expression
|
||||||
|
statements: List[Statement]
|
||||||
|
else_statements: List[Statement]
|
||||||
|
|
||||||
|
def __init__(self, test: Expression) -> None:
|
||||||
|
self.test = test
|
||||||
|
self.statements = []
|
||||||
|
self.else_statements = []
|
||||||
|
|
||||||
|
def render(self) -> List[str]:
|
||||||
|
"""
|
||||||
|
Renders the type back to source code format
|
||||||
|
|
||||||
|
This'll look like Python code.
|
||||||
|
"""
|
||||||
|
result = [f'if {self.test.render()}:']
|
||||||
|
|
||||||
|
for stmt in self.statements:
|
||||||
|
result.extend(
|
||||||
|
f' {line}' if line else ''
|
||||||
|
for line in stmt.render()
|
||||||
|
)
|
||||||
|
|
||||||
|
result.append('')
|
||||||
|
|
||||||
|
return result
|
||||||
|
|
||||||
|
class StatementPass(Statement):
|
||||||
|
"""
|
||||||
|
A pass statement
|
||||||
|
"""
|
||||||
|
__slots__ = ()
|
||||||
|
|
||||||
|
def render(self) -> List[str]:
|
||||||
|
return ['pass']
|
||||||
|
|
||||||
|
class Function:
|
||||||
|
"""
|
||||||
|
A function processes input and produces output
|
||||||
|
"""
|
||||||
|
__slots__ = ('name', 'lineno', 'exported', 'statements', 'returns', 'posonlyargs', )
|
||||||
|
|
||||||
|
name: str
|
||||||
|
lineno: int
|
||||||
|
exported: bool
|
||||||
|
statements: List[Statement]
|
||||||
|
returns: OurType
|
||||||
|
posonlyargs: List[Tuple[str, OurType]]
|
||||||
|
|
||||||
|
def __init__(self, name: str, lineno: int) -> None:
|
||||||
|
self.name = name
|
||||||
|
self.lineno = lineno
|
||||||
|
self.exported = False
|
||||||
|
self.statements = []
|
||||||
|
self.returns = OurTypeNone()
|
||||||
|
self.posonlyargs = []
|
||||||
|
|
||||||
|
def render(self) -> str:
|
||||||
|
"""
|
||||||
|
Renders the function back to source code format
|
||||||
|
|
||||||
|
This'll look like Python code.
|
||||||
|
"""
|
||||||
|
result = ''
|
||||||
|
if self.exported:
|
||||||
|
result += '@exported\n'
|
||||||
|
|
||||||
|
args = ', '.join(
|
||||||
|
f'{x}: {y.render()}'
|
||||||
|
for x, y in self.posonlyargs
|
||||||
|
)
|
||||||
|
|
||||||
|
result += f'def {self.name}({args}) -> {self.returns.render()}:\n'
|
||||||
|
for stmt in self.statements:
|
||||||
|
for line in stmt.render():
|
||||||
|
result += f' {line}\n' if line else '\n'
|
||||||
|
return result
|
||||||
|
|
||||||
|
class StructMember:
|
||||||
|
"""
|
||||||
|
Represents a struct member
|
||||||
|
"""
|
||||||
|
def __init__(self, name: str, type_: OurType, offset: int) -> None:
|
||||||
|
self.name = name
|
||||||
|
self.type = type_
|
||||||
|
self.offset = offset
|
||||||
|
|
||||||
|
class Struct:
|
||||||
|
"""
|
||||||
|
A struct has named properties
|
||||||
|
"""
|
||||||
|
__slots__ = ('name', 'lineno', 'members', )
|
||||||
|
|
||||||
|
name: str
|
||||||
|
lineno: int
|
||||||
|
members: List[StructMember]
|
||||||
|
|
||||||
|
def __init__(self, name: str, lineno: int) -> None:
|
||||||
|
self.name = name
|
||||||
|
self.lineno = lineno
|
||||||
|
self.members = []
|
||||||
|
|
||||||
|
def render(self) -> str:
|
||||||
|
"""
|
||||||
|
Renders the function back to source code format
|
||||||
|
|
||||||
|
This'll look like Python code.
|
||||||
|
"""
|
||||||
|
return '?'
|
||||||
|
|
||||||
|
class Module:
|
||||||
|
"""
|
||||||
|
A module is a file and consists of functions
|
||||||
|
"""
|
||||||
|
__slots__ = ('types', 'functions', 'structs', )
|
||||||
|
|
||||||
|
types: Dict[str, OurType]
|
||||||
|
functions: Dict[str, Function]
|
||||||
|
structs: Dict[str, Struct]
|
||||||
|
|
||||||
|
def __init__(self) -> None:
|
||||||
|
self.types = {
|
||||||
|
'i32': OurTypeInt32(),
|
||||||
|
'i64': OurTypeInt64(),
|
||||||
|
'f32': OurTypeFloat32(),
|
||||||
|
'f64': OurTypeFloat64(),
|
||||||
|
}
|
||||||
|
self.functions = {}
|
||||||
|
self.structs = {}
|
||||||
|
|
||||||
|
def render(self) -> str:
|
||||||
|
"""
|
||||||
|
Renders the module back to source code format
|
||||||
|
|
||||||
|
This'll look like Python code.
|
||||||
|
"""
|
||||||
|
result = ''
|
||||||
|
for function in self.functions.values():
|
||||||
|
if result:
|
||||||
|
result += '\n'
|
||||||
|
result += function.render()
|
||||||
|
return result
|
||||||
|
|
||||||
|
class StaticError(Exception):
|
||||||
|
"""
|
||||||
|
An error found during static analysis
|
||||||
|
"""
|
||||||
|
|
||||||
|
OurLocals = Dict[str, OurType]
|
||||||
|
|
||||||
|
class OurVisitor:
|
||||||
|
"""
|
||||||
|
Class to visit a Python syntax tree and create an ourlang syntax tree
|
||||||
|
"""
|
||||||
|
|
||||||
|
# pylint: disable=C0103,C0116,C0301,R0201,R0912
|
||||||
|
|
||||||
|
def __init__(self) -> None:
|
||||||
|
pass
|
||||||
|
|
||||||
|
def visit_Module(self, node: ast.Module) -> Module:
|
||||||
|
module = Module()
|
||||||
|
|
||||||
|
_not_implemented(not node.type_ignores, 'Module.type_ignores')
|
||||||
|
|
||||||
|
for stmt in node.body:
|
||||||
|
res = self.visit_Module_stmt(module, stmt)
|
||||||
|
|
||||||
|
if isinstance(res, Function):
|
||||||
|
if res.name in module.functions:
|
||||||
|
raise StaticError(
|
||||||
|
f'{res.name} already defined on line {module.functions[res.name].lineno}'
|
||||||
|
)
|
||||||
|
|
||||||
|
module.functions[res.name] = res
|
||||||
|
|
||||||
|
if isinstance(res, Struct):
|
||||||
|
if res.name in module.structs:
|
||||||
|
raise StaticError(
|
||||||
|
f'{res.name} already defined on line {module.structs[res.name].lineno}'
|
||||||
|
)
|
||||||
|
|
||||||
|
module.structs[res.name] = res
|
||||||
|
|
||||||
|
return module
|
||||||
|
|
||||||
|
def visit_Module_stmt(self, module: Module, node: ast.stmt) -> Union[Function, Struct]:
|
||||||
|
if isinstance(node, ast.FunctionDef):
|
||||||
|
return self.visit_Module_FunctionDef(module, node)
|
||||||
|
|
||||||
|
if isinstance(node, ast.ClassDef):
|
||||||
|
return self.visit_Module_ClassDef(module, node)
|
||||||
|
|
||||||
|
raise NotImplementedError(f'{node} on Module')
|
||||||
|
|
||||||
|
def visit_Module_FunctionDef(self, module: Module, node: ast.FunctionDef) -> Function:
|
||||||
|
function = Function(node.name, node.lineno)
|
||||||
|
|
||||||
|
_not_implemented(not node.args.posonlyargs, 'FunctionDef.args.posonlyargs')
|
||||||
|
|
||||||
|
for arg in node.args.args:
|
||||||
|
if not arg.annotation:
|
||||||
|
_raise_static_error(node, 'Type is required')
|
||||||
|
|
||||||
|
function.posonlyargs.append((
|
||||||
|
arg.arg,
|
||||||
|
self.visit_type(module, arg.annotation),
|
||||||
|
))
|
||||||
|
|
||||||
|
_not_implemented(not node.args.vararg, 'FunctionDef.args.vararg')
|
||||||
|
_not_implemented(not node.args.kwonlyargs, 'FunctionDef.args.kwonlyargs')
|
||||||
|
_not_implemented(not node.args.kw_defaults, 'FunctionDef.args.kw_defaults')
|
||||||
|
_not_implemented(not node.args.kwarg, 'FunctionDef.args.kwarg')
|
||||||
|
_not_implemented(not node.args.defaults, 'FunctionDef.args.defaults')
|
||||||
|
|
||||||
|
# Do stmts at the end so we have the return value
|
||||||
|
|
||||||
|
for decorator in node.decorator_list:
|
||||||
|
if not isinstance(decorator, ast.Name):
|
||||||
|
_raise_static_error(decorator, 'Function decorators must be string')
|
||||||
|
if not isinstance(decorator.ctx, ast.Load):
|
||||||
|
_raise_static_error(decorator, 'Must be load context')
|
||||||
|
_not_implemented(decorator.id != 'exports', 'Custom decorators')
|
||||||
|
function.exported = True
|
||||||
|
|
||||||
|
if node.returns:
|
||||||
|
function.returns = self.visit_type(module, node.returns)
|
||||||
|
|
||||||
|
_not_implemented(not node.type_comment, 'FunctionDef.type_comment')
|
||||||
|
|
||||||
|
# Deferred parsing
|
||||||
|
|
||||||
|
our_locals = dict(function.posonlyargs)
|
||||||
|
|
||||||
|
for stmt in node.body:
|
||||||
|
function.statements.append(
|
||||||
|
self.visit_Module_FunctionDef_stmt(module, function, our_locals, stmt)
|
||||||
|
)
|
||||||
|
|
||||||
|
return function
|
||||||
|
|
||||||
|
def visit_Module_ClassDef(self, module: Module, node: ast.ClassDef) -> Struct:
|
||||||
|
struct = Struct(node.name, node.lineno)
|
||||||
|
|
||||||
|
_not_implemented(not node.bases, 'ClassDef.bases')
|
||||||
|
_not_implemented(not node.keywords, 'ClassDef.keywords')
|
||||||
|
_not_implemented(not node.decorator_list, 'ClassDef.decorator_list')
|
||||||
|
|
||||||
|
offset = 0
|
||||||
|
|
||||||
|
for stmt in node.body:
|
||||||
|
if not isinstance(stmt, ast.AnnAssign):
|
||||||
|
raise NotImplementedError(f'Class with {stmt} nodes')
|
||||||
|
|
||||||
|
if not isinstance(stmt.target, ast.Name):
|
||||||
|
raise NotImplementedError('Class with default values')
|
||||||
|
|
||||||
|
if not stmt.value is None:
|
||||||
|
raise NotImplementedError('Class with default values')
|
||||||
|
|
||||||
|
if stmt.simple != 1:
|
||||||
|
raise NotImplementedError('Class with non-simple arguments')
|
||||||
|
|
||||||
|
member = StructMember(stmt.target.id, self.visit_type(module, stmt.annotation), offset)
|
||||||
|
|
||||||
|
struct.members.append(member)
|
||||||
|
offset += member.type.alloc_size()
|
||||||
|
|
||||||
|
return struct
|
||||||
|
|
||||||
|
def visit_Module_FunctionDef_stmt(self, module: Module, function: Function, our_locals: OurLocals, node: ast.stmt) -> Statement:
|
||||||
|
if isinstance(node, ast.Return):
|
||||||
|
if node.value is None:
|
||||||
|
# TODO: Implement methods without return values
|
||||||
|
_raise_static_error(node, 'Return must have an argument')
|
||||||
|
|
||||||
|
return StatementReturn(
|
||||||
|
self.visit_Module_FunctionDef_expr(module, function, our_locals, function.returns, node.value)
|
||||||
|
)
|
||||||
|
|
||||||
|
if isinstance(node, ast.If):
|
||||||
|
result = StatementIf(
|
||||||
|
self.visit_Module_FunctionDef_expr(module, function, our_locals, function.returns, node.test)
|
||||||
|
)
|
||||||
|
|
||||||
|
for stmt in node.body:
|
||||||
|
result.statements.append(
|
||||||
|
self.visit_Module_FunctionDef_stmt(module, function, our_locals, stmt)
|
||||||
|
)
|
||||||
|
|
||||||
|
for stmt in node.body:
|
||||||
|
result.else_statements.append(
|
||||||
|
self.visit_Module_FunctionDef_stmt(module, function, our_locals, stmt)
|
||||||
|
)
|
||||||
|
|
||||||
|
return result
|
||||||
|
|
||||||
|
raise NotImplementedError(f'{node} as stmt in FunctionDef')
|
||||||
|
|
||||||
|
def visit_Module_FunctionDef_expr(self, module: Module, function: Function, our_locals: OurLocals, exp_type: OurType, node: ast.expr) -> Expression:
|
||||||
|
if isinstance(node, ast.BinOp):
|
||||||
|
if isinstance(node.op, ast.Add):
|
||||||
|
operator = '+'
|
||||||
|
elif isinstance(node.op, ast.Sub):
|
||||||
|
operator = '-'
|
||||||
|
else:
|
||||||
|
raise NotImplementedError(f'Operator {node.op}')
|
||||||
|
|
||||||
|
# Assume the type doesn't change when descending into a binary operator
|
||||||
|
# e.g. you can do `"hello" * 3` with the code below (yet)
|
||||||
|
|
||||||
|
return BinaryOp(
|
||||||
|
operator,
|
||||||
|
self.visit_Module_FunctionDef_expr(module, function, our_locals, exp_type, node.left),
|
||||||
|
self.visit_Module_FunctionDef_expr(module, function, our_locals, exp_type, node.right),
|
||||||
|
)
|
||||||
|
|
||||||
|
if isinstance(node, ast.UnaryOp):
|
||||||
|
if isinstance(node.op, ast.UAdd):
|
||||||
|
operator = '+'
|
||||||
|
elif isinstance(node.op, ast.USub):
|
||||||
|
operator = '-'
|
||||||
|
else:
|
||||||
|
raise NotImplementedError(f'Operator {node.op}')
|
||||||
|
|
||||||
|
return UnaryOp(
|
||||||
|
operator,
|
||||||
|
self.visit_Module_FunctionDef_expr(module, function, our_locals, exp_type, node.operand),
|
||||||
|
)
|
||||||
|
|
||||||
|
if isinstance(node, ast.Compare):
|
||||||
|
if 1 < len(node.ops):
|
||||||
|
raise NotImplementedError('Multiple operators')
|
||||||
|
|
||||||
|
if isinstance(node.ops[0], ast.Gt):
|
||||||
|
operator = '>'
|
||||||
|
else:
|
||||||
|
raise NotImplementedError(f'Operator {node.ops}')
|
||||||
|
|
||||||
|
# Assume the type doesn't change when descending into a binary operator
|
||||||
|
# e.g. you can do `"hello" * 3` with the code below (yet)
|
||||||
|
|
||||||
|
return BinaryOp(
|
||||||
|
operator,
|
||||||
|
self.visit_Module_FunctionDef_expr(module, function, our_locals, exp_type, node.left),
|
||||||
|
self.visit_Module_FunctionDef_expr(module, function, our_locals, exp_type, node.comparators[0]),
|
||||||
|
)
|
||||||
|
|
||||||
|
if isinstance(node, ast.Call):
|
||||||
|
if node.keywords:
|
||||||
|
_raise_static_error(node, 'Keyword calling not supported') # Yet?
|
||||||
|
|
||||||
|
if not isinstance(node.func, ast.Name):
|
||||||
|
raise NotImplementedError(f'Calling methods that are not a name {node.func}')
|
||||||
|
if not isinstance(node.func.ctx, ast.Load):
|
||||||
|
_raise_static_error(node, 'Must be load context')
|
||||||
|
|
||||||
|
if node.func.id not in module.functions:
|
||||||
|
_raise_static_error(node, 'Call to undefined function')
|
||||||
|
|
||||||
|
func = module.functions[node.func.id]
|
||||||
|
if func.returns != exp_type:
|
||||||
|
_raise_static_error(node, f'Function does not return {exp_type.render()}')
|
||||||
|
|
||||||
|
result = FunctionCall(func)
|
||||||
|
result.arguments.extend(
|
||||||
|
self.visit_Module_FunctionDef_expr(module, function, our_locals, exp_type, expr)
|
||||||
|
for expr in node.args
|
||||||
|
)
|
||||||
|
return result
|
||||||
|
|
||||||
|
if isinstance(node, ast.Constant):
|
||||||
|
return self.visit_Module_FunctionDef_Constant(
|
||||||
|
module, function, exp_type, node,
|
||||||
|
)
|
||||||
|
|
||||||
|
if isinstance(node, ast.Name):
|
||||||
|
if not isinstance(node.ctx, ast.Load):
|
||||||
|
_raise_static_error(node, 'Must be load context')
|
||||||
|
|
||||||
|
if node.id in our_locals:
|
||||||
|
return VariableReference(our_locals[node.id], node.id)
|
||||||
|
|
||||||
|
raise NotImplementedError(f'{node} as expr in FunctionDef')
|
||||||
|
|
||||||
|
def visit_Module_FunctionDef_Constant(self, module: Module, function: Function, exp_type: OurType, node: ast.Constant) -> Expression:
|
||||||
|
del module
|
||||||
|
del function
|
||||||
|
|
||||||
|
_not_implemented(node.kind is None, 'Constant.kind')
|
||||||
|
|
||||||
|
if isinstance(exp_type, OurTypeInt32):
|
||||||
|
if not isinstance(node.value, int):
|
||||||
|
_raise_static_error(node, 'Expected integer value')
|
||||||
|
|
||||||
|
# FIXME: Range check
|
||||||
|
|
||||||
|
return ConstantInt32(exp_type, node.value)
|
||||||
|
|
||||||
|
if isinstance(exp_type, OurTypeInt64):
|
||||||
|
if not isinstance(node.value, int):
|
||||||
|
_raise_static_error(node, 'Expected integer value')
|
||||||
|
|
||||||
|
# FIXME: Range check
|
||||||
|
|
||||||
|
return ConstantInt64(exp_type, node.value)
|
||||||
|
|
||||||
|
if isinstance(exp_type, OurTypeFloat32):
|
||||||
|
if not isinstance(node.value, float):
|
||||||
|
_raise_static_error(node, 'Expected float value')
|
||||||
|
|
||||||
|
# FIXME: Range check
|
||||||
|
|
||||||
|
return ConstantFloat32(exp_type, node.value)
|
||||||
|
|
||||||
|
if isinstance(exp_type, OurTypeFloat64):
|
||||||
|
if not isinstance(node.value, float):
|
||||||
|
_raise_static_error(node, 'Expected float value')
|
||||||
|
|
||||||
|
# FIXME: Range check
|
||||||
|
|
||||||
|
return ConstantFloat64(exp_type, node.value)
|
||||||
|
|
||||||
|
raise NotImplementedError(f'{node} as const for type {exp_type.render()}')
|
||||||
|
|
||||||
|
def visit_type(self, module: Module, node: ast.expr) -> OurType:
|
||||||
|
if isinstance(node, ast.Name):
|
||||||
|
if not isinstance(node.ctx, ast.Load):
|
||||||
|
_raise_static_error(node, 'Must be load context')
|
||||||
|
|
||||||
|
if node.id not in module.types:
|
||||||
|
_raise_static_error(node, 'Unrecognized type')
|
||||||
|
|
||||||
|
return module.types[node.id]
|
||||||
|
|
||||||
|
raise NotImplementedError(f'{node} as type')
|
||||||
|
|
||||||
|
def _not_implemented(check: Any, msg: str) -> None:
|
||||||
|
if not check:
|
||||||
|
raise NotImplementedError(msg)
|
||||||
|
|
||||||
|
def _raise_static_error(node: Union[ast.mod, ast.stmt, ast.expr], msg: str) -> NoReturn:
|
||||||
|
raise StaticError(
|
||||||
|
f'Static error on line {node.lineno}: {msg}'
|
||||||
|
)
|
||||||
@ -4,7 +4,8 @@ Utility functions
|
|||||||
|
|
||||||
import ast
|
import ast
|
||||||
|
|
||||||
from py2wasm.python import Visitor
|
from .ourlang import OurVisitor, Module
|
||||||
|
from .python import Visitor
|
||||||
|
|
||||||
def process(source: str, input_name: str) -> str:
|
def process(source: str, input_name: str) -> str:
|
||||||
"""
|
"""
|
||||||
@ -16,3 +17,12 @@ def process(source: str, input_name: str) -> str:
|
|||||||
module = visitor.visit_Module(res)
|
module = visitor.visit_Module(res)
|
||||||
|
|
||||||
return module.generate()
|
return module.generate()
|
||||||
|
|
||||||
|
def our_process(source: str, input_name: str) -> Module:
|
||||||
|
"""
|
||||||
|
Processes the python code into web assembly code
|
||||||
|
"""
|
||||||
|
res = ast.parse(source, input_name)
|
||||||
|
|
||||||
|
our_visitor = OurVisitor()
|
||||||
|
return our_visitor.visit_Module(res)
|
||||||
|
|||||||
@ -0,0 +1,3 @@
|
|||||||
|
import pytest
|
||||||
|
|
||||||
|
pytest.register_assert_rewrite('tests.integration.helpers')
|
||||||
Loading…
x
Reference in New Issue
Block a user