CandidateTriggerObjectProducer.cc

Go to the documentation of this file.

#include "FWCore/Common/interface/TriggerNames.h"
#include "FWCore/Common/interface/TriggerResultsByName.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "DataFormats/Candidate/interface/CandidateFwd.h"
#include "DataFormats/Candidate/interface/Candidate.h"
#include "DataFormats/Candidate/interface/LeafCandidate.h"
#include "HLTrigger/HLTcore/interface/HLTConfigProvider.h"

#include "TString.h"
#include "TRegexp.h"

#include <cassert>

#include "CommonTools/TriggerUtils/interface/CandidateTriggerObjectProducer.h"


//
// constructors and destructor
//
CandidateTriggerObjectProducer::CandidateTriggerObjectProducer(const edm::ParameterSet& ps) :
  triggerResultsToken_(consumes<edm::TriggerResults>(ps.getParameter<edm::InputTag>("triggerResults"))),
  triggerEventTag_(ps.getParameter<edm::InputTag>("triggerEvent")),
  triggerEventToken_(consumes<trigger::TriggerEvent>(triggerEventTag_)),
  triggerName_(ps.getParameter<std::string>("triggerName")),
  hltPrescaleProvider_(ps, consumesCollector(), *this)
{

//   cout << "Trigger Object Producer:" << endl
//        << "   TriggerResultsTag = " << triggerResultsTag_.encode() << endl
//        << "   TriggerEventTag = " << triggerEventTag_.encode() << endl;

  produces<reco::CandidateCollection>();

}

CandidateTriggerObjectProducer::~CandidateTriggerObjectProducer()
{
}

//
// member functions
//
void
CandidateTriggerObjectProducer::beginRun(const edm::Run& iRun, edm::EventSetup const& iSetup)
{
  using namespace edm;

  bool changed(false);
  if(!hltPrescaleProvider_.init(iRun,iSetup,triggerEventTag_.process(),changed) ){
    edm::LogError( "CandidateTriggerObjectProducer" ) <<
      "Error! Can't initialize HLTConfigProvider";
    throw cms::Exception("HLTConfigProvider::init() returned non 0");
  }

  return;

}

// ------------ method called to produce the data  ------------
void
CandidateTriggerObjectProducer::produce(edm::Event & iEvent, const edm::EventSetup & iSetup)
{
  using namespace edm;
  using namespace reco;
  using namespace trigger;

  std::auto_ptr<reco::CandidateCollection> coll( new reco::CandidateCollection );

  // get event products
  iEvent.getByToken(triggerResultsToken_,triggerResultsHandle_);
  if (!triggerResultsHandle_.isValid()) {
     edm::LogError( "CandidateTriggerObjectProducer" ) << "CandidateTriggerObjectProducer::analyze: Error in getting TriggerResults product from Event!" ;
    return;
  }
  iEvent.getByToken(triggerEventToken_,triggerEventHandle_);
  if (!triggerEventHandle_.isValid()) {
    edm::LogError( "CandidateTriggerObjectProducer" ) << "CandidateTriggerObjectProducer::analyze: Error in getting TriggerEvent product from Event!" ;
    return;
  }

  HLTConfigProvider const&  hltConfig = hltPrescaleProvider_.hltConfigProvider();

  // sanity check
  //   std::cout << hltConfig.size() << std::endl;
  //   std::cout << triggerResultsHandle_->size() << std::endl;
  assert(triggerResultsHandle_->size()==hltConfig.size());

  const unsigned int n(hltConfig.size());
  std::vector<std::string> activeHLTPathsInThisEvent = hltConfig.triggerNames();
  std::map<std::string, bool> triggerInMenu;
  std::map<std::string, bool> triggerUnprescaled;

  for (std::vector<std::string>::const_iterator iHLT = activeHLTPathsInThisEvent.begin();
       iHLT != activeHLTPathsInThisEvent.end(); ++iHLT)
    {
      //matching with regexp filter name. More than 1 matching filter is allowed
      if (TString(*iHLT).Contains(TRegexp(TString(triggerName_))))
     {
       triggerInMenu[*iHLT] = true;
       const std::pair<int,int> prescales(hltPrescaleProvider_.prescaleValues(iEvent,iSetup,*iHLT));
       if (prescales.first * prescales.second == 1)
         triggerUnprescaled[*iHLT] = true;
     }
     }

  for (std::map<std::string, bool>::const_iterator iMyHLT = triggerInMenu.begin();
       iMyHLT != triggerInMenu.end(); ++iMyHLT)
    {
      //using only unprescaled triggers
      if (!(iMyHLT->second && triggerUnprescaled[iMyHLT->first]))
    continue;
      const unsigned int triggerIndex(hltConfig.triggerIndex(iMyHLT->first));

      assert(triggerIndex==iEvent.triggerNames(*triggerResultsHandle_).triggerIndex(iMyHLT->first));

      // abort on invalid trigger name
      if (triggerIndex>=n) {
    edm::LogError( "CandidateTriggerObjectProducer" ) << "CandidateTriggerObjectProducer::analyzeTrigger: path "
         << triggerName_ << " - not found!" ;
    return;
      }

      // modules on this trigger path
      //      const unsigned int m(hltConfig.size(triggerIndex));
      const std::vector<std::string>& moduleLabels(hltConfig.saveTagsModules(triggerIndex));

      // Results from TriggerResults product
      if (!(triggerResultsHandle_->wasrun(triggerIndex)) ||
      !(triggerResultsHandle_->accept(triggerIndex)) ||
      (triggerResultsHandle_->error(triggerIndex)))
    {
      continue;
    }

//       const unsigned int moduleIndex(triggerResultsHandle_->index(triggerIndex));

//       assert (moduleIndex<m);

      // Results from TriggerEvent product - Looking only on last filter since trigger is accepted
      for (unsigned int imodule=0;imodule<moduleLabels.size();++imodule)
    {
      const std::string& moduleLabel(moduleLabels[imodule]);
      const std::string  moduleType(hltConfig.moduleType(moduleLabel));
      //Avoiding L1 seeds
      if (moduleType.find("Level1GTSeed") != std::string::npos)
        continue;
      // check whether the module is packed up in TriggerEvent product
      const unsigned int filterIndex(triggerEventHandle_->filterIndex(InputTag(moduleLabel,"",triggerEventTag_.process())));
      if (filterIndex<triggerEventHandle_->sizeFilters()) {
        //      std::cout << " 'L3' filter in slot " << imodule << " - label/type " << moduleLabel << "/" << moduleType << std::endl;
        const Vids& VIDS (triggerEventHandle_->filterIds(filterIndex));
        const Keys& KEYS(triggerEventHandle_->filterKeys(filterIndex));
        const size_type nI(VIDS.size());
        const size_type nK(KEYS.size());
        assert(nI==nK);
        const size_type n(std::max(nI,nK));
        //      std::cout << "   " << n  << " accepted 'L3' objects found: " << std::endl;
        const TriggerObjectCollection& TOC(triggerEventHandle_->getObjects());
        for (size_type i=0; i!=n; ++i) {
          const TriggerObject& TO(TOC[KEYS[i]]);
          coll->push_back(reco::LeafCandidate( 0, TO.particle().p4(), reco::Particle::Point( 0., 0., 0. ), TO.id() ));
//        std::cout << "   " << i << " " << VIDS[i] << "/" << KEYS[i] << ": "
//          << TO.id() << " " << TO.pt() << " " << TO.eta() << " " << TO.phi() << " " << TO.mass()
//          << std::endl;
        }
      }
    }
    }


  iEvent.put(coll);
  return;
}

#include "FWCore/Framework/interface/MakerMacros.h"

DEFINE_FWK_MODULE(CandidateTriggerObjectProducer);