Add online stream. (#28)

sherpa-onnx/csrc/online-stream.cc

@@ -0,0 +1,89 @@
+// sherpa-onnx/csrc/online-stream.cc
+//
+// Copyright (c)  2023  Xiaomi Corporation
+#include "sherpa-onnx/csrc/online-stream.h"
+
+#include <memory>
+#include <vector>
+
+#include "sherpa-onnx/csrc/features.h"
+
+namespace sherpa_onnx {
+
+class OnlineStream::Impl {
+ public:
+  explicit Impl(const FeatureExtractorConfig &config)
+      : feat_extractor_(config) {}
+
+  void AcceptWaveform(float sampling_rate, const float *waveform, int32_t n) {
+    feat_extractor_.AcceptWaveform(sampling_rate, waveform, n);
+  }
+
+  void InputFinished() { feat_extractor_.InputFinished(); }
+
+  int32_t NumFramesReady() const { return feat_extractor_.NumFramesReady(); }
+
+  bool IsLastFrame(int32_t frame) const {
+    return feat_extractor_.IsLastFrame(frame);
+  }
+
+  std::vector<float> GetFrames(int32_t frame_index, int32_t n) const {
+    return feat_extractor_.GetFrames(frame_index, n);
+  }
+
+  void Reset() { feat_extractor_.Reset(); }
+
+  int32_t &GetNumProcessedFrames() { return num_processed_frames_; }
+
+  void SetResult(const OnlineTransducerDecoderResult &r) { result_ = r; }
+
+  const OnlineTransducerDecoderResult &GetResult() const { return result_; }
+
+  int32_t FeatureDim() const { return feat_extractor_.FeatureDim(); }
+
+ private:
+  FeatureExtractor feat_extractor_;
+  int32_t num_processed_frames_ = 0;  // before subsampling
+  OnlineTransducerDecoderResult result_;
+};
+
+OnlineStream::OnlineStream(const FeatureExtractorConfig &config /*= {}*/)
+    : impl_(std::make_unique<Impl>(config)) {}
+
+OnlineStream::~OnlineStream() = default;
+
+void OnlineStream::AcceptWaveform(float sampling_rate, const float *waveform,
+                                  int32_t n) {
+  impl_->AcceptWaveform(sampling_rate, waveform, n);
+}
+
+void OnlineStream::InputFinished() { impl_->InputFinished(); }
+
+int32_t OnlineStream::NumFramesReady() const { return impl_->NumFramesReady(); }
+
+bool OnlineStream::IsLastFrame(int32_t frame) const {
+  return impl_->IsLastFrame(frame);
+}
+
+std::vector<float> OnlineStream::GetFrames(int32_t frame_index,
+                                           int32_t n) const {
+  return impl_->GetFrames(frame_index, n);
+}
+
+void OnlineStream::Reset() { impl_->Reset(); }
+
+int32_t OnlineStream::FeatureDim() const { return impl_->FeatureDim(); }
+
+int32_t &OnlineStream::GetNumProcessedFrames() {
+  return impl_->GetNumProcessedFrames();
+}
+
+void OnlineStream::SetResult(const OnlineTransducerDecoderResult &r) {
+  impl_->SetResult(r);
+}
+
+const OnlineTransducerDecoderResult &OnlineStream::GetResult() const {
+  return impl_->GetResult();
+}
+
+}  // namespace sherpa_onnx