1
0
mirror of https://git.ffmpeg.org/ffmpeg.git synced 2026-10-24 02:09:03 +03:00

avutil/frame: let libavutil allocate a video frame's buffer with its own allocator

ff_frame_get_video_buffer() lays a video frame out like
av_frame_get_buffer() does, but takes the buffer from a callback, so
that the memory can come from a pool or from a device specific
allocator.

Assisted-by: Claude
This commit is contained in:
Michael Niedermayer
2026-10-04 02:50:51 +02:00
committed by michaelni
parent 8319541103
commit 501d815ec4
2 changed files with 43 additions and 3 deletions

View File

@@ -21,6 +21,7 @@
#include "buffer.h"
#include "dict.h"
#include "frame.h"
#include "frame_internal.h"
#include "imgutils.h"
#include "mem.h"
#include "refstruct.h"
@@ -72,7 +73,9 @@ void av_frame_free(AVFrame **frame)
#define ALIGN (HAVE_SIMD_ALIGN_64 ? 64 : 32)
static int get_video_buffer(AVFrame *frame, int align)
int ff_frame_get_video_buffer(AVFrame *frame, int align,
AVBufferRef *(*alloc)(void *opaque, size_t size),
void *opaque)
{
const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(frame->format);
int ret, padded_height;
@@ -119,7 +122,7 @@ static int get_video_buffer(AVFrame *frame, int align)
total_size += sizes[i];
}
frame->buf[0] = av_buffer_alloc(total_size);
frame->buf[0] = alloc ? alloc(opaque, total_size) : av_buffer_alloc(total_size);
if (!frame->buf[0]) {
ret = AVERROR(ENOMEM);
goto fail;
@@ -209,7 +212,7 @@ int av_frame_get_buffer(AVFrame *frame, int align)
return AVERROR(EINVAL);
if (frame->width > 0 && frame->height > 0)
return get_video_buffer(frame, align);
return ff_frame_get_video_buffer(frame, align, NULL, NULL);
else if (frame->nb_samples > 0 &&
(av_channel_layout_check(&frame->ch_layout)))
return get_audio_buffer(frame, align);

View File

@@ -0,0 +1,37 @@
/*
* 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
*/
#ifndef AVUTIL_FRAME_INTERNAL_H
#define AVUTIL_FRAME_INTERNAL_H
#include <stddef.h>
#include "buffer.h"
#include "frame.h"
/**
* Allocate the buffer of a video frame like av_frame_get_buffer().
*
* @param alloc allocates the buffer, av_buffer_alloc() is used if NULL
* @param opaque passed to alloc
*/
int ff_frame_get_video_buffer(AVFrame *frame, int align,
AVBufferRef *(*alloc)(void *opaque, size_t size),
void *opaque);
#endif /* AVUTIL_FRAME_INTERNAL_H */