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 Predicate> 15bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant// void wait(unique_lock<mutex>& lock, Predicate pred); 16bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant 17bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant#include <condition_variable> 18bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant#include <mutex> 19bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant#include <thread> 20bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant#include <functional> 21bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant#include <cassert> 22bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant 23bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantstd::condition_variable cv; 24bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantstd::mutex mut; 25bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant 26bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantint test1 = 0; 27bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantint test2 = 0; 28bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant 29bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantclass Pred 30bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant{ 31bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant int& i_; 32bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantpublic: 33bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant explicit Pred(int& i) : i_(i) {} 34bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant 35bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant bool operator()() {return i_ != 0;} 36bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant}; 37bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant 38bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantvoid f() 39bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant{ 40bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant std::unique_lock<std::mutex> lk(mut); 41bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(test2 == 0); 42bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant test1 = 1; 43bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant cv.notify_one(); 44bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant cv.wait(lk, Pred(test2)); 45bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(test2 != 0); 46bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant} 47bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant 48bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantint main() 49bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant{ 50bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant std::unique_lock<std::mutex>lk(mut); 51bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant std::thread t(f); 52bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(test1 == 0); 53bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant while (test1 == 0) 54bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant cv.wait(lk); 55bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(test1 != 0); 56bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant test2 = 1; 57bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant lk.unlock(); 58bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant cv.notify_one(); 59bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant t.join(); 60bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant} 61