/data/refman/pasoursint/CMSSW_4_4_5_patch3/src/Validation/MuonIsolation/src/MuIsoValidation.cc

Go to the documentation of this file.

// -*- C++ -*-
// MuIsoValidation.cc
// Package:    Muon Isolation Validation
// Class:      MuIsoValidation
// 
/*


 Description: Muon Isolation Validation class

 Implementation: This code will accept a data set and generate plots of
        various quantities relevent to the Muon Isolation module. We will 
        be using the IsoDeposit class, *not* the MuonIsolation struct.
         
        The sequence of events is... 
                * initalize statics (which variables to plot, axis limtis, etc.)
                * run over events
                        * for each event, run over the muons in that event
                                *record IsoDeposit data
                * transfer data to histograms, profile plots
                * save histograms to a root file
                
        Easy-peasy.
        
*/
//
// Original Author:  "C. Jess Riedel", UC Santa Barbara
//         Created:  Tue Jul 17 15:58:24 CDT 2007
//

//Class header file
#include "Validation/MuonIsolation/interface/MuIsoValidation.h"

//System included files
#include <memory>
#include <string>
#include <typeinfo>
#include <utility>

//Root included files
#include "TH1.h"
#include "TH2.h"
#include "TProfile.h"

//Event framework included files
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"

//Other included files
#include "DataFormats/TrackReco/interface/Track.h"

//Using declarations
using std::vector;
using std::pair;
using std::string;



//
//-----------------Constructors---------------------
//
MuIsoValidation::MuIsoValidation(const edm::ParameterSet& iConfig)
{
  
  //  rootfilename = iConfig.getUntrackedParameter<string>("rootfilename"); // comment out for inclusion
  requireCombinedMuon = iConfig.getUntrackedParameter<bool>("requireCombinedMuon");
  dirName = iConfig.getParameter<std::string>("directory");
  //  subDirName = iConfig.getParameter<std::string>("@module_label");
  
  //  dirName += subDirName;
  
  //--------Initialize tags-------
  Muon_Tag = iConfig.getUntrackedParameter<edm::InputTag>("Global_Muon_Label");
  
  //-------Initialize counters----------------
  nEvents = 0;
  nIncMuons = 0;   
  nCombinedMuons = 0;
  
  InitStatics();
  
  //Set up DAQ
  dbe = 0;
  dbe = edm::Service<DQMStore>().operator->();
  
  //------"allocate" space for the data vectors-------
  
  /*
    h_1D        is a 2D vector with indices [var][muon#]
    cd_plots    is a 2D vector with indices [var][muon#]  
    h_2D        is a 3D vector with indices [var][var][muon#]
    p_2D        is a 3D vector with indices [var][var][muon#]
  */
  //NOTE:the total number of muons and events is initially unknown, 
  //       so that dimension is not initialized. Hence, theMuonData
  //     needs no resizing.
  
  h_1D.resize    (NUM_VARS);
  cd_plots.resize(NUM_VARS);
  //  h_2D.resize(NUM_VARS, vector<MonitorElement*>     (NUM_VARS));
  p_2D.resize(NUM_VARS, vector<MonitorElement*>(NUM_VARS));
  
  dbe->cd();
}

//
//----------Destructor-----------------
//
MuIsoValidation::~MuIsoValidation(){
  
  //Deallocate memory
  
}

//
//------------Methods-----------------------
//
void MuIsoValidation::InitStatics(){
  
  //-----------Initialize primatives-----------
  S_BIN_WIDTH = 1.0;//in GeV
  L_BIN_WIDTH = 2.0;//in GeV
  LOG_BINNING_ENABLED = 1;
  NUM_LOG_BINS = 15;
  LOG_BINNING_RATIO = 1.1;//ratio by which each bin is wider than the last for log binning
  //i.e.  bin widths are (x), (r*x), (r^2*x), ..., (r^(nbins)*x)
  
  
  //-------Initialize Titles---------
  title_sam = "";//"[Sample b-jet events] ";
  title_cone = "";//" [in R=0.3 IsoDeposit Cone]";
  //The above two pieces of info will be printed on the title of the whole page,
  //not for each individual histogram
  title_cd = "C.D. of ";
  
  //-------"Allocate" memory for vectors
  main_titles.resize(NUM_VARS);
  axis_titles.resize(NUM_VARS);
  names.resize(NUM_VARS);
  param.resize(NUM_VARS, vector<double>(3) );
  isContinuous.resize(NUM_VARS);
  
  //-----Titles of the plots-----------
  main_titles[0 ] = "Total Tracker Momentum";
  main_titles[1 ] = "Total EM Cal Energy";
  main_titles[2 ] = "Total Had Cal Energy";
  main_titles[3 ] = "Total HO Cal Energy";
  main_titles[4 ] = "Number of Tracker Tracks";
  main_titles[5 ] = "Number of Jets around Muon";
  main_titles[6 ] = "Tracker p_{T} within veto cone";
  main_titles[7 ] = "EM E_{T} within veto cone";
  main_titles[8 ] = "Had E_{T} within veto cone";
  main_titles[9 ] = "HO E_{T} within veto cone";
  main_titles[10] = "Muon p_{T}";
  main_titles[11] = "Muon #eta";
  main_titles[12] = "Muon #phi";
  main_titles[13] = "Average Momentum per Track ";
  main_titles[14] = "Weighted Energy";

  //------Titles on the X or Y axis------------
  axis_titles[0 ] = "#Sigma p_{T}   (GeV)";
  axis_titles[1 ] = "#Sigma E_{T}^{EM}   (GeV)";
  axis_titles[2 ] = "#Sigma E_{T}^{Had}   (GeV)";
  axis_titles[3 ] = "#Sigma E_{T}^{HO}   (GeV)";
  axis_titles[4 ] = "N_{Tracks}";
  axis_titles[5 ] = "N_{Jets}";
  axis_titles[6 ] = "#Sigma p_{T,veto} (GeV)";
  axis_titles[7 ] = "#Sigma E_{T,veto}^{EM}   (GeV)";
  axis_titles[8 ] = "#Sigma E_{T,veto}^{Had}   (GeV)";
  axis_titles[9 ] = "#Sigma E_{T,veto}^{HO}   (GeV)";
  axis_titles[10] = "p_{T,#mu} (GeV)";
  axis_titles[11] = "#eta_{#mu}";
  axis_titles[12] = "#phi_{#mu}";
  axis_titles[13] = "#Sigma p_{T} / N_{Tracks} (GeV)";
  axis_titles[14] = "(1.5) X #Sigma E_{T}^{EM} + #Sigma E_{T}^{Had}";

  //-----------Names given for the root file----------
  names[0 ] = "sumPt";
  names[1 ] = "emEt";
  names[2 ] = "hadEt";
  names[3 ] = "hoEt";
  names[4 ] = "nTracks";
  names[5 ] = "nJets";
  names[6 ] = "trackerVetoPt";
  names[7 ] = "emVetoEt";
  names[8 ] = "hadVetoEt";
  names[9 ] = "hoVetoEt";
  names[10] = "muonPt";
  names[11] = "muonEta";
  names[12] = "muonPhi";
  names[13] = "avgPt";
  names[14] = "weightedEt";

  //----------Parameters for binning of histograms---------
  //param[var][0] is the number of bins
  //param[var][1] is the low edge of the low bin
  //param[var][2] is the high edge of the high bin
  //
  // maximum value------,
  //                    |
  //                    V                  
  param[0 ][0]= (int)( 20.0/S_BIN_WIDTH); param[0 ][1]=  0.0; param[0 ][2]= param[0 ][0]*S_BIN_WIDTH;
  param[1 ][0]= (int)( 20.0/S_BIN_WIDTH); param[1 ][1]=  0.0; param[1 ][2]= param[1 ][0]*S_BIN_WIDTH;
  param[2 ][0]= (int)( 20.0/S_BIN_WIDTH); param[2 ][1]=  0.0; param[2 ][2]= param[2 ][0]*S_BIN_WIDTH;
  param[3 ][0]=                       20; param[3 ][1]=  0.0; param[3 ][2]=                      2.0;
  param[4 ][0]=                       16; param[4 ][1]= -0.5; param[4 ][2]=         param[4 ][0]-0.5;
  param[5 ][0]=                        4; param[5 ][1]= -0.5; param[5 ][2]=         param[5 ][0]-0.5;
  param[6 ][0]= (int)( 40.0/S_BIN_WIDTH); param[6 ][1]=  0.0; param[6 ][2]= param[6 ][0]*S_BIN_WIDTH;
  param[7 ][0]=                       20; param[7 ][1]=  0.0; param[7 ][2]=                     10.0;
  param[8 ][0]= (int)( 20.0/S_BIN_WIDTH); param[8 ][1]=  0.0; param[8 ][2]= param[8 ][0]*S_BIN_WIDTH;
  param[9 ][0]=                       20; param[9 ][1]=  0.0; param[9 ][2]=                      5.0;
  param[10][0]= (int)( 40.0/S_BIN_WIDTH); param[10][1]=  0.0; param[10][2]= param[10][0]*S_BIN_WIDTH;
  param[11][0]=                       24; param[11][1]= -2.4; param[11][2]=                      2.4;
  param[12][0]=                       32; param[12][1]= -3.2; param[12][2]=                      3.2;
  param[13][0]= (int)( 15.0/S_BIN_WIDTH); param[13][1]=  0.0; param[13][2]= param[13][0]*S_BIN_WIDTH;
  param[14][0]= (int)( 20.0/S_BIN_WIDTH); param[14][1]=  0.0; param[14][2]= param[14][0]*S_BIN_WIDTH;

  //--------------Is the variable continuous (i.e. non-integer)?-------------
  //---------(Log binning will only be used for continuous variables)--------
  isContinuous[0 ] = 1;
  isContinuous[1 ] = 1;
  isContinuous[2 ] = 1;
  isContinuous[3 ] = 1;
  isContinuous[4 ] = 0;
  isContinuous[5 ] = 0;
  isContinuous[6 ] = 1;
  isContinuous[7 ] = 1;
  isContinuous[8 ] = 1;
  isContinuous[9 ] = 1;
  isContinuous[10] = 1;
  isContinuous[11] = 1;
  isContinuous[12] = 1;
  isContinuous[13] = 1;
  isContinuous[14] = 1;

}


// ------------ method called for each event  ------------
void MuIsoValidation::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup){
  
  ++nEvents;
  edm::LogInfo("Tutorial") << "\nInvestigating event #" << nEvents<<"\n";
  
  // Get Muon Collection 
  edm::Handle<edm::View<reco::Muon> > muonsHandle; // 
  iEvent.getByLabel(Muon_Tag, muonsHandle);
  
  //Fill event entry in histogram of number of muons
  edm::LogInfo("Tutorial") << "Number of Muons: " << muonsHandle->size();
  theMuonData = muonsHandle->size();
  h_nMuons->Fill(theMuonData);
  
  //Fill historgams concerning muon isolation 
  uint iMuon=0;
  dbe->setCurrentFolder(dirName.c_str());
  for (MuonIterator muon = muonsHandle->begin(); muon != muonsHandle->end(); ++muon, ++iMuon ) {
    ++nIncMuons;
    if (requireCombinedMuon) {
      if (muon->combinedMuon().isNull()) continue;
    }
    ++nCombinedMuons;
    RecordData(muon);
    FillHistos();
  }
  dbe->cd();
  
}

//---------------Record data for a signle muon's data---------------------
void MuIsoValidation::RecordData(MuonIterator muon){
  
  
  theData[0] = muon->isolationR03().sumPt;
  theData[1] = muon->isolationR03().emEt;
  theData[2] = muon->isolationR03().hadEt;
  theData[3] = muon->isolationR03().hoEt;

  theData[4] = muon->isolationR03().nTracks;
  theData[5] = muon->isolationR03().nJets;
  theData[6] = muon->isolationR03().trackerVetoPt;
  theData[7] = muon->isolationR03().emVetoEt;
  theData[8] = muon->isolationR03().hadVetoEt;
  theData[9] = muon->isolationR03().hoVetoEt;
  
  theData[10] = muon->pt();
  theData[11] = muon->eta();
  theData[12] = muon->phi();

  // make sure nTracks != 0 before filling this one
  if (theData[4] != 0) theData[13] = (double)theData[0] / (double)theData[4];
  else theData[13] = -99;

  theData[14] = 1.5 * theData[1] + theData[2];

}

// ------------ method called once each job just before starting event loop  ------------
void 
MuIsoValidation::beginJob()
{
  
  edm::LogInfo("Tutorial") << "\n#########################################\n\n"
                           << "Lets get started! " 
                           << "\n\n#########################################\n";
  dbe->setCurrentFolder(dirName.c_str());
  InitHistos();
  dbe->cd();
  
}

// ------------ method called once each job just after ending the event loop  ------------
void 
MuIsoValidation::endJob() {
  
  // check if ME still there (and not killed by MEtoEDM for memory saving)
  if( dbe )
    {
      // check existence of first histo in the list
      if (! dbe->get(dirName+"/nMuons")) return;
    }
  else
    return;

  edm::LogInfo("Tutorial") << "\n#########################################\n\n"
                           << "Total Number of Events: " << nEvents
                           << "\nTotal Number of Muons: " << nIncMuons
                           << "\n\n#########################################\n"
                           << "\nInitializing Histograms...\n";
  
  edm::LogInfo("Tutorial") << "\nIntializing Finished.  Filling...\n";
  NormalizeHistos();
  edm::LogInfo("Tutorial") << "\nFilled.  Saving...\n";
  //  dbe->save(rootfilename); // comment out for incorporation
  edm::LogInfo("Tutorial") << "\nSaved.  Peace, homie, I'm out.\n";
  
}

void MuIsoValidation::InitHistos(){
  
  //---initialize number of muons histogram---
  h_nMuons = dbe->book1D("nMuons", title_sam + "Number of Muons", 20, 0., 20.);
  h_nMuons->setAxisTitle("Number of Muons",XAXIS);
  h_nMuons->setAxisTitle("Fraction of Events",YAXIS);
  
  
  //---Initialize 1D Histograms---
  for(int var = 0; var < NUM_VARS; var++){
    h_1D[var] = dbe->book1D(
                            names[var], 
                            title_sam + main_titles[var] + title_cone, 
                            (int)param[var][0], 
                            param[var][1], 
                            param[var][2]
                            );
    cd_plots[var] = dbe->book1D(
                                names[var] + "_cd", 
                                title_sam + title_cd + main_titles[var] + title_cone, 
                                (int)param[var][0], 
                                param[var][1], 
                                param[var][2]
                                );
    
    h_1D[var]->setAxisTitle(axis_titles[var],XAXIS);
    h_1D[var]->setAxisTitle("Fraction of Muons",YAXIS);
    GetTH1FromMonitorElement(h_1D[var])->Sumw2();
    
    cd_plots[var]->setAxisTitle(axis_titles[var],XAXIS);
    cd_plots[var]->setAxisTitle("Fraction of Muons",YAXIS);
    GetTH1FromMonitorElement(cd_plots[var])->Sumw2();
    
  }//Finish 1D
  
  //---Initialize 2D Histograms---
  for(int var1 = 0; var1 < NUM_VARS; var1++){
    for(int var2 = 0; var2 < NUM_VARS; var2++){
      if(var1 == var2) continue;
      
      /*      h_2D[var1][var2] = dbe->book2D(
              names[var1] + "_" + names[var2] + "_s",
              //title is in "y-var vs. x-var" format
              title_sam + main_titles[var2] + " <vs> " + main_titles[var1] + title_cone, 
              (int)param[var1][0],
              param[var1][1],
              param[var1][2],
              (int)param[var2][0],
              param[var2][1],
              param[var2][2]
              );
      */
      //Monitor elements is weird and takes y axis parameters as well
      //as x axis parameters for a 1D profile plot
      p_2D[var1][var2] = dbe->bookProfile(
                                          names[var1] + "_" + names[var2],
                                          title_sam + main_titles[var2] + " <vs> " + main_titles[var1] + title_cone,
                                          (int)param[var1][0],
                                          param[var1][1],
                                          param[var1][2],
                                          (int)param[var2][0], //documentation says this is disregarded
                                          param[var2][1],      //does this do anything?
                                          param[var2][2],      //does this do anything?
                                          " "                  //profile errors = spread/sqrt(num_datums)
                                          );
      
      if(LOG_BINNING_ENABLED && isContinuous[var1]){
        Double_t * bin_edges = new Double_t[NUM_LOG_BINS+1];
        // nbins+1 because there is one more edge than there are bins
        MakeLogBinsForProfile(bin_edges, param[var1][1], param[var1][2]);
        GetTProfileFromMonitorElement(p_2D[var1][var2])->
          SetBins(NUM_LOG_BINS, bin_edges);
        delete[] bin_edges;
      }
      /*      h_2D[var1][var2]->setAxisTitle(axis_titles[var1],XAXIS);
              h_2D[var1][var2]->setAxisTitle(axis_titles[var2],YAXIS);
              GetTH2FromMonitorElement(h_2D[var1][var2])->Sumw2();
      */
      
      p_2D[var1][var2]->setAxisTitle(axis_titles[var1],XAXIS);
      p_2D[var1][var2]->setAxisTitle(axis_titles[var2],YAXIS);
      //                        GetTProfileFromMonitorElement(p_2D[var1][var2])->Sumw2();
    }
  }//Finish 2D
  
  
  
  //avg pT not defined for zero tracks.
  //MonitorElement is inflxible and won't let me change the
  //number of bins!  I guess all I'm doing here is changing 
  //range of the x axis when it is printed, not the actual
  //bins that are filled
  p_2D[4][9]->setAxisRange(0.5,15.5,XAXIS);
  
}

void MuIsoValidation::MakeLogBinsForProfile(Double_t* bin_edges,        const double min, 
                                            const double max){
  
  const double &r = LOG_BINNING_RATIO;
  const int &nbins = NUM_LOG_BINS;
  
  const double first_bin_width = (r > 1.0) ? //so we don't divide by zero
    (max - min)*(1-r)/(1-pow(r,nbins)) :
    (max - min)/nbins;
  
  bin_edges[0] = min;
  bin_edges[1] = min + first_bin_width;
  for(int n = 2; n<nbins; ++n){
    bin_edges[n] = bin_edges[n-1] + (bin_edges[n-1] - bin_edges[n-2])*r;
  }
  bin_edges[nbins] = max;
}

void MuIsoValidation::NormalizeHistos() {
  for(int var=0; var<NUM_VARS; var++){   
    //turn cd_plots into CDF's
    //underflow -> bin #0.  overflow -> bin #(nbins+1)
    //0th bin doesn't need changed
    
    double entries = GetTH1FromMonitorElement(h_1D[var])->GetEntries();
    if (entries==0)continue;
    int n_max = int(param[var][0])+1;
    for(int n=1; n<=n_max; ++n){
      cd_plots[var]->setBinContent(n, cd_plots[var]->getBinContent(n) + cd_plots[var]->getBinContent(n-1)); //Integrate.
    }
    //----normalize------
    if (requireCombinedMuon) {
      GetTH1FromMonitorElement(h_1D[var])->Scale(1./entries);
      GetTH1FromMonitorElement(cd_plots[var])->Scale(1./entries);
    }
    else {
      GetTH1FromMonitorElement(h_1D[var])->Scale(1./entries);
      GetTH1FromMonitorElement(cd_plots[var])->Scale(1./entries);
    }
  }
}

void MuIsoValidation::FillHistos() {
  
  int overFlowBin;
  double overFlow = 0;
  
  //----------Fill 1D histograms---------------
  for(int var=0; var<NUM_VARS; var++){  
    h_1D[var]->Fill(theData[var]);
    cd_plots[var]->Fill(theData[var]);//right now, this is a regular PDF (just like h_1D)
    if (theData[var] > param[var][2]) {
      // fill the overflow bin
      overFlowBin = (int) param[var][0] + 1;
      overFlow = GetTH1FromMonitorElement(h_1D[var])->GetBinContent(overFlowBin);
      GetTH1FromMonitorElement(h_1D[var])->SetBinContent(overFlowBin, overFlow + 1);
    }
  }//Finish 1D
  
  //----------Fill 2D histograms---------------
  for(int var1=0; var1<NUM_VARS; ++var1){
    for(int var2=0; var2<NUM_VARS; ++var2){
      if(var1 == var2) continue;
      //change below to regular int interating!
      //      h_2D[var1][var2]->Fill(theData[var1], theData[var2]);
      p_2D[var1][var2]->Fill(theData[var1], theData[var2]);
    }
  }//Finish 2D
}

TH1* MuIsoValidation::GetTH1FromMonitorElement(MonitorElement* me) {
  return me->getTH1();
}

TH2* MuIsoValidation::GetTH2FromMonitorElement(MonitorElement* me) {
  return me->getTH2F();
}

TProfile* MuIsoValidation::GetTProfileFromMonitorElement(MonitorElement* me) {
  return me->getTProfile();
}


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