Improved algo for edge computation.

2 files changed, 14 insertions, 1 deletions

diff --git a/src/sat/bsat/satSolver.c b/src/sat/bsat/satSolver.c
index b9ab0740..815626ad 100644
--- a/src/sat/bsat/satSolver.c
+++ b/src/sat/bsat/satSolver.c
@@ -211,6 +211,19 @@ static inline int  order_select(sat_solver* s, float random_var_freq) // selectv
     return var_Undef;
 }
 
+void sat_solver_set_var_activity(sat_solver* s, int * pVars, int nVars) 
+{
+    int i;
+    for (i = 0; i < s->size; i++)
+        s->activity[i] = 0;
+    s->var_inc = 1;
+    for ( i = 0; i < nVars; i++ )
+    {
+        s->activity[pVars[i]] = nVars-i;
+        order_update( s, pVars[i] );
+    }
+}
+
 //=================================================================================================
 // Activity functions:
 
diff --git a/src/sat/bsat/satSolver.h b/src/sat/bsat/satSolver.h
index e729d2c8..c4888c73 100644
--- a/src/sat/bsat/satSolver.h
+++ b/src/sat/bsat/satSolver.h
@@ -64,7 +64,7 @@ extern int         sat_solver_count_assigned(sat_solver* s);
 
 extern void        sat_solver_setnvars(sat_solver* s,int n);
 extern int         sat_solver_get_var_value(sat_solver* s, int v);
-
+extern void        sat_solver_set_var_activity(sat_solver* s, int * pVars, int nVars);
 
 extern void        Sat_SolverWriteDimacs( sat_solver * p, char * pFileName, lit* assumptionsBegin, lit* assumptionsEnd, int incrementVars );
 extern void        Sat_SolverPrintStats( FILE * pFile, sat_solver * p );