From 9f3b16927f39d64c09b1a6c1206a4f384d8c99a0 Mon Sep 17 00:00:00 2001
From: Andres Noetzli <andres.noetzli@gmail.com>
Date: Sat, 25 Jan 2020 13:51:39 -0800
Subject: Add stats

---
 src/theory/strings/skolem_cache.cpp | 16 ++++++++++++----
 src/theory/strings/skolem_cache.h   |  3 ++-
 2 files changed, 14 insertions(+), 5 deletions(-)

diff --git a/src/theory/strings/skolem_cache.cpp b/src/theory/strings/skolem_cache.cpp
index cd30d178a..c11833420 100644
--- a/src/theory/strings/skolem_cache.cpp
+++ b/src/theory/strings/skolem_cache.cpp
@@ -49,6 +49,11 @@ Node SkolemCache::mkTypedSkolemCached(
   a = a.isNull() ? a : Rewriter::rewrite(a);
   b = b.isNull() ? b : Rewriter::rewrite(b);
 
+  if (d_skolemCache[a][b].find(id) != d_skolemCache[a][b].end())
+  {
+    d_statistics.d_numCachedSkolemsPre += 1;
+  }
+
   if (options::skolemSharing() && tn == d_strType)
   {
     std::tie(id, a, b) = normalizeStringSkolem(id, a, b);
@@ -63,7 +68,7 @@ Node SkolemCache::mkTypedSkolemCached(
   }
   else
   {
-    d_statistics.d_numCachedSkolems += 1;
+    d_statistics.d_numCachedSkolemsPost += 1;
     d_statistics.d_numSkolems += 1;
   }
   return it->second;
@@ -215,15 +220,18 @@ SkolemCache::normalizeStringSkolem(SkolemId id, Node a, Node b)
 
 SkolemCache::Statistics::Statistics()
     : d_numSkolems("theory::strings::NumSkolems", 0),
-      d_numCachedSkolems("theory::strings::NumCachedSkolems", 0)
+      d_numCachedSkolemsPre("theory::strings::NumCachedSkolemsPre", 0),
+      d_numCachedSkolemsPost("theory::strings::NumCachedSkolemsPost", 0)
 {
   smtStatisticsRegistry()->registerStat(&d_numSkolems);
-  smtStatisticsRegistry()->registerStat(&d_numCachedSkolems);
+  smtStatisticsRegistry()->registerStat(&d_numCachedSkolemsPre);
+  smtStatisticsRegistry()->registerStat(&d_numCachedSkolemsPost);
 }
 
 SkolemCache::Statistics::~Statistics() {
   smtStatisticsRegistry()->unregisterStat(&d_numSkolems);
-  smtStatisticsRegistry()->unregisterStat(&d_numCachedSkolems);
+  smtStatisticsRegistry()->unregisterStat(&d_numCachedSkolemsPre);
+  smtStatisticsRegistry()->unregisterStat(&d_numCachedSkolemsPost);
 }
 
 }  // namespace strings
diff --git a/src/theory/strings/skolem_cache.h b/src/theory/strings/skolem_cache.h
index 3612f8503..3946d7622 100644
--- a/src/theory/strings/skolem_cache.h
+++ b/src/theory/strings/skolem_cache.h
@@ -174,7 +174,8 @@ class SkolemCache
   struct Statistics
   {
     IntStat d_numSkolems;
-    IntStat d_numCachedSkolems;
+    IntStat d_numCachedSkolemsPre;
+    IntStat d_numCachedSkolemsPost;
 
     Statistics();
     ~Statistics();
-- 
cgit v1.2.3