HcalDigiProducer.h

#ifndef HCAL_HCALDIGIPRODUCER_H_
#define HCAL_HCALDIGIPRODUCER_H_
 
//----------------//
//   C++ StdLib   //
//----------------//
#include <memory>  //for smart pointers
#include <random>  //for random num generators
 
//----------//
//   LDMX   //
//----------//
#include <TRandom2.h>
 
#include "Framework/EventProcessor.h"
#include "Tools/HgcrocEmulator.h"
#include "Tools/NoiseGenerator.h"
 
namespace hcal {
 
class HcalDigiProducer : public framework::Producer {
 public:
  HcalDigiProducer(const std::string& name, framework::Process& process);
 
  virtual ~HcalDigiProducer() = default;
 
  void configure(framework::config::Parameters&) override;
 
  void produce(framework::Event& event) override;
 
  virtual void onNewRun(const ldmx::RunHeader& runHeader) override;
 
  enum class TimeSpreadType {
    GAUSSIAN = 0,
    UNIFORM = 1,
    CONSTANT = 2,
  };
 
  double makeTimeDelta(TimeSpreadType kind,
                       std::vector<double>& parameters) const;
 
 private:
  // Python Configuration Parameters
 
  std::string input_coll_name_;
 
  std::string input_pass_name_;
 
  std::string digi_coll_name_;
 
  std::string pulse_truth_coll_name_;
 
  double clock_cycle_;
 
  int n_adcs_;
 
  int i_soi_;
 
  double mev_;
 
  double attlength_;
 
  // Other member variables
 
  bool noise_{true};
 
  bool save_pulse_truth_info_{false};
 
  bool zero_suppression_{true};
 
  std::unique_ptr<ldmx::HgcrocEmulator> hgcroc_;
 
  double ns_;
 
  std::unique_ptr<TRandom2> random_time_;
  bool do_time_spread_per_spill_{false};
  TimeSpreadType time_spread_per_spill_type_{TimeSpreadType::UNIFORM};
  std::vector<double> time_spread_per_spill_parameters_{};
 
  bool do_time_spread_per_hit_{false};
  TimeSpreadType time_spread_per_hit_type_{TimeSpreadType::GAUSSIAN};
  std::vector<double> time_spread_per_hit_parameters_{};
 
  double flat_time_shift_{0.};
 
  double readout_threshold_;
  double pedestal_;
  double gain_;
  double noise_rms_;
 
  std::unique_ptr<ldmx::NoiseGenerator> noise_generator_;
 
  std::mt19937 rng_;
};
}  // namespace hcal
 
#endif