146 lines
3.8 KiB
Go
146 lines
3.8 KiB
Go
package vosk
|
|
|
|
// #cgo CPPFLAGS: -I ${SRCDIR}/../src
|
|
// #cgo !windows LDFLAGS: -L ${SRCDIR}/../src -lvosk -ldl -lpthread
|
|
// #cgo windows LDFLAGS: -L ${SRCDIR}/../src -lvosk -lpthread
|
|
// #include <stdlib.h>
|
|
// #include <vosk_api.h>
|
|
import "C"
|
|
import "unsafe"
|
|
|
|
type VoskModel struct {
|
|
model *C.struct_VoskModel
|
|
}
|
|
|
|
func NewModel(modelPath string) (*VoskModel, error) {
|
|
cmodelPath := C.CString(modelPath)
|
|
defer C.free(unsafe.Pointer(cmodelPath))
|
|
internal := C.vosk_model_new(cmodelPath)
|
|
model := &VoskModel{model: internal}
|
|
return model, nil
|
|
}
|
|
|
|
func (m *VoskModel) Free() {
|
|
C.vosk_model_free(m.model)
|
|
}
|
|
|
|
func freeModel(model *VoskModel) {
|
|
C.vosk_model_free(model.model)
|
|
}
|
|
|
|
func (m *VoskModel) FindWord(word string) int {
|
|
cstr := C.CString(word)
|
|
defer C.free(unsafe.Pointer(cstr))
|
|
i := C.vosk_model_find_word(m.model, cstr)
|
|
return int(i)
|
|
}
|
|
|
|
type VoskSpkModel struct {
|
|
spkModel *C.struct_VoskSpkModel
|
|
}
|
|
|
|
func NewSpkModel(spkModelPath string) (*VoskSpkModel, error) {
|
|
cspkModelPath := C.CString(spkModelPath)
|
|
defer C.free(unsafe.Pointer(cspkModelPath))
|
|
internal := C.vosk_spk_model_new(cspkModelPath)
|
|
spkModel := &VoskSpkModel{spkModel: internal}
|
|
return spkModel, nil
|
|
}
|
|
|
|
func freeSpkModel(model *VoskSpkModel) {
|
|
C.vosk_spk_model_free(model.spkModel)
|
|
}
|
|
|
|
func(s *VoskSpkModel) Free() {
|
|
C.vosk_spk_model_free(s.spkModel)
|
|
}
|
|
|
|
type VoskRecognizer struct {
|
|
rec *C.struct_VoskRecognizer
|
|
}
|
|
|
|
func freeRecognizer(recognizer *VoskRecognizer) {
|
|
C.vosk_recognizer_free(recognizer.rec)
|
|
}
|
|
|
|
func (r *VoskRecognizer) Free() {
|
|
C.vosk_recognizer_free(r.rec)
|
|
}
|
|
|
|
func NewRecognizer(model *VoskModel, sampleRate float64) (*VoskRecognizer, error) {
|
|
internal := C.vosk_recognizer_new(model.model, C.float(sampleRate))
|
|
rec := &VoskRecognizer{rec: internal}
|
|
return rec, nil
|
|
}
|
|
|
|
func NewRecognizerSpk(model *VoskModel, sampleRate float64, spkModel *VoskSpkModel) (*VoskRecognizer, error) {
|
|
internal := C.vosk_recognizer_new_spk(model.model, C.float(sampleRate), spkModel.spkModel)
|
|
rec := &VoskRecognizer{rec: internal}
|
|
return rec, nil
|
|
}
|
|
|
|
func NewRecognizerGrm(model *VoskModel, sampleRate float64, grammar string) (*VoskRecognizer, error) {
|
|
cgrammar := C.CString(grammar)
|
|
defer C.free(unsafe.Pointer(cgrammar))
|
|
internal := C.vosk_recognizer_new_grm(model.model, C.float(sampleRate), cgrammar)
|
|
rec := &VoskRecognizer{rec: internal}
|
|
return rec, nil
|
|
}
|
|
|
|
func (r *VoskRecognizer) SetSpkModel(spkModel *VoskSpkModel) {
|
|
C.vosk_recognizer_set_spk_model(r.rec, spkModel.spkModel)
|
|
}
|
|
|
|
func (r *VoskRecognizer) SetGrm(grammar string) {
|
|
cgrammar := C.CString(grammar)
|
|
defer C.free(unsafe.Pointer(cgrammar))
|
|
C.vosk_recognizer_set_grm(r.rec, cgrammar)
|
|
}
|
|
|
|
func (r *VoskRecognizer) SetMaxAlternatives(maxAlternatives int) {
|
|
C.vosk_recognizer_set_max_alternatives(r.rec, C.int(maxAlternatives))
|
|
}
|
|
|
|
func (r *VoskRecognizer) SetWords(words int) {
|
|
C.vosk_recognizer_set_words(r.rec, C.int(words))
|
|
}
|
|
|
|
func (r *VoskRecognizer) SetPartialWords(words int) {
|
|
C.vosk_recognizer_set_partial_words(r.rec, C.int(words))
|
|
}
|
|
|
|
func (r *VoskRecognizer) AcceptWaveform(buffer []byte) int {
|
|
cbuf := C.CBytes(buffer)
|
|
defer C.free(cbuf)
|
|
i := C.vosk_recognizer_accept_waveform(r.rec, (*C.char)(cbuf), C.int(len(buffer)))
|
|
return int(i)
|
|
}
|
|
|
|
func (r *VoskRecognizer) Result() string {
|
|
return C.GoString(C.vosk_recognizer_result(r.rec))
|
|
}
|
|
|
|
func (r *VoskRecognizer) PartialResult() string {
|
|
return C.GoString(C.vosk_recognizer_partial_result(r.rec))
|
|
}
|
|
|
|
func (r *VoskRecognizer) FinalResult() string {
|
|
return C.GoString(C.vosk_recognizer_final_result(r.rec))
|
|
}
|
|
|
|
func (r *VoskRecognizer) Reset() {
|
|
C.vosk_recognizer_reset(r.rec)
|
|
}
|
|
|
|
func SetLogLevel(logLevel int) {
|
|
C.vosk_set_log_level(C.int(logLevel))
|
|
}
|
|
|
|
func GPUInit() {
|
|
C.vosk_gpu_init()
|
|
}
|
|
|
|
func GPUThreadInit() {
|
|
C.vosk_gpu_thread_init()
|
|
}
|