Merge pull request #304 from OpShin/feat/constant_fields

Use constant nth field access where possible
OpShin · Jan 9, 2024 · 20aa8e2 · 20aa8e2
2 parents 54b37d2 + 687379e
commit 20aa8e2
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/opshin/types.py b/opshin/types.py
@@ -429,9 +429,9 @@ def attribute(self, attr: str) -> plt.AST:
             return OLambda(
                 ["self"],
                 transform_ext_params_map(attr_typ)(
-                    plt.NthField(
+                    plt.ConstantNthField(
                         OVar("self"),
-                        plt.Integer(pos),
+                        pos,
                     ),
                 ),
             )
@@ -509,7 +509,7 @@ def stringify(self, recursive: bool = False) -> plt.AST:
                     plt.Apply(
                         field_type.stringify(recursive=True),
                         transform_ext_params_map(field_type)(
-                            plt.NthField(OVar("self"), plt.Integer(pos))
+                            plt.ConstantNthField(OVar("self"), pos)
                         ),
                     ),
                     map_fields,
@@ -520,7 +520,7 @@ def stringify(self, recursive: bool = False) -> plt.AST:
                 plt.Apply(
                     self.record.fields[0][1].stringify(recursive=True),
                     transform_ext_params_map(self.record.fields[0][1])(
-                        plt.NthField(OVar("self"), plt.Integer(pos))
+                        plt.ConstantNthField(OVar("self"), pos)
                     ),
                 ),
                 map_fields,