Implement full mtval (kuznia-rdzeni/coreblocks#712)

kuznia-rdzeni · Oct 15, 2024 · c057b7d · c057b7d
1 parent 40e20a9
commit c057b7d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/transactron/utils/transactron_helpers.py b/transactron/utils/transactron_helpers.py
@@ -19,6 +19,8 @@
     "mock_def_helper",
     "get_src_loc",
     "from_method_layout",
+    "make_layout",
+    "extend_layout",
 ]
 
 T = TypeVar("T")
@@ -147,6 +149,10 @@ def make_layout(*fields: LayoutListField) -> StructLayout:
     return from_method_layout(fields)
 
 
+def extend_layout(layout: StructLayout, *fields: LayoutListField) -> StructLayout:
+    return StructLayout(layout.members | from_method_layout(fields).members)
+
+
 def from_method_layout(layout: MethodLayout) -> StructLayout:
     if isinstance(layout, StructLayout):
         return layout