Add rnnoise_get_frame_size()

It provides a mean to know how many samples are processed at time.
desktop-app · Jan 22, 2021 · eca42a4 · eca42a4
1 parent 40a3816
commit eca42a4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/include/rnnoise.h b/include/rnnoise.h
@@ -53,6 +53,8 @@ typedef struct RNNModel RNNModel;
 
 RNNOISE_EXPORT int rnnoise_get_size();
 
+RNNOISE_EXPORT int rnnoise_get_frame_size();
+
 RNNOISE_EXPORT int rnnoise_init(DenoiseState *st, RNNModel *model);
 
 RNNOISE_EXPORT DenoiseState *rnnoise_create(RNNModel *model);

diff --git a/src/denoise.c b/src/denoise.c
@@ -257,6 +257,10 @@ int rnnoise_get_size() {
   return sizeof(DenoiseState);
 }
 
+int rnnoise_get_frame_size() {
+  return FRAME_SIZE;
+}
+
 int rnnoise_init(DenoiseState *st, RNNModel *model) {
   memset(st, 0, sizeof(*st));
   if (model)