added Recognizer.setGrammar(String) for java (#1229)

alphacep · Dec 17, 2022 · 64d84b0 · 64d84b0
1 parent cf2560c
commit 64d84b0
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/java/lib/src/main/java/org/vosk/LibVosk.java b/java/lib/src/main/java/org/vosk/LibVosk.java
@@ -78,6 +78,8 @@ private static void unpackDll(File targetDir, String lib) throws IOException {
 
     public static native String vosk_recognizer_partial_result(Pointer recognizer);
 
+    public static native void vosk_recognizer_set_grm(Pointer recognizer, String grammar);
+
     public static native void vosk_recognizer_reset(Pointer recognizer);
 
     public static native void vosk_recognizer_free(Pointer recognizer);

diff --git a/java/lib/src/main/java/org/vosk/Recognizer.java b/java/lib/src/main/java/org/vosk/Recognizer.java
@@ -60,6 +60,16 @@ public String getFinalResult() {
         return LibVosk.vosk_recognizer_final_result(this.getPointer());
     }
 
+    /**
+     * Reconfigures recognizer to use grammar.
+     *
+     * @param grammar      Set of phrases in JSON array of strings or "[]" to use default model graph.
+     * @see #Recognizer(Model, float, String)
+     */
+    public void setGrammar(String grammar) {
+        LibVosk.vosk_recognizer_set_grm(this.getPointer(), grammar);
+    }
+
     public void reset() {
         LibVosk.vosk_recognizer_reset(this.getPointer());
     }