LinearTrackFinder.h

#pragma once
 
//--- Framework ---//
#include "Framework/Configure/Parameters.h"
#include "Framework/EventProcessor.h"
 
//--- C++ ---//
 
//--- LDMX ---//
#include "Tracking/Reco/TrackingGeometryUser.h"
 
//--- Tracking ---//
#include "Tracking/Event/Measurement.h"
#include "Tracking/Event/StraightTrack.h"
 
namespace tracking {
namespace reco {
 
class LinearTrackFinder : public TrackingGeometryUser {
 public:
  LinearTrackFinder(const std::string& name, framework::Process& process);
 
  virtual ~LinearTrackFinder() = default;
 
  void onProcessEnd() override;
  void configure(framework::config::Parameters& parameters) override;
 
  void produce(framework::Event& event) override;
 
 private:
  int n_events_{0};
  double processing_time_{0.};
 
  // The output track collection
  std::string out_trk_collection_{"LinearRecoilTracks"};
 
  // The seed track collection
  std::string seed_collection_{"LinearRecoilSeedTracks"};
  std::string input_pass_name_{""};
 
  int n_seeds_{0};
  int n_tracks_{0};
 
  // Find tracks from seeds by taking the seeds with lowest chi2 for each RecHit
  std::vector<ldmx::StraightTrack> findTracks(
      const std::vector<ldmx::StraightTrack>& track_seeds);
 
  // Helper function to check if a measurement's position is already used
  bool isPositionUsed(
      const ldmx::Measurement& measurement,
      const std::set<std::tuple<float, float, float>>& used_sensor_positions);
 
};  // LinearTrackFinder
 
}  // namespace reco
}  // namespace tracking