G4AtlasRunManager.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#include "G4AtlasAlg/G4AtlasRunManager.h"
 
 
#include "G4GeometryManager.hh"
#include "G4LogicalVolumeStore.hh"
#include "G4ParallelWorldScoringProcess.hh"
#include "G4RegionStore.hh"
#include "G4Run.hh"
#include "G4ScoringManager.hh"
#include "G4StateManager.hh"
#include "G4TransportationManager.hh"
#include "G4UImanager.hh"
#include "G4UserRunAction.hh"
#include "G4RunManager.hh"
#include "G4UserWorkerInitialization.hh"
#include "G4Version.hh"
 
#include <string>
 
G4AtlasRunManager::G4AtlasRunManager()
  : G4RunManager()
  , AthMessaging("G4AtlasRunManager")
  , m_recordFlux(false)
  , m_physListSvc("PhysicsListSvc", "G4AtlasRunManager")
  , m_volumeSmartlessLevel({})
{  }
 
 
G4AtlasRunManager* G4AtlasRunManager::GetG4AtlasRunManager ATLAS_NOT_THREAD_SAFE ()
{
  static G4AtlasRunManager *thisManager = nullptr;
  if (!thisManager) { thisManager = new G4AtlasRunManager; } // Leaked
  return thisManager;
}
 
 
void G4AtlasRunManager::Initialize()
{
  // ADA 11/28.2018: switch initialization order to meet ISF requirements
  // Call the base class Initialize method. This will call
  // InitializeGeometry and InitializePhysics.
  if (m_quietMode) { SetVerboseLevel(0); } // HACK
  G4RunManager::Initialize();
}
 
void G4AtlasRunManager::RunInitialization()
{
  G4RunManager::RunInitialization();
  if(auto* uwi = GetUserWorkerInitialization()) {
    uwi->WorkerRunStart();
  }
}
 
void G4AtlasRunManager::InitializeGeometry()
{
  ATH_MSG_DEBUG( "InitializeGeometry()" );
 
  // Create/assign detector construction
  G4RunManager::SetUserInitialization(
      m_detConstruction->GetDetectorConstruction().release());
  if (userDetector) {
    G4RunManager::InitializeGeometry();
  }
  else {
    ATH_MSG_WARNING( " User Detector not set!!! Geometry NOT initialized!!!" );
  }
  return;
}
 
 
void G4AtlasRunManager::EndEvent()
{
  ATH_MSG_DEBUG( "G4AtlasRunManager::EndEvent" );
}
 
 
void G4AtlasRunManager::InitializePhysics()
{
  ATH_MSG_INFO( "InitializePhysics()" );
  kernel->InitializePhysics();
  physicsInitialized = true;
 
  // Grab the physics list tool and set the extra options
  if (m_physListSvc.retrieve().isFailure()) {
    ATH_MSG_ERROR ( "Could not retrieve the physics list tool" );
    G4ExceptionDescription description;
    description << "InitializePhysics: Failed to retrieve IPhysicsListSvc.";
    G4Exception("G4AtlasRunManager", "CouldNotRetrievePLTool", FatalException, description);
    abort(); // to keep Coverity happy
  }
  m_physListSvc->SetPhysicsOptions();
 
  if (m_recordFlux) {
    m_fluxRecorder->InitializeFluxRecording();
  }
 
  return;
}
 
bool G4AtlasRunManager::ProcessEvent(G4Event* event)
{
 
  G4StateManager* stateManager = G4StateManager::GetStateManager();
  stateManager->SetNewState(G4State_GeomClosed);
 
  currentEvent = event;
 
  eventManager->ProcessOneEvent(currentEvent);
  if (currentEvent->IsAborted()) {
    ATH_MSG_WARNING( "G4AtlasRunManager::ProcessEvent: Event Aborted at Detector Simulation level" );
    currentEvent = nullptr;
    return true;
  }
 
  if (m_recordFlux) { m_fluxRecorder->RecordFlux(currentEvent); }
 
  this->StackPreviousEvent(currentEvent);
  bool abort = currentEvent->IsAborted();
  currentEvent = nullptr;
 
  return abort;
}
 
void G4AtlasRunManager::RunTermination()
{
  ATH_MSG_DEBUG( " G4AtlasRunManager::RunTermination() " );
  if (m_recordFlux) {
    m_fluxRecorder->WriteFluxInformation();
  }
  
  this->CleanUpPreviousEvents();
  previousEvents->clear();
 
  if (userRunAction) {
    userRunAction->EndOfRunAction(currentRun);
  }
 
  delete currentRun;
  currentRun = nullptr;
  runIDCounter++;
 
  ATH_MSG_VERBOSE( "Changing the state..." );
  G4StateManager* stateManager = G4StateManager::GetStateManager();
  stateManager->SetNewState(G4State_Idle);
 
  ATH_MSG_VERBOSE( "Opening the geometry back up" );
  G4GeometryManager::GetInstance()->OpenGeometry();
 
  ATH_MSG_VERBOSE( "Terminating the run...  State is " << stateManager->GetStateString( stateManager->GetCurrentState() ) );
  kernel->RunTermination();
  ATH_MSG_VERBOSE( "All done..." );
 
  userRunAction = nullptr;
  userEventAction = nullptr;
  userSteppingAction = nullptr;
  userStackingAction = nullptr;
  userTrackingAction = nullptr;
  userDetector = nullptr;
  userPrimaryGeneratorAction = nullptr;
 
  return;
}