Geant4FastSimShowerModel.h

Go to the documentation of this file.

//==========================================================================
//  AIDA Detector description implementation
//--------------------------------------------------------------------------
// Copyright (C) Organisation europeenne pour la Recherche nucleaire (CERN)
// All rights reserved.
//
// For the licensing terms see $DD4hepINSTALL/LICENSE.
// For the list of contributors see $DD4hepINSTALL/doc/CREDITS.
//
// Author     : M.Frank
//
//==========================================================================
#ifndef DDG4_GEANT4FASTSIMSHOWERMODEL_H
#define DDG4_GEANT4FASTSIMSHOWERMODEL_H
 
// Framework include files
#include <DDG4/Geant4DetectorConstruction.h>
 
// Geant4 include files
#include <G4FastSimulationPhysics.hh>
 
// C/C++ include files
#include <set>
#include <vector>
#include <memory>
 
class G4FastStep;
class G4FastTrack;
class G4ParticleDefinition;
class G4VFastSimulationModel;
 
namespace dd4hep  {
 
  namespace sim  {
 
    class Geant4ShowerModelWrapper;
    
 
    class Geant4FastSimShowerModel : public Geant4DetectorConstruction    {
    protected:
      // typedef std::vector<std::pair<std::string, double> > ParticleConfig;
      typedef std::map<const G4ParticleDefinition*, double> ParticleCut;
      typedef std::map<std::string, std::string> ParticleConfig;
      typedef Geant4ShowerModelWrapper* Wrapper;
  
      std::string    m_regionName     { "Region-name-not-specified"};
      std::vector<std::string> m_applicablePartNames  { };
      bool           m_enable         { false };
      double         m_stepX0         { 0.1 };
      ParticleConfig m_eMin           { };
      ParticleConfig m_eMax           { };
      ParticleConfig m_eKill          { };
      ParticleConfig m_eTriggerNames  { };
 
      std::set<const G4ParticleDefinition*> m_applicableParticles  { };
      ParticleCut    m_eTriggerCut    { };
      G4VFastSimulationModel* m_model { nullptr };
      Wrapper        m_wrapper        { nullptr };
 
    protected:
      DDG4_DEFINE_ACTION_CONSTRUCTORS(Geant4FastSimShowerModel);
 
      G4Material* getMaterial(const std::string& name)   const;
      G4ParticleDefinition* getParticleDefinition(const std::string& name)   const;
      G4Region* getRegion(const std::string& nam)   const;
      void addShowerModel(G4Region* region);
      void killParticle(G4FastStep& step, double deposit, double step_length = 0e0);
 
    public:
      Geant4FastSimShowerModel(Geant4Context* context, const std::string& nam);
 
      virtual ~Geant4FastSimShowerModel();
 
      virtual void constructGeo(Geant4DetectorConstructionContext* ctxt)  override;
      virtual void constructField(Geant4DetectorConstructionContext* ctxt)  override;
      virtual void constructSensitives(Geant4DetectorConstructionContext* ctxt)  override;
 
      virtual bool check_applicability(const G4ParticleDefinition& particle);
      virtual bool check_trigger(const G4FastTrack& track);
      virtual void modelShower(const G4FastTrack& track, G4FastStep& step);
    };
 
 
 
    template <typename CONCRETE_MODEL>
    class Geant4FSShowerModel : public Geant4FastSimShowerModel    {
 
    public:
      CONCRETE_MODEL locals;
 
    public:
      Geant4FSShowerModel(Geant4Context* context, const std::string& nam);
 
      virtual ~Geant4FSShowerModel();
 
      void initialize();
 
      virtual void constructGeo(Geant4DetectorConstructionContext* ctxt)  override;
 
      virtual void constructField(Geant4DetectorConstructionContext* ctxt)  override;
 
      virtual void constructSensitives(Geant4DetectorConstructionContext* ctxt)  override;
 
 
      virtual bool check_applicability(const G4ParticleDefinition& particle)  override;
 
 
      virtual bool check_trigger(const G4FastTrack& track)  override;
 
      virtual void modelShower(const G4FastTrack& track, G4FastStep& step)  override;
    };
  }     /* End namespace sim   */
}       /* End namespace dd4hep */
#endif // DDG4_GEANT4FASTSIMSHOWERMODEL_H