Wrap punctuation APIs to C#. (#945)

2024-05-30 16:19:56 +08:00
parent 082f230dfb
commit a99c7cb35c
31 changed files with 263 additions and 55 deletions
--- a/scripts/dotnet/OfflinePunctuation.cs
+++ b/scripts/dotnet/OfflinePunctuation.cs
@@ -0,0 +1,88 @@
+/// 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 OfflinePunctuation : IDisposable
+    {
+        public OfflinePunctuation(OfflinePunctuationConfig config)
+        {
+            IntPtr h = SherpaOnnxCreateOfflinePunctuation(ref config);
+            _handle = new HandleRef(this, h);
+        }
+
+        public String AddPunct(String text)
+        {
+            IntPtr p = SherpaOfflinePunctuationAddPunct(_handle.Handle, text);
+
+            string s = "";
+            int length = 0;
+
+            unsafe
+            {
+                byte* b = (byte*)p;
+                if (b != null)
+                {
+                    while (*b != 0)
+                    {
+                        ++b;
+                        length += 1;
+                    }
+                }
+            }
+
+            if (length > 0)
+            {
+                byte[] stringBuffer = new byte[length];
+                Marshal.Copy(p, stringBuffer, 0, length);
+                s = Encoding.UTF8.GetString(stringBuffer);
+            }
+
+            SherpaOfflinePunctuationFreeText(p);
+
+            return s;
+        }
+
+        public void Dispose()
+        {
+            Cleanup();
+            // Prevent the object from being placed on the
+            // finalization queue
+            System.GC.SuppressFinalize(this);
+        }
+
+        ~OfflinePunctuation()
+        {
+            Cleanup();
+        }
+
+        private void Cleanup()
+        {
+            SherpaOnnxDestroyOfflinePunctuation(_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 SherpaOnnxCreateOfflinePunctuation(ref OfflinePunctuationConfig config);
+
+        [DllImport(Dll.Filename)]
+        private static extern void SherpaOnnxDestroyOfflinePunctuation(IntPtr handle);
+
+        [DllImport(Dll.Filename)]
+        private static extern IntPtr SherpaOfflinePunctuationAddPunct(IntPtr handle, [MarshalAs(UnmanagedType.LPStr)] string text);
+
+        [DllImport(Dll.Filename)]
+        private static extern void SherpaOfflinePunctuationFreeText(IntPtr p);
+    }
+}
+