LDMX Software
tracking::reco::LinearTrackFinder Class Reference

Public Member Functions

 LinearTrackFinder (const std::string &name, framework::Process &process)
 Constructor.
 
virtual ~LinearTrackFinder ()=default
 Destructor.
 
void onProcessEnd () override
 Output event statistics.
 
void configure (framework::config::Parameters &parameters) override
 Configure the processor using the given user specified parameters.
 
void produce (framework::Event &event) override
 Run the processor.
 
- Public Member Functions inherited from tracking::reco::TrackingGeometryUser
 TrackingGeometryUser (const std::string &name, framework::Process &p)
 
- Public Member Functions inherited from framework::Producer
 Producer (const std::string &name, Process &process)
 Class constructor.
 
virtual void process (Event &event) final
 Processing an event for a Producer is calling produce.
 
- Public Member Functions inherited from framework::EventProcessor
 DECLARE_FACTORY (EventProcessor, EventProcessor *, const std::string &, Process &)
 declare that we have a factory for this class
 
 EventProcessor (const std::string &name, Process &process)
 Class constructor.
 
virtual ~EventProcessor ()=default
 Class destructor.
 
virtual void beforeNewRun (ldmx::RunHeader &run_header)
 Callback for Producers to add parameters to the run header before conditions are initialized.
 
virtual void onNewRun (const ldmx::RunHeader &run_header)
 Callback for the EventProcessor to take any necessary action when the run being processed changes.
 
virtual void onFileOpen (EventFile &event_file)
 Callback for the EventProcessor to take any necessary action when a new event input ROOT file is opened.
 
virtual void onFileClose (EventFile &event_file)
 Callback for the EventProcessor to take any necessary action when a event input ROOT file is closed.
 
virtual void onProcessStart ()
 Callback for the EventProcessor to take any necessary action when the processing of events starts, such as creating histograms.
 
template<class T >
const T & getCondition (const std::string &condition_name)
 Access a conditions object for the current event.
 
TDirectory * getHistoDirectory ()
 Access/create a directory in the histogram file for this event processor to create histograms and analysis tuples.
 
void setStorageHint (framework::StorageControl::Hint hint)
 Mark the current event as having the given storage control hint from this module_.
 
void setStorageHint (framework::StorageControl::Hint hint, const std::string &purposeString)
 Mark the current event as having the given storage control hint from this module and the given purpose string.
 
int getLogFrequency () const
 Get the current logging frequency from the process.
 
int getRunNumber () const
 Get the run number from the process.
 
std::string getName () const
 Get the processor name.
 
void createHistograms (const std::vector< framework::config::Parameters > &histos)
 Internal function which is used to create histograms passed from the python configuration @parma histos vector of Parameters that configure histograms to create.
 

Private Member Functions

std::vector< ldmx::StraightTrackfindTracks (const std::vector< ldmx::StraightTrack > &track_seeds)
 
bool isPositionUsed (const ldmx::Measurement &measurement, const std::set< std::tuple< float, float, float > > &used_sensor_positions)
 

Private Attributes

int n_events_ {0}
 
double processing_time_ {0.}
 
std::string out_trk_collection_ {"LinearRecoilTracks"}
 
std::string seed_collection_ {"LinearRecoilSeedTracks"}
 
std::string input_pass_name_ {""}
 
int n_seeds_ {0}
 
int n_tracks_ {0}
 

Additional Inherited Members

- Protected Member Functions inherited from tracking::reco::TrackingGeometryUser
const Acts::GeometryContext & geometryContext ()
 
const Acts::MagneticFieldContext & magneticFieldContext ()
 
const Acts::CalibrationContext & calibrationContext ()
 
const geo::TrackersTrackingGeometrygeometry ()
 
- Protected Member Functions inherited from framework::EventProcessor
void abortEvent ()
 Abort the event immediately.
 
- Protected Attributes inherited from framework::EventProcessor
HistogramPool histograms_
 helper object for making and filling histograms
 
NtupleManagerntuple_ {NtupleManager::getInstance()}
 Manager for any ntuples.
 
logging::logger the_log_
 The logger for this EventProcessor.
 

Detailed Description

Definition at line 24 of file LinearTrackFinder.h.

Constructor & Destructor Documentation

◆ LinearTrackFinder()

tracking::reco::LinearTrackFinder::LinearTrackFinder ( const std::string & name,
framework::Process & process )

Constructor.

Parameters
nameThe name of the instance of this object.
processThe process running this producer.

Definition at line 15 of file LinearTrackFinder.cxx.

17 : TrackingGeometryUser(name, process) {}
virtual void process(Event &event) final
Processing an event for a Producer is calling produce.

Member Function Documentation

◆ configure()

void tracking::reco::LinearTrackFinder::configure ( framework::config::Parameters & parameters)
overridevirtual

Configure the processor using the given user specified parameters.

Parameters
parametersSet of parameters used to configure this processor.

Reimplemented from framework::EventProcessor.

Definition at line 19 of file LinearTrackFinder.cxx.

19 {
20 // seeds from the event
21 seed_collection_ =
22 parameters.get<std::string>("seed_collection", "LinearRecoilSeedTracks");
23 // output track collection
24 out_trk_collection_ =
25 parameters.get<std::string>("out_trk_collection", "LinearRecoilTracks");
26
27 input_pass_name_ = parameters.get<std::string>("input_pass_name");
28
29} // configure
const T & get(const std::string &name) const
Retrieve the parameter of the given name.
Definition Parameters.h:78

References framework::config::Parameters::get().

◆ findTracks()

std::vector< ldmx::StraightTrack > tracking::reco::LinearTrackFinder::findTracks ( const std::vector< ldmx::StraightTrack > & track_seeds)
private

Definition at line 71 of file LinearTrackFinder.cxx.

72 {
73 std::vector<ldmx::StraightTrack> best_tracks;
74 std::map<std::array<double, 3>, std::vector<ldmx::StraightTrack>>
75 seeds_by_rec_hit;
76
77 // Group seeds by their EcalRecHit point
78 for (const auto& seed : track_seeds) {
79 auto rec_hit_point = seed.getFirstLayerEcalRecHit();
80 seeds_by_rec_hit[rec_hit_point].push_back(seed);
81 }
82
83 // Track used sensor positions
84 std::set<std::tuple<float, float, float>> used_sensor_positions;
85
86 // Find the best seed for each RecHit
87 // numTracks <= number of RecHits
88 for (auto& entry : seeds_by_rec_hit) {
89 const auto& rec_hit_point = entry.first;
90 auto& seeds_with_same_rec_hit = entry.second;
91
92 ldmx_log(debug) << "Processing RecHit at: (" << rec_hit_point[0] << ", "
93 << rec_hit_point[1] << ", " << rec_hit_point[2] << ")\n";
94
95 // Main function to remove seeds with overlapping sensor positions
96 seeds_with_same_rec_hit.erase(
97 std::remove_if(
98 seeds_with_same_rec_hit.begin(), seeds_with_same_rec_hit.end(),
99 [&](const ldmx::StraightTrack& seed) {
100 for (const auto& measurement : seed.getAllSensorPoints()) {
101 // Check if this sensor's position is already used
102 if (isPositionUsed(measurement, used_sensor_positions)) {
103 // Mark this seed for removal
104 return true;
105 } // ifPositionUsed
106 } // for measurement
107 // Keep the seed if no position overlap
108 return false;
109 }),
110 seeds_with_same_rec_hit.end());
111
112 // If no valid seeds remain after filtering, skip to next RecHit
113 if (seeds_with_same_rec_hit.empty()) continue;
114
115 // Find the seed with the lowest chi2 for this RecHit, this is "best" seed
116 auto best_seed_it = std::min_element(
117 seeds_with_same_rec_hit.begin(), seeds_with_same_rec_hit.end(),
118 [](const ldmx::StraightTrack& trk_a, const ldmx::StraightTrack& trk_b) {
119 return trk_a.getChi2() < trk_b.getChi2();
120 });
121
122 // Store the best seed for this RecHit
123 ldmx::StraightTrack best_seed = *best_seed_it;
124 best_tracks.push_back(best_seed);
125
126 ldmx_log(debug) << "For RecHit at: (" << rec_hit_point[0] << ", "
127 << rec_hit_point[1] << ", " << rec_hit_point[2] << ")\n";
128
129 // Add best seed's sensor position to the global used positions set
130 auto best_seed_measurement = best_seed.getAllSensorPoints();
131
132 for (auto& position_object : best_seed_measurement) {
133 used_sensor_positions.insert(
134 std::make_tuple(position_object.getGlobalPosition()[0],
135 position_object.getGlobalPosition()[1],
136 position_object.getGlobalPosition()[2]));
137 ldmx_log(debug) << "We used the following point: ("
138 << position_object.getGlobalPosition()[0] << ", "
139 << position_object.getGlobalPosition()[1] << ", "
140 << position_object.getGlobalPosition()[2] << ")\n";
141 ldmx_log(debug) << "Which gave a track a distance: "
142 << best_seed.getDistanceToRecHit()
143 << " to the closest ECalRecHit\n";
144 } // for sensor points in "best" seed
145
146 } // for entry loop, everytime we loop onto a new RecHit, we will have fewer
147 // points to check
148
149 return best_tracks;
150
151} // findTracks

◆ isPositionUsed()

bool tracking::reco::LinearTrackFinder::isPositionUsed ( const ldmx::Measurement & measurement,
const std::set< std::tuple< float, float, float > > & used_sensor_positions )
private

Definition at line 153 of file LinearTrackFinder.cxx.

155 {
156 const auto& position = std::make_tuple(measurement.getGlobalPosition()[0],
157 measurement.getGlobalPosition()[1],
158 measurement.getGlobalPosition()[2]);
159
160 return used_sensor_positions.find(position) != used_sensor_positions.end();
161
162} // isPositionUsed
std::array< float, 3 > getGlobalPosition() const
Definition Measurement.h:48

◆ onProcessEnd()

void tracking::reco::LinearTrackFinder::onProcessEnd ( )
overridevirtual

Output event statistics.

Reimplemented from framework::EventProcessor.

Definition at line 64 of file LinearTrackFinder.cxx.

64 {
65 ldmx_log(info) << "found " << n_tracks_ << " tracks / " << n_events_
66 << " events.";
67 ldmx_log(info) << "AVG Time/Event: " << std::fixed << std::setprecision(1)
68 << processing_time_ / n_events_ << " ms";
69} // onProcessEnd

◆ produce()

void tracking::reco::LinearTrackFinder::produce ( framework::Event & event)
overridevirtual

Run the processor.

Parameters
eventThe event to process.

Implements framework::Producer.

Definition at line 31 of file LinearTrackFinder.cxx.

31 {
32 std::vector<ldmx::StraightTrack> straight_tracks;
33
34 auto start = std::chrono::high_resolution_clock::now();
35
36 n_events_++;
37 if (n_events_ % 1000 == 0) ldmx_log(info) << "events processed:" << n_events_;
38
39 ldmx_log(debug) << "Retrieve the seeds::" << seed_collection_;
40
41 const std::vector<ldmx::StraightTrack> seed_tracks =
42 event.getCollection<ldmx::StraightTrack>(seed_collection_,
43 input_pass_name_);
44
45 n_seeds_ = seed_tracks.size();
46 ldmx_log(debug) << "Number of seeds::" << n_seeds_;
47
48 if (n_seeds_ > 0) {
49 straight_tracks = findTracks(seed_tracks);
50 }
51
52 n_tracks_ += straight_tracks.size();
53
54 // Add the tracks to the event
55 event.add(out_trk_collection_, straight_tracks);
56
57 auto end = std::chrono::high_resolution_clock::now();
58 auto diff = end - start;
59 processing_time_ += std::chrono::duration<double, std::milli>(diff).count();
60
61 straight_tracks.clear();
62} // produce

Member Data Documentation

◆ input_pass_name_

std::string tracking::reco::LinearTrackFinder::input_pass_name_ {""}
private

Definition at line 64 of file LinearTrackFinder.h.

64{""};

◆ n_events_

int tracking::reco::LinearTrackFinder::n_events_ {0}
private

Definition at line 56 of file LinearTrackFinder.h.

56{0};

◆ n_seeds_

int tracking::reco::LinearTrackFinder::n_seeds_ {0}
private

Definition at line 66 of file LinearTrackFinder.h.

66{0};

◆ n_tracks_

int tracking::reco::LinearTrackFinder::n_tracks_ {0}
private

Definition at line 67 of file LinearTrackFinder.h.

67{0};

◆ out_trk_collection_

std::string tracking::reco::LinearTrackFinder::out_trk_collection_ {"LinearRecoilTracks"}
private

Definition at line 60 of file LinearTrackFinder.h.

60{"LinearRecoilTracks"};

◆ processing_time_

double tracking::reco::LinearTrackFinder::processing_time_ {0.}
private

Definition at line 57 of file LinearTrackFinder.h.

57{0.};

◆ seed_collection_

std::string tracking::reco::LinearTrackFinder::seed_collection_ {"LinearRecoilSeedTracks"}
private

Definition at line 63 of file LinearTrackFinder.h.

63{"LinearRecoilSeedTracks"};

The documentation for this class was generated from the following files: