Belle II Software  release-05-01-25
TrackLinker.cc
1 /**************************************************************************
2  * BASF2 (Belle Analysis Framework 2) *
3  * Copyright(C) 2015 - Belle II Collaboration *
4  * *
5  * Author: The Belle II Collaboration *
6  * Contributors: Oliver Frost *
7  * *
8  * This software is provided "as is" without any warranty. *
9  **************************************************************************/
10 #include <tracking/trackFindingCDC/findlets/minimal/TrackLinker.h>
11 
12 #include <tracking/trackFindingCDC/eventdata/tracks/CDCTrack.h>
13 
14 #include <tracking/trackFindingCDC/utilities/WeightedRelation.h>
15 #include <tracking/trackFindingCDC/utilities/Algorithms.h>
16 
17 using namespace Belle2;
18 using namespace TrackFindingCDC;
19 
21 {
23 }
24 
26 {
27  return "Links tracks by extraction of track paths in a cellular automaton.";
28 }
29 
30 void TrackLinker::exposeParameters(ModuleParamList* moduleParamList, const std::string& prefix)
31 {
32  m_trackRelationCreator.exposeParameters(moduleParamList, prefix);
33  m_cellularPathFinder.exposeParameters(moduleParamList, prefix);
34 }
35 
36 void TrackLinker::apply(const std::vector<CDCTrack>& inputTracks, std::vector<CDCTrack>& outputTracks)
37 {
38  // Obtain the tracks as pointers
39  std::vector<const CDCTrack*> trackPtrs = as_pointers<const CDCTrack>(inputTracks);
40 
41  // Create linking relations
42  m_trackRelations.clear();
43  m_trackRelationCreator.apply(trackPtrs, m_trackRelations);
44 
45  // Find linking paths
46  m_trackPaths.clear();
48 
49  // Put the linked tracks together
50  outputTracks.reserve(outputTracks.size() + m_trackPaths.size());
51  for (const std::vector<const CDCTrack*>& trackPath : m_trackPaths) {
52  // cppcheck-suppress useStlAlgorithm
53  outputTracks.push_back(CDCTrack::condense(trackPath));
54  }
55 }
Belle2::TrackFindingCDC::TrackLinker::TrackLinker
TrackLinker()
Constructor adding the filter as a subordinary processing signal listener.
Definition: TrackLinker.cc:20
Belle2::TrackFindingCDC::TrackLinker::m_cellularPathFinder
MultipassCellularPathFinder< const CDCTrack > m_cellularPathFinder
Instance of the cellular automaton path finder.
Definition: TrackLinker.h:66
Belle2::TrackFindingCDC::TrackLinker::apply
void apply(const std::vector< CDCTrack > &inputTracks, std::vector< CDCTrack > &outputTracks) final
Main algorithm.
Definition: TrackLinker.cc:36
Belle2::TrackFindingCDC::CompositeProcessingSignalListener::addProcessingSignalListener
void addProcessingSignalListener(ProcessingSignalListener *psl)
Register a processing signal listener to be notified.
Definition: CompositeProcessingSignalListener.cc:57
Belle2::TrackFindingCDC::TrackLinker::m_trackPaths
std::vector< Path< const CDCTrack > > m_trackPaths
Memory for the track paths generated from the graph.
Definition: TrackLinker.h:72
Belle2::TrackFindingCDC::CDCTrack::condense
static CDCTrack condense(const Path< const CDCTrack > &trackPath)
Concats several tracks from a path.
Belle2::TrackFindingCDC::TrackLinker::m_trackRelations
std::vector< WeightedRelation< const CDCTrack > > m_trackRelations
Memory for the relations between tracks to be followed on linking.
Definition: TrackLinker.h:69
Belle2
Abstract base class for different kinds of events.
Definition: MillepedeAlgorithm.h:19
Belle2::TrackFindingCDC::TrackLinker::exposeParameters
void exposeParameters(ModuleParamList *moduleParamList, const std::string &prefix) final
Expose the parameters to a module.
Definition: TrackLinker.cc:30
Belle2::ModuleParamList
The Module parameter list class.
Definition: ModuleParamList.h:46
Belle2::TrackFindingCDC::TrackLinker::m_trackRelationCreator
WeightedRelationCreator< const CDCTrack, ChooseableTrackRelationFilter > m_trackRelationCreator
Creator of the track relations for linking.
Definition: TrackLinker.h:63
Belle2::TrackFindingCDC::TrackLinker::getDescription
std::string getDescription() final
Short description of the findlet.
Definition: TrackLinker.cc:25