HLTMuonValidator.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:     HLTMuonValidator
// Class:       HLTMuonValidator
// 

//
// Jason Slaunwhite and Jeff Klukas
//
//

// system include files
#include <memory>
#include <iostream>

// user include files
#include "HLTriggerOffline/Muon/interface/HLTMuonPlotter.h"

#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDAnalyzer.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"


#include "HLTrigger/HLTcore/interface/HLTConfigProvider.h"
#include "FWCore/ServiceRegistry/interface/Service.h"


#include "TFile.h"
#include "TDirectory.h"
#include "TPRegexp.h"
#include "boost/tuple/tuple.hpp"






class HLTMuonValidator : public edm::EDAnalyzer {

public:

  explicit HLTMuonValidator(const edm::ParameterSet&);

private:

  // Analyzer Methods
  virtual void beginJob() override;
  virtual void beginRun(const edm::Run &, const edm::EventSetup &) override;
  virtual void analyze(const edm::Event &, const edm::EventSetup &) override;
  virtual void endRun(const edm::Run &, const edm::EventSetup &) override;
  virtual void endJob() override;

  // Extra Methods
  std::vector<std::string> moduleLabels(std::string);
  std::vector<std::string> stepLabels(const std::vector<std::string>&);

  // Input from Configuration File
  edm::ParameterSet pset_;
  std::string hltProcessName_;
  std::vector<std::string> hltPathsToCheck_;

  // Member Variables
  std::vector<HLTMuonPlotter> analyzers_;
  HLTConfigProvider hltConfig_;
  boost::tuple<
    edm::EDGetTokenT<trigger::TriggerEventWithRefs>,
    edm::EDGetTokenT<reco::GenParticleCollection>,
    edm::EDGetTokenT<reco::MuonCollection> > myTokens_;

  // Access to the DQM
  DQMStore * dbe_;

};




using namespace std;
using namespace edm;
using namespace reco;
using namespace trigger;

typedef vector<string> vstring;




HLTMuonValidator::HLTMuonValidator(const ParameterSet& pset) :
  pset_(pset),
  hltProcessName_(pset.getParameter<string>("hltProcessName")),
  hltPathsToCheck_(pset.getParameter<vstring>("hltPathsToCheck"))
{

  myTokens_ = HLTMuonPlotter::getTokens(pset_, consumesCollector());

}



vector<string> 
HLTMuonValidator::moduleLabels(string path) {

  vector<string> modules = hltConfig_.moduleLabels(path);
  vector<string>::iterator iter = modules.begin();

  while (iter != modules.end())
    if (iter->find("Filtered") == string::npos) 
      iter = modules.erase(iter);
    else
      ++iter;

  return modules;

}

vector<string>
HLTMuonValidator::stepLabels(const vector<string>& modules) {
  vector<string> steps(1, "All");
  for (size_t i = 0; i < modules.size(); i++) {
    if (modules[i].find("IsoFiltered") != string::npos) {
      if (modules[i].find("L3") != string::npos)
        steps.push_back("L3Iso");
      else
        steps.push_back("L2Iso");
    }
    else if (modules[i].find("L3") != string::npos)
      steps.push_back("L3");
    else if (modules[i].find("L2") != string::npos)
      steps.push_back("L2");
    else if (modules[i].find("L1") != string::npos)
      steps.push_back("L1");
    else
      return vector<string>();
  }
  if (steps.size() < 2 || steps[1] != "L1")
    return vector<string>();
  return steps;

}


void 
HLTMuonValidator::beginRun(const edm::Run & iRun, 
                         const edm::EventSetup & iSetup) 
{  
  // Initialize hltConfig
  bool changedConfig;
  if (!hltConfig_.init(iRun, iSetup, hltProcessName_, changedConfig)) {
    LogError("HLTMuonVal") << "Initialization of HLTConfigProvider failed!!"; 
    return;
  }
  
  // Get the set of trigger paths we want to make plots for
  set<string> hltPaths;
  for (size_t i = 0; i < hltPathsToCheck_.size(); i++) {
    TPRegexp pattern(hltPathsToCheck_[i]);
    for (size_t j = 0; j < hltConfig_.triggerNames().size(); j++)
      if (TString(hltConfig_.triggerNames()[j]).Contains(pattern))
        hltPaths.insert(hltConfig_.triggerNames()[j]);
  }
  
  // Initialize the analyzers
  analyzers_.clear();
  set<string>::iterator iPath;
  for (iPath = hltPaths.begin(); iPath != hltPaths.end(); iPath++) {
    
    string path = * iPath;
    string shortpath = path;
    if (path.rfind("_v") < path.length())
      shortpath = path.substr(0, path.rfind("_v"));
    
    vector<string> labels = moduleLabels(path);
    vector<string> steps = stepLabels(labels);
    
    if (labels.size() > 0 && steps.size() > 0) {
      HLTMuonPlotter analyzer(pset_, shortpath, labels, steps, myTokens_);
      analyzers_.push_back(analyzer);
    }
  }
  
  // Call the beginRun (which books all the histograms)
  vector<HLTMuonPlotter>::iterator iter;
  for (iter = analyzers_.begin(); iter != analyzers_.end(); ++iter) {
    iter->beginRun(iRun, iSetup);
  }
  
}

void
HLTMuonValidator::analyze(const Event& iEvent, 
                                     const EventSetup& iSetup)
{

  vector<HLTMuonPlotter>::iterator iter;
  for (iter = analyzers_.begin(); iter != analyzers_.end(); ++iter) {
    iter->analyze(iEvent, iSetup);
  }

}



void 
HLTMuonValidator::beginJob()
{
}



void 
HLTMuonValidator::endRun(const edm::Run & iRun, 
                                    const edm::EventSetup& iSetup)
{

  // vector<HLTMuonPlotter>::iterator iter;
  // for (iter = analyzers_.begin(); iter != analyzers_.end(); ++iter) {
  //   iter->endRun(iRun, iSetup);
  // }

}



void 
HLTMuonValidator::endJob()
{
}



//define this as a plug-in
DEFINE_FWK_MODULE(HLTMuonValidator);