90 lines
3.2 KiB
C++
90 lines
3.2 KiB
C++
#include "recognizer.h"
|
|
|
|
recognizer::recognizer(int index, float sampleRate, genericModel* model) : index(index) {
|
|
rec = vosk_recognizer_new(std::get<0>(model->mdl),sampleRate);
|
|
finishConstruction(model);
|
|
}
|
|
recognizer::recognizer(int index, float sampleRate, genericModel* model, genericModel* spkMdl) : index(index) {
|
|
rec = vosk_recognizer_new_spk(std::get<0>(model->mdl), sampleRate, std::get<1>(spkMdl->mdl));
|
|
finishConstruction(model, spkMdl);
|
|
}
|
|
recognizer::recognizer(int index, float sampleRate, genericModel* model, const std::string& grm, int dummy) : index(index) {
|
|
rec = vosk_recognizer_new_grm(std::get<0>(model->mdl), sampleRate, grm.c_str());
|
|
finishConstruction(model);
|
|
}
|
|
recognizer::~recognizer() {
|
|
done.test_and_set(std::memory_order_relaxed);
|
|
done.notify_one();
|
|
controller.test_and_set(std::memory_order_relaxed);
|
|
controller.notify_one();
|
|
vosk_recognizer_free(rec);
|
|
}
|
|
void recognizer::finishConstruction(genericModel* model, genericModel* spkModel) {
|
|
if(rec == nullptr) {
|
|
fireEv(index, "Unable to initialize recognizer");
|
|
return;
|
|
}
|
|
auto main {[this](){
|
|
emscripten_console_log("Recognizer loaded!");
|
|
std::string loadInfo{};
|
|
loadInfo += reinterpret_cast<int>(dataBuf);
|
|
loadInfo += ",";
|
|
loadInfo += reinterpret_cast<int>(&state);
|
|
fireEv(index, loadInfo.c_str());
|
|
while(!done.test(std::memory_order_relaxed)) {
|
|
controller.wait(!done.test(std::memory_order_relaxed), std::memory_order_relaxed);
|
|
controller.clear(std::memory_order_relaxed);
|
|
switch(vosk_recognizer_accept_waveform_f(rec, dataBuf, 512)) {
|
|
case 0:
|
|
fireEv(index, vosk_recognizer_result(rec), "result");
|
|
break;
|
|
case 1:
|
|
fireEv(index, vosk_recognizer_partial_result(rec), "partialResult");
|
|
}
|
|
state = 0;
|
|
}
|
|
}};
|
|
if(!model->recognizerUsedThrd) {
|
|
model->recognizerUsedThrd = true;
|
|
model->thrd.addTask(main);
|
|
emscripten_console_log("Adding task to model thread...");
|
|
return;
|
|
}
|
|
if(spkModel != nullptr && !spkModel->recognizerUsedThrd) {
|
|
spkModel->recognizerUsedThrd = true;
|
|
spkModel->thrd.addTask(main);
|
|
emscripten_console_log("Adding task to speaker model thread...");
|
|
return;
|
|
}
|
|
emscripten_console_log("New recognizer thread");
|
|
std::thread t{main};
|
|
t.detach();
|
|
}
|
|
void recognizer::setEndpointerMode(VoskEndpointerMode mode) {
|
|
vosk_recognizer_set_endpointer_mode(rec, mode);
|
|
}
|
|
void recognizer::setEndpointerDelays(float tStartMax, float tEnd, float tMax) {
|
|
vosk_recognizer_set_endpointer_delays(rec, tStartMax, tEnd, tMax);
|
|
}
|
|
void recognizer::acceptWaveForm() {
|
|
controller.test_and_set(std::memory_order_relaxed);
|
|
controller.notify_one();
|
|
}
|
|
void recognizer::setGrm(const std::string& grm) {
|
|
vosk_recognizer_set_grm(rec, grm.c_str());
|
|
}
|
|
void recognizer::setSpkModel(genericModel* spkModel) {
|
|
vosk_recognizer_set_spk_model(rec, std::get<1>(spkModel->mdl));
|
|
}
|
|
void recognizer::setWords(bool words) {
|
|
vosk_recognizer_set_words(rec,words);
|
|
}
|
|
void recognizer::setPartialWords(bool partialWords) {
|
|
vosk_recognizer_set_partial_words(rec, partialWords);
|
|
}
|
|
void recognizer::setNLSML(bool nlsml) {
|
|
vosk_recognizer_set_nlsml(rec, nlsml);
|
|
}
|
|
void recognizer::setMaxAlternatives(int alts) {
|
|
vosk_recognizer_set_max_alternatives(rec, alts);
|
|
} |