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 | #include "refstruct.h" | ||
36 | |||
37 | typedef struct FramePool { | ||
38 | /** | ||
39 | * Pools for each data plane. For audio all the planes have the same size, | ||
40 | * so only pools[0] is used. | ||
41 | */ | ||
42 | AVBufferPool *pools[4]; | ||
43 | |||
44 | /* | ||
45 | * Pool parameters | ||
46 | */ | ||
47 | int format; | ||
48 | int width, height; | ||
49 | int stride_align[AV_NUM_DATA_POINTERS]; | ||
50 | int linesize[4]; | ||
51 | int planes; | ||
52 | int channels; | ||
53 | int samples; | ||
54 | } FramePool; | ||
55 | |||
56 | 15342 | static void frame_pool_free(FFRefStructOpaque unused, void *obj) | |
57 | { | ||
58 | 15342 | FramePool *pool = obj; | |
59 | int i; | ||
60 | |||
61 |
2/2✓ Branch 0 taken 61368 times.
✓ Branch 1 taken 15342 times.
|
76710 | for (i = 0; i < FF_ARRAY_ELEMS(pool->pools); i++) |
62 | 61368 | av_buffer_pool_uninit(&pool->pools[i]); | |
63 | 15342 | } | |
64 | |||
65 | 389205 | static int update_frame_pool(AVCodecContext *avctx, AVFrame *frame) | |
66 | { | ||
67 | 389205 | FramePool *pool = avctx->internal->pool; | |
68 | int i, ret; | ||
69 | |||
70 |
4/4✓ Branch 0 taken 380654 times.
✓ Branch 1 taken 8551 times.
✓ Branch 2 taken 380474 times.
✓ Branch 3 taken 180 times.
|
389205 | if (pool && pool->format == frame->format) { |
71 |
2/2✓ Branch 0 taken 121220 times.
✓ Branch 1 taken 259254 times.
|
380474 | if (avctx->codec_type == AVMEDIA_TYPE_VIDEO && |
72 |
4/4✓ Branch 0 taken 120991 times.
✓ Branch 1 taken 229 times.
✓ Branch 2 taken 120978 times.
✓ Branch 3 taken 13 times.
|
121220 | pool->width == frame->width && pool->height == frame->height) |
73 | 120978 | return 0; | |
74 |
2/2✓ Branch 0 taken 259254 times.
✓ Branch 1 taken 242 times.
|
259496 | if (avctx->codec_type == AVMEDIA_TYPE_AUDIO && |
75 |
2/2✓ Branch 0 taken 259241 times.
✓ Branch 1 taken 13 times.
|
259254 | pool->channels == frame->ch_layout.nb_channels && |
76 |
2/2✓ Branch 0 taken 252885 times.
✓ Branch 1 taken 6356 times.
|
259241 | frame->nb_samples == pool->samples) |
77 | 252885 | return 0; | |
78 | } | ||
79 | |||
80 | 15342 | pool = ff_refstruct_alloc_ext(sizeof(*pool), 0, NULL, frame_pool_free); | |
81 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 15342 times.
|
15342 | if (!pool) |
82 | ✗ | return AVERROR(ENOMEM); | |
83 | |||
84 |
2/3✓ Branch 0 taken 7047 times.
✓ Branch 1 taken 8295 times.
✗ Branch 2 not taken.
|
15342 | switch (avctx->codec_type) { |
85 | 7047 | case AVMEDIA_TYPE_VIDEO: { | |
86 | int linesize[4]; | ||
87 | 7047 | int w = frame->width; | |
88 | 7047 | int h = frame->height; | |
89 | int unaligned; | ||
90 | ptrdiff_t linesize1[4]; | ||
91 | size_t size[4]; | ||
92 | |||
93 | 7047 | avcodec_align_dimensions2(avctx, &w, &h, pool->stride_align); | |
94 | |||
95 | do { | ||
96 | // NOTE: do not align linesizes individually, this breaks e.g. assumptions | ||
97 | // that linesize[0] == 2*linesize[1] in the MPEG-encoder for 4:2:2 | ||
98 | 14034 | ret = av_image_fill_linesizes(linesize, avctx->pix_fmt, w); | |
99 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 14034 times.
|
14034 | if (ret < 0) |
100 | ✗ | goto fail; | |
101 | // increase alignment of w for next try (rhs gives the lowest bit set in w) | ||
102 | 14034 | w += w & ~(w - 1); | |
103 | |||
104 | 14034 | unaligned = 0; | |
105 |
2/2✓ Branch 0 taken 56136 times.
✓ Branch 1 taken 14034 times.
|
70170 | for (i = 0; i < 4; i++) |
106 | 56136 | unaligned |= linesize[i] % pool->stride_align[i]; | |
107 |
2/2✓ Branch 0 taken 6987 times.
✓ Branch 1 taken 7047 times.
|
14034 | } while (unaligned); |
108 | |||
109 |
2/2✓ Branch 0 taken 28188 times.
✓ Branch 1 taken 7047 times.
|
35235 | for (i = 0; i < 4; i++) |
110 | 28188 | linesize1[i] = linesize[i]; | |
111 | 7047 | ret = av_image_fill_plane_sizes(size, avctx->pix_fmt, h, linesize1); | |
112 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 7047 times.
|
7047 | if (ret < 0) |
113 | ✗ | goto fail; | |
114 | |||
115 |
2/2✓ Branch 0 taken 28188 times.
✓ Branch 1 taken 7047 times.
|
35235 | for (i = 0; i < 4; i++) { |
116 | 28188 | pool->linesize[i] = linesize[i]; | |
117 |
2/2✓ Branch 0 taken 19584 times.
✓ Branch 1 taken 8604 times.
|
28188 | if (size[i]) { |
118 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 19584 times.
|
19584 | if (size[i] > INT_MAX - (16 + STRIDE_ALIGN - 1)) { |
119 | ✗ | ret = AVERROR(EINVAL); | |
120 | ✗ | goto fail; | |
121 | } | ||
122 | 19584 | pool->pools[i] = av_buffer_pool_init(size[i] + 16 + STRIDE_ALIGN - 1, | |
123 | CONFIG_MEMORY_POISONING ? | ||
124 | NULL : | ||
125 | av_buffer_allocz); | ||
126 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 19584 times.
|
19584 | if (!pool->pools[i]) { |
127 | ✗ | ret = AVERROR(ENOMEM); | |
128 | ✗ | goto fail; | |
129 | } | ||
130 | } | ||
131 | } | ||
132 | 7047 | pool->format = frame->format; | |
133 | 7047 | pool->width = frame->width; | |
134 | 7047 | pool->height = frame->height; | |
135 | |||
136 | 7047 | break; | |
137 | } | ||
138 | 8295 | case AVMEDIA_TYPE_AUDIO: { | |
139 | 8295 | ret = av_samples_get_buffer_size(&pool->linesize[0], | |
140 | frame->ch_layout.nb_channels, | ||
141 | 8295 | frame->nb_samples, frame->format, 0); | |
142 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 8295 times.
|
8295 | if (ret < 0) |
143 | ✗ | goto fail; | |
144 | |||
145 | 8295 | pool->pools[0] = av_buffer_pool_init(pool->linesize[0], NULL); | |
146 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 8295 times.
|
8295 | if (!pool->pools[0]) { |
147 | ✗ | ret = AVERROR(ENOMEM); | |
148 | ✗ | goto fail; | |
149 | } | ||
150 | |||
151 | 8295 | pool->format = frame->format; | |
152 | 8295 | pool->channels = frame->ch_layout.nb_channels; | |
153 | 8295 | pool->samples = frame->nb_samples; | |
154 |
2/2✓ Branch 1 taken 5480 times.
✓ Branch 2 taken 2815 times.
|
8295 | pool->planes = av_sample_fmt_is_planar(pool->format) ? pool->channels : 1; |
155 | 8295 | break; | |
156 | } | ||
157 | ✗ | default: av_assert0(0); | |
158 | } | ||
159 | |||
160 | 15342 | ff_refstruct_unref(&avctx->internal->pool); | |
161 | 15342 | avctx->internal->pool = pool; | |
162 | |||
163 | 15342 | return 0; | |
164 | ✗ | fail: | |
165 | ✗ | ff_refstruct_unref(&pool); | |
166 | ✗ | return ret; | |
167 | } | ||
168 | |||
169 | 261180 | static int audio_get_buffer(AVCodecContext *avctx, AVFrame *frame) | |
170 | { | ||
171 | 261180 | FramePool *pool = avctx->internal->pool; | |
172 | 261180 | int planes = pool->planes; | |
173 | int i; | ||
174 | |||
175 | 261180 | frame->linesize[0] = pool->linesize[0]; | |
176 | |||
177 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 261180 times.
|
261180 | if (planes > AV_NUM_DATA_POINTERS) { |
178 | ✗ | frame->extended_data = av_calloc(planes, sizeof(*frame->extended_data)); | |
179 | ✗ | frame->nb_extended_buf = planes - AV_NUM_DATA_POINTERS; | |
180 | ✗ | frame->extended_buf = av_calloc(frame->nb_extended_buf, | |
181 | sizeof(*frame->extended_buf)); | ||
182 | ✗ | if (!frame->extended_data || !frame->extended_buf) { | |
183 | ✗ | av_freep(&frame->extended_data); | |
184 | ✗ | av_freep(&frame->extended_buf); | |
185 | ✗ | return AVERROR(ENOMEM); | |
186 | } | ||
187 | } else { | ||
188 | 261180 | frame->extended_data = frame->data; | |
189 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 261180 times.
|
261180 | av_assert0(frame->nb_extended_buf == 0); |
190 | } | ||
191 | |||
192 |
2/2✓ Branch 0 taken 408329 times.
✓ Branch 1 taken 261180 times.
|
669509 | for (i = 0; i < FFMIN(planes, AV_NUM_DATA_POINTERS); i++) { |
193 | 408329 | frame->buf[i] = av_buffer_pool_get(pool->pools[0]); | |
194 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 408329 times.
|
408329 | if (!frame->buf[i]) |
195 | ✗ | goto fail; | |
196 | 408329 | frame->extended_data[i] = frame->data[i] = frame->buf[i]->data; | |
197 | } | ||
198 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 261180 times.
|
261180 | for (i = 0; i < frame->nb_extended_buf; i++) { |
199 | ✗ | frame->extended_buf[i] = av_buffer_pool_get(pool->pools[0]); | |
200 | ✗ | if (!frame->extended_buf[i]) | |
201 | ✗ | goto fail; | |
202 | ✗ | frame->extended_data[i + AV_NUM_DATA_POINTERS] = frame->extended_buf[i]->data; | |
203 | } | ||
204 | |||
205 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 261180 times.
|
261180 | if (avctx->debug & FF_DEBUG_BUFFERS) |
206 | ✗ | av_log(avctx, AV_LOG_DEBUG, "default_get_buffer called on frame %p", frame); | |
207 | |||
208 | 261180 | return 0; | |
209 | ✗ | fail: | |
210 | ✗ | av_frame_unref(frame); | |
211 | ✗ | return AVERROR(ENOMEM); | |
212 | } | ||
213 | |||
214 | 128025 | static int video_get_buffer(AVCodecContext *s, AVFrame *pic) | |
215 | { | ||
216 | 128025 | FramePool *pool = s->internal->pool; | |
217 | int i; | ||
218 | |||
219 |
4/8✓ Branch 0 taken 128025 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 128025 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 128025 times.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✓ Branch 7 taken 128025 times.
|
128025 | if (pic->data[0] || pic->data[1] || pic->data[2] || pic->data[3]) { |
220 | ✗ | av_log(s, AV_LOG_ERROR, "pic->data[*]!=NULL in avcodec_default_get_buffer\n"); | |
221 | ✗ | return -1; | |
222 | } | ||
223 | |||
224 | 128025 | memset(pic->data, 0, sizeof(pic->data)); | |
225 | 128025 | pic->extended_data = pic->data; | |
226 | |||
227 |
4/4✓ Branch 0 taken 489110 times.
✓ Branch 1 taken 691 times.
✓ Branch 2 taken 361776 times.
✓ Branch 3 taken 127334 times.
|
489801 | for (i = 0; i < 4 && pool->pools[i]; i++) { |
228 | 361776 | pic->linesize[i] = pool->linesize[i]; | |
229 | |||
230 | 361776 | pic->buf[i] = av_buffer_pool_get(pool->pools[i]); | |
231 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 361776 times.
|
361776 | if (!pic->buf[i]) |
232 | ✗ | goto fail; | |
233 | |||
234 | 361776 | pic->data[i] = pic->buf[i]->data; | |
235 | } | ||
236 |
2/2✓ Branch 0 taken 662424 times.
✓ Branch 1 taken 128025 times.
|
790449 | for (; i < AV_NUM_DATA_POINTERS; i++) { |
237 | 662424 | pic->data[i] = NULL; | |
238 | 662424 | pic->linesize[i] = 0; | |
239 | } | ||
240 | |||
241 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 128025 times.
|
128025 | if (s->debug & FF_DEBUG_BUFFERS) |
242 | ✗ | av_log(s, AV_LOG_DEBUG, "default_get_buffer called on pic %p\n", pic); | |
243 | |||
244 | 128025 | return 0; | |
245 | ✗ | fail: | |
246 | ✗ | av_frame_unref(pic); | |
247 | ✗ | return AVERROR(ENOMEM); | |
248 | } | ||
249 | |||
250 | 389205 | int avcodec_default_get_buffer2(AVCodecContext *avctx, AVFrame *frame, int flags) | |
251 | { | ||
252 | int ret; | ||
253 | |||
254 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 389205 times.
|
389205 | if (avctx->hw_frames_ctx) { |
255 | ✗ | ret = av_hwframe_get_buffer(avctx->hw_frames_ctx, frame, 0); | |
256 | ✗ | if (ret == AVERROR(ENOMEM)) { | |
257 | ✗ | AVHWFramesContext *frames_ctx = | |
258 | ✗ | (AVHWFramesContext*)avctx->hw_frames_ctx->data; | |
259 | ✗ | if (frames_ctx->initial_pool_size > 0 && | |
260 | ✗ | !avctx->internal->warned_on_failed_allocation_from_fixed_pool) { | |
261 | ✗ | av_log(avctx, AV_LOG_WARNING, "Failed to allocate a %s/%s " | |
262 | "frame from a fixed pool of hardware frames.\n", | ||
263 | av_get_pix_fmt_name(frames_ctx->format), | ||
264 | av_get_pix_fmt_name(frames_ctx->sw_format)); | ||
265 | ✗ | av_log(avctx, AV_LOG_WARNING, "Consider setting " | |
266 | "extra_hw_frames to a larger value " | ||
267 | "(currently set to %d, giving a pool size of %d).\n", | ||
268 | avctx->extra_hw_frames, frames_ctx->initial_pool_size); | ||
269 | ✗ | avctx->internal->warned_on_failed_allocation_from_fixed_pool = 1; | |
270 | } | ||
271 | } | ||
272 | ✗ | frame->width = avctx->coded_width; | |
273 | ✗ | frame->height = avctx->coded_height; | |
274 | ✗ | return ret; | |
275 | } | ||
276 | |||
277 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 389205 times.
|
389205 | if ((ret = update_frame_pool(avctx, frame)) < 0) |
278 | ✗ | return ret; | |
279 | |||
280 |
2/3✓ Branch 0 taken 128025 times.
✓ Branch 1 taken 261180 times.
✗ Branch 2 not taken.
|
389205 | switch (avctx->codec_type) { |
281 | 128025 | case AVMEDIA_TYPE_VIDEO: | |
282 | 128025 | return video_get_buffer(avctx, frame); | |
283 | 261180 | case AVMEDIA_TYPE_AUDIO: | |
284 | 261180 | return audio_get_buffer(avctx, frame); | |
285 | ✗ | default: | |
286 | ✗ | return -1; | |
287 | } | ||
288 | } | ||
289 |