Merge pull request #12644 from mrcdk/ogg_loop

Added loop_begin and loop_end to OggVorbisStream
This commit is contained in:
Rémi Verschelde 2017-11-20 09:14:16 +01:00 committed by GitHub
commit b91fa19582
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 72 additions and 22 deletions

View file

@ -36,36 +36,54 @@
#include "thirdparty/misc/stb_vorbis.c"
#pragma GCC diagnostic pop
#ifndef CLAMP
#define CLAMP(m_a, m_min, m_max) (((m_a) < (m_min)) ? (m_min) : (((m_a) > (m_max)) ? m_max : m_a))
#endif
void AudioStreamPlaybackOGGVorbis::_mix_internal(AudioFrame *p_buffer, int p_frames) {
ERR_FAIL_COND(!active);
int todo = p_frames;
while (todo && active) {
int start_buffer = 0;
int mixed = stb_vorbis_get_samples_float_interleaved(ogg_stream, 2, (float *)p_buffer, todo * 2);
while (todo > 0 && active) {
float *buffer = (float *)p_buffer;
if (start_buffer > 0) {
buffer = (buffer + start_buffer * 2);
}
int mixed = stb_vorbis_get_samples_float_interleaved(ogg_stream, 2, buffer, todo * 2);
if (vorbis_stream->channels == 1 && mixed > 0) {
//mix mono to stereo
for (int i = 0; i < mixed; i++) {
for (int i = start_buffer; i < mixed; i++) {
p_buffer[i].r = p_buffer[i].l;
}
}
todo -= mixed;
frames_mixed += mixed;
if (todo) {
if (todo > 0) {
//end of file!
if (vorbis_stream->loop) {
//loop
seek(vorbis_stream->loop_offset);
//loop to the loop_beginning
seek(vorbis_stream->loop_begin);
loops++;
// we still have buffer to fill, start from this element in the next iteration.
start_buffer = p_frames - todo;
} else {
for (int i = mixed; i < p_frames; i++) {
for (int i = p_frames - todo; i < p_frames; i++) {
p_buffer[i] = AudioFrame(0, 0);
}
active = false;
todo = 0;
}
} else if (vorbis_stream->loop && frames_mixed >= vorbis_stream->loop_end_frames) {
// We reached loop_end. Loop to loop_begin plus whatever extra length we already mixed
uint32_t frames_to_advance = uint32_t(frames_mixed - vorbis_stream->loop_end_frames);
float start_loop = vorbis_stream->loop_begin + (float(frames_to_advance) / vorbis_stream->sample_rate);
seek(start_loop);
loops++;
}
}
}
@ -199,6 +217,9 @@ void AudioStreamOGGVorbis::set_data(const PoolVector<uint8_t> &p_data) {
//print_line("succeeded "+itos(ogg_alloc.alloc_buffer_length_in_bytes)+" setup "+itos(info.setup_memory_required)+" setup temp "+itos(info.setup_temp_memory_required)+" temp "+itos(info.temp_memory_required)+" maxframe"+itos(info.max_frame_size));
length = stb_vorbis_stream_length_in_seconds(ogg_stream);
if (loop_end == 0) {
set_loop_end(length);
}
stb_vorbis_close(ogg_stream);
data = AudioServer::get_singleton()->audio_data_alloc(src_data_len, src_datar.ptr());
@ -233,12 +254,24 @@ bool AudioStreamOGGVorbis::has_loop() const {
return loop;
}
void AudioStreamOGGVorbis::set_loop_offset(float p_seconds) {
loop_offset = p_seconds;
void AudioStreamOGGVorbis::set_loop_begin(float p_seconds) {
p_seconds = CLAMP(p_seconds, 0, length);
loop_begin = p_seconds;
loop_begin_frames = uint32_t(sample_rate * p_seconds);
}
float AudioStreamOGGVorbis::get_loop_offset() const {
return loop_offset;
float AudioStreamOGGVorbis::get_loop_begin() const {
return loop_begin;
}
void AudioStreamOGGVorbis::set_loop_end(float p_seconds) {
p_seconds = CLAMP(p_seconds, 0, length);
loop_end = p_seconds;
loop_end_frames = uint32_t(sample_rate * p_seconds);
}
float AudioStreamOGGVorbis::get_loop_end() const {
return loop_end;
}
void AudioStreamOGGVorbis::_bind_methods() {
@ -249,12 +282,16 @@ void AudioStreamOGGVorbis::_bind_methods() {
ClassDB::bind_method(D_METHOD("set_loop", "enable"), &AudioStreamOGGVorbis::set_loop);
ClassDB::bind_method(D_METHOD("has_loop"), &AudioStreamOGGVorbis::has_loop);
ClassDB::bind_method(D_METHOD("set_loop_offset", "seconds"), &AudioStreamOGGVorbis::set_loop_offset);
ClassDB::bind_method(D_METHOD("get_loop_offset"), &AudioStreamOGGVorbis::get_loop_offset);
ClassDB::bind_method(D_METHOD("set_loop_begin", "seconds"), &AudioStreamOGGVorbis::set_loop_begin);
ClassDB::bind_method(D_METHOD("get_loop_begin"), &AudioStreamOGGVorbis::get_loop_begin);
ClassDB::bind_method(D_METHOD("set_loop_end", "seconds"), &AudioStreamOGGVorbis::set_loop_end);
ClassDB::bind_method(D_METHOD("get_loop_end"), &AudioStreamOGGVorbis::get_loop_end);
ADD_PROPERTY(PropertyInfo(Variant::POOL_BYTE_ARRAY, "data", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NOEDITOR), "set_data", "get_data");
ADD_PROPERTY(PropertyInfo(Variant::BOOL, "loop", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NOEDITOR), "set_loop", "has_loop");
ADD_PROPERTY(PropertyInfo(Variant::REAL, "loop_offset", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NOEDITOR), "set_loop_offset", "get_loop_offset");
ADD_PROPERTY(PropertyInfo(Variant::REAL, "loop_begin", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NOEDITOR), "set_loop_begin", "get_loop_begin");
ADD_PROPERTY(PropertyInfo(Variant::REAL, "loop_end", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NOEDITOR), "set_loop_end", "get_loop_end");
}
AudioStreamOGGVorbis::AudioStreamOGGVorbis() {
@ -263,7 +300,10 @@ AudioStreamOGGVorbis::AudioStreamOGGVorbis() {
length = 0;
sample_rate = 1;
channels = 1;
loop_offset = 0;
loop_begin = 0;
loop_end = 0;
loop_begin_frames = 0;
loop_end_frames = 0;
decode_mem_size = 0;
loop = false;
loop = true;
}

View file

@ -92,7 +92,11 @@ class AudioStreamOGGVorbis : public AudioStream {
int channels;
float length;
bool loop;
float loop_offset;
float loop_begin;
float loop_end;
uint32_t loop_begin_frames;
uint32_t loop_end_frames;
protected:
static void _bind_methods();
@ -101,8 +105,11 @@ public:
void set_loop(bool p_enable);
bool has_loop() const;
void set_loop_offset(float p_seconds);
float get_loop_offset() const;
void set_loop_begin(float p_seconds);
float get_loop_begin() const;
void set_loop_end(float p_seconds);
float get_loop_end() const;
virtual Ref<AudioStreamPlayback> instance_playback();
virtual String get_stream_name() const;

View file

@ -72,13 +72,15 @@ String ResourceImporterOGGVorbis::get_preset_name(int p_idx) const {
void ResourceImporterOGGVorbis::get_import_options(List<ImportOption> *r_options, int p_preset) const {
r_options->push_back(ImportOption(PropertyInfo(Variant::BOOL, "loop"), true));
r_options->push_back(ImportOption(PropertyInfo(Variant::REAL, "loop_offset"), 0));
r_options->push_back(ImportOption(PropertyInfo(Variant::REAL, "loop_begin"), 0));
r_options->push_back(ImportOption(PropertyInfo(Variant::REAL, "loop_end"), 0));
}
Error ResourceImporterOGGVorbis::import(const String &p_source_file, const String &p_save_path, const Map<StringName, Variant> &p_options, List<String> *r_platform_variants, List<String> *r_gen_files) {
bool loop = p_options["loop"];
float loop_offset = p_options["loop_offset"];
float loop_begin = p_options["loop_begin"];
float loop_end = p_options["loop_end"];
FileAccess *f = FileAccess::open(p_source_file, FileAccess::READ);
if (!f) {
@ -100,7 +102,8 @@ Error ResourceImporterOGGVorbis::import(const String &p_source_file, const Strin
ogg_stream->set_data(data);
ogg_stream->set_loop(loop);
ogg_stream->set_loop_offset(loop_offset);
ogg_stream->set_loop_begin(loop_begin);
if (loop_end > 0) ogg_stream->set_loop_end(loop_end);
return ResourceSaver::save(p_save_path + ".oggstr", ogg_stream);
}