added collect-statistics tactic

src/tactic/core/collect_statistics_tactic.cpp

@@ -0,0 +1,173 @@
+/*++
+Copyright (c) 2016 Microsoft Corporation
+
+Module Name:
+
+    collect_statistics_tactic.cpp
+
+Abstract:
+
+    A tactic for collection of various statistics.
+
+Author:
+
+    Mikolas Janota (mikjan) 2016-06-03
+    Christoph (cwinter) 2016-06-03
+
+Notes:
+
+--*/
+#include<string>
+#include<unordered_map>
+
+#include"ast.h"
+#include"params.h"
+#include"arith_decl_plugin.h"
+#include"array_decl_plugin.h"
+#include"bv_decl_plugin.h"
+#include"datatype_decl_plugin.h"
+#include"fpa_decl_plugin.h"
+#include"tactical.h"
+#include"stats.h"
+
+#include"collect_statistics_tactic.h"
+
+class collect_statistics_tactic : public tactic {
+    ast_manager &        m;
+    params_ref           m_params;
+    basic_decl_plugin    m_basic_pi;
+    arith_decl_plugin    m_arith_pi;
+    array_decl_plugin    m_array_pi;
+    bv_decl_plugin       m_bv_pi;
+    datatype_decl_plugin m_datatype_pi;
+    fpa_decl_plugin      m_fpa_pi;
+
+    typedef std::unordered_map<std::string, unsigned long> stats_type;
+    stats_type m_stats;
+
+public:
+    collect_statistics_tactic(ast_manager & m, params_ref const & p) :
+        m(m), 
+        m_params(p) {
+    }    
+        
+    virtual ~collect_statistics_tactic() {}
+
+    virtual tactic * translate(ast_manager & m_) {
+        return alloc(collect_statistics_tactic, m_, m_params);
+    }
+
+    virtual void updt_params(params_ref const & p) {
+        m_params = p;
+    }
+    
+    virtual void collect_param_descrs(param_descrs & r) {}
+
+    virtual void operator()(goal_ref const & g, goal_ref_buffer & result,
+                            model_converter_ref & mc, proof_converter_ref & pc, 
+                            expr_dependency_ref & core) {
+        mc = 0;
+        tactic_report report("collect-statistics", *g);        
+        
+        collect_proc cp(m, m_stats);
+        expr_mark visited;                            
+        const unsigned sz = g->size();
+        for (unsigned i = 0; i < sz; i++)
+            for_each_expr(cp, visited, g->form(i));
+        
+        std::cout << "(" << std::endl;
+        stats_type::iterator it = m_stats.begin();
+        stats_type::iterator end = m_stats.end();
+        for (; it != end; it++)
+            std::cout << " :" << it->first << "    " << it->second << std::endl;
+        std::cout << ")" << std::endl;
+
+        g->inc_depth();
+        result.push_back(g.get());
+    }    
+
+    virtual void cleanup() {}
+
+    virtual void collect_statistics(statistics & st) const {
+    }
+
+    virtual void reset_statistics() { reset();  }
+    virtual void reset() { cleanup(); }
+
+protected:
+    class collect_proc {
+    public:
+        ast_manager &            m;
+        stats_type &             m_stats;        
+        obj_hashtable<sort>      m_seen_sorts;
+        obj_hashtable<func_decl> m_seen_func_decls;
+
+        collect_proc(ast_manager & m, stats_type & s) : m(m), m_stats(s) {}
+
+        void operator()(var * v) {
+            m_stats["bound-variables"]++;
+            this->operator()(v->get_sort());
+        }
+
+        void operator()(quantifier * q) {
+            m_stats["quantifiers"]++;
+            SASSERT(is_app(q->get_expr()));
+            app * body = to_app(q->get_expr());
+            this->operator()(body);
+        }
+
+        void operator()(app * n) {
+            m_stats["function-applications"]++;
+            this->operator()(n->get_decl());
+        }
+
+        void operator()(sort * s) {            
+            if (m.is_uninterp(s)) {
+                if (!m_seen_sorts.contains(s)) {
+                    m_stats["uninterpreted-sorts"]++;
+                    m_seen_sorts.insert(s);
+                }
+                m_stats["uninterpreted-sort-occurrences"]++;
+            }
+            else {
+                params_ref prms;
+                prms.set_bool("pp.single_line", true);
+                std::stringstream ss;
+                ss << mk_ismt2_pp(s, m, prms);
+                m_stats[ss.str().c_str()]++;
+            }
+        }
+
+        void operator()(func_decl * f) {
+            for (unsigned i = 0; i < f->get_arity(); i++)
+                this->operator()(f->get_domain()[i]);
+            this->operator()(f->get_range());
+
+            if (f->get_family_id() == null_family_id) {
+                if (!m_seen_func_decls.contains(f)) {
+                    if (f->get_arity() == 0)
+                        m_stats["uninterpreted-constants"]++;
+                    else
+                        m_stats["uninterpreted-functions"]++;
+                    m_seen_func_decls.insert(f);
+                }
+
+                if (f->get_arity() > 0)
+                    m_stats["uninterpreted-function-occurrences"]++;
+            }
+            else {
+                params_ref prms;
+                prms.set_bool("pp.single_line", true);
+                std::stringstream ss;
+                ss << mk_ismt2_pp(f, m, prms);
+                m_stats[ss.str().c_str()]++;
+            }
+            m_stats["function-applications"]++;
+        }
+    };
+};
+
+
+tactic * mk_collect_statistics_tactic(ast_manager & m, params_ref const & p) {
+    return clean(alloc(collect_statistics_tactic, m, p));
+}

src/tactic/core/collect_statistics_tactic.h

@@ -0,0 +1,33 @@
+/*++
+Copyright (c) 2016 Microsoft Corporation
+
+Module Name:
+
+    collect_statistics_tactic.h
+
+Abstract:
+
+    A tactic for collection of various statistics. 
+
+Author:
+
+    Mikolas Janota (mikjan) 2016-06-03
+    Christoph (cwinter) 2016-06-03
+
+Notes:
+
+--*/
+#ifndef COLLECT_STATISTICS_H_
+#define COLLECT_STATISTICS_H_
+
+#include"params.h"
+class ast_manager;
+class tactic;
+
+tactic * mk_collect_statistics_tactic(ast_manager & m, params_ref const & p = params_ref());
+/*
+  ADD_TACTIC("collect-statistics", "Collects various statistics.", "mk_collect_statistics_tactic(m, p)")
+*/
+
+
+#endif