GCC Code Coverage Report
Directory: ../../../ffmpeg/ Exec Total Coverage
File: src/libavcodec/h265_metadata_bsf.c Lines: 46 175 26.3 %
Date: 2021-04-20 15:25:36 Branches: 35 164 21.3 %

Line Branch Exec Source
1
/*
2
 * This file is part of FFmpeg.
3
 *
4
 * FFmpeg is free software; you can redistribute it and/or
5
 * modify it under the terms of the GNU Lesser General Public
6
 * License as published by the Free Software Foundation; either
7
 * version 2.1 of the License, or (at your option) any later version.
8
 *
9
 * FFmpeg is distributed in the hope that it will be useful,
10
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12
 * Lesser General Public License for more details.
13
 *
14
 * You should have received a copy of the GNU Lesser General Public
15
 * License along with FFmpeg; if not, write to the Free Software
16
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
 */
18
19
#include "libavutil/common.h"
20
#include "libavutil/opt.h"
21
22
#include "bsf.h"
23
#include "cbs.h"
24
#include "cbs_bsf.h"
25
#include "cbs_h265.h"
26
#include "hevc.h"
27
#include "h265_profile_level.h"
28
29
enum {
30
    LEVEL_UNSET = -2,
31
    LEVEL_AUTO  = -1,
32
};
33
34
typedef struct H265MetadataContext {
35
    CBSBSFContext common;
36
37
    H265RawAUD aud_nal;
38
39
    int aud;
40
41
    AVRational sample_aspect_ratio;
42
43
    int video_format;
44
    int video_full_range_flag;
45
    int colour_primaries;
46
    int transfer_characteristics;
47
    int matrix_coefficients;
48
49
    int chroma_sample_loc_type;
50
51
    AVRational tick_rate;
52
    int poc_proportional_to_timing_flag;
53
    int num_ticks_poc_diff_one;
54
55
    int crop_left;
56
    int crop_right;
57
    int crop_top;
58
    int crop_bottom;
59
60
    int level;
61
    int level_guess;
62
    int level_warned;
63
} H265MetadataContext;
64
65
66
static void h265_metadata_guess_level(AVBSFContext *bsf,
67
                                      const CodedBitstreamFragment *au)
68
{
69
    H265MetadataContext *ctx = bsf->priv_data;
70
    const H265LevelDescriptor *desc;
71
    const H265RawProfileTierLevel *ptl = NULL;
72
    const H265RawHRDParameters    *hrd = NULL;
73
    int64_t bit_rate = 0;
74
    int width = 0, height = 0;
75
    int tile_cols = 0, tile_rows = 0;
76
    int max_dec_pic_buffering = 0;
77
    int i;
78
79
    for (i = 0; i < au->nb_units; i++) {
80
        const CodedBitstreamUnit *unit = &au->units[i];
81
82
        if (unit->type == HEVC_NAL_VPS) {
83
            const H265RawVPS *vps = unit->content;
84
85
            ptl = &vps->profile_tier_level;
86
            max_dec_pic_buffering = vps->vps_max_dec_pic_buffering_minus1[0] + 1;
87
88
            if (vps->vps_num_hrd_parameters > 0)
89
                hrd = &vps->hrd_parameters[0];
90
91
        } else if (unit->type == HEVC_NAL_SPS) {
92
            const H265RawSPS *sps = unit->content;
93
94
            ptl = &sps->profile_tier_level;
95
            max_dec_pic_buffering = sps->sps_max_dec_pic_buffering_minus1[0] + 1;
96
97
            width  = sps->pic_width_in_luma_samples;
98
            height = sps->pic_height_in_luma_samples;
99
100
            if (sps->vui.vui_hrd_parameters_present_flag)
101
                hrd = &sps->vui.hrd_parameters;
102
103
        } else if (unit->type == HEVC_NAL_PPS) {
104
            const H265RawPPS *pps = unit->content;
105
106
            if (pps->tiles_enabled_flag) {
107
                tile_cols = pps->num_tile_columns_minus1 + 1;
108
                tile_rows = pps->num_tile_rows_minus1 + 1;
109
            }
110
        }
111
    }
112
113
    if (hrd) {
114
        if (hrd->nal_hrd_parameters_present_flag) {
115
            bit_rate = (hrd->nal_sub_layer_hrd_parameters[0].bit_rate_value_minus1[0] + 1) *
116
                       (INT64_C(1) << hrd->bit_rate_scale + 6);
117
        } else if (hrd->vcl_hrd_parameters_present_flag) {
118
            bit_rate = (hrd->vcl_sub_layer_hrd_parameters[0].bit_rate_value_minus1[0] + 1) *
119
                       (INT64_C(1) << hrd->bit_rate_scale + 6);
120
            // Adjust for VCL vs. NAL limits.
121
            bit_rate = bit_rate * 11 / 10;
122
        }
123
    }
124
125
    desc = ff_h265_guess_level(ptl, bit_rate, width, height,
126
                               0, tile_rows, tile_cols,
127
                               max_dec_pic_buffering);
128
    if (desc) {
129
        av_log(bsf, AV_LOG_DEBUG, "Stream appears to conform to "
130
               "level %s.\n", desc->name);
131
        ctx->level_guess = desc->level_idc;
132
    }
133
}
134
135
98
static void h265_metadata_update_level(AVBSFContext *bsf,
136
                                       uint8_t *level_idc)
137
{
138
98
    H265MetadataContext *ctx = bsf->priv_data;
139
140
98
    if (ctx->level != LEVEL_UNSET) {
141
        if (ctx->level == LEVEL_AUTO) {
142
            if (ctx->level_guess) {
143
                *level_idc = ctx->level_guess;
144
            } else {
145
                if (!ctx->level_warned) {
146
                    av_log(bsf, AV_LOG_WARNING, "Unable to determine level "
147
                           "of stream: using level 8.5.\n");
148
                    ctx->level_warned = 1;
149
                }
150
                *level_idc = 255;
151
            }
152
        } else {
153
            *level_idc = ctx->level;
154
        }
155
    }
156
98
}
157
158
48
static int h265_metadata_update_vps(AVBSFContext *bsf,
159
                                    H265RawVPS *vps)
160
{
161
48
    H265MetadataContext *ctx = bsf->priv_data;
162
163

48
    if (ctx->tick_rate.num && ctx->tick_rate.den) {
164
        int num, den;
165
166
        av_reduce(&num, &den, ctx->tick_rate.num, ctx->tick_rate.den,
167
                  UINT32_MAX > INT_MAX ? UINT32_MAX : INT_MAX);
168
169
        vps->vps_time_scale        = num;
170
        vps->vps_num_units_in_tick = den;
171
172
        vps->vps_timing_info_present_flag = 1;
173
174
        if (ctx->num_ticks_poc_diff_one > 0) {
175
            vps->vps_num_ticks_poc_diff_one_minus1 =
176
                ctx->num_ticks_poc_diff_one - 1;
177
            vps->vps_poc_proportional_to_timing_flag = 1;
178
        } else if (ctx->num_ticks_poc_diff_one == 0) {
179
            vps->vps_poc_proportional_to_timing_flag = 0;
180
        }
181
    }
182
183
48
    h265_metadata_update_level(bsf, &vps->profile_tier_level.general_level_idc);
184
185
48
    return 0;
186
}
187
188
50
static int h265_metadata_update_sps(AVBSFContext *bsf,
189
                                    H265RawSPS *sps)
190
{
191
50
    H265MetadataContext *ctx = bsf->priv_data;
192
50
    int need_vui = 0;
193
    int crop_unit_x, crop_unit_y;
194
195

50
    if (ctx->sample_aspect_ratio.num && ctx->sample_aspect_ratio.den) {
196
        // Table E-1.
197
        static const AVRational sar_idc[] = {
198
            {   0,  0 }, // Unspecified (never written here).
199
            {   1,  1 }, {  12, 11 }, {  10, 11 }, {  16, 11 },
200
            {  40, 33 }, {  24, 11 }, {  20, 11 }, {  32, 11 },
201
            {  80, 33 }, {  18, 11 }, {  15, 11 }, {  64, 33 },
202
            { 160, 99 }, {   4,  3 }, {   3,  2 }, {   2,  1 },
203
        };
204
        int num, den, i;
205
206
        av_reduce(&num, &den, ctx->sample_aspect_ratio.num,
207
                  ctx->sample_aspect_ratio.den, 65535);
208
209
        for (i = 1; i < FF_ARRAY_ELEMS(sar_idc); i++) {
210
            if (num == sar_idc[i].num &&
211
                den == sar_idc[i].den)
212
                break;
213
        }
214
        if (i == FF_ARRAY_ELEMS(sar_idc)) {
215
            sps->vui.aspect_ratio_idc = 255;
216
            sps->vui.sar_width  = num;
217
            sps->vui.sar_height = den;
218
        } else {
219
            sps->vui.aspect_ratio_idc = i;
220
        }
221
        sps->vui.aspect_ratio_info_present_flag = 1;
222
        need_vui = 1;
223
    }
224
225
#define SET_OR_INFER(field, value, present_flag, infer) do { \
226
        if (value >= 0) { \
227
            field = value; \
228
            need_vui = 1; \
229
        } else if (!present_flag) \
230
            field = infer; \
231
    } while (0)
232
233
50
    if (ctx->video_format             >= 0 ||
234
50
        ctx->video_full_range_flag    >= 0 ||
235
50
        ctx->colour_primaries         >= 0 ||
236
50
        ctx->transfer_characteristics >= 0 ||
237
50
        ctx->matrix_coefficients      >= 0) {
238
239
        SET_OR_INFER(sps->vui.video_format, ctx->video_format,
240
                     sps->vui.video_signal_type_present_flag, 5);
241
242
        SET_OR_INFER(sps->vui.video_full_range_flag,
243
                     ctx->video_full_range_flag,
244
                     sps->vui.video_signal_type_present_flag, 0);
245
246
        if (ctx->colour_primaries         >= 0 ||
247
            ctx->transfer_characteristics >= 0 ||
248
            ctx->matrix_coefficients      >= 0) {
249
250
            SET_OR_INFER(sps->vui.colour_primaries,
251
                         ctx->colour_primaries,
252
                         sps->vui.colour_description_present_flag, 2);
253
254
            SET_OR_INFER(sps->vui.transfer_characteristics,
255
                         ctx->transfer_characteristics,
256
                         sps->vui.colour_description_present_flag, 2);
257
258
            SET_OR_INFER(sps->vui.matrix_coefficients,
259
                         ctx->matrix_coefficients,
260
                         sps->vui.colour_description_present_flag, 2);
261
262
            sps->vui.colour_description_present_flag = 1;
263
        }
264
        sps->vui.video_signal_type_present_flag = 1;
265
        need_vui = 1;
266
    }
267
268
50
    if (ctx->chroma_sample_loc_type >= 0) {
269
        sps->vui.chroma_sample_loc_type_top_field =
270
            ctx->chroma_sample_loc_type;
271
        sps->vui.chroma_sample_loc_type_bottom_field =
272
            ctx->chroma_sample_loc_type;
273
        sps->vui.chroma_loc_info_present_flag = 1;
274
        need_vui = 1;
275
    }
276
277

50
    if (ctx->tick_rate.num && ctx->tick_rate.den) {
278
        int num, den;
279
280
        av_reduce(&num, &den, ctx->tick_rate.num, ctx->tick_rate.den,
281
                  UINT32_MAX > INT_MAX ? UINT32_MAX : INT_MAX);
282
283
        sps->vui.vui_time_scale        = num;
284
        sps->vui.vui_num_units_in_tick = den;
285
286
        sps->vui.vui_timing_info_present_flag = 1;
287
        need_vui = 1;
288
289
        if (ctx->num_ticks_poc_diff_one > 0) {
290
            sps->vui.vui_num_ticks_poc_diff_one_minus1 =
291
                ctx->num_ticks_poc_diff_one - 1;
292
            sps->vui.vui_poc_proportional_to_timing_flag = 1;
293
        } else if (ctx->num_ticks_poc_diff_one == 0) {
294
            sps->vui.vui_poc_proportional_to_timing_flag = 0;
295
        }
296
    }
297
298

50
    if (sps->separate_colour_plane_flag || sps->chroma_format_idc == 0) {
299
        crop_unit_x = 1;
300
        crop_unit_y = 1;
301
    } else {
302
50
        crop_unit_x = 1 + (sps->chroma_format_idc < 3);
303
50
        crop_unit_y = 1 + (sps->chroma_format_idc < 2);
304
    }
305
#define CROP(border, unit) do { \
306
        if (ctx->crop_ ## border >= 0) { \
307
            if (ctx->crop_ ## border % unit != 0) { \
308
                av_log(bsf, AV_LOG_ERROR, "Invalid value for crop_%s: " \
309
                       "must be a multiple of %d.\n", #border, unit); \
310
                return AVERROR(EINVAL); \
311
            } \
312
            sps->conf_win_ ## border ## _offset = \
313
                ctx->crop_ ## border / unit; \
314
            sps->conformance_window_flag = 1; \
315
        } \
316
    } while (0)
317

50
    CROP(left,   crop_unit_x);
318

50
    CROP(right,  crop_unit_x);
319

50
    CROP(top,    crop_unit_y);
320

50
    CROP(bottom, crop_unit_y);
321
#undef CROP
322
323
50
    if (need_vui)
324
        sps->vui_parameters_present_flag = 1;
325
326
50
    h265_metadata_update_level(bsf, &sps->profile_tier_level.general_level_idc);
327
328
50
    return 0;
329
}
330
331
2152
static int h265_metadata_update_fragment(AVBSFContext *bsf, AVPacket *pkt,
332
                                         CodedBitstreamFragment *au)
333
{
334
2152
    H265MetadataContext *ctx = bsf->priv_data;
335
    int err, i;
336
337
    // If an AUD is present, it must be the first NAL unit.
338

2152
    if (au->nb_units && au->units[0].type == HEVC_NAL_AUD) {
339
60
        if (ctx->aud == BSF_ELEMENT_REMOVE)
340
            ff_cbs_delete_unit(au, 0);
341
    } else {
342

2092
        if (pkt && ctx->aud == BSF_ELEMENT_INSERT) {
343
            H265RawAUD *aud = &ctx->aud_nal;
344
            int pic_type = 0, temporal_id = 8, layer_id = 0;
345
346
            for (i = 0; i < au->nb_units; i++) {
347
                const H265RawNALUnitHeader *nal = au->units[i].content;
348
                if (!nal)
349
                    continue;
350
                if (nal->nuh_temporal_id_plus1 < temporal_id + 1)
351
                    temporal_id = nal->nuh_temporal_id_plus1 - 1;
352
353
                if (au->units[i].type <= HEVC_NAL_RSV_VCL31) {
354
                    const H265RawSlice *slice = au->units[i].content;
355
                    layer_id = nal->nuh_layer_id;
356
                    if (slice->header.slice_type == HEVC_SLICE_B &&
357
                        pic_type < 2)
358
                        pic_type = 2;
359
                    if (slice->header.slice_type == HEVC_SLICE_P &&
360
                        pic_type < 1)
361
                        pic_type = 1;
362
                }
363
            }
364
365
            aud->nal_unit_header = (H265RawNALUnitHeader) {
366
                .nal_unit_type         = HEVC_NAL_AUD,
367
                .nuh_layer_id          = layer_id,
368
                .nuh_temporal_id_plus1 = temporal_id + 1,
369
            };
370
            aud->pic_type = pic_type;
371
372
            err = ff_cbs_insert_unit_content(au, 0, HEVC_NAL_AUD, aud, NULL);
373
            if (err < 0) {
374
                av_log(bsf, AV_LOG_ERROR, "Failed to insert AUD.\n");
375
                return err;
376
            }
377
        }
378
    }
379
380

2152
    if (ctx->level == LEVEL_AUTO && !ctx->level_guess)
381
        h265_metadata_guess_level(bsf, au);
382
383
9984
    for (i = 0; i < au->nb_units; i++) {
384
7832
        if (au->units[i].type == HEVC_NAL_VPS) {
385
48
            err = h265_metadata_update_vps(bsf, au->units[i].content);
386
48
            if (err < 0)
387
                return err;
388
        }
389
7832
        if (au->units[i].type == HEVC_NAL_SPS) {
390
50
            err = h265_metadata_update_sps(bsf, au->units[i].content);
391
50
            if (err < 0)
392
                return err;
393
        }
394
    }
395
396
2152
    return 0;
397
}
398
399
static const CBSBSFType h265_metadata_type = {
400
    .codec_id        = AV_CODEC_ID_HEVC,
401
    .fragment_name   = "access unit",
402
    .unit_name       = "NAL unit",
403
    .update_fragment = &h265_metadata_update_fragment,
404
};
405
406
20
static int h265_metadata_init(AVBSFContext *bsf)
407
{
408
20
    return ff_cbs_bsf_generic_init(bsf, &h265_metadata_type);
409
}
410
411
#define OFFSET(x) offsetof(H265MetadataContext, x)
412
#define FLAGS (AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_BSF_PARAM)
413
static const AVOption h265_metadata_options[] = {
414
    BSF_ELEMENT_OPTIONS_PIR("aud", "Access Unit Delimiter NAL units",
415
                            aud, FLAGS),
416
417
    { "sample_aspect_ratio", "Set sample aspect ratio (table E-1)",
418
        OFFSET(sample_aspect_ratio), AV_OPT_TYPE_RATIONAL,
419
        { .dbl = 0.0 }, 0, 65535, FLAGS },
420
421
    { "video_format", "Set video format (table E-2)",
422
        OFFSET(video_format), AV_OPT_TYPE_INT,
423
        { .i64 = -1 }, -1, 7, FLAGS },
424
    { "video_full_range_flag", "Set video full range flag",
425
        OFFSET(video_full_range_flag), AV_OPT_TYPE_INT,
426
        { .i64 = -1 }, -1, 1, FLAGS },
427
    { "colour_primaries", "Set colour primaries (table E-3)",
428
        OFFSET(colour_primaries), AV_OPT_TYPE_INT,
429
        { .i64 = -1 }, -1, 255, FLAGS },
430
    { "transfer_characteristics", "Set transfer characteristics (table E-4)",
431
        OFFSET(transfer_characteristics), AV_OPT_TYPE_INT,
432
        { .i64 = -1 }, -1, 255, FLAGS },
433
    { "matrix_coefficients", "Set matrix coefficients (table E-5)",
434
        OFFSET(matrix_coefficients), AV_OPT_TYPE_INT,
435
        { .i64 = -1 }, -1, 255, FLAGS },
436
437
    { "chroma_sample_loc_type", "Set chroma sample location type (figure E-1)",
438
        OFFSET(chroma_sample_loc_type), AV_OPT_TYPE_INT,
439
        { .i64 = -1 }, -1, 6, FLAGS },
440
441
    { "tick_rate",
442
        "Set VPS and VUI tick rate (num_units_in_tick / time_scale)",
443
        OFFSET(tick_rate), AV_OPT_TYPE_RATIONAL,
444
        { .dbl = 0.0 }, 0, UINT_MAX, FLAGS },
445
    { "num_ticks_poc_diff_one",
446
        "Set VPS and VUI number of ticks per POC increment",
447
        OFFSET(num_ticks_poc_diff_one), AV_OPT_TYPE_INT,
448
        { .i64 = -1 }, -1, INT_MAX, FLAGS },
449
450
    { "crop_left", "Set left border crop offset",
451
        OFFSET(crop_left), AV_OPT_TYPE_INT,
452
        { .i64 = -1 }, -1, HEVC_MAX_WIDTH, FLAGS },
453
    { "crop_right", "Set right border crop offset",
454
        OFFSET(crop_right), AV_OPT_TYPE_INT,
455
        { .i64 = -1 }, -1, HEVC_MAX_WIDTH, FLAGS },
456
    { "crop_top", "Set top border crop offset",
457
        OFFSET(crop_top), AV_OPT_TYPE_INT,
458
        { .i64 = -1 }, -1, HEVC_MAX_HEIGHT, FLAGS },
459
    { "crop_bottom", "Set bottom border crop offset",
460
        OFFSET(crop_bottom), AV_OPT_TYPE_INT,
461
        { .i64 = -1 }, -1, HEVC_MAX_HEIGHT, FLAGS },
462
463
    { "level", "Set level (tables A.6 and A.7)",
464
        OFFSET(level), AV_OPT_TYPE_INT,
465
        { .i64 = LEVEL_UNSET }, LEVEL_UNSET, 0xff, FLAGS, "level" },
466
    { "auto", "Attempt to guess level from stream properties",
467
        0, AV_OPT_TYPE_CONST,
468
        { .i64 = LEVEL_AUTO }, .flags = FLAGS, .unit = "level" },
469
#define LEVEL(name, value) name, NULL, 0, AV_OPT_TYPE_CONST, \
470
        { .i64 = value },      .flags = FLAGS, .unit = "level"
471
    { LEVEL("1",    30) },
472
    { LEVEL("2",    60) },
473
    { LEVEL("2.1",  63) },
474
    { LEVEL("3",    90) },
475
    { LEVEL("3.1",  93) },
476
    { LEVEL("4",   120) },
477
    { LEVEL("4.1", 123) },
478
    { LEVEL("5",   150) },
479
    { LEVEL("5.1", 153) },
480
    { LEVEL("5.2", 156) },
481
    { LEVEL("6",   180) },
482
    { LEVEL("6.1", 183) },
483
    { LEVEL("6.2", 186) },
484
    { LEVEL("8.5", 255) },
485
#undef LEVEL
486
487
    { NULL }
488
};
489
490
static const AVClass h265_metadata_class = {
491
    .class_name = "h265_metadata_bsf",
492
    .item_name  = av_default_item_name,
493
    .option     = h265_metadata_options,
494
    .version    = LIBAVUTIL_VERSION_INT,
495
};
496
497
static const enum AVCodecID h265_metadata_codec_ids[] = {
498
    AV_CODEC_ID_HEVC, AV_CODEC_ID_NONE,
499
};
500
501
const AVBitStreamFilter ff_hevc_metadata_bsf = {
502
    .name           = "hevc_metadata",
503
    .priv_data_size = sizeof(H265MetadataContext),
504
    .priv_class     = &h265_metadata_class,
505
    .init           = &h265_metadata_init,
506
    .close          = &ff_cbs_bsf_generic_close,
507
    .filter         = &ff_cbs_bsf_generic_filter,
508
    .codec_ids      = h265_metadata_codec_ids,
509
};