FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/libavformat/vpcc.c
Date: 2024-05-03 15:42:48
Exec Total Coverage
Lines: 49 116 42.2%
Functions: 6 7 85.7%
Branches: 20 130 15.4%
Line Branch Exec Source
1 /*
2 * Copyright (c) 2016 Google Inc.
3 * Copyright (c) 2016 KongQun Yang (kqyang@google.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 "libavutil/pixdesc.h"
23 #include "libavutil/pixfmt.h"
24 #include "libavcodec/defs.h"
25 #include "libavcodec/get_bits.h"
26 #include "vpcc.h"
27
28 #define VP9_SYNCCODE 0x498342
29
30 enum VPX_CHROMA_SUBSAMPLING
31 {
32 VPX_SUBSAMPLING_420_VERTICAL = 0,
33 VPX_SUBSAMPLING_420_COLLOCATED_WITH_LUMA = 1,
34 VPX_SUBSAMPLING_422 = 2,
35 VPX_SUBSAMPLING_444 = 3,
36 };
37
38 1 static int get_vpx_chroma_subsampling(AVFormatContext *s,
39 enum AVPixelFormat pixel_format,
40 enum AVChromaLocation chroma_location)
41 {
42 int chroma_w, chroma_h;
43
1/2
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
 1 if (av_pix_fmt_get_chroma_sub_sample(pixel_format, &chroma_w, &chroma_h) == 0) {
44
2/4
✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
 1 if (chroma_w == 1 && chroma_h == 1) {
45 return (chroma_location == AVCHROMA_LOC_LEFT)
46 ? VPX_SUBSAMPLING_420_VERTICAL
47 1 : VPX_SUBSAMPLING_420_COLLOCATED_WITH_LUMA;
48 ✗ } else if (chroma_w == 1 && chroma_h == 0) {
49 ✗ return VPX_SUBSAMPLING_422;
50 ✗ } else if (chroma_w == 0 && chroma_h == 0) {
51 ✗ return VPX_SUBSAMPLING_444;
52 }
53 }
54 ✗ av_log(s, AV_LOG_ERROR, "Unsupported pixel format (%d)\n", pixel_format);
55 ✗ return -1;
56 }
57
58 1 static int get_bit_depth(AVFormatContext *s, enum AVPixelFormat pixel_format)
59 {
60 1 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(pixel_format);
61
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
 1 if (desc == NULL) {
62 ✗ av_log(s, AV_LOG_ERROR, "Unsupported pixel format (%d)\n",
63 pixel_format);
64 ✗ return -1;
65 }
66 1 return desc->comp[0].depth;
67 }
68
69 1 static int get_vpx_video_full_range_flag(enum AVColorRange color_range)
70 {
71 1 return color_range == AVCOL_RANGE_JPEG;
72 }
73
74 // Find approximate VP9 level based on the Luma's Sample rate and Picture size.
75 1 static int get_vp9_level(AVCodecParameters *par, AVRational *frame_rate) {
76 1 int picture_size = par->width * par->height;
77 int64_t sample_rate;
78
79 // All decisions will be based on picture_size, if frame rate is missing/invalid
80
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 1 if (!frame_rate || !frame_rate->den)
81 1 sample_rate = 0;
82 else
83 ✗ sample_rate = ((int64_t)picture_size * frame_rate->num) / frame_rate->den;
84
85
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
 1 if (picture_size <= 0) {
86 ✗ return 0;
87
2/4
✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 1 times.
 1 } else if (sample_rate <= 829440 && picture_size <= 36864) {
88 ✗ return 10;
89
2/4
✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 1 times.
 1 } else if (sample_rate <= 2764800 && picture_size <= 73728) {
90 ✗ return 11;
91
2/4
✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
 1 } else if (sample_rate <= 4608000 && picture_size <= 122880) {
92 1 return 20;
93 ✗ } else if (sample_rate <= 9216000 && picture_size <= 245760) {
94 ✗ return 21;
95 ✗ } else if (sample_rate <= 20736000 && picture_size <= 552960) {
96 ✗ return 30;
97 ✗ } else if (sample_rate <= 36864000 && picture_size <= 983040) {
98 ✗ return 31;
99 ✗ } else if (sample_rate <= 83558400 && picture_size <= 2228224) {
100 ✗ return 40;
101 ✗ } else if (sample_rate <= 160432128 && picture_size <= 2228224) {
102 ✗ return 41;
103 ✗ } else if (sample_rate <= 311951360 && picture_size <= 8912896) {
104 ✗ return 50;
105 ✗ } else if (sample_rate <= 588251136 && picture_size <= 8912896) {
106 ✗ return 51;
107 ✗ } else if (sample_rate <= 1176502272 && picture_size <= 8912896) {
108 ✗ return 52;
109 ✗ } else if (sample_rate <= 1176502272 && picture_size <= 35651584) {
110 ✗ return 60;
111 ✗ } else if (sample_rate <= 2353004544 && picture_size <= 35651584) {
112 ✗ return 61;
113 ✗ } else if (sample_rate <= 4706009088 && picture_size <= 35651584) {
114 ✗ return 62;
115 } else {
116 ✗ return 0;
117 }
118 }
119
120 ✗ static void parse_bitstream(GetBitContext *gb, int *profile, int *bit_depth) {
121 int keyframe, invisible;
122
123 ✗ if (get_bits(gb, 2) != 0x2) // frame marker
124 ✗ return;
125 ✗ *profile = get_bits1(gb);
126 ✗ *profile |= get_bits1(gb) << 1;
127 ✗ if (*profile == 3)
128 ✗ *profile += get_bits1(gb);
129
130 ✗ if (get_bits1(gb))
131 ✗ return;
132
133 ✗ keyframe = !get_bits1(gb);
134 ✗ invisible = !get_bits1(gb);
135 ✗ get_bits1(gb);
136
137 ✗ if (keyframe) {
138 ✗ if (get_bits(gb, 24) != VP9_SYNCCODE)
139 ✗ return;
140 } else {
141 ✗ int intraonly = invisible ? get_bits1(gb) : 0;
142 ✗ if (!intraonly || get_bits(gb, 24) != VP9_SYNCCODE)
143 ✗ return;
144 ✗ if (*profile < 1) {
145 ✗ *bit_depth = 8;
146 ✗ return;
147 }
148 }
149
150 ✗ *bit_depth = *profile <= 1 ? 8 : 10 + get_bits1(gb) * 2;
151 }
152
153 1 int ff_isom_get_vpcc_features(AVFormatContext *s, AVCodecParameters *par,
154 const uint8_t *data, int len,
155 AVRational *frame_rate, VPCC *vpcc)
156 {
157 1 int profile = par->profile;
158 2 int level = par->level == AV_LEVEL_UNKNOWN ?
159
1/2
✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
 1 get_vp9_level(par, frame_rate) : par->level;
160 1 int bit_depth = get_bit_depth(s, par->format);
161 int vpx_chroma_subsampling =
162 1 get_vpx_chroma_subsampling(s, par->format, par->chroma_location);
163 int vpx_video_full_range_flag =
164 1 get_vpx_video_full_range_flag(par->color_range);
165
166
2/4
✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 1 times.
 1 if (bit_depth < 0 || vpx_chroma_subsampling < 0)
167 ✗ return AVERROR_INVALIDDATA;
168
169
1/6
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
 1 if (len && (profile == AV_PROFILE_UNKNOWN || !bit_depth)) {
170 GetBitContext gb;
171
172 ✗ int ret = init_get_bits8(&gb, data, len);
173 ✗ if (ret < 0)
174 ✗ return ret;
175
176 ✗ parse_bitstream(&gb, &profile, &bit_depth);
177 }
178
179
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 1 if (profile == AV_PROFILE_UNKNOWN && bit_depth) {
180 ✗ if (vpx_chroma_subsampling == VPX_SUBSAMPLING_420_VERTICAL ||
181 vpx_chroma_subsampling == VPX_SUBSAMPLING_420_COLLOCATED_WITH_LUMA) {
182 ✗ profile = (bit_depth == 8) ? AV_PROFILE_VP9_0 : AV_PROFILE_VP9_2;
183 } else {
184 ✗ profile = (bit_depth == 8) ? AV_PROFILE_VP9_1 : AV_PROFILE_VP9_3;
185 }
186 }
187
188
2/4
✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 1 times.
 1 if (profile == AV_PROFILE_UNKNOWN || !bit_depth)
189 ✗ av_log(s, AV_LOG_WARNING, "VP9 profile and/or bit depth not set or could not be derived\n");
190
191 1 vpcc->profile = profile;
192 1 vpcc->level = level;
193 1 vpcc->bitdepth = bit_depth;
194 1 vpcc->chroma_subsampling = vpx_chroma_subsampling;
195 1 vpcc->full_range_flag = vpx_video_full_range_flag;
196
197 1 return 0;
198 }
199
200 1 int ff_isom_write_vpcc(AVFormatContext *s, AVIOContext *pb,
201 const uint8_t *data, int len,
202 AVCodecParameters *par)
203 {
204 VPCC vpcc;
205 int ret;
206
207 1 ret = ff_isom_get_vpcc_features(s, par, data, len, NULL, &vpcc);
208
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
 1 if (ret < 0)
209 ✗ return ret;
210
211 1 avio_w8(pb, 1); /* version */
212 1 avio_wb24(pb, 0); /* flags */
213 1 avio_w8(pb, vpcc.profile);
214 1 avio_w8(pb, vpcc.level);
215 1 avio_w8(pb, (vpcc.bitdepth << 4) | (vpcc.chroma_subsampling << 1) | vpcc.full_range_flag);
216 1 avio_w8(pb, par->color_primaries);
217 1 avio_w8(pb, par->color_trc);
218 1 avio_w8(pb, par->color_space);
219
220 // vp9 does not have codec initialization data.
221 1 avio_wb16(pb, 0);
222 1 return 0;
223 }
224