40 lines
1.0 KiB
C
40 lines
1.0 KiB
C
|
#pragma once
|
||
|
|
|
||
|
|
#include "time/BoundedTimeline.h"
|
||
|
|
#include "core/Phone.h"
|
||
|
|
#include "audio/AudioClip.h"
|
||
|
|
#include "tools/ProgressBar.h"
|
||
|
|
#include <boost/filesystem/path.hpp>
|
||
|
|
|
||
|
|
extern "C" {
|
||
|
|
#include <pocketsphinx.h>
|
||
|
|
}
|
||
|
|
|
||
|
|
typedef std::function<lambda_unique_ptr<ps_decoder_t>(
|
||
|
|
boost::optional<std::string> dialog
|
||
|
|
)> decoderFactory;
|
||
|
|
|
||
|
|
typedef std::function<Timeline<Phone>(
|
||
|
|
const AudioClip& audioClip,
|
||
|
|
TimeRange utteranceTimeRange,
|
||
|
|
ps_decoder_t& decoder,
|
||
|
|
ProgressSink& utteranceProgressSink
|
||
|
|
)> utteranceToPhonesFunction;
|
||
|
|
|
||
|
|
BoundedTimeline<Phone> recognizePhones(
|
||
|
|
const AudioClip& inputAudioClip,
|
||
|
|
boost::optional<std::string> dialog,
|
||
|
|
decoderFactory createDecoder,
|
||
|
|
utteranceToPhonesFunction utteranceToPhones,
|
||
|
|
int maxThreadCount,
|
||
|
|
ProgressSink& progressSink
|
||
|
|
);
|
||
|
|
|
||
|
|
constexpr int sphinxSampleRate = 16000;
|
||
|
|
|
||
|
|
const boost::filesystem::path& getSphinxModelDirectory();
|
||
|
|
|
||
|
|
JoiningTimeline<void> getNoiseSounds(TimeRange utteranceTimeRange, const Timeline<Phone>& phones);
|
||
|
|
|
||
|
|
BoundedTimeline<std::string> recognizeWords(const std::vector<int16_t>& audioBuffer, ps_decoder_t& decoder);
|