Cache macro logics

tm/macro.py

@@ -54,15 +54,15 @@ def make_block_macro(
         blocks: int,
         params: Params,
 ) -> MacroProg:
-    return MacroProg(comp, BlockLogic(blocks, *params))
+    return MacroProg(comp, make_block(blocks, params))
 
 
 def make_backsymbol_macro(
         comp: GetInstr,
         backsym: int,
         params: Params,
 ) -> MacroProg:
-    return MacroProg(comp, BacksymbolLogic(backsym, *params))
+    return MacroProg(comp, make_backsym(backsym, params))
 
 ########################################
 
@@ -377,3 +377,33 @@ def reconstruct_outputs(self, config: Config) -> Instr:
                      * self.backsymbols)))
             ,
         )
+
+
+BLOCKS: dict[
+    Params,
+    dict[int, BlockLogic]
+] = defaultdict(dict)
+
+
+BACKSYMS: dict[
+    Params,
+    dict[int, BacksymbolLogic]
+] = defaultdict(dict)
+
+
+def make_block(blocks: int, params: Params) -> BlockLogic:
+    if (cached := BLOCKS[params].get(blocks)) is not None:
+        return cached
+
+    block = BlockLogic(blocks, *params)
+    BLOCKS[params][blocks] = block
+    return block
+
+
+def make_backsym(backsyms: int, params: Params) -> BacksymbolLogic:
+    if (cached := BACKSYMS[params].get(backsyms)) is not None:
+        return cached
+
+    backsym = BacksymbolLogic(backsyms, *params)
+    BACKSYMS[params][backsyms] = backsym
+    return backsym