TrackProcessorUserActionBase.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
// class header
#include "TrackProcessorUserActionBase.h"
 
// ISF includes
#include "ISF_Event/ISFParticle.h"
#include "ISF_Event/EntryLayer.h"
 
#include "ISF_Interfaces/IParticleBroker.h"
 
// ISF Geant4 includes
#include "ISF_Geant4Event/ISFG4Helper.h"
 
// Athena includes
#include "AtlasDetDescr/AtlasRegion.h"
#include "CxxUtils/checker_macros.h"
 
#include "MCTruth/AtlasG4EventUserInfo.h"
#include "MCTruth/PrimaryParticleInformation.h"
#include "MCTruth/TrackHelper.h"
#include "MCTruth/TrackInformation.h"
 
#include "StoreGate/StoreGateSvc.h"
 
// Geant4 includes
#include "G4ParticleDefinition.hh"
#include "G4DynamicParticle.hh"
#include "G4TouchableHistory.hh"
#include "G4Step.hh"
#include "G4TransportationManager.hh"
#include "G4LogicalVolumeStore.hh"
 
#include "G4EventManager.hh"
#include "G4Event.hh"
#include "G4PrimaryParticle.hh"
 
//#include "G4VPhysicalVolume.hh"
 
#include <iostream>
 
namespace G4UA {
 
namespace iGeant4 {
 
void TrackProcessorUserActionBase::BeginOfEventAction(const G4Event*)
{
  m_curBaseISP = nullptr;
  m_atlasG4EvtUserInfo = ::iGeant4::ISFG4Helper::getAtlasG4EventUserInfo();
  return;
}
 
void TrackProcessorUserActionBase::EndOfEventAction(const G4Event*)
{
  m_curBaseISP = nullptr;
  m_atlasG4EvtUserInfo = nullptr;
  return;
}
 
void TrackProcessorUserActionBase::UserSteppingAction(const G4Step* aStep)
{
  // get geoID from parent
  //TODO ELLI AtlasDetDescr::AtlasRegion curGeoID = m_curBaseISP->nextGeoID();
  //TODO ELLI ATH_MSG_DEBUG( "Currently simulating TrackID = " << aStep->GetTrack()->GetTrackID() <<
  //TODO ELLI                " inside geoID = " << curGeoID );
 
  // AdePT currently does not return the G4Secondary vector per G4Step. As the vector is required to attach
  // the G4VUserInfo in the PreUserTrackingAction below, the G4VUserInfo is not available, which
  // would lead to nullptr access crashes
#ifdef ATHSIMULATION_USE_ADEPT
  return;
#endif
  //
  // call the ISFSteppingAction method of the implementation
  //
  ISFSteppingAction( aStep, m_curBaseISP );
 
  //
  // propagate the current ISFParticle link to all secondaries
  //
  const std::vector<const G4Track*>  *secondaryVector = aStep->GetSecondaryInCurrentStep();
  if (secondaryVector && !secondaryVector->empty()) {
    for ( auto* aConstSecondaryTrack : *secondaryVector ) {
      // get a non-const G4Track for current secondary (nasty!)
      G4Track* aSecondaryTrack ATLAS_THREAD_SAFE = const_cast<G4Track*>( aConstSecondaryTrack ); // imposed by Geant4 interface
 
      auto *trackInfo = ::iGeant4::ISFG4Helper::getISFTrackInfo(*aSecondaryTrack);
 
      // G4Tracks aready returned to ISF will have a TrackInformation attached to them
      bool particleReturnedToISF = trackInfo && trackInfo->GetReturnedToISF();
      if (!particleReturnedToISF) {
        HepMC::GenParticlePtr generationZeroGenParticle{};
        ::iGeant4::ISFG4Helper::attachTrackInfoToNewG4Track( *aSecondaryTrack,
                                                  *m_curBaseISP,
                                                  VTrackInformation::Secondary,
                                                  generationZeroGenParticle );
      }
    } // <- loop over secondaries from this step
  }
 
  return;
}
 
void TrackProcessorUserActionBase::PreUserTrackingAction(const G4Track* aTrack)
{
  // AdePT currently does not return the G4Secondary vector per G4Step. As the vector is required to attach
  // the G4VUserInfo below the G4VUserInfo is not available, which would lead to nullptr access crashes
#ifdef ATHSIMULATION_USE_ADEPT
  return;
#endif
 
  bool isPrimary = ! aTrack->GetParentID();
  if (isPrimary) {
    G4Track* nonConstTrack ATLAS_THREAD_SAFE = const_cast<G4Track*> (aTrack); // imposed by Geant4 interface
    setupPrimary(*nonConstTrack);
  } else {
    setupSecondary(*aTrack);
  }
 
  return;
}
 
void TrackProcessorUserActionBase::setupPrimary(G4Track& aTrack)
{
  //
  // Get PrimaryParticleInformation from G4PrimaryParticle (assigned by TransportTool::addPrimaryVertex)
  //
 
  auto* trackInfo = ::iGeant4::ISFG4Helper::getISFTrackInfo(aTrack);
  if ( trackInfo ) {
    G4ExceptionDescription description;
    description << G4String("PreUserTrackingAction: ")
                << "Started simulation of primary particle which already has a TrackInformation/TrackBarcodeInfo object attached (trackID: "
                << aTrack.GetTrackID() << ", track pos: "<<aTrack.GetPosition() << ", mom: "<<aTrack.GetMomentum()
                << ", parentID " << aTrack.GetParentID() << ")";
    G4Exception("iGeant4::TrackProcessorUserActionBase", "TrackInformationAlreadyExists", FatalException, description);
    return; // The G4Exception call above should abort the job, but Coverity does not seem to pick this up.
  }
 
  auto* primaryPartInfo = dynamic_cast <PrimaryParticleInformation*> (aTrack.GetDynamicParticle()->GetPrimaryParticle()->GetUserInformation());
  if (!primaryPartInfo) {
    G4ExceptionDescription description;
    description << G4String("PreUserTrackingAction: ") + "NULL PrimaryParticleInformation pointer for current G4Step (trackID "
                << aTrack.GetTrackID() << ", track pos: "<<aTrack.GetPosition() << ", mom: "<<aTrack.GetMomentum()
                << ", parentID " << aTrack.GetParentID() << ")";
    G4Exception("iGeant4::TrackProcessorUserActionBase", "NoPPInfo", FatalException, description);
    return; // The G4Exception call above should abort the job, but Coverity does not seem to pick this up.
  }
 
  // get base ISFParticle and link to TrackInformation
  auto* baseISP = primaryPartInfo->GetISFParticle();
  if (!baseISP) {
    G4ExceptionDescription description;
    description << G4String("PreUserTrackingAction: ") + "No ISFParticle associated with primary particle (trackID: "
                << aTrack.GetTrackID() << ", track pos: "<<aTrack.GetPosition() << ", mom: "<<aTrack.GetMomentum()
                << ", parentID " << aTrack.GetParentID() << ")";
    G4Exception("iGeant4::TrackProcessorUserActionBase", "NoISFParticle", FatalException, description);
    return; // The G4Exception call above should abort the job, but Coverity does not seem to pick this up.
  }
 
  ISF::TruthBinding* truthBinding = baseISP->getTruthBinding();
  if (!truthBinding) {
    G4ExceptionDescription description;
    description << G4String("PreUserTrackingAction: ") + "No ISF::TruthBinding associated with primary particle (trackID: "
                << aTrack.GetTrackID() << ", track pos: "<<aTrack.GetPosition() << ", mom: "<<aTrack.GetMomentum()
                << ", parentID " << aTrack.GetParentID() << ")";
    G4Exception("iGeant4::TrackProcessorUserActionBase", "NoISFTruthBinding", FatalException, description);
    return; // The G4Exception call above should abort the job, but Coverity does not seem to pick this up.
  }
 
  int regenerationNr = primaryPartInfo->GetRegenerationNr();
 
  HepMC::GenParticlePtr primaryGenParticle = truthBinding->getGenerationZeroGenParticle();
  HepMC::GenParticlePtr generationZeroGenParticle = truthBinding->getGenerationZeroGenParticle();
  HepMC::GenParticlePtr currentGenParticle = truthBinding->getCurrentGenParticle();
 
  auto classification = classify(primaryGenParticle,
                                 generationZeroGenParticle,
                                 currentGenParticle,
                                 regenerationNr);
 
  auto* newTrackInfo = ::iGeant4::ISFG4Helper::attachTrackInfoToNewG4Track(aTrack,
                                                                 *baseISP,
                                                                 classification,
                                                                 generationZeroGenParticle );
  newTrackInfo->SetRegenerationNr(regenerationNr);
 
  updateCachedParticleInfo(baseISP,
                     primaryGenParticle,
                     currentGenParticle);
 
  return;
}
 
void TrackProcessorUserActionBase::setupSecondary(const G4Track& aTrack)
{
  auto* trackInfo = ::iGeant4::ISFG4Helper::getISFTrackInfo(aTrack);
 
  HepMC::GenParticlePtr currentGenParticle = trackInfo->GetCurrentGenParticle();
  HepMC::GenParticlePtr primaryGenParticle = trackInfo->GetPrimaryGenParticle();
  ISF::ISFParticle* baseISFParticle = trackInfo->GetBaseISFParticle();
 
  updateCachedParticleInfo(baseISFParticle, primaryGenParticle, currentGenParticle);
 
  return;
}
 
void TrackProcessorUserActionBase::updateCachedParticleInfo(ISF::ISFParticle* baseISFParticle,
                                                      HepMC::ConstGenParticlePtr primaryGenParticle,
                                                      HepMC::GenParticlePtr currentGenParticle)
{
  m_curBaseISP = baseISFParticle;
  m_atlasG4EvtUserInfo->SetCurrentPrimaryGenParticle( primaryGenParticle );
  m_atlasG4EvtUserInfo->SetCurrentGenParticle( currentGenParticle );
  return;
}

VTrackInformation::TrackClassification 
TrackProcessorUserActionBase::classify(HepMC::ConstGenParticlePtr primaryGenParticle,
                                                           HepMC::ConstGenParticlePtr generationZeroGenParticle,
                                                           HepMC::ConstGenParticlePtr currentGenParticle,
                                                           int regenerationNumber) const
{
  // if particle points to a non-zero truth particle it can not just be a 'simple' Secondary
 
  if (currentGenParticle) {
    if (currentGenParticle == primaryGenParticle) {
      return VTrackInformation::Primary;
    }
    else if (generationZeroGenParticle == primaryGenParticle && regenerationNumber>0) {
      return VTrackInformation::RegeneratedPrimary;
    }
    else {
      return VTrackInformation::RegisteredSecondary;
    }
  }
 
  return VTrackInformation::Secondary;
}
 
 
 
void TrackProcessorUserActionBase::PostUserTrackingAction(const G4Track*)
{
  m_curBaseISP = nullptr;
  return;
}
 
ISF::ISFParticleContainer TrackProcessorUserActionBase::ReturnSecondaries(ISF::ISFParticle const* /*parent*/)
{
  // For now, just return all particles
  ISF::ISFParticleContainer result;
  std::swap( result, m_storedSecondaries );
  return result;
}
 
} // namespace iGeant4
 
} // namespace G4UA