a14 af

[master.git] / ar / assignments / src / 4.py

diff --git a/ar/assignments/src/4.py b/ar/assignments/src/4.py
index e260264..51d5347 100644 (file)
--- a/ar/assignments/src/4.py
+++ b/ar/assignments/src/4.py
@@ -1,5 +1,7 @@
-iterations = 11
-numa = 8
+#!/usr/bin/env python3
+import sys
+iterations = int(sys.argv[1]) if len(sys.argv) > 1 else 11
+numa = int(sys.argv[2]) if len(sys.argv) > 2 else 7
 
 ##Print preamble
 print("(benchmark a4.smt")
@@ -52,8 +54,7 @@ for v in range(2, numa):
     for ov in [k for k in range(2, numa) if k != v]:
         print("(= i{0}a{1} i{0}a{2})".format(iterations-1, v, ov))
     print("))")
-
-
 print(")")
+
 ## Close the and,benchmark parenthesis
 print("))")