From bf6b0b076d06b965a2894683aa030f7bef7f6e11 Mon Sep 17 00:00:00 2001
From: Mate Soos <soos.mate@gmail.com>
Date: Thu, 25 Apr 2024 15:05:50 +0200
Subject: [PATCH] Adding for unigen

---
 src/arjun.cpp | 4 ++++
 src/arjun.h   | 1 +
 2 files changed, 5 insertions(+)

diff --git a/src/arjun.cpp b/src/arjun.cpp
index b27986b..44bb37c 100644
--- 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
index bb752a0..47459ca 100644
--- 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(