Merge pull request #323 from michaelmackenzie/TriggerInfo

Add Trigger rooutil object

Author: AndrewEdmonds11

rooutil/examples/PrintEvents.C

@@ -259,8 +259,8 @@ void PrintEvents(std::string filename) {
     }
 
     // trigger branches
-    for (const auto& pair : event.trigNameMap) {
-      std::cout << pair.first << ": " << event.triginfo._triggerArray[pair.second] << std::endl;
+    for (const auto& pair : event.trigger.NameToIndexMap()) {
+      std::cout << pair.first << ": " << event.trigger.Fired(pair.second) << std::endl;
     }
 
   }

rooutil/inc/Event.hh

@@ -41,6 +41,7 @@
 #include "EventNtuple/rooutil/inc/Track.hh"
 #include "EventNtuple/rooutil/inc/CrvCoinc.hh"
 #include "EventNtuple/rooutil/inc/CaloCluster.hh"
+#include "EventNtuple/rooutil/inc/Trigger.hh"
 
 #include "TChain.h"
 
@@ -50,16 +51,7 @@ namespace rooutil {
       CheckForBranch(ntuple, "evtinfo", &this->evtinfo);
       CheckForBranch(ntuple, "hitcount", &this->hitcount);
       CheckForBranch(ntuple, "crvsummary", &this->crvsummary);
-      const auto& branches = ntuple->GetListOfBranches();
-      int i_trig_branch = 0;
-      for (const auto& branch : *branches) {
-        std::string brname = branch->GetName();
-        if (brname.substr(0, 5) == "trig_") {
-          ntuple->SetBranchAddress(brname.c_str(), &this->triginfo._triggerArray[i_trig_branch]);
-          trigNameMap.insert({brname, i_trig_branch});
-          i_trig_branch++;
-        }
-      }
+      AddTriggerInfo(ntuple);
 
       CheckForBranch(ntuple, "trk", &this->trk);
       CheckForBranch(ntuple, "trksegs", &this->trksegs);
@@ -104,6 +96,24 @@ namespace rooutil {
       return true;
     }
 
+    // Add trigger branches and store the path name information
+    void AddTriggerInfo(TChain* ntuple) {
+      trigger.SetTrigInfo(&triginfo); // pointer to the underlying trigger data read in event-by-event
+
+      // trigger branches are named "trig_<trigger path name>" --> look for these
+      const auto& branches = ntuple->GetListOfBranches();
+      int i_trig_branch = 0;
+      for (const auto& branch : *branches) {
+        std::string brname = branch->GetName();
+        if (brname.substr(0, 5) == "trig_") {
+          ntuple->SetBranchAddress(brname.c_str(), &this->triginfo._triggerArray[i_trig_branch]);
+          const std::string trigname = brname.substr(5); // name of the trigger path
+          trigger.AssignIndex(i_trig_branch, trigname); // map the array index to the path name
+          i_trig_branch++;
+        }
+      }
+    }
+
     void Update(bool debug = false) {
       if (debug) { std::cout << "Event::Update(): Clearing previous Tracks... " << std::endl; }
       tracks.clear();
@@ -300,6 +310,7 @@ namespace rooutil {
     mu2e::CrvSummaryReco* crvsummary = nullptr;
     mu2e::CrvSummaryMC* crvsummarymc = nullptr;
     mu2e::TrigInfo triginfo; // not a pointer because we give the address of array elements inside this
+    Trigger trigger; // contains additional trigger information
 
     std::vector<mu2e::TrkInfo>* trk = nullptr;
     std::vector<mu2e::TrkInfoMC>* trkmc = nullptr;
@@ -331,9 +342,6 @@ namespace rooutil {
     std::vector<mu2e::CrvPlaneInfoMC>* crvcoincsmcplane = nullptr;
 
     std::vector<std::vector<mu2e::SimInfo>>* trkmcsim = nullptr;
-
-    // Need to keep track of trigger name to element in triginfo
-    std::map<std::string, unsigned int> trigNameMap;
   };
 } // namespace rooutil
 #endif

rooutil/inc/RooUtil.hh

@@ -144,8 +144,8 @@ namespace rooutil {
 
       if(event->trkmcsim) { output_ntuple->Branch("trkmcsim", event->trkmcsim); }
 
-      for (const auto& pair : event->trigNameMap) {
-        output_ntuple->Branch(pair.first.c_str(), &event->triginfo._triggerArray[pair.second]);
+      for (const auto& pair : event->trigger.NameToIndexMap()) {
+        output_ntuple->Branch(("trig_" + pair.first).c_str(), &event->triginfo._triggerArray[pair.second]);
       }
 
       // Write out histograms from input to output

rooutil/inc/Trigger.hh

@@ -0,0 +1,66 @@
+#ifndef rooutil_Trigger_hh_
+#define rooutil_Trigger_hh_
+
+#include <functional>
+#include <map>
+#include "EventNtuple/inc/TrigInfo.hh"
+
+namespace rooutil {
+  struct Trigger {
+    Trigger(mu2e::TrigInfo* trig = nullptr)
+      : _trig(trig) {
+
+    }
+
+    void Update(bool debug = false) {
+    }
+
+    void SetTrigInfo(const mu2e::TrigInfo* trig) { _trig = trig; }
+
+    void AssignIndex(const int index, const std::string& name) {
+      _indexToName[index] = name;
+      _nameToIndex[name]  = index;
+    }
+
+    int GetIndex(const std::string& name) const {
+      if(_nameToIndex.count(name) == 0) return -1;
+      return _nameToIndex.at(name);
+    }
+
+    bool Fired(const int index) const {
+      if(!_trig) return false;
+      if(index >= int(mu2e::TrigInfo::ntrig_)) return false;
+      if(_indexToName.count(index) == 0) return false; // not in the trigger menu
+      return _trig->_triggerArray[index];
+    }
+
+    bool Fired(const std::string& name) const {
+      if(_nameToIndex.count(name) == 0) return false;
+      return Fired(_nameToIndex.at(name));
+    }
+
+    // Check if any trigger with this pattern string was fired
+    bool FiredByTag(const std::string pattern) const {
+      bool fired = false;
+      for(const auto& itr : NameToIndexMap()) {
+        const std::string name = itr.first;
+        if(name.find(pattern) != std::string::npos) fired |= Fired(itr.second);
+        if(fired) break;
+      }
+      return fired;
+    }
+
+    // Data accessors
+    const mu2e::TrigInfo*             TrigInfo      () const { return _trig; }
+    const std::map<std::string, int>& NameToIndexMap() const { return _nameToIndex; }
+    const std::map<int, std::string>& IndexToNameMap() const { return _indexToName; }
+
+    // Pointer to the data
+    const mu2e::TrigInfo* _trig = nullptr;
+    std::map<int, std::string> _indexToName;
+    std::map<std::string, int> _nameToIndex;
+  };
+
+  typedef std::function<bool(Trigger&)> TriggerCut;
+} // namespace rooutil
+#endif

rooutil/inc/common_cuts.hh

@@ -286,10 +286,9 @@ bool from_start_process(Track& track, mu2e::ProcessCode::enum_type proc) { // Tr
 
 //+ Trigger Cut
 bool passes_trigger(Event& event, std::string trigname) { // true if the event passed the named trigger
-  if (trigname.substr(0, 5) != "trig_") { // we need to add the prefix to get the correct element
-    trigname = "trig_" + trigname;
+  if (trigname.substr(0, 5) == "trig_") { // drop this prefix
+    trigname = trigname.substr(5);
   }
-  const auto i_trig_element = event.trigNameMap[trigname];
-  return event.triginfo._triggerArray[i_trig_element];
+  return event.trigger.Fired(trigname);
 }
 #endif

validation/create_val_file_rooutil.C

@@ -507,7 +507,7 @@ void create_val_file_rooutil(std::string filename, std::string outfilename) {
   TH1F* h_triginfo = new TH1F("h_triginfo", "", mu2e::TrigInfo::ntrig_,0,mu2e::TrigInfo::ntrig_);
   // Grab the first event to name the bins
   const auto& evt = util.GetEvent(0);
-  for (const auto pair : evt.trigNameMap) {
+  for (const auto pair : evt.trigger.NameToIndexMap()) {
     h_triginfo->GetXaxis()->SetBinLabel(pair.second+1, pair.first.c_str());
   }