check

Author: jsiek

tests/var/add.golden

@@ -0,0 +1 @@
+42

tests/var/add.in

@@ -0,0 +1 @@
+

tests/var/add.py

@@ -0,0 +1 @@
+print(40 + 2)

tests/var/input.golden

@@ -0,0 +1 @@
+42

tests/var/input.in

@@ -0,0 +1 @@
+42

tests/var/input.py

@@ -0,0 +1 @@
+print(input_int())

tests/var/zero.golden

@@ -0,0 +1 @@
+0

tests/var/zero.in

@@ -0,0 +1 @@
+

tests/var/zero.py

@@ -0,0 +1 @@
+print(0)

type_check_Pvar.py

@@ -0,0 +1,54 @@
+from ast import *
+
+def check_type_equal(t1, t2, e):
+  if t1 != t2:
+    raise Exception('error: ' + repr(t1) + ' != ' + repr(t2) + ' in ' + repr(e))
+
+class TypeCheckPvar:
+          
+  def type_check_exp(self, e, env):
+    match e:
+      case BinOp(left, Add(), right):
+        l = self.type_check_exp(left, env)
+        check_type_equal(l, int, left)
+        r = self.type_check_exp(right, env)
+        check_type_equal(r, int, right)
+        return int
+      case UnaryOp(USub(), v):
+        t = self.type_check_exp(v, env)
+        check_type_equal(t, int, v)
+        return int
+      case Name(id):
+        return env[id]
+      case Constant(value) if isinstance(value, int):
+        return int
+      case Call(Name('input_int'), []):
+        return int
+      case _:
+        raise Exception('error in TypeCheckPvar.type_check_exp, unhandled ' + repr(e))
+
+  def type_check_stmts(self, ss, env):
+    if len(ss) == 0:
+      return
+    match ss[0]:
+      case Assign([lhs], value):
+        t = self.type_check_exp(value, env)
+        if lhs.id in env:
+          check_type_equal(env[lhs.id], t, value)
+        else:
+          env[lhs.id] = t
+        return self.type_check_stmts(ss[1:], env)
+      case Expr(Call(Name('print'), [arg])):
+        t = self.type_check_exp(arg, env)
+        check_type_equal(t, int, arg)
+        return self.type_check_stmts(ss[1:], env)
+      case Expr(value):
+        self.type_check_exp(value, env)
+        return self.type_check_stmts(ss[1:], env)
+      case _:
+        raise Exception('error in TypeCheckPvar.type_check_stmt, unhandled ' + repr(s))
+
+  def type_check_P(self, p):
+    match p:
+      case Module(body):
+        self.type_check_stmts(body, {})

x86_ast.py

@@ -0,0 +1,133 @@
+from typing import List
+
+class X86Program:
+    __match_args__ = ("body",)
+    def __init__(self, body):
+        self.body = body
+    def __repr__(self):
+        result = ''
+        if type(self.body) == dict:
+            for (l,ss) in self.body.items():
+                result += l + ':\n'
+                result += '\n'.join([repr(s) for s in ss]) + '\n\n'
+        else:
+            result = '\n'.join([repr(s) for s in self.body])
+        return result
+                
+class instr: ...
+class arg: ...
+class location(arg): ...
+    
+class Instr(instr):
+    instr: str
+    args: List[arg]
+    
+    __match_args__ = ("instr", "args")
+    def __init__(self, instr, args):
+        self.instr = instr
+        self.args = args
+    def source(self):
+        return self.args[0]
+    def target(self):
+        return self.args[-1]
+    def __repr__(self):
+        return self.instr + ' ' + ', '.join(repr(a) for a in self.args)
+    
+class Callq(instr):
+    __match_args__ = ("func", "num_args")
+    def __init__(self, func, num_args):
+        self.func = func
+        self.num_args = num_args
+    def __repr__(self):
+        return 'callq' + ' ' + self.func
+    
+class Variable(location):
+    __match_args__ = ("id",)
+    def __init__(self, id):
+        self.id = id
+    def __repr__(self):
+        return self.id
+    def __eq__(self, other):
+        if isinstance(other, Variable):
+            return self.id == other.id
+        else:
+            return False
+    def __hash__(self):
+        return hash(self.id)
+
+class Immediate(arg):
+    __match_args__ = ("value",)
+    def __init__(self, value):
+        self.value = value
+    def __repr__(self):
+        return repr(self.value)
+    def __eq__(self, other):
+        if isinstance(other, Immediate):
+            return self.value == other.value
+        else:
+            return False
+    def __hash__(self):
+        return hash(self.value)
+    
+class Reg(location):
+    __match_args__ = ("id",)
+    def __init__(self, id):
+        self.id = id
+    def __repr__(self):
+        return '%' + self.id
+    def __eq__(self, other):
+        if isinstance(other, Reg):
+            return self.id == other.id
+        else:
+            return False
+    def __hash__(self):
+        return hash(self.id)
+        
+class ByteReg(arg):
+    __match_args__ = ("id",)
+    def __init__(self, id):
+        self.id = id
+    def __repr__(self):
+        return '%' + self.id
+    def __eq__(self, other):
+        if isinstance(other, ByteReg):
+            return self.id == other.id
+        else:
+            return False
+    def __hash__(self):
+        return hash(self.id)
+        
+class Deref(arg):
+    __match_args__ = ("reg", "offset")
+    def __init__(self, reg, offset):
+        self.reg = reg
+        self.offset = offset
+    def __repr__(self):
+        return repr(self.offset) + '(%' + self.reg + ')'
+    def __eq__(self, other):
+        if isinstance(other, Deref):
+            return self.reg == other.reg and self.offset == other.offset
+        else:
+            return False
+    def __hash__(self):
+        return hash((self.reg, self.offset))
+
+class JumpIf(instr):
+    cc: str
+    label: str
+    
+    __match_args__ = ("cc", "label")
+    def __init__(self, cc, label):
+        self.cc = cc
+        self.label = label
+    def __repr__(self):
+        return 'j' + self.cc + ' ' + self.label
+
+class Jump(instr):
+    label: str
+    
+    __match_args__ = ("label",)
+    def __init__(self, label):
+        self.label = label
+    def __repr__(self):
+        return 'jmp ' + self.label