StripClusterer.h

#pragma once
 
#include <vector>
 
#include "Tracking/Event/FittedSiStripHit.h"
 
namespace tracking::digitization {
 
class StripClusterer {
 public:
  // I like not having underscores for struct members,
  // but that's not settable until a later version of clang-tidy
  // NOLINTBEGIN(readability-identifier-naming)
  struct ClusterCandidate {
    double centroid_strip{0};   
    double total_amplitude{0};  
    double time_ns{0};          
    double sigma_strip{0};      
    int n_strips{0};
    int layer_id{-1};
    std::vector<int> strip_ids;
  };
  // NOLINTEND(readability-identifier-naming)
 
  StripClusterer(double seed_threshold = 4.0, double neighbor_threshold = 3.0,
                 double cluster_threshold = 4.0, double noise_sigma_adc = 5.0,
                 double mean_time_ns = 0.0, double time_window_ns = -1.0,
                 double neighbor_delta_t_ns = -1.0, double max_chi2_ndf = -1.0);
 
  std::vector<ClusterCandidate> findClusters(
      const std::vector<ldmx::FittedSiStripHit>& hits) const;
 
 private:
  bool passesSeedCuts(const ldmx::FittedSiStripHit& h) const;
  bool passesNeighborCuts(const ldmx::FittedSiStripHit& h,
                          double cluster_weighted_t,
                          double cluster_total_amp) const;
 
  double hitNoise(const ldmx::FittedSiStripHit& h) const {
    const double n = h.getNoise();
    return n > 0.0 ? n : noise_sigma_adc_;
  }
 
  double seed_threshold_;
  double neighbor_threshold_;
  double cluster_threshold_;
  double noise_sigma_adc_;
  double mean_time_ns_;
  double time_window_ns_;
  double neighbor_delta_t_ns_;
  double max_chi2_ndf_;
};
 
}  // namespace tracking::digitization