| //===----------------------------------------------------------------------===// |
| // |
| // The LLVM Compiler Infrastructure |
| // |
| // This file is dual licensed under the MIT and the University of Illinois Open |
| // Source Licenses. See LICENSE.TXT for details. |
| // |
| //===----------------------------------------------------------------------===// |
| |
| // <condition_variable> |
| |
| // class condition_variable; |
| |
| // template <class Rep, class Period, class Predicate> |
| // bool |
| // wait_for(unique_lock<mutex>& lock, |
| // const chrono::duration<Rep, Period>& rel_time, |
| // Predicate pred); |
| |
| #include <condition_variable> |
| #include <mutex> |
| #include <thread> |
| #include <chrono> |
| #include <cassert> |
| |
| class Pred |
| { |
| int& i_; |
| public: |
| explicit Pred(int& i) : i_(i) {} |
| |
| bool operator()() {return i_ != 0;} |
| }; |
| |
| std::condition_variable cv; |
| std::mutex mut; |
| |
| int test1 = 0; |
| int test2 = 0; |
| |
| int runs = 0; |
| |
| void f() |
| { |
| typedef std::chrono::system_clock Clock; |
| typedef std::chrono::milliseconds milliseconds; |
| std::unique_lock<std::mutex> lk(mut); |
| assert(test2 == 0); |
| test1 = 1; |
| cv.notify_one(); |
| Clock::time_point t0 = Clock::now(); |
| bool r = cv.wait_for(lk, milliseconds(250), Pred(test2)); |
| Clock::time_point t1 = Clock::now(); |
| if (runs == 0) |
| { |
| assert(t1 - t0 < milliseconds(250)); |
| assert(test2 != 0); |
| } |
| else |
| { |
| assert(t1 - t0 - milliseconds(250) < milliseconds(50)); |
| assert(test2 == 0); |
| } |
| ++runs; |
| } |
| |
| int main() |
| { |
| { |
| std::unique_lock<std::mutex>lk(mut); |
| std::thread t(f); |
| assert(test1 == 0); |
| while (test1 == 0) |
| cv.wait(lk); |
| assert(test1 != 0); |
| test2 = 1; |
| lk.unlock(); |
| cv.notify_one(); |
| t.join(); |
| } |
| test1 = 0; |
| test2 = 0; |
| { |
| std::unique_lock<std::mutex>lk(mut); |
| std::thread t(f); |
| assert(test1 == 0); |
| while (test1 == 0) |
| cv.wait(lk); |
| assert(test1 != 0); |
| lk.unlock(); |
| t.join(); |
| } |
| } |