SVDDigitizerModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010-2011  Belle II Collaboration                         *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Zbynek Drasal, Martin Ritter, Peter Kvasnicka            *
 *                                                                        *
 **************************************************************************/
 
#ifndef SVDDigitizerModule_H
#define SVDDigitizerModule_H
 
#include <framework/core/Module.h>
#include <svd/dataobjects/SVDSimHit.h>
#include <svd/simulation/SVDSignal.h>
#include <svd/geometry/SensorInfo.h>
#include <svd/calibration/SVDFADCMaskedStrips.h>
#include <svd/online/SVDOnlineToOfflineMap.h>
#include <framework/database/PayloadFile.h>
 
#include <string>
 
#include <root/TVector3.h>
#include <root/TFile.h>
#include <root/TTree.h>
#include <root/TH1D.h>
 
namespace Belle2 {
  namespace SVD {
 
    typedef std::map<short int, SVDSignal> StripSignals;
 
    typedef std::pair<StripSignals, StripSignals> Sensor;
 
    typedef std::map<VxdID, Sensor> Sensors;
 
    class SVDDigitizerModule : public Module {
    public:
      SVDDigitizerModule();
 
      void processHit();
 
      void driftCharge(const TVector3& position, double carriers, SVD::SensorInfo::CarrierType carrierType);
 
      double addNoise(double charge, double noise);
 
      void saveDigits();
 
      void saveWaveforms();
 
      void saveSignals();
 
      virtual void initialize() override;
      virtual void beginRun() override;
      virtual void event() override;
      virtual void terminate() override;
 
    protected:
 
      // Members holding module parameters:
 
      // 1. Collections
      std::string m_storeMCParticlesName = "";
      std::string m_storeSimHitsName = "";
      std::string m_storeTrueHitsName = "";
      std::string m_relMCParticleSimHitName = "";
      std::string m_relTrueHitSimHitName = "";
      std::string m_storeShaperDigitsName = "";
      std::string m_relShaperDigitMCParticleName = "";
      std::string m_relShaperDigitTrueHitName = "";
      std::string m_svdEventInfoName = "SVDEventInfoSim";
 
      // 2. Physics
      double m_segmentLength = 0.020;
      double m_widthOfDiffusCloud = 3.0;
 
      // 3. Noise
      bool  m_applyPoisson = true;
      bool  m_applyNoise = false;
      double m_SNAdjacent = 3.0;
      bool m_roundZS = true;
      unsigned short m_nSamplesOverZS = 1;
      double m_noiseFraction = 0.01;
 
      // 4. Timing
      double m_shapingTime = 250.0;
      double m_samplingTime = 16000. / 509.;
      bool m_randomizeEventTimes = false;
      float m_minTimeFrame = -300;
      float m_maxTimeFrame = 150;
      float m_currentEventTime = 0.0;
 
 
      double m_startSampling = -2.0;
 
      // 5. Reporting
      std::string m_rootFilename = "";
      bool m_storeWaveforms = false;
      std::string m_signalsList = "";
 
      // Other data members:
 
      Sensors m_sensors;
 
      const SVDSimHit*   m_currentHit = nullptr;
      int                m_currentParticle = -1;
      int                m_currentTrueHit = -1;
      Sensor*            m_currentSensor = NULL;
      const SensorInfo*  m_currentSensorInfo = nullptr;
      double m_currentTime = 0;
      double m_sensorThickness = 0.03;
      double m_depletionVoltage = 40;
      double m_biasVoltage = 100;
      double m_backplaneCapacitanceU = 0.1; //pF/cm
      double m_interstripCapacitanceU = 0.7; //pF/cm
      double m_couplingCapacitanceU = 15; //pF/cm
      double m_backplaneCapacitanceV = 0.4; //pF/cm
      double m_interstripCapacitanceV = 0.7; //pF/cm
      double m_couplingCapacitanceV = 30; //pF/cm
      //      double m_aduEquivalentU;
      //      double m_aduEquivalentV;
      double m_elNoiseU = 500; //e-
      double m_elNoiseV = 500; //e-
 
      //run-dependent MC payloads:
      SVDFADCMaskedStrips m_MaskedStr; 
      static std::string m_xmlFileName ;
      DBObjPtr<PayloadFile> m_mapping; 
      std::unique_ptr<SVDOnlineToOfflineMap> m_map; 
      // ROOT stuff:
      TFile* m_rootFile = nullptr;
      TH1D*  m_histChargeSharing_u = nullptr;
      TH1D*  m_histChargeSharing_v = nullptr;
 
      TH1D*  m_histMobility_e = nullptr;
      TH1D*  m_histMobility_h = nullptr;
      TH1D*  m_histVelocity_e = nullptr;
      TH1D*  m_histVelocity_h = nullptr;
      TH1D*  m_histDistanceToPlane_e = nullptr;
      TH1D*  m_histDistanceToPlane_h = nullptr;
      TH1D*  m_histDriftTime_e = nullptr;
      TH1D*  m_histDriftTime_h = nullptr;
 
      TH1D*  m_histLorentz_u = nullptr;
      TH1D*  m_histLorentz_v = nullptr;
      TH1D*  m_signalDist_u = nullptr;
      TH1D*  m_signalDist_v = nullptr;
      TTree* m_waveTree = nullptr;
 
    };//end class declaration
 
 
  } // end namespace SVD
} // end namespace Belle2
 
#endif // SVDDigitizerModule_H