Merge pull request #453 from rest-for-physics/issue452

Adding TRestEvent ptr while opening input file

Author: juanangp

macros/REST_OpenInputFile.C

@@ -1,40 +1,51 @@
-TRestRun* run0 = nullptr;
-TRestAnalysisTree* ana_tree0 = nullptr;
-TTree* ev_tree0 = nullptr;
-TRestDataSet* dSet0 = nullptr;
-std::map<std::string, TRestMetadata*> metadata0;
+TRestRun* run = nullptr;
+TRestAnalysisTree* ana_tree = nullptr;
+TTree* ev_tree = nullptr;
+TRestDataSet* dSet = nullptr;
 
 void REST_OpenInputFile(const std::string& fileName) {
     if (TRestTools::isRunFile(fileName)) {
-        printf("\n%s\n", "REST processed file identified. It contains a valid TRestRun.");
-        run0 = new TRestRun(fileName);
-        printf("\nAttaching TRestRun %s as run0...\n", fileName.c_str());
-        ana_tree0 = run0->GetAnalysisTree();
-        printf("\nAttaching TRestAnalysisTree as ana_tree0...\n");
-        ev_tree0 = run0->GetEventTree();
-        printf("\nAttaching event tree as ev_tree0...\n");
-        std::map<std::string, int> metanames;
-        for (auto& [name, meta] : metadata0) delete meta;
-        metadata0.clear();
-        for (int n = 0; n < run0->GetNumberOfMetadata(); n++) {
-            std::string metaName = run0->GetMetadataNames()[n];
+        printf("\n%s\n\n", "REST processed file identified. It contains a valid TRestRun.");
+        run = new TRestRun(fileName);
+        printf("\nAttaching TRestRun %s as run...\n", fileName.c_str());
+        ana_tree = run->GetAnalysisTree();
+        printf("Attaching TRestAnalysisTree as ana_tree...\n");
+        ev_tree = run->GetEventTree();
+        printf("Attaching event tree as ev_tree...\n");
+        std::string eventType = run->GetInputEvent()->ClassName();
+        std::string evcmd = Form("%s* ev = (%s*)run->GetInputEvent();", eventType.c_str(), eventType.c_str());
+        gROOT->ProcessLine(evcmd.c_str());
+        run->GetEntry(0);
+        printf("Attaching input event %s as ev...\n", eventType.c_str());
+        for (int n = 0; n < run->GetNumberOfMetadata(); n++) {
+            if (n == 0) printf("Attaching Metadata classes:\n");
+            std::string metaName = run->GetMetadataNames()[n];
             if (metaName.find("Historic") != string::npos) continue;
-            TRestMetadata* md = run0->GetMetadata(metaName);
-            metadata0[metaName] = md;
-            printf("\nAttaching Metadata class %s as metadata0[\"%s\"]...\n", md->ClassName(),
-                   metaName.c_str());
+            TRestMetadata* md = run->GetMetadata(metaName);
+            if (md == nullptr) {
+                printf("\nERROR Cannot get metadata pointer for class %s and name%s\n", md->ClassName(),
+                       metaName.c_str());
+                continue;
+            }
+            std::string mName = Replace(metaName, " ", "");
+            mName = Replace(mName, ".", "_");
+            std::string mdcmd = Form("%s* %s = (%s*)run->GetMetadata(\"%s\");", md->ClassName(),
+                                     mName.c_str(), md->ClassName(), metaName.c_str());
+            gROOT->ProcessLine(mdcmd.c_str());
+            printf("- %s as %s\n", md->ClassName(), mName.c_str());
         }
+        printf("\n");
 
     } else if (TRestTools::isDataSet(fileName)) {
         printf("\n%s\n", "REST dataset file identified. It contains a valid TRestDataSet.");
-        printf("\nImporting dataset %s as `dSet0`\n", fileName.c_str());
+        printf("\nImporting dataset %s as `dSet`\n", fileName.c_str());
         printf("\n%s\n", "The dataset is ready. You may now access the dataset using:");
-        printf("\n%s\n", " - dSet0->PrintMetadata()");
-        printf("%s\n", " - dSet0->GetDataFrame().GetColumnNames()");
-        printf("%s\n\n", " - dSet0->GetTree()->GetEntries()");
-        if (dSet0) delete dSet0;
-        dSet0 = new TRestDataSet();
-        dSet0->Import(fileName);
+        printf("\n%s\n", " - dSet->PrintMetadata()");
+        printf("%s\n", " - dSet->GetDataFrame().GetColumnNames()");
+        printf("%s\n\n", " - dSet->GetTree()->GetEntries()");
+        if (dSet) delete dSet;
+        dSet = new TRestDataSet();
+        dSet->Import(fileName);
     } else {
         printf("\n%s is not a valid TRestRun or TRestDataSet\n", fileName.c_str());
     }

source/framework/core/inc/TRestRun.h

@@ -167,6 +167,8 @@ class TRestRun : public TRestMetadata {
     inline TTree* GetEventTree() const { return fEventTree; }
     inline Int_t GetInputFileNumber() const { return fFileProcess == nullptr ? fInputFileNames.size() : 1; }
 
+    std::vector<std::string> GetEventTypesList();
+
     TRestMetadata* GetMetadata(const TString& name, TFile* file = nullptr);
     TRestMetadata* GetMetadataClass(const TString& type, TFile* file = nullptr);
     std::vector<std::string> GetMetadataNames();

source/framework/core/src/TRestRun.cxx

@@ -887,8 +887,25 @@ Int_t TRestRun::GetNextEvent(TRestEvent* targetEvent, TRestAnalysisTree* targetT
     return 0;
 }
 
+///////////////////////////////////////////////
+/// \brief It returns a list of available event types inside the file
+///
+std::vector<std::string> TRestRun::GetEventTypesList() {
+    std::vector<std::string> list;
+    TObjArray* branches = GetEventTree()->GetListOfBranches();
+    for (int i = 0; i <= branches->GetLast(); i++) {
+        auto branch = (TBranch*)branches->At(i);
+        if (((std::string)branch->GetName()).find("EventBranch") != string::npos) {
+            std::string eventType = Replace((string)branch->GetName(), "Branch", "");
+            list.emplace_back(eventType);
+        }
+    }
+    return list;
+}
+
 ///////////////////////////////////////////////
 /// \brief Calls GetEntry() for both AnalysisTree and EventTree
+///
 void TRestRun::GetEntry(Long64_t entry) {
     if (entry >= GetEntries()) {
         RESTWarning << "TRestRun::GetEntry. Entry requested out of limits" << RESTendl;

source/framework/tools/src/TRestTools.cxx

@@ -1137,7 +1137,10 @@ int TRestTools::DownloadRemoteFile(string remoteFile, string localFile) {
             return 0;
         }
     } else {
-        RESTError << "unknown protocol!" << RESTendl;
+        if (!TRestTools::fileExists(remoteFile)) {
+            RESTWarning << "Trying to download: " << remoteFile << RESTendl;
+            RESTWarning << "Unknown protocol!" << RESTendl;
+        }
     }
 
     return -1;