33 lines
1.1 KiB
C++
33 lines
1.1 KiB
C++
// sherpa-onnx/jni/online-stream.cc
|
|
//
|
|
// Copyright (c) 2024 Xiaomi Corporation
|
|
|
|
#include "sherpa-onnx/csrc/online-stream.h"
|
|
|
|
#include "sherpa-onnx/jni/common.h"
|
|
|
|
SHERPA_ONNX_EXTERN_C
|
|
JNIEXPORT void JNICALL Java_com_k2fsa_sherpa_onnx_OnlineStream_delete(
|
|
JNIEnv * /*env*/, jobject /*obj*/, jlong ptr) {
|
|
delete reinterpret_cast<sherpa_onnx::OnlineStream *>(ptr);
|
|
}
|
|
|
|
SHERPA_ONNX_EXTERN_C
|
|
JNIEXPORT void JNICALL Java_com_k2fsa_sherpa_onnx_OnlineStream_acceptWaveform(
|
|
JNIEnv *env, jobject /*obj*/, jlong ptr, jfloatArray samples,
|
|
jint sample_rate) {
|
|
auto stream = reinterpret_cast<sherpa_onnx::OnlineStream *>(ptr);
|
|
|
|
jfloat *p = env->GetFloatArrayElements(samples, nullptr);
|
|
jsize n = env->GetArrayLength(samples);
|
|
stream->AcceptWaveform(sample_rate, p, n);
|
|
env->ReleaseFloatArrayElements(samples, p, JNI_ABORT);
|
|
}
|
|
|
|
SHERPA_ONNX_EXTERN_C
|
|
JNIEXPORT void JNICALL Java_com_k2fsa_sherpa_onnx_OnlineStream_inputFinished(
|
|
JNIEnv * /*env*/, jobject /*obj*/, jlong ptr) {
|
|
auto stream = reinterpret_cast<sherpa_onnx::OnlineStream *>(ptr);
|
|
stream->InputFinished();
|
|
}
|