From 5cde7a99b537c096b6fd76209ea90c44b6afac64 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Erik=20Fr=C3=B6jdh?= <erik.frojdh@gmail.com>
Date: Thu, 14 Nov 2024 17:02:48 +0100
Subject: [PATCH] WIP

---
 include/aare/ClusterFile.hpp | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/include/aare/ClusterFile.hpp b/include/aare/ClusterFile.hpp
index f42b24f..76937e0 100644
--- a/include/aare/ClusterFile.hpp
+++ b/include/aare/ClusterFile.hpp
@@ -41,9 +41,10 @@ struct ClusterAnalysis {
 class ClusterFile {
     FILE *fp{};
     uint32_t m_num_left{};
+    size_t m_chunk_size{};
 
   public:
-    ClusterFile(const std::filesystem::path &fname);
+    ClusterFile(const std::filesystem::path &fname, size_t chunk_size = 1000);
     std::vector<Cluster> read_clusters(size_t n_clusters);
     std::vector<Cluster>
     read_cluster_with_cut(size_t n_clusters, double *noise_map, int nx, int ny);
@@ -54,6 +55,8 @@ class ClusterFile {
                     double *eta2x, double *eta2y, double *eta3x,
                     double *eta3y);
 
+    size_t chunk_size() const { return m_chunk_size; }
+
 };
 
 } // namespace aare