HLTHiggsPlotter.cc

Go to the documentation of this file.

#include "FWCore/ServiceRegistry/interface/Service.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/Candidate/interface/CandMatchMap.h"
#include "DataFormats/MuonReco/interface/Muon.h"
#include "DataFormats/MuonReco/interface/MuonFwd.h"

#include "DataFormats/HLTReco/interface/TriggerEvent.h"

#include "HLTriggerOffline/Higgs/interface/HLTHiggsPlotter.h"
#include "HLTriggerOffline/Higgs/interface/HLTHiggsSubAnalysis.h"
#include "HLTriggerOffline/Higgs/src/EVTColContainer.cc"

#include "TPRegexp.h"


#include<set>
#include<cctype>

HLTHiggsPlotter::HLTHiggsPlotter(const edm::ParameterSet & pset,
                                 const std::string & hltPath,
                                 const std::vector<unsigned int> & objectsType,
                                 DQMStore * dbe) :
    _hltPath(hltPath),
    _hltProcessName(pset.getParameter<std::string>("hltProcessName")),
    _objectsType(std::set<unsigned int>(objectsType.begin(),objectsType.end())),
    _nObjects(objectsType.size()),
    _parametersEta(pset.getParameter<std::vector<double> >("parametersEta")),
    _parametersPhi(pset.getParameter<std::vector<double> >("parametersPhi")),
    _parametersTurnOn(pset.getParameter<std::vector<double> >("parametersTurnOn")),
    _dbe(dbe)
{
  for(std::set<unsigned int>::iterator it = _objectsType.begin();
      it != _objectsType.end(); ++it)
  {
    // Some parameters extracted from the .py
    std::string objStr = EVTColContainer::getTypeString( *it );
    _cutMinPt[*it] = pset.getParameter<double>( std::string(objStr+"_cutMinPt").c_str() );
    _cutMaxEta[*it] = pset.getParameter<double>( std::string(objStr+"_cutMaxEta").c_str() );
  }
}

HLTHiggsPlotter::~HLTHiggsPlotter()
{
}


void HLTHiggsPlotter::beginJob() 
{
}



void HLTHiggsPlotter::beginRun(const edm::Run & iRun,
                               const edm::EventSetup & iSetup)
{
  for (std::set<unsigned int>::iterator it = _objectsType.begin(); 
      it != _objectsType.end(); ++it)
  {
    std::vector<std::string> sources(2);
    sources[0] = "gen";
    sources[1] = "rec";

    const std::string objTypeStr = EVTColContainer::getTypeString(*it);
      
    for (size_t i = 0; i < sources.size(); i++) 
    {
      std::string source = sources[i];
      bookHist(source, objTypeStr, "Eta");
      bookHist(source, objTypeStr, "Phi");
      bookHist(source, objTypeStr, "MaxPt1");
      bookHist(source, objTypeStr, "MaxPt2");
    }
  }
}

void HLTHiggsPlotter::analyze(const bool & isPassTrigger,
                              const std::string & source,
                              const std::vector<MatchStruct> & matches)
{
  if ( !isPassTrigger )
  {
    return;
  }
  std::map<unsigned int,int> countobjects;
  // Initializing the count of the used object
  for(std::set<unsigned int>::iterator co = _objectsType.begin();
      co != _objectsType.end(); ++co)
  {
    countobjects[*co] = 0;
  }
    
  int counttotal = 0;
  const int totalobjectssize2 = 2*countobjects.size();
  // Fill the histos if pass the trigger (just the two with higher pt)
  for (size_t j = 0; j < matches.size(); ++j)
  {
    // Is this object owned by this trigger? If not we are not interested...
    if ( _objectsType.find( matches[j].objType) == _objectsType.end() )
    {
      continue;
    }

    const unsigned int objType = matches[j].objType;
    const std::string objTypeStr = EVTColContainer::getTypeString(matches[j].objType);
        
    float pt  = matches[j].pt;
    float eta = matches[j].eta;
    float phi = matches[j].phi;
    this->fillHist(isPassTrigger,source,objTypeStr,"Eta",eta);
    this->fillHist(isPassTrigger,source,objTypeStr,"Phi",phi);
    if ( countobjects[objType] == 0 )
    {
      this->fillHist(isPassTrigger,source,objTypeStr,"MaxPt1",pt);
      // Filled the high pt ...
      ++(countobjects[objType]);
      ++counttotal;
    }
    else if ( countobjects[objType] == 1 )
    {
      this->fillHist(isPassTrigger,source,objTypeStr,"MaxPt2",pt);
      // Filled the second high pt ...
      ++(countobjects[objType]);
      ++counttotal;
    }
    else
    {
      if ( counttotal == totalobjectssize2 ) 
      {
        break;
      }
    }               
  }
}


void HLTHiggsPlotter::bookHist(const std::string & source, 
                               const std::string & objType,
                               const std::string & variable)
{
  std::string sourceUpper = source; 
  sourceUpper[0] = std::toupper(sourceUpper[0]);
  std::string name = source + objType + variable + "_" + _hltPath;
  TH1F * h = 0;

  if (variable.find("MaxPt") != std::string::npos) 
  {
    std::string desc = (variable == "MaxPt1") ? "Leading" : "Next-to-Leading";
    std::string title = "pT of " + desc + " " + sourceUpper + " " + objType + " "
        "where event pass the "+ _hltPath;
    const size_t nBins = _parametersTurnOn.size() - 1;
    float * edges = new float[nBins + 1];
    for(size_t i = 0; i < nBins + 1; i++)
    {
      edges[i] = _parametersTurnOn[i];
    }
    h = new TH1F(name.c_str(), title.c_str(), nBins, edges);
    delete [] edges;
  }
  else 
  {
    std::string symbol = (variable == "Eta") ? "#eta" : "#phi";
    std::string title  = symbol + " of " + sourceUpper + " " + objType + " "+
        "where event pass the "+ _hltPath;
    std::vector<double> params = (variable == "Eta") ? _parametersEta : _parametersPhi;

    int    nBins = (int)params[0];
    double min   = params[1];
    double max   = params[2];
    h = new TH1F(name.c_str(), title.c_str(), nBins, min, max);
  }
  h->Sumw2();
  _elements[name] = _dbe->book1D(name, h);
  delete h;
}

void HLTHiggsPlotter::fillHist(const bool & passTrigger,
                               const std::string & source, 
                               const std::string & objType,
                               const std::string & variable,
                               const float & value )
{
  std::string sourceUpper = source; 
  sourceUpper[0] = toupper(sourceUpper[0]);
  std::string name = source + objType + variable + "_" + _hltPath;

  _elements[name]->Fill(value);
}