blob: 6f9020b9fe0e29acb3fa4d1fe46e5a6242b0b8fd
1 | /* |
2 | * Copyright 2007 Bobby Bingham |
3 | * Copyright Stefano Sabatini <stefasab gmail com> |
4 | * Copyright Vitor Sessak <vitor1001 gmail com> |
5 | * |
6 | * This file is part of FFmpeg. |
7 | * |
8 | * FFmpeg is free software; you can redistribute it and/or |
9 | * modify it under the terms of the GNU Lesser General Public |
10 | * License as published by the Free Software Foundation; either |
11 | * version 2.1 of the License, or (at your option) any later version. |
12 | * |
13 | * FFmpeg is distributed in the hope that it will be useful, |
14 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
15 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
16 | * Lesser General Public License for more details. |
17 | * |
18 | * You should have received a copy of the GNU Lesser General Public |
19 | * License along with FFmpeg; if not, write to the Free Software |
20 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
21 | */ |
22 | |
23 | #include <string.h> |
24 | #include <stdio.h> |
25 | |
26 | #include "libavutil/avassert.h" |
27 | #include "libavutil/buffer.h" |
28 | #include "libavutil/hwcontext.h" |
29 | #include "libavutil/imgutils.h" |
30 | #include "libavutil/mem.h" |
31 | |
32 | #include "avfilter.h" |
33 | #include "internal.h" |
34 | #include "video.h" |
35 | |
36 | #define BUFFER_ALIGN 32 |
37 | |
38 | |
39 | AVFrame *ff_null_get_video_buffer(AVFilterLink *link, int w, int h) |
40 | { |
41 | return ff_get_video_buffer(link->dst->outputs[0], w, h); |
42 | } |
43 | |
44 | AVFrame *ff_default_get_video_buffer(AVFilterLink *link, int w, int h) |
45 | { |
46 | int pool_width = 0; |
47 | int pool_height = 0; |
48 | int pool_align = 0; |
49 | enum AVPixelFormat pool_format = AV_PIX_FMT_NONE; |
50 | |
51 | if (link->hw_frames_ctx && |
52 | ((AVHWFramesContext*)link->hw_frames_ctx->data)->format == link->format) { |
53 | int ret; |
54 | AVFrame *frame = av_frame_alloc(); |
55 | |
56 | if (!frame) |
57 | return NULL; |
58 | |
59 | ret = av_hwframe_get_buffer(link->hw_frames_ctx, frame, 0); |
60 | if (ret < 0) |
61 | av_frame_free(&frame); |
62 | |
63 | return frame; |
64 | } |
65 | |
66 | if (!link->frame_pool) { |
67 | link->frame_pool = ff_frame_pool_video_init(av_buffer_allocz, w, h, |
68 | link->format, BUFFER_ALIGN); |
69 | if (!link->frame_pool) |
70 | return NULL; |
71 | } else { |
72 | if (ff_frame_pool_get_video_config(link->frame_pool, |
73 | &pool_width, &pool_height, |
74 | &pool_format, &pool_align) < 0) { |
75 | return NULL; |
76 | } |
77 | |
78 | if (pool_width != w || pool_height != h || |
79 | pool_format != link->format || pool_align != BUFFER_ALIGN) { |
80 | |
81 | ff_frame_pool_uninit((FFFramePool **)&link->frame_pool); |
82 | link->frame_pool = ff_frame_pool_video_init(av_buffer_allocz, w, h, |
83 | link->format, BUFFER_ALIGN); |
84 | if (!link->frame_pool) |
85 | return NULL; |
86 | } |
87 | } |
88 | |
89 | return ff_frame_pool_get(link->frame_pool); |
90 | } |
91 | |
92 | AVFrame *ff_get_video_buffer(AVFilterLink *link, int w, int h) |
93 | { |
94 | AVFrame *ret = NULL; |
95 | |
96 | FF_TPRINTF_START(NULL, get_video_buffer); ff_tlog_link(NULL, link, 0); |
97 | |
98 | if (link->dstpad->get_video_buffer) |
99 | ret = link->dstpad->get_video_buffer(link, w, h); |
100 | |
101 | if (!ret) |
102 | ret = ff_default_get_video_buffer(link, w, h); |
103 | |
104 | return ret; |
105 | } |
106 |