feat: add little optimization for O2

Author: boriel

src/arch/z80/visitor/translator.py

@@ -22,6 +22,7 @@
 from src.arch.z80.visitor.unary_op_translator import UnaryOpTranslator
 from src.symbols import sym as symbols
 from src.symbols.arrayaccess import SymbolARRAYACCESS
+from src.symbols.binary import SymbolBINARY
 from src.symbols.id_ import ref
 from src.symbols.type_ import Type
 
@@ -216,7 +217,7 @@ def visit_ARGUMENT(self, node):
             return
 
         # ByRef argument
-        if node.value.token != "ARRAYLOAD":
+        if node.value.token not in ("ARRAYLOAD", "ARRAYACCESS"):
             scope = node.value.scope
             if node.t[0] == "_":
                 t = optemps.new_t()
@@ -238,8 +239,18 @@ def visit_ARGUMENT(self, node):
             return
 
         # Must compute Address of @array(...)
-        node.value = SymbolARRAYACCESS.copy_from(node.value)
-        node.value = symbols.UNARY("ADDRESS", node.value, node.lineno, type_=self.TYPE(gl.PTR_TYPE))
+        if node.value.scope == SCOPE.global_ and self.O_LEVEL > 1:  # Calculate offset if global variable
+            node.value = SymbolBINARY.make_node(
+                "PLUS",
+                symbols.UNARY("ADDRESS", node.value.entry, node.value.lineno, type_=self.TYPE(gl.PTR_TYPE)),
+                symbols.NUMBER(node.value.offset, lineno=node.value.lineno, type_=self.TYPE(gl.PTR_TYPE)),
+                lineno=node.lineno,
+                func=lambda x, y: x + y,
+            )
+        else:
+            node.value = SymbolARRAYACCESS.copy_from(node.value)
+            node.value = symbols.UNARY("ADDRESS", node.value, node.lineno, type_=self.TYPE(gl.PTR_TYPE))
+
         yield node.value
 
     def visit_ARRAYLOAD(self, node):