git.sesse.net Git - ffmpeg/blob - libavcodec/h264_levels.c

   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 "avcodec.h"
  20 #include "h264_levels.h"
  21
  22 // H.264 table A-1.
  23 static const H264LevelDescriptor h264_levels[] = {
  24     // Name          MaxMBPS                   MaxBR              MinCR
  25     //  | level_idc     |       MaxFS            |    MaxCPB        | MaxMvsPer2Mb
  26     //  |     | cs3f    |         |  MaxDpbMbs   |       |  MaxVmvR |   |
  27     { "1",   10, 0,     1485,     99,    396,     64,    175,   64, 2,  0 },
  28     { "1b",  10, 1,     1485,     99,    396,    128,    350,   64, 2,  0 },
  29     { "1b",   9, 0,     1485,     99,    396,    128,    350,   64, 2,  0 },
  30     { "1.1", 11, 0,     3000,    396,    900,    192,    500,  128, 2,  0 },
  31     { "1.2", 12, 0,     6000,    396,   2376,    384,   1000,  128, 2,  0 },
  32     { "1.3", 13, 0,    11880,    396,   2376,    768,   2000,  128, 2,  0 },
  33     { "2",   20, 0,    11880,    396,   2376,   2000,   2000,  128, 2,  0 },
  34     { "2.1", 21, 0,    19800,    792,   4752,   4000,   4000,  256, 2,  0 },
  35     { "2.2", 22, 0,    20250,   1620,   8100,   4000,   4000,  256, 2,  0 },
  36     { "3",   30, 0,    40500,   1620,   8100,  10000,  10000,  256, 2, 32 },
  37     { "3.1", 31, 0,   108000,   3600,  18000,  14000,  14000,  512, 4, 16 },
  38     { "3.2", 32, 0,   216000,   5120,  20480,  20000,  20000,  512, 4, 16 },
  39     { "4",   40, 0,   245760,   8192,  32768,  20000,  25000,  512, 4, 16 },
  40     { "4.1", 41, 0,   245760,   8192,  32768,  50000,  62500,  512, 2, 16 },
  41     { "4.2", 42, 0,   522240,   8704,  34816,  50000,  62500,  512, 2, 16 },
  42     { "5",   50, 0,   589824,  22080, 110400, 135000, 135000,  512, 2, 16 },
  43     { "5.1", 51, 0,   983040,  36864, 184320, 240000, 240000,  512, 2, 16 },
  44     { "5.2", 52, 0,  2073600,  36864, 184320, 240000, 240000,  512, 2, 16 },
  45     { "6",   60, 0,  4177920, 139264, 696320, 240000, 240000, 8192, 2, 16 },
  46     { "6.1", 61, 0,  8355840, 139264, 696320, 480000, 480000, 8192, 2, 16 },
  47     { "6.2", 62, 0, 16711680, 139264, 696320, 800000, 800000, 8192, 2, 16 },
  48 };
  49
  50 // H.264 table A-2 plus values from A-1.
  51 static const struct {
  52     int profile_idc;
  53     int cpb_br_vcl_factor;
  54     int cpb_br_nal_factor;
  55 } h264_br_factors[] = {
  56     {  66, 1000, 1200 },
  57     {  77, 1000, 1200 },
  58     {  88, 1000, 1200 },
  59     { 100, 1250, 1500 },
  60     { 110, 3000, 3600 },
  61     { 122, 4000, 4800 },
  62     { 244, 4000, 4800 },
  63     {  44, 4000, 4800 },
  64 };
  65
  66 // We are only ever interested in the NAL bitrate factor.
  67 static int h264_get_br_factor(int profile_idc)
  68 {
  69     int i;
  70     for (i = 0; i < FF_ARRAY_ELEMS(h264_br_factors); i++) {
  71         if (h264_br_factors[i].profile_idc == profile_idc)
  72             return h264_br_factors[i].cpb_br_nal_factor;
  73     }
  74     // Default to the non-high profile value if not specified.
  75     return 1200;
  76 }
  77
  78 const H264LevelDescriptor *ff_h264_get_level(int level_idc,
  79                                              int constraint_set3_flag)
  80 {
  81     int i;
  82     for (i = 0; i < FF_ARRAY_ELEMS(h264_levels); i++) {
  83         if (h264_levels[i].level_idc            == level_idc &&
  84             h264_levels[i].constraint_set3_flag == constraint_set3_flag)
  85             return &h264_levels[i];
  86     }
  87     return NULL;
  88 }
  89
  90 const H264LevelDescriptor *ff_h264_guess_level(int profile_idc,
  91                                                int64_t bitrate,
  92                                                int width, int height,
  93                                                int max_dec_frame_buffering)
  94 {
  95     int width_mbs  = (width  + 15) / 16;
  96     int height_mbs = (height + 15) / 16;
  97     int no_cs3f = !(profile_idc == 66 ||
  98                     profile_idc == 77 ||
  99                     profile_idc == 88);
 100     int i;
 101
 102     for (i = 0; i < FF_ARRAY_ELEMS(h264_levels); i++) {
 103         const H264LevelDescriptor *level = &h264_levels[i];
 104
 105         if (level->constraint_set3_flag && no_cs3f)
 106             continue;
 107
 108         if (bitrate > (int64_t)level->max_br * h264_get_br_factor(profile_idc))
 109             continue;
 110
 111         if (width_mbs  * height_mbs > level->max_fs)
 112             continue;
 113         if (width_mbs  * width_mbs  > 8 * level->max_fs)
 114             continue;
 115         if (height_mbs * height_mbs > 8 * level->max_fs)
 116             continue;
 117
 118         if (width_mbs && height_mbs) {
 119             int max_dpb_frames =
 120                 FFMIN(level->max_dpb_mbs / (width_mbs * height_mbs), 16);
 121             if (max_dec_frame_buffering > max_dpb_frames)
 122                 continue;
 123         }
 124
 125         return level;
 126     }
 127
 128     // No usable levels found - frame is too big or bitrate is too high.
 129     return NULL;
 130 }