TrackProcessorUserActionBase.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ISF_GEANT4TOOLS_TRACKPROCESSORUSERACTIONBASE_H
#define ISF_GEANT4TOOLS_TRACKPROCESSORUSERACTIONBASE_H
 
#include "G4UserTrackingAction.hh"
#include "G4UserSteppingAction.hh"
#include "G4UserEventAction.hh"
 
#include "MCTruth/VTrackInformation.h"
 
#include "ISF_Event/ISFParticleContainer.h"
 
// forward declarations
class AtlasG4EventUserInfo;
 
#include "AtlasHepMC/GenParticle_fwd.h"
 
namespace ISF {
  class ISFParticle;
}
 
 
namespace G4UA{
 
namespace iGeant4 {
 
class TrackProcessorUserActionBase: public G4UserTrackingAction,
                                    public G4UserSteppingAction,
                                    public G4UserEventAction
{
 
public:
  TrackProcessorUserActionBase() = default;
  virtual void BeginOfEventAction(const G4Event*) override final;
  virtual void EndOfEventAction(const G4Event*) override final;
 
  virtual void PreUserTrackingAction(const G4Track*) override;
  virtual void PostUserTrackingAction(const G4Track*) override final;
 
  virtual void UserSteppingAction(const G4Step*) override final;
 
  ISF::ISFParticleContainer ReturnSecondaries(ISF::ISFParticle const* parent);
 
protected:
  AtlasG4EventUserInfo *m_atlasG4EvtUserInfo{};   
 
  ISF::ISFParticleContainer m_storedSecondaries;

  virtual void ISFSteppingAction(const G4Step*, ISF::ISFParticle*) = 0;
 
private:
  void setupPrimary(G4Track&);

  void setupSecondary(const G4Track&);

  void updateCachedParticleInfo(ISF::ISFParticle* baseISFParticle,
                          HepMC::ConstGenParticlePtr primaryGenParticle,
                          HepMC::GenParticlePtr currentGenParticle);

  VTrackInformation::TrackClassification classify(HepMC::ConstGenParticlePtr primaryGenParticle,
                               HepMC::ConstGenParticlePtr generationZeroGenParticle,
                               HepMC::ConstGenParticlePtr currentGenParticle,
                               int regenerationNumber) const;

  ISF::ISFParticle* m_curBaseISP{};
 
}; // class TrackProcessorUserActionBase
 
}// iGeant4
 
}// G4UA
 
#endif // ISF_GEANT4TOOLS_TRACKPROCESSORUSERACTIONBASE_H