LArFastShower.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef LARG4FASTSIMULATION_LARFASTSHOWER_H
#define LARG4FASTSIMULATION_LARFASTSHOWER_H
 
// STL includes
#include <string>
#include <map>
 
// Geant4 includes
#include "G4VFastSimulationModel.hh"
 
// Local includes
#include "FastShowerConfigStruct.h"
 
// Forward declarations
#include "AtlasHepMC/GenEvent_fwd.h"
#include "AtlasHepMC/IO_GenEvent.h"
#ifdef HEPMC3
#include "HepMC3/WriterAscii.h"
#endif
 
class IFastSimDedicatedSD;
class ILArG4ShowerLibSvc;
 
 
class LArFastShower : public G4VFastSimulationModel
{
  public:
 
    LArFastShower(const std::string& name, const FastShowerConfigStruct& config,
                  IFastSimDedicatedSD* fastSimDedicatedSD);
 
    virtual ~LArFastShower() {}
 
    G4bool IsApplicable(const G4ParticleDefinition&) override;
 
    virtual G4bool ModelTrigger(const G4FastTrack &) override;
 
    virtual G4bool ForcedAccept(const G4FastTrack &);
    virtual G4bool ForcedDeny  (const G4FastTrack &); 
 
    void DoIt(const G4FastTrack&, G4FastStep&) override;
 
  protected:
 
    IFastSimDedicatedSD* fastShowerSD();
    ILArG4ShowerLibSvc* showerLibSvc();
 
    void KillParticle(const G4FastTrack&, G4FastStep&);
    void UseShowerLib(const G4FastTrack&, G4FastStep&);
 
    virtual G4bool CheckContainment(const G4FastTrack &fastTrack);
 
    std::unique_ptr<const HepMC::GenEvent> GetGenEvent(const G4FastTrack &fastTrack);
 
    bool   flagToShowerLib  ( const G4ParticleDefinition& particleType ) const;
    double maxEneToShowerLib( const G4ParticleDefinition& particleType ) const;
    double minEneToShowerLib( const G4ParticleDefinition& particleType ) const;
 
    bool generateFSStartingPoint( std::unique_ptr<const HepMC::GenEvent> &ge) const;
 
  private:
    const FastShowerConfigStruct m_configuration;
 
    IFastSimDedicatedSD *m_fastSimDedicatedSD;
    ILArG4ShowerLibSvc *m_showerLibSvc;
 
    // data members for configuration
    bool m_generate_starting_points;
#ifdef HEPMC3
    std::shared_ptr<HepMC3::WriterAscii>  m_starting_points_file;
#else
    std::shared_ptr<HepMC::IO_GenEvent>  m_starting_points_file;
#endif
    std::string m_detector_tag_str;
    std::map<std::string,int> m_detmap;
 
    int m_eventNum;
    std::map<int,bool> m_applicableMap;
 
};
 
#endif //LARG4FASTSIMULATION_LARFASTSHOWER_H