Add separate speech probability method

Author: henkelmax

native/src/rnnoise.c

@@ -143,9 +143,9 @@ JNIEXPORT jshortArray JNICALL Java_de_maxhenkel_rnnoise4j_Denoiser_denoise0(
     return pcm_output;
 }
 
-JNIEXPORT jfloat JNICALL Java_de_maxhenkel_rnnoise4j_Denoiser_denoiseInPlace0(
+jfloat denoiseInPlace(
     JNIEnv *env,
-    jobject obj,
+    bool denoise,
     const jlong denoiser_pointer,
     const jshortArray input
 ) {
@@ -200,7 +200,9 @@ JNIEXPORT jfloat JNICALL Java_de_maxhenkel_rnnoise4j_Denoiser_denoiseInPlace0(
         if (speech_probability > total_speech_probability) {
             total_speech_probability = speech_probability;
         }
-        (*env)->SetShortArrayRegion(env, input, i * frame_size, frame_size, output_pcm_buffer);
+        if (denoise) {
+            (*env)->SetShortArrayRegion(env, input, i * frame_size, frame_size, output_pcm_buffer);
+        }
     }
 
     free(input_pcm_buffer);
@@ -211,6 +213,24 @@ JNIEXPORT jfloat JNICALL Java_de_maxhenkel_rnnoise4j_Denoiser_denoiseInPlace0(
     return total_speech_probability;
 }
 
+JNIEXPORT jfloat JNICALL Java_de_maxhenkel_rnnoise4j_Denoiser_denoiseInPlace0(
+    JNIEnv *env,
+    jobject obj,
+    const jlong denoiser_pointer,
+    const jshortArray input
+) {
+    return denoiseInPlace(env, true, denoiser_pointer, input);
+}
+
+JNIEXPORT jfloat JNICALL Java_de_maxhenkel_rnnoise4j_Denoiser_getSpeechProbability0(
+    JNIEnv *env,
+    jobject obj,
+    const jlong denoiser_pointer,
+    const jshortArray input
+) {
+    return denoiseInPlace(env, false, denoiser_pointer, input);
+}
+
 JNIEXPORT void JNICALL Java_de_maxhenkel_rnnoise4j_Denoiser_destroyDenoiser0(
     JNIEnv *env,
     jobject obj,

src/main/java/de/maxhenkel/rnnoise4j/Denoiser.java

@@ -88,6 +88,21 @@ public float denoiseInPlace(short[] input) {
         }
     }
 
+    private native float getSpeechProbability0(long denoiserPointer, short[] input);
+
+    /**
+     * Does the same as {@link #denoiseInPlace(short[])} but does not modify the input.
+     * Used for getting the probability of speech without denoising.
+     *
+     * @param input the input pcm samples
+     * @return the probability of speech (0-1)
+     */
+    public float getSpeechProbability(short[] input) {
+        synchronized (this) {
+            return getSpeechProbability0(pointer, input);
+        }
+    }
+
     private native void destroyDenoiser0(long denoiserPointer);
 
     /**

src/test/java/de/maxhenkel/rnnoise4j/DenoiserTest.java

@@ -48,6 +48,11 @@ void speechProbability() throws IOException, UnknownPlatformException {
             float probability = denoiser.denoiseInPlace(buffer);
             assertEquals(0F, probability);
         }
+        try (Denoiser denoiser = new Denoiser()) {
+            short[] buffer = new short[denoiser.getFrameSize()];
+            float probability = denoiser.getSpeechProbability(buffer);
+            assertEquals(0F, probability);
+        }
     }
 
     @Test