|
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
|
#include "DataFormats/ParticleFlowReco/interface/PFTrajectoryPoint.h"
#include <array>
#include <ostream>
#include <algorithm>
using namespace reco;
// To be kept in synch with the enumerator definitions in PFTrajectoryPoint.h file
// Don't consider "Unknown" and "NLayers"
std::array<std::string, PFTrajectoryPoint::NLayers> const PFTrajectoryPoint::layerTypeNames{{"ClosestApproach",
"BeamPipeOrEndVertex",
"PS1",
"PS2",
"ECALEntrance",
"ECALShowerMax",
"HCALEntrance",
"HCALExit",
"HOLayer",
"VFcalEntrance"}};
PFTrajectoryPoint::PFTrajectoryPoint() : isTrackerLayer_(false), detId_(-1), layer_(-1) {}
PFTrajectoryPoint::PFTrajectoryPoint(int detId,
int layer,
const math::XYZPoint& posxyz,
const math::XYZTLorentzVector& momentum)
: isTrackerLayer_(false), detId_(detId), layer_(layer), posxyz_(posxyz), momentum_(momentum) {
if (detId)
isTrackerLayer_ = true;
posrep_.SetCoordinates(posxyz_.Rho(), posxyz_.Eta(), posxyz_.Phi());
}
PFTrajectoryPoint::LayerType PFTrajectoryPoint::layerTypeByName(const std::string& name) {
auto it = std::find(layerTypeNames.begin(), layerTypeNames.end(), name);
if (it == layerTypeNames.end()) {
return Unknown; // better this or throw()?
}
int index = std::distance(layerTypeNames.begin(), it);
return LayerType(index);
}
std::ostream& reco::operator<<(std::ostream& out, const reco::PFTrajectoryPoint& trajPoint) {
if (!out)
return out;
const math::XYZPoint& posxyz = trajPoint.position();
out << "Traj point id = " << trajPoint.detId() << ", layer = " << trajPoint.layer() << ", Eta,Phi = " << posxyz.Eta()
<< "," << posxyz.Phi() << ", X,Y = " << posxyz.X() << "," << posxyz.Y() << ", R,Z = " << posxyz.Rho() << ","
<< posxyz.Z() << ", E,Pt = " << trajPoint.momentum().E() << "," << trajPoint.momentum().Pt();
return out;
}
|