TriggerOutputBranches.cc

Go to the documentation of this file.

#include "PhysicsTools/NanoAOD/plugins/TriggerOutputBranches.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/Registry.h"

#include <iostream>

void 
TriggerOutputBranches::updateTriggerNames(TTree & tree, const edm::TriggerNames & names, const edm::TriggerResults & triggers) 
{  
   std::vector<std::string> newNames(triggers.getTriggerNames());
   if(newNames.empty()) {
       for(unsigned int j=0;j<triggers.size();j++) {
           newNames.push_back(names.triggerName(j));
       }
   }

   for(auto & existing : m_triggerBranches)
   {
       existing.idx = -1;    // reset all triggers as not found and zero buffer
       existing.buffer = 0;  // reset all triggers as not found and zero buffer
       for(unsigned int j=0;j<newNames.size();j++) {
     std::string name=newNames[j]; // no const & as it will be modified below!
     std::size_t vfound = name.rfind("_v");
     if (vfound!=std::string::npos && (name.compare(0, 3, "HLT") == 0 || name.compare(0, 2, "L1") == 0)){
           name.replace(vfound,name.size()-vfound,"");
     }
     if(name==existing.name) existing.idx=j;
       }
   }
   // Find new ones
   for(unsigned int j=0;j<newNames.size();j++) {
       std::string name=newNames[j]; // no const & as it will be modified below!
       std::size_t vfound = name.rfind("_v");
       if (vfound!=std::string::npos && (name.compare(0, 3, "HLT") == 0 || name.compare(0, 2, "L1") == 0)){
           name.replace(vfound,name.size()-vfound,"");
       }
       bool found=false;
       if(name.compare(0,3,"HLT")==0 || name.compare(0,4,"Flag")==0 || name.compare(0,2,"L1")==0 ){
           for(auto & existing : m_triggerBranches) {if(name==existing.name) found=true;}
           if(!found){
            NamedBranchPtr nb(name,
                  std::string("Trigger/flag bit (process: ") + m_processName +
                  ")");  //FIXME: If the title can be updated we can use it to list the versions _v* that were seen in this file
                uint8_t backFillValue=0;
        bool found_duplicate = verifyBranchUniqueName(tree, nb.name);
        std::string brname = nb.name + (found_duplicate ? (std::string("_p") + m_processName) : "");
        nb.branch = tree.Branch(brname.c_str(), &backFillValue, (brname + "/O").c_str());
                nb.branch->SetTitle(nb.title.c_str());
                nb.idx=j;
                m_triggerBranches.push_back(nb);
                for(size_t i=0;i<m_fills;i++) nb.branch->Fill(); // Back fill
           }
       }
   }
}

edm::TriggerNames TriggerOutputBranches::triggerNames(const edm::TriggerResults triggerResults){
    edm::pset::Registry* psetRegistry = edm::pset::Registry::instance();
    edm::ParameterSet const* pset=nullptr;
    if (nullptr!=(pset=psetRegistry->getMapped(triggerResults.parameterSetID()))) {

         if (pset->existsAs<std::vector<std::string> >("@trigger_paths", true)) {
            edm::TriggerNames triggerNames(*pset);

            // This should never happen
            if (triggerNames.size() != triggerResults.size()) {
               throw cms::Exception("LogicError")
                  << "edm::EventBase::triggerNames_ Encountered vector\n"
                     "of trigger names and a TriggerResults object with\n"
                     "different sizes.  This should be impossible.\n"
                     "Please send information to reproduce this problem to\n"
                     "the edm developers.\n";
            }
            return triggerNames;
         }
      }
    return edm::TriggerNames();
}

void TriggerOutputBranches::fill(const edm::EventForOutput &iEvent,TTree & tree) 
{
    edm::Handle<edm::TriggerResults> handle;
    iEvent.getByToken(m_token, handle);
    const edm::TriggerResults & triggers = *handle;
    const edm::TriggerNames &names = triggerNames(triggers);

    if(m_lastRun!=iEvent.id().run()) {
        m_lastRun=iEvent.id().run();
        updateTriggerNames(tree,names,triggers);
    }
    for (auto & pair : m_triggerBranches) fillColumn<uint8_t>(pair, triggers);
    m_fills++; 
}

bool TriggerOutputBranches::verifyBranchUniqueName(TTree& tree, std::string name) const {
  auto const branches = tree.GetListOfBranches();
  for (int i = 0; i < branches->GetEntries(); i++) {
    if (name == std::string(branches->At(i)->GetName())) {
      edm::LogWarning("TriggerOutputBranches")
          << "Found a branch with name " << std::string(branches->At(i)->GetName()) << " already present with title "
          << std::string(branches->At(i)->GetTitle()) << ": will add suffix _p" << m_processName
          << " to the new branch.\n";
      return true;
    }
  }
  return false;
}