Geant4Manager.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file Geant4Manager.hpp

// This file is part of the Acts project.
//
// Copyright (C) 2023 CERN for the benefit of the Acts project
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#pragma once

#include <memory>
#include <mutex>
#include <string>
#include <unordered_map>

class G4RunManager;
class G4VUserPhysicsList;

namespace ActsExamples {

class PhysicsListFactory;
class Geant4Manager;

struct Geant4Handle {
  std::mutex mutex;
  int logLevel{};
  std::unique_ptr<G4RunManager> runManager;
  G4VUserPhysicsList *physicsList;
  std::string physicsListName;

  Geant4Handle(int logLevel, std::unique_ptr<G4RunManager> runManager,
               std::unique_ptr<G4VUserPhysicsList> physicsList,
               std::string physicsListName);
  Geant4Handle(const Geant4Handle &) = delete;
  Geant4Handle &operator=(const Geant4Handle &) = delete;
  ~Geant4Handle();

  void tweakLogging(int level) const;
};

class Geant4Manager {
 public:
  static Geant4Manager &instance();

  static void tweakLogging(G4RunManager &runManager, int level);

  std::shared_ptr<Geant4Handle> currentHandle() const;

  std::shared_ptr<Geant4Handle> createHandle(int logLevel,
                                             const std::string &physicsList);

  std::shared_ptr<Geant4Handle> createHandle(
      int logLevel, std::unique_ptr<G4VUserPhysicsList> physicsList,
      std::string physicsListName);

  void registerPhysicsListFactory(
      std::string name, std::shared_ptr<PhysicsListFactory> physicsListFactory);
  std::unique_ptr<G4VUserPhysicsList> createPhysicsList(
      const std::string &name) const;

  const std::unordered_map<std::string, std::shared_ptr<PhysicsListFactory>>
      &getPhysicsListFactories() const;

 private:
  Geant4Manager();
  ~Geant4Manager();

  bool m_created = false;
  std::weak_ptr<Geant4Handle> m_handle;
  std::unordered_map<std::string, std::shared_ptr<PhysicsListFactory>>
      m_physicsListFactories;
};

}  // namespace ActsExamples