19 #include "AthLinks/ElementLink.h"
37 return StatusCode::FAILURE;
48 return StatusCode::FAILURE;
61 return StatusCode::SUCCESS;
69 m_muonCollectionKey(
"Muons"),
70 m_TrkParticleCollection(
"TrackParticleCandidate"),
71 m_iVertexFitter(
"Trk::TrkVKalVrtFitter"),
72 m_iV0VertexFitter(
"Trk::V0VertexFitter"),
73 m_trkSelector(
"InDet::TrackSelectorTool")
99 pairVxContainer->setStore(pairVxAuxContainer);
103 quadVxContainer->setStore(quadVxAuxContainer);
114 ATH_MSG_DEBUG(
"ID TrackParticle container size "<< importedTrackCollection->
size());
117 std::vector<const xAOD::Muon*> theMuonsAfterSelection;
119 unsigned int nCombMuons = 0;
120 unsigned int nSegmentTaggedMuons = 0;
122 for (
auto muItr=importedMuonCollection->
begin(); muItr!=importedMuonCollection->
end(); ++muItr) {
123 if ( *muItr == NULL )
continue;
124 muonDecorator(**muItr) = -1;
125 if ( ((*muItr)->muonType() !=
xAOD::Muon::Combined ) && ((*muItr)->muonType() != xAOD::Muon::SegmentTagged ) )
continue;
126 if (!(*muItr)->inDetTrackParticleLink().isValid())
continue;
127 auto& link = (*muItr)->inDetTrackParticleLink();
129 if ( muonTrk==NULL)
continue;
132 if ( fabs(muonTrk->
pt())<
m_ptCut )
continue;
135 if ( (*muItr)->muonType() == xAOD::Muon::SegmentTagged ) ++nSegmentTaggedMuons;
136 theMuonsAfterSelection.push_back(*muItr);
138 unsigned int nSelectedMuons = theMuonsAfterSelection.size();
139 ATH_MSG_DEBUG(
"Number of muons after selection: " << nSelectedMuons);
141 ATH_MSG_DEBUG(
"and " << nSegmentTaggedMuons <<
" are segment tagged");
142 if ( (nSelectedMuons < 4) || (nCombMuons < 1) ) {
144 return StatusCode::SUCCESS;
155 std::sort(theMuonsAfterSelection.begin(), theMuonsAfterSelection.end(), [](
const xAOD::Muon *
a,
const xAOD::Muon *
b) {
156 return b->pt() < a->pt();
160 unsigned int muonIndex(0);
161 for (
auto selMuon : theMuonsAfterSelection) {
162 muonDecorator(*selMuon) = muonIndex;
167 std::vector<Combination> quadruplets;
168 std::vector<Combination>
pairs;
170 if (quadruplets.size()==0) {
172 return StatusCode::SUCCESS;
183 std::vector<const xAOD::TrackParticle*> theTracks = (*pairItr).trackParticles(
"pair1");
185 if (pairVxCandidate != 0) {
187 indexDecorator(*pairVxCandidate) = (*pairItr).combinationIndices();
188 chargeDecorator(*pairVxCandidate) = (*pairItr).combinationCharges();
192 std::vector<const xAOD::Muon*> theStoredMuons;
193 theStoredMuons = (*pairItr).muons;
194 helper.setMuons(theStoredMuons,importedMuonCollection.
get());
196 pairVxContainer->
push_back(pairVxCandidate);
197 ATH_MSG_DEBUG(
"..... indices: " << (*pairItr).combinationIndices() <<
198 " charges: " << (*pairItr).combinationCharges() <<
209 std::vector<const xAOD::TrackParticle*> theDCTracks; theDCTracks.clear();
212 theDCTracks = (*quadItr).trackParticles(
"DC");
223 if (dcVxCandidate != 0) {
225 indexDecorator(*dcVxCandidate) = (*quadItr).combinationIndices();
226 chargeDecorator(*dcVxCandidate) = (*quadItr).combinationCharges();
234 const std::vector<const xAOD::Muon*> &theStoredMuons = (*quadItr).muons;
235 helper.setMuons(theStoredMuons,importedMuonCollection.
get());
237 quadVxContainer->
push_back(dcVxCandidate);
238 ATH_MSG_DEBUG(
"..... indices: " << (*quadItr).combinationIndices() <<
239 " charges: " << (*quadItr).combinationCharges() <<
240 " chi2(DC): " << dcChi2);
249 return StatusCode::SUCCESS;;
266 if(concreteVertexFitter == 0) {
267 ATH_MSG_FATAL(
"The vertex fitter passed is not a V0 Vertex Fitter");
281 return myVxCandidate;
294 std::vector<std::vector<unsigned int> > quadIndices =
mFromN(4,
length);
306 std::vector<std::vector<unsigned int> > allCombinations;
307 std::vector<unsigned int> mainList;
308 std::vector<unsigned int> combination;
309 for (
unsigned int i=0;
i<
N; ++
i) mainList.push_back(
i);
311 return allCombinations;
316 std::vector<unsigned int> &combination,
317 std::vector<unsigned int> &mainList,
318 std::vector<std::vector<unsigned int> > &allCombinations) {
320 allCombinations.push_back(combination);
324 for (
unsigned int i=
offset;
i<=mainList.size()-
k; ++
i) {
325 combination.push_back(mainList[
i]);
327 combination.pop_back();
338 std::vector<std::pair<unsigned int, unsigned int> > uniquePairs;
339 std::vector<std::vector<unsigned int> > doublets =
mFromN(2,
length);
340 for (std::vector<std::vector<unsigned int> >::
iterator it=doublets.begin();
it!=doublets.end(); ++
it) {
341 std::pair<unsigned int, unsigned int> tmpPair = std::make_pair((*it).at(0),(*it).at(1));
342 uniquePairs.push_back(tmpPair);
357 std::vector<Combination> &
pairs,
358 std::vector<Combination> &quadruplets,
359 unsigned int nSelectedMuons) {
361 std::vector<std::vector<unsigned int> > quadrupletIndices =
getQuadIndices(nSelectedMuons);
362 std::vector<std::pair<unsigned int, unsigned int> > pairIndices =
getPairIndices(nSelectedMuons);
365 std::vector<std::vector<unsigned int> >
::iterator quadItr;
366 for (quadItr=quadrupletIndices.begin(); quadItr!=quadrupletIndices.end(); ++quadItr) {
367 const std::vector<unsigned int> &
quad = (*quadItr);
368 std::vector<const xAOD::Muon*> theMuons = {muonsIn[
quad[0]],muonsIn[
quad[1]],muonsIn[
quad[2]],muonsIn[
quad[3]]};
371 tmpQuad.
muons = std::move(theMuons);
373 quadruplets.emplace_back(std::move(tmpQuad));
375 if (quadruplets.size() == 0)
return;
378 std::vector<std::pair<unsigned int, unsigned int> >
::iterator pairItr;
379 for (pairItr=pairIndices.begin(); pairItr!=pairIndices.end(); ++pairItr) {
380 std::pair<unsigned int, unsigned int> pair = (*pairItr);
382 std::vector<const xAOD::Muon*> theMuons = {muonsIn[pair.first],muonsIn[pair.second]};
383 tmpPair.
muons = std::move(theMuons);
385 pairs.emplace_back(std::move(tmpPair));