Exposed epsilon configuration parameter for PGM

1 files changed, 5 insertions, 4 deletions

diff --git a/include/shard/PGM.h b/include/shard/PGM.h
index 0d4268a..65d548e 100644
--- a/include/shard/PGM.h
+++ b/include/shard/PGM.h
@@ -53,12 +53,13 @@ struct PGMBufferState {
 
 };
 
-template <RecordInterface R>
+template <RecordInterface R, size_t epsilon=128>
 class PGM {
 private:
     typedef decltype(R::key) K;
     typedef decltype(R::value) V;
 
+
 public:
 
     // FIXME: there has to be a better way to do this
@@ -112,7 +113,7 @@ public:
         }
 
         if (m_reccnt > 0) {
-            m_pgm = pgm::PGMIndex<K>(keys);
+            m_pgm = pgm::PGMIndex<K, epsilon>(keys);
         }
     }
 
@@ -175,7 +176,7 @@ public:
         }
 
         if (m_reccnt > 0) {
-            m_pgm = pgm::PGMIndex<K>(keys);
+            m_pgm = pgm::PGMIndex<K, epsilon>(keys);
         }
    }
 
@@ -270,7 +271,7 @@ private:
     size_t m_alloc_size;
     K m_max_key;
     K m_min_key;
-    pgm::PGMIndex<K> m_pgm;
+    pgm::PGMIndex<K, epsilon> m_pgm;
     BloomFilter<R> *m_bf;
 };