G4AtlasMTRunManager.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef G4ATLASALG_G4ATLASMTRUNMANAGER_H
#define G4ATLASALG_G4ATLASMTRUNMANAGER_H
 
// Hide multi-threading classes from builds without G4MT
#include "G4Types.hh"
#ifdef G4MULTITHREADED
 
// Geant4 includes
#include "G4MTRunManager.hh"
 
// Framework includes
#include "GaudiKernel/ServiceHandle.h"
#include <GaudiKernel/ToolHandle.h>
#include "AthenaBaseComps/AthMessaging.h"
#include "CxxUtils/checker_macros.h"
 
// G4Atlas includes
#include "G4AtlasInterfaces/IDetectorConstructionTool.h"
#include "G4AtlasInterfaces/IPhysicsListSvc.h"
 

class G4AtlasMTRunManager: public G4MTRunManager, public AthMessaging {
 
public:

  static G4AtlasMTRunManager* GetG4AtlasMTRunManager ATLAS_NOT_THREAD_SAFE ();

  void RunTermination() override final;

  void Initialize() override final;

  virtual void ThisWorkerReady() override final {};
  virtual void ThisWorkerEndEventLoop() override final {};

  void SetDetConstructionTool(IDetectorConstructionTool* detConstruction) {
    m_detConstruction = detConstruction;
  }

  void SetPhysListSvc(const std::string& typeAndName) {
    m_physListSvc.setTypeAndName(typeAndName);
  }

  void SetQuietMode(bool quietMode) {
    m_quietMode = quietMode;
  }
 
 protected:

  void InitializeGeometry() override final;
 
  // Initialize the physics list on the master
  void InitializePhysics() override final;

  virtual void WaitForReadyWorkers() override final {};
  virtual void WaitForEndEventLoopWorkers() override final {};
 
private:

  G4AtlasMTRunManager();
 
private:
  IDetectorConstructionTool* m_detConstruction{nullptr};

  ServiceHandle<IPhysicsListSvc> m_physListSvc;

  bool m_quietMode{true};
 
}; // class G4AtlasMTRunManager
 
#endif // G4MULTITHREADED
 
#endif