Line | Branch | Exec | Source |
---|---|---|---|
1 | /* | ||
2 | * The default get_buffer2() implementation | ||
3 | * | ||
4 | * This file is part of FFmpeg. | ||
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 <stdint.h> | ||
22 | |||
23 | #include "libavutil/avassert.h" | ||
24 | #include "libavutil/avutil.h" | ||
25 | #include "libavutil/buffer.h" | ||
26 | #include "libavutil/frame.h" | ||
27 | #include "libavutil/hwcontext.h" | ||
28 | #include "libavutil/imgutils.h" | ||
29 | #include "libavutil/mem.h" | ||
30 | #include "libavutil/samplefmt.h" | ||
31 | #include "libavutil/version.h" | ||
32 | |||
33 | #include "avcodec.h" | ||
34 | #include "internal.h" | ||
35 | |||
36 | typedef struct FramePool { | ||
37 | /** | ||
38 | * Pools for each data plane. For audio all the planes have the same size, | ||
39 | * so only pools[0] is used. | ||
40 | */ | ||
41 | AVBufferPool *pools[4]; | ||
42 | |||
43 | /* | ||
44 | * Pool parameters | ||
45 | */ | ||
46 | int format; | ||
47 | int width, height; | ||
48 | int stride_align[AV_NUM_DATA_POINTERS]; | ||
49 | int linesize[4]; | ||
50 | int planes; | ||
51 | int channels; | ||
52 | int samples; | ||
53 | } FramePool; | ||
54 | |||
55 | 14949 | static void frame_pool_free(void *opaque, uint8_t *data) | |
56 | { | ||
57 | 14949 | FramePool *pool = (FramePool*)data; | |
58 | int i; | ||
59 | |||
60 |
2/2✓ Branch 0 taken 59796 times.
✓ Branch 1 taken 14949 times.
|
74745 | for (i = 0; i < FF_ARRAY_ELEMS(pool->pools); i++) |
61 | 59796 | av_buffer_pool_uninit(&pool->pools[i]); | |
62 | |||
63 | 14949 | av_freep(&data); | |
64 | 14949 | } | |
65 | |||
66 | 14949 | static AVBufferRef *frame_pool_alloc(void) | |
67 | { | ||
68 | 14949 | FramePool *pool = av_mallocz(sizeof(*pool)); | |
69 | AVBufferRef *buf; | ||
70 | |||
71 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 14949 times.
|
14949 | if (!pool) |
72 | ✗ | return NULL; | |
73 | |||
74 | 14949 | buf = av_buffer_create((uint8_t*)pool, sizeof(*pool), | |
75 | frame_pool_free, NULL, 0); | ||
76 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 14949 times.
|
14949 | if (!buf) { |
77 | ✗ | av_freep(&pool); | |
78 | ✗ | return NULL; | |
79 | } | ||
80 | |||
81 | 14949 | return buf; | |
82 | } | ||
83 | |||
84 | 394224 | static int update_frame_pool(AVCodecContext *avctx, AVFrame *frame) | |
85 | { | ||
86 | 788448 | FramePool *pool = avctx->internal->pool ? | |
87 |
2/2✓ Branch 0 taken 386236 times.
✓ Branch 1 taken 7988 times.
|
394224 | (FramePool*)avctx->internal->pool->data : NULL; |
88 | AVBufferRef *pool_buf; | ||
89 | int i, ret, ch, planes; | ||
90 | |||
91 |
2/2✓ Branch 0 taken 303126 times.
✓ Branch 1 taken 91098 times.
|
394224 | if (avctx->codec_type == AVMEDIA_TYPE_AUDIO) { |
92 | 303126 | int planar = av_sample_fmt_is_planar(frame->format); | |
93 | 303126 | ch = frame->ch_layout.nb_channels; | |
94 | #if FF_API_OLD_CHANNEL_LAYOUT | ||
95 | FF_DISABLE_DEPRECATION_WARNINGS | ||
96 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 303126 times.
|
303126 | if (!ch) |
97 | ✗ | ch = frame->channels; | |
98 | FF_ENABLE_DEPRECATION_WARNINGS | ||
99 | #endif | ||
100 |
2/2✓ Branch 0 taken 162258 times.
✓ Branch 1 taken 140868 times.
|
303126 | planes = planar ? ch : 1; |
101 | } | ||
102 | |||
103 |
4/4✓ Branch 0 taken 386236 times.
✓ Branch 1 taken 7988 times.
✓ Branch 2 taken 386056 times.
✓ Branch 3 taken 180 times.
|
394224 | if (pool && pool->format == frame->format) { |
104 |
2/2✓ Branch 0 taken 84785 times.
✓ Branch 1 taken 301271 times.
|
386056 | if (avctx->codec_type == AVMEDIA_TYPE_VIDEO && |
105 |
4/4✓ Branch 0 taken 84562 times.
✓ Branch 1 taken 223 times.
✓ Branch 2 taken 84551 times.
✓ Branch 3 taken 11 times.
|
84785 | pool->width == frame->width && pool->height == frame->height) |
106 | 84551 | return 0; | |
107 |
4/4✓ Branch 0 taken 301271 times.
✓ Branch 1 taken 234 times.
✓ Branch 2 taken 301258 times.
✓ Branch 3 taken 13 times.
|
301505 | if (avctx->codec_type == AVMEDIA_TYPE_AUDIO && pool->planes == planes && |
108 |
3/4✓ Branch 0 taken 301258 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 294724 times.
✓ Branch 3 taken 6534 times.
|
301258 | pool->channels == ch && frame->nb_samples == pool->samples) |
109 | 294724 | return 0; | |
110 | } | ||
111 | |||
112 | 14949 | pool_buf = frame_pool_alloc(); | |
113 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 14949 times.
|
14949 | if (!pool_buf) |
114 | ✗ | return AVERROR(ENOMEM); | |
115 | 14949 | pool = (FramePool*)pool_buf->data; | |
116 | |||
117 |
2/3✓ Branch 0 taken 6547 times.
✓ Branch 1 taken 8402 times.
✗ Branch 2 not taken.
|
14949 | switch (avctx->codec_type) { |
118 | 6547 | case AVMEDIA_TYPE_VIDEO: { | |
119 | int linesize[4]; | ||
120 | 6547 | int w = frame->width; | |
121 | 6547 | int h = frame->height; | |
122 | int unaligned; | ||
123 | ptrdiff_t linesize1[4]; | ||
124 | size_t size[4]; | ||
125 | |||
126 | 6547 | avcodec_align_dimensions2(avctx, &w, &h, pool->stride_align); | |
127 | |||
128 | do { | ||
129 | // NOTE: do not align linesizes individually, this breaks e.g. assumptions | ||
130 | // that linesize[0] == 2*linesize[1] in the MPEG-encoder for 4:2:2 | ||
131 | 13083 | ret = av_image_fill_linesizes(linesize, avctx->pix_fmt, w); | |
132 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 13083 times.
|
13083 | if (ret < 0) |
133 | ✗ | goto fail; | |
134 | // increase alignment of w for next try (rhs gives the lowest bit set in w) | ||
135 | 13083 | w += w & ~(w - 1); | |
136 | |||
137 | 13083 | unaligned = 0; | |
138 |
2/2✓ Branch 0 taken 52332 times.
✓ Branch 1 taken 13083 times.
|
65415 | for (i = 0; i < 4; i++) |
139 | 52332 | unaligned |= linesize[i] % pool->stride_align[i]; | |
140 |
2/2✓ Branch 0 taken 6536 times.
✓ Branch 1 taken 6547 times.
|
13083 | } while (unaligned); |
141 | |||
142 |
2/2✓ Branch 0 taken 26188 times.
✓ Branch 1 taken 6547 times.
|
32735 | for (i = 0; i < 4; i++) |
143 | 26188 | linesize1[i] = linesize[i]; | |
144 | 6547 | ret = av_image_fill_plane_sizes(size, avctx->pix_fmt, h, linesize1); | |
145 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6547 times.
|
6547 | if (ret < 0) |
146 | ✗ | goto fail; | |
147 | |||
148 |
2/2✓ Branch 0 taken 26188 times.
✓ Branch 1 taken 6547 times.
|
32735 | for (i = 0; i < 4; i++) { |
149 | 26188 | pool->linesize[i] = linesize[i]; | |
150 |
2/2✓ Branch 0 taken 18106 times.
✓ Branch 1 taken 8082 times.
|
26188 | if (size[i]) { |
151 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18106 times.
|
18106 | if (size[i] > INT_MAX - (16 + STRIDE_ALIGN - 1)) { |
152 | ✗ | ret = AVERROR(EINVAL); | |
153 | ✗ | goto fail; | |
154 | } | ||
155 | 18106 | pool->pools[i] = av_buffer_pool_init(size[i] + 16 + STRIDE_ALIGN - 1, | |
156 | CONFIG_MEMORY_POISONING ? | ||
157 | NULL : | ||
158 | av_buffer_allocz); | ||
159 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18106 times.
|
18106 | if (!pool->pools[i]) { |
160 | ✗ | ret = AVERROR(ENOMEM); | |
161 | ✗ | goto fail; | |
162 | } | ||
163 | } | ||
164 | } | ||
165 | 6547 | pool->format = frame->format; | |
166 | 6547 | pool->width = frame->width; | |
167 | 6547 | pool->height = frame->height; | |
168 | |||
169 | 6547 | break; | |
170 | } | ||
171 | 8402 | case AVMEDIA_TYPE_AUDIO: { | |
172 | 8402 | ret = av_samples_get_buffer_size(&pool->linesize[0], ch, | |
173 | 8402 | frame->nb_samples, frame->format, 0); | |
174 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 8402 times.
|
8402 | if (ret < 0) |
175 | ✗ | goto fail; | |
176 | |||
177 | 8402 | pool->pools[0] = av_buffer_pool_init(pool->linesize[0], NULL); | |
178 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 8402 times.
|
8402 | if (!pool->pools[0]) { |
179 | ✗ | ret = AVERROR(ENOMEM); | |
180 | ✗ | goto fail; | |
181 | } | ||
182 | |||
183 | 8402 | pool->format = frame->format; | |
184 | 8402 | pool->planes = planes; | |
185 | 8402 | pool->channels = ch; | |
186 | 8402 | pool->samples = frame->nb_samples; | |
187 | 8402 | break; | |
188 | } | ||
189 | ✗ | default: av_assert0(0); | |
190 | } | ||
191 | |||
192 | 14949 | av_buffer_unref(&avctx->internal->pool); | |
193 | 14949 | avctx->internal->pool = pool_buf; | |
194 | |||
195 | 14949 | return 0; | |
196 | ✗ | fail: | |
197 | ✗ | av_buffer_unref(&pool_buf); | |
198 | ✗ | return ret; | |
199 | } | ||
200 | |||
201 | 303126 | static int audio_get_buffer(AVCodecContext *avctx, AVFrame *frame) | |
202 | { | ||
203 | 303126 | FramePool *pool = (FramePool*)avctx->internal->pool->data; | |
204 | 303126 | int planes = pool->planes; | |
205 | int i; | ||
206 | |||
207 | 303126 | frame->linesize[0] = pool->linesize[0]; | |
208 | |||
209 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 303126 times.
|
303126 | if (planes > AV_NUM_DATA_POINTERS) { |
210 | ✗ | frame->extended_data = av_calloc(planes, sizeof(*frame->extended_data)); | |
211 | ✗ | frame->nb_extended_buf = planes - AV_NUM_DATA_POINTERS; | |
212 | ✗ | frame->extended_buf = av_calloc(frame->nb_extended_buf, | |
213 | sizeof(*frame->extended_buf)); | ||
214 | ✗ | if (!frame->extended_data || !frame->extended_buf) { | |
215 | ✗ | av_freep(&frame->extended_data); | |
216 | ✗ | av_freep(&frame->extended_buf); | |
217 | ✗ | return AVERROR(ENOMEM); | |
218 | } | ||
219 | } else { | ||
220 | 303126 | frame->extended_data = frame->data; | |
221 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 303126 times.
|
303126 | av_assert0(frame->nb_extended_buf == 0); |
222 | } | ||
223 | |||
224 |
2/2✓ Branch 0 taken 447446 times.
✓ Branch 1 taken 303126 times.
|
750572 | for (i = 0; i < FFMIN(planes, AV_NUM_DATA_POINTERS); i++) { |
225 | 447446 | frame->buf[i] = av_buffer_pool_get(pool->pools[0]); | |
226 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 447446 times.
|
447446 | if (!frame->buf[i]) |
227 | ✗ | goto fail; | |
228 | 447446 | frame->extended_data[i] = frame->data[i] = frame->buf[i]->data; | |
229 | } | ||
230 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 303126 times.
|
303126 | for (i = 0; i < frame->nb_extended_buf; i++) { |
231 | ✗ | frame->extended_buf[i] = av_buffer_pool_get(pool->pools[0]); | |
232 | ✗ | if (!frame->extended_buf[i]) | |
233 | ✗ | goto fail; | |
234 | ✗ | frame->extended_data[i + AV_NUM_DATA_POINTERS] = frame->extended_buf[i]->data; | |
235 | } | ||
236 | |||
237 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 303126 times.
|
303126 | if (avctx->debug & FF_DEBUG_BUFFERS) |
238 | ✗ | av_log(avctx, AV_LOG_DEBUG, "default_get_buffer called on frame %p", frame); | |
239 | |||
240 | 303126 | return 0; | |
241 | ✗ | fail: | |
242 | ✗ | av_frame_unref(frame); | |
243 | ✗ | return AVERROR(ENOMEM); | |
244 | } | ||
245 | |||
246 | 91098 | static int video_get_buffer(AVCodecContext *s, AVFrame *pic) | |
247 | { | ||
248 | 91098 | FramePool *pool = (FramePool*)s->internal->pool->data; | |
249 | int i; | ||
250 | |||
251 |
4/8✓ Branch 0 taken 91098 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 91098 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 91098 times.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✓ Branch 7 taken 91098 times.
|
91098 | if (pic->data[0] || pic->data[1] || pic->data[2] || pic->data[3]) { |
252 | ✗ | av_log(s, AV_LOG_ERROR, "pic->data[*]!=NULL in avcodec_default_get_buffer\n"); | |
253 | ✗ | return -1; | |
254 | } | ||
255 | |||
256 | 91098 | memset(pic->data, 0, sizeof(pic->data)); | |
257 | 91098 | pic->extended_data = pic->data; | |
258 | |||
259 |
4/4✓ Branch 0 taken 349887 times.
✓ Branch 1 taken 889 times.
✓ Branch 2 taken 259678 times.
✓ Branch 3 taken 90209 times.
|
350776 | for (i = 0; i < 4 && pool->pools[i]; i++) { |
260 | 259678 | pic->linesize[i] = pool->linesize[i]; | |
261 | |||
262 | 259678 | pic->buf[i] = av_buffer_pool_get(pool->pools[i]); | |
263 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 259678 times.
|
259678 | if (!pic->buf[i]) |
264 | ✗ | goto fail; | |
265 | |||
266 | 259678 | pic->data[i] = pic->buf[i]->data; | |
267 | } | ||
268 |
2/2✓ Branch 0 taken 469106 times.
✓ Branch 1 taken 91098 times.
|
560204 | for (; i < AV_NUM_DATA_POINTERS; i++) { |
269 | 469106 | pic->data[i] = NULL; | |
270 | 469106 | pic->linesize[i] = 0; | |
271 | } | ||
272 | |||
273 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 91098 times.
|
91098 | if (s->debug & FF_DEBUG_BUFFERS) |
274 | ✗ | av_log(s, AV_LOG_DEBUG, "default_get_buffer called on pic %p\n", pic); | |
275 | |||
276 | 91098 | return 0; | |
277 | ✗ | fail: | |
278 | ✗ | av_frame_unref(pic); | |
279 | ✗ | return AVERROR(ENOMEM); | |
280 | } | ||
281 | |||
282 | 394224 | int avcodec_default_get_buffer2(AVCodecContext *avctx, AVFrame *frame, int flags) | |
283 | { | ||
284 | int ret; | ||
285 | |||
286 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 394224 times.
|
394224 | if (avctx->hw_frames_ctx) { |
287 | ✗ | ret = av_hwframe_get_buffer(avctx->hw_frames_ctx, frame, 0); | |
288 | ✗ | frame->width = avctx->coded_width; | |
289 | ✗ | frame->height = avctx->coded_height; | |
290 | ✗ | return ret; | |
291 | } | ||
292 | |||
293 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 394224 times.
|
394224 | if ((ret = update_frame_pool(avctx, frame)) < 0) |
294 | ✗ | return ret; | |
295 | |||
296 |
2/3✓ Branch 0 taken 91098 times.
✓ Branch 1 taken 303126 times.
✗ Branch 2 not taken.
|
394224 | switch (avctx->codec_type) { |
297 | 91098 | case AVMEDIA_TYPE_VIDEO: | |
298 | 91098 | return video_get_buffer(avctx, frame); | |
299 | 303126 | case AVMEDIA_TYPE_AUDIO: | |
300 | 303126 | return audio_get_buffer(avctx, frame); | |
301 | ✗ | default: | |
302 | ✗ | return -1; | |
303 | } | ||
304 | } | ||
305 |