Update main of generator_honk.py to produce both keccak and starknet …

…honk flavors
keep-starknet-strange · Jan 10, 2025 · 0f5c48e · 0f5c48e
1 parent fa34622
commit 0f5c48e
Showing 1 changed file with 16 additions and 12 deletions.
diff --git a/hydra/garaga/starknet/honk_contract_generator/generator_honk.py b/hydra/garaga/starknet/honk_contract_generator/generator_honk.py
@@ -451,17 +451,21 @@ def gen_honk_verifier(
 
 if __name__ == "__main__":
 
-    VK_PATH = (
-        "hydra/garaga/starknet/honk_contract_generator/examples/vk_ultra_keccak.bin"
-    )
-    VK_LARGE_PATH = (
-        "hydra/garaga/starknet/honk_contract_generator/examples/vk_large.bin"
-    )
-    CONTRACTS_FOLDER = "src/contracts/"  # Do not change this
+    for system in [ProofSystem.UltraKeccakHonk, ProofSystem.UltraStarknetHonk]:
 
-    FOLDER_NAME = (
-        "noir_ultra_keccak_honk_example"  # '_curve_id' is appended in the end.
-    )
+        flavor = "keccak" if system == ProofSystem.UltraKeccakHonk else "starknet"
+
+        VK_PATH = (
+            "hydra/garaga/starknet/honk_contract_generator/examples/vk_ultra_keccak.bin"
+        )
+        VK_LARGE_PATH = (
+            "hydra/garaga/starknet/honk_contract_generator/examples/vk_large.bin"
+        )
+        CONTRACTS_FOLDER = "src/contracts/"  # Do not change this
+
+        FOLDER_NAME = (
+            f"noir_ultra_{flavor}_honk_example"  # '_curve_id' is appended in the end.
+        )
 
-    gen_honk_verifier(VK_PATH, CONTRACTS_FOLDER, FOLDER_NAME)
-    # gen_honk_verifier(VK_LARGE_PATH, CONTRACTS_FOLDER, FOLDER_NAME + "_large")
+        gen_honk_verifier(VK_PATH, CONTRACTS_FOLDER, FOLDER_NAME, system=system)
+        # gen_honk_verifier(VK_LARGE_PATH, CONTRACTS_FOLDER, FOLDER_NAME + "_large", system=system)