decoder/ffmpeg: move code to class FfmpegBuffer

This commit is contained in:
Max Kellermann
2014-12-18 21:28:50 +01:00
parent 2dfe2fb22c
commit a142f93dca
3 changed files with 84 additions and 17 deletions

View File

@@ -24,6 +24,7 @@
#include "FfmpegDecoderPlugin.hxx"
#include "lib/ffmpeg/Domain.hxx"
#include "lib/ffmpeg/LogError.hxx"
#include "lib/ffmpeg/Buffer.hxx"
#include "../DecoderAPI.hxx"
#include "FfmpegMetaData.hxx"
#include "tag/TagBuilder.hxx"
@@ -275,7 +276,7 @@ static int
copy_interleave_frame(const AVCodecContext *codec_context,
const AVFrame *frame,
uint8_t **output_buffer,
uint8_t **global_buffer, int *global_buffer_size)
FfmpegBuffer &global_buffer)
{
int plane_size;
const int data_size =
@@ -288,17 +289,11 @@ copy_interleave_frame(const AVCodecContext *codec_context,
if (av_sample_fmt_is_planar(codec_context->sample_fmt) &&
codec_context->channels > 1) {
if(*global_buffer_size < data_size) {
av_freep(global_buffer);
*output_buffer = global_buffer.GetT<uint8_t>(data_size);
if (*output_buffer == nullptr)
/* Not enough memory - shouldn't happen */
return AVERROR(ENOMEM);
*global_buffer = (uint8_t*)av_malloc(data_size);
if (!*global_buffer)
/* Not enough memory - shouldn't happen */
return AVERROR(ENOMEM);
*global_buffer_size = data_size;
}
*output_buffer = *global_buffer;
copy_interleave_frame2(*output_buffer, frame->extended_data,
frame->nb_samples,
codec_context->channels,
@@ -316,7 +311,7 @@ ffmpeg_send_packet(Decoder &decoder, InputStream &is,
AVCodecContext *codec_context,
const AVStream *stream,
AVFrame *frame,
uint8_t **buffer, int *buffer_size)
FfmpegBuffer &buffer)
{
if (packet->pts >= 0 && packet->pts != (int64_t)AV_NOPTS_VALUE) {
auto start = start_time_fallback(*stream);
@@ -346,7 +341,7 @@ ffmpeg_send_packet(Decoder &decoder, InputStream &is,
audio_size = copy_interleave_frame(codec_context,
frame,
&output_buffer,
buffer, buffer_size);
buffer);
if (audio_size < 0) {
/* this must be a serious error,
e.g. OOM */
@@ -667,8 +662,7 @@ ffmpeg_decode(Decoder &decoder, InputStream &input)
return;
}
uint8_t *interleaved_buffer = nullptr;
int interleaved_buffer_size = 0;
FfmpegBuffer interleaved_buffer;
DecoderCommand cmd;
do {
@@ -686,7 +680,7 @@ ffmpeg_decode(Decoder &decoder, InputStream &input)
&packet, codec_context,
av_stream,
frame,
&interleaved_buffer, &interleaved_buffer_size);
interleaved_buffer);
else
cmd = decoder_get_command(decoder);
@@ -715,7 +709,6 @@ ffmpeg_decode(Decoder &decoder, InputStream &input)
#else
av_freep(&frame);
#endif
av_freep(&interleaved_buffer);
avcodec_close(codec_context);
avformat_close_input(&format_context);