143 lines
5.9 KiB
C++
143 lines
5.9 KiB
C++
/*************************************************************************/
|
|
/* audio_effect_capture.cpp */
|
|
/*************************************************************************/
|
|
/* This file is part of: */
|
|
/* GODOT ENGINE */
|
|
/* https://godotengine.org */
|
|
/*************************************************************************/
|
|
/* Copyright (c) 2007-2022 Juan Linietsky, Ariel Manzur. */
|
|
/* Copyright (c) 2014-2022 Godot Engine contributors (cf. AUTHORS.md). */
|
|
/* */
|
|
/* 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 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 "audio_effect_capture.h"
|
|
|
|
bool AudioEffectCapture::can_get_buffer(int p_frames) const {
|
|
return buffer.data_left() >= p_frames;
|
|
}
|
|
|
|
PoolVector2Array AudioEffectCapture::get_buffer(int p_frames) {
|
|
ERR_FAIL_COND_V(!buffer_initialized, PoolVector2Array());
|
|
ERR_FAIL_INDEX_V(p_frames, buffer.size(), PoolVector2Array());
|
|
int data_left = buffer.data_left();
|
|
if (data_left < p_frames || p_frames == 0) {
|
|
return PoolVector2Array();
|
|
}
|
|
|
|
PoolVector2Array ret;
|
|
ret.resize(p_frames);
|
|
|
|
{
|
|
PoolVector<AudioFrame> streaming_data;
|
|
streaming_data.resize(p_frames);
|
|
buffer.read(streaming_data.write().ptr(), p_frames);
|
|
|
|
PoolVector2Array::Write retw = ret.write();
|
|
for (int32_t i = 0; i < p_frames; i++) {
|
|
retw[i] = Vector2(streaming_data[i].l, streaming_data[i].r);
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
void AudioEffectCapture::clear_buffer() {
|
|
const int32_t data_left = buffer.data_left();
|
|
buffer.advance_read(data_left);
|
|
}
|
|
|
|
void AudioEffectCapture::_bind_methods() {
|
|
ClassDB::bind_method(D_METHOD("can_get_buffer", "frames"), &AudioEffectCapture::can_get_buffer);
|
|
ClassDB::bind_method(D_METHOD("get_buffer", "frames"), &AudioEffectCapture::get_buffer);
|
|
ClassDB::bind_method(D_METHOD("clear_buffer"), &AudioEffectCapture::clear_buffer);
|
|
ClassDB::bind_method(D_METHOD("set_buffer_length", "buffer_length_seconds"), &AudioEffectCapture::set_buffer_length);
|
|
ClassDB::bind_method(D_METHOD("get_buffer_length"), &AudioEffectCapture::get_buffer_length);
|
|
ClassDB::bind_method(D_METHOD("get_frames_available"), &AudioEffectCapture::get_frames_available);
|
|
ClassDB::bind_method(D_METHOD("get_discarded_frames"), &AudioEffectCapture::get_discarded_frames);
|
|
ClassDB::bind_method(D_METHOD("get_buffer_length_frames"), &AudioEffectCapture::get_buffer_length_frames);
|
|
ClassDB::bind_method(D_METHOD("get_pushed_frames"), &AudioEffectCapture::get_pushed_frames);
|
|
|
|
ADD_PROPERTY(PropertyInfo(Variant::REAL, "buffer_length", PROPERTY_HINT_RANGE, "0.01,10,0.01"), "set_buffer_length", "get_buffer_length");
|
|
}
|
|
|
|
Ref<AudioEffectInstance> AudioEffectCapture::instance() {
|
|
if (!buffer_initialized) {
|
|
float target_buffer_size = AudioServer::get_singleton()->get_mix_rate() * buffer_length_seconds;
|
|
ERR_FAIL_COND_V(target_buffer_size <= 0 || target_buffer_size >= (1 << 27), Ref<AudioEffectInstance>());
|
|
buffer.resize(nearest_shift((int)target_buffer_size));
|
|
buffer_initialized = true;
|
|
}
|
|
|
|
clear_buffer();
|
|
|
|
Ref<AudioEffectCaptureInstance> ins;
|
|
ins.instance();
|
|
ins->base = Ref<AudioEffectCapture>(this);
|
|
|
|
return ins;
|
|
}
|
|
|
|
void AudioEffectCapture::set_buffer_length(float p_buffer_length_seconds) {
|
|
buffer_length_seconds = p_buffer_length_seconds;
|
|
}
|
|
|
|
float AudioEffectCapture::get_buffer_length() {
|
|
return buffer_length_seconds;
|
|
}
|
|
|
|
int AudioEffectCapture::get_frames_available() const {
|
|
ERR_FAIL_COND_V(!buffer_initialized, 0);
|
|
return buffer.data_left();
|
|
}
|
|
|
|
int64_t AudioEffectCapture::get_discarded_frames() const {
|
|
return discarded_frames;
|
|
}
|
|
|
|
int AudioEffectCapture::get_buffer_length_frames() const {
|
|
ERR_FAIL_COND_V(!buffer_initialized, 0);
|
|
return buffer.size();
|
|
}
|
|
|
|
int64_t AudioEffectCapture::get_pushed_frames() const {
|
|
return pushed_frames;
|
|
}
|
|
|
|
void AudioEffectCaptureInstance::process(const AudioFrame *p_src_frames, AudioFrame *p_dst_frames, int p_frame_count) {
|
|
RingBuffer<AudioFrame> &buffer = base->buffer;
|
|
|
|
for (int i = 0; i < p_frame_count; i++) {
|
|
p_dst_frames[i] = p_src_frames[i];
|
|
}
|
|
|
|
if (buffer.space_left() >= p_frame_count) {
|
|
// Add incoming audio frames to the IO ring buffer
|
|
int32_t ret = buffer.write(p_src_frames, p_frame_count);
|
|
ERR_FAIL_COND_MSG(ret != p_frame_count, "Failed to add data to effect capture ring buffer despite sufficient space.");
|
|
base->pushed_frames += p_frame_count;
|
|
} else {
|
|
base->discarded_frames += p_frame_count;
|
|
}
|
|
}
|
|
|
|
bool AudioEffectCaptureInstance::process_silence() const {
|
|
return true;
|
|
}
|