1 /*
2 * This file is part of FFmpeg.
3 *
4 * Copyright (c) 2015 Matthieu Bouron <matthieu.bouron stupeflix.com>
5 *
6 * FFmpeg is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * FFmpeg is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with FFmpeg; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21 #include "framepool.h"
22 #include "libavutil/avassert.h"
23 #include "libavutil/avutil.h"
24 #include "libavutil/buffer.h"
25 #include "libavutil/frame.h"
26 #include "libavutil/imgutils.h"
27 #include "libavutil/mem.h"
28 #include "libavutil/pixfmt.h"
29
30 struct FFFramePool {
31
32 enum AVMediaType type;
33
34 /* video */
35 int width;
36 int height;
37
38 /* audio */
39 int planes;
40 int channels;
41 int nb_samples;
42
43 /* common */
44 int format;
45 int align;
46 int linesize[4];
47 AVBufferPool *pools[4];
48
49 };
50
ff_frame_pool_video_init(AVBufferRef * (* alloc)(buffer_size_t size),int width,int height,enum AVPixelFormat format,int align)51 FFFramePool *ff_frame_pool_video_init(AVBufferRef* (*alloc)(buffer_size_t size),
52 int width,
53 int height,
54 enum AVPixelFormat format,
55 int align)
56 {
57 int i, ret;
58 FFFramePool *pool;
59 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(format);
60
61 if (!desc)
62 return NULL;
63
64 pool = av_mallocz(sizeof(FFFramePool));
65 if (!pool)
66 return NULL;
67
68 pool->type = AVMEDIA_TYPE_VIDEO;
69 pool->width = width;
70 pool->height = height;
71 pool->format = format;
72 pool->align = align;
73
74 if ((ret = av_image_check_size2(width, height, INT64_MAX, format, 0, NULL)) < 0) {
75 goto fail;
76 }
77
78 if (!pool->linesize[0]) {
79 for(i = 1; i <= align; i += i) {
80 ret = av_image_fill_linesizes(pool->linesize, pool->format,
81 FFALIGN(pool->width, i));
82 if (ret < 0) {
83 goto fail;
84 }
85 if (!(pool->linesize[0] & (pool->align - 1)))
86 break;
87 }
88
89 for (i = 0; i < 4 && pool->linesize[i]; i++) {
90 pool->linesize[i] = FFALIGN(pool->linesize[i], pool->align);
91 }
92 }
93
94 for (i = 0; i < 4 && pool->linesize[i]; i++) {
95 int h = FFALIGN(pool->height, 32);
96 if (i == 1 || i == 2)
97 h = AV_CEIL_RSHIFT(h, desc->log2_chroma_h);
98
99 pool->pools[i] = av_buffer_pool_init(pool->linesize[i] * h + 16 + 16 - 1,
100 alloc);
101 if (!pool->pools[i])
102 goto fail;
103 }
104
105 if (desc->flags & AV_PIX_FMT_FLAG_PAL ||
106 desc->flags & FF_PSEUDOPAL) {
107 pool->pools[1] = av_buffer_pool_init(AVPALETTE_SIZE, alloc);
108 if (!pool->pools[1])
109 goto fail;
110 }
111
112 return pool;
113
114 fail:
115 ff_frame_pool_uninit(&pool);
116 return NULL;
117 }
118
ff_frame_pool_audio_init(AVBufferRef * (* alloc)(buffer_size_t size),int channels,int nb_samples,enum AVSampleFormat format,int align)119 FFFramePool *ff_frame_pool_audio_init(AVBufferRef* (*alloc)(buffer_size_t size),
120 int channels,
121 int nb_samples,
122 enum AVSampleFormat format,
123 int align)
124 {
125 int ret, planar;
126 FFFramePool *pool;
127
128 pool = av_mallocz(sizeof(FFFramePool));
129 if (!pool)
130 return NULL;
131
132 planar = av_sample_fmt_is_planar(format);
133
134 pool->type = AVMEDIA_TYPE_AUDIO;
135 pool->planes = planar ? channels : 1;
136 pool->channels = channels;
137 pool->nb_samples = nb_samples;
138 pool->format = format;
139 pool->align = align;
140
141 ret = av_samples_get_buffer_size(&pool->linesize[0], channels,
142 nb_samples, format, 0);
143 if (ret < 0)
144 goto fail;
145
146 pool->pools[0] = av_buffer_pool_init(pool->linesize[0], NULL);
147 if (!pool->pools[0])
148 goto fail;
149
150 return pool;
151
152 fail:
153 ff_frame_pool_uninit(&pool);
154 return NULL;
155 }
156
ff_frame_pool_get_video_config(FFFramePool * pool,int * width,int * height,enum AVPixelFormat * format,int * align)157 int ff_frame_pool_get_video_config(FFFramePool *pool,
158 int *width,
159 int *height,
160 enum AVPixelFormat *format,
161 int *align)
162 {
163 if (!pool)
164 return AVERROR(EINVAL);
165
166 av_assert0(pool->type == AVMEDIA_TYPE_VIDEO);
167
168 *width = pool->width;
169 *height = pool->height;
170 *format = pool->format;
171 *align = pool->align;
172
173 return 0;
174 }
175
ff_frame_pool_get_audio_config(FFFramePool * pool,int * channels,int * nb_samples,enum AVSampleFormat * format,int * align)176 int ff_frame_pool_get_audio_config(FFFramePool *pool,
177 int *channels,
178 int *nb_samples,
179 enum AVSampleFormat *format,
180 int *align)
181 {
182 if (!pool)
183 return AVERROR(EINVAL);
184
185 av_assert0(pool->type == AVMEDIA_TYPE_AUDIO);
186
187 *channels = pool->channels;
188 *nb_samples = pool->nb_samples;
189 *format = pool->format;
190 *align = pool->align;
191
192 return 0;
193 }
194
ff_frame_pool_get(FFFramePool * pool)195 AVFrame *ff_frame_pool_get(FFFramePool *pool)
196 {
197 int i;
198 AVFrame *frame;
199 const AVPixFmtDescriptor *desc;
200
201 frame = av_frame_alloc();
202 if (!frame) {
203 return NULL;
204 }
205
206 switch(pool->type) {
207 case AVMEDIA_TYPE_VIDEO:
208 desc = av_pix_fmt_desc_get(pool->format);
209 if (!desc) {
210 goto fail;
211 }
212
213 frame->width = pool->width;
214 frame->height = pool->height;
215 frame->format = pool->format;
216
217 for (i = 0; i < 4; i++) {
218 frame->linesize[i] = pool->linesize[i];
219 if (!pool->pools[i])
220 break;
221
222 frame->buf[i] = av_buffer_pool_get(pool->pools[i]);
223 if (!frame->buf[i])
224 goto fail;
225
226 frame->data[i] = frame->buf[i]->data;
227 }
228
229 if (desc->flags & AV_PIX_FMT_FLAG_PAL ||
230 desc->flags & FF_PSEUDOPAL) {
231 enum AVPixelFormat format =
232 pool->format == AV_PIX_FMT_PAL8 ? AV_PIX_FMT_BGR8 : pool->format;
233
234 av_assert0(frame->data[1] != NULL);
235 if (avpriv_set_systematic_pal2((uint32_t *)frame->data[1], format) < 0)
236 goto fail;
237 }
238
239 frame->extended_data = frame->data;
240 break;
241 case AVMEDIA_TYPE_AUDIO:
242 frame->nb_samples = pool->nb_samples;
243 frame->channels = pool->channels;
244 frame->format = pool->format;
245 frame->linesize[0] = pool->linesize[0];
246
247 if (pool->planes > AV_NUM_DATA_POINTERS) {
248 frame->extended_data = av_mallocz_array(pool->planes,
249 sizeof(*frame->extended_data));
250 frame->nb_extended_buf = pool->planes - AV_NUM_DATA_POINTERS;
251 frame->extended_buf = av_mallocz_array(frame->nb_extended_buf,
252 sizeof(*frame->extended_buf));
253 if (!frame->extended_data || !frame->extended_buf)
254 goto fail;
255 } else {
256 frame->extended_data = frame->data;
257 av_assert0(frame->nb_extended_buf == 0);
258 }
259
260 for (i = 0; i < FFMIN(pool->planes, AV_NUM_DATA_POINTERS); i++) {
261 frame->buf[i] = av_buffer_pool_get(pool->pools[0]);
262 if (!frame->buf[i])
263 goto fail;
264 frame->extended_data[i] = frame->data[i] = frame->buf[i]->data;
265 }
266 for (i = 0; i < frame->nb_extended_buf; i++) {
267 frame->extended_buf[i] = av_buffer_pool_get(pool->pools[0]);
268 if (!frame->extended_buf[i])
269 goto fail;
270 frame->extended_data[i + AV_NUM_DATA_POINTERS] = frame->extended_buf[i]->data;
271 }
272
273 break;
274 default:
275 av_assert0(0);
276 }
277
278 return frame;
279 fail:
280 av_frame_free(&frame);
281 return NULL;
282 }
283
ff_frame_pool_uninit(FFFramePool ** pool)284 void ff_frame_pool_uninit(FFFramePool **pool)
285 {
286 int i;
287
288 if (!pool || !*pool)
289 return;
290
291 for (i = 0; i < 4; i++) {
292 av_buffer_pool_uninit(&(*pool)->pools[i]);
293 }
294
295 av_freep(pool);
296 }
297