1 |
amagnan |
1.1 |
#include <iostream>
|
2 |
|
|
#include <fstream>
|
3 |
|
|
|
4 |
amagnan |
1.2 |
//#include "DataFormats/EgammaReco/interface/Electron.h"
|
5 |
amagnan |
1.1 |
|
6 |
|
|
|
7 |
amagnan |
1.2 |
#include "FWCore/MessageLogger/interface/MessageLogger.h"
|
8 |
|
|
|
9 |
amagnan |
1.1 |
#include "UserCode/HbbAnalysis/interface/HistosElecs.hh"
|
10 |
|
|
|
11 |
|
|
namespace HbbAnalysis {//namespace
|
12 |
|
|
|
13 |
amagnan |
1.2 |
void HistosElecs::Initialise(TFileDirectory & aDir, std::string aName, bool aDoGenMatched){
|
14 |
amagnan |
1.1 |
|
15 |
amagnan |
1.2 |
doGenMatched_ = aDoGenMatched;
|
16 |
amagnan |
1.1 |
CreateHistos(aName,aDir);
|
17 |
|
|
|
18 |
amagnan |
1.2 |
p_nElectrons = aDir.make<TH1F>("p_nElectrons",";N_{electrons};N_{entries}",20,0,20);
|
19 |
|
|
|
20 |
amagnan |
1.1 |
p_nChambers = aDir.make<TH1F>("p_nChambers",";n_{chambers};n_{entries}",20,0,20);
|
21 |
|
|
|
22 |
|
|
}
|
23 |
|
|
|
24 |
amagnan |
1.2 |
void HistosElecs::FillEventHistograms(edm::Handle<std::vector<pat::Electron> > aCol){
|
25 |
|
|
|
26 |
|
|
if (doGenMatched_) {
|
27 |
|
|
unsigned int nMatched = 0;
|
28 |
|
|
for (std::vector<pat::Electron>::const_iterator iElec = aCol->begin();
|
29 |
|
|
iElec != aCol->end();
|
30 |
|
|
iElec++)
|
31 |
|
|
{
|
32 |
|
|
if (MatchesGenElectron(*iElec)) nMatched++;
|
33 |
|
|
}
|
34 |
|
|
p_nElectrons->Fill(nMatched);
|
35 |
|
|
}
|
36 |
|
|
else p_nElectrons->Fill(aCol->size());
|
37 |
|
|
|
38 |
|
|
}
|
39 |
|
|
|
40 |
|
|
void HistosElecs::FillHistograms(const pat::Electron & aElec, bool isLead){//FillHistograms
|
41 |
amagnan |
1.1 |
|
42 |
amagnan |
1.2 |
bool lIsGenMatched = !doGenMatched_ || (doGenMatched_ && MatchesGenElectron(aElec));
|
43 |
|
|
if (lIsGenMatched) {//genMatched
|
44 |
|
|
if (isLead) {//isLead
|
45 |
|
|
FillBaseHistograms(aElec.pt(),aElec.eta(),aElec.phi());
|
46 |
amagnan |
1.1 |
|
47 |
amagnan |
1.2 |
}//isLead
|
48 |
|
|
}//genMatched
|
49 |
amagnan |
1.1 |
|
50 |
|
|
}//FillHistograms
|
51 |
|
|
|
52 |
amagnan |
1.2 |
bool HistosElecs::MatchesGenElectron(const pat::Electron& aPatElec)
|
53 |
|
|
{
|
54 |
|
|
|
55 |
|
|
bool isGenMatched = false;
|
56 |
|
|
|
57 |
|
|
const std::vector<reco::GenParticleRef> & lVec = aPatElec.genParticleRefs();
|
58 |
|
|
for ( std::vector<reco::GenParticleRef>::const_iterator it = lVec.begin();
|
59 |
|
|
it != lVec.end(); ++it ) {
|
60 |
|
|
if ( it->ref().isNonnull() && it->ref().isValid() ) {
|
61 |
|
|
const reco::GenParticleRef & genParticle = (*it);
|
62 |
|
|
if ( genParticle->pdgId() == -11 || genParticle->pdgId() == +11 ) isGenMatched = true;
|
63 |
|
|
} else {
|
64 |
|
|
edm::LogWarning("MatchesGenElectron") << " edm::Ref of genParticle associated to pat::Electron is invalid !!";
|
65 |
|
|
}
|
66 |
|
|
}
|
67 |
|
|
return isGenMatched;
|
68 |
|
|
}
|
69 |
|
|
|
70 |
amagnan |
1.1 |
}//namespace
|
71 |
|
|
|
72 |
|
|
|