/src/ffmpeg/libavformat/vvcdec.c
Line | Count | Source |
1 | | /* |
2 | | * RAW H.266 / VVC video demuxer |
3 | | * Copyright (c) 2020 Nuo Mi <nuomi2021@gmail.com> |
4 | | * |
5 | | * This file is part of FFmpeg. |
6 | | * |
7 | | * FFmpeg is free software; you can redistribute it and/or |
8 | | * modify it under the terms of the GNU Lesser General Public |
9 | | * License as published by the Free Software Foundation; either |
10 | | * version 2.1 of the License, or (at your option) any later version. |
11 | | * |
12 | | * FFmpeg is distributed in the hope that it will be useful, |
13 | | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
14 | | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
15 | | * Lesser General Public License for more details. |
16 | | * |
17 | | * You should have received a copy of the GNU Lesser General Public |
18 | | * License along with FFmpeg; if not, write to the Free Software |
19 | | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
20 | | */ |
21 | | |
22 | | #include "libavcodec/vvc.h" |
23 | | |
24 | | #include "avformat.h" |
25 | | #include "rawdec.h" |
26 | | |
27 | | static int check_temporal_id(uint8_t nuh_temporal_id_plus1, int type) |
28 | 805k | { |
29 | 805k | if (nuh_temporal_id_plus1 == 0) |
30 | 203k | return 0; |
31 | | |
32 | 602k | if (nuh_temporal_id_plus1 != 1) { |
33 | 319k | if (type >= VVC_IDR_W_RADL && type <= VVC_RSV_IRAP_11 |
34 | 285k | || type == VVC_DCI_NUT || type == VVC_OPI_NUT |
35 | 278k | || type == VVC_VPS_NUT || type == VVC_SPS_NUT |
36 | 269k | || type == VVC_EOS_NUT || type == VVC_EOB_NUT) |
37 | 116k | return 0; |
38 | 319k | } |
39 | | |
40 | 486k | return 1; |
41 | 602k | } |
42 | | |
43 | | static int vvc_probe(const AVProbeData *p) |
44 | 959k | { |
45 | 959k | uint32_t code = -1; |
46 | 959k | int sps = 0, pps = 0, irap = 0; |
47 | 959k | int valid_pps = 0, valid_irap = 0; |
48 | 959k | int i; |
49 | | |
50 | 390M | for (i = 0; i < p->buf_size - 1; i++) { |
51 | 389M | code = (code << 8) + p->buf[i]; |
52 | 389M | if ((code & 0xffffff00) == 0x100) { |
53 | 1.01M | uint8_t nal2 = p->buf[i + 1]; |
54 | 1.01M | int type = (nal2 & 0xF8) >> 3; |
55 | | |
56 | 1.01M | if (code & 0x80) // forbidden_zero_bit |
57 | 198k | return 0; |
58 | | |
59 | 815k | if ((code & 0x3F) > 55) // nuh_layer_id must be in [0, 55] |
60 | 9.64k | return 0; |
61 | | |
62 | 805k | if (!check_temporal_id(nal2 & 0x7, type)) |
63 | 319k | return 0; |
64 | | |
65 | 486k | switch (type) { |
66 | 24.6k | case VVC_SPS_NUT: sps++; break; |
67 | 25.5k | case VVC_PPS_NUT: |
68 | 25.5k | pps++; |
69 | 25.5k | if (sps) |
70 | 19.0k | valid_pps++; |
71 | 25.5k | break; |
72 | 18.3k | case VVC_IDR_N_LP: |
73 | 20.9k | case VVC_IDR_W_RADL: |
74 | 28.6k | case VVC_CRA_NUT: |
75 | 32.6k | case VVC_GDR_NUT: |
76 | 32.6k | irap++; |
77 | 32.6k | if (valid_pps) |
78 | 12.8k | valid_irap++; |
79 | 32.6k | break; |
80 | 486k | } |
81 | 486k | } |
82 | 389M | } |
83 | | |
84 | 432k | if (valid_irap) |
85 | 3.31k | return AVPROBE_SCORE_EXTENSION + 1; // 1 more than .mpg |
86 | 428k | if (sps && pps && irap) |
87 | 478 | return AVPROBE_SCORE_EXTENSION / 2; |
88 | 428k | if (sps || pps || irap) |
89 | 6.04k | return AVPROBE_SCORE_EXTENSION / 4; |
90 | 422k | return 0; |
91 | 428k | } |
92 | | |
93 | | FF_DEF_RAWVIDEO_DEMUXER(vvc, "raw H.266/VVC video", vvc_probe, "h266,266,vvc", AV_CODEC_ID_VVC) |