ProjDataInMemory::read_from_file to return a ProjDataInMemory object

robbietuk · Sep 10, 2024 · 01e448f · 01e448f
1 parent e168b07
commit 01e448f
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/src/buildblock/ProjDataInMemory.cxx b/src/buildblock/ProjDataInMemory.cxx
@@ -350,6 +350,18 @@ ProjDataInMemory::ProjDataInMemory(const ProjDataInMemory& proj_data)
   std::copy(proj_data.begin_all(), proj_data.end_all(), this->begin_all());
 }
 
+shared_ptr<ProjDataInMemory>
+ProjDataInMemory::read_from_file(const std::string& filename, const std::ios::openmode openmode)
+{
+#if 1
+  const auto proj_data = ProjData::read_from_file(filename, openmode);
+  return std::make_shared<ProjDataInMemory>(*proj_data);
+#else
+  // Alternative implementation
+  return std::make_shared<ProjDataInMemory>(*ProjData::read_from_file(filename, openmode));
+#endif
+}
+
 float
 ProjDataInMemory::get_bin_value(Bin& bin)
 {

diff --git a/src/include/stir/ProjDataInMemory.h b/src/include/stir/ProjDataInMemory.h
@@ -62,6 +62,9 @@ class ProjDataInMemory : public ProjData
   //! Copy constructor
   ProjDataInMemory(const ProjDataInMemory& proj_data);
 
+  //! A static member to get the projection data in memory from a file
+  static shared_ptr<ProjDataInMemory> read_from_file(const std::string& filename, std::ios::openmode open_mode = std::ios::in);
+
   Viewgram<float> get_viewgram(const int view_num,
                                const int segment_num,
                                const bool make_num_tangential_poss_odd = false,