C++ API for speaker diarization (#1396)

sherpa-onnx/csrc/offline-speaker-segmentation-pyannote-model-config.cc

@@ -0,0 +1,38 @@
+// sherpa-onnx/csrc/offline-speaker-segmentation-pyannote-model-config.cc
+//
+// Copyright (c)  2024  Xiaomi Corporation
+#include "sherpa-onnx/csrc/offline-speaker-segmentation-pyannote-model-config.h"
+
+#include <sstream>
+#include <string>
+
+#include "sherpa-onnx/csrc/file-utils.h"
+#include "sherpa-onnx/csrc/macros.h"
+
+namespace sherpa_onnx {
+
+void OfflineSpeakerSegmentationPyannoteModelConfig::Register(ParseOptions *po) {
+  po->Register("pyannote-model", &model,
+               "Path to model.onnx of the Pyannote segmentation model.");
+}
+
+bool OfflineSpeakerSegmentationPyannoteModelConfig::Validate() const {
+  if (!FileExists(model)) {
+    SHERPA_ONNX_LOGE("Pyannote segmentation model: '%s' does not exist",
+                     model.c_str());
+    return false;
+  }
+
+  return true;
+}
+
+std::string OfflineSpeakerSegmentationPyannoteModelConfig::ToString() const {
+  std::ostringstream os;
+
+  os << "OfflineSpeakerSegmentationPyannoteModelConfig(";
+  os << "model=\"" << model << "\")";
+
+  return os.str();
+}
+
+}  // namespace sherpa_onnx