|
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
95
96
97
|
#include "DataFormats/ParticleFlowReco/interface/PFDisplacedVertexSeed.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/Math/interface/Point3D.h"
using namespace std;
using namespace reco;
PFDisplacedVertexSeed::PFDisplacedVertexSeed() : seedPoint_(GlobalPoint(0, 0, 0)), totalWeight_(0) {}
void PFDisplacedVertexSeed::addElement(TrackBaseRef element) {
if (std::find(elements_.begin(), elements_.end(), element) == elements_.end()) {
elements_.emplace_back(std::move(element));
}
}
void PFDisplacedVertexSeed::reserveElements(size_t newSize) { elements_.reserve(newSize); }
void PFDisplacedVertexSeed::updateSeedPoint(const GlobalPoint& dcaPoint,
TrackBaseRef r1,
TrackBaseRef r2,
double weight) {
if (isEmpty()) {
seedPoint_ = dcaPoint;
totalWeight_ = weight;
} else {
Basic3DVector<double> vertexSeedVector(seedPoint_);
Basic3DVector<double> dcaVector(dcaPoint);
dcaVector = (dcaVector * weight + vertexSeedVector * totalWeight_) / (totalWeight_ + weight);
GlobalPoint P(dcaVector.x(), dcaVector.y(), dcaVector.z());
totalWeight_ += weight;
seedPoint_ = P;
}
reserveElements(elements_.size() + 2);
addElement(r1);
addElement(r2);
}
void PFDisplacedVertexSeed::mergeWith(const PFDisplacedVertexSeed& displacedVertex) {
double weight = displacedVertex.totalWeight();
const GlobalPoint& dcaPoint = displacedVertex.seedPoint();
Basic3DVector<double> vertexSeedVector(seedPoint_);
Basic3DVector<double> dcaVector(dcaPoint);
dcaVector = (dcaVector * weight + vertexSeedVector * totalWeight_) / (totalWeight_ + weight);
GlobalPoint P(dcaVector.x(), dcaVector.y(), dcaVector.z());
totalWeight_ += weight;
seedPoint_ = P;
reserveElements(elements_.size() + displacedVertex.elements().size());
auto const oldSize = elements_.size();
//avoid checking elements we just added from displacedVertex.elements()
for (auto const& e : displacedVertex.elements()) {
if (std::find(elements_.begin(), elements_.begin() + oldSize, e) == elements_.begin() + oldSize) {
elements_.emplace_back(e);
}
}
}
void PFDisplacedVertexSeed::Dump(ostream& out) const {
if (!out)
return;
out << "\t--- DisplacedVertexSeed --- " << endl;
out << "\tnumber of elements: " << elements_.size() << endl;
out << "\t Seed Point x = " << seedPoint().x() << "\t Seed Point y = " << seedPoint().y()
<< "\t Seed Point z = " << seedPoint().z() << endl;
// Build element label (string) : elid from type, layer and occurence number
// use stringstream instead of sprintf to concatenate string and integer into string
for (auto const& ie : elements_) {
math::XYZPoint Pi(ie.get()->innerPosition());
math::XYZPoint Po(ie.get()->outerPosition());
float innermost_radius = sqrt(Pi.x() * Pi.x() + Pi.y() * Pi.y() + Pi.z() * Pi.z());
float outermost_radius = sqrt(Po.x() * Po.x() + Po.y() * Po.y() + Po.z() * Po.z());
float innermost_rho = sqrt(Pi.x() * Pi.x() + Pi.y() * Pi.y());
float outermost_rho = sqrt(Po.x() * Po.x() + Po.y() * Po.y());
double pt = ie->pt();
out << "ie = " << ie.key() << " pt = " << pt << " innermost hit radius = " << innermost_radius
<< " rho = " << innermost_rho << " outermost hit radius = " << outermost_radius << " rho = " << outermost_rho
<< endl;
out << "ie = " << ie.key() << " pt = "
<< pt
// <<" inn hit pos x = " << Pi.x() << " y = " << Pi.y() << " z = " << Pi.z()
<< " out hit pos x = " << Po.x() << " y = " << Po.y() << " z = " << Po.z() << endl;
}
out << endl;
}
|