RawTowerZDCDigitizer.h

#ifndef EICZDCRECO_RAWTOWERZDCDIGITIZER_H
#define EICZDCRECO_RAWTOWERZDCDIGITIZER_H

#include <fun4all/SubsysReco.h>

#include <phparameter/PHParameters.h>

#include <string>

class PHCompositeNode;
class RawTowerZDCContainer;
class RawTowerZDCGeomContainer;
class RawTowerZDCDeadMap;

class RawTowerZDC;

#include <gsl/gsl_rng.h>

//! simple tower digitizer which sum all cell to produce photon yield and pedestal noises
//! default input DST node is TOWER_SIM_DETECTOR
//! default output DST node is TOWER_RAW_DETECTOR
class RawTowerZDCDigitizer : public SubsysReco
{
 public:
  RawTowerZDCDigitizer(const std::string &name = "RawTowerZDCDigitizer");
  ~RawTowerZDCDigitizer() override;

  int InitRun(PHCompositeNode *topNode) override;
  int process_event(PHCompositeNode *topNode) override;
  void Detector(const std::string &d) { m_Detector = d; _tower_params.set_name(d);}
  void TowerType(const int type) { m_TowerType = type; }
  void set_seed(const unsigned int iseed);
  unsigned int get_seed() const { return m_Seed; }
  enum enu_digi_algorithm
  {
    //! directly pass the energy of sim tower to digitized tower
    kNo_digitization = 0,

    //! simple digitization with photon statistics, single amplitude ADC conversion and pedestal
    kSimple_photon_digitization = 1,

    //! digitization with photon statistics on SiPM with an effective pixel N, ADC conversion and pedestal
    kSiPM_photon_digitization = 2
  };

  enu_digi_algorithm
  get_digi_algorithm() const
  {
    return m_DigiAlgorithm;
  }

  void
  set_digi_algorithm(enu_digi_algorithm digiAlgorithm)
  {
    m_DigiAlgorithm = digiAlgorithm;
  }

  double
  get_pedestal_central_ADC() const
  {
    return m_PedestalCentralADC;
  }

  void
  set_pedestal_central_ADC(const double pedestalCentralAdc)
  {
    m_PedestalCentralADC = pedestalCentralAdc;
  }

  double
  get_pedestal_width_ADC() const
  {
    return m_PedestalWidthADC;
  }

  void
  set_pedestal_width_ADC(const double pedestalWidthAdc)
  {
    m_PedestalWidthADC = pedestalWidthAdc;
  }

  double
  get_photonelec_ADC() const
  {
    return m_PhotonElecADC;
  }

  void
  set_photonelec_ADC(const double photonelecAdc)
  {
    m_PhotonElecADC = photonelecAdc;
  }

  double
  get_photonelec_yield_visible_GeV() const
  {
    return m_PhotonElecYieldVisibleGeV;
  }

  void
  set_photonelec_yield_visible_GeV(const double photonelecYieldVisibleGeV)
  {
    m_PhotonElecYieldVisibleGeV = photonelecYieldVisibleGeV;
  }

  double
  get_zero_suppression_ADC() const
  {
    return m_ZeroSuppressionADC;
  }

  void
  set_zero_suppression_ADC(const double zeroSuppressionAdc)
  {
    m_ZeroSuppressionADC = zeroSuppressionAdc;
  }

  void
  set_variable_zero_suppression(const bool value)
  {
    m_ZeroSuppressionFile = value;
  }

  void
  set_variable_pedestal(const bool value)
  {
    m_pedestalFile = value;
  }

  PHParameters &
  GetParameters()
  {
    return _tower_params;
  }

  std::string
  get_raw_tower_node_prefix() const
  {
    return m_RawTowerNodePrefix;
  }

  void
  set_raw_tower_node_prefix(const std::string &rawTowerNodePrefix)
  {
    m_RawTowerNodePrefix = rawTowerNodePrefix;
  }

  std::string
  get_sim_tower_node_prefix() const
  {
    return m_SimTowerNodePrefix;
  }

  void
  set_sim_tower_node_prefix(const std::string &simTowerNodePrefix)
  {
    m_SimTowerNodePrefix = simTowerNodePrefix;
  }

  // ! SiPM effective pixel per tower, only used with kSiPM_photon_digitalization
  void set_sipm_effective_pixel(const unsigned int &d) { m_SiPMEffectivePixel = d; }

  // ! SiPM effective pixel per tower, only used with kSiPM_photon_digitalization
  unsigned int get_sipm_effective_pixel() { return m_SiPMEffectivePixel; }

 private:
  void CreateNodes(PHCompositeNode *topNode);

  //! simple digitization with photon statistics, ADC conversion and pedestal
  //! \param  sim_tower simulation tower input
  //! \return a new RawTowerZDC object contain digitalized value of ADC output in RawTowerZDC::get_energy()
  RawTowerZDC *simple_photon_digitization(RawTowerZDC *sim_tower);

  //! digitization with photon statistics on SiPM with an effective pixel N, ADC conversion and pedestal
  //! this function use the effective pixel to count for the effect that the sipm is not evenly lit
  RawTowerZDC *sipm_photon_digitization(RawTowerZDC *sim_tower);

  enu_digi_algorithm m_DigiAlgorithm;

  RawTowerZDCContainer *m_SimTowers;
  RawTowerZDCContainer *m_RawTowers;
  RawTowerZDCGeomContainer *m_RawTowerGeom;
  RawTowerZDCDeadMap *m_DeadMap;

  std::string m_Detector;

  std::string m_SimTowerNodePrefix;
  std::string m_RawTowerNodePrefix;

  //! photon electron yield per GeV of visible energy
  double m_PhotonElecYieldVisibleGeV;

  //! photon electron per ADC unit
  double m_PhotonElecADC;

  //! pedestal central in unit of ADC
  double m_PedestalCentralADC;

  //! pedestal width in unit of ADC
  double m_PedestalWidthADC;

  //! pedestal from file
  bool m_pedestalFile;

  //! zero suppression in unit of ADC
  double m_ZeroSuppressionADC;

  //! zero suppression from file
  bool m_ZeroSuppressionFile;

  //! tower type to act on
  int m_TowerType;

  unsigned int m_Seed;

  // ! SiPM effective pixel per tower, only used with kSiPM_photon_digitalization
  unsigned int m_SiPMEffectivePixel;

  PHParameters _tower_params;

  gsl_rng *m_RandomGenerator;
};

#endif /* G4CALO_RAWTOWERDIGITIZER_H */