Add CI

Author: csukuangfj

.github/workflows/c-api.yaml

@@ -108,6 +108,8 @@ jobs:
           cp -v inp_16k.wav denoised-wavs
           cp -v enhanced_16k.wav denoised-wavs
 
+          rm $name
+
       - uses: actions/upload-artifact@v4
         with:
           name: denoised-wavs-${{ matrix.os }}

.github/workflows/cxx-api.yaml

@@ -81,6 +81,44 @@ jobs:
             otool -L ./install/lib/libsherpa-onnx-cxx-api.dylib
           fi
 
+      - name: Test Speech Enhancement (GTCRN)
+        shell: bash
+        run: |
+          name=speech-enhancement-gtcrn-cxx-api
+          g++ -std=c++17 -o $name ./cxx-api-examples/$name.cc \
+            -I ./build/install/include \
+            -L ./build/install/lib/ \
+            -l sherpa-onnx-cxx-api \
+            -l sherpa-onnx-c-api \
+            -l onnxruntime
+
+          ls -lh $name
+
+          export LD_LIBRARY_PATH=$PWD/build/install/lib:$LD_LIBRARY_PATH
+          export DYLD_LIBRARY_PATH=$PWD/build/install/lib:$DYLD_LIBRARY_PATH
+
+          if [[ ${{ matrix.os }} == ubuntu-latest || ${{ matrix.os }} == ubuntu-22.04-arm ]]; then
+            ldd ./$name
+            echo "----"
+            readelf -d ./$name
+          fi
+
+          curl -SL -O https://github.com/k2-fsa/sherpa-onnx/releases/download/speech-enhancement-models/gtcrn_simple.onnx
+          curl -SL -O https://github.com/k2-fsa/sherpa-onnx/releases/download/speech-enhancement-models/inp_16k.wav
+
+          ./$name
+
+          mkdir denoised-wavs
+          cp -v inp_16k.wav denoised-wavs
+          cp -v enhanced_16k.wav denoised-wavs
+
+          rm $name
+
+      - uses: actions/upload-artifact@v4
+        with:
+          name: denoised-wavs-cxx-${{ matrix.os }}
+          path: ./denoised-wavs/*.wav
+
       - name: Test FireRedAsr
         shell: bash
         run: |

cxx-api-examples/speech-enhancement-gtcrn-cxx-api.cc

@@ -14,7 +14,9 @@ wget https://github.com/k2-fsa/sherpa-onnx/releases/download/speech-enhancement-
 wget https://github.com/k2-fsa/sherpa-onnx/releases/download/speech-enhancement-models/inp_16k.wav
 */
 // clang-format on
+#include <chrono>  // NOLINT
 #include <iostream>
+#include <string>
 
 #include "sherpa-onnx/c-api/cxx-api.h"
 

sherpa-onnx/c-api/cxx-api.cc

@@ -513,4 +513,49 @@ void KeywordSpotter::Reset(const OnlineStream *s) const {
   SherpaOnnxResetKeywordStream(p_, s->Get());
 }
 
+// ============================================================
+// For Offline Speech Enhancement
+// ============================================================
+
+OfflineSpeechDenoiser OfflineSpeechDenoiser::Create(
+    const OfflineSpeechDenoiserConfig &config) {
+  struct SherpaOnnxOfflineSpeechDenoiserConfig c;
+  memset(&c, 0, sizeof(c));
+
+  c.model.gtcrn.model = config.model.gtcrn.model.c_str();
+
+  c.model.num_threads = config.model.num_threads;
+  c.model.provider = config.model.provider.c_str();
+  c.model.debug = config.model.debug;
+
+  auto p = SherpaOnnxCreateOfflineSpeechDenoiser(&c);
+
+  return OfflineSpeechDenoiser(p);
+}
+
+void OfflineSpeechDenoiser::Destroy(
+    const SherpaOnnxOfflineSpeechDenoiser *p) const {
+  SherpaOnnxDestroyOfflineSpeechDenoiser(p);
+}
+
+OfflineSpeechDenoiser::OfflineSpeechDenoiser(
+    const SherpaOnnxOfflineSpeechDenoiser *p)
+    : MoveOnly<OfflineSpeechDenoiser, SherpaOnnxOfflineSpeechDenoiser>(p) {}
+
+DenoisedAudio OfflineSpeechDenoiser::Run(const float *samples, int32_t n,
+                                         int32_t sample_rate) const {
+  auto audio = SherpaOnnxOfflineSpeechDenoiserRun(p_, samples, n, sample_rate);
+
+  DenoisedAudio ans;
+  ans.samples = {audio->samples, audio->samples + audio->n};
+  ans.sample_rate = audio->sample_rate;
+  SherpaOnnxDestroyDenoisedAudio(audio);
+
+  return ans;
+}
+
+int32_t OfflineSpeechDenoiser::GetSampleRate() const {
+  return SherpaOnnxOfflineSpeechDenoiserGetSampleRate(p_);
+}
+
 }  // namespace sherpa_onnx::cxx

sherpa-onnx/c-api/cxx-api.h

@@ -464,6 +464,42 @@ class SHERPA_ONNX_API KeywordSpotter
   explicit KeywordSpotter(const SherpaOnnxKeywordSpotter *p);
 };
 
+struct OfflineSpeechDenoiserGtcrnModelConfig {
+  std::string model;
+};
+
+struct OfflineSpeechDenoiserModelConfig {
+  OfflineSpeechDenoiserGtcrnModelConfig gtcrn;
+  int32_t num_threads = 1;
+  int32_t debug = false;
+  std::string provider = "cpu";
+};
+
+struct OfflineSpeechDenoiserConfig {
+  OfflineSpeechDenoiserModelConfig model;
+};
+
+struct DenoisedAudio {
+  std::vector<float> samples;  // in the range [-1, 1]
+  int32_t sample_rate;
+};
+
+class SHERPA_ONNX_API OfflineSpeechDenoiser
+    : public MoveOnly<OfflineSpeechDenoiser, SherpaOnnxOfflineSpeechDenoiser> {
+ public:
+  static OfflineSpeechDenoiser Create(
+      const OfflineSpeechDenoiserConfig &config);
+
+  void Destroy(const SherpaOnnxOfflineSpeechDenoiser *p) const;
+
+  DenoisedAudio Run(const float *samples, int32_t n, int32_t sample_rate) const;
+
+  int32_t GetSampleRate() const;
+
+ private:
+  explicit OfflineSpeechDenoiser(const SherpaOnnxOfflineSpeechDenoiser *p);
+};
+
 }  // namespace sherpa_onnx::cxx
 
 #endif  // SHERPA_ONNX_C_API_CXX_API_H_