SensitiveDetector.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Poyuan Chen                                              *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#ifndef ECLSENSITIVEDETECTOR_H_
#define ECLSENSITIVEDETECTOR_H_
 
#include <simulation/kernel/SensitiveDetectorBase.h>
#include <simulation/dataobjects/BeamBackHit.h>
#include <framework/datastore/StoreArray.h>
#include <framework/datastore/RelationArray.h>
#include <framework/database/DBObjPtr.h>
#include <ecl/dataobjects/ECLSimHit.h>
#include <ecl/dataobjects/ECLHit.h>
#include <ecl/dbobjects/ECLHadronComponentEmissionFunction.h>
 
#include "G4EmCalculator.hh"
 
#include "TGraph.h"
 
namespace Belle2 {
  namespace ECL {
    class ECLGeometryPar;
    class SensitiveDetector: public Simulation::SensitiveDetectorBase {
    public:
      SensitiveDetector(G4String, G4double, G4double);
 
      ~SensitiveDetector();
 
      void Initialize(G4HCofThisEvent* HCTE) override;
 
      bool step(G4Step* aStep, G4TouchableHistory* history) override;
 
      void EndOfEvent(G4HCofThisEvent* eventHC) override;
 
    private:
      TGraph* m_HadronEmissionFunction = nullptr;  
      double GetHadronIntensityFromDEDX(double);  
      int saveSimHit(G4int, G4int, G4int, G4double, G4double, const G4ThreeVector&, const G4ThreeVector&, double);
      // members of SensitiveDetector
      // G4double m_thresholdEnergyDeposit;// Energy Deposit  threshold
      // G4double m_thresholdKineticEnergy;// Kinetic Energy  threshold
      G4EmCalculator m_emCal;  
      StoreArray<ECLSimHit> m_eclSimHits;   
      StoreArray<ECLHit> m_eclHits;         
      StoreArray<MCParticle> m_mcParticles; 
      RelationArray m_eclSimHitRel; 
      RelationArray m_eclHitRel;    
      int m_trackID;                
      double m_WeightedTime;        
      double m_energyDeposit;       
      double m_hadronenergyDeposit;     
      G4ThreeVector m_WeightedPos;  
      G4ThreeVector m_momentum;     
      DBObjPtr<ECLHadronComponentEmissionFunction> m_ECLHadronComponentEmissionFunction;  
    };
    class SensitiveDiode: public Simulation::SensitiveDetectorBase {
    public:
      explicit SensitiveDiode(const G4String&);
 
      ~SensitiveDiode();
 
      void Initialize(G4HCofThisEvent* HCTE) override;
 
      bool step(G4Step* aStep, G4TouchableHistory* history) override;
 
      void EndOfEvent(G4HCofThisEvent* eventHC) override;
 
    private:
      // members of SensitiveDiode
      ECLGeometryPar* m_eclp; 
      struct hit_t {
        int cellId; 
        double e; 
        double t; 
      };
      int m_trackID; 
      double m_tsum; 
      double m_esum; 
      std::vector<hit_t> m_hits; 
      std::vector<int> m_cells; 
      StoreArray<ECLHit> m_eclHits;  
    };
 
    class BkgSensitiveDiode: public Simulation::SensitiveDetectorBase {
    public:
      explicit BkgSensitiveDiode(const G4String&);
 
      bool step(G4Step* aStep, G4TouchableHistory* history) override;
    private:
      int m_trackID;          
      TVector3 m_startPos;    
      TVector3 m_startMom;    
      double m_startTime;     
      double m_startEnergy;   
      double m_energyDeposit; 
      double m_trackLength;   
      ECLGeometryPar* m_eclp; 
      StoreArray<MCParticle> m_mcParticles;     
      StoreArray<BeamBackHit> m_eclBeamBkgHits; 
      RelationArray m_eclBeamBkgHitRel;         
    };
  } // end of namespace ecl
} // end of namespace Belle2
 
#endif /* ECLSENSITIVEDETECTOR_H_ */