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 "AthenaKernel/ExtendedEventContext.h"
#include "AthenaKernel/IProxyDict.h"
#include "AtlasHepMC/GenEvent_fwd.h"
#include "AtlasHepMC/GenParticle.h"
#include "HitManagement/HitCollectionMap.h"
 
#include <G4EventManager.hh>
#include "G4VUserEventInformation.hh"
 
#include <memory>
class AtlasG4EventUserInfo: public G4VUserEventInformation {
public:
  AtlasG4EventUserInfo(const EventContext& ctx)
    : G4VUserEventInformation()
    , m_eventContext(ctx)
    , m_eventStore(Atlas::getExtendedEventContext(ctx).proxy())
  {}

  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; }
 
  IProxyDict* GetEventStore() { return m_eventStore; }
 
  void Print() const {}
 
  // Static helper method to get the AtlasG4EventUserInfo from G4EventManager
  static AtlasG4EventUserInfo* GetEventUserInfo()
  {
      // Event manager may be null in unit tests.
      G4EventManager* eventManager = G4EventManager::GetEventManager();
      if (!eventManager) {
          return nullptr;
      }
      // User info may be null
      return static_cast<AtlasG4EventUserInfo*>(eventManager->GetUserInformation());
  }
 
 
private:
  const EventContext& m_eventContext;
  IProxyDict* m_eventStore{};
  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