GCC Code Coverage Report
Directory: ../../../ffmpeg/ Exec Total Coverage
File: src/libavcodec/utils.c Lines: 944 1281 73.7 %
Date: 2020-10-23 17:01:47 Branches: 671 973 69.0 %

Line Branch Exec Source
1
/*
2
 * utils for libavcodec
3
 * Copyright (c) 2001 Fabrice Bellard
4
 * Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at>
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
/**
24
 * @file
25
 * utils.
26
 */
27
28
#include "config.h"
29
#include "libavutil/attributes.h"
30
#include "libavutil/avassert.h"
31
#include "libavutil/avstring.h"
32
#include "libavutil/bprint.h"
33
#include "libavutil/channel_layout.h"
34
#include "libavutil/crc.h"
35
#include "libavutil/frame.h"
36
#include "libavutil/hwcontext.h"
37
#include "libavutil/internal.h"
38
#include "libavutil/mathematics.h"
39
#include "libavutil/mem_internal.h"
40
#include "libavutil/pixdesc.h"
41
#include "libavutil/imgutils.h"
42
#include "libavutil/samplefmt.h"
43
#include "libavutil/dict.h"
44
#include "libavutil/thread.h"
45
#include "avcodec.h"
46
#include "decode.h"
47
#include "hwconfig.h"
48
#include "libavutil/opt.h"
49
#include "mpegvideo.h"
50
#include "thread.h"
51
#include "frame_thread_encoder.h"
52
#include "internal.h"
53
#include "packet_internal.h"
54
#include "put_bits.h"
55
#include "raw.h"
56
#include "bytestream.h"
57
#include "version.h"
58
#include <stdlib.h>
59
#include <stdarg.h>
60
#include <stdatomic.h>
61
#include <limits.h>
62
#include <float.h>
63
#if CONFIG_ICONV
64
# include <iconv.h>
65
#endif
66
67
#include "libavutil/ffversion.h"
68
const char av_codec_ffversion[] = "FFmpeg version " FFMPEG_VERSION;
69
70
static AVMutex codec_mutex = AV_MUTEX_INITIALIZER;
71
72
134062
void av_fast_padded_malloc(void *ptr, unsigned int *size, size_t min_size)
73
{
74
134062
    uint8_t **p = ptr;
75
134062
    if (min_size > SIZE_MAX - AV_INPUT_BUFFER_PADDING_SIZE) {
76
        av_freep(p);
77
        *size = 0;
78
        return;
79
    }
80
134062
    if (!ff_fast_malloc(p, size, min_size + AV_INPUT_BUFFER_PADDING_SIZE, 1))
81
105948
        memset(*p + min_size, 0, AV_INPUT_BUFFER_PADDING_SIZE);
82
}
83
84
485
void av_fast_padded_mallocz(void *ptr, unsigned int *size, size_t min_size)
85
{
86
485
    uint8_t **p = ptr;
87
485
    if (min_size > SIZE_MAX - AV_INPUT_BUFFER_PADDING_SIZE) {
88
        av_freep(p);
89
        *size = 0;
90
        return;
91
    }
92
485
    if (!ff_fast_malloc(p, size, min_size + AV_INPUT_BUFFER_PADDING_SIZE, 1))
93
374
        memset(*p, 0, min_size + AV_INPUT_BUFFER_PADDING_SIZE);
94
}
95
96
6233663
int av_codec_is_encoder(const AVCodec *codec)
97
{
98


6233663
    return codec && (codec->encode_sub || codec->encode2 || codec->receive_packet);
99
}
100
101
11183803
int av_codec_is_decoder(const AVCodec *codec)
102
{
103

11183803
    return codec && (codec->decode || codec->receive_frame);
104
}
105
106
29284
int ff_set_dimensions(AVCodecContext *s, int width, int height)
107
{
108
29284
    int ret = av_image_check_size2(width, height, s->max_pixels, AV_PIX_FMT_NONE, 0, s);
109
110
29284
    if (ret < 0)
111
        width = height = 0;
112
113
29284
    s->coded_width  = width;
114
29284
    s->coded_height = height;
115
29284
    s->width        = AV_CEIL_RSHIFT(width,  s->lowres);
116
29284
    s->height       = AV_CEIL_RSHIFT(height, s->lowres);
117
118
29284
    return ret;
119
}
120
121
2470
int ff_set_sar(AVCodecContext *avctx, AVRational sar)
122
{
123
2470
    int ret = av_image_check_sar(avctx->width, avctx->height, sar);
124
125
2470
    if (ret < 0) {
126
14
        av_log(avctx, AV_LOG_WARNING, "ignoring invalid SAR: %d/%d\n",
127
               sar.num, sar.den);
128
14
        avctx->sample_aspect_ratio = (AVRational){ 0, 1 };
129
14
        return ret;
130
    } else {
131
2456
        avctx->sample_aspect_ratio = sar;
132
    }
133
2456
    return 0;
134
}
135
136
20526
int ff_side_data_update_matrix_encoding(AVFrame *frame,
137
                                        enum AVMatrixEncoding matrix_encoding)
138
{
139
    AVFrameSideData *side_data;
140
    enum AVMatrixEncoding *data;
141
142
20526
    side_data = av_frame_get_side_data(frame, AV_FRAME_DATA_MATRIXENCODING);
143
20526
    if (!side_data)
144
20526
        side_data = av_frame_new_side_data(frame, AV_FRAME_DATA_MATRIXENCODING,
145
                                           sizeof(enum AVMatrixEncoding));
146
147
20526
    if (!side_data)
148
        return AVERROR(ENOMEM);
149
150
20526
    data  = (enum AVMatrixEncoding*)side_data->data;
151
20526
    *data = matrix_encoding;
152
153
20526
    return 0;
154
}
155
156
8464
void avcodec_align_dimensions2(AVCodecContext *s, int *width, int *height,
157
                               int linesize_align[AV_NUM_DATA_POINTERS])
158
{
159
    int i;
160
8464
    int w_align = 1;
161
8464
    int h_align = 1;
162
8464
    AVPixFmtDescriptor const *desc = av_pix_fmt_desc_get(s->pix_fmt);
163
164
8464
    if (desc) {
165
8464
        w_align = 1 << desc->log2_chroma_w;
166
8464
        h_align = 1 << desc->log2_chroma_h;
167
    }
168
169


8464
    switch (s->pix_fmt) {
170
7556
    case AV_PIX_FMT_YUV420P:
171
    case AV_PIX_FMT_YUYV422:
172
    case AV_PIX_FMT_YVYU422:
173
    case AV_PIX_FMT_UYVY422:
174
    case AV_PIX_FMT_YUV422P:
175
    case AV_PIX_FMT_YUV440P:
176
    case AV_PIX_FMT_YUV444P:
177
    case AV_PIX_FMT_GBRP:
178
    case AV_PIX_FMT_GBRAP:
179
    case AV_PIX_FMT_GRAY8:
180
    case AV_PIX_FMT_GRAY16BE:
181
    case AV_PIX_FMT_GRAY16LE:
182
    case AV_PIX_FMT_YUVJ420P:
183
    case AV_PIX_FMT_YUVJ422P:
184
    case AV_PIX_FMT_YUVJ440P:
185
    case AV_PIX_FMT_YUVJ444P:
186
    case AV_PIX_FMT_YUVA420P:
187
    case AV_PIX_FMT_YUVA422P:
188
    case AV_PIX_FMT_YUVA444P:
189
    case AV_PIX_FMT_YUV420P9LE:
190
    case AV_PIX_FMT_YUV420P9BE:
191
    case AV_PIX_FMT_YUV420P10LE:
192
    case AV_PIX_FMT_YUV420P10BE:
193
    case AV_PIX_FMT_YUV420P12LE:
194
    case AV_PIX_FMT_YUV420P12BE:
195
    case AV_PIX_FMT_YUV420P14LE:
196
    case AV_PIX_FMT_YUV420P14BE:
197
    case AV_PIX_FMT_YUV420P16LE:
198
    case AV_PIX_FMT_YUV420P16BE:
199
    case AV_PIX_FMT_YUVA420P9LE:
200
    case AV_PIX_FMT_YUVA420P9BE:
201
    case AV_PIX_FMT_YUVA420P10LE:
202
    case AV_PIX_FMT_YUVA420P10BE:
203
    case AV_PIX_FMT_YUVA420P16LE:
204
    case AV_PIX_FMT_YUVA420P16BE:
205
    case AV_PIX_FMT_YUV422P9LE:
206
    case AV_PIX_FMT_YUV422P9BE:
207
    case AV_PIX_FMT_YUV422P10LE:
208
    case AV_PIX_FMT_YUV422P10BE:
209
    case AV_PIX_FMT_YUV422P12LE:
210
    case AV_PIX_FMT_YUV422P12BE:
211
    case AV_PIX_FMT_YUV422P14LE:
212
    case AV_PIX_FMT_YUV422P14BE:
213
    case AV_PIX_FMT_YUV422P16LE:
214
    case AV_PIX_FMT_YUV422P16BE:
215
    case AV_PIX_FMT_YUVA422P9LE:
216
    case AV_PIX_FMT_YUVA422P9BE:
217
    case AV_PIX_FMT_YUVA422P10LE:
218
    case AV_PIX_FMT_YUVA422P10BE:
219
    case AV_PIX_FMT_YUVA422P12LE:
220
    case AV_PIX_FMT_YUVA422P12BE:
221
    case AV_PIX_FMT_YUVA422P16LE:
222
    case AV_PIX_FMT_YUVA422P16BE:
223
    case AV_PIX_FMT_YUV440P10LE:
224
    case AV_PIX_FMT_YUV440P10BE:
225
    case AV_PIX_FMT_YUV440P12LE:
226
    case AV_PIX_FMT_YUV440P12BE:
227
    case AV_PIX_FMT_YUV444P9LE:
228
    case AV_PIX_FMT_YUV444P9BE:
229
    case AV_PIX_FMT_YUV444P10LE:
230
    case AV_PIX_FMT_YUV444P10BE:
231
    case AV_PIX_FMT_YUV444P12LE:
232
    case AV_PIX_FMT_YUV444P12BE:
233
    case AV_PIX_FMT_YUV444P14LE:
234
    case AV_PIX_FMT_YUV444P14BE:
235
    case AV_PIX_FMT_YUV444P16LE:
236
    case AV_PIX_FMT_YUV444P16BE:
237
    case AV_PIX_FMT_YUVA444P9LE:
238
    case AV_PIX_FMT_YUVA444P9BE:
239
    case AV_PIX_FMT_YUVA444P10LE:
240
    case AV_PIX_FMT_YUVA444P10BE:
241
    case AV_PIX_FMT_YUVA444P12LE:
242
    case AV_PIX_FMT_YUVA444P12BE:
243
    case AV_PIX_FMT_YUVA444P16LE:
244
    case AV_PIX_FMT_YUVA444P16BE:
245
    case AV_PIX_FMT_GBRP9LE:
246
    case AV_PIX_FMT_GBRP9BE:
247
    case AV_PIX_FMT_GBRP10LE:
248
    case AV_PIX_FMT_GBRP10BE:
249
    case AV_PIX_FMT_GBRP12LE:
250
    case AV_PIX_FMT_GBRP12BE:
251
    case AV_PIX_FMT_GBRP14LE:
252
    case AV_PIX_FMT_GBRP14BE:
253
    case AV_PIX_FMT_GBRP16LE:
254
    case AV_PIX_FMT_GBRP16BE:
255
    case AV_PIX_FMT_GBRAP12LE:
256
    case AV_PIX_FMT_GBRAP12BE:
257
    case AV_PIX_FMT_GBRAP16LE:
258
    case AV_PIX_FMT_GBRAP16BE:
259
7556
        w_align = 16; //FIXME assume 16 pixel per macroblock
260
7556
        h_align = 16 * 2; // interlaced needs 2 macroblocks height
261
7556
        break;
262
26
    case AV_PIX_FMT_YUV411P:
263
    case AV_PIX_FMT_YUVJ411P:
264
    case AV_PIX_FMT_UYYVYY411:
265
26
        w_align = 32;
266
26
        h_align = 16 * 2;
267
26
        break;
268
21
    case AV_PIX_FMT_YUV410P:
269
21
        if (s->codec_id == AV_CODEC_ID_SVQ1) {
270
10
            w_align = 64;
271
10
            h_align = 64;
272
        }
273
21
        break;
274
34
    case AV_PIX_FMT_RGB555:
275
34
        if (s->codec_id == AV_CODEC_ID_RPZA) {
276
1
            w_align = 4;
277
1
            h_align = 4;
278
        }
279
34
        if (s->codec_id == AV_CODEC_ID_INTERPLAY_VIDEO) {
280
1
            w_align = 8;
281
1
            h_align = 8;
282
        }
283
34
        break;
284
121
    case AV_PIX_FMT_PAL8:
285
    case AV_PIX_FMT_BGR8:
286
    case AV_PIX_FMT_RGB8:
287
121
        if (s->codec_id == AV_CODEC_ID_SMC ||
288
120
            s->codec_id == AV_CODEC_ID_CINEPAK) {
289
2
            w_align = 4;
290
2
            h_align = 4;
291
        }
292
121
        if (s->codec_id == AV_CODEC_ID_JV ||
293
120
            s->codec_id == AV_CODEC_ID_INTERPLAY_VIDEO) {
294
2
            w_align = 8;
295
2
            h_align = 8;
296
        }
297
121
        break;
298
72
    case AV_PIX_FMT_BGR24:
299
72
        if ((s->codec_id == AV_CODEC_ID_MSZH) ||
300
71
            (s->codec_id == AV_CODEC_ID_ZLIB)) {
301
6
            w_align = 4;
302
6
            h_align = 4;
303
        }
304
72
        break;
305
193
    case AV_PIX_FMT_RGB24:
306
193
        if (s->codec_id == AV_CODEC_ID_CINEPAK) {
307
6
            w_align = 4;
308
6
            h_align = 4;
309
        }
310
193
        break;
311
441
    default:
312
441
        break;
313
    }
314
315
8464
    if (s->codec_id == AV_CODEC_ID_IFF_ILBM) {
316
2
        w_align = FFMAX(w_align, 8);
317
    }
318
319
8464
    *width  = FFALIGN(*width, w_align);
320
8464
    *height = FFALIGN(*height, h_align);
321

8464
    if (s->codec_id == AV_CODEC_ID_H264 || s->lowres ||
322

7937
        s->codec_id == AV_CODEC_ID_VP5  || s->codec_id == AV_CODEC_ID_VP6 ||
323

7929
        s->codec_id == AV_CODEC_ID_VP6F || s->codec_id == AV_CODEC_ID_VP6A
324
    ) {
325
        // some of the optimized chroma MC reads one line too much
326
        // which is also done in mpeg decoders with lowres > 0
327
539
        *height += 2;
328
329
        // H.264 uses edge emulation for out of frame motion vectors, for this
330
        // it requires a temporary area large enough to hold a 21x21 block,
331
        // increasing witdth ensure that the temporary area is large enough,
332
        // the next rounded up width is 32
333
539
        *width = FFMAX(*width, 32);
334
    }
335
336
42320
    for (i = 0; i < 4; i++)
337
33856
        linesize_align[i] = STRIDE_ALIGN;
338
8464
}
339
340
void avcodec_align_dimensions(AVCodecContext *s, int *width, int *height)
341
{
342
    const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(s->pix_fmt);
343
    int chroma_shift = desc->log2_chroma_w;
344
    int linesize_align[AV_NUM_DATA_POINTERS];
345
    int align;
346
347
    avcodec_align_dimensions2(s, width, height, linesize_align);
348
    align               = FFMAX(linesize_align[0], linesize_align[3]);
349
    linesize_align[1] <<= chroma_shift;
350
    linesize_align[2] <<= chroma_shift;
351
    align               = FFMAX3(align, linesize_align[1], linesize_align[2]);
352
    *width              = FFALIGN(*width, align);
353
}
354
355
1
int avcodec_enum_to_chroma_pos(int *xpos, int *ypos, enum AVChromaLocation pos)
356
{
357

1
    if (pos <= AVCHROMA_LOC_UNSPECIFIED || pos >= AVCHROMA_LOC_NB)
358
        return AVERROR(EINVAL);
359
1
    pos--;
360
361
1
    *xpos = (pos&1) * 128;
362
1
    *ypos = ((pos>>1)^(pos<4)) * 128;
363
364
1
    return 0;
365
}
366
367
1
enum AVChromaLocation avcodec_chroma_pos_to_enum(int xpos, int ypos)
368
{
369
    int pos, xout, yout;
370
371
1
    for (pos = AVCHROMA_LOC_UNSPECIFIED + 1; pos < AVCHROMA_LOC_NB; pos++) {
372

1
        if (avcodec_enum_to_chroma_pos(&xout, &yout, pos) == 0 && xout == xpos && yout == ypos)
373
1
            return pos;
374
    }
375
    return AVCHROMA_LOC_UNSPECIFIED;
376
}
377
378
int avcodec_fill_audio_frame(AVFrame *frame, int nb_channels,
379
                             enum AVSampleFormat sample_fmt, const uint8_t *buf,
380
                             int buf_size, int align)
381
{
382
    int ch, planar, needed_size, ret = 0;
383
384
    needed_size = av_samples_get_buffer_size(NULL, nb_channels,
385
                                             frame->nb_samples, sample_fmt,
386
                                             align);
387
    if (buf_size < needed_size)
388
        return AVERROR(EINVAL);
389
390
    planar = av_sample_fmt_is_planar(sample_fmt);
391
    if (planar && nb_channels > AV_NUM_DATA_POINTERS) {
392
        if (!(frame->extended_data = av_mallocz_array(nb_channels,
393
                                                sizeof(*frame->extended_data))))
394
            return AVERROR(ENOMEM);
395
    } else {
396
        frame->extended_data = frame->data;
397
    }
398
399
    if ((ret = av_samples_fill_arrays(frame->extended_data, &frame->linesize[0],
400
                                      (uint8_t *)(intptr_t)buf, nb_channels, frame->nb_samples,
401
                                      sample_fmt, align)) < 0) {
402
        if (frame->extended_data != frame->data)
403
            av_freep(&frame->extended_data);
404
        return ret;
405
    }
406
    if (frame->extended_data != frame->data) {
407
        for (ch = 0; ch < AV_NUM_DATA_POINTERS; ch++)
408
            frame->data[ch] = frame->extended_data[ch];
409
    }
410
411
    return ret;
412
}
413
414
14
void ff_color_frame(AVFrame *frame, const int c[4])
415
{
416
14
    const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(frame->format);
417
    int p, y;
418
419
14
    av_assert0(desc->flags & AV_PIX_FMT_FLAG_PLANAR);
420
421
56
    for (p = 0; p<desc->nb_components; p++) {
422
42
        uint8_t *dst = frame->data[p];
423

42
        int is_chroma = p == 1 || p == 2;
424
42
        int bytes  = is_chroma ? AV_CEIL_RSHIFT(frame->width,  desc->log2_chroma_w) : frame->width;
425
42
        int height = is_chroma ? AV_CEIL_RSHIFT(frame->height, desc->log2_chroma_h) : frame->height;
426
42
        if (desc->comp[0].depth >= 9) {
427
            ((uint16_t*)dst)[0] = c[p];
428
            av_memcpy_backptr(dst + 2, 2, bytes - 2);
429
            dst += frame->linesize[p];
430
            for (y = 1; y < height; y++) {
431
                memcpy(dst, frame->data[p], 2*bytes);
432
                dst += frame->linesize[p];
433
            }
434
        } else {
435
18666
            for (y = 0; y < height; y++) {
436
18624
                memset(dst, c[p], bytes);
437
18624
                dst += frame->linesize[p];
438
            }
439
        }
440
    }
441
14
}
442
443
46936
int avcodec_default_execute(AVCodecContext *c, int (*func)(AVCodecContext *c2, void *arg2), void *arg, int *ret, int count, int size)
444
{
445
    int i;
446
447
1380084
    for (i = 0; i < count; i++) {
448
1333148
        int r = func(c, (char *)arg + i * size);
449
1333148
        if (ret)
450
28391
            ret[i] = r;
451
    }
452
46936
    emms_c();
453
46936
    return 0;
454
}
455
456
6850
int avcodec_default_execute2(AVCodecContext *c, int (*func)(AVCodecContext *c2, void *arg2, int jobnr, int threadnr), void *arg, int *ret, int count)
457
{
458
    int i;
459
460
210060
    for (i = 0; i < count; i++) {
461
203210
        int r = func(c, arg, i, 0);
462
203210
        if (ret)
463
50
            ret[i] = r;
464
    }
465
6850
    emms_c();
466
6850
    return 0;
467
}
468
469
1362
enum AVPixelFormat avpriv_find_pix_fmt(const PixelFormatTag *tags,
470
                                       unsigned int fourcc)
471
{
472
12588
    while (tags->pix_fmt >= 0) {
473
12390
        if (tags->fourcc == fourcc)
474
1164
            return tags->pix_fmt;
475
11226
        tags++;
476
    }
477
198
    return AV_PIX_FMT_NONE;
478
}
479
480
#if FF_API_CODEC_GET_SET
481
MAKE_ACCESSORS(AVCodecContext, codec, AVRational, pkt_timebase)
482
MAKE_ACCESSORS(AVCodecContext, codec, const AVCodecDescriptor *, codec_descriptor)
483
MAKE_ACCESSORS(AVCodecContext, codec, int, lowres)
484
MAKE_ACCESSORS(AVCodecContext, codec, int, seek_preroll)
485
MAKE_ACCESSORS(AVCodecContext, codec, uint16_t*, chroma_intra_matrix)
486
487
unsigned av_codec_get_codec_properties(const AVCodecContext *codec)
488
{
489
    return codec->properties;
490
}
491
492
int av_codec_get_max_lowres(const AVCodec *codec)
493
{
494
    return codec->max_lowres;
495
}
496
#endif
497
498
164717
int avpriv_codec_get_cap_skip_frame_fill_param(const AVCodec *codec){
499
164717
    return !!(codec->caps_internal & FF_CODEC_CAP_SKIP_FRAME_FILL_PARAM);
500
}
501
502
21102
static int64_t get_bit_rate(AVCodecContext *ctx)
503
{
504
    int64_t bit_rate;
505
    int bits_per_sample;
506
507
21102
    switch (ctx->codec_type) {
508
17622
    case AVMEDIA_TYPE_VIDEO:
509
    case AVMEDIA_TYPE_DATA:
510
    case AVMEDIA_TYPE_SUBTITLE:
511
    case AVMEDIA_TYPE_ATTACHMENT:
512
17622
        bit_rate = ctx->bit_rate;
513
17622
        break;
514
3480
    case AVMEDIA_TYPE_AUDIO:
515
3480
        bits_per_sample = av_get_bits_per_sample(ctx->codec_id);
516
3480
        bit_rate = bits_per_sample ? ctx->sample_rate * (int64_t)ctx->channels * bits_per_sample : ctx->bit_rate;
517
3480
        break;
518
    default:
519
        bit_rate = 0;
520
        break;
521
    }
522
21102
    return bit_rate;
523
}
524
525
526
24652
static void ff_lock_avcodec(AVCodecContext *log_ctx, const AVCodec *codec)
527
{
528

24652
    if (!(codec->caps_internal & FF_CODEC_CAP_INIT_THREADSAFE) && codec->init)
529
17549
        ff_mutex_lock(&codec_mutex);
530
24652
}
531
532
24652
static void ff_unlock_avcodec(const AVCodec *codec)
533
{
534

24652
    if (!(codec->caps_internal & FF_CODEC_CAP_INIT_THREADSAFE) && codec->init)
535
17549
        ff_mutex_unlock(&codec_mutex);
536
24652
}
537
538
13
int attribute_align_arg ff_codec_open2_recursive(AVCodecContext *avctx, const AVCodec *codec, AVDictionary **options)
539
{
540
13
    int ret = 0;
541
542
13
    ff_unlock_avcodec(codec);
543
544
13
    ret = avcodec_open2(avctx, codec, options);
545
546
13
    ff_lock_avcodec(avctx, codec);
547
13
    return ret;
548
}
549
550
18649
int attribute_align_arg avcodec_open2(AVCodecContext *avctx, const AVCodec *codec, AVDictionary **options)
551
{
552
18649
    int ret = 0;
553
18649
    int codec_init_ok = 0;
554
18649
    AVDictionary *tmp = NULL;
555
    const AVPixFmtDescriptor *pixdesc;
556
    AVCodecInternal *avci;
557
558
18649
    if (avcodec_is_open(avctx))
559
        return 0;
560
561

18649
    if (!codec && !avctx->codec) {
562
        av_log(avctx, AV_LOG_ERROR, "No codec provided to avcodec_open2()\n");
563
        return AVERROR(EINVAL);
564
    }
565

18649
    if (codec && avctx->codec && codec != avctx->codec) {
566
        av_log(avctx, AV_LOG_ERROR, "This AVCodecContext was allocated for %s, "
567
                                    "but %s passed to avcodec_open2()\n", avctx->codec->name, codec->name);
568
        return AVERROR(EINVAL);
569
    }
570
18649
    if (!codec)
571
2
        codec = avctx->codec;
572
573

18649
    if (avctx->extradata_size < 0 || avctx->extradata_size >= FF_MAX_EXTRADATA_SIZE)
574
        return AVERROR(EINVAL);
575
576
18649
    if (options)
577
18567
        av_dict_copy(&tmp, *options, 0);
578
579
18649
    ff_lock_avcodec(avctx, codec);
580
581
18649
    avci = av_mallocz(sizeof(*avci));
582
18649
    if (!avci) {
583
        ret = AVERROR(ENOMEM);
584
        goto end;
585
    }
586
18649
    avctx->internal = avci;
587
588
18649
    avci->to_free = av_frame_alloc();
589
18649
    avci->compat_decode_frame = av_frame_alloc();
590
18649
    avci->compat_encode_packet = av_packet_alloc();
591
18649
    avci->buffer_frame = av_frame_alloc();
592
18649
    avci->buffer_pkt = av_packet_alloc();
593
18649
    avci->es.in_frame = av_frame_alloc();
594
18649
    avci->ds.in_pkt = av_packet_alloc();
595
18649
    avci->last_pkt_props = av_packet_alloc();
596

18649
    if (!avci->compat_decode_frame || !avci->compat_encode_packet ||
597

18649
        !avci->buffer_frame || !avci->buffer_pkt          ||
598

18649
        !avci->es.in_frame  || !avci->ds.in_pkt           ||
599

18649
        !avci->to_free      || !avci->last_pkt_props) {
600
        ret = AVERROR(ENOMEM);
601
        goto free_and_end;
602
    }
603
604
18649
    avci->skip_samples_multiplier = 1;
605
606
18649
    if (codec->priv_data_size > 0) {
607
13252
        if (!avctx->priv_data) {
608
6597
            avctx->priv_data = av_mallocz(codec->priv_data_size);
609
6597
            if (!avctx->priv_data) {
610
                ret = AVERROR(ENOMEM);
611
                goto free_and_end;
612
            }
613
6597
            if (codec->priv_class) {
614
1675
                *(const AVClass **)avctx->priv_data = codec->priv_class;
615
1675
                av_opt_set_defaults(avctx->priv_data);
616
            }
617
        }
618

13252
        if (codec->priv_class && (ret = av_opt_set_dict(avctx->priv_data, &tmp)) < 0)
619
            goto free_and_end;
620
    } else {
621
5397
        avctx->priv_data = NULL;
622
    }
623
18649
    if ((ret = av_opt_set_dict(avctx, &tmp)) < 0)
624
        goto free_and_end;
625
626

18649
    if (avctx->codec_whitelist && av_match_list(codec->name, avctx->codec_whitelist, ',') <= 0) {
627
        av_log(avctx, AV_LOG_ERROR, "Codec (%s) not on whitelist \'%s\'\n", codec->name, avctx->codec_whitelist);
628
        ret = AVERROR(EINVAL);
629
        goto free_and_end;
630
    }
631
632
    // only call ff_set_dimensions() for non H.264/VP6F/DXV codecs so as not to overwrite previously setup dimensions
633


18649
    if (!(avctx->coded_width && avctx->coded_height && avctx->width && avctx->height &&
634

192
          (avctx->codec_id == AV_CODEC_ID_H264 || avctx->codec_id == AV_CODEC_ID_VP6F || avctx->codec_id == AV_CODEC_ID_DXV))) {
635

18638
        if (avctx->coded_width && avctx->coded_height)
636
191
            ret = ff_set_dimensions(avctx, avctx->coded_width, avctx->coded_height);
637

18447
        else if (avctx->width && avctx->height)
638
11130
            ret = ff_set_dimensions(avctx, avctx->width, avctx->height);
639
18638
        if (ret < 0)
640
            goto free_and_end;
641
    }
642
643


18649
    if ((avctx->coded_width || avctx->coded_height || avctx->width || avctx->height)
644
11332
        && (  av_image_check_size2(avctx->coded_width, avctx->coded_height, avctx->max_pixels, AV_PIX_FMT_NONE, 0, avctx) < 0
645
11332
           || av_image_check_size2(avctx->width,       avctx->height,       avctx->max_pixels, AV_PIX_FMT_NONE, 0, avctx) < 0)) {
646
        av_log(avctx, AV_LOG_WARNING, "Ignoring invalid width/height values\n");
647
        ff_set_dimensions(avctx, 0, 0);
648
    }
649
650

18649
    if (avctx->width > 0 && avctx->height > 0) {
651
11332
        if (av_image_check_sar(avctx->width, avctx->height,
652
                               avctx->sample_aspect_ratio) < 0) {
653
            av_log(avctx, AV_LOG_WARNING, "ignoring invalid SAR: %u/%u\n",
654
                   avctx->sample_aspect_ratio.num,
655
                   avctx->sample_aspect_ratio.den);
656
            avctx->sample_aspect_ratio = (AVRational){ 0, 1 };
657
        }
658
    }
659
660
    /* if the decoder init function was already called previously,
661
     * free the already allocated subtitle_header before overwriting it */
662
18649
    if (av_codec_is_decoder(codec))
663
12659
        av_freep(&avctx->subtitle_header);
664
665

18649
    if (avctx->channels > FF_SANE_NB_CHANNELS || avctx->channels < 0) {
666
        av_log(avctx, AV_LOG_ERROR, "Too many or invalid channels: %d\n", avctx->channels);
667
        ret = AVERROR(EINVAL);
668
        goto free_and_end;
669
    }
670
18649
    if (avctx->sample_rate < 0) {
671
        av_log(avctx, AV_LOG_ERROR, "Invalid sample rate: %d\n", avctx->sample_rate);
672
        ret = AVERROR(EINVAL);
673
        goto free_and_end;
674
    }
675
18649
    if (avctx->block_align < 0) {
676
        av_log(avctx, AV_LOG_ERROR, "Invalid block align: %d\n", avctx->block_align);
677
        ret = AVERROR(EINVAL);
678
        goto free_and_end;
679
    }
680
681
18649
    avctx->codec = codec;
682

18649
    if ((avctx->codec_type == AVMEDIA_TYPE_UNKNOWN || avctx->codec_type == codec->type) &&
683
18649
        avctx->codec_id == AV_CODEC_ID_NONE) {
684
        avctx->codec_type = codec->type;
685
        avctx->codec_id   = codec->id;
686
    }
687

18649
    if (avctx->codec_id != codec->id || (avctx->codec_type != codec->type
688
                                         && avctx->codec_type != AVMEDIA_TYPE_ATTACHMENT)) {
689
        av_log(avctx, AV_LOG_ERROR, "Codec type or id mismatches\n");
690
        ret = AVERROR(EINVAL);
691
        goto free_and_end;
692
    }
693
18649
    avctx->frame_number = 0;
694
18649
    avctx->codec_descriptor = avcodec_descriptor_get(avctx->codec_id);
695
696
18649
    if ((avctx->codec->capabilities & AV_CODEC_CAP_EXPERIMENTAL) &&
697
7
        avctx->strict_std_compliance > FF_COMPLIANCE_EXPERIMENTAL) {
698
        const char *codec_string = av_codec_is_encoder(codec) ? "encoder" : "decoder";
699
        const AVCodec *codec2;
700
        av_log(avctx, AV_LOG_ERROR,
701
               "The %s '%s' is experimental but experimental codecs are not enabled, "
702
               "add '-strict %d' if you want to use it.\n",
703
               codec_string, codec->name, FF_COMPLIANCE_EXPERIMENTAL);
704
        codec2 = av_codec_is_encoder(codec) ? avcodec_find_encoder(codec->id) : avcodec_find_decoder(codec->id);
705
        if (!(codec2->capabilities & AV_CODEC_CAP_EXPERIMENTAL))
706
            av_log(avctx, AV_LOG_ERROR, "Alternatively use the non experimental %s '%s'.\n",
707
                codec_string, codec2->name);
708
        ret = AVERROR_EXPERIMENTAL;
709
        goto free_and_end;
710
    }
711
712
18649
    if (avctx->codec_type == AVMEDIA_TYPE_AUDIO &&
713

3894
        (!avctx->time_base.num || !avctx->time_base.den)) {
714
2774
        avctx->time_base.num = 1;
715
2774
        avctx->time_base.den = avctx->sample_rate;
716
    }
717
718
    if (!HAVE_THREADS)
719
        av_log(avctx, AV_LOG_WARNING, "Warning: not compiled with thread support, using thread emulation\n");
720
721
18649
    if (CONFIG_FRAME_THREAD_ENCODER && av_codec_is_encoder(avctx->codec)) {
722
5990
        ff_unlock_avcodec(codec); //we will instantiate a few encoders thus kick the counter to prevent false detection of a problem
723
5990
        ret = ff_frame_thread_encoder_init(avctx, options ? *options : NULL);
724
5990
        ff_lock_avcodec(avctx, codec);
725
5990
        if (ret < 0)
726
            goto free_and_end;
727
    }
728
729
18649
    if (av_codec_is_decoder(avctx->codec)) {
730
12659
        ret = ff_decode_bsfs_init(avctx);
731
12659
        if (ret < 0)
732
            goto free_and_end;
733
    }
734
735
18649
    if (HAVE_THREADS
736

18649
        && !(avci->frame_thread_encoder && (avctx->active_thread_type&FF_THREAD_FRAME))) {
737
18631
        ret = ff_thread_init(avctx);
738
18631
        if (ret < 0) {
739
            goto free_and_end;
740
        }
741
    }
742
    if (!HAVE_THREADS && !(codec->capabilities & AV_CODEC_CAP_AUTO_THREADS))
743
        avctx->thread_count = 1;
744
745

18649
    if (avctx->codec->max_lowres < avctx->lowres || avctx->lowres < 0) {
746
        av_log(avctx, AV_LOG_WARNING, "The maximum value for lowres supported by the decoder is %d\n",
747
               avctx->codec->max_lowres);
748
        avctx->lowres = avctx->codec->max_lowres;
749
    }
750
751
18649
    if (av_codec_is_encoder(avctx->codec)) {
752
        int i;
753
#if FF_API_CODED_FRAME
754
FF_DISABLE_DEPRECATION_WARNINGS
755
5990
        avctx->coded_frame = av_frame_alloc();
756
5990
        if (!avctx->coded_frame) {
757
            ret = AVERROR(ENOMEM);
758
            goto free_and_end;
759
        }
760
FF_ENABLE_DEPRECATION_WARNINGS
761
#endif
762
763

5990
        if (avctx->time_base.num <= 0 || avctx->time_base.den <= 0) {
764
            av_log(avctx, AV_LOG_ERROR, "The encoder timebase is not set.\n");
765
            ret = AVERROR(EINVAL);
766
            goto free_and_end;
767
        }
768
769
5990
        if (avctx->codec->sample_fmts) {
770
1148
            for (i = 0; avctx->codec->sample_fmts[i] != AV_SAMPLE_FMT_NONE; i++) {
771
1148
                if (avctx->sample_fmt == avctx->codec->sample_fmts[i])
772
1136
                    break;
773

15
                if (avctx->channels == 1 &&
774
3
                    av_get_planar_sample_fmt(avctx->sample_fmt) ==
775
3
                    av_get_planar_sample_fmt(avctx->codec->sample_fmts[i])) {
776
                    avctx->sample_fmt = avctx->codec->sample_fmts[i];
777
                    break;
778
                }
779
            }
780
1136
            if (avctx->codec->sample_fmts[i] == AV_SAMPLE_FMT_NONE) {
781
                char buf[128];
782
                snprintf(buf, sizeof(buf), "%d", avctx->sample_fmt);
783
                av_log(avctx, AV_LOG_ERROR, "Specified sample format %s is invalid or not supported\n",
784
                       (char *)av_x_if_null(av_get_sample_fmt_name(avctx->sample_fmt), buf));
785
                ret = AVERROR(EINVAL);
786
                goto free_and_end;
787
            }
788
        }
789
5990
        if (avctx->codec->pix_fmts) {
790
2076
            for (i = 0; avctx->codec->pix_fmts[i] != AV_PIX_FMT_NONE; i++)
791
2076
                if (avctx->pix_fmt == avctx->codec->pix_fmts[i])
792
697
                    break;
793
697
            if (avctx->codec->pix_fmts[i] == AV_PIX_FMT_NONE
794
                && !((avctx->codec_id == AV_CODEC_ID_MJPEG || avctx->codec_id == AV_CODEC_ID_LJPEG)
795
                     && avctx->strict_std_compliance <= FF_COMPLIANCE_UNOFFICIAL)) {
796
                char buf[128];
797
                snprintf(buf, sizeof(buf), "%d", avctx->pix_fmt);
798
                av_log(avctx, AV_LOG_ERROR, "Specified pixel format %s is invalid or not supported\n",
799
                       (char *)av_x_if_null(av_get_pix_fmt_name(avctx->pix_fmt), buf));
800
                ret = AVERROR(EINVAL);
801
                goto free_and_end;
802
            }
803
697
            if (avctx->codec->pix_fmts[i] == AV_PIX_FMT_YUVJ420P ||
804
675
                avctx->codec->pix_fmts[i] == AV_PIX_FMT_YUVJ411P ||
805
675
                avctx->codec->pix_fmts[i] == AV_PIX_FMT_YUVJ422P ||
806
671
                avctx->codec->pix_fmts[i] == AV_PIX_FMT_YUVJ440P ||
807
671
                avctx->codec->pix_fmts[i] == AV_PIX_FMT_YUVJ444P)
808
34
                avctx->color_range = AVCOL_RANGE_JPEG;
809
        }
810
5990
        if (avctx->codec->supported_samplerates) {
811
103
            for (i = 0; avctx->codec->supported_samplerates[i] != 0; i++)
812
103
                if (avctx->sample_rate == avctx->codec->supported_samplerates[i])
813
44
                    break;
814
44
            if (avctx->codec->supported_samplerates[i] == 0) {
815
                av_log(avctx, AV_LOG_ERROR, "Specified sample rate %d is not supported\n",
816
                       avctx->sample_rate);
817
                ret = AVERROR(EINVAL);
818
                goto free_and_end;
819
            }
820
        }
821
5990
        if (avctx->sample_rate < 0) {
822
            av_log(avctx, AV_LOG_ERROR, "Specified sample rate %d is not supported\n",
823
                    avctx->sample_rate);
824
            ret = AVERROR(EINVAL);
825
            goto free_and_end;
826
        }
827
5990
        if (avctx->codec->channel_layouts) {
828
42
            if (!avctx->channel_layout) {
829
                av_log(avctx, AV_LOG_WARNING, "Channel layout not specified\n");
830
            } else {
831
58
                for (i = 0; avctx->codec->channel_layouts[i] != 0; i++)
832
58
                    if (avctx->channel_layout == avctx->codec->channel_layouts[i])
833
42
                        break;
834
42
                if (avctx->codec->channel_layouts[i] == 0) {
835
                    char buf[512];
836
                    av_get_channel_layout_string(buf, sizeof(buf), -1, avctx->channel_layout);
837
                    av_log(avctx, AV_LOG_ERROR, "Specified channel layout '%s' is not supported\n", buf);
838
                    ret = AVERROR(EINVAL);
839
                    goto free_and_end;
840
                }
841
            }
842
        }
843

7102
        if (avctx->channel_layout && avctx->channels) {
844
1112
            int channels = av_get_channel_layout_nb_channels(avctx->channel_layout);
845
1112
            if (channels != avctx->channels) {
846
                char buf[512];
847
                av_get_channel_layout_string(buf, sizeof(buf), -1, avctx->channel_layout);
848
                av_log(avctx, AV_LOG_ERROR,
849
                       "Channel layout '%s' with %d channels does not match number of specified channels %d\n",
850
                       buf, channels, avctx->channels);
851
                ret = AVERROR(EINVAL);
852
                goto free_and_end;
853
            }
854
4878
        } else if (avctx->channel_layout) {
855
16
            avctx->channels = av_get_channel_layout_nb_channels(avctx->channel_layout);
856
        }
857
5990
        if (avctx->channels < 0) {
858
            av_log(avctx, AV_LOG_ERROR, "Specified number of channels %d is not supported\n",
859
                    avctx->channels);
860
            ret = AVERROR(EINVAL);
861
            goto free_and_end;
862
        }
863
5990
        if(avctx->codec_type == AVMEDIA_TYPE_VIDEO) {
864
4823
            pixdesc = av_pix_fmt_desc_get(avctx->pix_fmt);
865
4823
            if (    avctx->bits_per_raw_sample < 0
866

4823
                || (avctx->bits_per_raw_sample > 8 && pixdesc->comp[0].depth <= 8)) {
867
                av_log(avctx, AV_LOG_WARNING, "Specified bit depth %d not possible with the specified pixel formats depth %d\n",
868
                    avctx->bits_per_raw_sample, pixdesc->comp[0].depth);
869
                avctx->bits_per_raw_sample = pixdesc->comp[0].depth;
870
            }
871

4823
            if (avctx->width <= 0 || avctx->height <= 0) {
872
                av_log(avctx, AV_LOG_ERROR, "dimensions not set\n");
873
                ret = AVERROR(EINVAL);
874
                goto free_and_end;
875
            }
876
        }
877

5990
        if (   (avctx->codec_type == AVMEDIA_TYPE_VIDEO || avctx->codec_type == AVMEDIA_TYPE_AUDIO)
878

5959
            && avctx->bit_rate>0 && avctx->bit_rate<1000) {
879
            av_log(avctx, AV_LOG_WARNING, "Bitrate %"PRId64" is extremely low, maybe you mean %"PRId64"k\n", avctx->bit_rate, avctx->bit_rate);
880
        }
881
882
5990
        if (!avctx->rc_initial_buffer_occupancy)
883
5989
            avctx->rc_initial_buffer_occupancy = avctx->rc_buffer_size * 3LL / 4;
884
885

5990
        if (avctx->ticks_per_frame && avctx->time_base.num &&
886
5959
            avctx->ticks_per_frame > INT_MAX / avctx->time_base.num) {
887
            av_log(avctx, AV_LOG_ERROR,
888
                   "ticks_per_frame %d too large for the timebase %d/%d.",
889
                   avctx->ticks_per_frame,
890
                   avctx->time_base.num,
891
                   avctx->time_base.den);
892
            goto free_and_end;
893
        }
894
895
5990
        if (avctx->hw_frames_ctx) {
896
            AVHWFramesContext *frames_ctx = (AVHWFramesContext*)avctx->hw_frames_ctx->data;
897
            if (frames_ctx->format != avctx->pix_fmt) {
898
                av_log(avctx, AV_LOG_ERROR,
899
                       "Mismatching AVCodecContext.pix_fmt and AVHWFramesContext.format\n");
900
                ret = AVERROR(EINVAL);
901
                goto free_and_end;
902
            }
903
            if (avctx->sw_pix_fmt != AV_PIX_FMT_NONE &&
904
                avctx->sw_pix_fmt != frames_ctx->sw_format) {
905
                av_log(avctx, AV_LOG_ERROR,
906
                       "Mismatching AVCodecContext.sw_pix_fmt (%s) "
907
                       "and AVHWFramesContext.sw_format (%s)\n",
908
                       av_get_pix_fmt_name(avctx->sw_pix_fmt),
909
                       av_get_pix_fmt_name(frames_ctx->sw_format));
910
                ret = AVERROR(EINVAL);
911
                goto free_and_end;
912
            }
913
            avctx->sw_pix_fmt = frames_ctx->sw_format;
914
        }
915
    }
916
917
18649
    avctx->pts_correction_num_faulty_pts =
918
18649
    avctx->pts_correction_num_faulty_dts = 0;
919
18649
    avctx->pts_correction_last_pts =
920
18649
    avctx->pts_correction_last_dts = INT64_MIN;
921
922
18649
    if (   !CONFIG_GRAY && avctx->flags & AV_CODEC_FLAG_GRAY
923
        && avctx->codec_descriptor->type == AVMEDIA_TYPE_VIDEO)
924
        av_log(avctx, AV_LOG_WARNING,
925
               "gray decoding requested but not enabled at configuration time\n");
926
18649
    if (avctx->flags2 & AV_CODEC_FLAG2_EXPORT_MVS) {
927
5
        avctx->export_side_data |= AV_CODEC_EXPORT_DATA_MVS;
928
    }
929
930

18649
    if (   avctx->codec->init && (!(avctx->active_thread_type&FF_THREAD_FRAME)
931
30
        || avci->frame_thread_encoder)) {
932
13584
        ret = avctx->codec->init(avctx);
933
13584
        if (ret < 0) {
934
8
            codec_init_ok = -1;
935
8
            goto free_and_end;
936
        }
937
13576
        codec_init_ok = 1;
938
    }
939
940
18641
    ret=0;
941
942
18641
    if (av_codec_is_decoder(avctx->codec)) {
943
12651
        if (!avctx->bit_rate)
944
8506
            avctx->bit_rate = get_bit_rate(avctx);
945
        /* validate channel layout from the decoder */
946
12651
        if (avctx->channel_layout) {
947
1671
            int channels = av_get_channel_layout_nb_channels(avctx->channel_layout);
948
1671
            if (!avctx->channels)
949
16
                avctx->channels = channels;
950
1655
            else if (channels != avctx->channels) {
951
                char buf[512];
952
7
                av_get_channel_layout_string(buf, sizeof(buf), -1, avctx->channel_layout);
953
7
                av_log(avctx, AV_LOG_WARNING,
954
                       "Channel layout '%s' with %d channels does not match specified number of channels %d: "
955
                       "ignoring specified channel layout\n",
956
                       buf, channels, avctx->channels);
957
7
                avctx->channel_layout = 0;
958
            }
959
        }
960

12651
        if (avctx->channels && avctx->channels < 0 ||
961
12651
            avctx->channels > FF_SANE_NB_CHANNELS) {
962
            ret = AVERROR(EINVAL);
963
            goto free_and_end;
964
        }
965
12651
        if (avctx->bits_per_coded_sample < 0) {
966
            ret = AVERROR(EINVAL);
967
            goto free_and_end;
968
        }
969
12651
        if (avctx->sub_charenc) {
970
6
            if (avctx->codec_type != AVMEDIA_TYPE_SUBTITLE) {
971
                av_log(avctx, AV_LOG_ERROR, "Character encoding is only "
972
                       "supported with subtitles codecs\n");
973
                ret = AVERROR(EINVAL);
974
                goto free_and_end;
975
6
            } else if (avctx->codec_descriptor->props & AV_CODEC_PROP_BITMAP_SUB) {
976
                av_log(avctx, AV_LOG_WARNING, "Codec '%s' is bitmap-based, "
977
                       "subtitles character encoding will be ignored\n",
978
                       avctx->codec_descriptor->name);
979
                avctx->sub_charenc_mode = FF_SUB_CHARENC_MODE_DO_NOTHING;
980
            } else {
981
                /* input character encoding is set for a text based subtitle
982
                 * codec at this point */
983
6
                if (avctx->sub_charenc_mode == FF_SUB_CHARENC_MODE_AUTOMATIC)
984
6
                    avctx->sub_charenc_mode = FF_SUB_CHARENC_MODE_PRE_DECODER;
985
986
6
                if (avctx->sub_charenc_mode == FF_SUB_CHARENC_MODE_PRE_DECODER) {
987
#if CONFIG_ICONV
988
6
                    iconv_t cd = iconv_open("UTF-8", avctx->sub_charenc);
989
6
                    if (cd == (iconv_t)-1) {
990
                        ret = AVERROR(errno);
991
                        av_log(avctx, AV_LOG_ERROR, "Unable to open iconv context "
992
                               "with input character encoding \"%s\"\n", avctx->sub_charenc);
993
                        goto free_and_end;
994
                    }
995
6
                    iconv_close(cd);
996
#else
997
                    av_log(avctx, AV_LOG_ERROR, "Character encoding subtitles "
998
                           "conversion needs a libavcodec built with iconv support "
999
                           "for this codec\n");
1000
                    ret = AVERROR(ENOSYS);
1001
                    goto free_and_end;
1002
#endif
1003
                }
1004
            }
1005
        }
1006
1007
#if FF_API_AVCTX_TIMEBASE
1008

12651
        if (avctx->framerate.num > 0 && avctx->framerate.den > 0)
1009
4388
            avctx->time_base = av_inv_q(av_mul_q(avctx->framerate, (AVRational){avctx->ticks_per_frame, 1}));
1010
#endif
1011
    }
1012

18641
    if (codec->priv_data_size > 0 && avctx->priv_data && codec->priv_class) {
1013
3861
        av_assert0(*(const AVClass **)avctx->priv_data == codec->priv_class);
1014
    }
1015
1016
18641
end:
1017
18649
    ff_unlock_avcodec(codec);
1018
18649
    if (options) {
1019
18567
        av_dict_free(options);
1020
18567
        *options = tmp;
1021
    }
1022
1023
18649
    return ret;
1024
8
free_and_end:
1025

8
    if (avctx->codec && avctx->codec->close &&
1026
6
        (codec_init_ok > 0 || (codec_init_ok < 0 &&
1027
6
         avctx->codec->caps_internal & FF_CODEC_CAP_INIT_CLEANUP)))
1028
        avctx->codec->close(avctx);
1029
1030
8
    if (HAVE_THREADS && avci->thread_ctx)
1031
        ff_thread_free(avctx);
1032
1033

8
    if (codec->priv_class && avctx->priv_data)
1034
        av_opt_free(avctx->priv_data);
1035
8
    av_opt_free(avctx);
1036
1037
8
    if (av_codec_is_encoder(avctx->codec)) {
1038
#if FF_API_CODED_FRAME
1039
FF_DISABLE_DEPRECATION_WARNINGS
1040
        av_frame_free(&avctx->coded_frame);
1041
FF_ENABLE_DEPRECATION_WARNINGS
1042
#endif
1043
        av_freep(&avctx->extradata);
1044
        avctx->extradata_size = 0;
1045
    }
1046
1047
8
    av_dict_free(&tmp);
1048
8
    av_freep(&avctx->priv_data);
1049
8
    av_freep(&avctx->subtitle_header);
1050
1051
8
    av_frame_free(&avci->to_free);
1052
8
    av_frame_free(&avci->compat_decode_frame);
1053
8
    av_frame_free(&avci->buffer_frame);
1054
8
    av_packet_free(&avci->compat_encode_packet);
1055
8
    av_packet_free(&avci->buffer_pkt);
1056
8
    av_packet_free(&avci->last_pkt_props);
1057
1058
8
    av_packet_free(&avci->ds.in_pkt);
1059
8
    av_frame_free(&avci->es.in_frame);
1060
8
    av_bsf_free(&avci->bsf);
1061
1062
8
    av_buffer_unref(&avci->pool);
1063
8
    av_freep(&avci);
1064
8
    avctx->internal = NULL;
1065
8
    avctx->codec = NULL;
1066
8
    goto end;
1067
}
1068
1069
28
void avcodec_flush_buffers(AVCodecContext *avctx)
1070
{
1071
28
    AVCodecInternal *avci = avctx->internal;
1072
1073
28
    if (av_codec_is_encoder(avctx->codec)) {
1074
        int caps = avctx->codec->capabilities;
1075
1076
        if (!(caps & AV_CODEC_CAP_ENCODER_FLUSH)) {
1077
            // Only encoders that explicitly declare support for it can be
1078
            // flushed. Otherwise, this is a no-op.
1079
            av_log(avctx, AV_LOG_WARNING, "Ignoring attempt to flush encoder "
1080
                   "that doesn't support it\n");
1081
            return;
1082
        }
1083
1084
        // We haven't implemented flushing for frame-threaded encoders.
1085
        av_assert0(!(caps & AV_CODEC_CAP_FRAME_THREADS));
1086
    }
1087
1088
28
    avci->draining      = 0;
1089
28
    avci->draining_done = 0;
1090
28
    avci->nb_draining_errors = 0;
1091
28
    av_frame_unref(avci->buffer_frame);
1092
28
    av_frame_unref(avci->compat_decode_frame);
1093
28
    av_packet_unref(avci->compat_encode_packet);
1094
28
    av_packet_unref(avci->buffer_pkt);
1095
1096
28
    av_frame_unref(avci->es.in_frame);
1097
28
    av_packet_unref(avci->ds.in_pkt);
1098
1099
28
    if (HAVE_THREADS && avctx->active_thread_type & FF_THREAD_FRAME)
1100
        ff_thread_flush(avctx);
1101
28
    else if (avctx->codec->flush)
1102
        avctx->codec->flush(avctx);
1103
1104
28
    avctx->pts_correction_last_pts =
1105
28
    avctx->pts_correction_last_dts = INT64_MIN;
1106
1107
28
    if (av_codec_is_decoder(avctx->codec))
1108
28
        av_bsf_flush(avci->bsf);
1109
1110
28
    if (!avctx->refcounted_frames)
1111
28
        av_frame_unref(avci->to_free);
1112
}
1113
1114
7016
void avsubtitle_free(AVSubtitle *sub)
1115
{
1116
    int i;
1117
1118
7723
    for (i = 0; i < sub->num_rects; i++) {
1119
707
        av_freep(&sub->rects[i]->data[0]);
1120
707
        av_freep(&sub->rects[i]->data[1]);
1121
707
        av_freep(&sub->rects[i]->data[2]);
1122
707
        av_freep(&sub->rects[i]->data[3]);
1123
707
        av_freep(&sub->rects[i]->text);
1124
707
        av_freep(&sub->rects[i]->ass);
1125
707
        av_freep(&sub->rects[i]);
1126
    }
1127
1128
7016
    av_freep(&sub->rects);
1129
1130
7016
    memset(sub, 0, sizeof(*sub));
1131
7016
}
1132
1133
63900
av_cold int avcodec_close(AVCodecContext *avctx)
1134
{
1135
    int i;
1136
1137
63900
    if (!avctx)
1138
        return 0;
1139
1140
63900
    if (avcodec_is_open(avctx)) {
1141
18641
        if (CONFIG_FRAME_THREAD_ENCODER &&
1142

18641
            avctx->internal->frame_thread_encoder && avctx->thread_count > 1) {
1143
18
            ff_frame_thread_encoder_free(avctx);
1144
        }
1145
18641
        if (HAVE_THREADS && avctx->internal->thread_ctx)
1146
28
            ff_thread_free(avctx);
1147

18641
        if (avctx->codec && avctx->codec->close)
1148
6084
            avctx->codec->close(avctx);
1149
18641
        avctx->internal->byte_buffer_size = 0;
1150
18641
        av_freep(&avctx->internal->byte_buffer);
1151
18641
        av_frame_free(&avctx->internal->to_free);
1152
18641
        av_frame_free(&avctx->internal->compat_decode_frame);
1153
18641
        av_frame_free(&avctx->internal->buffer_frame);
1154
18641
        av_packet_free(&avctx->internal->compat_encode_packet);
1155
18641
        av_packet_free(&avctx->internal->buffer_pkt);
1156
18641
        av_packet_free(&avctx->internal->last_pkt_props);
1157
18641
        avpriv_packet_list_free(&avctx->internal->pkt_props,
1158
18641
                                &avctx->internal->pkt_props_tail);
1159
1160
18641
        av_packet_free(&avctx->internal->ds.in_pkt);
1161
18641
        av_frame_free(&avctx->internal->es.in_frame);
1162
1163
18641
        av_buffer_unref(&avctx->internal->pool);
1164
1165

18641
        if (avctx->hwaccel && avctx->hwaccel->uninit)
1166
            avctx->hwaccel->uninit(avctx);
1167
18641
        av_freep(&avctx->internal->hwaccel_priv_data);
1168
1169
18641
        av_bsf_free(&avctx->internal->bsf);
1170
1171
18641
        av_freep(&avctx->internal);
1172
    }
1173
1174
64294
    for (i = 0; i < avctx->nb_coded_side_data; i++)
1175
394
        av_freep(&avctx->coded_side_data[i].data);
1176
63900
    av_freep(&avctx->coded_side_data);
1177
63900
    avctx->nb_coded_side_data = 0;
1178
1179
63900
    av_buffer_unref(&avctx->hw_frames_ctx);
1180
63900
    av_buffer_unref(&avctx->hw_device_ctx);
1181
1182

63900
    if (avctx->priv_data && avctx->codec && avctx->codec->priv_class)
1183
4015
        av_opt_free(avctx->priv_data);
1184
63900
    av_opt_free(avctx);
1185
63900
    av_freep(&avctx->priv_data);
1186
63900
    if (av_codec_is_encoder(avctx->codec)) {
1187
17652
        av_freep(&avctx->extradata);
1188
#if FF_API_CODED_FRAME
1189
FF_DISABLE_DEPRECATION_WARNINGS
1190
17652
        av_frame_free(&avctx->coded_frame);
1191
FF_ENABLE_DEPRECATION_WARNINGS
1192
#endif
1193
    }
1194
63900
    avctx->codec = NULL;
1195
63900
    avctx->active_thread_type = 0;
1196
1197
63900
    return 0;
1198
}
1199
1200
16194
const char *avcodec_get_name(enum AVCodecID id)
1201
{
1202
    const AVCodecDescriptor *cd;
1203
    const AVCodec *codec;
1204
1205
16194
    if (id == AV_CODEC_ID_NONE)
1206
154
        return "none";
1207
16040
    cd = avcodec_descriptor_get(id);
1208
16040
    if (cd)
1209
16038
        return cd->name;
1210
2
    av_log(NULL, AV_LOG_WARNING, "Codec 0x%x is not in the full list.\n", id);
1211
2
    codec = avcodec_find_decoder(id);
1212
2
    if (codec)
1213
        return codec->name;
1214
2
    codec = avcodec_find_encoder(id);
1215
2
    if (codec)
1216
        return codec->name;
1217
2
    return "unknown_codec";
1218
}
1219
1220
#if FF_API_TAG_STRING
1221
size_t av_get_codec_tag_string(char *buf, size_t buf_size, unsigned int codec_tag)
1222
{
1223
    int i, len, ret = 0;
1224
1225
#define TAG_PRINT(x)                                              \
1226
    (((x) >= '0' && (x) <= '9') ||                                \
1227
     ((x) >= 'a' && (x) <= 'z') || ((x) >= 'A' && (x) <= 'Z') ||  \
1228
     ((x) == '.' || (x) == ' ' || (x) == '-' || (x) == '_'))
1229
1230
    for (i = 0; i < 4; i++) {
1231
        len = snprintf(buf, buf_size,
1232
                       TAG_PRINT(codec_tag & 0xFF) ? "%c" : "[%d]", codec_tag & 0xFF);
1233
        buf        += len;
1234
        buf_size    = buf_size > len ? buf_size - len : 0;
1235
        ret        += len;
1236
        codec_tag >>= 8;
1237
    }
1238
    return ret;
1239
}
1240
#endif
1241
1242
12599
void avcodec_string(char *buf, int buf_size, AVCodecContext *enc, int encode)
1243
{
1244
    const char *codec_type;
1245
    const char *codec_name;
1246
12599
    const char *profile = NULL;
1247
    int64_t bitrate;
1248
12599
    int new_line = 0;
1249
    AVRational display_aspect_ratio;
1250
12599
    const char *separator = enc->dump_separator ? (const char *)enc->dump_separator : ", ";
1251
1252

12599
    if (!buf || buf_size <= 0)
1253
3
        return;
1254
12599
    codec_type = av_get_media_type_string(enc->codec_type);
1255
12599
    codec_name = avcodec_get_name(enc->codec_id);
1256
12599
    profile = avcodec_profile_name(enc->codec_id, enc->profile);
1257
1258
12599
    snprintf(buf, buf_size, "%s: %s", codec_type ? codec_type : "unknown",
1259
             codec_name);
1260
12599
    buf[0] ^= 'a' ^ 'A'; /* first letter in uppercase */
1261
1262

12599
    if (enc->codec && strcmp(enc->codec->name, codec_name))
1263
105
        snprintf(buf + strlen(buf), buf_size - strlen(buf), " (%s)", enc->codec->name);
1264
1265
12599
    if (profile)
1266
1550
        snprintf(buf + strlen(buf), buf_size - strlen(buf), " (%s)", profile);
1267
12599
    if (   enc->codec_type == AVMEDIA_TYPE_VIDEO
1268
9786
        && av_log_get_level() >= AV_LOG_VERBOSE
1269
        && enc->refs)
1270
        snprintf(buf + strlen(buf), buf_size - strlen(buf),
1271
                 ", %d reference frame%s",
1272
                 enc->refs, enc->refs > 1 ? "s" : "");
1273
1274
12599
    if (enc->codec_tag)
1275
7292
        snprintf(buf + strlen(buf), buf_size - strlen(buf), " (%s / 0x%04X)",
1276
7292
                 av_fourcc2str(enc->codec_tag), enc->codec_tag);
1277
1278

12599
    switch (enc->codec_type) {
1279
9786
    case AVMEDIA_TYPE_VIDEO:
1280
        {
1281
9786
            char detail[256] = "(";
1282
1283
9786
            av_strlcat(buf, separator, buf_size);
1284
1285
77
            snprintf(buf + strlen(buf), buf_size - strlen(buf),
1286
9786
                 "%s", enc->pix_fmt == AV_PIX_FMT_NONE ? "none" :
1287
9709
                     av_get_pix_fmt_name(enc->pix_fmt));
1288

9786
            if (enc->bits_per_raw_sample && enc->pix_fmt != AV_PIX_FMT_NONE &&
1289
3408
                enc->bits_per_raw_sample < av_pix_fmt_desc_get(enc->pix_fmt)->comp[0].depth)
1290
9
                av_strlcatf(detail, sizeof(detail), "%d bpc, ", enc->bits_per_raw_sample);
1291
9786
            if (enc->color_range != AVCOL_RANGE_UNSPECIFIED)
1292
960
                av_strlcatf(detail, sizeof(detail), "%s, ",
1293
                            av_color_range_name(enc->color_range));
1294
1295
9786
            if (enc->colorspace != AVCOL_SPC_UNSPECIFIED ||
1296
9513
                enc->color_primaries != AVCOL_PRI_UNSPECIFIED ||
1297
9513
                enc->color_trc != AVCOL_TRC_UNSPECIFIED) {
1298
287
                if (enc->colorspace != (int)enc->color_primaries ||
1299
96
                    enc->colorspace != (int)enc->color_trc) {
1300
214
                    new_line = 1;
1301
214
                    av_strlcatf(detail, sizeof(detail), "%s/%s/%s, ",
1302
                                av_color_space_name(enc->colorspace),
1303
                                av_color_primaries_name(enc->color_primaries),
1304
                                av_color_transfer_name(enc->color_trc));
1305
                } else
1306
73
                    av_strlcatf(detail, sizeof(detail), "%s, ",
1307
                                av_get_colorspace_name(enc->colorspace));
1308
            }
1309
1310
9786
            if (enc->field_order != AV_FIELD_UNKNOWN) {
1311
573
                const char *field_order = "progressive";
1312
573
                if (enc->field_order == AV_FIELD_TT)
1313
102
                    field_order = "top first";
1314
471
                else if (enc->field_order == AV_FIELD_BB)
1315
34
                    field_order = "bottom first";
1316
437
                else if (enc->field_order == AV_FIELD_TB)
1317
2
                    field_order = "top coded first (swapped)";
1318
435
                else if (enc->field_order == AV_FIELD_BT)
1319
18
                    field_order = "bottom coded first (swapped)";
1320
1321
573
                av_strlcatf(detail, sizeof(detail), "%s, ", field_order);
1322
            }
1323
1324
9786
            if (av_log_get_level() >= AV_LOG_VERBOSE &&
1325
                enc->chroma_sample_location != AVCHROMA_LOC_UNSPECIFIED)
1326
                av_strlcatf(detail, sizeof(detail), "%s, ",
1327
                            av_chroma_location_name(enc->chroma_sample_location));
1328
1329
9786
            if (strlen(detail) > 1) {
1330
1418
                detail[strlen(detail) - 2] = 0;
1331
1418
                av_strlcatf(buf, buf_size, "%s)", detail);
1332
            }
1333
        }
1334
1335
9786
        if (enc->width) {
1336
9780
            av_strlcat(buf, new_line ? separator : ", ", buf_size);
1337
1338
9780
            snprintf(buf + strlen(buf), buf_size - strlen(buf),
1339
                     "%dx%d",
1340
                     enc->width, enc->height);
1341
1342
9780
            if (av_log_get_level() >= AV_LOG_VERBOSE &&
1343
                (enc->width != enc->coded_width ||
1344
                 enc->height != enc->coded_height))
1345
                snprintf(buf + strlen(buf), buf_size - strlen(buf),
1346
                         " (%dx%d)", enc->coded_width, enc->coded_height);
1347
1348
9780
            if (enc->sample_aspect_ratio.num) {
1349
1209
                av_reduce(&display_aspect_ratio.num, &display_aspect_ratio.den,
1350
1209
                          enc->width * (int64_t)enc->sample_aspect_ratio.num,
1351
1209
                          enc->height * (int64_t)enc->sample_aspect_ratio.den,
1352
                          1024 * 1024);
1353
1209
                snprintf(buf + strlen(buf), buf_size - strlen(buf),
1354
                         " [SAR %d:%d DAR %d:%d]",
1355
                         enc->sample_aspect_ratio.num, enc->sample_aspect_ratio.den,
1356
                         display_aspect_ratio.num, display_aspect_ratio.den);
1357
            }
1358
9780
            if (av_log_get_level() >= AV_LOG_DEBUG) {
1359
                int g = av_gcd(enc->time_base.num, enc->time_base.den);
1360
                snprintf(buf + strlen(buf), buf_size - strlen(buf),
1361
                         ", %d/%d",
1362
                         enc->time_base.num / g, enc->time_base.den / g);
1363
            }
1364
        }
1365
9786
        if (encode) {
1366
4840
            snprintf(buf + strlen(buf), buf_size - strlen(buf),
1367
                     ", q=%d-%d", enc->qmin, enc->qmax);
1368
        } else {
1369
4946
            if (enc->properties & FF_CODEC_PROPERTY_CLOSED_CAPTIONS)
1370
2
                snprintf(buf + strlen(buf), buf_size - strlen(buf),
1371
                         ", Closed Captions");
1372
4946
            if (enc->properties & FF_CODEC_PROPERTY_LOSSLESS)
1373
19
                snprintf(buf + strlen(buf), buf_size - strlen(buf),
1374
                         ", lossless");
1375
        }
1376
9786
        break;
1377
2660
    case AVMEDIA_TYPE_AUDIO:
1378
2660
        av_strlcat(buf, separator, buf_size);
1379
1380
2660
        if (enc->sample_rate) {
1381
2658
            snprintf(buf + strlen(buf), buf_size - strlen(buf),
1382
                     "%d Hz, ", enc->sample_rate);
1383
        }
1384
2660
        av_get_channel_layout_string(buf + strlen(buf), buf_size - strlen(buf), enc->channels, enc->channel_layout);
1385
2660
        if (enc->sample_fmt != AV_SAMPLE_FMT_NONE) {
1386
2659
            snprintf(buf + strlen(buf), buf_size - strlen(buf),
1387
                     ", %s", av_get_sample_fmt_name(enc->sample_fmt));
1388
        }
1389
2660
        if (   enc->bits_per_raw_sample > 0
1390
368
            && enc->bits_per_raw_sample != av_get_bytes_per_sample(enc->sample_fmt) * 8)
1391
117
            snprintf(buf + strlen(buf), buf_size - strlen(buf),
1392
                     " (%d bit)", enc->bits_per_raw_sample);
1393
2660
        if (av_log_get_level() >= AV_LOG_VERBOSE) {
1394
4
            if (enc->initial_padding)
1395
2
                snprintf(buf + strlen(buf), buf_size - strlen(buf),
1396
                         ", delay %d", enc->initial_padding);
1397
4
            if (enc->trailing_padding)
1398
                snprintf(buf + strlen(buf), buf_size - strlen(buf),
1399
                         ", padding %d", enc->trailing_padding);
1400
        }
1401
2660
        break;
1402
64
    case AVMEDIA_TYPE_DATA:
1403
64
        if (av_log_get_level() >= AV_LOG_DEBUG) {
1404
            int g = av_gcd(enc->time_base.num, enc->time_base.den);
1405
            if (g)
1406
                snprintf(buf + strlen(buf), buf_size - strlen(buf),
1407
                         ", %d/%d",
1408
                         enc->time_base.num / g, enc->time_base.den / g);
1409
        }
1410
64
        break;
1411
86
    case AVMEDIA_TYPE_SUBTITLE:
1412
86
        if (enc->width)
1413
17
            snprintf(buf + strlen(buf), buf_size - strlen(buf),
1414
                     ", %dx%d", enc->width, enc->height);
1415
86
        break;
1416
3
    default:
1417
3
        return;
1418
    }
1419
12596
    if (encode) {
1420
6118
        if (enc->flags & AV_CODEC_FLAG_PASS1)
1421
            snprintf(buf + strlen(buf), buf_size - strlen(buf),
1422
                     ", pass 1");
1423
6118
        if (enc->flags & AV_CODEC_FLAG_PASS2)
1424
            snprintf(buf + strlen(buf), buf_size - strlen(buf),
1425
                     ", pass 2");
1426
    }
1427
12596
    bitrate = get_bit_rate(enc);
1428
12596
    if (bitrate != 0) {
1429
8217
        snprintf(buf + strlen(buf), buf_size - strlen(buf),
1430
                 ", %"PRId64" kb/s", bitrate / 1000);
1431
4379
    } else if (enc->rc_max_rate > 0) {
1432
        snprintf(buf + strlen(buf), buf_size - strlen(buf),
1433
                 ", max. %"PRId64" kb/s", enc->rc_max_rate / 1000);
1434
    }
1435
}
1436
1437
const char *av_get_profile_name(const AVCodec *codec, int profile)
1438
{
1439
    const AVProfile *p;
1440
    if (profile == FF_PROFILE_UNKNOWN || !codec->profiles)
1441
        return NULL;
1442
1443
    for (p = codec->profiles; p->profile != FF_PROFILE_UNKNOWN; p++)
1444
        if (p->profile == profile)
1445
            return p->name;
1446
1447
    return NULL;
1448
}
1449
1450
12608
const char *avcodec_profile_name(enum AVCodecID codec_id, int profile)
1451
{
1452
12608
    const AVCodecDescriptor *desc = avcodec_descriptor_get(codec_id);
1453
    const AVProfile *p;
1454
1455

12608
    if (profile == FF_PROFILE_UNKNOWN || !desc || !desc->profiles)
1456
11041
        return NULL;
1457
1458
4142
    for (p = desc->profiles; p->profile != FF_PROFILE_UNKNOWN; p++)
1459
4132
        if (p->profile == profile)
1460
1557
            return p->name;
1461
1462
10
    return NULL;
1463
}
1464
1465
6049
unsigned avcodec_version(void)
1466
{
1467
    av_assert0(AV_CODEC_ID_PCM_S8_PLANAR==65563);
1468
    av_assert0(AV_CODEC_ID_ADPCM_G722==69660);
1469
    av_assert0(AV_CODEC_ID_SRT==94216);
1470
    av_assert0(LIBAVCODEC_VERSION_MICRO >= 100);
1471
1472
6049
    return LIBAVCODEC_VERSION_INT;
1473
}
1474
1475
6049
const char *avcodec_configuration(void)
1476
{
1477
6049
    return FFMPEG_CONFIGURATION;
1478
}
1479
1480
const char *avcodec_license(void)
1481
{
1482
#define LICENSE_PREFIX "libavcodec license: "
1483
    return &LICENSE_PREFIX FFMPEG_LICENSE[sizeof(LICENSE_PREFIX) - 1];
1484
}
1485
1486
809102
int av_get_exact_bits_per_sample(enum AVCodecID codec_id)
1487
{
1488

809102
    switch (codec_id) {
1489
10482
    case AV_CODEC_ID_8SVX_EXP:
1490
    case AV_CODEC_ID_8SVX_FIB:
1491
    case AV_CODEC_ID_ADPCM_ARGO:
1492
    case AV_CODEC_ID_ADPCM_CT:
1493
    case AV_CODEC_ID_ADPCM_IMA_APC:
1494
    case AV_CODEC_ID_ADPCM_IMA_APM:
1495
    case AV_CODEC_ID_ADPCM_IMA_EA_SEAD:
1496
    case AV_CODEC_ID_ADPCM_IMA_OKI:
1497
    case AV_CODEC_ID_ADPCM_IMA_WS:
1498
    case AV_CODEC_ID_ADPCM_IMA_SSI:
1499
    case AV_CODEC_ID_ADPCM_G722:
1500
    case AV_CODEC_ID_ADPCM_YAMAHA:
1501
    case AV_CODEC_ID_ADPCM_AICA:
1502
10482
        return 4;
1503
11448
    case AV_CODEC_ID_DSD_LSBF:
1504
    case AV_CODEC_ID_DSD_MSBF:
1505
    case AV_CODEC_ID_DSD_LSBF_PLANAR:
1506
    case AV_CODEC_ID_DSD_MSBF_PLANAR:
1507
    case AV_CODEC_ID_PCM_ALAW:
1508
    case AV_CODEC_ID_PCM_MULAW:
1509
    case AV_CODEC_ID_PCM_VIDC:
1510
    case AV_CODEC_ID_PCM_S8:
1511
    case AV_CODEC_ID_PCM_S8_PLANAR:
1512
    case AV_CODEC_ID_PCM_U8:
1513
    case AV_CODEC_ID_SDX2_DPCM:
1514
    case AV_CODEC_ID_DERF_DPCM:
1515
11448
        return 8;
1516
543626
    case AV_CODEC_ID_PCM_S16BE:
1517
    case AV_CODEC_ID_PCM_S16BE_PLANAR:
1518
    case AV_CODEC_ID_PCM_S16LE:
1519
    case AV_CODEC_ID_PCM_S16LE_PLANAR:
1520
    case AV_CODEC_ID_PCM_U16BE:
1521
    case AV_CODEC_ID_PCM_U16LE:
1522
543626
        return 16;
1523
45030
    case AV_CODEC_ID_PCM_S24DAUD:
1524
    case AV_CODEC_ID_PCM_S24BE:
1525
    case AV_CODEC_ID_PCM_S24LE:
1526
    case AV_CODEC_ID_PCM_S24LE_PLANAR:
1527
    case AV_CODEC_ID_PCM_U24BE:
1528
    case AV_CODEC_ID_PCM_U24LE:
1529
45030
        return 24;
1530
27125
    case AV_CODEC_ID_PCM_S32BE:
1531
    case AV_CODEC_ID_PCM_S32LE:
1532
    case AV_CODEC_ID_PCM_S32LE_PLANAR:
1533
    case AV_CODEC_ID_PCM_U32BE:
1534
    case AV_CODEC_ID_PCM_U32LE:
1535
    case AV_CODEC_ID_PCM_F32BE:
1536
    case AV_CODEC_ID_PCM_F32LE:
1537
    case AV_CODEC_ID_PCM_F24LE:
1538
    case AV_CODEC_ID_PCM_F16LE:
1539
27125
        return 32;
1540
21300
    case AV_CODEC_ID_PCM_F64BE:
1541
    case AV_CODEC_ID_PCM_F64LE:
1542
    case AV_CODEC_ID_PCM_S64BE:
1543
    case AV_CODEC_ID_PCM_S64LE:
1544
21300
        return 64;
1545
150091
    default:
1546
150091
        return 0;
1547
    }
1548
}
1549
1550
16
enum AVCodecID av_get_pcm_codec(enum AVSampleFormat fmt, int be)
1551
{
1552
    static const enum AVCodecID map[][2] = {
1553
        [AV_SAMPLE_FMT_U8  ] = { AV_CODEC_ID_PCM_U8,    AV_CODEC_ID_PCM_U8    },
1554
        [AV_SAMPLE_FMT_S16 ] = { AV_CODEC_ID_PCM_S16LE, AV_CODEC_ID_PCM_S16BE },
1555
        [AV_SAMPLE_FMT_S32 ] = { AV_CODEC_ID_PCM_S32LE, AV_CODEC_ID_PCM_S32BE },
1556
        [AV_SAMPLE_FMT_FLT ] = { AV_CODEC_ID_PCM_F32LE, AV_CODEC_ID_PCM_F32BE },
1557
        [AV_SAMPLE_FMT_DBL ] = { AV_CODEC_ID_PCM_F64LE, AV_CODEC_ID_PCM_F64BE },
1558
        [AV_SAMPLE_FMT_U8P ] = { AV_CODEC_ID_PCM_U8,    AV_CODEC_ID_PCM_U8    },
1559
        [AV_SAMPLE_FMT_S16P] = { AV_CODEC_ID_PCM_S16LE, AV_CODEC_ID_PCM_S16BE },
1560
        [AV_SAMPLE_FMT_S32P] = { AV_CODEC_ID_PCM_S32LE, AV_CODEC_ID_PCM_S32BE },
1561
        [AV_SAMPLE_FMT_S64P] = { AV_CODEC_ID_PCM_S64LE, AV_CODEC_ID_PCM_S64BE },
1562
        [AV_SAMPLE_FMT_FLTP] = { AV_CODEC_ID_PCM_F32LE, AV_CODEC_ID_PCM_F32BE },
1563
        [AV_SAMPLE_FMT_DBLP] = { AV_CODEC_ID_PCM_F64LE, AV_CODEC_ID_PCM_F64BE },
1564
    };
1565

16
    if (fmt < 0 || fmt >= FF_ARRAY_ELEMS(map))
1566
        return AV_CODEC_ID_NONE;
1567

16
    if (be < 0 || be > 1)
1568
16
        be = AV_NE(1, 0);
1569
16
    return map[fmt][be];
1570
}
1571
1572
323599
int av_get_bits_per_sample(enum AVCodecID codec_id)
1573
{
1574

323599
    switch (codec_id) {
1575
3
    case AV_CODEC_ID_ADPCM_SBPRO_2:
1576
3
        return 2;
1577
3
    case AV_CODEC_ID_ADPCM_SBPRO_3:
1578
3
        return 3;
1579
64
    case AV_CODEC_ID_ADPCM_SBPRO_4:
1580
    case AV_CODEC_ID_ADPCM_IMA_WAV:
1581
    case AV_CODEC_ID_ADPCM_IMA_QT:
1582
    case AV_CODEC_ID_ADPCM_SWF:
1583
    case AV_CODEC_ID_ADPCM_MS:
1584
64
        return 4;
1585
323529
    default:
1586
323529
        return av_get_exact_bits_per_sample(codec_id);
1587
    }
1588
}
1589
1590
479715
static int get_audio_frame_duration(enum AVCodecID id, int sr, int ch, int ba,
1591
                                    uint32_t tag, int bits_per_coded_sample, int64_t bitrate,
1592
                                    uint8_t * extradata, int frame_size, int frame_bytes)
1593
{
1594
479715
    int bps = av_get_exact_bits_per_sample(id);
1595

479715
    int framecount = (ba > 0 && frame_bytes / ba > 0) ? frame_bytes / ba : 1;
1596
1597
    /* codecs with an exact constant bits per sample */
1598


479715
    if (bps > 0 && ch > 0 && frame_bytes > 0 && ch < 32768 && bps < 32768)
1599
336682
        return (frame_bytes * 8LL) / (bps * ch);
1600
143033
    bps = bits_per_coded_sample;
1601
1602
    /* codecs with a fixed packet duration */
1603



143033
    switch (id) {
1604
16542
    case AV_CODEC_ID_ADPCM_ADX:    return   32;
1605
8274
    case AV_CODEC_ID_ADPCM_IMA_QT: return   64;
1606
600
    case AV_CODEC_ID_ADPCM_EA_XAS: return  128;
1607
3933
    case AV_CODEC_ID_AMR_NB:
1608
    case AV_CODEC_ID_EVRC:
1609
    case AV_CODEC_ID_GSM:
1610
    case AV_CODEC_ID_QCELP:
1611
3933
    case AV_CODEC_ID_RA_288:       return  160;
1612
    case AV_CODEC_ID_AMR_WB:
1613
    case AV_CODEC_ID_GSM_MS:       return  320;
1614
    case AV_CODEC_ID_MP1:          return  384;
1615
3274
    case AV_CODEC_ID_ATRAC1:       return  512;
1616
2756
    case AV_CODEC_ID_ATRAC9:
1617
    case AV_CODEC_ID_ATRAC3:
1618
2756
        if (framecount > INT_MAX/1024)
1619
            return 0;
1620
2756
        return 1024 * framecount;
1621
813
    case AV_CODEC_ID_ATRAC3P:      return 2048;
1622
10201
    case AV_CODEC_ID_MP2:
1623
10201
    case AV_CODEC_ID_MUSEPACK7:    return 1152;
1624
1167
    case AV_CODEC_ID_AC3:          return 1536;
1625
    }
1626
1627
95473
    if (sr > 0) {
1628
        /* calc from sample rate */
1629
59901
        if (id == AV_CODEC_ID_TTA)
1630
9
            return 256 * sr / 245;
1631
59892
        else if (id == AV_CODEC_ID_DST)
1632
            return 588 * sr / 44100;
1633
1634
59892
        if (ch > 0) {
1635
            /* calc from sample rate and channels */
1636
59892
            if (id == AV_CODEC_ID_BINKAUDIO_DCT)
1637
65
                return (480 << (sr / 22050)) / ch;
1638
        }
1639
1640
59827
        if (id == AV_CODEC_ID_MP3)
1641
2690
            return sr <= 24000 ? 576 : 1152;
1642
    }
1643
1644
92709
    if (ba > 0) {
1645
        /* calc from block_align */
1646
19371
        if (id == AV_CODEC_ID_SIPR) {
1647

6972
            switch (ba) {
1648
3250
            case 20: return 160;
1649
1706
            case 19: return 144;
1650
1680
            case 29: return 288;
1651
336
            case 37: return 480;
1652
            }
1653
12399
        } else if (id == AV_CODEC_ID_ILBC) {
1654
            switch (ba) {
1655
            case 38: return 160;
1656
            case 50: return 240;
1657
            }
1658
        }
1659
    }
1660
1661
85737
    if (frame_bytes > 0) {
1662
        /* calc from frame_bytes only */
1663
83059
        if (id == AV_CODEC_ID_TRUESPEECH)
1664
12
            return 240 * (frame_bytes / 32);
1665
83047
        if (id == AV_CODEC_ID_NELLYMOSER)
1666
2074
            return 256 * (frame_bytes / 64);
1667
80973
        if (id == AV_CODEC_ID_RA_144)
1668
1057
            return 160 * (frame_bytes / 20);
1669
1670
79916
        if (bps > 0) {
1671
            /* calc from frame_bytes and bits_per_coded_sample */
1672

15379
            if (id == AV_CODEC_ID_ADPCM_G726 || id == AV_CODEC_ID_ADPCM_G726LE)
1673
106
                return frame_bytes * 8 / bps;
1674
        }
1675
1676

79810
        if (ch > 0 && ch < INT_MAX/16) {
1677
            /* calc from frame_bytes and channels */
1678




46852
            switch (id) {
1679
            case AV_CODEC_ID_FASTAUDIO:
1680
                return frame_bytes / (40 * ch) * 256;
1681
            case AV_CODEC_ID_ADPCM_IMA_MOFLEX:
1682
                return (frame_bytes - 4 * ch) / (128 * ch) * 256;
1683
36
            case AV_CODEC_ID_ADPCM_AFC:
1684
36
                return frame_bytes / (9 * ch) * 16;
1685
33
            case AV_CODEC_ID_ADPCM_PSX:
1686
            case AV_CODEC_ID_ADPCM_DTK:
1687
33
                return frame_bytes / (16 * ch) * 28;
1688
452
            case AV_CODEC_ID_ADPCM_4XM:
1689
            case AV_CODEC_ID_ADPCM_IMA_DAT4:
1690
            case AV_CODEC_ID_ADPCM_IMA_ISS:
1691
452
                return (frame_bytes - 4 * ch) * 2 / ch;
1692
1601
            case AV_CODEC_ID_ADPCM_IMA_SMJPEG:
1693
1601
                return (frame_bytes - 4) * 2 / ch;
1694
162
            case AV_CODEC_ID_ADPCM_IMA_AMV:
1695
162
                return (frame_bytes - 8) * 2 / ch;
1696
1792
            case AV_CODEC_ID_ADPCM_THP:
1697
            case AV_CODEC_ID_ADPCM_THP_LE:
1698
1792
                if (extradata)
1699
1767
                    return frame_bytes * 14 / (8 * ch);
1700
25
                break;
1701
111
            case AV_CODEC_ID_ADPCM_XA:
1702
111
                return (frame_bytes / 128) * 224 / ch;
1703
327
            case AV_CODEC_ID_INTERPLAY_DPCM:
1704
327
                return (frame_bytes - 6 - ch) / ch;
1705
680
            case AV_CODEC_ID_ROQ_DPCM:
1706
680
                return (frame_bytes - 8) / ch;
1707
112
            case AV_CODEC_ID_XAN_DPCM:
1708
112
                return (frame_bytes - 2 * ch) / ch;
1709
            case AV_CODEC_ID_MACE3:
1710
                return 3 * frame_bytes / ch;
1711
69
            case AV_CODEC_ID_MACE6:
1712
69
                return 6 * frame_bytes / ch;
1713
            case AV_CODEC_ID_PCM_LXF:
1714
                return 2 * (frame_bytes / (5 * ch));
1715
1311
            case AV_CODEC_ID_IAC:
1716
            case AV_CODEC_ID_IMC:
1717
1311
                return 4 * frame_bytes / ch;
1718
            }
1719
1720
40191
            if (tag) {
1721
                /* calc from frame_bytes, channels, and codec_tag */
1722
9276
                if (id == AV_CODEC_ID_SOL_DPCM) {
1723
74
                    if (tag == 3)
1724
74
                        return frame_bytes / ch;
1725
                    else
1726
                        return frame_bytes * 2 / ch;
1727
                }
1728
            }
1729
1730
40117
            if (ba > 0) {
1731
                /* calc from frame_bytes, channels, and block_align */
1732
9288
                int blocks = frame_bytes / ba;
1733

9288
                switch (id) {
1734
916
                case AV_CODEC_ID_ADPCM_IMA_WAV:
1735

916
                    if (bps < 2 || bps > 5)
1736
                        return 0;
1737
916
                    return blocks * (1 + (ba - 4 * ch) / (bps * ch) * 8);
1738
669
                case AV_CODEC_ID_ADPCM_IMA_DK3:
1739
669
                    return blocks * (((ba - 16) * 2 / 3 * 4) / ch);
1740
649
                case AV_CODEC_ID_ADPCM_IMA_DK4:
1741
649
                    return blocks * (1 + (ba - 4 * ch) * 2 / ch);
1742
1001
                case AV_CODEC_ID_ADPCM_IMA_RAD:
1743
1001
                    return blocks * ((ba - 4 * ch) * 2 / ch);
1744
890
                case AV_CODEC_ID_ADPCM_MS:
1745
890
                    return blocks * (2 + (ba - 7 * ch) * 2 / ch);
1746
                case AV_CODEC_ID_ADPCM_MTAF:
1747
                    return blocks * (ba - 16) * 2 / ch;
1748
                }
1749
            }
1750
1751
35992
            if (bps > 0) {
1752
                /* calc from frame_bytes, channels, and bits_per_coded_sample */
1753

7608
                switch (id) {
1754
654
                case AV_CODEC_ID_PCM_DVD:
1755

654
                    if(bps<4 || frame_bytes<3)
1756
                        return 0;
1757
654
                    return 2 * ((frame_bytes - 3) / ((bps * 2 / 8) * ch));
1758
                case AV_CODEC_ID_PCM_BLURAY:
1759
                    if(bps<4 || frame_bytes<4)
1760
                        return 0;
1761
                    return (frame_bytes - 4) / ((FFALIGN(ch, 2) * bps) / 8);
1762
                case AV_CODEC_ID_S302M:
1763
                    return 2 * (frame_bytes / ((bps + 4) / 4)) / ch;
1764
                }
1765
            }
1766
        }
1767
    }
1768
1769
    /* Fall back on using frame_size */
1770

70974
    if (frame_size > 1 && frame_bytes)
1771
21615
        return frame_size;
1772
1773
    //For WMA we currently have no other means to calculate duration thus we
1774
    //do it here by assuming CBR, which is true for all known cases.
1775


49359
    if (bitrate > 0 && frame_bytes > 0 && sr > 0 && ba > 1) {
1776

4318
        if (id == AV_CODEC_ID_WMAV1 || id == AV_CODEC_ID_WMAV2)
1777
770
            return  (frame_bytes * 8LL * sr) / bitrate;
1778
    }
1779
1780
48589
    return 0;
1781
}
1782
1783
382877
int av_get_audio_frame_duration(AVCodecContext *avctx, int frame_bytes)
1784
{
1785
382877
    return get_audio_frame_duration(avctx->codec_id, avctx->sample_rate,
1786
                                    avctx->channels, avctx->block_align,
1787
                                    avctx->codec_tag, avctx->bits_per_coded_sample,
1788
                                    avctx->bit_rate, avctx->extradata, avctx->frame_size,
1789
                                    frame_bytes);
1790
}
1791
1792
96838
int av_get_audio_frame_duration2(AVCodecParameters *par, int frame_bytes)
1793
{
1794
96838
    return get_audio_frame_duration(par->codec_id, par->sample_rate,
1795
                                    par->channels, par->block_align,
1796
                                    par->codec_tag, par->bits_per_coded_sample,
1797
                                    par->bit_rate, par->extradata, par->frame_size,
1798
                                    frame_bytes);
1799
}
1800
1801
#if !HAVE_THREADS
1802
int ff_thread_init(AVCodecContext *s)
1803
{
1804
    return -1;
1805
}
1806
1807
#endif
1808
1809
54
unsigned int av_xiphlacing(unsigned char *s, unsigned int v)
1810
{
1811
54
    unsigned int n = 0;
1812
1813
54
    while (v >= 0xff) {
1814
        *s++ = 0xff;
1815
        v -= 0xff;
1816
        n++;
1817
    }
1818
54
    *s = v;
1819
54
    n++;
1820
54
    return n;
1821
}
1822
1823
476
int ff_match_2uint16(const uint16_t(*tab)[2], int size, int a, int b)
1824
{
1825
    int i;
1826

1924
    for (i = 0; i < size && !(tab[i][0] == a && tab[i][1] == b); i++) ;
1827
476
    return i;
1828
}
1829
1830
13218
const AVCodecHWConfig *avcodec_get_hw_config(const AVCodec *codec, int index)
1831
{
1832
    int i;
1833

13218
    if (!codec->hw_configs || index < 0)
1834
10783
        return NULL;
1835
6404
    for (i = 0; i <= index; i++)
1836
4803
        if (!codec->hw_configs[i])
1837
834
            return NULL;
1838
1601
    return &codec->hw_configs[index]->public;
1839
}
1840
1841
#if FF_API_USER_VISIBLE_AVHWACCEL
1842
AVHWAccel *av_hwaccel_next(const AVHWAccel *hwaccel)
1843
{
1844
    return NULL;
1845
}
1846
1847
void av_register_hwaccel(AVHWAccel *hwaccel)
1848
{
1849
}
1850
#endif
1851
1852
#if FF_API_LOCKMGR
1853
int av_lockmgr_register(int (*cb)(void **mutex, enum AVLockOp op))
1854
{
1855
    return 0;
1856
}
1857
#endif
1858
1859
623037
unsigned int avpriv_toupper4(unsigned int x)
1860
{
1861
623037
    return av_toupper(x & 0xFF) +
1862
623037
          (av_toupper((x >>  8) & 0xFF) << 8)  +
1863
1246074
          (av_toupper((x >> 16) & 0xFF) << 16) +
1864
623037
((unsigned)av_toupper((x >> 24) & 0xFF) << 24);
1865
}
1866
1867
137114
int ff_thread_ref_frame(ThreadFrame *dst, const ThreadFrame *src)
1868
{
1869
    int ret;
1870
1871
137114
    dst->owner[0] = src->owner[0];
1872
137114
    dst->owner[1] = src->owner[1];
1873
1874
137114
    ret = av_frame_ref(dst->f, src->f);
1875
137114
    if (ret < 0)
1876
6
        return ret;
1877
1878
137108
    av_assert0(!dst->progress);
1879
1880
137108
    if (src->progress &&
1881
333
        !(dst->progress = av_buffer_ref(src->progress))) {
1882
        ff_thread_release_buffer(dst->owner[0], dst);
1883
        return AVERROR(ENOMEM);
1884
    }
1885
1886
137108
    return 0;
1887
}
1888
1889
#if !HAVE_THREADS
1890
1891
enum AVPixelFormat ff_thread_get_format(AVCodecContext *avctx, const enum AVPixelFormat *fmt)
1892
{
1893
    return ff_get_format(avctx, fmt);
1894
}
1895
1896
int ff_thread_get_buffer(AVCodecContext *avctx, ThreadFrame *f, int flags)
1897
{
1898
    f->owner[0] = f->owner[1] = avctx;
1899
    return ff_get_buffer(avctx, f->f, flags);
1900
}
1901
1902
void ff_thread_release_buffer(AVCodecContext *avctx, ThreadFrame *f)
1903
{
1904
    if (f->f)
1905
        av_frame_unref(f->f);
1906
}
1907
1908
void ff_thread_finish_setup(AVCodecContext *avctx)
1909
{
1910
}
1911
1912
void ff_thread_report_progress(ThreadFrame *f, int progress, int field)
1913
{
1914
}
1915
1916
void ff_thread_await_progress(ThreadFrame *f, int progress, int field)
1917
{
1918
}
1919
1920
int ff_thread_can_start_frame(AVCodecContext *avctx)
1921
{
1922
    return 1;
1923
}
1924
1925
int ff_alloc_entries(AVCodecContext *avctx, int count)
1926
{
1927
    return 0;
1928
}
1929
1930
void ff_reset_entries(AVCodecContext *avctx)
1931
{
1932
}
1933
1934
void ff_thread_await_progress2(AVCodecContext *avctx, int field, int thread, int shift)
1935
{
1936
}
1937
1938
void ff_thread_report_progress2(AVCodecContext *avctx, int field, int thread, int n)
1939
{
1940
}
1941
1942
#endif
1943
1944
2623953
int avcodec_is_open(AVCodecContext *s)
1945
{
1946
2623953
    return !!s->internal;
1947
}
1948
1949
1
int avpriv_bprint_to_extradata(AVCodecContext *avctx, struct AVBPrint *buf)
1950
{
1951
    int ret;
1952
    char *str;
1953
1954
1
    ret = av_bprint_finalize(buf, &str);
1955
1
    if (ret < 0)
1956
        return ret;
1957
1
    if (!av_bprint_is_complete(buf)) {
1958
        av_free(str);
1959
        return AVERROR(ENOMEM);
1960
    }
1961
1962
1
    avctx->extradata = str;
1963
    /* Note: the string is NUL terminated (so extradata can be read as a
1964
     * string), but the ending character is not accounted in the size (in
1965
     * binary formats you are likely not supposed to mux that character). When
1966
     * extradata is copied, it is also padded with AV_INPUT_BUFFER_PADDING_SIZE
1967
     * zeros. */
1968
1
    avctx->extradata_size = buf->len;
1969
1
    return 0;
1970
}
1971
1972
568437
const uint8_t *avpriv_find_start_code(const uint8_t *av_restrict p,
1973
                                      const uint8_t *end,
1974
                                      uint32_t *av_restrict state)
1975
{
1976
    int i;
1977
1978
568437
    av_assert0(p <= end);
1979
568437
    if (p >= end)
1980
553
        return end;
1981
1982
2262446
    for (i = 0; i < 3; i++) {
1983
1698268
        uint32_t tmp = *state << 8;
1984
1698268
        *state = tmp + *(p++);
1985

1698268
        if (tmp == 0x100 || p == end)
1986
3706
            return p;
1987
    }
1988
1989
161232291
    while (p < end) {
1990
161175219
        if      (p[-1] > 1      ) p += 3;
1991
20649686
        else if (p[-2]          ) p += 2;
1992
12741652
        else if (p[-3]|(p[-1]-1)) p++;
1993
        else {
1994
507106
            p++;
1995
507106
            break;
1996
        }
1997
    }
1998
1999
564178
    p = FFMIN(p, end) - 4;
2000
564178
    *state = AV_RB32(p);
2001
2002
564178
    return p + 4;
2003
}
2004
2005
419
AVCPBProperties *av_cpb_properties_alloc(size_t *size)
2006
{
2007
419
    AVCPBProperties *props = av_mallocz(sizeof(AVCPBProperties));
2008
419
    if (!props)
2009
        return NULL;
2010
2011
419
    if (size)
2012
419
        *size = sizeof(*props);
2013
2014
419
    props->vbv_delay = UINT64_MAX;
2015
2016
419
    return props;
2017
}
2018
2019
831
AVCPBProperties *ff_add_cpb_side_data(AVCodecContext *avctx)
2020
{
2021
    AVPacketSideData *tmp;
2022
    AVCPBProperties  *props;
2023
    size_t size;
2024
    int i;
2025
2026
831
    for (i = 0; i < avctx->nb_coded_side_data; i++)
2027
437
        if (avctx->coded_side_data[i].type == AV_PKT_DATA_CPB_PROPERTIES)
2028
437
            return (AVCPBProperties *)avctx->coded_side_data[i].data;
2029
2030
394
    props = av_cpb_properties_alloc(&size);
2031
394
    if (!props)
2032
        return NULL;
2033
2034
394
    tmp = av_realloc_array(avctx->coded_side_data, avctx->nb_coded_side_data + 1, sizeof(*tmp));
2035
394
    if (!tmp) {
2036
        av_freep(&props);
2037
        return NULL;
2038
    }
2039
2040
394
    avctx->coded_side_data = tmp;
2041
394
    avctx->nb_coded_side_data++;
2042
2043
394
    avctx->coded_side_data[avctx->nb_coded_side_data - 1].type = AV_PKT_DATA_CPB_PROPERTIES;
2044
394
    avctx->coded_side_data[avctx->nb_coded_side_data - 1].data = (uint8_t*)props;
2045
394
    avctx->coded_side_data[avctx->nb_coded_side_data - 1].size = size;
2046
2047
394
    return props;
2048
}
2049
2050
155692
static void codec_parameters_reset(AVCodecParameters *par)
2051
{
2052
155692
    av_freep(&par->extradata);
2053
2054
155692
    memset(par, 0, sizeof(*par));
2055
2056
155692
    par->codec_type          = AVMEDIA_TYPE_UNKNOWN;
2057
155692
    par->codec_id            = AV_CODEC_ID_NONE;
2058
155692
    par->format              = -1;
2059
155692
    par->field_order         = AV_FIELD_UNKNOWN;
2060
155692
    par->color_range         = AVCOL_RANGE_UNSPECIFIED;
2061
155692
    par->color_primaries     = AVCOL_PRI_UNSPECIFIED;
2062
155692
    par->color_trc           = AVCOL_TRC_UNSPECIFIED;
2063
155692
    par->color_space         = AVCOL_SPC_UNSPECIFIED;
2064
155692
    par->chroma_location     = AVCHROMA_LOC_UNSPECIFIED;
2065
155692
    par->sample_aspect_ratio = (AVRational){ 0, 1 };
2066
155692
    par->profile             = FF_PROFILE_UNKNOWN;
2067
155692
    par->level               = FF_LEVEL_UNKNOWN;
2068
155692
}
2069
2070
48484
AVCodecParameters *avcodec_parameters_alloc(void)
2071
{
2072
48484
    AVCodecParameters *par = av_mallocz(sizeof(*par));
2073
2074
48484
    if (!par)
2075
        return NULL;
2076
48484
    codec_parameters_reset(par);
2077
48484
    return par;
2078
}
2079
2080
48478
void avcodec_parameters_free(AVCodecParameters **ppar)
2081
{
2082
48478
    AVCodecParameters *par = *ppar;
2083
2084
48478
    if (!par)
2085
        return;
2086
48478
    codec_parameters_reset(par);
2087
2088
48478
    av_freep(ppar);
2089
}
2090
2091
26883
int avcodec_parameters_copy(AVCodecParameters *dst, const AVCodecParameters *src)
2092
{
2093
26883
    codec_parameters_reset(dst);
2094
26883
    memcpy(dst, src, sizeof(*dst));
2095
2096
26883
    dst->extradata      = NULL;
2097
26883
    dst->extradata_size = 0;
2098
26883
    if (src->extradata) {
2099
4131
        dst->extradata = av_mallocz(src->extradata_size + AV_INPUT_BUFFER_PADDING_SIZE);
2100
4131
        if (!dst->extradata)
2101
            return AVERROR(ENOMEM);
2102
4131
        memcpy(dst->extradata, src->extradata, src->extradata_size);
2103
4131
        dst->extradata_size = src->extradata_size;
2104
    }
2105
2106
26883
    return 0;
2107
}
2108
2109
31847
int avcodec_parameters_from_context(AVCodecParameters *par,
2110
                                    const AVCodecContext *codec)
2111
{
2112
31847
    codec_parameters_reset(par);
2113
2114
31847
    par->codec_type = codec->codec_type;
2115
31847
    par->codec_id   = codec->codec_id;
2116
31847
    par->codec_tag  = codec->codec_tag;
2117
2118
31847
    par->bit_rate              = codec->bit_rate;
2119
31847
    par->bits_per_coded_sample = codec->bits_per_coded_sample;
2120
31847
    par->bits_per_raw_sample   = codec->bits_per_raw_sample;
2121
31847
    par->profile               = codec->profile;
2122
31847
    par->level                 = codec->level;
2123
2124

31847
    switch (par->codec_type) {
2125
24602
    case AVMEDIA_TYPE_VIDEO:
2126
24602
        par->format              = codec->pix_fmt;
2127
24602
        par->width               = codec->width;
2128
24602
        par->height              = codec->height;
2129
24602
        par->field_order         = codec->field_order;
2130
24602
        par->color_range         = codec->color_range;
2131
24602
        par->color_primaries     = codec->color_primaries;
2132
24602
        par->color_trc           = codec->color_trc;
2133
24602
        par->color_space         = codec->colorspace;
2134
24602
        par->chroma_location     = codec->chroma_sample_location;
2135
24602
        par->sample_aspect_ratio = codec->sample_aspect_ratio;
2136
24602
        par->video_delay         = codec->has_b_frames;
2137
24602
        break;
2138
6920
    case AVMEDIA_TYPE_AUDIO:
2139
6920
        par->format           = codec->sample_fmt;
2140
6920
        par->channel_layout   = codec->channel_layout;
2141
6920
        par->channels         = codec->channels;
2142
6920
        par->sample_rate      = codec->sample_rate;
2143
6920
        par->block_align      = codec->block_align;
2144
6920
        par->frame_size       = codec->frame_size;
2145
6920
        par->initial_padding  = codec->initial_padding;
2146
6920
        par->trailing_padding = codec->trailing_padding;
2147
6920
        par->seek_preroll     = codec->seek_preroll;
2148
6920
        break;
2149
213
    case AVMEDIA_TYPE_SUBTITLE:
2150
213
        par->width  = codec->width;
2151
213
        par->height = codec->height;
2152
213
        break;
2153
    }
2154
2155
31847
    if (codec->extradata) {
2156
4973
        par->extradata = av_mallocz(codec->extradata_size + AV_INPUT_BUFFER_PADDING_SIZE);
2157
4973
        if (!par->extradata)
2158
            return AVERROR(ENOMEM);
2159
4973
        memcpy(par->extradata, codec->extradata, codec->extradata_size);
2160
4973
        par->extradata_size = codec->extradata_size;
2161
    }
2162
2163
31847
    return 0;
2164
}
2165
2166
46849
int avcodec_parameters_to_context(AVCodecContext *codec,
2167
                                  const AVCodecParameters *par)
2168
{
2169
46849
    codec->codec_type = par->codec_type;
2170
46849
    codec->codec_id   = par->codec_id;
2171
46849
    codec->codec_tag  = par->codec_tag;
2172
2173
46849
    codec->bit_rate              = par->bit_rate;
2174
46849
    codec->bits_per_coded_sample = par->bits_per_coded_sample;
2175
46849
    codec->bits_per_raw_sample   = par->bits_per_raw_sample;
2176
46849
    codec->profile               = par->profile;
2177
46849
    codec->level                 = par->level;
2178
2179

46849
    switch (par->codec_type) {
2180
35205
    case AVMEDIA_TYPE_VIDEO:
2181
35205
        codec->pix_fmt                = par->format;
2182
35205
        codec->width                  = par->width;
2183
35205
        codec->height                 = par->height;
2184
35205
        codec->field_order            = par->field_order;
2185
35205
        codec->color_range            = par->color_range;
2186
35205
        codec->color_primaries        = par->color_primaries;
2187
35205
        codec->color_trc              = par->color_trc;
2188
35205
        codec->colorspace             = par->color_space;
2189
35205
        codec->chroma_sample_location = par->chroma_location;
2190
35205
        codec->sample_aspect_ratio    = par->sample_aspect_ratio;
2191
35205
        codec->has_b_frames           = par->video_delay;
2192
35205
        break;
2193
10956
    case AVMEDIA_TYPE_AUDIO:
2194
10956
        codec->sample_fmt       = par->format;
2195
10956
        codec->channel_layout   = par->channel_layout;
2196
10956
        codec->channels         = par->channels;
2197
10956
        codec->sample_rate      = par->sample_rate;
2198
10956
        codec->block_align      = par->block_align;
2199
10956
        codec->frame_size       = par->frame_size;
2200
10956
        codec->delay            =
2201
10956
        codec->initial_padding  = par->initial_padding;
2202
10956
        codec->trailing_padding = par->trailing_padding;
2203
10956
        codec->seek_preroll     = par->seek_preroll;
2204
10956
        break;
2205
331
    case AVMEDIA_TYPE_SUBTITLE:
2206
331
        codec->width  = par->width;
2207
331
        codec->height = par->height;
2208
331
        break;
2209
    }
2210
2211
46849
    if (par->extradata) {
2212
7013
        av_freep(&codec->extradata);
2213
7013
        codec->extradata = av_mallocz(par->extradata_size + AV_INPUT_BUFFER_PADDING_SIZE);
2214
7013
        if (!codec->extradata)
2215
            return AVERROR(ENOMEM);
2216
7013
        memcpy(codec->extradata, par->extradata, par->extradata_size);
2217
7013
        codec->extradata_size = par->extradata_size;
2218
    }
2219
2220
46849
    return 0;
2221
}
2222
2223
static unsigned bcd2uint(uint8_t bcd)
2224
{
2225
    unsigned low  = bcd & 0xf;
2226
    unsigned high = bcd >> 4;
2227
    if (low > 9 || high > 9)
2228
        return 0;
2229
    return low + 10*high;
2230
}
2231
2232
int ff_alloc_timecode_sei(const AVFrame *frame, AVRational rate, size_t prefix_len,
2233
                     void **data, size_t *sei_size)
2234
{
2235
    AVFrameSideData *sd = NULL;
2236
    uint8_t *sei_data;
2237
    PutBitContext pb;
2238
    uint32_t *tc;
2239
    int m;
2240
2241
    if (frame)
2242
        sd = av_frame_get_side_data(frame, AV_FRAME_DATA_S12M_TIMECODE);
2243
2244
    if (!sd) {
2245
        *data = NULL;
2246
        return 0;
2247
    }
2248
    tc =  (uint32_t*)sd->data;
2249
    m  = tc[0] & 3;
2250
2251
    *sei_size = sizeof(uint32_t) * 4;
2252
    *data = av_mallocz(*sei_size + prefix_len);
2253
    if (!*data)
2254
        return AVERROR(ENOMEM);
2255
    sei_data = (uint8_t*)*data + prefix_len;
2256
2257
    init_put_bits(&pb, sei_data, *sei_size);
2258
    put_bits(&pb, 2, m); // num_clock_ts
2259
2260
    for (int j = 1; j <= m; j++) {
2261
        uint32_t tcsmpte = tc[j];
2262
        unsigned hh   = bcd2uint(tcsmpte     & 0x3f);    // 6-bit hours
2263
        unsigned mm   = bcd2uint(tcsmpte>>8  & 0x7f);    // 7-bit minutes
2264
        unsigned ss   = bcd2uint(tcsmpte>>16 & 0x7f);    // 7-bit seconds
2265
        unsigned ff   = bcd2uint(tcsmpte>>24 & 0x3f);    // 6-bit frames
2266
        unsigned drop = tcsmpte & 1<<30 && !0;  // 1-bit drop if not arbitrary bit
2267
2268
        /* Calculate frame number of HEVC by SMPTE ST 12-1:2014 Sec 12.2 if rate > 30FPS */
2269
        if (av_cmp_q(rate, (AVRational) {30, 1}) == 1) {
2270
            unsigned pc;
2271
            ff *= 2;
2272
            if (av_cmp_q(rate, (AVRational) {50, 1}) == 0)
2273
                pc = !!(tcsmpte & 1 << 7);
2274
            else
2275
                pc = !!(tcsmpte & 1 << 23);
2276
            ff = (ff + pc) & 0x7f;
2277
        }
2278
2279
        put_bits(&pb, 1, 1); // clock_timestamp_flag
2280
        put_bits(&pb, 1, 1); // units_field_based_flag
2281
        put_bits(&pb, 5, 0); // counting_type
2282
        put_bits(&pb, 1, 1); // full_timestamp_flag
2283
        put_bits(&pb, 1, 0); // discontinuity_flag
2284
        put_bits(&pb, 1, drop);
2285
        put_bits(&pb, 9, ff);
2286
        put_bits(&pb, 6, ss);
2287
        put_bits(&pb, 6, mm);
2288
        put_bits(&pb, 5, hh);
2289
        put_bits(&pb, 5, 0);
2290
    }
2291
    flush_put_bits(&pb);
2292
2293
    return 0;
2294
}
2295
2296
4127
int64_t ff_guess_coded_bitrate(AVCodecContext *avctx)
2297
{
2298
4127
    AVRational framerate = avctx->framerate;
2299
4127
    int bits_per_coded_sample = avctx->bits_per_coded_sample;
2300
    int64_t bitrate;
2301
2302

4127
    if (!(framerate.num && framerate.den))
2303
1
        framerate = av_inv_q(avctx->time_base);
2304

4127
    if (!(framerate.num && framerate.den))
2305
        return 0;
2306
2307
4127
    if (!bits_per_coded_sample) {
2308
        const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(avctx->pix_fmt);
2309
        bits_per_coded_sample = av_get_bits_per_pixel(desc);
2310
    }
2311
4127
    bitrate = (int64_t)bits_per_coded_sample * avctx->width * avctx->height *
2312
4127
              framerate.num / framerate.den;
2313
2314
4127
    return bitrate;
2315
}
2316
2317
2400
int ff_int_from_list_or_default(void *ctx, const char * val_name, int val,
2318
                                const int * array_valid_values, int default_value)
2319
{
2320
2400
    int i = 0, ref_val;
2321
2322
    while (1) {
2323
6400
        ref_val = array_valid_values[i];
2324
6400
        if (ref_val == INT_MAX)
2325
            break;
2326
6400
        if (val == ref_val)
2327
2400
            return val;
2328
4000
        i++;
2329
    }
2330
    /* val is not a valid value */
2331
    av_log(ctx, AV_LOG_DEBUG,
2332
           "%s %d are not supported. Set to default value : %d\n", val_name, val, default_value);
2333
    return default_value;
2334
}