made the tool compatible with newer versions of minizinc

hadipourh · Jul 4, 2024 · 19ff127 · 19ff127
1 parent 255d1b9
commit 19ff127
Show file tree

Hide file tree

Showing 7 changed files with 114 additions and 24 deletions.
diff --git a/ascon/impossible/distinguisher.py b/ascon/impossible/distinguisher.py
@@ -35,6 +35,19 @@
 from argparse import ArgumentParser, RawTextHelpFormatter
 from drawdistinguisher import DrawDL
 
+# Check if "OR Tools" appears in the output of "minizinc --solvers" command 
+import subprocess
+try:
+    output = subprocess.run(['minizinc', '--solvers'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    if "com.google.ortools.sat" in output.stdout.decode("utf-8"):
+        ortools_available = True
+        print("OR Tools is available")
+    else:
+        ortools_available = False
+        print("OR Tools is not available")
+except FileNotFoundError:
+    ortools_available = False
+
 
 class ID:
     DL_counter = 0
@@ -48,10 +61,10 @@ def __init__(self, param) -> None:
         self.cp_solver_name = param["solver"]
         self.num_of_threads = param["threads"]
         ##################################################
-        # Use this block if you install Or-Tools bundeled with MiniZinc
-        if self.cp_solver_name == "ortools":
-            self.cp_solver_name = "com.google.ortools.sat"
-        ##################################################
+        if ortools_available:
+            if self.cp_solver_name == "ortools":
+                self.cp_solver_name = "com.google.ortools.sat"
+        #################################################
         self.cp_solver = minizinc.Solver.lookup(self.cp_solver_name)
         self.time_limit = param["timelimit"]
         self.mzn_file_name = None

diff --git a/ascon/zero-correlation/distinguisher.py b/ascon/zero-correlation/distinguisher.py
@@ -34,6 +34,18 @@
 import datetime
 from argparse import ArgumentParser, RawTextHelpFormatter
 from drawdistinguisher import DrawDL
+# Check if "OR Tools" appears in the output of "minizinc --solvers" command 
+import subprocess
+try:
+    output = subprocess.run(['minizinc', '--solvers'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    if "com.google.ortools.sat" in output.stdout.decode("utf-8"):
+        ortools_available = True
+        print("OR Tools is available")
+    else:
+        ortools_available = False
+        print("OR Tools is not available")
+except FileNotFoundError:
+    ortools_available = False
 
 
 class ID:
@@ -48,10 +60,10 @@ def __init__(self, param) -> None:
         self.cp_solver_name = param["solver"]
         self.num_of_threads = param["threads"]
         ##################################################
-        # Use this block if you install Or-Tools bundeled with MiniZinc
-        if self.cp_solver_name == "ortools":
-            self.cp_solver_name = "com.google.ortools.sat"
-        ##################################################        
+        if ortools_available:
+            if self.cp_solver_name == "ortools":
+                self.cp_solver_name = "com.google.ortools.sat"
+        #################################################      
         self.cp_solver = minizinc.Solver.lookup(self.cp_solver_name)
         self.time_limit = param["timelimit"]
         self.mzn_file_name = None

diff --git a/forkskinny/impossible/attack.py b/forkskinny/impossible/attack.py
@@ -38,6 +38,19 @@
 from tweakeyschedule import *
 line_separator = "#"*55
 
+# Check if "OR Tools" appears in the output of "minizinc --solvers" command 
+import subprocess
+try:
+    output = subprocess.run(['minizinc', '--solvers'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    if "com.google.ortools.sat" in output.stdout.decode("utf-8"):
+        ortools_available = True
+        print("OR Tools is available")
+    else:
+        ortools_available = False
+        print("OR Tools is not available")
+except FileNotFoundError:
+    ortools_available = False
+
 class ID:
     ID_counter = 0
 
@@ -66,10 +79,10 @@ def __init__(self, params) -> None:
                                      'picat', 'scip', 'choco', 'ortools']
         assert(self.cp_solver_name in self.supported_cp_solvers)
         ##################################################
-        # Use this block if you install Or-Tools bundeled with MiniZinc
-        # if self.cp_solver_name == "ortools":
-        #     self.cp_solver_name = "com.google.ortools.sat"
-        ##################################################        
+        if ortools_available:
+            if self.cp_solver_name == "ortools":
+                self.cp_solver_name = "com.google.ortools.sat"
+        #################################################
         self.cp_solver = minizinc.Solver.lookup(self.cp_solver_name)
 
         if self.RB + self.RF == 0:

diff --git a/forkskinny/integral/attack.py b/forkskinny/integral/attack.py
@@ -35,6 +35,18 @@
 from pathlib import Path
 from drawattack import *
 line_separator = "#"*55
+# Check if "OR Tools" appears in the output of "minizinc --solvers" command 
+import subprocess
+try:
+    output = subprocess.run(['minizinc', '--solvers'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    if "com.google.ortools.sat" in output.stdout.decode("utf-8"):
+        ortools_available = True
+        print("OR Tools is available")
+    else:
+        ortools_available = False
+        print("OR Tools is not available")
+except FileNotFoundError:
+    ortools_available = False
 
 class IntegralAttack:
     Integral_counter = 0

diff --git a/present/impossible/distinguisher.py b/present/impossible/distinguisher.py
@@ -36,6 +36,19 @@
 from drawdistinguisher import DrawDL
 import itertools
 
+# Check if "OR Tools" appears in the output of "minizinc --solvers" command 
+import subprocess
+try:
+    output = subprocess.run(['minizinc', '--solvers'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    if "com.google.ortools.sat" in output.stdout.decode("utf-8"):
+        ortools_available = True
+        print("OR Tools is available")
+    else:
+        ortools_available = False
+        print("OR Tools is not available")
+except FileNotFoundError:
+    ortools_available = False
+
 
 class ID:
     DL_counter = 0
@@ -49,10 +62,10 @@ def __init__(self, param) -> None:
         self.cp_solver_name = param["solver"]
         self.num_of_threads = param["threads"]
         ##################################################
-        # Use this block if you install Or-Tools bundeled with MiniZinc
-        if self.cp_solver_name == "ortools":
-            self.cp_solver_name = "com.google.ortools.sat"
-        ##################################################        
+        if ortools_available:
+            if self.cp_solver_name == "ortools":
+                self.cp_solver_name = "com.google.ortools.sat"
+        #################################################      
         self.cp_solver = minizinc.Solver.lookup(self.cp_solver_name)
         self.time_limit = param["timelimit"]
         self.mzn_file_name = None

diff --git a/present/zero-correlation/distinguisher.py b/present/zero-correlation/distinguisher.py
@@ -36,6 +36,20 @@
 from drawdistinguisher import DrawDL
 import itertools
 
+# Check if "OR Tools" appears in the output of "minizinc --solvers" command 
+import subprocess
+try:
+    output = subprocess.run(['minizinc', '--solvers'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    if "com.google.ortools.sat" in output.stdout.decode("utf-8"):
+        ortools_available = True
+        print("OR Tools is available")
+    else:
+        ortools_available = False
+        print("OR Tools is not available")
+except FileNotFoundError:
+    ortools_available = False
+
+
 
 class ZC:
     DL_counter = 0
@@ -49,10 +63,10 @@ def __init__(self, param) -> None:
         self.cp_solver_name = param["solver"]
         self.num_of_threads = param["threads"]
         ##################################################
-        # Use this block if you install Or-Tools bundeled with MiniZinc
-        if self.cp_solver_name == "ortools":
-            self.cp_solver_name = "com.google.ortools.sat"
-        ##################################################        
+        if ortools_available:
+            if self.cp_solver_name == "ortools":
+                self.cp_solver_name = "com.google.ortools.sat"
+        #################################################     
         self.cp_solver = minizinc.Solver.lookup(self.cp_solver_name)
         self.time_limit = param["timelimit"]
         self.mzn_file_name = None

diff --git a/skinny/impossible/attack.py b/skinny/impossible/attack.py
@@ -38,6 +38,19 @@
 from tweakeyschedule import *
 line_separator = "#"*55
 
+# Check if "OR Tools" appears in the output of "minizinc --solvers" command 
+import subprocess
+try:
+    output = subprocess.run(['minizinc', '--solvers'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    if "com.google.ortools.sat" in output.stdout.decode("utf-8"):
+        ortools_available = True
+        print("OR Tools is available")
+    else:
+        ortools_available = False
+        print("OR Tools is not available")
+except FileNotFoundError:
+    ortools_available = False
+
 class ID:
     ID_counter = 0
 
@@ -66,10 +79,10 @@ def __init__(self, params) -> None:
                                      'picat', 'scip', 'choco', 'ortools']
         assert(self.cp_solver_name in self.supported_cp_solvers)
         ##################################################
-        # Use this block if you install Or-Tools bundeled with MiniZinc
-        if self.cp_solver_name == "ortools":
-            self.cp_solver_name = "com.google.ortools.sat"
-        ##################################################        
+        if ortools_available:
+            if self.cp_solver_name == "ortools":
+                self.cp_solver_name = "com.google.ortools.sat"
+        #################################################      
         self.cp_solver = minizinc.Solver.lookup(self.cp_solver_name)
 
         if self.RB + self.RF == 0: