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

Handling zero-length audio files

This commit is contained in:
Daniel Wolf 2016-04-12 18:07:53 +02:00
parent 7bc4e37a1a
commit 90e1375f1b
3 changed files with 13 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/audio/WaveFileReader.cpp
View File

@ -185,7 +185,7 @@ int WaveFileReader::getSampleIndex() {
} }
void WaveFileReader::seek(int sampleIndex) { void WaveFileReader::seek(int sampleIndex) {
if (sampleIndex < 0 || sampleIndex >= sampleCount) throw std::invalid_argument("sampleIndex out of range."); if (sampleIndex < 0 || sampleIndex > sampleCount) throw std::invalid_argument("sampleIndex out of range.");
file.seekg(dataOffset + static_cast<std::streamoff>(sampleIndex * channelCount * bytesPerSample)); file.seekg(dataOffset + static_cast<std::streamoff>(sampleIndex * channelCount * bytesPerSample));
this->sampleIndex = sampleIndex; this->sampleIndex = sampleIndex;

12
src/main.cpp
View File

@ -50,20 +50,22 @@ ptree createXmlTree(const path& filePath, const Timeline<Phone>& phones, const T
ptree tree; ptree tree;
// Add sound file path // Add sound file path
tree.add("rhubarbResult.info.soundFile", filePath.string()); tree.put("rhubarbResult.info.soundFile", filePath.string());
// Add phones // Add phones
tree.put("rhubarbResult.phones", "");
for (auto& timedPhone : phones) { for (auto& timedPhone : phones) {
ptree& phoneElement = tree.add("rhubarbResult.phones.phone", timedPhone.getValue()); ptree& phoneElement = tree.add("rhubarbResult.phones.phone", timedPhone.getValue());
phoneElement.add("<xmlattr>.start", formatDuration(timedPhone.getStart())); phoneElement.put("<xmlattr>.start", formatDuration(timedPhone.getStart()));
phoneElement.add("<xmlattr>.duration", formatDuration(timedPhone.getLength())); phoneElement.put("<xmlattr>.duration", formatDuration(timedPhone.getLength()));
} }
// Add mouth cues // Add mouth cues
tree.put("rhubarbResult.mouthCues", "");
for (auto& timedShape : shapes) { for (auto& timedShape : shapes) {
ptree& mouthCueElement = tree.add("rhubarbResult.mouthCues.mouthCue", timedShape.getValue()); ptree& mouthCueElement = tree.add("rhubarbResult.mouthCues.mouthCue", timedShape.getValue());
mouthCueElement.add("<xmlattr>.start", formatDuration(timedShape.getStart())); mouthCueElement.put("<xmlattr>.start", formatDuration(timedShape.getStart()));
mouthCueElement.add("<xmlattr>.duration", formatDuration(timedShape.getLength())); mouthCueElement.put("<xmlattr>.duration", formatDuration(timedShape.getLength()));
} }
return tree; return tree;

5
src/phoneExtraction.cpp
View File

@ -288,6 +288,11 @@ Timeline<Phone> detectPhones(
boost::optional<std::string> dialog, boost::optional<std::string> dialog,
ProgressSink& progressSink) ProgressSink& progressSink)
{ {
// Pocketsphinx doesn't like empty input
if (audioStream->getTruncatedRange().getLength() == centiseconds::zero()) {
return Timeline<Phone>{};
}
// Discard Pocketsphinx output // Discard Pocketsphinx output
err_set_logfp(nullptr); err_set_logfp(nullptr);