CaloWaveformProcessing.h

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

#ifndef CALORECO_CALOWAVEFORMPROCESSING_H
#define CALORECO_CALOWAVEFORMPROCESSING_H

#include <fun4all/SubsysReco.h>

#include <string>
#include <vector>

class CaloWaveformFitting;

class CaloWaveformProcessing : public SubsysReco
{
 public:
  enum process
  {
    NONE = 0,
    TEMPLATE = 1,
    ONNX = 2,
    FAST = 3,
  };

  CaloWaveformProcessing() = default;
  ~CaloWaveformProcessing() override;

  void set_processing_type(CaloWaveformProcessing::process modelno)
  {
    m_processingtype = modelno;
    return;
  }

  CaloWaveformProcessing::process get_processing_type()
  {
    return m_processingtype;
  }

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

  void set_template_name(const std::string &template_name)
  {
    m_template_name = template_name;
    return;
  }
  void set_model_file(const std::string &model_name)
  {
    m_model_name = model_name;
    return;
  }

  void set_nthreads(int nthreads);

  int get_nthreads();

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


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

  void initialize_processing();

 private:
  CaloWaveformFitting *m_Fitter = nullptr;

  CaloWaveformProcessing::process m_processingtype = CaloWaveformProcessing::TEMPLATE;
  int _nthreads = 1;
  int _nsoftwarezerosuppression = 40;
  bool _bdosoftwarezerosuppression = false;

  std::string m_template_input_file;
  std::string url_template;
  std::string m_template_name = "NONE";

  std::string url_onnx;
  std::string m_model_name = "CEMC_ONNX";
};
#endif