Merge pull request #511 from rest-for-physics/lobis-download

Add option to PID prefix on downloads to avoid faults

Author: lobis

source/framework/core/src/TRestEvent.cxx

@@ -49,6 +49,8 @@
 
 #include "TRestEvent.h"
 
+#include "TRestRun.h"
+
 using namespace std;
 
 ClassImp(TRestEvent);
@@ -170,7 +172,11 @@ void TRestEvent::RestartPad(Int_t nElements) {
 
 void TRestEvent::InitializeWithMetadata(TRestRun* run) { Initialize(); }
 
-void TRestEvent::InitializeReferences(TRestRun* run) { fRun = run; }
+void TRestEvent::InitializeReferences(TRestRun* run) {
+    fRun = run;
+    SetRunOrigin(fRun->GetRunNumber());
+    SetSubRunOrigin(fRun->GetSubRunNumber());
+}
 
 //////////////////////////////////////////////////////////////////////////
 /// Run to print event data info on console

source/framework/core/src/TRestMetadata.cxx

@@ -1284,7 +1284,7 @@ void TRestMetadata::ExpandIncludeFile(TiXmlElement* e) {
             url = _filename;
         }
 
-        filename = TRestTools::DownloadRemoteFile(url);
+        filename = TRestTools::DownloadRemoteFile(url, true);
     } else {
         filename = SearchFile(_filename);
     }

source/framework/tools/inc/TRestTools.h

@@ -127,7 +127,7 @@ class TRestTools {
 
     static std::string Execute(std::string cmd);
 
-    static std::string DownloadRemoteFile(const std::string& remoteFile);
+    static std::string DownloadRemoteFile(const std::string& remoteFile, bool pidPrefix = false);
     static int DownloadRemoteFile(std::string remoteFile, std::string localFile);
     static int UploadToServer(std::string localFile, std::string remoteFile, std::string methodUrl = "");
 

source/framework/tools/src/TRestTools.cxx

@@ -1121,7 +1121,7 @@ std::istream& TRestTools::GetLine(std::istream& is, std::string& t) {
 /// will be used, including scp, wget. Downloads to REST_USER_PATH + "/download/" + filename
 /// by default.
 ///
-std::string TRestTools::DownloadRemoteFile(const string& url) {
+std::string TRestTools::DownloadRemoteFile(const string& url, bool pidPrefix) {
     string pureName = TRestTools::GetPureFileName(url);
     if (pureName.empty()) {
         RESTWarning << "error! (TRestTools::DownloadRemoteFile): url is not a file!" << RESTendl;
@@ -1133,7 +1133,8 @@ std::string TRestTools::DownloadRemoteFile(const string& url) {
     if (url.find("local:") == 0) {
         return Replace(url, "local:", "");
     } else {
-        string fullpath = REST_USER_PATH + "/download/" + pureName;
+        string fullpath =
+            REST_USER_PATH + "/download/" + (pidPrefix ? "PID_" + ToString(getpid()) + "_" : "") + pureName;
         int out;
         int attempts = 10;
         do {
@@ -1153,7 +1154,6 @@ std::string TRestTools::DownloadRemoteFile(const string& url) {
             return "";
         }
     }
-    return "";
 }
 
 ///////////////////////////////////////////////