FourVectorHLT.cc

Go to the documentation of this file.

#include "FWCore/Framework/interface/EDAnalyzer.h"
#include "FWCore/Framework/interface/Run.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/HLTReco/interface/TriggerObject.h"
#include "DataFormats/HLTReco/interface/TriggerEvent.h"
#include "DQMServices/Core/interface/MonitorElement.h"

// see header file for information. 
#include "FourVectorHLT.h"

using namespace edm;

FourVectorHLT::FourVectorHLT(const edm::ParameterSet& iConfig)
{
  LogDebug("FourVectorHLT") << "constructor...." ;

  dbe_ = Service < DQMStore > ().operator->();
  if ( ! dbe_ ) {
    LogWarning("Status") << "unable to get DQMStore service?";
  }
  if (iConfig.getUntrackedParameter < bool > ("DQMStore", false)) {
    dbe_->setVerbose(0);
  }
  
  
  dirname_="HLT/FourVectorHLT" ;
  
  if (dbe_ != nullptr ) {
    LogDebug("Status") << "Setting current directory to " << dirname_;
    dbe_->setCurrentFolder(dirname_);
  }
  
  
  // plotting paramters
  ptMin_ = iConfig.getUntrackedParameter<double>("ptMin",0.);
  ptMax_ = iConfig.getUntrackedParameter<double>("ptMax",200.);
  nBins_ = iConfig.getUntrackedParameter<unsigned int>("Nbins",50);
  
  plotAll_ = iConfig.getUntrackedParameter<bool>("plotAll", false);

  // this is the list of paths to look at.
  std::vector<edm::ParameterSet> filters = 
    iConfig.getParameter<std::vector<edm::ParameterSet> >("filters");
  for(std::vector<edm::ParameterSet>::iterator 
    filterconf = filters.begin() ; filterconf != filters.end(); 
      filterconf++) {
    std::string me  = filterconf->getParameter<std::string>("name");
    int objectType = filterconf->getParameter<unsigned int>("type");
    float ptMin = filterconf->getUntrackedParameter<double>("ptMin");
    float ptMax = filterconf->getUntrackedParameter<double>("ptMax");
    hltPaths_.push_back(PathInfo(me, objectType, ptMin, ptMax));
  }
  if ( !hltPaths_.empty() && plotAll_) {
    // these two ought to be mutually exclusive....
    LogWarning("Configuration") << "Using both plotAll and a list. "
      "list will be ignored." ;
    hltPaths_.clear();
  }
  triggerSummaryLabel_ = 
    iConfig.getParameter<edm::InputTag>("triggerSummaryLabel");

  //set Token(-s)
  triggerSummaryToken_ = consumes<trigger::TriggerEvent>(iConfig.getParameter<edm::InputTag>("triggerSummaryLabel"));
}


FourVectorHLT::~FourVectorHLT()
{
 
   // do anything here that needs to be done at desctruction time
   // (e.g. close files, deallocate resources etc.)

}


//
// member functions
//

// ------------ method called to for each event  ------------
void
FourVectorHLT::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup)
{
  using namespace edm;
  using namespace trigger;
  ++nev_;
  LogDebug("Status")<< "analyze" ;
  
  edm::Handle<TriggerEvent> triggerObj;
  iEvent.getByToken(triggerSummaryToken_, triggerObj);
  if(!triggerObj.isValid()) { 
    edm::LogInfo("Status") << "Summary HLT object (TriggerEvent) not found, "
      "skipping event"; 
    return;
  }
  
  const trigger::TriggerObjectCollection & toc(triggerObj->getObjects());

  if ( plotAll_ ) {
    for ( size_t ia = 0; ia < triggerObj->sizeFilters(); ++ ia) {
      std::string fullname = triggerObj->filterTag(ia).encode();
      // the name can have in it the module label as well as the process and
      // other labels - strip 'em
      std::string name;
      size_t p = fullname.find_first_of(':');
      if ( p != std::string::npos) {
    name = fullname.substr(0, p);
      }
      else {
    name = fullname;
      }

      LogDebug("Parameter") << "filter " << ia << ", full name = " << fullname
                << ", p = " << p 
                << ", abbreviated = " << name ;
      
      PathInfoCollection::iterator pic =  hltPaths_.find(name);
      if ( pic == hltPaths_.end() ) {
    // doesn't exist - add it
    MonitorElement *et(nullptr), *eta(nullptr), *phi(nullptr), *etavsphi(nullptr);
    
    std::string histoname(name+"_et");
    LogDebug("Status") << "new histo with name "<<  histoname ;
    dbe_->setCurrentFolder(dirname_);
    std::string title(name+" E_{T}");
    et =  dbe_->book1D(histoname.c_str(),
              title.c_str(),nBins_, 0, 100);
      
    histoname = name+"_eta";
    title = name+" #eta";
    eta =  dbe_->book1D(histoname.c_str(),
               title.c_str(),nBins_,-2.7,2.7);
      
    histoname = name+"_phi";
    title = name+" #phi";
    phi =  dbe_->book1D(histoname.c_str(),
               title.c_str(),nBins_,-3.14,3.14);
      
      
    histoname = name+"_etaphi";
    title = name+" #eta vs #phi";
    etavsphi =  dbe_->book2D(histoname.c_str(),
                title.c_str(),
                nBins_,-2.7,2.7,
                nBins_,-3.14, 3.14);
      
    // no idea how to get the bin boundries in this mode
    PathInfo e(name,0, et, eta, phi, etavsphi, ptMin_,ptMax_);
    hltPaths_.push_back(e);  
    pic = hltPaths_.begin() + hltPaths_.size()-1;
      }
      const trigger::Keys & k = triggerObj->filterKeys(ia);
      for (trigger::Keys::const_iterator ki = k.begin(); ki !=k.end(); ++ki ) {
    LogDebug("Parameters") << "pt, eta, phi = " 
                   << toc[*ki].pt() << ", "
                   << toc[*ki].eta() << ", "
                   << toc[*ki].phi() ;
    pic->getEtHisto()->Fill(toc[*ki].pt());
    pic->getEtaHisto()->Fill(toc[*ki].eta());
    pic->getPhiHisto()->Fill(toc[*ki].phi());
    pic->getEtaVsPhiHisto()->Fill(toc[*ki].eta(), toc[*ki].phi());
      }  

    }

  }
  else { // not plotAll_
    for(PathInfoCollection::iterator v = hltPaths_.begin();
    v!= hltPaths_.end(); ++v ) {
      const int index = triggerObj->filterIndex(v->getName());
      if ( index >= triggerObj->sizeFilters() ) {
    continue; // not in this event
      }
      LogDebug("Status") << "filling ... " ;
      const trigger::Keys & k = triggerObj->filterKeys(index);
      for (trigger::Keys::const_iterator ki = k.begin(); ki !=k.end(); ++ki ) {
    v->getEtHisto()->Fill(toc[*ki].pt());
    v->getEtaHisto()->Fill(toc[*ki].eta());
    v->getPhiHisto()->Fill(toc[*ki].phi());
    v->getEtaVsPhiHisto()->Fill(toc[*ki].eta(), toc[*ki].phi());
      }  
    }
  }
}


// -- method called once each job just before starting event loop  --------
void 
FourVectorHLT::beginJob()
{
  nev_ = 0;
  DQMStore *dbe = nullptr;
  dbe = Service<DQMStore>().operator->();
  
  if (dbe) {
    dbe->setCurrentFolder(dirname_);
    dbe->rmdir(dirname_);
  }
  
  
  if (dbe) {
    dbe->setCurrentFolder(dirname_);

    if ( ! plotAll_ ) {
      for(PathInfoCollection::iterator v = hltPaths_.begin();
      v!= hltPaths_.end(); ++v ) {
    MonitorElement *et, *eta, *phi, *etavsphi=nullptr;
    std::string histoname(v->getName()+"_et");
    std::string title(v->getName()+" E_t");
    et =  dbe->book1D(histoname.c_str(),
              title.c_str(),nBins_,
              v->getPtMin(),
              v->getPtMax());
      
    histoname = v->getName()+"_eta";
    title = v->getName()+" #eta";
    eta =  dbe->book1D(histoname.c_str(),
               title.c_str(),nBins_,-2.7,2.7);

    histoname = v->getName()+"_phi";
    title = v->getName()+" #phi";
    phi =  dbe->book1D(histoname.c_str(),
               histoname.c_str(),nBins_,-3.14,3.14);
 

    histoname = v->getName()+"_etaphi";
    title = v->getName()+" #eta vs #phi";
    etavsphi =  dbe->book2D(histoname.c_str(),
                title.c_str(),
                nBins_,-2.7,2.7,
                nBins_,-3.14, 3.14);
      
    v->setHistos( et, eta, phi, etavsphi);
      } 
    } // ! plotAll_ - for plotAll we discover it during the event
  }
}

// - method called once each job just after ending the event loop  ------------
void 
FourVectorHLT::endJob() 
{
   LogInfo("Status") << "endJob: analyzed " << nev_ << " events";
   return;
}


// BeginRun
void FourVectorHLT::beginRun(const edm::Run& run, const edm::EventSetup& c)
{
  LogDebug("Status") << "beginRun, run " << run.id();
}

void FourVectorHLT::endRun(const edm::Run& run, const edm::EventSetup& c)
{
  LogDebug("Status") << "endRun, run " << run.id();
}