Blame view
src/util/kaldi-semaphore.cc
1.43 KB
8dcb6dfcb first commit |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
// util/kaldi-semaphore.cc // Copyright 2012 Karel Vesely (Brno University of Technology) // 2017 Dogan Can (University of Southern California) // See ../../COPYING for clarification regarding multiple authors // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // THIS CODE IS PROVIDED *AS IS* BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY // KIND, EITHER EXPRESS OR IMPLIED, INCLUDING WITHOUT LIMITATION ANY IMPLIED // WARRANTIES OR CONDITIONS OF TITLE, FITNESS FOR A PARTICULAR PURPOSE, // MERCHANTABLITY OR NON-INFRINGEMENT. // See the Apache 2 License for the specific language governing permissions and // limitations under the License. #include "base/kaldi-error.h" #include "util/kaldi-semaphore.h" namespace kaldi { Semaphore::Semaphore(int32 count) { KALDI_ASSERT(count >= 0); count_ = count; } Semaphore::~Semaphore() {} bool Semaphore::TryWait() { std::unique_lock<std::mutex> lock(mutex_); if(count_) { count_--; return true; } return false; } void Semaphore::Wait() { std::unique_lock<std::mutex> lock(mutex_); while(!count_) condition_variable_.wait(lock); count_--; } void Semaphore::Signal() { std::unique_lock<std::mutex> lock(mutex_); count_++; condition_variable_.notify_one(); } } // namespace kaldi |