You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

625 lines
24KB

  1. /*
  2. * H.26L/H.264/AVC/JVT/14496-10/... parameter set decoding
  3. * Copyright (c) 2003 Michael Niedermayer <michaelni@gmx.at>
  4. *
  5. * This file is part of Libav.
  6. *
  7. * Libav 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. * Libav 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 Libav; if not, write to the Free Software
  19. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20. */
  21. /**
  22. * @file
  23. * H.264 / AVC / MPEG4 part10 parameter set decoding.
  24. * @author Michael Niedermayer <michaelni@gmx.at>
  25. */
  26. #include "libavutil/imgutils.h"
  27. #include "internal.h"
  28. #include "avcodec.h"
  29. #include "h264.h"
  30. #include "h264data.h" //FIXME FIXME FIXME (just for zigzag_scan)
  31. #include "golomb.h"
  32. //#undef NDEBUG
  33. #include <assert.h>
  34. #define MAX_LOG2_MAX_FRAME_NUM (12 + 4)
  35. #define MIN_LOG2_MAX_FRAME_NUM 4
  36. static const AVRational pixel_aspect[17]={
  37. {0, 1},
  38. {1, 1},
  39. {12, 11},
  40. {10, 11},
  41. {16, 11},
  42. {40, 33},
  43. {24, 11},
  44. {20, 11},
  45. {32, 11},
  46. {80, 33},
  47. {18, 11},
  48. {15, 11},
  49. {64, 33},
  50. {160,99},
  51. {4, 3},
  52. {3, 2},
  53. {2, 1},
  54. };
  55. #define QP(qP,depth) ( (qP)+6*((depth)-8) )
  56. #define CHROMA_QP_TABLE_END(d) \
  57. QP(0,d), QP(1,d), QP(2,d), QP(3,d), QP(4,d), QP(5,d),\
  58. QP(6,d), QP(7,d), QP(8,d), QP(9,d), QP(10,d), QP(11,d),\
  59. QP(12,d), QP(13,d), QP(14,d), QP(15,d), QP(16,d), QP(17,d),\
  60. QP(18,d), QP(19,d), QP(20,d), QP(21,d), QP(22,d), QP(23,d),\
  61. QP(24,d), QP(25,d), QP(26,d), QP(27,d), QP(28,d), QP(29,d),\
  62. QP(29,d), QP(30,d), QP(31,d), QP(32,d), QP(32,d), QP(33,d),\
  63. QP(34,d), QP(34,d), QP(35,d), QP(35,d), QP(36,d), QP(36,d),\
  64. QP(37,d), QP(37,d), QP(37,d), QP(38,d), QP(38,d), QP(38,d),\
  65. QP(39,d), QP(39,d), QP(39,d), QP(39,d)
  66. const uint8_t ff_h264_chroma_qp[3][QP_MAX_NUM+1] = {
  67. {
  68. CHROMA_QP_TABLE_END(8)
  69. },
  70. {
  71. 0, 1, 2, 3, 4, 5,
  72. CHROMA_QP_TABLE_END(9)
  73. },
  74. {
  75. 0, 1, 2, 3, 4, 5,
  76. 6, 7, 8, 9, 10, 11,
  77. CHROMA_QP_TABLE_END(10)
  78. },
  79. };
  80. static const uint8_t default_scaling4[2][16]={
  81. { 6,13,20,28,
  82. 13,20,28,32,
  83. 20,28,32,37,
  84. 28,32,37,42
  85. },{
  86. 10,14,20,24,
  87. 14,20,24,27,
  88. 20,24,27,30,
  89. 24,27,30,34
  90. }};
  91. static const uint8_t default_scaling8[2][64]={
  92. { 6,10,13,16,18,23,25,27,
  93. 10,11,16,18,23,25,27,29,
  94. 13,16,18,23,25,27,29,31,
  95. 16,18,23,25,27,29,31,33,
  96. 18,23,25,27,29,31,33,36,
  97. 23,25,27,29,31,33,36,38,
  98. 25,27,29,31,33,36,38,40,
  99. 27,29,31,33,36,38,40,42
  100. },{
  101. 9,13,15,17,19,21,22,24,
  102. 13,13,17,19,21,22,24,25,
  103. 15,17,19,21,22,24,25,27,
  104. 17,19,21,22,24,25,27,28,
  105. 19,21,22,24,25,27,28,30,
  106. 21,22,24,25,27,28,30,32,
  107. 22,24,25,27,28,30,32,33,
  108. 24,25,27,28,30,32,33,35
  109. }};
  110. static inline int decode_hrd_parameters(H264Context *h, SPS *sps){
  111. int cpb_count, i;
  112. cpb_count = get_ue_golomb_31(&h->gb) + 1;
  113. if(cpb_count > 32U){
  114. av_log(h->avctx, AV_LOG_ERROR, "cpb_count %d invalid\n", cpb_count);
  115. return -1;
  116. }
  117. get_bits(&h->gb, 4); /* bit_rate_scale */
  118. get_bits(&h->gb, 4); /* cpb_size_scale */
  119. for(i=0; i<cpb_count; i++){
  120. get_ue_golomb_long(&h->gb); /* bit_rate_value_minus1 */
  121. get_ue_golomb_long(&h->gb); /* cpb_size_value_minus1 */
  122. get_bits1(&h->gb); /* cbr_flag */
  123. }
  124. sps->initial_cpb_removal_delay_length = get_bits(&h->gb, 5) + 1;
  125. sps->cpb_removal_delay_length = get_bits(&h->gb, 5) + 1;
  126. sps->dpb_output_delay_length = get_bits(&h->gb, 5) + 1;
  127. sps->time_offset_length = get_bits(&h->gb, 5);
  128. sps->cpb_cnt = cpb_count;
  129. return 0;
  130. }
  131. static inline int decode_vui_parameters(H264Context *h, SPS *sps){
  132. int aspect_ratio_info_present_flag;
  133. unsigned int aspect_ratio_idc;
  134. aspect_ratio_info_present_flag= get_bits1(&h->gb);
  135. if( aspect_ratio_info_present_flag ) {
  136. aspect_ratio_idc= get_bits(&h->gb, 8);
  137. if( aspect_ratio_idc == EXTENDED_SAR ) {
  138. sps->sar.num= get_bits(&h->gb, 16);
  139. sps->sar.den= get_bits(&h->gb, 16);
  140. }else if(aspect_ratio_idc < FF_ARRAY_ELEMS(pixel_aspect)){
  141. sps->sar= pixel_aspect[aspect_ratio_idc];
  142. }else{
  143. av_log(h->avctx, AV_LOG_ERROR, "illegal aspect ratio\n");
  144. return -1;
  145. }
  146. }else{
  147. sps->sar.num=
  148. sps->sar.den= 0;
  149. }
  150. // s->avctx->aspect_ratio= sar_width*s->width / (float)(s->height*sar_height);
  151. if(get_bits1(&h->gb)){ /* overscan_info_present_flag */
  152. get_bits1(&h->gb); /* overscan_appropriate_flag */
  153. }
  154. sps->video_signal_type_present_flag = get_bits1(&h->gb);
  155. if(sps->video_signal_type_present_flag){
  156. get_bits(&h->gb, 3); /* video_format */
  157. sps->full_range = get_bits1(&h->gb); /* video_full_range_flag */
  158. sps->colour_description_present_flag = get_bits1(&h->gb);
  159. if(sps->colour_description_present_flag){
  160. sps->color_primaries = get_bits(&h->gb, 8); /* colour_primaries */
  161. sps->color_trc = get_bits(&h->gb, 8); /* transfer_characteristics */
  162. sps->colorspace = get_bits(&h->gb, 8); /* matrix_coefficients */
  163. if (sps->color_primaries >= AVCOL_PRI_NB)
  164. sps->color_primaries = AVCOL_PRI_UNSPECIFIED;
  165. if (sps->color_trc >= AVCOL_TRC_NB)
  166. sps->color_trc = AVCOL_TRC_UNSPECIFIED;
  167. if (sps->colorspace >= AVCOL_SPC_NB)
  168. sps->colorspace = AVCOL_SPC_UNSPECIFIED;
  169. }
  170. }
  171. if(get_bits1(&h->gb)){ /* chroma_location_info_present_flag */
  172. h->avctx->chroma_sample_location = get_ue_golomb(&h->gb)+1; /* chroma_sample_location_type_top_field */
  173. get_ue_golomb(&h->gb); /* chroma_sample_location_type_bottom_field */
  174. }
  175. sps->timing_info_present_flag = get_bits1(&h->gb);
  176. if(sps->timing_info_present_flag){
  177. sps->num_units_in_tick = get_bits_long(&h->gb, 32);
  178. sps->time_scale = get_bits_long(&h->gb, 32);
  179. if(!sps->num_units_in_tick || !sps->time_scale){
  180. av_log(h->avctx, AV_LOG_ERROR, "time_scale/num_units_in_tick invalid or unsupported (%d/%d)\n", sps->time_scale, sps->num_units_in_tick);
  181. return -1;
  182. }
  183. sps->fixed_frame_rate_flag = get_bits1(&h->gb);
  184. }
  185. sps->nal_hrd_parameters_present_flag = get_bits1(&h->gb);
  186. if(sps->nal_hrd_parameters_present_flag)
  187. if(decode_hrd_parameters(h, sps) < 0)
  188. return -1;
  189. sps->vcl_hrd_parameters_present_flag = get_bits1(&h->gb);
  190. if(sps->vcl_hrd_parameters_present_flag)
  191. if(decode_hrd_parameters(h, sps) < 0)
  192. return -1;
  193. if(sps->nal_hrd_parameters_present_flag || sps->vcl_hrd_parameters_present_flag)
  194. get_bits1(&h->gb); /* low_delay_hrd_flag */
  195. sps->pic_struct_present_flag = get_bits1(&h->gb);
  196. sps->bitstream_restriction_flag = get_bits1(&h->gb);
  197. if(sps->bitstream_restriction_flag){
  198. get_bits1(&h->gb); /* motion_vectors_over_pic_boundaries_flag */
  199. get_ue_golomb(&h->gb); /* max_bytes_per_pic_denom */
  200. get_ue_golomb(&h->gb); /* max_bits_per_mb_denom */
  201. get_ue_golomb(&h->gb); /* log2_max_mv_length_horizontal */
  202. get_ue_golomb(&h->gb); /* log2_max_mv_length_vertical */
  203. sps->num_reorder_frames= get_ue_golomb(&h->gb);
  204. get_ue_golomb(&h->gb); /*max_dec_frame_buffering*/
  205. if (get_bits_left(&h->gb) < 0) {
  206. sps->num_reorder_frames=0;
  207. sps->bitstream_restriction_flag= 0;
  208. }
  209. if(sps->num_reorder_frames > 16U /*max_dec_frame_buffering || max_dec_frame_buffering > 16*/){
  210. av_log(h->avctx, AV_LOG_ERROR, "illegal num_reorder_frames %d\n", sps->num_reorder_frames);
  211. return -1;
  212. }
  213. }
  214. if (get_bits_left(&h->gb) < 0) {
  215. av_log(h->avctx, AV_LOG_ERROR, "Overread VUI by %d bits\n", -get_bits_left(&h->gb));
  216. return AVERROR_INVALIDDATA;
  217. }
  218. return 0;
  219. }
  220. static void decode_scaling_list(H264Context *h, uint8_t *factors, int size,
  221. const uint8_t *jvt_list, const uint8_t *fallback_list){
  222. int i, last = 8, next = 8;
  223. const uint8_t *scan = size == 16 ? zigzag_scan : ff_zigzag_direct;
  224. if(!get_bits1(&h->gb)) /* matrix not written, we use the predicted one */
  225. memcpy(factors, fallback_list, size*sizeof(uint8_t));
  226. else
  227. for(i=0;i<size;i++){
  228. if(next)
  229. next = (last + get_se_golomb(&h->gb)) & 0xff;
  230. if(!i && !next){ /* matrix not written, we use the preset one */
  231. memcpy(factors, jvt_list, size*sizeof(uint8_t));
  232. break;
  233. }
  234. last = factors[scan[i]] = next ? next : last;
  235. }
  236. }
  237. static void decode_scaling_matrices(H264Context *h, SPS *sps, PPS *pps, int is_sps,
  238. uint8_t (*scaling_matrix4)[16], uint8_t (*scaling_matrix8)[64]){
  239. int fallback_sps = !is_sps && sps->scaling_matrix_present;
  240. const uint8_t *fallback[4] = {
  241. fallback_sps ? sps->scaling_matrix4[0] : default_scaling4[0],
  242. fallback_sps ? sps->scaling_matrix4[3] : default_scaling4[1],
  243. fallback_sps ? sps->scaling_matrix8[0] : default_scaling8[0],
  244. fallback_sps ? sps->scaling_matrix8[3] : default_scaling8[1]
  245. };
  246. if(get_bits1(&h->gb)){
  247. sps->scaling_matrix_present |= is_sps;
  248. decode_scaling_list(h,scaling_matrix4[0],16,default_scaling4[0],fallback[0]); // Intra, Y
  249. decode_scaling_list(h,scaling_matrix4[1],16,default_scaling4[0],scaling_matrix4[0]); // Intra, Cr
  250. decode_scaling_list(h,scaling_matrix4[2],16,default_scaling4[0],scaling_matrix4[1]); // Intra, Cb
  251. decode_scaling_list(h,scaling_matrix4[3],16,default_scaling4[1],fallback[1]); // Inter, Y
  252. decode_scaling_list(h,scaling_matrix4[4],16,default_scaling4[1],scaling_matrix4[3]); // Inter, Cr
  253. decode_scaling_list(h,scaling_matrix4[5],16,default_scaling4[1],scaling_matrix4[4]); // Inter, Cb
  254. if(is_sps || pps->transform_8x8_mode){
  255. decode_scaling_list(h,scaling_matrix8[0],64,default_scaling8[0],fallback[2]); // Intra, Y
  256. if(sps->chroma_format_idc == 3){
  257. decode_scaling_list(h,scaling_matrix8[1],64,default_scaling8[0],scaling_matrix8[0]); // Intra, Cr
  258. decode_scaling_list(h,scaling_matrix8[2],64,default_scaling8[0],scaling_matrix8[1]); // Intra, Cb
  259. }
  260. decode_scaling_list(h,scaling_matrix8[3],64,default_scaling8[1],fallback[3]); // Inter, Y
  261. if(sps->chroma_format_idc == 3){
  262. decode_scaling_list(h,scaling_matrix8[4],64,default_scaling8[1],scaling_matrix8[3]); // Inter, Cr
  263. decode_scaling_list(h,scaling_matrix8[5],64,default_scaling8[1],scaling_matrix8[4]); // Inter, Cb
  264. }
  265. }
  266. }
  267. }
  268. int ff_h264_decode_seq_parameter_set(H264Context *h){
  269. int profile_idc, level_idc, constraint_set_flags = 0;
  270. unsigned int sps_id;
  271. int i, log2_max_frame_num_minus4;
  272. SPS *sps;
  273. profile_idc= get_bits(&h->gb, 8);
  274. constraint_set_flags |= get_bits1(&h->gb) << 0; //constraint_set0_flag
  275. constraint_set_flags |= get_bits1(&h->gb) << 1; //constraint_set1_flag
  276. constraint_set_flags |= get_bits1(&h->gb) << 2; //constraint_set2_flag
  277. constraint_set_flags |= get_bits1(&h->gb) << 3; //constraint_set3_flag
  278. get_bits(&h->gb, 4); // reserved
  279. level_idc= get_bits(&h->gb, 8);
  280. sps_id= get_ue_golomb_31(&h->gb);
  281. if(sps_id >= MAX_SPS_COUNT) {
  282. av_log(h->avctx, AV_LOG_ERROR, "sps_id (%d) out of range\n", sps_id);
  283. return -1;
  284. }
  285. sps= av_mallocz(sizeof(SPS));
  286. if(sps == NULL)
  287. return -1;
  288. sps->time_offset_length = 24;
  289. sps->profile_idc= profile_idc;
  290. sps->constraint_set_flags = constraint_set_flags;
  291. sps->level_idc= level_idc;
  292. memset(sps->scaling_matrix4, 16, sizeof(sps->scaling_matrix4));
  293. memset(sps->scaling_matrix8, 16, sizeof(sps->scaling_matrix8));
  294. sps->scaling_matrix_present = 0;
  295. if (sps->profile_idc == 100 || sps->profile_idc == 110 ||
  296. sps->profile_idc == 122 || sps->profile_idc == 244 ||
  297. sps->profile_idc == 44 || sps->profile_idc == 83 ||
  298. sps->profile_idc == 86 || sps->profile_idc == 118 ||
  299. sps->profile_idc == 128 || sps->profile_idc == 144) {
  300. sps->chroma_format_idc= get_ue_golomb_31(&h->gb);
  301. if(sps->chroma_format_idc > 3) {
  302. av_log(h->avctx, AV_LOG_ERROR, "chroma_format_idc (%u) out of range\n", sps->chroma_format_idc);
  303. goto fail;
  304. } else if(sps->chroma_format_idc == 3) {
  305. sps->residual_color_transform_flag = get_bits1(&h->gb);
  306. }
  307. sps->bit_depth_luma = get_ue_golomb(&h->gb) + 8;
  308. sps->bit_depth_chroma = get_ue_golomb(&h->gb) + 8;
  309. sps->transform_bypass = get_bits1(&h->gb);
  310. decode_scaling_matrices(h, sps, NULL, 1, sps->scaling_matrix4, sps->scaling_matrix8);
  311. }else{
  312. sps->chroma_format_idc= 1;
  313. sps->bit_depth_luma = 8;
  314. sps->bit_depth_chroma = 8;
  315. }
  316. log2_max_frame_num_minus4 = get_ue_golomb(&h->gb);
  317. if (log2_max_frame_num_minus4 < MIN_LOG2_MAX_FRAME_NUM - 4 ||
  318. log2_max_frame_num_minus4 > MAX_LOG2_MAX_FRAME_NUM - 4) {
  319. av_log(h->avctx, AV_LOG_ERROR,
  320. "log2_max_frame_num_minus4 out of range (0-12): %d\n",
  321. log2_max_frame_num_minus4);
  322. goto fail;
  323. }
  324. sps->log2_max_frame_num = log2_max_frame_num_minus4 + 4;
  325. sps->poc_type= get_ue_golomb_31(&h->gb);
  326. if(sps->poc_type == 0){ //FIXME #define
  327. sps->log2_max_poc_lsb= get_ue_golomb(&h->gb) + 4;
  328. } else if(sps->poc_type == 1){//FIXME #define
  329. sps->delta_pic_order_always_zero_flag= get_bits1(&h->gb);
  330. sps->offset_for_non_ref_pic= get_se_golomb(&h->gb);
  331. sps->offset_for_top_to_bottom_field= get_se_golomb(&h->gb);
  332. sps->poc_cycle_length = get_ue_golomb(&h->gb);
  333. if((unsigned)sps->poc_cycle_length >= FF_ARRAY_ELEMS(sps->offset_for_ref_frame)){
  334. av_log(h->avctx, AV_LOG_ERROR, "poc_cycle_length overflow %u\n", sps->poc_cycle_length);
  335. goto fail;
  336. }
  337. for(i=0; i<sps->poc_cycle_length; i++)
  338. sps->offset_for_ref_frame[i]= get_se_golomb(&h->gb);
  339. }else if(sps->poc_type != 2){
  340. av_log(h->avctx, AV_LOG_ERROR, "illegal POC type %d\n", sps->poc_type);
  341. goto fail;
  342. }
  343. sps->ref_frame_count= get_ue_golomb_31(&h->gb);
  344. if(sps->ref_frame_count > MAX_PICTURE_COUNT-2 || sps->ref_frame_count >= 32U){
  345. av_log(h->avctx, AV_LOG_ERROR, "too many reference frames\n");
  346. goto fail;
  347. }
  348. sps->gaps_in_frame_num_allowed_flag= get_bits1(&h->gb);
  349. sps->mb_width = get_ue_golomb(&h->gb) + 1;
  350. sps->mb_height= get_ue_golomb(&h->gb) + 1;
  351. if((unsigned)sps->mb_width >= INT_MAX/16 || (unsigned)sps->mb_height >= INT_MAX/16 ||
  352. av_image_check_size(16*sps->mb_width, 16*sps->mb_height, 0, h->avctx)){
  353. av_log(h->avctx, AV_LOG_ERROR, "mb_width/height overflow\n");
  354. goto fail;
  355. }
  356. sps->frame_mbs_only_flag= get_bits1(&h->gb);
  357. if(!sps->frame_mbs_only_flag)
  358. sps->mb_aff= get_bits1(&h->gb);
  359. else
  360. sps->mb_aff= 0;
  361. sps->direct_8x8_inference_flag= get_bits1(&h->gb);
  362. if(!sps->frame_mbs_only_flag && !sps->direct_8x8_inference_flag){
  363. av_log(h->avctx, AV_LOG_ERROR, "This stream was generated by a broken encoder, invalid 8x8 inference\n");
  364. goto fail;
  365. }
  366. #ifndef ALLOW_INTERLACE
  367. if(sps->mb_aff)
  368. av_log(h->avctx, AV_LOG_ERROR, "MBAFF support not included; enable it at compile-time.\n");
  369. #endif
  370. sps->crop= get_bits1(&h->gb);
  371. if(sps->crop){
  372. int crop_left = get_ue_golomb(&h->gb);
  373. int crop_right = get_ue_golomb(&h->gb);
  374. int crop_top = get_ue_golomb(&h->gb);
  375. int crop_bottom = get_ue_golomb(&h->gb);
  376. if (h->avctx->flags2 & CODEC_FLAG2_IGNORE_CROP) {
  377. av_log(h->avctx, AV_LOG_DEBUG, "discarding sps cropping, original "
  378. "values are l:%u r:%u t:%u b:%u\n", crop_left, crop_right,
  379. crop_top, crop_bottom);
  380. sps->crop_left =
  381. sps->crop_right =
  382. sps->crop_top =
  383. sps->crop_bottom = 0;
  384. } else {
  385. int vsub = (sps->chroma_format_idc == 1) ? 1 : 0;
  386. int hsub = (sps->chroma_format_idc == 1 || sps->chroma_format_idc == 2) ? 1 : 0;
  387. int step_x = 1 << hsub;
  388. int step_y = (2 - sps->frame_mbs_only_flag) << vsub;
  389. if (crop_left & (0x1F >> (sps->bit_depth_luma > 8)) &&
  390. !(h->avctx->flags & CODEC_FLAG_UNALIGNED)) {
  391. crop_left &= ~(0x1F >> (sps->bit_depth_luma > 8));
  392. av_log(h->avctx, AV_LOG_WARNING, "Reducing left cropping to %d "
  393. "chroma samples to preserve alignment.\n",
  394. crop_left);
  395. }
  396. sps->crop_left = crop_left * step_x;
  397. sps->crop_right = crop_right * step_x;
  398. sps->crop_top = crop_top * step_y;
  399. sps->crop_bottom = crop_bottom * step_y;
  400. }
  401. }else{
  402. sps->crop_left =
  403. sps->crop_right =
  404. sps->crop_top =
  405. sps->crop_bottom= 0;
  406. sps->crop = 0;
  407. }
  408. sps->vui_parameters_present_flag= get_bits1(&h->gb);
  409. if( sps->vui_parameters_present_flag )
  410. if (decode_vui_parameters(h, sps) < 0)
  411. goto fail;
  412. if(!sps->sar.den)
  413. sps->sar.den= 1;
  414. if(h->avctx->debug&FF_DEBUG_PICT_INFO){
  415. static const char csp[4][5] = { "Gray", "420", "422", "444" };
  416. av_log(h->avctx, AV_LOG_DEBUG, "sps:%u profile:%d/%d poc:%d ref:%d %dx%d %s %s crop:%d/%d/%d/%d %s %s %d/%d\n",
  417. sps_id, sps->profile_idc, sps->level_idc,
  418. sps->poc_type,
  419. sps->ref_frame_count,
  420. sps->mb_width, sps->mb_height,
  421. sps->frame_mbs_only_flag ? "FRM" : (sps->mb_aff ? "MB-AFF" : "PIC-AFF"),
  422. sps->direct_8x8_inference_flag ? "8B8" : "",
  423. sps->crop_left, sps->crop_right,
  424. sps->crop_top, sps->crop_bottom,
  425. sps->vui_parameters_present_flag ? "VUI" : "",
  426. csp[sps->chroma_format_idc],
  427. sps->timing_info_present_flag ? sps->num_units_in_tick : 0,
  428. sps->timing_info_present_flag ? sps->time_scale : 0
  429. );
  430. }
  431. sps->new = 1;
  432. av_free(h->sps_buffers[sps_id]);
  433. h->sps_buffers[sps_id] = sps;
  434. h->sps = *sps;
  435. h->current_sps_id = sps_id;
  436. return 0;
  437. fail:
  438. av_free(sps);
  439. return -1;
  440. }
  441. static void
  442. build_qp_table(PPS *pps, int t, int index, const int depth)
  443. {
  444. int i;
  445. const int max_qp = 51 + 6*(depth-8);
  446. for(i = 0; i < max_qp+1; i++)
  447. pps->chroma_qp_table[t][i] = ff_h264_chroma_qp[depth-8][av_clip(i + index, 0, max_qp)];
  448. }
  449. int ff_h264_decode_picture_parameter_set(H264Context *h, int bit_length){
  450. unsigned int pps_id= get_ue_golomb(&h->gb);
  451. PPS *pps;
  452. const int qp_bd_offset = 6*(h->sps.bit_depth_luma-8);
  453. int bits_left;
  454. if(pps_id >= MAX_PPS_COUNT) {
  455. av_log(h->avctx, AV_LOG_ERROR, "pps_id (%d) out of range\n", pps_id);
  456. return -1;
  457. } else if (h->sps.bit_depth_luma > 10) {
  458. av_log(h->avctx, AV_LOG_ERROR, "Unimplemented luma bit depth=%d (max=10)\n", h->sps.bit_depth_luma);
  459. return AVERROR_PATCHWELCOME;
  460. }
  461. pps= av_mallocz(sizeof(PPS));
  462. if(pps == NULL)
  463. return -1;
  464. pps->sps_id= get_ue_golomb_31(&h->gb);
  465. if((unsigned)pps->sps_id>=MAX_SPS_COUNT || h->sps_buffers[pps->sps_id] == NULL){
  466. av_log(h->avctx, AV_LOG_ERROR, "sps_id out of range\n");
  467. goto fail;
  468. }
  469. pps->cabac= get_bits1(&h->gb);
  470. pps->pic_order_present= get_bits1(&h->gb);
  471. pps->slice_group_count= get_ue_golomb(&h->gb) + 1;
  472. if(pps->slice_group_count > 1 ){
  473. pps->mb_slice_group_map_type= get_ue_golomb(&h->gb);
  474. av_log(h->avctx, AV_LOG_ERROR, "FMO not supported\n");
  475. switch(pps->mb_slice_group_map_type){
  476. case 0:
  477. #if 0
  478. | for( i = 0; i <= num_slice_groups_minus1; i++ ) | | |
  479. | run_length[ i ] |1 |ue(v) |
  480. #endif
  481. break;
  482. case 2:
  483. #if 0
  484. | for( i = 0; i < num_slice_groups_minus1; i++ ) | | |
  485. |{ | | |
  486. | top_left_mb[ i ] |1 |ue(v) |
  487. | bottom_right_mb[ i ] |1 |ue(v) |
  488. | } | | |
  489. #endif
  490. break;
  491. case 3:
  492. case 4:
  493. case 5:
  494. #if 0
  495. | slice_group_change_direction_flag |1 |u(1) |
  496. | slice_group_change_rate_minus1 |1 |ue(v) |
  497. #endif
  498. break;
  499. case 6:
  500. #if 0
  501. | slice_group_id_cnt_minus1 |1 |ue(v) |
  502. | for( i = 0; i <= slice_group_id_cnt_minus1; i++ | | |
  503. |) | | |
  504. | slice_group_id[ i ] |1 |u(v) |
  505. #endif
  506. break;
  507. }
  508. }
  509. pps->ref_count[0]= get_ue_golomb(&h->gb) + 1;
  510. pps->ref_count[1]= get_ue_golomb(&h->gb) + 1;
  511. if(pps->ref_count[0]-1 > 32-1 || pps->ref_count[1]-1 > 32-1){
  512. av_log(h->avctx, AV_LOG_ERROR, "reference overflow (pps)\n");
  513. goto fail;
  514. }
  515. pps->weighted_pred= get_bits1(&h->gb);
  516. pps->weighted_bipred_idc= get_bits(&h->gb, 2);
  517. pps->init_qp= get_se_golomb(&h->gb) + 26 + qp_bd_offset;
  518. pps->init_qs= get_se_golomb(&h->gb) + 26 + qp_bd_offset;
  519. pps->chroma_qp_index_offset[0]= get_se_golomb(&h->gb);
  520. pps->deblocking_filter_parameters_present= get_bits1(&h->gb);
  521. pps->constrained_intra_pred= get_bits1(&h->gb);
  522. pps->redundant_pic_cnt_present = get_bits1(&h->gb);
  523. pps->transform_8x8_mode= 0;
  524. h->dequant_coeff_pps= -1; //contents of sps/pps can change even if id doesn't, so reinit
  525. memcpy(pps->scaling_matrix4, h->sps_buffers[pps->sps_id]->scaling_matrix4, sizeof(pps->scaling_matrix4));
  526. memcpy(pps->scaling_matrix8, h->sps_buffers[pps->sps_id]->scaling_matrix8, sizeof(pps->scaling_matrix8));
  527. bits_left = bit_length - get_bits_count(&h->gb);
  528. if (bits_left && (bits_left > 8 ||
  529. show_bits(&h->gb, bits_left) != 1 << (bits_left - 1))) {
  530. pps->transform_8x8_mode= get_bits1(&h->gb);
  531. decode_scaling_matrices(h, h->sps_buffers[pps->sps_id], pps, 0, pps->scaling_matrix4, pps->scaling_matrix8);
  532. pps->chroma_qp_index_offset[1]= get_se_golomb(&h->gb); //second_chroma_qp_index_offset
  533. } else {
  534. pps->chroma_qp_index_offset[1]= pps->chroma_qp_index_offset[0];
  535. }
  536. build_qp_table(pps, 0, pps->chroma_qp_index_offset[0], h->sps.bit_depth_luma);
  537. build_qp_table(pps, 1, pps->chroma_qp_index_offset[1], h->sps.bit_depth_luma);
  538. if(pps->chroma_qp_index_offset[0] != pps->chroma_qp_index_offset[1])
  539. pps->chroma_qp_diff= 1;
  540. if(h->avctx->debug&FF_DEBUG_PICT_INFO){
  541. av_log(h->avctx, AV_LOG_DEBUG, "pps:%u sps:%u %s slice_groups:%d ref:%d/%d %s qp:%d/%d/%d/%d %s %s %s %s\n",
  542. pps_id, pps->sps_id,
  543. pps->cabac ? "CABAC" : "CAVLC",
  544. pps->slice_group_count,
  545. pps->ref_count[0], pps->ref_count[1],
  546. pps->weighted_pred ? "weighted" : "",
  547. pps->init_qp, pps->init_qs, pps->chroma_qp_index_offset[0], pps->chroma_qp_index_offset[1],
  548. pps->deblocking_filter_parameters_present ? "LPAR" : "",
  549. pps->constrained_intra_pred ? "CONSTR" : "",
  550. pps->redundant_pic_cnt_present ? "REDU" : "",
  551. pps->transform_8x8_mode ? "8x8DCT" : ""
  552. );
  553. }
  554. av_free(h->pps_buffers[pps_id]);
  555. h->pps_buffers[pps_id]= pps;
  556. return 0;
  557. fail:
  558. av_free(pps);
  559. return -1;
  560. }