mirror of
https://gitlab.freedesktop.org/pipewire/pipewire.git
synced 2025-11-04 13:30:12 -05:00
module-echo-cancel: Change extension .cc to .cpp for aec-webrtc
Makes this consistent with the rest of the code base.
This commit is contained in:
parent
d95870d8d3
commit
d2d0375c39
2 changed files with 1 additions and 1 deletions
134
src/modules/module-echo-cancel/aec-webrtc.cpp
Normal file
134
src/modules/module-echo-cancel/aec-webrtc.cpp
Normal file
|
|
@ -0,0 +1,134 @@
|
|||
/* PipeWire
|
||||
*
|
||||
* Copyright © 2021 Wim Taymans <wim.taymans@gmail.com>
|
||||
* © 2021 Arun Raghavan <arun@asymptotic.io>
|
||||
*
|
||||
* Permission is hereby granted, free of charge, to any person obtaining a
|
||||
* copy of this software and associated documentation files (the "Software"),
|
||||
* to deal in the Software without restriction, including without limitation
|
||||
* the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
||||
* and/or sell copies of the Software, and to permit persons to whom the
|
||||
* Software is furnished to do so, subject to the following conditions:
|
||||
*
|
||||
* The above copyright notice and this permission notice (including the next
|
||||
* paragraph) shall be included in all copies or substantial portions of the
|
||||
* Software.
|
||||
*
|
||||
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
|
||||
* THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
||||
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
||||
* DEALINGS IN THE SOFTWARE.
|
||||
*/
|
||||
|
||||
#include "echo-cancel.h"
|
||||
|
||||
#include <pipewire/pipewire.h>
|
||||
|
||||
#include <webrtc/modules/audio_processing/include/audio_processing.h>
|
||||
#include <webrtc/modules/interface/module_common_types.h>
|
||||
#include <webrtc/system_wrappers/include/trace.h>
|
||||
|
||||
struct impl {
|
||||
webrtc::AudioProcessing *apm = NULL;
|
||||
spa_audio_info_raw info;
|
||||
};
|
||||
|
||||
static void *webrtc_create(const struct pw_properties *args, const spa_audio_info_raw *info)
|
||||
{
|
||||
struct impl *impl;
|
||||
webrtc::AudioProcessing *apm;
|
||||
webrtc::ProcessingConfig pconfig;
|
||||
webrtc::Config config;
|
||||
|
||||
apm = webrtc::AudioProcessing::Create(config);
|
||||
|
||||
pconfig = {{
|
||||
webrtc::StreamConfig(info->rate, info->channels, false), /* input stream */
|
||||
webrtc::StreamConfig(info->rate, info->channels, false), /* output stream */
|
||||
webrtc::StreamConfig(info->rate, info->channels, false), /* reverse input stream */
|
||||
webrtc::StreamConfig(info->rate, info->channels, false), /* reverse output stream */
|
||||
}};
|
||||
|
||||
if (apm->Initialize(pconfig) != webrtc::AudioProcessing::kNoError) {
|
||||
pw_log_error("Error initialising webrtc audio processing module");
|
||||
goto error;
|
||||
}
|
||||
|
||||
// TODO: wire up args to control these
|
||||
apm->high_pass_filter()->Enable(true);
|
||||
apm->echo_cancellation()->enable_drift_compensation(false);
|
||||
apm->echo_cancellation()->Enable(true);
|
||||
apm->noise_suppression()->set_level(webrtc::NoiseSuppression::kHigh);
|
||||
apm->noise_suppression()->Enable(true);
|
||||
apm->gain_control()->set_analog_level_limits(0, 255);
|
||||
// FIXME: can we hook up AGC?
|
||||
apm->gain_control()->set_mode(webrtc::GainControl::kAdaptiveDigital);
|
||||
apm->gain_control()->Enable(true);
|
||||
apm->voice_detection()->Enable(true);
|
||||
|
||||
impl = (struct impl *)calloc(1, sizeof(struct impl));
|
||||
impl->info = *info;
|
||||
|
||||
impl->apm = apm;
|
||||
|
||||
return impl;
|
||||
|
||||
error:
|
||||
if (apm)
|
||||
delete apm;
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static void webrtc_destroy(void *ec)
|
||||
{
|
||||
struct impl *impl = (struct impl*)ec;
|
||||
|
||||
delete impl->apm;
|
||||
free(impl);
|
||||
}
|
||||
|
||||
static int webrtc_run(void *ec, const float *rec[], const float *play[], float *out[], uint32_t n_samples)
|
||||
{
|
||||
struct impl *impl = (struct impl*)ec;
|
||||
webrtc::StreamConfig config =
|
||||
webrtc::StreamConfig(impl->info.rate, impl->info.channels, false);
|
||||
|
||||
if (n_samples * 1000 / impl->info.rate != 10) {
|
||||
pw_log_error("Buffers must be 10ms in length (currently %u samples)", n_samples);
|
||||
return -1;
|
||||
}
|
||||
|
||||
/* FIXME: ProcessReverseStream may change the playback buffer, in which
|
||||
* case we should use that, if we ever expose the intelligibility
|
||||
* enhancer */
|
||||
if (impl->apm->ProcessReverseStream(play, config, config, (float**)play) !=
|
||||
webrtc::AudioProcessing::kNoError) {
|
||||
pw_log_error("Processing reverse stream failed");
|
||||
}
|
||||
|
||||
impl->apm->set_stream_delay_ms(0);
|
||||
|
||||
if (impl->apm->ProcessStream(rec, config, config, out) !=
|
||||
webrtc::AudioProcessing::kNoError) {
|
||||
pw_log_error("Processing stream failed");
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static const struct echo_cancel_info echo_cancel_webrtc_impl = {
|
||||
.name = "webrtc",
|
||||
.info = SPA_DICT_INIT(NULL, 0),
|
||||
.latency = "480/48000",
|
||||
|
||||
.create = webrtc_create,
|
||||
.destroy = webrtc_destroy,
|
||||
|
||||
.run = webrtc_run,
|
||||
};
|
||||
|
||||
const struct echo_cancel_info *echo_cancel_webrtc = &echo_cancel_webrtc_impl;
|
||||
Loading…
Add table
Add a link
Reference in a new issue