|
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
|
#ifndef RecoCandidate_IsoDepositDirection_H
#define RecoCandidate_IsoDepositDirection_H
/** \class isolation::Direction
* Simple eta-phi direction
* \author M. Zanetti
*/
#include <cmath>
#include <sstream>
#include <iostream>
#include "DataFormats/Math/interface/deltaR.h"
#include "DataFormats/Math/interface/deltaPhi.h"
namespace reco {
namespace isodeposit {
class Direction {
public:
struct Distance {
float deltaR;
float relativeAngle;
bool operator<(const Distance& rd2) const { return deltaR < rd2.deltaR; };
};
Direction(double eta = 0., double phi = 0.) : theEta(eta), thePhi(phi) {
while (thePhi < 0.0)
thePhi += 2 * M_PI;
while (thePhi >= 2 * M_PI)
thePhi -= 2 * M_PI;
};
double eta() const { return theEta; }
double phi() const { return thePhi; }
double theta() const { return acos(tanh(theEta)); }
inline bool operator==(const Direction& d2) {
if (this == &d2)
return true;
if (deltaR(d2) < 1.e-4)
return true;
return false;
}
inline double deltaR2(const Direction& dir2) const { return reco::deltaR2(*this, dir2); }
inline double deltaR(const Direction& dir2) const { return reco::deltaR(*this, dir2); }
Distance operator-(const Direction& dir2) const {
Distance result;
double dR = deltaR(dir2);
double dEta = theEta - dir2.eta();
double dPhi = reco::deltaPhi(thePhi, dir2.phi());
result.relativeAngle = (dR > 1.e-4) ? atan2(dPhi, dEta) : 0.;
result.deltaR = dR;
return result;
}
Direction operator+(const Distance& relDir) const {
double eta = theEta + relDir.deltaR * cos(relDir.relativeAngle);
double phi = thePhi + relDir.deltaR * sin(relDir.relativeAngle);
return Direction(eta, phi);
}
std::string print() const {
std::ostringstream str;
str << " (Eta=" << theEta << ","
<< "Phi=" << thePhi << ")";
return str.str();
}
private:
float theEta;
float thePhi;
};
} //namespace isodeposit
} //namespace reco
#endif
|