|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
#include "CommonTools/ParticleFlow/interface/PFIsoDepositAlgo.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
#include "DataFormats/Math/interface/LorentzVector.h"
#include "FWCore/Framework/interface/EventSetup.h"
using namespace std;
using namespace edm;
using namespace reco;
using namespace math;
using namespace pf2pat;
PFIsoDepositAlgo::PFIsoDepositAlgo(const edm::ParameterSet& iConfig)
: verbose_(iConfig.getUntrackedParameter<bool>("verbose", false))
{}
PFIsoDepositAlgo::~PFIsoDepositAlgo() {}
const PFIsoDepositAlgo::IsoDeposits& PFIsoDepositAlgo::produce(const ParticleCollection& toBeIsolated,
const ParticleCollection& forIsolation) {
isoDeposits_.clear();
isoDeposits_.reserve(toBeIsolated.size());
for (unsigned i = 0; i < toBeIsolated.size(); i++) {
const reco::PFCandidate& toBeIso = toBeIsolated[i];
if (verbose_)
cout << "to be isolated: " << toBeIso << endl;
isoDeposits_.push_back(buildIsoDeposit(toBeIso, forIsolation));
}
if (verbose_) {
cout << "PFIsoDepositAlgo " << endl;
}
return isoDeposits_;
}
IsoDeposit PFIsoDepositAlgo::buildIsoDeposit(const Particle& particle, const ParticleCollection& forIsolation) const {
reco::isodeposit::Direction pfDir(particle.eta(), particle.phi());
// reco::IsoDeposit::Veto veto;
// veto.vetoDir = pfDir;
// veto.dR = 0.05;
IsoDeposit isoDep(pfDir);
for (unsigned i = 0; i < forIsolation.size(); i++) {
const reco::PFCandidate& pfc = forIsolation[i];
// need to remove "particle"!
if (sameParticle(particle, pfc))
continue;
const XYZTLorentzVector& pvi(pfc.p4());
reco::isodeposit::Direction dirPfc(pfc.eta(), pfc.phi());
double dR = pfDir.deltaR(dirPfc);
//COLIN make a parameter
double maxDeltaRForIsoDep_ = 1;
if (dR > maxDeltaRForIsoDep_) {
// if( verbose_ ) cout<<"OUT OF CONE"<<endl;
continue;
}
// else if(verbose_) cout<<endl;
if (verbose_)
cout << "\t" << pfc << endl;
double pt = pvi.Pt();
isoDep.addDeposit(dirPfc, pt);
}
return isoDep;
}
bool PFIsoDepositAlgo::sameParticle(const Particle& particle1, const Particle& particle2) const {
double smallNumber = 1e-15;
if (particle1.particleId() != particle2.particleId())
return false;
else if (fabs(particle1.energy() - particle2.energy()) > smallNumber)
return false;
else if (fabs(particle1.eta() - particle2.eta()) > smallNumber)
return false;
else if (fabs(particle1.eta() - particle2.eta()) > smallNumber)
return false;
else
return true;
}
|