Merging.
55 {
56
58
59
62 NewContainer->setStore(auxNewContainer);
63
69
70 static const SG::Accessor<float> mAcc_sumPt2("sumPt2");
71 static const SG::Accessor<int> mAcc_momdir ("MomentaDirection");
72 static const SG::Accessor<float> mAcc_mass("mass");
73 static const SG::Accessor<float> mAcc_energy("ee");
74 static const SG::Accessor<int> mAcc_intrk("NumInputTrk");
75 static const SG::Accessor<float> mAcc_radpat("radiiPattern");
77 static const SG::Accessor<int> mAcc_numtav("NumTrkAtVtx");
79
80 bool moreDeco = mAcc_momdir.isAvailable(*MyVxCont.
front());
81
82 if (!moreDeco)
84
92
93
94 std::vector<bool> remerged(MyVxCont.
size(),
false);
95
98 unsigned int Ni = 0;
101 auto vx = std::make_unique<xAOD::Vertex>(**i);
102
103 if (remerged[Ni])
104 continue;
105
106 std::vector<const xAOD::TrackParticle*> combinedTracks;
107 std::vector<ElementLink<xAOD::TrackParticleContainer>> tpLinks1 =
108 vx->trackParticleLinks();
109 if (!tpLinks1.empty()) {
110 for (const auto& tp_EL : tpLinks1) {
112 combinedTracks.push_back(trk);
113 }
114
115 unsigned int Nj = Ni + 1;
116 bool newmerge = false;
120 if (remerged[Nj])
121 continue;
122
123 if (newmerge) {
124 combinedTracks.clear();
125 tpLinks1 = vx->trackParticleLinks();
126 if (tpLinks1.empty())
127 break;
128 for (const auto& tp_EL : tpLinks1) {
130 combinedTracks.push_back(trk);
131 }
132 newmerge = false;
133 }
134
135
136
138 continue;
139
141
142
143 const std::vector<ElementLink<xAOD::TrackParticleContainer>>
145 if (tpLinks2.empty())
146 continue;
147
148 for (const auto& tp_EL : tpLinks2) {
150 combinedTracks.push_back(trk);
151 }
152
154 << tpLinks2.size());
155
156
157
158
159
161 std::unique_ptr<xAOD::Vertex> mergedVtx =
m_iVertexFitter->fit(ctx, combinedTracks, start);
162
163 ATH_MSG_DEBUG(
"Merged vertices " << mergedVtx->nTrackParticles());
164
165 remerged[Nj] = true;
166 remerged[Ni] = true;
167 newmerge = true;
168
169
170 float pt1 = sqrt(mAcc_sumPt2(*vx));
171 float pt2 = sqrt(mAcc_sumPt2(*mergeCand));
172 float ntrk1 = 1.0 * ((vx->trackParticleLinks()).
size());
174 float wght1 =
175 0.6 * pt1 / (pt1 + pt2) + 0.4 * ntrk1 / (ntrk1 + ntrk2);
176 float wght2 =
177 0.6 * pt2 / (pt1 + pt2) + 0.4 * ntrk2 / (ntrk1 + ntrk2);
178
181 float mas1 = mAcc_mass(*vx);
182 float mas2 = mAcc_mass(*mergeCand);
183 float e1 = mAcc_energy(*vx);
184 float e2 = mAcc_energy(*mergeCand);
185 int inNtrk1 = mAcc_intrk(*vx);
186 int inNtrk2 = mAcc_intrk(*mergeCand);
187
188 int ntrks = 0;
189 float md1 = 0., md2 = 0., hf1 = 0., hf2 = 0.;
190 std::vector<float> trkW1, trkW2, doe1, doe2;
191 if (moreDeco) {
192 doe1 = mAcc_trkdoe(*vx);
193 doe2 = mAcc_trkdoe(*mergeCand);
194 doe2.insert(doe2.end(), doe1.begin(), doe1.end());
195 md1 = mAcc_momdir(*vx);
196 md2 = mAcc_momdir(*mergeCand);
197 hf1 = mAcc_radpat(*vx);
198 hf2 = mAcc_radpat(*mergeCand);
199 trkW1 = mAcc_trkwt(*vx);
200 trkW2 = mAcc_trkwt(*mergeCand);
201 trkW2.insert(trkW2.end(), trkW1.begin(), trkW1.end());
202 ntrks = mAcc_numtav(*vx) + mAcc_numtav(*mergeCand);
203 }
204
205
206 vx = std::move(mergedVtx);
207
208 if (wght1 >= wght2)
209 vx->setVertexType(typ1);
210 else
211 vx->setVertexType(typ2);
212
213 if (moreDeco) {
214 mAcc_trkdoe(*vx) = doe2;
215 mAcc_momdir(*vx) = wght1 * md1 + wght2 * md2;
216 mAcc_radpat(*vx) = wght1 * hf1 + wght2 * hf2;
217 mAcc_trkwt(*vx) = trkW2;
218 mAcc_numtav(*vx) = ntrks;
219 }
220
221 mDecor_sumPt2(*vx) = pt1 * pt1 + pt2 * pt2;
222 mDecor_mass(*vx) = wght1 * mas1 + wght2 * mas2;
223 mDecor_energy(*vx) = wght1 *
e1 + wght2 *
e2;
224 mDecor_nrobbed(*vx) = 0;
225 mDecor_intrk(*vx) = (
int)(wght1 * inNtrk1 + wght1 * inNtrk2);
226
227 }
228 }
229
230
231 if (vx != nullptr){
234 }
235 }
236
237 return std::make_pair(NewContainer, auxNewContainer);
238
239 }
size_t size() const
Number of registered mappings.
DataModel_detail::const_iterator< DataVector > const_iterator
value_type push_back(value_type pElem)
Add an element to the end of the collection.
const_iterator end() const noexcept
Return a const_iterator pointing past the end of the collection.
const_iterator begin() const noexcept
Return a const_iterator pointing at the beginning of the collection.
const T * front() const
Access the first element in the collection as an rvalue.
size_type size() const noexcept
Returns the number of elements in the collection.
const TrackParticleLinks_t & trackParticleLinks() const
Get all the particles associated with the vertex.
VxType::VertexType vertexType() const
The type of the vertex.
const Amg::Vector3D & position() const
Returns the 3-pos.
SG::Accessor< T, ALLOC > Accessor
SG::Decorator< T, ALLOC > Decorator
Helper class to provide type-safe access to aux data, specialized for JaggedVecElt.
float j(const xAOD::IParticle &, const xAOD::TrackMeasurementValidation &hit, const Eigen::Matrix3d &jab_inv)
double e2(const xAOD::CaloCluster &cluster)
return the uncorrected cluster energy in 2nd sampling
double e1(const xAOD::CaloCluster &cluster)
return the uncorrected cluster energy in 1st sampling
VertexAuxContainer_v1 VertexAuxContainer
Definition of the current jet auxiliary container.
TrackParticle_v1 TrackParticle
Reference the current persistent version:
VertexContainer_v1 VertexContainer
Definition of the current "Vertex container version".
Vertex_v1 Vertex
Define the latest version of the vertex class.
1.16.1