You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
1.4 KiB
58 lines
1.4 KiB
3 years ago
|
// 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
|