1bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant//===----------------------------------------------------------------------===// 2bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant// 3f5256e16dfc425c1d466f6308d4026d529ce9e0bHoward Hinnant// The LLVM Compiler Infrastructure 4bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant// 5b64f8b07c104c6cc986570ac8ee0ed16a9f23976Howard Hinnant// This file is dual licensed under the MIT and the University of Illinois Open 6b64f8b07c104c6cc986570ac8ee0ed16a9f23976Howard Hinnant// Source Licenses. See LICENSE.TXT for details. 7bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant// 8bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant//===----------------------------------------------------------------------===// 9bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant 10bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant// <condition_variable> 11bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant 12bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant// class condition_variable; 13bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant 14bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant// template <class Clock, class Duration, class Predicate> 15bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant// bool 16bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant// wait_until(unique_lock<mutex>& lock, 17bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant// const chrono::time_point<Clock, Duration>& abs_time, 18bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant// Predicate pred); 19bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant 20bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant#include <condition_variable> 21bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant#include <mutex> 22bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant#include <thread> 23bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant#include <chrono> 24bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant#include <cassert> 25bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant 26bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantstruct Clock 27bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant{ 28bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant typedef std::chrono::milliseconds duration; 29bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant typedef duration::rep rep; 30bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant typedef duration::period period; 31bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant typedef std::chrono::time_point<Clock> time_point; 32f8f852138f86e4588916021e1afedfcab25298c0Howard Hinnant static const bool is_steady = true; 33bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant 34bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant static time_point now() 35bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant { 36bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant using namespace std::chrono; 37bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant return time_point(duration_cast<duration>( 38f8f852138f86e4588916021e1afedfcab25298c0Howard Hinnant steady_clock::now().time_since_epoch() 39bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant )); 40bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant } 41bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant}; 42bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant 43bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantclass Pred 44bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant{ 45bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant int& i_; 46bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantpublic: 47bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant explicit Pred(int& i) : i_(i) {} 48bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant 49bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant bool operator()() {return i_ != 0;} 50bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant}; 51bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant 52bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantstd::condition_variable cv; 53bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantstd::mutex mut; 54bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant 55bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantint test1 = 0; 56bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantint test2 = 0; 57bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant 58bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantint runs = 0; 59bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant 60bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantvoid f() 61bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant{ 62bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant std::unique_lock<std::mutex> lk(mut); 63bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(test2 == 0); 64bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant test1 = 1; 65bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant cv.notify_one(); 66bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant Clock::time_point t0 = Clock::now(); 67bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant Clock::time_point t = t0 + Clock::duration(250); 68bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant bool r = cv.wait_until(lk, t, Pred(test2)); 69bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant Clock::time_point t1 = Clock::now(); 70bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant if (runs == 0) 71bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant { 72bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(t1 - t0 < Clock::duration(250)); 73bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(test2 != 0); 74bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(r); 75bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant } 76bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant else 77bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant { 788eba41324c5519557e76092fe8fe98ebb8aa5f5bDaniel Dunbar assert(t1 - t0 - Clock::duration(250) < Clock::duration(50)); 79bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(test2 == 0); 80bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(!r); 81bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant } 82bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant ++runs; 83bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant} 84bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant 85bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantint main() 86bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant{ 87bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant { 88bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant std::unique_lock<std::mutex>lk(mut); 89bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant std::thread t(f); 90bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(test1 == 0); 91bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant while (test1 == 0) 92bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant cv.wait(lk); 93bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(test1 != 0); 94bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant test2 = 1; 95bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant lk.unlock(); 96bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant cv.notify_one(); 97bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant t.join(); 98bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant } 99bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant test1 = 0; 100bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant test2 = 0; 101bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant { 102bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant std::unique_lock<std::mutex>lk(mut); 103bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant std::thread t(f); 104bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(test1 == 0); 105bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant while (test1 == 0) 106bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant cv.wait(lk); 107bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(test1 != 0); 108bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant lk.unlock(); 109bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant t.join(); 110bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant } 111bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant} 112