jamosdev
/
rhubarb-lip-sync
mirror of https://github.com/DanielSWolf/rhubarb-lip-sync.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fixed VAD error handling

This commit is contained in:
Daniel Wolf 2016-06-30 20:17:28 +02:00
parent ed27b8470c
commit 6de7ba020a
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/audio/voiceActivityDetection.cpp
View File

@ -31,14 +31,20 @@ BoundedTimeline<void> webRtcDetectVoiceActivity(AudioStream& audioStream, Progre
// Detect activity
BoundedTimeline<void> activity(audioStream.getTruncatedRange());
centiseconds time = centiseconds::zero();
const size_t bufferCapacity = audioStream.getSampleRate() / 100;
auto processBuffer = [&](const vector<int16_t>& buffer) {
bool isActive = WebRtcVad_Process(vadHandle, audioStream.getSampleRate(), buffer.data(), buffer.size()) == 1;
// WebRTC is picky regarding buffer size
if (buffer.size() < bufferCapacity) return;
int result = WebRtcVad_Process(vadHandle, audioStream.getSampleRate(), buffer.data(), buffer.size()) == 1;
if (result == -1) throw runtime_error("Error processing audio buffer using WebRTC VAD.");
bool isActive = result != 0;
if (isActive) {
activity.set(time, time + centiseconds(1));
}
time += centiseconds(1);
};
const size_t bufferCapacity = audioStream.getSampleRate() / 100;
process16bitAudioStream(audioStream, processBuffer, bufferCapacity, progressSink);
return activity;