Add unique read_csv in utility (cms-patatrack#87)

* Add unique `read_csv` in utility * Update benchmarking scripts
sbaldu · Feb 7, 2025 · e73a582 · e73a582
1 parent 816f15c
commit e73a582
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 58 deletions.
diff --git a/benchmark/dataset_size/main.cpp b/benchmark/dataset_size/main.cpp
@@ -12,7 +12,7 @@
 #include "CLUEstering/DataFormats/Points.hpp"
 #include "CLUEstering/DataFormats/alpaka/PointsAlpaka.hpp"
 
-#include "read_csv.hpp"
+#include "CLUEstering/utility/read_csv.hpp"
 
 #include <pybind11/embed.h>
 #include <pybind11/stl_bind.h>

diff --git a/benchmark/profiling/main.cpp b/benchmark/profiling/main.cpp
@@ -8,7 +8,7 @@
 #include "CLUEstering/DataFormats/Points.hpp"
 #include "CLUEstering/DataFormats/alpaka/PointsAlpaka.hpp"
 
-#include "read_csv.hpp"
+#include "CLUEstering/utility/read_csv.hpp"
 
 namespace ALPAKA_ACCELERATOR_NAMESPACE_CLUE {
   void run(const std::string& input_file) {

diff --git a/benchmark/profiling/read_csv.hpp b/benchmark/profiling/read_csv.hpp
diff --git a/benchmark/dataset_size/read_csv.hpp → include/CLUEstering/utility/read_csv.hpp b/benchmark/dataset_size/read_csv.hpp → include/CLUEstering/utility/read_csv.hpp
@@ -1,17 +1,11 @@
 
-#ifndef read_csv_hpp
-#define read_csv_hpp
+#pragma once
 
 #include <fstream>
 #include <sstream>
 #include <string>
-#include <utility>
 #include <vector>
 
-#include "CLUEstering/DataFormats/alpaka/AlpakaVecArray.hpp"
-
-using clue::VecArray;
-
 template <typename T, size_t NDim>
 std::vector<T> read_csv(const std::string& file_path) {
   std::fstream file(file_path);
@@ -20,9 +14,9 @@ std::vector<T> read_csv(const std::string& file_path) {
   }
   auto n_points = std::count(
       std::istreambuf_iterator<char>(file), std::istreambuf_iterator<char>(), '\n');
-
   std::vector<float> coords((NDim + 1) * n_points);
 
+  file = std::fstream(file_path);
   // discard the header
   std::string buffer;
   getline(file, buffer);
@@ -42,5 +36,3 @@ std::vector<T> read_csv(const std::string& file_path) {
 
   return coords;
 }
-
-#endif