CsIDigitizerModule.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#ifndef CSIDIGITIZERMODULE_H
#define CSIDIGITIZERMODULE_H
 
#include <framework/core/Module.h>
#include <framework/datastore/StoreArray.h>
#include <beast/csi/dataobjects/CsiDigiHit.h>
#include <beast/csi/dataobjects/CsiSimHit.h>
#include <beast/csi/dataobjects/CsiHit.h>
 
#include <vector>
 
 
 
namespace Belle2 {
  namespace csi {
 
    typedef std::vector<double> Signal;
    typedef std::vector<int> DigitalSignal;
 
    class CsIDigitizerModule : public Module {
 
    public:
 
      CsIDigitizerModule();
 
      virtual ~CsIDigitizerModule();
 
      virtual void initialize() override;
 
      virtual void beginRun() override;
 
      virtual void event() override;
 
      virtual void endRun() override;
 
      virtual void terminate() override;
 
 
    private:
 
      Signal genTimeSignal(double _energy, double _timeAvg, double _timeRMS, int iChannel, bool _save = 0);
 
      double genTimeSignal(Signal* _output, Signal _energies, Signal _times,  int _iChannel, int _dt, int _nsam, bool _save = 0);
 
      Signal genSignalTemplate(int _n, int _i0, double _t1, double _t2 = 0.0, double _rFastTot = 1.0);
 
 
      Signal firstOrderResponse(double _gain, Signal _u, double _y0, double _dt, double _tSlow, double _delay);
 
 
      //      Signal firstOrderResponse(double _gain, Signal _u, double _y0, double _dt, double _tSlow, double _delay, double _tRatio, double _tFast);
 
 
 
      uint16_t  doChargeIntegration(Signal _u, int _NsamBL, uint16_t* BSL, uint32_t* Q, uint32_t* t,
                                    std::vector<uint16_t>* _Waveform, std::vector<uint8_t>* _DPPCIBits,
                                    int _Treshold, double _TriggerHoldoff = 0.0, double _GateWidth = 320.0,
                                    double _GateOffset = 40.0, bool _recordTraces = false);
 
 
      int addNoise(Signal* y, double _rms, double  _offset);
 
      DigitalSignal  doDigitization(Signal _v, double _LSB);
 
 
      double f(double fi, double u_i, double u_j, double y, double invtau);
 
 
 
 
      int getnSamples() const { return m_nSamples; }
 
      void setnSamples(int nsamples) { m_nSamples = nsamples; }
 
 
    private:
 
      int    m_hitNum;
 
      // Member data objects
      double m_TrueEdep;          
      double m_Resolution;  
      double m_SampleRate;  
      double m_dt;          
      int m_nWaveforms;     
      int m_nWFcounter;     
      uint8_t   m_CellId;          
      uint16_t  m_Baseline;        
      uint32_t  m_Charge;          
      uint16_t  m_MaxADC;          
      uint32_t  m_Time;            
      int       m_nSamples;               
      std::vector<uint16_t>  m_Waveform;  
      std::vector<uint8_t>   m_DPPCIBits; 
      const double m_tRisePMT = 2;   
      const double m_tTransitPMT = 48; 
      const double m_Zl = 50;    
      StoreArray<CsiHit>     m_aHit; 
      StoreArray<CsiSimHit>  m_aSimHit; 
      StoreArray<CsiDigiHit> m_aDigiHit; 
      Signal m_CsITlSignalTemplate; 
      std::vector<double> m_calibConstants; 
      std::vector<double> m_noiseLevels;    
      std::vector<double> m_LY;  
      std::vector<double> m_tRatio; 
      std::vector<double> m_tFast; 
      std::vector<double> m_tSlow; 
      std::vector<double> m_LCE; 
      std::vector<double> m_PmtQE; 
      std::vector<double> m_PmtGain; 
      Signal m_SimHitTimes[16]; 
      Signal m_SimHitEdeps[16];  
    };
 
  }
}
 
#endif /* CSIDIGITIZERMODULE_H */