Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2021-02-14 12:48:02

0001 
0002 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0003 
0004 #include "DataFormats/EcalRecHit/interface/EcalRecHitCollections.h"
0005 #include "DataFormats/EcalDetId/interface/EBDetId.h"
0006 #include "DataFormats/DetId/interface/DetId.h"
0007 #include "CondFormats/EcalObjects/interface/EcalIntercalibConstants.h"
0008 #include "CondFormats/DataRecord/interface/EcalIntercalibConstantsRcd.h"
0009 #include "Calibration/Tools/interface/calibXMLwriter.h"
0010 #include "Calibration/Tools/interface/CalibrationCluster.h"
0011 #include "Calibration/Tools/interface/CalibElectron.h"
0012 #include "Calibration/Tools/interface/HouseholderDecomposition.h"
0013 #include "Calibration/Tools/interface/MinL3Algorithm.h"
0014 #include "Calibration/EcalCalibAlgos/interface/ZeePlots.h"
0015 #include "Calibration/EcalCalibAlgos/interface/ZeeKinematicTools.h"
0016 #include "DataFormats/EgammaCandidates/interface/Electron.h"
0017 #include "DataFormats/EgammaReco/interface/SuperCluster.h"
0018 #include "DataFormats/TrackReco/interface/Track.h"
0019 #include "DataFormats/TrackReco/interface/TrackFwd.h"
0020 #include "DataFormats/TrackReco/interface/TrackExtraFwd.h"
0021 #include "TFile.h"
0022 #include "TH1.h"
0023 #include "TH2.h"
0024 #include "TF1.h"
0025 #include "TRandom.h"
0026 
0027 #include <iostream>
0028 #include <string>
0029 #include <stdexcept>
0030 #include <vector>
0031 
0032 #include "DataFormats/EgammaCandidates/interface/GsfElectron.h"
0033 
0034 ZeePlots::ZeePlots(const char* fileName) {
0035   fileName_ = fileName;
0036   file_ = new TFile(fileName_, "RECREATE");
0037 }
0038 
0039 ZeePlots::~ZeePlots() {
0040   file_->Close();
0041 
0042   delete file_;
0043 }
0044 
0045 //========================================================================
0046 
0047 void ZeePlots::openFile() { file_->cd(); }
0048 //========================================================================
0049 
0050 void ZeePlots::bookZMCHistograms() {
0051   file_->cd();
0052 
0053   h1_gen_ZMass_ = new TH1F("gen_ZMass", "Generated Z mass", 200, 0., 150.);
0054   h1_gen_ZMass_->SetXTitle("gen_ZMass (GeV)");
0055   h1_gen_ZMass_->SetYTitle("events");
0056 
0057   h1_gen_ZEta_ = new TH1F("gen_ZEta", "Eta of gen Z", 200, -6., 6.);
0058   h1_gen_ZEta_->SetXTitle("#eta");
0059   h1_gen_ZEta_->SetYTitle("events");
0060 
0061   h1_gen_ZPhi_ = new TH1F("gen_ZPhi", "Phi of gen Z", 200, -4., 4.);
0062   h1_gen_ZPhi_->SetXTitle("#phi");
0063   h1_gen_ZPhi_->SetYTitle("events");
0064 
0065   h1_gen_ZRapidity_ = new TH1F("gen_ZRapidity", "Rapidity of gen Z", 200, -6., 6.);
0066   h1_gen_ZRapidity_->SetXTitle("Y");
0067   h1_gen_ZRapidity_->SetYTitle("events");
0068 
0069   h1_gen_ZPt_ = new TH1F("gen_ZPt", "Pt of gen Z", 200, 0., 100.);
0070   h1_gen_ZPt_->SetXTitle("p_{T} (GeV/c)");
0071   h1_gen_ZPt_->SetYTitle("events");
0072 }
0073 
0074 void ZeePlots::bookZHistograms() {
0075   file_->cd();
0076 
0077   h1_reco_ZEta_ = new TH1F("reco_ZEta", "Eta of reco Z", 200, -6., 6.);
0078   h1_reco_ZEta_->SetXTitle("#eta");
0079   h1_reco_ZEta_->SetYTitle("events");
0080 
0081   h1_reco_ZTheta_ = new TH1F("reco_ZTheta", "Theta of reco Z", 200, 0., 4.);
0082   h1_reco_ZTheta_->SetXTitle("#theta");
0083   h1_reco_ZTheta_->SetYTitle("events");
0084 
0085   h1_reco_ZRapidity_ = new TH1F("reco_ZRapidity", "Rapidity of reco Z", 200, -6., 6.);
0086   h1_reco_ZRapidity_->SetXTitle("Y");
0087   h1_reco_ZRapidity_->SetYTitle("events");
0088 
0089   h1_reco_ZPhi_ = new TH1F("reco_ZPhi", "Phi of reco Z", 100, -4., 4.);
0090   h1_reco_ZPhi_->SetXTitle("#phi");
0091   h1_reco_ZPhi_->SetYTitle("events");
0092 
0093   h1_reco_ZPt_ = new TH1F("reco_ZPt", "Pt of reco Z", 200, 0., 100.);
0094   h1_reco_ZPt_->SetXTitle("p_{T} (GeV/c)");
0095   h1_reco_ZPt_->SetYTitle("events");
0096 }
0097 
0098 //========================================================================
0099 
0100 void ZeePlots::fillZInfo(std::pair<calib::CalibElectron*, calib::CalibElectron*> myZeeCandidate) {
0101   h1_reco_ZEta_->Fill(ZeeKinematicTools::calculateZEta(myZeeCandidate));
0102   h1_reco_ZTheta_->Fill(ZeeKinematicTools::calculateZTheta(myZeeCandidate));
0103   h1_reco_ZRapidity_->Fill(ZeeKinematicTools::calculateZRapidity(myZeeCandidate));
0104   h1_reco_ZPhi_->Fill(ZeeKinematicTools::calculateZPhi(myZeeCandidate));
0105   h1_reco_ZPt_->Fill(ZeeKinematicTools::calculateZPt(myZeeCandidate));
0106 }
0107 
0108 //========================================================================
0109 
0110 void ZeePlots::writeZHistograms() {
0111   file_->cd();
0112 
0113   h1_reco_ZEta_->Write();
0114   h1_reco_ZTheta_->Write();
0115   h1_reco_ZRapidity_->Write();
0116   h1_reco_ZPhi_->Write();
0117   h1_reco_ZPt_->Write();
0118 }
0119 
0120 //========================================================================
0121 
0122 void ZeePlots::writeMCZHistograms() {
0123   file_->cd();
0124 
0125   h1_gen_ZRapidity_->Write();
0126   h1_gen_ZPt_->Write();
0127   h1_gen_ZPhi_->Write();
0128 }
0129 
0130 //========================================================================
0131 
0132 void ZeePlots::fillZMCInfo(const HepMC::GenEvent* myGenEvent) {
0133   file_->cd();
0134 
0135   for (HepMC::GenEvent::particle_const_iterator p = myGenEvent->particles_begin(); p != myGenEvent->particles_end();
0136        ++p) {  //loop over MC particles
0137 
0138     if ((*p)->pdg_id() == 23 && (*p)->status() == 2) {
0139       h1_gen_ZMass_->Fill((*p)->momentum().m());
0140       h1_gen_ZEta_->Fill((*p)->momentum().eta());
0141 
0142       float genZ_Y =
0143           0.5 * log(((*p)->momentum().e() + (*p)->momentum().pz()) / ((*p)->momentum().e() - (*p)->momentum().pz()));
0144 
0145       h1_gen_ZRapidity_->Fill(genZ_Y);
0146       h1_gen_ZPt_->Fill((*p)->momentum().perp());
0147       h1_gen_ZPhi_->Fill((*p)->momentum().phi());
0148     }
0149   }  //end loop over MC particles
0150 
0151   return;
0152 }
0153 
0154 //========================================================================
0155 
0156 void ZeePlots::bookEleMCHistograms() {
0157   file_->cd();
0158 
0159   h1_mcEle_Energy_ = new TH1F("mcEleEnergy", "mc EleEnergy", 300, 0., 300.);
0160   h1_mcEle_Energy_->SetXTitle("E (GeV)");
0161   h1_mcEle_Energy_->SetYTitle("events");
0162 
0163   h1_mcElePt_ = new TH1F("mcElePt", "p_{T} of MC electrons", 300, 0., 300.);
0164   h1_mcElePt_->SetXTitle("p_{T}(GeV/c)");
0165   h1_mcElePt_->SetYTitle("events");
0166 
0167   h1_mcEleEta_ = new TH1F("mcEleEta", "Eta of MC electrons", 100, -4., 4.);
0168   h1_mcEleEta_->SetXTitle("#eta");
0169   h1_mcEleEta_->SetYTitle("events");
0170 
0171   h1_mcElePhi_ = new TH1F("mcElePhi", "Phi of MC electrons", 100, -4., 4.);
0172   h1_mcElePhi_->SetXTitle("#phi");
0173   h1_mcElePhi_->SetYTitle("events");
0174 }
0175 
0176 //========================================================================
0177 
0178 void ZeePlots::fillEleMCInfo(const HepMC::GenEvent* myGenEvent) {
0179   file_->cd();
0180 
0181   for (HepMC::GenEvent::particle_const_iterator p = myGenEvent->particles_begin(); p != myGenEvent->particles_end();
0182        ++p) {
0183     if (abs((*p)->pdg_id()) == 11) {
0184       h1_mcEle_Energy_->Fill((*p)->momentum().e());
0185       h1_mcElePt_->Fill((*p)->momentum().perp());
0186       h1_mcEleEta_->Fill((*p)->momentum().eta());
0187       h1_mcElePhi_->Fill((*p)->momentum().phi());
0188 
0189     }  //matches if (  abs( (*p)->pdg_id() ) == 11 )
0190 
0191   }  //end loop over MC particles
0192 }
0193 
0194 //========================================================================
0195 void ZeePlots::bookEleHistograms() {
0196   file_->cd();
0197 
0198   h1_nEleReco_ = new TH1F("h1_nEleReco", "h1_nEleReco", 20, 0, 20);
0199   h1_nEleReco_->SetXTitle("Num. of reco electrons");
0200 
0201   h1_recoEleEnergy_ = new TH1F("recoEleEnergy", "EleEnergy from SC", 300, 0., 300.);
0202   h1_recoEleEnergy_->SetXTitle("eleSCEnergy(GeV)");
0203   h1_recoEleEnergy_->SetYTitle("events");
0204 
0205   h1_recoElePt_ = new TH1F("recoElePt", "p_{T} of reco electrons", 300, 0., 300.);
0206   h1_recoElePt_->SetXTitle("p_{T}(GeV/c)");
0207   h1_recoElePt_->SetYTitle("events");
0208 
0209   h1_recoEleEta_ = new TH1F("recoEleEta", "Eta of reco electrons", 100, -4., 4.);
0210   h1_recoEleEta_->SetXTitle("#eta");
0211   h1_recoEleEta_->SetYTitle("events");
0212 
0213   h1_recoElePhi_ = new TH1F("recoElePhi", "Phi of reco electrons", 100, -4., 4.);
0214   h1_recoElePhi_->SetXTitle("#phi");
0215   h1_recoElePhi_->SetYTitle("events");
0216 }
0217 
0218 //========================================================================
0219 
0220 void ZeePlots::fillEleInfo(const reco::GsfElectronCollection* electronCollection) {
0221   file_->cd();
0222 
0223   h1_nEleReco_->Fill(electronCollection->size());
0224 
0225   for (reco::GsfElectronCollection::const_iterator eleIt = electronCollection->begin();
0226        eleIt != electronCollection->end();
0227        eleIt++) {
0228     file_->cd();
0229 
0230     h1_recoEleEnergy_->Fill(eleIt->superCluster()->energy());
0231     h1_recoElePt_->Fill(eleIt->pt());
0232     h1_recoEleEta_->Fill(eleIt->eta());
0233     h1_recoElePhi_->Fill(eleIt->phi());
0234 
0235   }  //end loop on electrons
0236 }
0237 
0238 //========================================================================
0239 
0240 void ZeePlots::writeEleHistograms() {
0241   file_->cd();
0242 
0243   std::cout << "Start with ZeePlots::writeEleHistograms(), done file_->cd(); " << std::endl;
0244 
0245   h1_recoEleEnergy_->Write();
0246   h1_recoElePt_->Write();
0247   h1_recoEleEta_->Write();
0248   h1_recoElePhi_->Write();
0249 
0250   std::cout << "Done with ZeePlots::writeEleHistograms() " << std::endl;
0251 }
0252 
0253 //========================================================================
0254 
0255 void ZeePlots::writeMCEleHistograms() {
0256   file_->cd();
0257 
0258   std::cout << "Start with ZeePlots::writeMCEleHistograms(), done file_->cd(); " << std::endl;
0259 
0260   h1_mcEle_Energy_->Write();
0261   h1_mcElePt_->Write();
0262   h1_mcEleEta_->Write();
0263   h1_mcElePhi_->Write();
0264 
0265   std::cout << "Done with ZeePlots::writeMCEleHistograms() " << std::endl;
0266 }
0267 
0268 //========================================================================
0269 
0270 void ZeePlots::bookHLTHistograms() {
0271   file_->cd();
0272 
0273   h1_FiredTriggers_ = new TH1F("h1_FiredTriggers", "h1_FiredTriggers", 5, 0, 5);
0274 
0275   h1_HLTVisitedEvents_ = new TH1F("h1_HLTVisitedEvents", "h1_HLTVisitedEvents", 5, 0, 5);
0276 
0277   h1_HLT1Electron_FiredEvents_ = new TH1F("h1_HLT1Electron_FiredEvents", "h1_HLT1Electron_FiredEvents", 5, 0, 5);
0278   h1_HLT2Electron_FiredEvents_ = new TH1F("h1_HLT2Electron_FiredEvents", "h1_HLT2Electron_FiredEvents", 5, 0, 5);
0279   h1_HLT2ElectronRelaxed_FiredEvents_ =
0280       new TH1F("h1_HLT2ElectronRelaxed_FiredEvents", "h1_HLT2ElectronRelaxed_FiredEvents", 5, 0, 5);
0281 
0282   h1_HLT1Electron_HLT2Electron_FiredEvents_ =
0283       new TH1F("h1_HLT1Electron_HLT2Electron_FiredEvents", "h1_HLT1Electron_HLT2Electron_FiredEvents", 5, 0, 5);
0284   h1_HLT1Electron_HLT2ElectronRelaxed_FiredEvents_ = new TH1F(
0285       "h1_HLT1Electron_HLT2ElectronRelaxed_FiredEvents", "h1_HLT1Electron_HLT2ElectronRelaxed_FiredEvents", 5, 0, 5);
0286   h1_HLT2Electron_HLT2ElectronRelaxed_FiredEvents_ = new TH1F(
0287       "h1_HLT2Electron_HLT2ElectronRelaxed_FiredEvents", "h1_HLT2Electron_HLT2ElectronRelaxed_FiredEvents", 5, 0, 5);
0288 
0289   h1_HLT1Electron_HLT2Electron_HLT2ElectronRelaxed_FiredEvents_ =
0290       new TH1F("h1_HLT1Electron_HLT2Electron_HLT2ElectronRelaxed_FiredEvents",
0291                "h1_HLT1Electron_HLT2Electron_HLT2ElectronRelaxed_FiredEvents",
0292                5,
0293                0,
0294                5);
0295 }
0296 
0297 //========================================================================
0298 
0299 void ZeePlots::fillHLTInfo(edm::Handle<edm::TriggerResults> hltTriggerResultHandle) {
0300   file_->cd();
0301 
0302   int hltCount = hltTriggerResultHandle->size();
0303 
0304   bool aHLTResults[200] = {false};
0305 
0306   for (int i = 0; i < hltCount; i++) {
0307     aHLTResults[i] = hltTriggerResultHandle->accept(i);
0308     if (aHLTResults[i])
0309       h1_FiredTriggers_->Fill(i);
0310 
0311     //HLT bit 32 = HLT1Electron
0312     //HLT bit 34 = HLT2Electron
0313     //HLT bit 35 = HLT2ElectronRelaxed
0314   }
0315 
0316   h1_HLTVisitedEvents_->Fill(1);
0317 
0318   if (aHLTResults[32] && !aHLTResults[34] && !aHLTResults[35])
0319     h1_HLT1Electron_FiredEvents_->Fill(1);
0320 
0321   if (aHLTResults[34] && !aHLTResults[32] && !aHLTResults[35])
0322     h1_HLT2Electron_FiredEvents_->Fill(1);
0323 
0324   if (aHLTResults[35] && !aHLTResults[32] && !aHLTResults[34])
0325     h1_HLT2ElectronRelaxed_FiredEvents_->Fill(1);
0326 
0327   if (aHLTResults[32] && aHLTResults[34] && !aHLTResults[35])
0328     h1_HLT1Electron_HLT2Electron_FiredEvents_->Fill(1);
0329 
0330   if (aHLTResults[32] && aHLTResults[35] && !aHLTResults[34])
0331     h1_HLT1Electron_HLT2ElectronRelaxed_FiredEvents_->Fill(1);
0332 
0333   if (aHLTResults[34] && aHLTResults[35] && !aHLTResults[32])
0334     h1_HLT2Electron_HLT2ElectronRelaxed_FiredEvents_->Fill(1);
0335 
0336   if (aHLTResults[32] && aHLTResults[34] && aHLTResults[35])
0337     h1_HLT1Electron_HLT2Electron_HLT2ElectronRelaxed_FiredEvents_->Fill(1);
0338 }
0339 
0340 void ZeePlots::fillEleClassesPlots(calib::CalibElectron* myEle) {
0341   int myClass = myEle->getRecoElectron()->classification();
0342 
0343   float myEta = myEle->getRecoElectron()->eta();
0344 
0345   if (myClass == 0 || myClass == 100)
0346     h1_occupancyVsEtaGold_->Fill(myEta);
0347 
0348   std::cout << "[ZeePlots::fillEleClassesPlots]Done gold" << std::endl;
0349 
0350   if (myClass == 40 || myClass == 140)
0351     h1_occupancyVsEtaCrack_->Fill(myEta);
0352 
0353   std::cout << "[ZeePlots::fillEleClassesPlots]Done crack" << std::endl;
0354 
0355   if ((myClass >= 30 && myClass <= 34) || (myClass >= 130 && myClass <= 134))
0356     h1_occupancyVsEtaShower_->Fill(myEta);
0357 
0358   std::cout << "[ZeePlots::fillEleClassesPlots]Done shower" << std::endl;
0359 
0360   if (myClass == 10 || myClass == 20 || myClass == 110 || myClass == 120)
0361     h1_occupancyVsEtaSilver_->Fill(myEta);
0362 
0363   std::cout << "[ZeePlots::fillEleClassesPlots]Done" << std::endl;
0364 }
0365 
0366 void ZeePlots::bookEleClassesPlots() {
0367   file_->cd();
0368 
0369   h1_occupancyVsEtaGold_ = new TH1F("occupancyVsEtaGold", "occupancyVsEtaGold", 200, -4., 4.);
0370   h1_occupancyVsEtaGold_->SetYTitle("Electron statistics");
0371   h1_occupancyVsEtaGold_->SetXTitle("Eta channel");
0372 
0373   h1_occupancyVsEtaSilver_ = new TH1F("occupancyVsEtaSilver", "occupancyVsEtaSilver", 200, -4., 4.);
0374   h1_occupancyVsEtaSilver_->SetYTitle("Electron statistics");
0375   h1_occupancyVsEtaSilver_->SetXTitle("Eta channel");
0376 
0377   h1_occupancyVsEtaShower_ = new TH1F("occupancyVsEtaShower", "occupancyVsEtaShower", 200, -4., 4.);
0378   h1_occupancyVsEtaShower_->SetYTitle("Electron statistics");
0379   h1_occupancyVsEtaShower_->SetXTitle("Eta channel");
0380 
0381   h1_occupancyVsEtaCrack_ = new TH1F("occupancyVsEtaCrack", "occupancyVsEtaCrack", 200, -4., 4.);
0382   h1_occupancyVsEtaCrack_->SetYTitle("Electron statistics");
0383   h1_occupancyVsEtaCrack_->SetXTitle("Eta channel");
0384 }
0385 
0386 void ZeePlots::writeEleClassesPlots() {
0387   file_->cd();
0388 
0389   h1_occupancyVsEtaGold_->Write();
0390   h1_occupancyVsEtaSilver_->Write();
0391   h1_occupancyVsEtaShower_->Write();
0392   h1_occupancyVsEtaCrack_->Write();
0393 }