Move atomic variables into Parallel_callback

This commit is contained in:
Simon Giraudot 2018-03-09 14:13:17 +01:00
parent 7fba36b8dd
commit 45bad1258e
2 changed files with 55 additions and 25 deletions

View File

@ -243,19 +243,15 @@ compute_average_spacing(
#else #else
if (boost::is_convertible<ConcurrencyTag,Parallel_tag>::value) if (boost::is_convertible<ConcurrencyTag,Parallel_tag>::value)
{ {
// tbb::atomic only has default constructor, initialization done in two steps
tbb::atomic<std::size_t> advancement;
advancement = 0;
tbb::atomic<bool> interrupted;
interrupted = false;
internal::Point_set_processing_3::Parallel_callback internal::Point_set_processing_3::Parallel_callback
parallel_callback (callback, interrupted, advancement, kd_tree_points.size()); parallel_callback (callback, kd_tree_points.size());
std::thread* callback_thread = (callback ? new std::thread (parallel_callback) : NULL); std::thread* callback_thread = (callback ? new std::thread (parallel_callback) : NULL);
std::vector<FT> spacings (kd_tree_points.size (), -1); std::vector<FT> spacings (kd_tree_points.size (), -1);
CGAL::internal::Compute_average_spacings<Kernel, Tree> CGAL::internal::Compute_average_spacings<Kernel, Tree>
f (tree, k, kd_tree_points, spacings, advancement, interrupted); f (tree, k, kd_tree_points, spacings,
parallel_callback.advancement(),
parallel_callback.interrupted());
tbb::parallel_for(tbb::blocked_range<size_t>(0, kd_tree_points.size ()), f); tbb::parallel_for(tbb::blocked_range<size_t>(0, kd_tree_points.size ()), f);
for (unsigned int i = 0; i < spacings.size (); ++ i) for (unsigned int i = 0; i < spacings.size (); ++ i)

View File

@ -33,36 +33,70 @@ namespace Point_set_processing_3 {
class Parallel_callback class Parallel_callback
{ {
const cpp11::function<bool(double)>& callback; const cpp11::function<bool(double)>& m_callback;
tbb::atomic<std::size_t>& advancement; tbb::atomic<std::size_t>* m_advancement;
tbb::atomic<bool>& interrupted; tbb::atomic<bool>* m_interrupted;
std::size_t size; std::size_t m_size;
bool m_creator;
public: public:
Parallel_callback (const cpp11::function<bool(double)>& callback, Parallel_callback (const cpp11::function<bool(double)>& callback,
tbb::atomic<bool>& interrupted, std::size_t size,
tbb::atomic<std::size_t>& advancement, std::size_t advancement = 0,
std::size_t size) bool interrupted = false)
: callback (callback) : m_callback (callback)
, advancement (advancement) , m_advancement (new tbb::atomic<std::size_t>())
, interrupted (interrupted) , m_interrupted (new tbb::atomic<bool>())
, size (size) , m_size (size)
{ } , m_creator (true)
{
// tbb::atomic only has default constructor, initialization done in two steps
*m_advancement = advancement;
*m_interrupted = interrupted;
}
Parallel_callback (const Parallel_callback& other)
: m_callback (other.m_callback)
, m_advancement (other.m_advancement)
, m_interrupted (other.m_interrupted)
, m_size (other.m_size)
, m_creator (false)
{
}
Parallel_callback& operator= (const Parallel_callback& other)
{
Parallel_callback out (other);
return out;
}
~Parallel_callback ()
{
if (m_creator)
{
delete m_advancement;
delete m_interrupted;
}
}
tbb::atomic<std::size_t>& advancement() { return *m_advancement; }
tbb::atomic<bool>& interrupted() { return *m_interrupted; }
void operator()() void operator()()
{ {
tbb::tick_count::interval_t sleeping_time(0.00001); tbb::tick_count::interval_t sleeping_time(0.00001);
while (advancement != size) while (*m_advancement != m_size)
{ {
if (!callback (advancement / double(size))) if (!m_callback (*m_advancement / double(m_size)))
{ {
interrupted = true; *m_interrupted = true;
return; return;
} }
std::this_thread::sleep_for(sleeping_time); std::this_thread::sleep_for(sleeping_time);
} }
callback (1.); m_callback (1.);
} }
}; };