PtrRetrieval.h

#ifndef SIMCORE_PTRRETRIEVAL_H
#define SIMCORE_PTRRETRIEVAL_H
 
//----------------//
//   C++ StdLib   //
//----------------//
#include <iostream>
#include <string>
 
//------------//
//   Geant4   //
//------------//
#include "G4Gamma.hh"
#include "G4LogicalVolume.hh"
#include "G4LogicalVolumeStore.hh"
#include "G4ParticleDefinition.hh"
#include "G4PhysicalVolumeStore.hh"
#include "G4ProcessManager.hh"
#include "G4Region.hh"
#include "G4RegionStore.hh"
#include "G4VPhysicalVolume.hh"
 
namespace simcore {
namespace g4user {
namespace ptrretrieval {
 
inline const G4VProcess* getProcess(const G4ParticleDefinition* particle,
                                    const std::string& processName) {
  if (!particle) return nullptr;
 
  const auto manager = particle->GetProcessManager();
  if (!manager) return nullptr;
 
  const auto processes = manager->GetProcessList();
  for (int i = 0; i < processes->size(); ++i) {
    const auto process = (*processes)[i];
    if (process->GetProcessName().find(processName) != std::string::npos) {
      return process;
    }
  }
  return nullptr;
}
 
inline const G4VProcess* getPhotonuclearProcess() {
  return getProcess(G4Gamma::Definition(), "photonNuclear");
}
 
inline G4Region* getRegion(const std::string& name) {
  G4Region* region = G4RegionStore::GetInstance()->GetRegion(name);
  if (!region) {
    return nullptr;
  }
  return region;
}
 
inline G4VPhysicalVolume* getPhysicalVolume(const std::string& name) {
  auto* volume = G4PhysicalVolumeStore::GetInstance()->GetVolume(name);
  if (!volume) {
    return nullptr;
  }
  return volume;
}
 
inline G4LogicalVolume* getLogicalVolume(const std::string& name) {
  auto* volume = G4LogicalVolumeStore::GetInstance()->GetVolume(name);
  if (!volume) {
    return nullptr;
  }
  return volume;
}
 
}  // namespace ptrretrieval
}  // namespace g4user
}  // namespace simcore
#endif  // SIMCORE_PTRRETRIEVAL_H