91 lines
2.5 KiB
C
91 lines
2.5 KiB
C
// Copyright 2015-2019 Espressif Systems (Shanghai) PTE LTD
|
|
//
|
|
// 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
|
|
#ifndef _ESP_WEBRTC_H_
|
|
#define _ESP_WEBRTC_H_
|
|
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
#include <stdint.h>
|
|
#include "sr_ringbuf.h"
|
|
#include "esp_log.h"
|
|
#include "esp_agc.h"
|
|
#include "esp_ns.h"
|
|
|
|
#include "esp_heap_caps.h"
|
|
|
|
typedef struct {
|
|
void* ns_handle;
|
|
void* agc_handle;
|
|
int frame_size;
|
|
int sample_rate;
|
|
int16_t *buff;
|
|
int16_t *out_data;
|
|
sr_ringbuf_handle_t rb;
|
|
}webrtc_handle_t;
|
|
|
|
/**
|
|
* @brief Creates an instance of webrtc.
|
|
*
|
|
* @warning frame_length can supports be 10 ms, 20 ms, 30 ms, 32 ms.
|
|
*
|
|
* @param frame_length_ms The length of the audio processing
|
|
* @param ns_mode The mode of NS. -1 means NS is disabled. 0: Mild, 1: Medium, 2: Aggressive
|
|
* @param agc_mode The model of AGC
|
|
* @param agc_gain The gain of AGC. default is 9
|
|
* @param agc_target_level The target level of AGC. default is -3 dbfs
|
|
* @param sample_rate The sample rate of the audio.
|
|
*
|
|
* @return
|
|
* - NULL: Create failed
|
|
* - Others: The instance of webrtc
|
|
*/
|
|
webrtc_handle_t* webrtc_create(
|
|
int frame_length_ms,
|
|
int ns_mode,
|
|
agc_mode_t agc_mode,
|
|
int agc_gain,
|
|
int agc_target_level,
|
|
int sample_rate);
|
|
|
|
/**
|
|
* @brief Feed samples of an audio stream to the webrtc and get the audio stream after Noise suppression.
|
|
*
|
|
* @param handle The instance of NS.
|
|
* @param in_data An array of 16-bit signed audio samples.
|
|
* @param out_size The sample size of output data
|
|
* @param enable_ns Enable noise suppression
|
|
* @param enable_agc Enable automatic gain control
|
|
*
|
|
* @return data after noise suppression
|
|
*/
|
|
int16_t* webrtc_process(webrtc_handle_t *handle, int16_t *indata, int *size, bool enable_ns, bool enable_agc);
|
|
|
|
/**
|
|
* @brief Free the webrtc instance
|
|
*
|
|
* @param handle The instance of webrtc.
|
|
*
|
|
* @return None
|
|
*
|
|
*/
|
|
void webrtc_destroy(webrtc_handle_t *handle);
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif //_ESP_NS_H_
|