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