Merge pull request #141 from rest-for-physics/nkx-patch-endprocess

fixed calling of external process's endprocess() method
rest-for-physics · Mar 8, 2022 · b8f82a4 · b8f82a4
2 parents 2b04de4 + 55bf0fd
commit b8f82a4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/source/framework/core/src/TRestProcessRunner.cxx b/source/framework/core/src/TRestProcessRunner.cxx
@@ -466,11 +466,14 @@ void TRestProcessRunner::RunProcess() {
         if (finish) break;
     }
 
+    // make analysis tree filled with observables, which may used in EndProcess()
     fAnalysisTree->GetEntry(fAnalysisTree->GetEntries() - 1);
     for (int i = 0; i < fThreadNumber; i++) {
         fThreads[i]->EndProcess();
     }
-
+    if (fRunInfo->GetFileProcess()) {
+        fRunInfo->GetFileProcess()->EndProcess();
+    }
     fProcStatus = kFinished;
 
 #ifdef TIME_MEASUREMENT

diff --git a/source/framework/core/src/TRestRun.cxx b/source/framework/core/src/TRestRun.cxx
@@ -804,7 +804,7 @@ Int_t TRestRun::GetNextEvent(TRestEvent* targetevt, TRestAnalysisTree* targettre
             goto GetEventExt;
         }
         fInputEvent = eve;
-        if (fFileProcess != nullptr) fFileProcess->EndProcess();
+        // if (fFileProcess != nullptr) fFileProcess->EndProcess();
         return -1;
     }