123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567 |
- #ifndef BOOST_THREAD_CONDITION_VARIABLE_WIN32_HPP
- #define BOOST_THREAD_CONDITION_VARIABLE_WIN32_HPP
- // Distributed under the Boost Software License, Version 1.0. (See
- // accompanying file LICENSE_1_0.txt or copy at
- // http://www.boost.org/LICENSE_1_0.txt)
- // (C) Copyright 2007-8 Anthony Williams
- // (C) Copyright 2011-2012 Vicente J. Botet Escriba
- #include <boost/thread/win32/thread_primitives.hpp>
- #include <boost/thread/win32/thread_data.hpp>
- #include <boost/thread/win32/thread_data.hpp>
- #include <boost/thread/win32/interlocked_read.hpp>
- #include <boost/thread/cv_status.hpp>
- #if defined BOOST_THREAD_USES_DATETIME
- #include <boost/thread/xtime.hpp>
- #endif
- #include <boost/thread/mutex.hpp>
- #include <boost/thread/thread_time.hpp>
- #include <boost/thread/lock_guard.hpp>
- #include <boost/thread/lock_types.hpp>
- #include <boost/assert.hpp>
- #include <boost/intrusive_ptr.hpp>
- #ifdef BOOST_THREAD_USES_CHRONO
- #include <boost/chrono/system_clocks.hpp>
- #include <boost/chrono/ceil.hpp>
- #endif
- #include <limits.h>
- #include <algorithm>
- #include <vector>
- #include <boost/config/abi_prefix.hpp>
- namespace boost
- {
- namespace detail
- {
- class basic_cv_list_entry;
- void intrusive_ptr_add_ref(basic_cv_list_entry * p);
- void intrusive_ptr_release(basic_cv_list_entry * p);
- class basic_cv_list_entry
- {
- private:
- detail::win32::handle_manager semaphore;
- detail::win32::handle_manager wake_sem;
- long waiters;
- bool notified;
- long references;
- public:
- BOOST_THREAD_NO_COPYABLE(basic_cv_list_entry)
- explicit basic_cv_list_entry(detail::win32::handle_manager const& wake_sem_):
- semaphore(detail::win32::create_anonymous_semaphore(0,LONG_MAX)),
- wake_sem(wake_sem_.duplicate()),
- waiters(1),notified(false),references(0)
- {}
- static bool no_waiters(boost::intrusive_ptr<basic_cv_list_entry> const& entry)
- {
- return !detail::interlocked_read_acquire(&entry->waiters);
- }
- void add_waiter()
- {
- BOOST_INTERLOCKED_INCREMENT(&waiters);
- }
- void remove_waiter()
- {
- BOOST_INTERLOCKED_DECREMENT(&waiters);
- }
- void release(unsigned count_to_release)
- {
- notified=true;
- detail::win32::ReleaseSemaphore(semaphore,count_to_release,0);
- }
- void release_waiters()
- {
- release(detail::interlocked_read_acquire(&waiters));
- }
- bool is_notified() const
- {
- return notified;
- }
- bool wait(timeout abs_time)
- {
- return this_thread::interruptible_wait(semaphore,abs_time);
- }
- bool woken()
- {
- unsigned long const woken_result=detail::win32::WaitForSingleObjectEx(wake_sem,0,0);
- BOOST_ASSERT((woken_result==detail::win32::timeout) || (woken_result==0));
- return woken_result==0;
- }
- friend void intrusive_ptr_add_ref(basic_cv_list_entry * p);
- friend void intrusive_ptr_release(basic_cv_list_entry * p);
- };
- inline void intrusive_ptr_add_ref(basic_cv_list_entry * p)
- {
- BOOST_INTERLOCKED_INCREMENT(&p->references);
- }
- inline void intrusive_ptr_release(basic_cv_list_entry * p)
- {
- if(!BOOST_INTERLOCKED_DECREMENT(&p->references))
- {
- delete p;
- }
- }
- class basic_condition_variable
- {
- boost::mutex internal_mutex;
- long total_count;
- unsigned active_generation_count;
- typedef basic_cv_list_entry list_entry;
- typedef boost::intrusive_ptr<list_entry> entry_ptr;
- typedef std::vector<entry_ptr> generation_list;
- generation_list generations;
- detail::win32::handle_manager wake_sem;
- void wake_waiters(long count_to_wake)
- {
- detail::interlocked_write_release(&total_count,total_count-count_to_wake);
- detail::win32::ReleaseSemaphore(wake_sem,count_to_wake,0);
- }
- template<typename lock_type>
- struct relocker
- {
- BOOST_THREAD_NO_COPYABLE(relocker)
- lock_type& lock;
- bool unlocked;
- relocker(lock_type& lock_):
- lock(lock_),unlocked(false)
- {}
- void unlock()
- {
- lock.unlock();
- unlocked=true;
- }
- ~relocker()
- {
- if(unlocked)
- {
- lock.lock();
- }
- }
- };
- entry_ptr get_wait_entry()
- {
- boost::lock_guard<boost::mutex> internal_lock(internal_mutex);
- if(!wake_sem)
- {
- wake_sem=detail::win32::create_anonymous_semaphore(0,LONG_MAX);
- BOOST_ASSERT(wake_sem);
- }
- detail::interlocked_write_release(&total_count,total_count+1);
- if(generations.empty() || generations.back()->is_notified())
- {
- entry_ptr new_entry(new list_entry(wake_sem));
- generations.push_back(new_entry);
- return new_entry;
- }
- else
- {
- generations.back()->add_waiter();
- return generations.back();
- }
- }
- struct entry_manager
- {
- entry_ptr const entry;
- boost::mutex& internal_mutex;
- BOOST_THREAD_NO_COPYABLE(entry_manager)
- entry_manager(entry_ptr const& entry_, boost::mutex& mutex_):
- entry(entry_), internal_mutex(mutex_)
- {}
- ~entry_manager()
- {
- boost::lock_guard<boost::mutex> internal_lock(internal_mutex);
- entry->remove_waiter();
- }
- list_entry* operator->()
- {
- return entry.get();
- }
- };
- protected:
- template<typename lock_type>
- bool do_wait(lock_type& lock,timeout abs_time)
- {
- relocker<lock_type> locker(lock);
- entry_manager entry(get_wait_entry(), internal_mutex);
- locker.unlock();
- bool woken=false;
- while(!woken)
- {
- if(!entry->wait(abs_time))
- {
- return false;
- }
- woken=entry->woken();
- }
- return woken;
- }
- template<typename lock_type,typename predicate_type>
- bool do_wait(lock_type& m,timeout const& abs_time,predicate_type pred)
- {
- while (!pred())
- {
- if(!do_wait(m, abs_time))
- return pred();
- }
- return true;
- }
- basic_condition_variable(const basic_condition_variable& other);
- basic_condition_variable& operator=(const basic_condition_variable& other);
- public:
- basic_condition_variable():
- total_count(0),active_generation_count(0),wake_sem(0)
- {}
- ~basic_condition_variable()
- {}
- void notify_one() BOOST_NOEXCEPT
- {
- if(detail::interlocked_read_acquire(&total_count))
- {
- boost::lock_guard<boost::mutex> internal_lock(internal_mutex);
- if(!total_count)
- {
- return;
- }
- wake_waiters(1);
- for(generation_list::iterator it=generations.begin(),
- end=generations.end();
- it!=end;++it)
- {
- (*it)->release(1);
- }
- generations.erase(std::remove_if(generations.begin(),generations.end(),&basic_cv_list_entry::no_waiters),generations.end());
- }
- }
- void notify_all() BOOST_NOEXCEPT
- {
- if(detail::interlocked_read_acquire(&total_count))
- {
- boost::lock_guard<boost::mutex> internal_lock(internal_mutex);
- if(!total_count)
- {
- return;
- }
- wake_waiters(total_count);
- for(generation_list::iterator it=generations.begin(),
- end=generations.end();
- it!=end;++it)
- {
- (*it)->release_waiters();
- }
- generations.clear();
- wake_sem=detail::win32::handle(0);
- }
- }
- };
- }
- class condition_variable:
- private detail::basic_condition_variable
- {
- public:
- BOOST_THREAD_NO_COPYABLE(condition_variable)
- condition_variable()
- {}
- using detail::basic_condition_variable::notify_one;
- using detail::basic_condition_variable::notify_all;
- void wait(unique_lock<mutex>& m)
- {
- do_wait(m,detail::timeout::sentinel());
- }
- template<typename predicate_type>
- void wait(unique_lock<mutex>& m,predicate_type pred)
- {
- while(!pred()) wait(m);
- }
- #if defined BOOST_THREAD_USES_DATETIME
- bool timed_wait(unique_lock<mutex>& m,boost::system_time const& abs_time)
- {
- return do_wait(m,abs_time);
- }
- bool timed_wait(unique_lock<mutex>& m,boost::xtime const& abs_time)
- {
- return do_wait(m,system_time(abs_time));
- }
- template<typename duration_type>
- bool timed_wait(unique_lock<mutex>& m,duration_type const& wait_duration)
- {
- if (wait_duration.is_pos_infinity())
- {
- wait(m); // or do_wait(m,detail::timeout::sentinel());
- return true;
- }
- if (wait_duration.is_special())
- {
- return true;
- }
- return do_wait(m,wait_duration.total_milliseconds());
- }
- template<typename predicate_type>
- bool timed_wait(unique_lock<mutex>& m,boost::system_time const& abs_time,predicate_type pred)
- {
- return do_wait(m,abs_time,pred);
- }
- template<typename predicate_type>
- bool timed_wait(unique_lock<mutex>& m,boost::xtime const& abs_time,predicate_type pred)
- {
- return do_wait(m,system_time(abs_time),pred);
- }
- template<typename duration_type,typename predicate_type>
- bool timed_wait(unique_lock<mutex>& m,duration_type const& wait_duration,predicate_type pred)
- {
- if (wait_duration.is_pos_infinity())
- {
- while (!pred())
- {
- wait(m); // or do_wait(m,detail::timeout::sentinel());
- }
- return true;
- }
- if (wait_duration.is_special())
- {
- return pred();
- }
- return do_wait(m,wait_duration.total_milliseconds(),pred);
- }
- #endif
- #ifdef BOOST_THREAD_USES_CHRONO
- template <class Clock, class Duration>
- cv_status
- wait_until(
- unique_lock<mutex>& lock,
- const chrono::time_point<Clock, Duration>& t)
- {
- using namespace chrono;
- chrono::time_point<Clock, Duration> now = Clock::now();
- if (t<=now) {
- return cv_status::timeout;
- }
- do_wait(lock, ceil<milliseconds>(t-now).count());
- return Clock::now() < t ? cv_status::no_timeout :
- cv_status::timeout;
- }
- template <class Rep, class Period>
- cv_status
- wait_for(
- unique_lock<mutex>& lock,
- const chrono::duration<Rep, Period>& d)
- {
- using namespace chrono;
- if (d<=chrono::duration<Rep, Period>::zero()) {
- return cv_status::timeout;
- }
- steady_clock::time_point c_now = steady_clock::now();
- do_wait(lock, ceil<milliseconds>(d).count());
- return steady_clock::now() - c_now < d ? cv_status::no_timeout :
- cv_status::timeout;
- }
- template <class Clock, class Duration, class Predicate>
- bool
- wait_until(
- unique_lock<mutex>& lock,
- const chrono::time_point<Clock, Duration>& t,
- Predicate pred)
- {
- while (!pred())
- {
- if (wait_until(lock, t) == cv_status::timeout)
- return pred();
- }
- return true;
- }
- template <class Rep, class Period, class Predicate>
- bool
- wait_for(
- unique_lock<mutex>& lock,
- const chrono::duration<Rep, Period>& d,
- Predicate pred)
- {
- return wait_until(lock, chrono::steady_clock::now() + d, boost::move(pred));
- }
- #endif
- };
- class condition_variable_any:
- private detail::basic_condition_variable
- {
- public:
- BOOST_THREAD_NO_COPYABLE(condition_variable_any)
- condition_variable_any()
- {}
- using detail::basic_condition_variable::notify_one;
- using detail::basic_condition_variable::notify_all;
- template<typename lock_type>
- void wait(lock_type& m)
- {
- do_wait(m,detail::timeout::sentinel());
- }
- template<typename lock_type,typename predicate_type>
- void wait(lock_type& m,predicate_type pred)
- {
- while(!pred()) wait(m);
- }
- #if defined BOOST_THREAD_USES_DATETIME
- template<typename lock_type>
- bool timed_wait(lock_type& m,boost::system_time const& abs_time)
- {
- return do_wait(m,abs_time);
- }
- template<typename lock_type>
- bool timed_wait(lock_type& m,boost::xtime const& abs_time)
- {
- return do_wait(m,system_time(abs_time));
- }
- template<typename lock_type,typename duration_type>
- bool timed_wait(lock_type& m,duration_type const& wait_duration)
- {
- return do_wait(m,wait_duration.total_milliseconds());
- }
- template<typename lock_type,typename predicate_type>
- bool timed_wait(lock_type& m,boost::system_time const& abs_time,predicate_type pred)
- {
- return do_wait(m,abs_time,pred);
- }
- template<typename lock_type,typename predicate_type>
- bool timed_wait(lock_type& m,boost::xtime const& abs_time,predicate_type pred)
- {
- return do_wait(m,system_time(abs_time),pred);
- }
- template<typename lock_type,typename duration_type,typename predicate_type>
- bool timed_wait(lock_type& m,duration_type const& wait_duration,predicate_type pred)
- {
- return do_wait(m,wait_duration.total_milliseconds(),pred);
- }
- #endif
- #ifdef BOOST_THREAD_USES_CHRONO
- template <class lock_type, class Clock, class Duration>
- cv_status
- wait_until(
- lock_type& lock,
- const chrono::time_point<Clock, Duration>& t)
- {
- using namespace chrono;
- chrono::time_point<Clock, Duration> now = Clock::now();
- if (t<=now) {
- return cv_status::timeout;
- }
- do_wait(lock, ceil<milliseconds>(t-now).count());
- return Clock::now() < t ? cv_status::no_timeout :
- cv_status::timeout;
- }
- template <class lock_type, class Rep, class Period>
- cv_status
- wait_for(
- lock_type& lock,
- const chrono::duration<Rep, Period>& d)
- {
- using namespace chrono;
- if (d<=chrono::duration<Rep, Period>::zero()) {
- return cv_status::timeout;
- }
- steady_clock::time_point c_now = steady_clock::now();
- do_wait(lock, ceil<milliseconds>(d).count());
- return steady_clock::now() - c_now < d ? cv_status::no_timeout :
- cv_status::timeout;
- }
- template <class lock_type, class Clock, class Duration, class Predicate>
- bool
- wait_until(
- lock_type& lock,
- const chrono::time_point<Clock, Duration>& t,
- Predicate pred)
- {
- while (!pred())
- {
- if (wait_until(lock, t) == cv_status::timeout)
- return pred();
- }
- return true;
- }
- template <class lock_type, class Rep, class Period, class Predicate>
- bool
- wait_for(
- lock_type& lock,
- const chrono::duration<Rep, Period>& d,
- Predicate pred)
- {
- return wait_until(lock, chrono::steady_clock::now() + d, boost::move(pred));
- }
- #endif
- };
- BOOST_THREAD_DECL void notify_all_at_thread_exit(condition_variable& cond, unique_lock<mutex> lk);
- }
- #include <boost/config/abi_suffix.hpp>
- #endif
|