G4InitTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#include "G4InitTool.h"
 
#include "G4WorkerRunManager.hh"
 
#include "G4UImanager.hh"
 
#include "G4MTRunManager.hh"
#include "G4WorkerThread.hh"
#include "G4UserWorkerThreadInitialization.hh"
#include "G4RunManager.hh"
#include "G4VUserActionInitialization.hh"
#include "G4UserWorkerInitialization.hh"
#include "G4AutoDelete.hh"
 
#include "CxxUtils/checker_macros.h"
 
#include <unistd.h>
#include <sys/syscall.h>
 
 
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
 
G4InitTool::G4InitTool( const std::string& type, const std::string& name,
                        const IInterface* parent )
  : base_class(type, name, parent),
    m_nInitThreads(0)
 
{
}
 
 
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
 
void
G4InitTool::initThread() {
 
  ATH_MSG_INFO ("==> tbb thread started with id: 0x" << std::hex
                << pthread_self() << std::dec);
 
  // The following code is modeled after G4MTRunManagerKernel::StartThread
 
  // Define the G4 worker thread context
  auto wThreadContext = new G4WorkerThread;
  G4AutoDelete::Register(wThreadContext);
 
  // Assign the thread id
  // TODO: implement a better solution.
  // WARNING: not portable!
#ifdef __USE_GNU
  pid_t tid = gettid();
  wThreadContext->SetThreadId( tid );
#else
  #error "gettid() unavailable on this system"
#endif
  //wThreadContext->SetThreadId( pthread_self() );
  G4int thisID = wThreadContext->GetThreadId();
  G4Threading::G4SetThreadId(thisID);
 
  // Retrieve the G4AtlasMTRunManager
  G4MTRunManager* masterRM = G4MTRunManager::GetMasterRunManager();
  // Worker thread initialization
  const G4UserWorkerThreadInitialization* workerInitializer = masterRM->GetUserWorkerThreadInitialization();
 
  const CLHEP::HepRandomEngine* masterEngine = masterRM->getMasterRandomEngine();
  workerInitializer->SetupRNGEngine(masterEngine);
 
  // Setup up thread-local workspace
  wThreadContext->BuildGeometryAndPhysicsVector();
 
  // Create the thread-local worker run manager (G4AtlasWorkerRunManager)
  ATH_MSG_INFO("Creating worker RM");
  G4WorkerRunManager* wrm = workerInitializer->CreateWorkerRunManager();
  wrm->SetWorkerThread(wThreadContext);
 
  // Share detector from master with worker.
  ATH_MSG_INFO("Assigning detector construction");
  auto detector ATLAS_THREAD_SAFE = const_cast<G4VUserDetectorConstruction*>
    (masterRM->GetUserDetectorConstruction());
  wrm->G4RunManager::SetUserInitialization(detector);
 
  // Share physics list from master with worker.
  auto physicslist ATLAS_THREAD_SAFE = const_cast<G4VUserPhysicsList*>(masterRM->GetUserPhysicsList());
  wrm->SetUserInitialization(physicslist);
 
  // Build thread-local user actions - NOT CURRENTLY USED.
  if(masterRM->GetUserActionInitialization()) {
    masterRM->GetNonConstUserActionInitialization()->Build();
  }
 
  // Start user worker initialization
  if(masterRM->GetUserWorkerInitialization()) {
    masterRM->GetUserWorkerInitialization()->WorkerStart();
  }
 
  // Initialize the worker run manager
  ATH_MSG_INFO("Initializing worker RM");
  wrm->Initialize();
 
  // Atomic increment number of initialized threads
  m_nInitThreads++;
 
  ATH_MSG_INFO ("==> tbb thread end of initThread with id: 0x" << std::hex <<
                pthread_self() << std::dec );
 
}
 
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
 
void
G4InitTool::terminateThread() {
 
  ATH_MSG_INFO ("terminateThread ==> tbb thread 0x"
                << std::hex << pthread_self() << std::dec );
 
   // call the run termination method on the G4 side
  G4RunManager::GetRunManager()->RunTermination();
 
  m_nInitThreads--;
 
}