mirror of
https://github.com/pineappleEA/pineapple-src.git
synced 2024-12-05 10:58:25 -05:00
160 lines
4.4 KiB
C
Executable File
160 lines
4.4 KiB
C
Executable File
/*
|
|
* Copyright (c) 2014 Luca Barbato <lu_zero@gentoo.org>
|
|
* Copyright (c) 2014 Michael Niedermayer <michaelni@gmx.at>
|
|
*
|
|
* This file is part of FFmpeg.
|
|
*
|
|
* FFmpeg is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* FFmpeg is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with FFmpeg; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
#include "swresample_internal.h"
|
|
#include "libavutil/frame.h"
|
|
#include "libavutil/opt.h"
|
|
|
|
int swr_config_frame(SwrContext *s, const AVFrame *out, const AVFrame *in)
|
|
{
|
|
swr_close(s);
|
|
|
|
if (in) {
|
|
if (av_opt_set_int(s, "icl", in->channel_layout, 0) < 0)
|
|
goto fail;
|
|
if (av_opt_set_int(s, "isf", in->format, 0) < 0)
|
|
goto fail;
|
|
if (av_opt_set_int(s, "isr", in->sample_rate, 0) < 0)
|
|
goto fail;
|
|
}
|
|
|
|
if (out) {
|
|
if (av_opt_set_int(s, "ocl", out->channel_layout, 0) < 0)
|
|
goto fail;
|
|
if (av_opt_set_int(s, "osf", out->format, 0) < 0)
|
|
goto fail;
|
|
if (av_opt_set_int(s, "osr", out->sample_rate, 0) < 0)
|
|
goto fail;
|
|
}
|
|
|
|
return 0;
|
|
fail:
|
|
av_log(s, AV_LOG_ERROR, "Failed to set option\n");
|
|
return AVERROR(EINVAL);
|
|
}
|
|
|
|
static int config_changed(SwrContext *s,
|
|
const AVFrame *out, const AVFrame *in)
|
|
{
|
|
int ret = 0;
|
|
|
|
if (in) {
|
|
if (s->in_ch_layout != in->channel_layout ||
|
|
s->in_sample_rate != in->sample_rate ||
|
|
s->in_sample_fmt != in->format) {
|
|
ret |= AVERROR_INPUT_CHANGED;
|
|
}
|
|
}
|
|
|
|
if (out) {
|
|
if (s->out_ch_layout != out->channel_layout ||
|
|
s->out_sample_rate != out->sample_rate ||
|
|
s->out_sample_fmt != out->format) {
|
|
ret |= AVERROR_OUTPUT_CHANGED;
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
static inline int convert_frame(SwrContext *s,
|
|
AVFrame *out, const AVFrame *in)
|
|
{
|
|
int ret;
|
|
uint8_t **out_data = NULL;
|
|
const uint8_t **in_data = NULL;
|
|
int out_nb_samples = 0, in_nb_samples = 0;
|
|
|
|
if (out) {
|
|
out_data = out->extended_data;
|
|
out_nb_samples = out->nb_samples;
|
|
}
|
|
|
|
if (in) {
|
|
in_data = (const uint8_t **)in->extended_data;
|
|
in_nb_samples = in->nb_samples;
|
|
}
|
|
|
|
ret = swr_convert(s, out_data, out_nb_samples, in_data, in_nb_samples);
|
|
|
|
if (ret < 0) {
|
|
if (out)
|
|
out->nb_samples = 0;
|
|
return ret;
|
|
}
|
|
|
|
if (out)
|
|
out->nb_samples = ret;
|
|
|
|
return 0;
|
|
}
|
|
|
|
static inline int available_samples(AVFrame *out)
|
|
{
|
|
int bytes_per_sample = av_get_bytes_per_sample(out->format);
|
|
int samples = out->linesize[0] / bytes_per_sample;
|
|
|
|
if (av_sample_fmt_is_planar(out->format)) {
|
|
return samples;
|
|
} else {
|
|
int channels = av_get_channel_layout_nb_channels(out->channel_layout);
|
|
return samples / channels;
|
|
}
|
|
}
|
|
|
|
int swr_convert_frame(SwrContext *s,
|
|
AVFrame *out, const AVFrame *in)
|
|
{
|
|
int ret, setup = 0;
|
|
|
|
if (!swr_is_initialized(s)) {
|
|
if ((ret = swr_config_frame(s, out, in)) < 0)
|
|
return ret;
|
|
if ((ret = swr_init(s)) < 0)
|
|
return ret;
|
|
setup = 1;
|
|
} else {
|
|
// return as is or reconfigure for input changes?
|
|
if ((ret = config_changed(s, out, in)))
|
|
return ret;
|
|
}
|
|
|
|
if (out) {
|
|
if (!out->linesize[0]) {
|
|
out->nb_samples = swr_get_delay(s, s->out_sample_rate) + 3;
|
|
if (in) {
|
|
out->nb_samples += in->nb_samples*(int64_t)s->out_sample_rate / s->in_sample_rate;
|
|
}
|
|
if ((ret = av_frame_get_buffer(out, 0)) < 0) {
|
|
if (setup)
|
|
swr_close(s);
|
|
return ret;
|
|
}
|
|
} else {
|
|
if (!out->nb_samples)
|
|
out->nb_samples = available_samples(out);
|
|
}
|
|
}
|
|
|
|
return convert_frame(s, out, in);
|
|
}
|
|
|