rnnoise.h 2.31 KB
Newer Older
Gregor Richards's avatar
Gregor Richards committed
1
2
/* Copyright (c) 2018 Gregor Richards
 * Copyright (c) 2017 Mozilla */
Jean-Marc Valin's avatar
Jean-Marc Valin committed
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
/*
   Redistribution and use in source and binary forms, with or without
   modification, are permitted provided that the following conditions
   are met:

   - Redistributions of source code must retain the above copyright
   notice, this list of conditions and the following disclaimer.

   - Redistributions in binary form must reproduce the above copyright
   notice, this list of conditions and the following disclaimer in the
   documentation and/or other materials provided with the distribution.

   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
   ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
   A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR
   CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
   EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
   PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
   PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
   LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
   NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
   SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/

Gregor Richards's avatar
Gregor Richards committed
28
29
30
#ifndef RNNOISE_H
#define RNNOISE_H 1

Gregor Richards's avatar
Gregor Richards committed
31
32
#include <stdio.h>

33
34
35
#ifdef __cplusplus
extern "C" {
#endif
Gregor Richards's avatar
Gregor Richards committed
36

Jean-Marc Valin's avatar
Jean-Marc Valin committed
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
#ifndef RNNOISE_EXPORT
# if defined(WIN32)
#  if defined(RNNOISE_BUILD) && defined(DLL_EXPORT)
#   define RNNOISE_EXPORT __declspec(dllexport)
#  else
#   define RNNOISE_EXPORT
#  endif
# elif defined(__GNUC__) && defined(RNNOISE_BUILD)
#  define RNNOISE_EXPORT __attribute__ ((visibility ("default")))
# else
#  define RNNOISE_EXPORT
# endif
#endif

typedef struct DenoiseState DenoiseState;
Gregor Richards's avatar
Gregor Richards committed
52
typedef struct RNNModel RNNModel;
Jean-Marc Valin's avatar
Jean-Marc Valin committed
53
54
55

RNNOISE_EXPORT int rnnoise_get_size();

Gregor Richards's avatar
Gregor Richards committed
56
RNNOISE_EXPORT int rnnoise_init(DenoiseState *st, RNNModel *model);
Jean-Marc Valin's avatar
Jean-Marc Valin committed
57

Gregor Richards's avatar
Gregor Richards committed
58
RNNOISE_EXPORT DenoiseState *rnnoise_create(RNNModel *model);
Jean-Marc Valin's avatar
Jean-Marc Valin committed
59
60
61

RNNOISE_EXPORT void rnnoise_destroy(DenoiseState *st);

Jean-Marc Valin's avatar
Jean-Marc Valin committed
62
RNNOISE_EXPORT float rnnoise_process_frame(DenoiseState *st, float *out, const float *in);
Gregor Richards's avatar
Gregor Richards committed
63

Gregor Richards's avatar
Gregor Richards committed
64
65
66
67
RNNOISE_EXPORT RNNModel *rnnoise_model_from_file(FILE *f);

RNNOISE_EXPORT void rnnoise_model_free(RNNModel *model);

68
69
70
71
#ifdef __cplusplus
}
#endif

Gregor Richards's avatar
Gregor Richards committed
72
#endif