AtlasG4EventUserInfo.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MCTRUTH_ATLASG4EVENTUSERINFO_H
#define MCTRUTH_ATLASG4EVENTUSERINFO_H
 
#include <memory>
#include "AtlasHepMC/GenEvent_fwd.h"
#include "AtlasHepMC/GenParticle.h"
#include "G4VUserEventInformation.hh"
#include "HitManagement/HitCollectionMap.h"
 
class AtlasG4EventUserInfo: public G4VUserEventInformation {
public:
  AtlasG4EventUserInfo()
    : G4VUserEventInformation()
  {}
 
  HepMC::GenEvent* GetHepMCEvent() ;
  void SetHepMCEvent(HepMC::GenEvent*);
 
  HepMC::ConstGenParticlePtr GetCurrentPrimaryGenParticle() const {return m_currentPrimaryGenParticle;}
  void SetCurrentPrimaryGenParticle(HepMC::ConstGenParticlePtr p) {m_currentPrimaryGenParticle = 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; }
 
  void Print() const {}
 
private:
  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