Wrap VAD APIs to C# (#946)

2024-05-30 18:29:38 +08:00
parent a99c7cb35c
commit 0196f1d546
17 changed files with 474 additions and 10 deletions
--- a/scripts/dotnet/VoiceActivityDetector.cs
+++ b/scripts/dotnet/VoiceActivityDetector.cs
@@ -0,0 +1,115 @@
+/// Copyright (c)  2024  Xiaomi Corporation (authors: Fangjun Kuang)
+
+using System.Linq;
+using System.Collections.Generic;
+using System.Runtime.InteropServices;
+using System.Text;
+using System;
+
+namespace SherpaOnnx
+{
+    public class VoiceActivityDetector : IDisposable
+    {
+        public VoiceActivityDetector(VadModelConfig config, float bufferSizeInSeconds)
+        {
+            IntPtr h = SherpaOnnxCreateVoiceActivityDetector(ref config, bufferSizeInSeconds);
+            _handle = new HandleRef(this, h);
+        }
+
+        public void AcceptWaveform(float[] samples)
+        {
+            SherpaOnnxVoiceActivityDetectorAcceptWaveform(_handle.Handle, samples, samples.Length);
+        }
+
+        public bool IsEmpty()
+        {
+            return SherpaOnnxVoiceActivityDetectorEmpty(_handle.Handle) == 1;
+        }
+
+        public bool IsSpeechDetected()
+        {
+            return SherpaOnnxVoiceActivityDetectorDetected(_handle.Handle) == 1;
+        }
+
+        public void Pop()
+        {
+            SherpaOnnxVoiceActivityDetectorPop(_handle.Handle);
+        }
+
+        public SpeechSegment Front()
+        {
+            IntPtr p = SherpaOnnxVoiceActivityDetectorFront(_handle.Handle);
+
+            SpeechSegment segment = new SpeechSegment(p);
+
+            SherpaOnnxDestroySpeechSegment(p);
+
+            return segment;
+        }
+
+        public void Clear()
+        {
+            SherpaOnnxVoiceActivityDetectorClear(_handle.Handle);
+        }
+
+        public void Reset()
+        {
+            SherpaOnnxVoiceActivityDetectorReset(_handle.Handle);
+        }
+
+        public void Dispose()
+        {
+            Cleanup();
+            // Prevent the object from being placed on the
+            // finalization queue
+            System.GC.SuppressFinalize(this);
+        }
+
+        ~VoiceActivityDetector()
+        {
+            Cleanup();
+        }
+
+        private void Cleanup()
+        {
+            SherpaOnnxDestroyVoiceActivityDetector(_handle.Handle);
+
+            // Don't permit the handle to be used again.
+            _handle = new HandleRef(this, IntPtr.Zero);
+        }
+
+        private HandleRef _handle;
+
+        [DllImport(Dll.Filename)]
+        private static extern IntPtr SherpaOnnxCreateVoiceActivityDetector(ref VadModelConfig config, float bufferSizeInSeconds);
+
+        [DllImport(Dll.Filename)]
+        private static extern void SherpaOnnxDestroyVoiceActivityDetector(IntPtr handle);
+
+        [DllImport(Dll.Filename)]
+        private static extern void SherpaOnnxVoiceActivityDetectorAcceptWaveform(IntPtr handle, float[] samples, int n);
+
+        [DllImport(Dll.Filename)]
+        private static extern int SherpaOnnxVoiceActivityDetectorEmpty(IntPtr handle);
+
+        [DllImport(Dll.Filename)]
+        private static extern int SherpaOnnxVoiceActivityDetectorDetected(IntPtr handle);
+
+        [DllImport(Dll.Filename)]
+        private static extern void SherpaOnnxVoiceActivityDetectorPop(IntPtr handle);
+
+        [DllImport(Dll.Filename)]
+        private static extern void SherpaOnnxVoiceActivityDetectorClear(IntPtr handle);
+
+        [DllImport(Dll.Filename)]
+        private static extern IntPtr SherpaOnnxVoiceActivityDetectorFront(IntPtr handle);
+
+        [DllImport(Dll.Filename)]
+        private static extern void SherpaOnnxDestroySpeechSegment(IntPtr segment);
+
+        [DllImport(Dll.Filename)]
+        private static extern void SherpaOnnxVoiceActivityDetectorReset(IntPtr handle);
+
+    }
+}
+