SiStripWaveform.h

#ifndef TRACKING_EVENT_SISTRIPWAVEFORM_H_
#define TRACKING_EVENT_SISTRIPWAVEFORM_H_
 
#include <algorithm>
#include <iostream>
#include <vector>
 
#include "TObject.h"
 
namespace ldmx {
 
class SiStripWaveform {
 public:
  SiStripWaveform() = default;
 
  SiStripWaveform(std::vector<short> samples, int16_t pchannel,
                  uint8_t hybrid_id, uint8_t feb_id, uint8_t n_triggers);
 
  virtual ~SiStripWaveform() {}
 
  void clear();
 
  const std::vector<short>& getSamples() const { return samples_; }
  int16_t getPchannel() const { return pchannel_; }
  uint8_t getHybridId() const { return hybrid_id_; }
  uint8_t getFebId() const { return feb_id_; }
  uint8_t getNTriggers() const { return n_triggers_; }
 
  void setFitResult(float amplitude, float t0, float chi2, int ndf,
                    bool converged) {
    fit_amplitude_ = amplitude;
    fit_t0_        = t0;
    fit_chi2_      = chi2;
    fit_ndf_       = ndf;
    fit_converged_ = converged;
  }
 
  float getFitAmplitude() const { return fit_amplitude_; }
  float getFitT0()        const { return fit_t0_; }
  float getFitChi2()      const { return fit_chi2_; }
  int   getFitNDF()       const { return fit_ndf_; }
  bool  isFitConverged()  const { return fit_converged_; }
 
  short getSample(uint8_t t, uint8_t s) const { return samples_[t * 3 + s]; }
 
  short peakAmplitude() const {
    if (samples_.empty()) return 0;
    return *std::max_element(samples_.begin(), samples_.end());
  }
 
  uint8_t peakTrigger() const {
    if (samples_.empty()) return 0;
    auto it = std::max_element(samples_.begin(), samples_.end());
    return static_cast<uint8_t>(std::distance(samples_.begin(), it) / 3);
  }
 
  friend std::ostream& operator<<(std::ostream& o, const SiStripWaveform& w);
 
 protected:
  std::vector<short>
      samples_;          
  int16_t pchannel_{0};  
  uint8_t hybrid_id_{0};
  uint8_t feb_id_{0};
  uint8_t n_triggers_{0};  
 
  // Pulse-shape fit result (filled by SiStripWaveformBuilder).
  float fit_amplitude_{0};       
  float fit_t0_{0};              
  float fit_chi2_{0};            
  int   fit_ndf_{0};             
  bool  fit_converged_{false};   
 
  ClassDef(SiStripWaveform, 2);
};
 
}  // namespace ldmx
 
#endif  // TRACKING_EVENT_SISTRIPWAVEFORM_H_