1 | #ifndef PUPPIPARTICLE_HH
|
---|
2 | #define PUPPIPARTICLE_HH
|
---|
3 |
|
---|
4 | #include <functional>
|
---|
5 | #include <algorithm>
|
---|
6 |
|
---|
7 |
|
---|
8 | //............ class to keep track of particles used in the puppi code storing pt,eta,metric value, id (LV,PU) and position in the orignal vector of particles
|
---|
9 | class puppiParticle {
|
---|
10 |
|
---|
11 | public:
|
---|
12 |
|
---|
13 | puppiParticle(){
|
---|
14 | fPt_ = 0.;
|
---|
15 | fEta_ = 0.;
|
---|
16 | fPval_ = 1.;
|
---|
17 | fParticleId_ = -1;
|
---|
18 | fPosition_ = -1;
|
---|
19 | };
|
---|
20 |
|
---|
21 | puppiParticle(const float & pt, const float & eta, const float & pval, const int & particleId, const int & position):
|
---|
22 | fPt_(pt),
|
---|
23 | fEta_(eta),
|
---|
24 | fPval_(pval),
|
---|
25 | fParticleId_(particleId),
|
---|
26 | fPosition_(position)
|
---|
27 | {};
|
---|
28 |
|
---|
29 | puppiParticle(const puppiParticle & particle) {
|
---|
30 | fPt_ = particle.fPt_;
|
---|
31 | fEta_ = particle.fEta_;
|
---|
32 | fPval_ = particle.fPval_;
|
---|
33 | fParticleId_ = particle.fParticleId_;
|
---|
34 | fPosition_ = particle.fPosition_;
|
---|
35 |
|
---|
36 | }
|
---|
37 |
|
---|
38 | virtual ~puppiParticle(){};
|
---|
39 |
|
---|
40 | // sort asaf of pt
|
---|
41 | bool operator < (const puppiParticle & particle2) const {
|
---|
42 | if(fPt_ < particle2.fPt_) return true;
|
---|
43 | else return false;
|
---|
44 | };
|
---|
45 |
|
---|
46 |
|
---|
47 | // oper operator =
|
---|
48 | bool operator == (const puppiParticle & particle2) const {
|
---|
49 | if( fPt_ == particle2.fPt_ and fEta_ == particle2.fEta_ and
|
---|
50 | fPval_ == particle2.fPval_ and fParticleId_ == particle2.fParticleId_ and
|
---|
51 | fPosition_ == particle2.fPosition_) return true;
|
---|
52 | else return false ;
|
---|
53 | };
|
---|
54 |
|
---|
55 | float fPt_; // pt of the candidate
|
---|
56 | float fEta_; // eta of the candidate
|
---|
57 | float fPval_; // metric value
|
---|
58 | int fParticleId_; // particle id means user_index
|
---|
59 | int fPosition_; // position in the original particle vector
|
---|
60 |
|
---|
61 | };
|
---|
62 |
|
---|
63 | class puppiValSort : public std::binary_function<int,int,bool> {
|
---|
64 | public:
|
---|
65 |
|
---|
66 | puppiValSort(){};
|
---|
67 |
|
---|
68 | ~puppiValSort(){};
|
---|
69 |
|
---|
70 | bool operator() (const puppiParticle & x, const puppiParticle & y){
|
---|
71 | if(x.fPval_ < y.fPval_ ) return true;
|
---|
72 | return false;
|
---|
73 | }
|
---|
74 | };
|
---|
75 |
|
---|
76 | #endif
|
---|