yading@11
|
1 /*
|
yading@11
|
2 * Copyright 2007 Bobby Bingham
|
yading@11
|
3 * Copyright Stefano Sabatini <stefasab gmail com>
|
yading@11
|
4 * Copyright Vitor Sessak <vitor1001 gmail com>
|
yading@11
|
5 *
|
yading@11
|
6 * This file is part of FFmpeg.
|
yading@11
|
7 *
|
yading@11
|
8 * FFmpeg is free software; you can redistribute it and/or
|
yading@11
|
9 * modify it under the terms of the GNU Lesser General Public
|
yading@11
|
10 * License as published by the Free Software Foundation; either
|
yading@11
|
11 * version 2.1 of the License, or (at your option) any later version.
|
yading@11
|
12 *
|
yading@11
|
13 * FFmpeg is distributed in the hope that it will be useful,
|
yading@11
|
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
yading@11
|
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
yading@11
|
16 * Lesser General Public License for more details.
|
yading@11
|
17 *
|
yading@11
|
18 * You should have received a copy of the GNU Lesser General Public
|
yading@11
|
19 * License along with FFmpeg; if not, write to the Free Software
|
yading@11
|
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
yading@11
|
21 */
|
yading@11
|
22
|
yading@11
|
23 #include <string.h>
|
yading@11
|
24 #include <stdio.h>
|
yading@11
|
25
|
yading@11
|
26 #include "libavutil/avassert.h"
|
yading@11
|
27 #include "libavutil/buffer.h"
|
yading@11
|
28 #include "libavutil/imgutils.h"
|
yading@11
|
29 #include "libavutil/mem.h"
|
yading@11
|
30
|
yading@11
|
31 #include "avfilter.h"
|
yading@11
|
32 #include "internal.h"
|
yading@11
|
33 #include "video.h"
|
yading@11
|
34
|
yading@11
|
35 AVFrame *ff_null_get_video_buffer(AVFilterLink *link, int w, int h)
|
yading@11
|
36 {
|
yading@11
|
37 return ff_get_video_buffer(link->dst->outputs[0], w, h);
|
yading@11
|
38 }
|
yading@11
|
39
|
yading@11
|
40 /* TODO: set the buffer's priv member to a context structure for the whole
|
yading@11
|
41 * filter chain. This will allow for a buffer pool instead of the constant
|
yading@11
|
42 * alloc & free cycle currently implemented. */
|
yading@11
|
43 AVFrame *ff_default_get_video_buffer(AVFilterLink *link, int w, int h)
|
yading@11
|
44 {
|
yading@11
|
45 AVFrame *frame = av_frame_alloc();
|
yading@11
|
46 int ret;
|
yading@11
|
47
|
yading@11
|
48 #if 0 //POOL
|
yading@11
|
49 AVFilterPool *pool = link->pool;
|
yading@11
|
50 if (pool) {
|
yading@11
|
51 for (i = 0; i < POOL_SIZE; i++) {
|
yading@11
|
52 picref = pool->pic[i];
|
yading@11
|
53 if (picref && picref->buf->format == link->format && picref->buf->w == w && picref->buf->h == h) {
|
yading@11
|
54 AVFilterBuffer *pic = picref->buf;
|
yading@11
|
55 pool->pic[i] = NULL;
|
yading@11
|
56 pool->count--;
|
yading@11
|
57 av_assert0(!picref->video->qp_table);
|
yading@11
|
58 picref->video->w = w;
|
yading@11
|
59 picref->video->h = h;
|
yading@11
|
60 picref->perms = full_perms;
|
yading@11
|
61 picref->format = link->format;
|
yading@11
|
62 pic->refcount = 1;
|
yading@11
|
63 memcpy(picref->data, pic->data, sizeof(picref->data));
|
yading@11
|
64 memcpy(picref->linesize, pic->linesize, sizeof(picref->linesize));
|
yading@11
|
65 pool->refcount++;
|
yading@11
|
66 return picref;
|
yading@11
|
67 }
|
yading@11
|
68 }
|
yading@11
|
69 } else {
|
yading@11
|
70 pool = link->pool = av_mallocz(sizeof(AVFilterPool));
|
yading@11
|
71 pool->refcount = 1;
|
yading@11
|
72 }
|
yading@11
|
73 #endif
|
yading@11
|
74 if (!frame)
|
yading@11
|
75 return NULL;
|
yading@11
|
76
|
yading@11
|
77 frame->width = w;
|
yading@11
|
78 frame->height = h;
|
yading@11
|
79 frame->format = link->format;
|
yading@11
|
80
|
yading@11
|
81 ret = av_frame_get_buffer(frame, 32);
|
yading@11
|
82 if (ret < 0)
|
yading@11
|
83 av_frame_free(&frame);
|
yading@11
|
84
|
yading@11
|
85 #if 0 //POOL
|
yading@11
|
86 memset(data[0], 128, i);
|
yading@11
|
87
|
yading@11
|
88 picref->buf->priv = pool;
|
yading@11
|
89 picref->buf->free = NULL;
|
yading@11
|
90 pool->refcount++;
|
yading@11
|
91 #endif
|
yading@11
|
92
|
yading@11
|
93 return frame;
|
yading@11
|
94 }
|
yading@11
|
95
|
yading@11
|
96 #if FF_API_AVFILTERBUFFER
|
yading@11
|
97 AVFilterBufferRef *
|
yading@11
|
98 avfilter_get_video_buffer_ref_from_arrays(uint8_t * const data[4], const int linesize[4], int perms,
|
yading@11
|
99 int w, int h, enum AVPixelFormat format)
|
yading@11
|
100 {
|
yading@11
|
101 AVFilterBuffer *pic = av_mallocz(sizeof(AVFilterBuffer));
|
yading@11
|
102 AVFilterBufferRef *picref = av_mallocz(sizeof(AVFilterBufferRef));
|
yading@11
|
103
|
yading@11
|
104 if (!pic || !picref)
|
yading@11
|
105 goto fail;
|
yading@11
|
106
|
yading@11
|
107 picref->buf = pic;
|
yading@11
|
108 picref->buf->free = ff_avfilter_default_free_buffer;
|
yading@11
|
109 if (!(picref->video = av_mallocz(sizeof(AVFilterBufferRefVideoProps))))
|
yading@11
|
110 goto fail;
|
yading@11
|
111
|
yading@11
|
112 pic->w = picref->video->w = w;
|
yading@11
|
113 pic->h = picref->video->h = h;
|
yading@11
|
114
|
yading@11
|
115 /* make sure the buffer gets read permission or it's useless for output */
|
yading@11
|
116 picref->perms = perms | AV_PERM_READ;
|
yading@11
|
117
|
yading@11
|
118 pic->refcount = 1;
|
yading@11
|
119 picref->type = AVMEDIA_TYPE_VIDEO;
|
yading@11
|
120 pic->format = picref->format = format;
|
yading@11
|
121
|
yading@11
|
122 memcpy(pic->data, data, 4*sizeof(data[0]));
|
yading@11
|
123 memcpy(pic->linesize, linesize, 4*sizeof(linesize[0]));
|
yading@11
|
124 memcpy(picref->data, pic->data, sizeof(picref->data));
|
yading@11
|
125 memcpy(picref->linesize, pic->linesize, sizeof(picref->linesize));
|
yading@11
|
126
|
yading@11
|
127 pic-> extended_data = pic->data;
|
yading@11
|
128 picref->extended_data = picref->data;
|
yading@11
|
129
|
yading@11
|
130 picref->pts = AV_NOPTS_VALUE;
|
yading@11
|
131
|
yading@11
|
132 return picref;
|
yading@11
|
133
|
yading@11
|
134 fail:
|
yading@11
|
135 if (picref && picref->video)
|
yading@11
|
136 av_free(picref->video);
|
yading@11
|
137 av_free(picref);
|
yading@11
|
138 av_free(pic);
|
yading@11
|
139 return NULL;
|
yading@11
|
140 }
|
yading@11
|
141 #endif
|
yading@11
|
142
|
yading@11
|
143 AVFrame *ff_get_video_buffer(AVFilterLink *link, int w, int h)
|
yading@11
|
144 {
|
yading@11
|
145 AVFrame *ret = NULL;
|
yading@11
|
146
|
yading@11
|
147 av_unused char buf[16];
|
yading@11
|
148 FF_TPRINTF_START(NULL, get_video_buffer); ff_tlog_link(NULL, link, 0);
|
yading@11
|
149
|
yading@11
|
150 if (link->dstpad->get_video_buffer)
|
yading@11
|
151 ret = link->dstpad->get_video_buffer(link, w, h);
|
yading@11
|
152
|
yading@11
|
153 if (!ret)
|
yading@11
|
154 ret = ff_default_get_video_buffer(link, w, h);
|
yading@11
|
155
|
yading@11
|
156 return ret;
|
yading@11
|
157 }
|