adding unit test for arith_plugin

src/ast/euf/euf_ac_plugin.cpp

@@ -41,8 +41,6 @@ More notes:
 
 --*/
 
-#pragma once
-
 #include "ast/euf/euf_ac_plugin.h"
 #include "ast/euf/euf_egraph.h"
 
@@ -275,7 +273,7 @@ namespace euf {
             return m_nodes[id];
         auto* r = node::mk(get_region(), n);
         push_undo(is_add_node);
-        m_nodes.set(id, r);
+        m_nodes.setx(id, r, nullptr);
         m_node_trail.push_back(r);
         return r;
     }

src/ast/euf/euf_ac_plugin.h

@@ -48,7 +48,7 @@ namespace euf {
 
             unsigned root_id() const { return root->n->get_id(); }
             ~node() {}
-            static node* mk(region& r, enode* n);            
+            static node* mk(region& r, enode* n);
         };
 
         class equiv {
@@ -71,17 +71,17 @@ namespace euf {
             iterator end() const { return iterator(&n, &n); }
         };
 
-        struct eq {   
+        struct eq {
             unsigned l, r;   // refer to monomials
             bool is_processed = false;
-            justification j; 
+            justification j;
         };
 
         unsigned                 m_fid;
-        unsigned                 m_op;        
+        unsigned                 m_op;
         vector<eq>               m_eqs;
         ptr_vector<node>         m_nodes;
-        vector<ptr_vector<node>> m_monomials; 
+        vector<ptr_vector<node>> m_monomials;
         enode_vector             m_monomial_enodes;
         justification::dependency_manager m_dep_manager;
 
@@ -103,8 +103,8 @@ namespace euf {
         svector<std::pair<unsigned, eq>> m_update_eq_trail;
 
         node* mk_node(enode* n);
-        void merge(node* r1, node* r2, justification j);        
-         
+        void merge(node* r1, node* r2, justification j);
+
         bool is_op(enode* n) const { auto d = n->get_decl(); return d && m_fid == d->get_family_id() && m_op == d->get_kind(); }
 
         std::function<void(void)> m_undo_notify;
@@ -114,9 +114,9 @@ namespace euf {
         unsigned to_monomial(enode* n, ptr_vector<node> const& ms);
         ptr_vector<node> const& monomial(unsigned i) const { return m_monomials[i]; }
         ptr_vector<node>& monomial(unsigned i) { return m_monomials[i]; }
-        
+
         void init_equation(eq const& e);
-        bool orient_equation(eq & e);
+        bool orient_equation(eq& e);
         void set_processed(unsigned eq_id, bool f);
         unsigned pick_next_eq();
         bool is_trivial(unsigned eq_id) const { throw default_exception("NYI"); }
@@ -143,7 +143,6 @@ namespace euf {
         bool is_processed(unsigned eq) const { return m_eqs[eq].is_processed; }
 
         justification justify_rewrite(unsigned eq1, unsigned eq2);
-        justification justify_superpose(justification j1, justification j2);
         justification::dependency* justify_monomial(justification::dependency* d, ptr_vector<node> const& m);
 
         void propagate_shared();
@@ -152,6 +151,8 @@ namespace euf {
     public:
 
         ac_plugin(egraph& g, unsigned fid, unsigned op);
+
+        ~ac_plugin() override {}
         
         unsigned get_id() const override { return m_fid; }
 

src/ast/euf/euf_arith_plugin.cpp

@@ -52,6 +52,11 @@ namespace euf {
         // no-op
     }
 
+    void arith_plugin::propagate() {
+        m_add.propagate();
+        m_mul.propagate();
+    }
+
     void arith_plugin::undo() {
         auto k = m_undo.back();
         m_undo.pop_back();

src/ast/euf/euf_arith_plugin.h

@@ -33,6 +33,8 @@ namespace euf {
     public:
         arith_plugin(egraph& g);
 
+        ~arith_plugin() override {}
+
         unsigned get_id() const override { return a.get_family_id(); }
 
         void register_node(enode* n) override;
@@ -44,6 +46,8 @@ namespace euf {
         void diseq_eh(enode* n1, enode* n2) override;
 
         void undo() override;
+
+        void propagate() override;
         
         std::ostream& display(std::ostream& out) const override;
             

src/ast/euf/euf_bv_plugin.h

@@ -80,6 +80,8 @@ namespace euf {
     public:
         bv_plugin(egraph& g);
 
+        ~bv_plugin() override {}
+
         unsigned get_id() const override { return bv.get_family_id(); }
 
         void register_node(enode* n) override;

src/ast/euf/euf_egraph.cpp

@@ -19,6 +19,7 @@ Notes:
 
 #include "ast/euf/euf_egraph.h"
 #include "ast/euf/euf_bv_plugin.h"
+#include "ast/euf/euf_arith_plugin.h"
 #include "ast/ast_pp.h"
 #include "ast/ast_translation.h"
 
@@ -159,9 +160,13 @@ namespace euf {
     }
 
     void egraph::add_plugins() {
-        auto* plugin = alloc(bv_plugin, *this);
-        m_plugins.reserve(plugin->get_id() + 1);
-        m_plugins.set(plugin->get_id(), plugin);    
+        auto insert = [&](plugin* p) {
+            m_plugins.reserve(p->get_id() + 1);
+            m_plugins.set(p->get_id(), p);
+        };
+
+        insert(alloc(bv_plugin, *this));
+        insert(alloc(arith_plugin, *this));
     }
 
     void egraph::propagate_plugins() {

src/ast/euf/euf_plugin.h

@@ -38,6 +38,8 @@ namespace euf {
             g(g)
         {}
 
+        virtual ~plugin() {}
+
         virtual unsigned get_id() const = 0;
 
         virtual void register_node(enode* n) = 0;