Add event info

alja · Apr 17, 2020 · b43ce77 · b43ce77
1 parent a526d1b
commit b43ce77
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/CmsNanoAod.cxx b/CmsNanoAod.cxx
@@ -25,6 +25,11 @@ MamaCollection::MamaCollection(const std::string& name) :
 
       s = "^"; s += name; s += "_(.*)$";
       m_data_re.Reset(s, "o", 1);
+
+      if (name == "EventInfo")
+      {
+         m_data_re.Reset("^(run|luminosityBlock|event)$", "o", 1);
+      }
    }
 
 MamaCollection::~MamaCollection()
@@ -95,6 +100,11 @@ void MamaCollection::goto_event(Long64_t ev)
       m_num_ents_leaf->GetBranch()->GetEntry(ev);
       m_entries = ((UInt_t*)(m_num_ents_leaf->GetValuePointer()))[0];
    }
+   else
+   {
+      // If not a vector (does not have nClass branch), assume single entry.
+      m_entries = 1;
+   }
 
    for (auto &m : m_members)
    {

diff --git a/cms_nano_aod_bootstrap.C b/cms_nano_aod_bootstrap.C
@@ -2,6 +2,9 @@ nanoaod::Event * g_event = nullptr;
 
 void cms_nano_aod_config_event(nanoaod::Event *e)
 {
+   // EventInfo is handled specifically to pick up run / lumi / event ids.
+   e->RegisterMamaCollection("EventInfo");
+
    e->RegisterMamaCollection("Electron");
    e->RegisterMamaCollection("Jet");
    e->RegisterMamaCollection("MET");