Import Geant4 10.7.0.beta source tree
This commit is contained in:
@@ -0,0 +1,97 @@
|
||||
//
|
||||
// MIT License
|
||||
// Copyright (c) 2020 Jonathan R. Madsen
|
||||
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
// of this software and associated documentation files (the "Software"), to deal
|
||||
// in the Software without restriction, including without limitation the rights
|
||||
// to use, copy, modify, merge, publish, distribute, sublicense, and
|
||||
// copies of the Software, and to permit persons to whom the Software is
|
||||
// furnished to do so, subject to the following conditions:
|
||||
// The above copyright notice and this permission notice shall be included in
|
||||
// all copies or substantial portions of the Software. THE SOFTWARE IS PROVIDED
|
||||
// "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT
|
||||
// LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR
|
||||
// PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
|
||||
// HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
|
||||
// ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
||||
// WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
||||
//
|
||||
// class description:
|
||||
// This is a class for run control in Tasking for multi-threaded runs
|
||||
// It extends RunManager re-implementing multi-threaded behavior in
|
||||
// key methods. See documentation for RunManager
|
||||
// Users initializes an instance of this class instead of RunManager
|
||||
// to start a multi-threaded simulation.
|
||||
|
||||
#pragma once
|
||||
|
||||
#include "PTL/TBBTaskGroup.hh"
|
||||
#include "PTL/TaskGroup.hh"
|
||||
#include "PTL/ThreadPool.hh"
|
||||
#include "PTL/Threading.hh"
|
||||
#include "PTL/VUserTaskQueue.hh"
|
||||
|
||||
#include <list>
|
||||
#include <map>
|
||||
|
||||
namespace PTL
|
||||
{
|
||||
class ThreadPool;
|
||||
class TaskManager;
|
||||
|
||||
//======================================================================================//
|
||||
|
||||
class TaskRunManager
|
||||
{
|
||||
public:
|
||||
typedef TaskGroup<void> RunTaskGroup;
|
||||
typedef TBBTaskGroup<void> RunTaskGroupTBB;
|
||||
typedef TaskRunManager* pointer;
|
||||
|
||||
public:
|
||||
// Parameters:
|
||||
// m_task_queue: provide a custom task queue
|
||||
// useTBB: only relevant if PTL_USE_TBB defined
|
||||
// grainsize: 0 = auto
|
||||
explicit TaskRunManager(bool useTBB = false);
|
||||
virtual ~TaskRunManager();
|
||||
|
||||
public:
|
||||
virtual int GetNumberOfThreads() const
|
||||
{
|
||||
return (m_thread_pool) ? m_thread_pool->size() : 0;
|
||||
}
|
||||
virtual size_t GetNumberActiveThreads() const
|
||||
{
|
||||
return (m_thread_pool) ? m_thread_pool->size() : 0;
|
||||
}
|
||||
|
||||
public:
|
||||
// Inherited methods to re-implement for MT case
|
||||
virtual void Initialize(uint64_t n = std::thread::hardware_concurrency());
|
||||
virtual void Terminate();
|
||||
ThreadPool* GetThreadPool() const { return m_thread_pool; }
|
||||
TaskManager* GetTaskManager() const { return m_task_manager; }
|
||||
bool IsInitialized() const { return m_is_initialized; }
|
||||
int GetVerbose() const { return m_verbose; }
|
||||
void SetVerbose(int val) { m_verbose = val; }
|
||||
|
||||
public: // with description
|
||||
// Singleton implementing master thread behavior
|
||||
static TaskRunManager* GetInstance(bool useTBB = false);
|
||||
static TaskRunManager* GetMasterRunManager(bool useTBB = false);
|
||||
|
||||
private:
|
||||
static pointer& GetPrivateMasterRunManager(bool init, bool useTBB = false);
|
||||
|
||||
protected:
|
||||
// Barriers: synch points between master and workers
|
||||
bool m_is_initialized = false;
|
||||
int m_verbose = 0;
|
||||
uint64_t m_workers = 0;
|
||||
VUserTaskQueue* m_task_queue = nullptr;
|
||||
ThreadPool* m_thread_pool = nullptr;
|
||||
TaskManager* m_task_manager = nullptr;
|
||||
};
|
||||
|
||||
} // namespace PTL
|
||||
Reference in New Issue
Block a user