Fix rankTable reading for file-based encoding

openPMD · Jun 20, 2023 · a90dc99 · a90dc99
1 parent 7d1d002
commit a90dc99
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/src/Series.cpp b/src/Series.cpp
@@ -189,6 +189,17 @@ chunk_assignment::RankMeta Series::mpiRanksMetaInfo()
     {
         return series.m_bufferedRanktableReadonly.value();
     }
+    if (iterationEncoding() == IterationEncoding::fileBased)
+    {
+        if (iterations.empty())
+        {
+            return {};
+        }
+        Parameter<Operation::OPEN_FILE> openFile;
+        openFile.encoding = IterationEncoding::fileBased;
+        openFile.name = iterationFilename(iterations.begin()->first);
+        IOHandler()->enqueue(IOTask(this, openFile));
+    }
     Parameter<Operation::LIST_DATASETS> listDatasets;
     IOHandler()->enqueue(IOTask(this, listDatasets));
     IOHandler()->flush(internal::defaultFlushParams);
@@ -267,6 +278,15 @@ chunk_assignment::RankMeta Series::mpiRanksMetaInfo()
     doReadDataset();
 #endif
 
+    if (iterationEncoding() == IterationEncoding::fileBased)
+    {
+        auto &it = iterations.begin()->second;
+        Parameter<Operation::CLOSE_FILE> closeFile;
+        IOHandler()->enqueue(IOTask(&it, closeFile));
+        it.get().m_closed = internal::CloseStatus::ClosedTemporarily;
+        IOHandler()->flush(internal::defaultFlushParams);
+    }
+
     chunk_assignment::RankMeta res;
     for (size_t i = 0; i < writerRanks; ++i)
     {