From 11736f078e1469d51938689f88c2fd82a5d75e21 Mon Sep 17 00:00:00 2001
From: Nikolaj Bjorner <nbjorner@microsoft.com>
Date: Fri, 18 Oct 2019 19:22:36 -0700
Subject: [PATCH] ensure statistics survive cancelation in tactics, fix
 propagation for smtfd

Signed-off-by: Nikolaj Bjorner <nbjorner@microsoft.com>
---
 src/solver/solver2tactic.cpp          | 11 +++++++++--
 src/tactic/fd_solver/smtfd_solver.cpp |  8 ++++++--
 2 files changed, 15 insertions(+), 4 deletions(-)

diff --git a/src/solver/solver2tactic.cpp b/src/solver/solver2tactic.cpp
index 2bd5f59c7..5d35b879c 100644
--- a/src/solver/solver2tactic.cpp
+++ b/src/solver/solver2tactic.cpp
@@ -111,10 +111,18 @@ public:
         ref<solver> local_solver = m_solver->translate(m, m_params);
         local_solver->assert_expr(clauses);
         TRACE("solver2tactic", tout << "clauses asserted\n";);
-        lbool r = local_solver->check_sat(assumptions.size(), assumptions.c_ptr()); 
+        lbool r;
+        try {
+            r = local_solver->check_sat(assumptions.size(), assumptions.c_ptr()); 
+        }
+        catch (...) {
+            local_solver->collect_statistics(m_st);
+            throw;
+        }
         TRACE("solver2tactic", tout << "check sat result " << r << "\n";);
         proof* pr = local_solver->get_proof();
         if (pr) in->set(proof2proof_converter(m, pr));
+        local_solver->collect_statistics(m_st);
         switch (r) {
         case l_true: 
             if (in->models_enabled()) {
@@ -160,7 +168,6 @@ public:
             result.push_back(in.get());
             break;
         }
-        local_solver->collect_statistics(m_st);
     }
 
     void collect_statistics(statistics & st) const override {
diff --git a/src/tactic/fd_solver/smtfd_solver.cpp b/src/tactic/fd_solver/smtfd_solver.cpp
index 94b1974dc..e185d779e 100644
--- a/src/tactic/fd_solver/smtfd_solver.cpp
+++ b/src/tactic/fd_solver/smtfd_solver.cpp
@@ -584,6 +584,8 @@ namespace smtfd {
             m_pinned.reset();
             m_tables.reset();
             m_ast2table.reset();
+            m_values.reset();
+            m_model = nullptr;
         }
     };
 
@@ -875,7 +877,7 @@ namespace smtfd {
             expr_ref val2 = eval_abs(stored_value);
             // A[i] = v
             if (val1 != val2) {
-                TRACE("smtfd", tout << "select/store: " << mk_pp(t, m) << "\n";);
+                TRACE("smtfd", tout << "select/store: " << mk_bounded_pp(t, m, 2) << "\n";);
                 add_lemma(m.mk_eq(sel, stored_value));
             }
             m_pinned.push_back(sel);
@@ -1080,6 +1082,7 @@ namespace smtfd {
         {}
         
         void check_term(expr* t, unsigned round) override {
+            TRACE("smtfd", tout << mk_bounded_pp(t, m, 2) << "\n";);
             switch (round) {
             case 0:
                 if (m_autil.is_select(t)) {
@@ -1797,8 +1800,8 @@ namespace smtfd {
         lbool refine_core(expr_ref_vector & core) {
             lbool r = l_true;
             unsigned round = 0;
+            m_context.reset(m_model);
             while (true) {
-                m_context.reset(m_model);
                 if (!m_context.add_theory_axioms(core, round)) {
                     break;
                 }
@@ -1822,6 +1825,7 @@ namespace smtfd {
                 case l_true:
                     m_fd_sat_solver->get_model(m_model);
                     m_model->set_model_completion(true);
+                    m_context.reset(m_model);
                     break;
                 default:
                     return r;