aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorLuca Barbato <lu_zero@gentoo.org>2020-07-27 12:11:20 +0200
committerpetterreinholdtsen <pere-github@hungry.com>2021-01-22 21:17:01 +0100
commiteca42a4b9035a9fd66095183cc195a9f488c3e12 (patch)
treec9c3164ddeabee451f915afebff416e5cf5e0d5d
parent40a38162a43384d53e4cd81ae3f96fbaa723da55 (diff)
downloadrnnoise-eca42a4b9035a9fd66095183cc195a9f488c3e12.tar.gz
Add rnnoise_get_frame_size()
It provides a mean to know how many samples are processed at time.
-rw-r--r--include/rnnoise.h2
-rw-r--r--src/denoise.c4
2 files changed, 6 insertions, 0 deletions
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)