interpreters and AST nodes for tuple chapter

Author: jsiek

interp_Ctup.py

@@ -0,0 +1,38 @@
+from ast import *
+from interp_Cif import InterpCif
+from utils import *
+
+class InterpCtup(InterpCif):
+
+  def interp_exp(self, e, env):
+    match e:
+      case Tuple(es, Load()):
+        return tuple([self.interp_exp(e, env) for e in es])
+      case Subscript(tup, index, Load()):
+        t = self.interp_exp(tup, env)
+        n = self.interp_exp(index, env)
+        return t[n]
+      case Allocate(length, typ):
+        array = [None] * length
+        return array
+      case Begin(ss, e):
+        self.interp_stmts(ss, env)
+        return self.interp_exp(e, env)
+      case GlobalValue(name):
+        return 0 # bogus
+      case _:
+        return super().interp_exp(e, env)
+
+  def interp_stmts(self, ss, env):
+    if len(ss) == 0:
+      return
+    match ss[0]:
+      case Collect(size):
+        return self.interp_stmts(ss[1:], env)
+      case Assign([Subscript(tup, index)], value):
+        tup = self.interp_exp(tup, env)
+        index = self.interp_exp(index, env)
+        tup[index] = self.interp_exp(value, env)
+        return self.interp_stmts(ss[1:], env)
+      case _:
+        return super().interp_stmts(ss, env)

interp_Ltup.py

@@ -0,0 +1,53 @@
+from ast import *
+from interp_Lwhile import InterpLwhile
+from utils import *
+
+class InterpLtup(InterpLwhile):
+
+  def interp_cmp(self, cmp):
+    match cmp:
+      case Is():
+        return lambda x, y: x is y
+      case _:
+        return super().interp_cmp(cmp)      
+    
+  def interp_exp(self, e, env):
+    match e:
+      case Tuple(es, Load()):
+        return tuple([self.interp_exp(e, env) for e in es])
+      case Subscript(tup, index, Load()):
+        t = self.interp_exp(tup, env)
+        n = self.interp_exp(index, env)
+        return t[n]
+      case Allocate(length, typ):
+        array = [None] * length
+        return array
+      case Begin(ss, e):
+        self.interp_stmts(ss, env)
+        return self.interp_exp(e, env)
+      case GlobalValue(name):
+        return 0 # ???
+      case _:
+        return super().interp_exp(e, env)
+
+  def interp_stmts(self, ss, env):
+    if len(ss) == 0:
+      return
+    match ss[0]:
+      case Collect(size):
+        return self.interp_stmts(ss[1:], env)
+      case Assign([Subscript(tup, index)], value):
+        tup = self.interp_exp(tup, env)
+        index = self.interp_exp(index, env)
+        tup[index] = self.interp_exp(value, env)
+        return self.interp_stmts(ss[1:], env)
+      case _:
+        return super().interp_stmts(ss, env)
+    
+if __name__ == "__main__":
+  t1 = Tuple([Constant(1), Constant(2)], Load())
+  t1_at_0 = Subscript(t1, Constant(0), Load())
+  pr = Expr(Call(Name('print'), [t1_at_0]))
+  p = Module([pr])
+  InterpLtup().interp(p)
+    

interp_x86/__pycache__/convert_x86.cpython-310.pyc

@@ -4,7 +4,7 @@
 from lark import Tree
 from ast import Name, Constant
 from x86_ast import *
-from utils import label_name
+from utils import label_name, GlobalValue
 
 def convert_int(value):
     if value >= 0:
@@ -24,6 +24,8 @@ def convert_arg(arg):
             return Tree('mem_a', [convert_int(offset), reg])
         case ByteReg(id):
             return Tree('reg_a', [id])
+        case GlobalValue(id):
+            return Tree('global_val_a', [id, 'rip'])
         case _:
             raise Exception('convert_arg: unhandled ' + repr(arg))
 

interp_x86/__pycache__/eval_x86.cpython-310.pyc

@@ -275,11 +275,13 @@ def eval_instrs(self, instrs, blocks, output):
                         raise Exception('jump to invalid target ' + target)
                     return # after jumping, toss continuation
 
-            elif instr.data in ['sete', 'setl', 'setle', 'setg', 'setge']:
+            elif instr.data in ['sete', 'setne', 'setl', 'setle', 'setg', 'setge']:
                 a1 = instr.children[0]
 
                 if instr.data == 'sete' and self.registers['EFLAGS'] == 'e':
                     self.store_arg(a1, 1)
+                elif instr.data == 'setne' and self.registers['EFLAGS'] != 'e':
+                    self.store_arg(a1, 1)
                 elif instr.data == 'setl' and self.registers['EFLAGS'] == 'l':
                     self.store_arg(a1, 1)
                 elif instr.data == 'setle' \