1//===----------------------------------------------------------------------===// 2// 3// The LLVM Compiler Infrastructure 4// 5// This file is dual licensed under the MIT and the University of Illinois Open 6// Source Licenses. See LICENSE.TXT for details. 7// 8//===----------------------------------------------------------------------===// 9// 10// UNSUPPORTED: libcpp-has-no-threads 11 12// <condition_variable> 13 14// class condition_variable_any; 15 16// template <class Lock, class Rep, class Period, class Predicate> 17// bool 18// wait_for(Lock& lock, const chrono::duration<Rep, Period>& rel_time, 19// Predicate pred); 20 21#include <condition_variable> 22#include <mutex> 23#include <thread> 24#include <chrono> 25#include <cassert> 26 27class Pred 28{ 29 int& i_; 30public: 31 explicit Pred(int& i) : i_(i) {} 32 33 bool operator()() {return i_ != 0;} 34}; 35 36std::condition_variable_any cv; 37 38typedef std::timed_mutex L0; 39typedef std::unique_lock<L0> L1; 40 41L0 m0; 42 43int test1 = 0; 44int test2 = 0; 45 46int runs = 0; 47 48void f() 49{ 50 typedef std::chrono::system_clock Clock; 51 typedef std::chrono::milliseconds milliseconds; 52 L1 lk(m0); 53 assert(test2 == 0); 54 test1 = 1; 55 cv.notify_one(); 56 Clock::time_point t0 = Clock::now(); 57 bool r = cv.wait_for(lk, milliseconds(250), Pred(test2)); 58 Clock::time_point t1 = Clock::now(); 59 if (runs == 0) 60 { 61 assert(t1 - t0 < milliseconds(250)); 62 assert(test2 != 0); 63 } 64 else 65 { 66 assert(t1 - t0 - milliseconds(250) < milliseconds(50)); 67 assert(test2 == 0); 68 } 69 ++runs; 70} 71 72int main() 73{ 74 { 75 L1 lk(m0); 76 std::thread t(f); 77 assert(test1 == 0); 78 while (test1 == 0) 79 cv.wait(lk); 80 assert(test1 != 0); 81 test2 = 1; 82 lk.unlock(); 83 cv.notify_one(); 84 t.join(); 85 } 86 test1 = 0; 87 test2 = 0; 88 { 89 L1 lk(m0); 90 std::thread t(f); 91 assert(test1 == 0); 92 while (test1 == 0) 93 cv.wait(lk); 94 assert(test1 != 0); 95 lk.unlock(); 96 t.join(); 97 } 98} 99