Adding for unigen

meelgroup · Apr 25, 2024 · bf6b0b0 · bf6b0b0
1 parent f11b3c7
commit bf6b0b0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/arjun.cpp b/src/arjun.cpp
@@ -348,6 +348,10 @@ DLL_PUBLIC const std::vector<uint32_t>& Arjun::get_set_sampling_vars() const {
     return arjdata->common.set_sampling_vars;
 }
 
+DLL_PUBLIC const std::vector<uint32_t>& Arjun::get_sampl_vars() const {
+    return arjdata->common.sampling_set;
+}
+
 DLL_PUBLIC void Arjun::set_multiplier_weight(const mpz_class mult) {
     arjdata->common.solver->set_multiplier_weight(mult);
 }

diff --git a/src/arjun.h b/src/arjun.h
@@ -159,6 +159,7 @@ namespace ArjunNS {
         const std::vector<uint32_t>& get_empty_sampl_vars() const;
         bool sampling_vars_set = false;
         bool get_sampl_vars_set() const { return sampling_vars_set; }
+        const std::vector<uint32_t>& get_sampl_vars() const;
 
         //Get clauses
         void start_getting_constraints(