minor fixes

sherpa-onnx/CMakeLists.txt

@@ -0,0 +1 @@
+add_subdirectory(csrc)

sherpa-onnx/csrc/main.cpp

@@ -12,9 +12,14 @@
 
 
 int main(int argc, char* argv[]) {
-    char* filename = argv[1];
-    std::string search_method = argv[2];
-    int num_active_paths = atoi(argv[3]);
+    char* encoder_path = argv[1];
+    char* decoder_path = argv[2];
+    char* joiner_path = argv[3];
+    char* joiner_encoder_proj_path = argv[4];
+    char* joiner_decoder_proj_path = argv[5];
+    char* token_path = argv[6];
+    std::string search_method = argv[7];
+    char* filename = argv[8];
 
     // General parameters
     int numberOfThreads = 16;
@@ -44,17 +49,16 @@ int main(int argc, char* argv[]) {
 
     // Define model
     auto model = get_model(
-        "/mnt/local4/sr/k2_sherpa/models/exp_en2/encoder_simp.onnx",
-        "/mnt/local4/sr/k2_sherpa/models/exp_en2/decoder_simp.onnx",
-        "/mnt/local4/sr/k2_sherpa/models/exp_en2/joiner_simp.onnx",
-        "/mnt/local4/sr/k2_sherpa/models/exp_en2/joiner_encoder_proj_simp.onnx",
-        "/mnt/local4/sr/k2_sherpa/models/exp_en2/joiner_decoder_proj_simp.onnx",
-        "/mnt/local4/sr/k2_sherpa/models/exp_en2/enUS_tokens.txt"
+        encoder_path,
+        decoder_path,
+        joiner_path,
+        joiner_encoder_proj_path,
+        joiner_decoder_proj_path,
+        token_path
     );
     
     std::vector<std::string> filename_list {
-        "/mnt/local4/sr/k2_sherpa/test_wavs/cnn_15sec.wav",
-        //"/mnt/local4/sr/k2_sherpa/test_wavs/1089-134686-0001.wav"
+        filename
     };
 
     for (auto filename : filename_list){

sherpa-onnx/csrc/online-fbank-test.cc

@@ -0,0 +1,48 @@
+/**
+ * Copyright (c)  2022  Xiaomi Corporation (authors: Fangjun Kuang)
+ *
+ * See LICENSE for clarification regarding multiple authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include <iostream>
+
+#include "kaldi-native-fbank/csrc/online-feature.h"
+
+int main() {
+  knf::FbankOptions opts;
+  opts.frame_opts.dither = 0;
+  opts.mel_opts.num_bins = 10;
+
+  knf::OnlineFbank fbank(opts);
+  for (int32_t i = 0; i < 1600; ++i) {
+    float s = (i * i - i / 2) / 32767.;
+    fbank.AcceptWaveform(16000, &s, 1);
+  }
+
+  std::ostringstream os;
+
+  int32_t n = fbank.NumFramesReady();
+  for (int32_t i = 0; i != n; ++i) {
+    const float *frame = fbank.GetFrame(i);
+    for (int32_t k = 0; k != opts.mel_opts.num_bins; ++k) {
+      os << frame[k] << ", ";
+    }
+    os << "\n";
+  }
+
+  std::cout << os.str() << "\n";
+
+  return 0;
+}