switched k to a double

1 files changed, 5 insertions, 3 deletions

diff --git a/include/framework/reconstruction/TieringPolicy.h b/include/framework/reconstruction/TieringPolicy.h
index 55f5f47..b1fcb49 100644
--- a/include/framework/reconstruction/TieringPolicy.h
+++ b/include/framework/reconstruction/TieringPolicy.h
@@ -21,7 +21,7 @@ class TieringPolicy : public ReconstructionPolicy<ShardType, QueryType> {
       LevelVector;
 
 public:
-  TieringPolicy(size_t scale_factor, size_t buffer_size, size_t modifier=0)
+  TieringPolicy(size_t scale_factor, size_t buffer_size, double modifier=0)
       : m_scale_factor(scale_factor), m_buffer_size(buffer_size), m_size_modifier(modifier) {}
 
   std::vector<ReconstructionVector> get_reconstruction_tasks(
@@ -73,10 +73,12 @@ private:
     return target_level;
   }
 
-  inline size_t capacity(size_t reccnt) const { return m_scale_factor * std::pow(std::log(reccnt), m_size_modifier); }
+  inline size_t capacity(size_t reccnt) const {
+    return std::ceil((double) m_scale_factor * std::pow<double>(std::log10(reccnt), m_size_modifier));
+  }
 
   size_t m_scale_factor;
   size_t m_buffer_size;
-  size_t m_size_modifier;
+  double m_size_modifier;
 };
 } // namespace de