oppc/code: support subscript exprs

diff --git a/src/openpower/oppc/pc_code.py b/src/openpower/oppc/pc_code.py
index 67d4f42feba24e743f08b1fc10b88d9ca04f6a24..78d3ce55781b2c9e309b08581425af9da5a66f35 100644 (file)
--- a/src/openpower/oppc/pc_code.py
+++ b/src/openpower/oppc/pc_code.py
@@ -153,6 +153,21 @@ class CodeVisitor(pc_util.Visitor):
         yield node
         self[node].emit(f"ctx->XLEN")
 
+    @pc_util.Hook(pc_ast.SubscriptExpr)
+    def SubscriptExpr(self, node):
+        yield node
+        index = str(self[node.index])
+        subject = str(self[node.subject])
+        self[node].emit(f"oppc_subscript({subject}, {index})")
+
+    @pc_util.Hook(pc_ast.RangeSubscriptExpr)
+    def RangeSubscriptExpr(self, node):
+        yield node
+        start = str(self[node.start])
+        end = str(self[node.end])
+        subject = str(self[node.subject])
+        self[node].emit(f"oppc_range_subscript({subject}, {start}, {end})")
+
     @pc_util.Hook(pc_ast.Call.Name)
     def CallName(self, node):
         yield node