lavfi: Make default get_video_buffer work with hardware frames
[ffmpeg.git] / libavfilter / video.c
1 /*
2  * This file is part of Libav.
3  *
4  * Libav is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * Libav is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with Libav; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17  */
18
19 #include <string.h>
20 #include <stdio.h>
21
22 #include "libavutil/buffer.h"
23 #include "libavutil/hwcontext.h"
24 #include "libavutil/imgutils.h"
25 #include "libavutil/mem.h"
26
27 #include "avfilter.h"
28 #include "internal.h"
29 #include "video.h"
30
31 AVFrame *ff_null_get_video_buffer(AVFilterLink *link, int w, int h)
32 {
33     return ff_get_video_buffer(link->dst->outputs[0], w, h);
34 }
35
36 /* TODO: set the buffer's priv member to a context structure for the whole
37  * filter chain.  This will allow for a buffer pool instead of the constant
38  * alloc & free cycle currently implemented. */
39 AVFrame *ff_default_get_video_buffer(AVFilterLink *link, int w, int h)
40 {
41     AVFrame *frame = av_frame_alloc();
42     int ret;
43
44     if (!frame)
45         return NULL;
46
47     if (link->hw_frames_ctx &&
48         ((AVHWFramesContext*)link->hw_frames_ctx->data)->format == link->format) {
49         ret = av_hwframe_get_buffer(link->hw_frames_ctx, frame, 0);
50     } else {
51         frame->width  = w;
52         frame->height = h;
53         frame->format = link->format;
54
55         ret = av_frame_get_buffer(frame, 32);
56     }
57     if (ret < 0)
58         av_frame_free(&frame);
59
60     return frame;
61 }
62
63 AVFrame *ff_get_video_buffer(AVFilterLink *link, int w, int h)
64 {
65     AVFrame *ret = NULL;
66
67     FF_DPRINTF_START(NULL, get_video_buffer); ff_dlog_link(NULL, link, 0);
68
69     if (link->dstpad->get_video_buffer)
70         ret = link->dstpad->get_video_buffer(link, w, h);
71
72     if (!ret)
73         ret = ff_default_get_video_buffer(link, w, h);
74
75     return ret;
76 }