From eca42a4b9035a9fd66095183cc195a9f488c3e12 Mon Sep 17 00:00:00 2001 From: Luca Barbato Date: Mon, 27 Jul 2020 12:11:20 +0200 Subject: Add rnnoise_get_frame_size() It provides a mean to know how many samples are processed at time. --- include/rnnoise.h | 2 ++ src/denoise.c | 4 ++++ 2 files changed, 6 insertions(+) diff --git a/include/rnnoise.h b/include/rnnoise.h index 511a976..f7f5007 100644 --- 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 index d1c21dc..5a62844 100644 --- 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) -- cgit v1.2.3