mirror of https://github.com/CGAL/cgal
Move atomic variables into Parallel_callback
This commit is contained in:
parent
7fba36b8dd
commit
45bad1258e
|
|
@ -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)
|
||||||
|
|
|
||||||
|
|
@ -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.);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue