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 "h265_profile_level.h"
20
21
22static const H265LevelDescriptor h265_levels[] = {
23    // Name             CpbFactor-Main    MaxSliceSegmentsPerPicture
24    // |  level_idc            | CpbFactor-High           MaxLumaSr      BrFactor-High
25    // |      |   MaxLumaPs    |       |      | MaxTileRows   |   BrFactor-Main | MinCr-Main
26    // |      |      |         |       |      |   | MaxTileCols         |       |    |  MinCr-High
27    { "1",    30,    36864,    350,      0,  16,  1,  1,     552960,    128,      0, 2, 2 },
28    { "2",    60,   122880,   1500,      0,  16,  1,  1,    3686400,   1500,      0, 2, 2 },
29    { "2.1",  63,   245760,   3000,      0,  20,  1,  1,    7372800,   3000,      0, 2, 2 },
30    { "3",    90,   552960,   6000,      0,  30,  2,  2,   16588800,   6000,      0, 2, 2 },
31    { "3.1",  93,   983040,  10000,      0,  40,  3,  3,   33177600,  10000,      0, 2, 2 },
32    { "4",   120,  2228224,  12000,  30000,  75,  5,  5,   66846720,  12000,  30000, 4, 4 },
33    { "4.1", 123,  2228224,  20000,  50000,  75,  5,  5,  133693440,  20000,  50000, 4, 4 },
34    { "5",   150,  8912896,  25000, 100000, 200, 11, 10,  267386880,  25000, 100000, 6, 4 },
35    { "5.1", 153,  8912896,  40000, 160000, 200, 11, 10,  534773760,  40000, 160000, 8, 4 },
36    { "5.2", 156,  8912896,  60000, 240000, 200, 11, 10, 1069547520,  60000, 240000, 8, 4 },
37    { "6",   180, 35651584,  60000, 240000, 600, 22, 20, 1069547520,  60000, 240000, 8, 4 },
38    { "6.1", 183, 35651584, 120000, 480000, 600, 22, 20, 2139095040, 120000, 480000, 8, 4 },
39    { "6.2", 186, 35651584, 240000, 800000, 600, 22, 20, 4278190080, 240000, 800000, 6, 4 },
40};
41
42static const H265ProfileDescriptor h265_profiles[] = {
43    // profile_idc   8bit       one-picture
44    //   HT-profile  | 422chroma    | lower-bit-rate
45    //   |  14bit    |  | 420chroma |  | CpbVclFactor     MinCrScaleFactor
46    //   |  |  12bit |  |  | monochrome|    | CpbNalFactor    | maxDpbPicBuf
47    //   |  |  |  10bit |  |  | intra  |    |     | FormatCapabilityFactor
48    { "Monochrome", //  |  |  |  |  |  |    |     |     |     |   |
49      4, 0, 2, 1, 1, 1, 1, 1, 1, 0, 0, 1,  667,  733, 1.000, 1.0, 6 },
50    { "Monochrome 10",
51      4, 0, 2, 1, 1, 0, 1, 1, 1, 0, 0, 1,  833,  917, 1.250, 1.0, 6 },
52    { "Monochrome 12",
53      4, 0, 2, 1, 0, 0, 1, 1, 1, 0, 0, 1, 1000, 1100, 1.500, 1.0, 6 },
54    { "Monochrome 16",
55      4, 0, 2, 0, 0, 0, 1, 1, 1, 0, 0, 1, 1333, 1467, 2.000, 1.0, 6 },
56    { "Main",
57      1, 0, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 1000, 1100, 1.500, 1.0, 6 },
58    { "Screen-Extended Main",
59      9, 0, 1, 1, 1, 1, 1, 1, 0, 0, 0, 1, 1000, 1100, 1.500, 1.0, 7 },
60    { "Main 10",
61      2, 0, 2, 2, 2, 2, 2, 2, 2, 2, 0, 2, 1000, 1100, 1.875, 1.0, 6 },
62    { "Screen-Extended Main 10",
63      9, 0, 1, 1, 1, 0, 1, 1, 0, 0, 0, 1, 1000, 1100, 1.875, 1.0, 7 },
64    { "Main 12",
65      4, 0, 2, 1, 0, 0, 1, 1, 0, 0, 0, 1, 1500, 1650, 2.250, 1.0, 6 },
66    { "Main Still Picture",
67      3, 0, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 1000, 1100, 1.500, 1.0, 6 },
68    { "Main 10 Still Picture",
69      2, 0, 2, 2, 2, 2, 2, 2, 2, 2, 1, 2, 1000, 1100, 1.875, 1.0, 6 },
70    { "Main 4:2:2 10",
71      4, 0, 2, 1, 1, 0, 1, 0, 0, 0, 0, 1, 1667, 1833, 2.500, 0.5, 6 },
72    { "Main 4:2:2 12",
73      4, 0, 2, 1, 0, 0, 1, 0, 0, 0, 0, 1, 2000, 2200, 3.000, 0.5, 6 },
74    { "Main 4:4:4",
75      4, 0, 2, 1, 1, 1, 0, 0, 0, 0, 0, 1, 2000, 2200, 3.000, 0.5, 6 },
76    { "High Throughput 4:4:4",
77      5, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 1, 2000, 2200, 3.000, 0.5, 6 },
78    { "Screen-Extended Main 4:4:4",
79      9, 0, 1, 1, 1, 1, 0, 0, 0, 0, 0, 1, 2000, 2200, 3.000, 0.5, 7 },
80    { "Screen-Extended High Throughput 4:4:4",
81      9, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 1, 2000, 2200, 3.000, 0.5, 7 },
82    { "Main 4:4:4 10",
83      4, 0, 2, 1, 1, 0, 0, 0, 0, 0, 0, 1, 2500, 2750, 3.750, 0.5, 6 },
84    { "High Throughput 4:4:4 10",
85      5, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 1, 2500, 2750, 3.750, 0.5, 6 },
86    { "Screen-Extended Main 4:4:4 10",
87      9, 0, 1, 1, 1, 0, 0, 0, 0, 0, 0, 1, 2500, 2750, 3.750, 0.5, 7 },
88    { "Screen-Extended High Throughput 4:4:4 10",
89      9, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 1, 2500, 2750, 3.750, 0.5, 7 },
90    { "Main 4:4:4 12",
91      4, 0, 2, 1, 0, 0, 0, 0, 0, 0, 0, 1, 3000, 3300, 4.500, 0.5, 6 },
92    { "High Throughput 4:4:4 14",
93      5, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 1, 3500, 3850, 5.250, 0.5, 6 },
94    { "Screen-Extended High Throughput 4:4:4 14",
95      9, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 1, 3500, 3850, 5.250, 0.5, 7 },
96    { "Main Intra",
97      4, 0, 2, 1, 1, 1, 1, 1, 0, 1, 0, 2, 1000, 1100, 1.500, 1.0, 6 },
98    { "Main 10 Intra",
99      4, 0, 2, 1, 1, 0, 1, 1, 0, 1, 0, 2, 1000, 1100, 1.875, 1.0, 6 },
100    { "Main 12 Intra",
101      4, 0, 2, 1, 0, 0, 1, 1, 0, 1, 0, 2, 1500, 1650, 2.250, 1.0, 6 },
102    { "Main 4:2:2 10 Intra",
103      4, 0, 2, 1, 1, 0, 1, 0, 0, 1, 0, 2, 1667, 1833, 2.500, 0.5, 6 },
104    { "Main 4:2:2 12 Intra",
105      4, 0, 2, 1, 0, 0, 1, 0, 0, 1, 0, 2, 2000, 2200, 3.000, 0.5, 6 },
106    { "Main 4:4:4 Intra",
107      4, 0, 2, 1, 1, 1, 0, 0, 0, 1, 0, 2, 2000, 2200, 3.000, 0.5, 6 },
108    { "Main 4:4:4 10 Intra",
109      4, 0, 2, 1, 1, 0, 0, 0, 0, 1, 0, 2, 2500, 2750, 3.750, 0.5, 6 },
110    { "Main 4:4:4 12 Intra",
111      4, 0, 2, 1, 0, 0, 0, 0, 0, 1, 0, 2, 3000, 3300, 4.500, 0.5, 6 },
112    { "Main 4:4:4 16 Intra",
113      4, 0, 2, 0, 0, 0, 0, 0, 0, 1, 0, 2, 4000, 4400, 6.000, 0.5, 6 },
114    { "Main 4:4:4 Still Picture",
115      4, 0, 2, 1, 1, 1, 0, 0, 0, 1, 1, 2, 2000, 2200, 3.000, 0.5, 6 },
116    { "Main 4:4:4 16 Still Picture",
117      4, 0, 2, 0, 0, 0, 0, 0, 0, 1, 1, 2, 4000, 4400, 6.000, 0.5, 6 },
118    { "High Throughput 4:4:4 16 Intra",
119      5, 1, 0, 0, 0, 0, 0, 0, 0, 1, 0, 2, 4000, 4400, 6.000, 0.5, 6 },
120};
121
122
123const H265ProfileDescriptor *ff_h265_get_profile(const H265RawProfileTierLevel *ptl)
124{
125    int i;
126
127    if (ptl->general_profile_space)
128        return NULL;
129
130    for (i = 0; i < FF_ARRAY_ELEMS(h265_profiles); i++) {
131        const H265ProfileDescriptor *profile = &h265_profiles[i];
132
133        if (ptl->general_profile_idc &&
134            ptl->general_profile_idc != profile->profile_idc)
135            continue;
136        if (!ptl->general_profile_compatibility_flag[profile->profile_idc])
137            continue;
138
139#define check_flag(name) \
140        if (profile->name < 2) { \
141            if (profile->name != ptl->general_ ## name ## _constraint_flag) \
142                continue; \
143        }
144        check_flag(max_14bit);
145        check_flag(max_12bit);
146        check_flag(max_10bit);
147        check_flag(max_8bit);
148        check_flag(max_422chroma);
149        check_flag(max_420chroma);
150        check_flag(max_monochrome);
151        check_flag(intra);
152        check_flag(one_picture_only);
153        check_flag(lower_bit_rate);
154#undef check_flag
155
156        return profile;
157    }
158
159    return NULL;
160}
161
162const H265LevelDescriptor *ff_h265_guess_level(const H265RawProfileTierLevel *ptl,
163                                               int64_t bitrate,
164                                               int width, int height,
165                                               int slice_segments,
166                                               int tile_rows, int tile_cols,
167                                               int max_dec_pic_buffering)
168{
169    const H265ProfileDescriptor *profile;
170    int pic_size, tier_flag, lbr_flag, hbr_factor;
171    int i;
172
173    if (ptl)
174        profile = ff_h265_get_profile(ptl);
175    else
176        profile = NULL;
177    if (!profile) {
178        // Default to using multiplication factors for Main profile.
179        profile = &h265_profiles[4];
180    }
181
182    pic_size = width * height;
183
184    if (ptl) {
185        tier_flag = ptl->general_tier_flag;
186        lbr_flag  = ptl->general_lower_bit_rate_constraint_flag;
187    } else {
188        tier_flag = 0;
189        lbr_flag  = profile->lower_bit_rate > 0;
190    }
191    if (profile->profile_idc == 1 || profile->profile_idc == 2) {
192        hbr_factor = 1;
193    } else if (profile->high_throughput) {
194        if (profile->intra)
195            hbr_factor = 24 - 12 * lbr_flag;
196        else
197            hbr_factor = 6;
198    } else {
199        hbr_factor = 2 - lbr_flag;
200    }
201
202    for (i = 0; i < FF_ARRAY_ELEMS(h265_levels); i++) {
203        const H265LevelDescriptor *level = &h265_levels[i];
204        int max_br, max_dpb_size;
205
206        if (tier_flag && !level->max_br_high)
207            continue;
208
209        if (pic_size > level->max_luma_ps)
210            continue;
211        if (width  * width  > 8 * level->max_luma_ps)
212            continue;
213        if (height * height > 8 * level->max_luma_ps)
214            continue;
215
216        if (slice_segments > level->max_slice_segments_per_picture)
217            continue;
218        if (tile_rows > level->max_tile_rows)
219            continue;
220        if (tile_cols > level->max_tile_cols)
221            continue;
222
223        if (tier_flag)
224            max_br = level->max_br_high;
225        else
226            max_br = level->max_br_main;
227        if (!max_br)
228            continue;
229        if (bitrate > (int64_t)profile->cpb_nal_factor * hbr_factor * max_br)
230            continue;
231
232        if (pic_size <= (level->max_luma_ps >> 2))
233            max_dpb_size = FFMIN(4 * profile->max_dpb_pic_buf, 16);
234        else if (pic_size <= (level->max_luma_ps >> 1))
235            max_dpb_size = FFMIN(2 * profile->max_dpb_pic_buf, 16);
236        else if (pic_size <= (3 * level->max_luma_ps >> 2))
237            max_dpb_size = FFMIN(4 * profile->max_dpb_pic_buf / 3, 16);
238        else
239            max_dpb_size = profile->max_dpb_pic_buf;
240        if (max_dec_pic_buffering > max_dpb_size)
241            continue;
242
243        return level;
244    }
245
246    return NULL;
247}
248