EngineX-Iluvatar/enginex-mr_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

Support decoding multiple streams in Java API. (#2149)

Browse Source
This commit is contained in:
Fangjun Kuang
2025-04-25 11:18:57 +08:00
committed by GitHub
parent 48ab90aadc
commit e3280027f9
7 changed files with 160 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
sherpa-onnx/jni/offline-recognizer.cc
View File

@@ -366,21 +366,43 @@ Java_com_k2fsa_sherpa_onnx_OfflineRecognizer_createStream(JNIEnv * /*env*/,
SHERPA_ONNX_EXTERN_C
JNIEXPORT void JNICALL Java_com_k2fsa_sherpa_onnx_OfflineRecognizer_decode(
JNIEnv *env, jobject /*obj*/, jlong ptr, jlong streamPtr) {
JNIEnv *env, jobject /*obj*/, jlong ptr, jlong stream_ptr) {
SafeJNI(env, "OfflineRecognizer_decode", [&] {
if (!ValidatePointer(env, ptr, "OfflineRecognizer_decode",
"OfflineRecognizer pointer is null.") ||
!ValidatePointer(env, streamPtr, "OfflineRecognizer_decode",
!ValidatePointer(env, stream_ptr, "OfflineRecognizer_decode",
"OfflineStream pointer is null.")) {
return;
}
auto recognizer = reinterpret_cast<sherpa_onnx::OfflineRecognizer *>(ptr);
auto stream = reinterpret_cast<sherpa_onnx::OfflineStream *>(streamPtr);
auto stream = reinterpret_cast<sherpa_onnx::OfflineStream *>(stream_ptr);
recognizer->DecodeStream(stream);
});
}
SHERPA_ONNX_EXTERN_C
JNIEXPORT void JNICALL
Java_com_k2fsa_sherpa_onnx_OfflineRecognizer_decodeStreams(
JNIEnv *env, jobject /*obj*/, jlong ptr, jlongArray stream_ptrs) {
SafeJNI(env, "OfflineRecognizer_decode_streams", [&] {
if (!ValidatePointer(env, ptr, "OfflineRecognizer_decode_streams",
"OfflineRecognizer pointer is null.")) {
return;
}
auto recognizer = reinterpret_cast<sherpa_onnx::OfflineRecognizer *>(ptr);
jlong *p = env->GetLongArrayElements(stream_ptrs, nullptr);
jsize n = env->GetArrayLength(stream_ptrs);
auto ss = reinterpret_cast<sherpa_onnx::OfflineStream **>(p);
recognizer->DecodeStreams(ss, n);
env->ReleaseLongArrayElements(stream_ptrs, p, JNI_ABORT);
});
}
SHERPA_ONNX_EXTERN_C
JNIEXPORT jobjectArray JNICALL
Java_com_k2fsa_sherpa_onnx_OfflineRecognizer_getResult(JNIEnv *env,

16
sherpa-onnx/jni/online-recognizer.cc
View File

@@ -339,6 +339,22 @@ JNIEXPORT void JNICALL Java_com_k2fsa_sherpa_onnx_OnlineRecognizer_decode(
recognizer->DecodeStream(stream);
}
SHERPA_ONNX_EXTERN_C
JNIEXPORT void JNICALL
Java_com_k2fsa_sherpa_onnx_OnlineRecognizer_decodeStreams(
JNIEnv *env, jobject /*obj*/, jlong ptr, jlongArray stream_ptrs) {
auto recognizer = reinterpret_cast<sherpa_onnx::OnlineRecognizer *>(ptr);
jlong *p = env->GetLongArrayElements(stream_ptrs, nullptr);
jsize n = env->GetArrayLength(stream_ptrs);
auto ss = reinterpret_cast<sherpa_onnx::OnlineStream **>(p);
recognizer->DecodeStreams(ss, n);
env->ReleaseLongArrayElements(stream_ptrs, p, JNI_ABORT);
}
SHERPA_ONNX_EXTERN_C
JNIEXPORT jlong JNICALL
Java_com_k2fsa_sherpa_onnx_OnlineRecognizer_createStream(JNIEnv *env,