AtlasG4EventUserInfo.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MCTRUTH_ATLASG4EVENTUSERINFO_H
#define MCTRUTH_ATLASG4EVENTUSERINFO_H
 
 
#include <GaudiKernel/EventContext.h>
#include "AtlasHepMC/GenEvent_fwd.h"
#include "AtlasHepMC/GenParticle.h"
#include "G4VUserEventInformation.hh"
#include "HitManagement/HitCollectionMap.h"
#include <memory>
class AtlasG4EventUserInfo: public G4VUserEventInformation {
public:
  AtlasG4EventUserInfo(const EventContext& ctx)
    : G4VUserEventInformation()
    , m_eventContext(ctx)
  {}
 
  HepMC::GenEvent* GetHepMCEvent() ;
  void SetHepMCEvent(HepMC::GenEvent*);
 
  HepMC::ConstGenParticlePtr GetCurrentPrimaryGenParticle() const {return m_currentPrimaryGenParticle;}
  void SetCurrentPrimaryGenParticle(HepMC::ConstGenParticlePtr p) {m_currentPrimaryGenParticle = std::move(p);}
 
  HepMC::GenParticlePtr GetCurrentGenParticle() {return m_currentGenParticle;}
  HepMC::ConstGenParticlePtr GetCurrentGenParticle() const {return m_currentGenParticle;}
  void SetCurrentGenParticle(HepMC::GenParticlePtr p) {m_currentGenParticle = std::move(p);}
 
  int GetLastProcessedTrackID() const { return m_lastProcessedTrackID; }
  void SetLastProcessedTrackID(int trackID) { m_lastProcessedTrackID = trackID; }
 
  int GetLastProcessedStep() const { return m_lastProcessedStep; }
  void SetLastProcessedStep(int stepNumber) { m_lastProcessedStep = stepNumber; }
 
  std::shared_ptr<HitCollectionMap> GetHitCollectionMap() const { return m_hitCollectionMap; }
 
  void SetHitCollectionMap(std::shared_ptr<HitCollectionMap> hitCollections) {  m_hitCollectionMap = hitCollections; }
 
  const EventContext& GetEventContext() const { return m_eventContext; }
 
  void Print() const {}
 
private:
  const EventContext& m_eventContext;
  HepMC::GenEvent *m_theEvent{};
  HepMC::ConstGenParticlePtr m_currentPrimaryGenParticle{};
  HepMC::GenParticlePtr m_currentGenParticle{};
 
  std::shared_ptr<HitCollectionMap> m_hitCollectionMap{std::make_shared<HitCollectionMap>()};
 
  // These next two variables are used by the CaloCalibrationHit
  // recording code as event-level flags They correspond to the Track
  // ID and step number of the last G4Step processed by a
  // CaloCalibrationHit SD Both are needed, because a particle might
  // have only one step
  int m_lastProcessedTrackID{0};
  int m_lastProcessedStep{0};
};
 
#endif // MCTRUTH_ATLASG4EVENTUSERINFO_H