/* * Copyright 2007 Bobby Bingham * Copyright Stefano Sabatini <stefasab gmail com> * Copyright Vitor Sessak <vitor1001 gmail com> * * 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 <string.h> #include <stdio.h> #include "libavutil/buffer.h" #include "libavutil/cpu.h" #include "libavutil/hwcontext.h" #include "libavutil/imgutils.h" #include "avfilter.h" #include "framepool.h" #include "internal.h" #include "video.h" AVFrame *ff_null_get_video_buffer(AVFilterLink *link, int w, int h) { return ff_get_video_buffer(link->dst->outputs[0], w, h); } AVFrame *ff_default_get_video_buffer(AVFilterLink *link, int w, int h) { AVFrame *frame = NULL; int pool_width = 0; int pool_height = 0; int pool_align = 0; int align = av_cpu_max_align(); enum AVPixelFormat pool_format = AV_PIX_FMT_NONE; if (link->hw_frames_ctx && ((AVHWFramesContext*)link->hw_frames_ctx->data)->format == link->format) { int ret; AVFrame *frame = av_frame_alloc(); if (!frame) return NULL; ret = av_hwframe_get_buffer(link->hw_frames_ctx, frame, 0); if (ret < 0) av_frame_free(&frame); return frame; } if (!link->frame_pool) { link->frame_pool = ff_frame_pool_video_init(av_buffer_allocz, w, h, link->format, align); if (!link->frame_pool) return NULL; } else { if (ff_frame_pool_get_video_config(link->frame_pool, &pool_width, &pool_height, &pool_format, &pool_align) < 0) { return NULL; } if (pool_width != w || pool_height != h || pool_format != link->format || pool_align != align) { ff_frame_pool_uninit((FFFramePool **)&link->frame_pool); link->frame_pool = ff_frame_pool_video_init(av_buffer_allocz, w, h, link->format, align); if (!link->frame_pool) return NULL; } } frame = ff_frame_pool_get(link->frame_pool); if (!frame) return NULL; frame->sample_aspect_ratio = link->sample_aspect_ratio; return frame; } AVFrame *ff_get_video_buffer(AVFilterLink *link, int w, int h) { AVFrame *ret = NULL; FF_TPRINTF_START(NULL, get_video_buffer); ff_tlog_link(NULL, link, 0); if (link->dstpad->get_buffer.video) ret = link->dstpad->get_buffer.video(link, w, h); if (!ret) ret = ff_default_get_video_buffer(link, w, h); return ret; }