MCTruthSteppingAction.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MCTruthBase_MCTruthSteppingAction_H
#define MCTruthBase_MCTruthSteppingAction_H
 
// System includes
#include <map>
#include <string>
#include <vector>
 
// Framework includes
#include "AthenaBaseComps/AthMessaging.h"
 
// Geant4 includes
#include "G4UserEventAction.hh"
#include "G4UserSteppingAction.hh"
 
// Local includes
#include "RecordingEnvelope.h"
 
namespace ISF
{
  class IGeoIDSvc;
  class ITruthSvc;
}
 
namespace G4UA
{

  class MCTruthSteppingAction : public G4UserEventAction,
                                public G4UserSteppingAction,
                                public AthMessaging
  {
 
    public:

      using VolumeCollectionMap_t = std::map<std::string, std::string>;

      MCTruthSteppingAction(const VolumeCollectionMap_t& collMap,
                            int secondarySavingLevel,
                            int subDetVolLevel,
                            ISF::ITruthSvc& truthRecordSvc,
                            ISF::IGeoIDSvc& geoIDSvc,
                            IMessageSvc* msgSvc, MSG::Level level);

      virtual void BeginOfEventAction(const G4Event*) override final;

      virtual void UserSteppingAction(const G4Step*) override final;
 
    private:

      void setupRecEnvelopes();

      void propagatePrimaryInfoToSecondaries(const G4Step*) const;

      void createTruthIncident(const G4Step*) const;

      bool m_isInitialized;

      int m_secondarySavingLevel;

      int m_subDetVolLevel;

      ISF::ITruthSvc& m_truthRecordSvc;

      ISF::IGeoIDSvc& m_geoIDSvc;

      VolumeCollectionMap_t m_volumeCollectionMap;

      std::vector<RecordingEnvelope> m_recordingEnvelopes;
 
  }; // class MCTruthSteppingAction
 
} // namespace G4UA
 
#endif