PHG4TpcDigitizer.h

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

// Tell emacs that this is a C++ source
//  -*- C++ -*-.
#ifndef G4TPC_PHG4TPCDIGITIZER_H
#define G4TPC_PHG4TPCDIGITIZER_H

#include <fun4all/SubsysReco.h>

#include <trackbase/TrkrDefs.h>
#include <trackbase/TrkrHitSet.h>

#include <map>
#include <string>   // for string
#include <utility>  // for pair, make_pair
#include <vector>

#include <gsl/gsl_rng.h>

class PHCompositeNode;

class PHG4TpcDigitizer : public SubsysReco
{
 public:
  PHG4TpcDigitizer(const std::string &name = "PHG4TpcDigitizer");
  ~PHG4TpcDigitizer() override;

  int Init(PHCompositeNode * /*topNode*/) override { return 0; }

  int InitRun(PHCompositeNode *topNode) override;

  int process_event(PHCompositeNode *topNode) override;

  int End(PHCompositeNode * /*topNode*/) override { return 0; };

  void set_adc_scale(const int layer, const unsigned int max_adc, const float energy_per_adc)
  {
    _max_adc.insert(std::make_pair(layer, max_adc));
    _energy_scale.insert(std::make_pair(layer, energy_per_adc));
  }

  void SetTpcMinLayer(const int minlayer) { TpcMinLayer = minlayer; };
  void SetADCThreshold(const float thresh) { ADCThreshold = thresh; };
  void SetENC(const float enc) { TpcEnc = enc; };
  void set_drift_velocity(float vd) {_drift_velocity = vd;}
  void set_skip_noise_flag(const bool skip) {skip_noise = skip;}

 private:
  void CalculateCylinderCellADCScale(PHCompositeNode *topNode);
  void DigitizeCylinderCells(PHCompositeNode *topNode);
  float added_noise();
  float add_noise_to_bin(float signal);
  
  unsigned int TpcMinLayer;
  unsigned int TpcNLayers;
  float ADCThreshold;
  float ADCThreshold_mV = 0;
  float TpcEnc;
  float Pedestal;
  float ChargeToPeakVolts;
  float _drift_velocity = 8.0e-3;  // override from macro with simulation drift velocity

  float ADCSignalConversionGain;
  float ADCNoiseConversionGain;

  bool skip_noise = false;

  std::vector<std::vector<TrkrHitSet::ConstIterator> > phi_sorted_hits;
  std::vector<std::vector<TrkrHitSet::ConstIterator> > t_sorted_hits;

  std::vector<float> adc_input;
  std::vector<TrkrDefs::hitkey> adc_hitid;
  std::vector<int> is_populated;

  // settings
  std::map<int, unsigned int> _max_adc;
  std::map<int, float> _energy_scale;

  gsl_rng *RandomGenerator;
};

#endif