Loading...
Searching...
No Matches
Invoke.h
Go to the documentation of this file.
// Copyright Contributors to the OpenVDB Project
// SPDX-License-Identifier: MPL-2.0
/*!
\file Invoke.h
\author Ken Museth
\date August 24, 2020
\brief A unified wrapper for tbb::parallel_invoke and a naive std::thread analog
@code
template<typename Func0, typename Func1, ..., typename FuncN>
int invoke(const Func0& f0, const Func1& f1, ..., const FuncN& fN);
@endcode
*/
#ifndef NANOVDB_INVOKE_H_HAS_BEEN_INCLUDED
#define NANOVDB_INVOKE_H_HAS_BEEN_INCLUDED
#include "../NanoVDB.h"// for nanovdb::CoordBBox
#ifdef NANOVDB_USE_TBB
#include <tbb/parallel_invoke.h>
#endif
#include <thread>
#include <mutex>
#include <vector>
namespace nanovdb {
namespace {
#ifndef NANOVDB_USE_TBB
// Base case
template<typename Func>
void parallel_invoke(std::vector<std::thread> &threadPool, const Func &taskFunc) {
threadPool.emplace_back(taskFunc);
}
// Iterative call
template<typename Func, typename... Rest>
void parallel_invoke(std::vector<std::thread> &threadPool, const Func &taskFunc1, Rest... taskFuncN) {
threadPool.emplace_back(taskFunc1);
parallel_invoke(threadPool, taskFuncN...);
}
// Base case
template<typename Func>
void serial_invoke(const Func &taskFunc) {taskFunc();}
// Iterative call
template<typename Func, typename... Rest>
void serial_invoke(const Func &taskFunc1, Rest... taskFuncN) {
taskFunc1();
serial_invoke(taskFuncN...);
}
#endif
}// unnamed namespace
/// @return 1 for serial, 2 for tbb multi-threading, and 3 for std multi-threading
template<typename Func, typename... Rest>
int invoke(const Func &taskFunc1, Rest... taskFuncN) {
#ifdef NANOVDB_USE_TBB
tbb::parallel_invoke(taskFunc1, taskFuncN...);
return 2;
#else
const auto threadCount = std::thread::hardware_concurrency()>>1;
if (1 + sizeof...(Rest) <= threadCount) {
std::vector<std::thread> threadPool;
threadPool.emplace_back(taskFunc1);
parallel_invoke(threadPool, taskFuncN...);
for (auto &t : threadPool) t.join();
return 3;// std multi-threading
} else {
taskFunc1();
serial_invoke(taskFuncN...);
return 1;// serial
}
#endif
return -1;// should never happen
}
}// namespace nanovdb
#endif // NANOVDB_INVOKE_H_HAS_BEEN_INCLUDED
NanoVDB.h
Implements a light-weight self-contained VDB data-structure in a single file! In other words,...
nanovdb
Definition NanoVDB.h:208
nanovdb::invoke
int invoke(const Func &taskFunc1, Rest... taskFuncN)
Definition Invoke.h:64