CaloWaveformFitting.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file CaloWaveformFitting.h

#ifndef CALORECO_CALOWAVEFORMFITTING_H
#define CALORECO_CALOWAVEFORMFITTING_H

#include <string>
#include <vector>

class TProfile;

class CaloWaveformFitting
{
 public:
  CaloWaveformFitting() = default;
  ~CaloWaveformFitting() = default;

  void set_template_file(const std::string &template_input_file)
  {
    m_template_input_file = template_input_file;
    return;
  }

  void set_nthreads(int nthreads)
  {
    _nthreads = nthreads;
    return;
  }

  void set_softwarezerosuppression(bool usezerosuppression,int softwarezerosuppression)
  {
    _nsoftwarezerosuppression = softwarezerosuppression;
    _bdosoftwarezerosuppression = usezerosuppression;
  }

  int get_nthreads() 
  {
    return _nthreads;
  }

  std::vector<std::vector<float>> process_waveform(std::vector<std::vector<float>> waveformvector);
  std::vector<std::vector<float>> calo_processing_templatefit(std::vector<std::vector<float>> chnlvector);
  std::vector<std::vector<float>> calo_processing_fast(std::vector<std::vector<float>> chnlvector);

  void initialize_processing(const std::string &templatefile);

 private:
  void FastMax(float x0, float x1, float x2, float y0, float y1, float y2, float &xmax, float &ymax);
  TProfile *h_template = nullptr;
  double template_function(double *x, double *par);
  int _nthreads = 1;
  int _nzerosuppresssamples = 2;
  int _nsoftwarezerosuppression = 40;
  bool _bdosoftwarezerosuppression = false;
  std::string m_template_input_file;
  std::string url_template;
  double m_peakTimeTemp = 0;

  std::string url_onnx;
  std::string m_model_name;
};
#endif