Update offline-recognizer.cc (#1253)

Adding setConfig method to JNI to support setting a config on the previously initialized offline-recognizer.
k2-fsa · Aug 13, 2024 · 9f06b05 · 9f06b05
1 parent 619279b
commit 9f06b05
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/sherpa-onnx/jni/offline-recognizer.cc b/sherpa-onnx/jni/offline-recognizer.cc
@@ -264,6 +264,20 @@ Java_com_k2fsa_sherpa_onnx_OfflineRecognizer_newFromFile(JNIEnv *env,
   return (jlong)model;
 }
 
+
+SHERPA_ONNX_EXTERN_C
+JNIEXPORT void JNICALL
+Java_com_k2fsa_sherpa_onnx_OfflineRecognizer_setConfig(JNIEnv *env,
+                                                         jobject /*obj*/,
+                                                         jlong ptr,
+                                                         jobject _config) {
+  auto config = sherpa_onnx::GetOfflineConfig(env, _config);
+  SHERPA_ONNX_LOGE("config:\n%s", config.ToString().c_str());
+
+  auto recognizer = reinterpret_cast<sherpa_onnx::OfflineRecognizer *>(ptr);
+  recognizer->SetConfig(config);
+}
+
 SHERPA_ONNX_EXTERN_C
 JNIEXPORT void JNICALL Java_com_k2fsa_sherpa_onnx_OfflineRecognizer_delete(
     JNIEnv * /*env*/, jobject /*obj*/, jlong ptr) {