using single script

Author: mahi045

.github/workflows/build.yml

@@ -33,19 +33,29 @@ jobs:
       - name: Sample Hashcount run
         run: |
           cd scripts
-          python run_hashcount_on_modelcounting.py -i mccmc2021_track1_058.cnf
+          python run_hashcount.py -i mccmc2021_track1_058.cnf
 
       - name: Sample Hashcount run
         run: |
           cd scripts
-          python run_hashcount_on_itemmining.py -i vote.cnf
+          python run_hashcount.py -i minimal_vote.cnf
 
       - name: Sample ProjEnum Run
         run: |
           cd scripts
           python run_projenum.py -i mccmc2021_track1_058.cnf
 
+      - name: Sample ProjEnum Run
+        run: |
+          cd scripts
+          python run_projenum.py -i minimal_vote.cnf
+
+      - name: Sample MinLB Run
+        run: |
+          cd scripts
+          python run_minlb.py -i mccmc2021_track1_058.cnf
+
       - name: Sample MinLB Run
         run: |
           cd scripts
-          python run_minlb.py -i mccmc2021_track1_058.cnf
+          python run_minlb.py -i minimal_vote.cnf

README.md

@@ -36,11 +36,7 @@ First `cd scripts`:
 
 The command to run ``HashCount`` on model counting benchmark:
 ```
-python run_hashcount_on_modelcounting.py -i mccmc2021_track1_058.cnf
-```
-The command to run ``HashCount`` on item mining benchmark:
-```
-python run_hashcount_on_itemmining.py -i vote.cnf
+python run_hashcount.py -i mccmc2021_track1_058.cnf
 ```
 
 ## Run ProjEnum

scripts/compute_dlp.py

@@ -11,12 +11,23 @@ def compute_disjunctive_program(file_name):
     file_pointer = open(file_name, 'r')
     output_file_pointer = open(output_file, 'w')
     for line in file_pointer:
-        if line.startswith("c"):
-            continue
-        elif line.startswith("p cnf"):
+        if line.startswith("p cnf"):
             l = line.split()
             output_file_pointer.write("%rule size: {0}\n".format(int(l[-1])))
             # print("The number of literals: {0} and clauses: {1} [Unpreprocessed]".format(l[-2], l[-1]))
+        elif line.startswith("c ind"):
+            # for some benchmarks (e.g., item mining), we can compute independent support easily
+            # prepare independent support file
+            IS_file = open("IS_dlp_" + file_name, 'w') 
+            numbers = [int(x) for x in line.split()[2:] if int(x) != 0]
+            IS_str = "c ind "
+            for item in numbers:
+                IS_str += "v{0} ".format(item)
+            IS_str += "0\n"
+            IS_file.write(IS_str)
+            IS_file.close()
+        elif line.startswith("c"):
+            continue
         else:
             l = line.split()
             lit_list = []

scripts/minimal_vote.cnf

@@ -35,35 +35,36 @@
 
         transaction_id += 1
 
-asp_file_name = "dlp_" + file_name
-IS_file_name = "IS_dlp_" + file_name
+# asp_file_name = "dlp_" + file_name
+# IS_file_name = "IS_dlp_" + file_name
 cut_file_name = "minimal_" + file_name
-asp_file = open(asp_file_name, 'w')
-IS_file = open(IS_file_name, 'w')
+# asp_file = open(asp_file_name, 'w')
+# IS_file = open(IS_file_name, 'w')
 cut_file = open(cut_file_name, 'w')
 cut_str = "p cnf {0} {1}\n".format(int(max(union_of_items)) + transaction_id - 1, transaction_id - 1)
 cut_file.write(cut_str)
 transaction_id = int(max(union_of_items))
 
+IS_str = "c ind "
+for item in union_of_items:
+    IS_str += "{0} ".format(item)
+IS_str += "0\n"
+cut_file.write(IS_str)
+
 for index, transaction in enumerate(transaction_items):
-    rule_str = "t({0})".format(index + 1)
+    # rule_str = "t({0})".format(index + 1)
     clause_str = "{0} ".format(transaction_id + index + 1)
     if len(transaction_items) > 0:
-        rule_str += "".join("; i(" + str(_) + ")" for _ in transaction)
+        # rule_str += "".join("; i(" + str(_) + ")" for _ in transaction)
         clause_str += "".join(str(_) + " " for _ in transaction)
-    rule_str += ".\n"
+    # rule_str += ".\n"
     clause_str += "0\n"
-    asp_file.write(rule_str)
+    # asp_file.write(rule_str)
     cut_file.write(clause_str)
 
 
-IS_str = "c ind "
-for item in union_of_items:
-    IS_str += "i({0}) ".format(item)
-IS_str += "0\n"
-IS_file.write(IS_str)
-IS_file.close()
-asp_file.close()
+# IS_file.close()
+# asp_file.close()
 cut_file.close()
 
 # ctl.load(asp_file_name)

scripts/propagator.py

@@ -26,11 +26,16 @@ def run(cmd, timeout, ttl = 3, silent = False):
     os.system("cp {0} {1}".format(args.i, temp_file))
     input_file = os.path.basename(temp_file)
 
+if os.path.exists(f"IS_dlp_{input_file}"):
+    os.remove(f"IS_dlp_{input_file}")
+
 cmd = 'python compute_dlp.py -i {0}'.format(input_file)
 out = run(cmd, 100)
-# print(" === Computing independent suport === ")
-cmd = 'python compute_independent_support.py -i dlp_{0}'.format(input_file)
-out = run(cmd, 250)
+if not os.path.exists(f"IS_dlp_{input_file}"):
+    # print(" === Computing independent suport === ")
+    cmd = 'python compute_independent_support.py -i dlp_{0}'.format(input_file)
+    out = run(cmd, 250)
+
 # print(" === Running HashCount === ")
 cmd = './hashcount --useind IS_dlp_{0} --asp dlp_{0}'.format(input_file)
 out = run(cmd, 5000)
@@ -42,6 +47,6 @@ def run(cmd, timeout, ttl = 3, silent = False):
         print("hashcount: {0}".format(line))
 
 if cnt is None:
-    print("No estimate found in the hashcount.")
+    print("No estimate found from hashcount.")
 
 os.system(f'rm -f {input_file} dlp_{input_file} IS_dlp_{input_file}')

scripts/run_hashcount_on_modelcounting.py scripts/run_hashcount.pyscripts/run_hashcount_on_modelcounting.py renamed to scripts/run_hashcount.py

@@ -39,6 +39,6 @@ def run(cmd, timeout, ttl = 3, silent = False):
         print("hashcount: {0}".format(line))
 
 if cnt is None:
-    print("No estimate found in the hashcount.")
+    print("No estimate found from hashcount.")
 
 os.system(f'rm -f {input_file} IS_dlp_{input_file} dlp_{input_file}')

scripts/run_hashcount_on_itemmining.py

@@ -2,6 +2,7 @@
 import subprocess as sp
 parser = argparse.ArgumentParser()
 parser.add_argument('-i','--i', help='input CNF file', required=True)
+parser.add_argument('-thresh','--thresh', help='threshold cut size', default=25, required=False)
 args = parser.parse_args()
 file_name = args.i
 
@@ -26,6 +27,9 @@ def run(cmd, timeout, ttl = 3, silent = False):
     os.system("cp {0} {1}".format(args.i, temp_file))
     input_file = os.path.basename(temp_file)
 
+if os.path.exists(f"IS_dlp_{input_file}"):
+    os.remove(f"IS_dlp_{input_file}")
+
 cmd = 'python compute_dlp.py -i {0}'.format(input_file)
 out = run(cmd, 100)
 # print(" === Computing a cut === ")
@@ -36,7 +40,7 @@ def run(cmd, timeout, ttl = 3, silent = False):
 # out = run(cmd, 100)
 
 cut_size = None
-threshold = 50
+threshold = int(args.thresh)
 cut_string = ""
 for line in out.splitlines():
     if line.startswith("c the size of cut:"):
@@ -70,12 +74,13 @@ def run(cmd, timeout, ttl = 3, silent = False):
     if cnt is None:
         print("No estimate found in the projenum.")
 
-    os.system(f'rm -f {input_file} cut_{input_file} minimal_{input_file}')
+    os.system(f'rm -f {input_file} cut_{input_file} dlp_{input_file} minimal_{input_file}')
 else:
-    # we can run Hashcount
-    # print(" === Computing independent suport === ")
-    cmd = 'python compute_independent_support.py -i dlp_{0}'.format(input_file)
-    out = run(cmd, 250)
+    # we run Hashcount
+    if not os.path.exists(f"IS_dlp_{input_file}"):
+        # print(" === Computing independent suport === ")   
+        cmd = 'python compute_independent_support.py -i dlp_{0}'.format(input_file)
+        out = run(cmd, 250)
     # print(" === Running HashCount === ")
     cmd = './hashcount --useind IS_dlp_{0} --asp dlp_{0}'.format(input_file)
     out = run(cmd, 5000)
@@ -90,4 +95,4 @@ def run(cmd, timeout, ttl = 3, silent = False):
     if cnt is None:
         print("No estimate found in the hashcount.")
 
-    os.system(f'rm -f {input_file} dlp_{input_file} IS_dlp_{input_file}')
+    os.system(f'rm -f {input_file} dlp_{input_file} IS_dlp_{input_file} minimal_{input_file} {new_input_file}')

scripts/run_minlb.py

@@ -31,7 +31,7 @@ def run(cmd, timeout, ttl = 3, silent = False):
 
 out = run(cmd, 100)
 # print(" === Computing a cut === ")
-cmd = './td -decot 100 -decow 100 -tmpdir . -cs 3500 minimal_{0}'.format(input_file)
+cmd = './td -decot 100 -decow 100 -tmpdir . -cs 4000 minimal_{0}'.format(input_file)
 out = run(cmd, 120)
 cut_size = None
 cut_string = ""