parset.c
上传用户:hjq518
上传日期:2021-12-09
资源大小:5084k
文件大小:29k
源码类别:

Audio

开发平台:

Visual C++

  1. /*!
  2.  ************************************************************************
  3.  *  file
  4.  *     parset.c
  5.  *  brief
  6.  *     Parameter Sets
  7.  *  author
  8.  *     Main contributors (see contributors.h for copyright, address and affiliation details)
  9.  *     - Stephan Wenger          <stewe@cs.tu-berlin.de>
  10.  *
  11.  ***********************************************************************
  12.  */
  13. #include "global.h"
  14. #include "parsetcommon.h"
  15. #include "parset.h"
  16. #include "nalu.h"
  17. #include "memalloc.h"
  18. #include "fmo.h"
  19. #include "cabac.h"
  20. #include "vlc.h"
  21. #include "mbuffer.h"
  22. #include "erc_api.h"
  23. #if TRACE
  24. #define SYMTRACESTRING(s) strncpy(sym->tracestring,s,TRACESTRING_SIZE)
  25. #else
  26. #define SYMTRACESTRING(s) // do nothing
  27. #endif
  28. const byte ZZ_SCAN[16]  =
  29. {  0,  1,  4,  8,  5,  2,  3,  6,  9, 12, 13, 10,  7, 11, 14, 15
  30. };
  31. const byte ZZ_SCAN8[64] =
  32. {  0,  1,  8, 16,  9,  2,  3, 10, 17, 24, 32, 25, 18, 11,  4,  5,
  33.    12, 19, 26, 33, 40, 48, 41, 34, 27, 20, 13,  6,  7, 14, 21, 28,
  34.    35, 42, 49, 56, 57, 50, 43, 36, 29, 22, 15, 23, 30, 37, 44, 51,
  35.    58, 59, 52, 45, 38, 31, 39, 46, 53, 60, 61, 54, 47, 55, 62, 63
  36. };
  37. extern int UsedBits;      // for internal statistics, is adjusted by se_v, ue_v, u_1
  38. extern ColocatedParams *Co_located;
  39. extern ColocatedParams *Co_located_JV[MAX_PLANE];  //!< Co_located to be used during 4:4:4 independent mode decoding
  40. extern int quant_intra_default[16];
  41. extern int quant_inter_default[16];
  42. extern int quant8_intra_default[64];
  43. extern int quant8_inter_default[64];
  44. seq_parameter_set_rbsp_t SeqParSet[MAXSPS];
  45. pic_parameter_set_rbsp_t PicParSet[MAXPPS];
  46. extern StorablePicture* dec_picture;
  47. extern void init_frext(ImageParameters *img);
  48. // syntax for scaling list matrix values
  49. void Scaling_List(int *scalingList, int sizeOfScalingList, Boolean *UseDefaultScalingMatrix, Bitstream *s)
  50. {
  51.   int j, scanj;
  52.   int delta_scale, lastScale, nextScale;
  53.   lastScale      = 8;
  54.   nextScale      = 8;
  55.   for(j=0; j<sizeOfScalingList; j++)
  56.   {
  57.     scanj = (sizeOfScalingList==16) ? ZZ_SCAN[j]:ZZ_SCAN8[j];
  58.     if(nextScale!=0)
  59.     {
  60.       delta_scale = se_v (   "   : delta_sl   "                           , s);
  61.       nextScale = (lastScale + delta_scale + 256) % 256;
  62.       *UseDefaultScalingMatrix = (Boolean) (scanj==0 && nextScale==0);
  63.     }
  64.     scalingList[scanj] = (nextScale==0) ? lastScale:nextScale;
  65.     lastScale = scalingList[scanj];
  66.   }
  67. }
  68. // fill sps with content of p
  69. int InterpretSPS (DataPartition *p, seq_parameter_set_rbsp_t *sps)
  70. {
  71.   unsigned i;
  72.   unsigned n_ScalingList;
  73.   int reserved_zero;
  74.   Bitstream *s = p->bitstream;
  75.   assert (p != NULL);
  76.   assert (p->bitstream != NULL);
  77.   assert (p->bitstream->streamBuffer != 0);
  78.   assert (sps != NULL);
  79.   UsedBits = 0;
  80.   sps->profile_idc                            = u_v  (8, "SPS: profile_idc"                           , s);
  81.   if ((sps->profile_idc!=66 ) &&
  82.       (sps->profile_idc!=77 ) &&
  83.       (sps->profile_idc!=88 ) &&
  84.       (sps->profile_idc!=FREXT_HP    ) &&
  85.       (sps->profile_idc!=FREXT_Hi10P ) &&
  86.       (sps->profile_idc!=FREXT_Hi422 ) &&
  87.       (sps->profile_idc!=FREXT_Hi444 ) &&
  88.       (sps->profile_idc!=FREXT_CAVLC444 ))
  89.   {
  90.     return UsedBits;
  91.   }
  92.   sps->constrained_set0_flag                  = u_1  (   "SPS: constrained_set0_flag"                 , s);
  93.   sps->constrained_set1_flag                  = u_1  (   "SPS: constrained_set1_flag"                 , s);
  94.   sps->constrained_set2_flag                  = u_1  (   "SPS: constrained_set2_flag"                 , s);
  95.   sps->constrained_set3_flag                  = u_1  (   "SPS: constrained_set3_flag"                 , s);
  96.   reserved_zero                               = u_v  (4, "SPS: reserved_zero_4bits"                   , s);
  97.   assert (reserved_zero==0);
  98.   sps->level_idc                              = u_v  (8, "SPS: level_idc"                             , s);
  99.   sps->seq_parameter_set_id                   = ue_v ("SPS: seq_parameter_set_id"                     , s);
  100.   // Fidelity Range Extensions stuff
  101.   sps->chroma_format_idc = 1;
  102.   sps->bit_depth_luma_minus8   = 0;
  103.   sps->bit_depth_chroma_minus8 = 0;
  104.   img->lossless_qpprime_flag   = 0;
  105.   sps->separate_colour_plane_flag = 0;
  106.   if((sps->profile_idc==FREXT_HP   ) ||
  107.      (sps->profile_idc==FREXT_Hi10P) ||
  108.      (sps->profile_idc==FREXT_Hi422) ||
  109.      (sps->profile_idc==FREXT_Hi444) ||
  110.      (sps->profile_idc==FREXT_CAVLC444))
  111.   {
  112.     sps->chroma_format_idc                      = ue_v ("SPS: chroma_format_idc"                       , s);
  113.     if(sps->chroma_format_idc == YUV444)
  114.     {
  115.       sps->separate_colour_plane_flag           = u_1  ("SPS: separate_colour_plane_flag"              , s);
  116.     }
  117.     sps->bit_depth_luma_minus8                  = ue_v ("SPS: bit_depth_luma_minus8"                   , s);
  118.     sps->bit_depth_chroma_minus8                = ue_v ("SPS: bit_depth_chroma_minus8"                 , s);
  119.     img->lossless_qpprime_flag                  = u_1  ("SPS: lossless_qpprime_y_zero_flag"            , s);
  120.     sps->seq_scaling_matrix_present_flag        = u_1  (   "SPS: seq_scaling_matrix_present_flag"       , s);
  121.     if(sps->seq_scaling_matrix_present_flag)
  122.     {
  123.       n_ScalingList = (sps->chroma_format_idc != YUV444) ? 8 : 12;
  124.       for(i=0; i<n_ScalingList; i++)
  125.       {
  126.         sps->seq_scaling_list_present_flag[i]   = u_1  (   "SPS: seq_scaling_list_present_flag"         , s);
  127.         if(sps->seq_scaling_list_present_flag[i])
  128.         {
  129.           if(i<6)
  130.             Scaling_List(sps->ScalingList4x4[i], 16, &sps->UseDefaultScalingMatrix4x4Flag[i], s);
  131.           else
  132.             Scaling_List(sps->ScalingList8x8[i-6], 64, &sps->UseDefaultScalingMatrix8x8Flag[i-6], s);
  133.         }
  134.       }
  135.     }
  136.   }
  137.   sps->log2_max_frame_num_minus4              = ue_v ("SPS: log2_max_frame_num_minus4"                , s);
  138.   sps->pic_order_cnt_type                     = ue_v ("SPS: pic_order_cnt_type"                       , s);
  139.   if (sps->pic_order_cnt_type == 0)
  140.     sps->log2_max_pic_order_cnt_lsb_minus4 = ue_v ("SPS: log2_max_pic_order_cnt_lsb_minus4"           , s);
  141.   else if (sps->pic_order_cnt_type == 1)
  142.   {
  143.     sps->delta_pic_order_always_zero_flag      = u_1  ("SPS: delta_pic_order_always_zero_flag"       , s);
  144.     sps->offset_for_non_ref_pic                = se_v ("SPS: offset_for_non_ref_pic"                 , s);
  145.     sps->offset_for_top_to_bottom_field        = se_v ("SPS: offset_for_top_to_bottom_field"         , s);
  146.     sps->num_ref_frames_in_pic_order_cnt_cycle = ue_v ("SPS: num_ref_frames_in_pic_order_cnt_cycle"  , s);
  147.     for(i=0; i<sps->num_ref_frames_in_pic_order_cnt_cycle; i++)
  148.       sps->offset_for_ref_frame[i]               = se_v ("SPS: offset_for_ref_frame[i]"              , s);
  149.   }
  150.   sps->num_ref_frames                        = ue_v ("SPS: num_ref_frames"                         , s);
  151.   sps->gaps_in_frame_num_value_allowed_flag  = u_1  ("SPS: gaps_in_frame_num_value_allowed_flag"   , s);
  152.   sps->pic_width_in_mbs_minus1               = ue_v ("SPS: pic_width_in_mbs_minus1"                , s);
  153.   sps->pic_height_in_map_units_minus1        = ue_v ("SPS: pic_height_in_map_units_minus1"         , s);
  154.   sps->frame_mbs_only_flag                   = u_1  ("SPS: frame_mbs_only_flag"                    , s);
  155.   if (!sps->frame_mbs_only_flag)
  156.   {
  157.     sps->mb_adaptive_frame_field_flag        = u_1  ("SPS: mb_adaptive_frame_field_flag"           , s);
  158.   }
  159.   sps->direct_8x8_inference_flag             = u_1  ("SPS: direct_8x8_inference_flag"              , s);
  160.   sps->frame_cropping_flag                   = u_1  ("SPS: frame_cropping_flag"                    , s);
  161.   if (sps->frame_cropping_flag)
  162.   {
  163.     sps->frame_cropping_rect_left_offset      = ue_v ("SPS: frame_cropping_rect_left_offset"           , s);
  164.     sps->frame_cropping_rect_right_offset     = ue_v ("SPS: frame_cropping_rect_right_offset"          , s);
  165.     sps->frame_cropping_rect_top_offset       = ue_v ("SPS: frame_cropping_rect_top_offset"            , s);
  166.     sps->frame_cropping_rect_bottom_offset    = ue_v ("SPS: frame_cropping_rect_bottom_offset"         , s);
  167.   }
  168.   sps->vui_parameters_present_flag           = (Boolean) u_1  ("SPS: vui_parameters_present_flag"      , s);
  169.   InitVUI(sps);
  170.   ReadVUI(p, sps);
  171.   sps->Valid = TRUE;
  172.   return UsedBits;
  173. }
  174. void InitVUI(seq_parameter_set_rbsp_t *sps)
  175. {
  176.   sps->vui_seq_parameters.matrix_coefficients = 2;
  177. }
  178. int ReadVUI(DataPartition *p, seq_parameter_set_rbsp_t *sps)
  179. {
  180.   Bitstream *s = p->bitstream;
  181.   if (sps->vui_parameters_present_flag)
  182.   {
  183.     sps->vui_seq_parameters.aspect_ratio_info_present_flag = u_1  ("VUI: aspect_ratio_info_present_flag"   , s);
  184.     if (sps->vui_seq_parameters.aspect_ratio_info_present_flag)
  185.     {
  186.       sps->vui_seq_parameters.aspect_ratio_idc             = u_v  ( 8, "VUI: aspect_ratio_idc"              , s);
  187.       if (255==sps->vui_seq_parameters.aspect_ratio_idc)
  188.       {
  189.         sps->vui_seq_parameters.sar_width                  = u_v  (16, "VUI: sar_width"                     , s);
  190.         sps->vui_seq_parameters.sar_height                 = u_v  (16, "VUI: sar_height"                    , s);
  191.       }
  192.   }
  193.     sps->vui_seq_parameters.overscan_info_present_flag     = u_1  ("VUI: overscan_info_present_flag"        , s);
  194.     if (sps->vui_seq_parameters.overscan_info_present_flag)
  195.     {
  196.       sps->vui_seq_parameters.overscan_appropriate_flag    = u_1  ("VUI: overscan_appropriate_flag"         , s);
  197.     }
  198.     sps->vui_seq_parameters.video_signal_type_present_flag = u_1  ("VUI: video_signal_type_present_flag"    , s);
  199.     if (sps->vui_seq_parameters.video_signal_type_present_flag)
  200.     {
  201.       sps->vui_seq_parameters.video_format                    = u_v  ( 3,"VUI: video_format"                      , s);
  202.       sps->vui_seq_parameters.video_full_range_flag           = u_1  (   "VUI: video_full_range_flag"             , s);
  203.       sps->vui_seq_parameters.colour_description_present_flag = u_1  (   "VUI: color_description_present_flag"    , s);
  204.       if(sps->vui_seq_parameters.colour_description_present_flag)
  205.       {
  206.         sps->vui_seq_parameters.colour_primaries              = u_v  ( 8,"VUI: colour_primaries"                  , s);
  207.         sps->vui_seq_parameters.transfer_characteristics      = u_v  ( 8,"VUI: transfer_characteristics"          , s);
  208.         sps->vui_seq_parameters.matrix_coefficients           = u_v  ( 8,"VUI: matrix_coefficients"               , s);
  209.       }
  210.     }
  211.     sps->vui_seq_parameters.chroma_location_info_present_flag = u_1  (   "VUI: chroma_loc_info_present_flag"      , s);
  212.     if(sps->vui_seq_parameters.chroma_location_info_present_flag)
  213.     {
  214.       sps->vui_seq_parameters.chroma_sample_loc_type_top_field     = ue_v  ( "VUI: chroma_sample_loc_type_top_field"    , s);
  215.       sps->vui_seq_parameters.chroma_sample_loc_type_bottom_field  = ue_v  ( "VUI: chroma_sample_loc_type_bottom_field" , s);
  216.     }
  217.     sps->vui_seq_parameters.timing_info_present_flag          = u_1  ("VUI: timing_info_present_flag"           , s);
  218.     if (sps->vui_seq_parameters.timing_info_present_flag)
  219.     {
  220.       sps->vui_seq_parameters.num_units_in_tick               = u_v  (32,"VUI: num_units_in_tick"               , s);
  221.       sps->vui_seq_parameters.time_scale                      = u_v  (32,"VUI: time_scale"                      , s);
  222.       sps->vui_seq_parameters.fixed_frame_rate_flag           = u_1  (   "VUI: fixed_frame_rate_flag"           , s);
  223.     }
  224.     sps->vui_seq_parameters.nal_hrd_parameters_present_flag   = u_1  ("VUI: nal_hrd_parameters_present_flag"    , s);
  225.     if (sps->vui_seq_parameters.nal_hrd_parameters_present_flag)
  226.     {
  227.       ReadHRDParameters(p, &(sps->vui_seq_parameters.nal_hrd_parameters));
  228.     }
  229.     sps->vui_seq_parameters.vcl_hrd_parameters_present_flag   = u_1  ("VUI: vcl_hrd_parameters_present_flag"    , s);
  230.     if (sps->vui_seq_parameters.vcl_hrd_parameters_present_flag)
  231.     {
  232.       ReadHRDParameters(p, &(sps->vui_seq_parameters.vcl_hrd_parameters));
  233.     }
  234.     if (sps->vui_seq_parameters.nal_hrd_parameters_present_flag || sps->vui_seq_parameters.vcl_hrd_parameters_present_flag)
  235.     {
  236.       sps->vui_seq_parameters.low_delay_hrd_flag             =  u_1  ("VUI: low_delay_hrd_flag"                 , s);
  237.     }
  238.     sps->vui_seq_parameters.pic_struct_present_flag          =  u_1  ("VUI: pic_struct_present_flag   "         , s);
  239.     sps->vui_seq_parameters.bitstream_restriction_flag       =  u_1  ("VUI: bitstream_restriction_flag"         , s);
  240.     if (sps->vui_seq_parameters.bitstream_restriction_flag)
  241.     {
  242.       sps->vui_seq_parameters.motion_vectors_over_pic_boundaries_flag =  u_1  ("VUI: motion_vectors_over_pic_boundaries_flag", s);
  243.       sps->vui_seq_parameters.max_bytes_per_pic_denom                 =  ue_v ("VUI: max_bytes_per_pic_denom"                , s);
  244.       sps->vui_seq_parameters.max_bits_per_mb_denom                   =  ue_v ("VUI: max_bits_per_mb_denom"                  , s);
  245.       sps->vui_seq_parameters.log2_max_mv_length_horizontal           =  ue_v ("VUI: log2_max_mv_length_horizontal"          , s);
  246.       sps->vui_seq_parameters.log2_max_mv_length_vertical             =  ue_v ("VUI: log2_max_mv_length_vertical"            , s);
  247.       sps->vui_seq_parameters.num_reorder_frames                      =  ue_v ("VUI: num_reorder_frames"                     , s);
  248.       sps->vui_seq_parameters.max_dec_frame_buffering                 =  ue_v ("VUI: max_dec_frame_buffering"                , s);
  249.     }
  250.   }
  251.   return 0;
  252. }
  253. int ReadHRDParameters(DataPartition *p, hrd_parameters_t *hrd)
  254. {
  255.   Bitstream *s = p->bitstream;
  256.   unsigned int SchedSelIdx;
  257.   hrd->cpb_cnt_minus1                                      = ue_v (   "VUI: cpb_cnt_minus1"                       , s);
  258.   hrd->bit_rate_scale                                      = u_v  ( 4,"VUI: bit_rate_scale"                       , s);
  259.   hrd->cpb_size_scale                                      = u_v  ( 4,"VUI: cpb_size_scale"                       , s);
  260.   for( SchedSelIdx = 0; SchedSelIdx <= hrd->cpb_cnt_minus1; SchedSelIdx++ )
  261.   {
  262.     hrd->bit_rate_value_minus1[ SchedSelIdx ]             = ue_v  ( "VUI: bit_rate_value_minus1"                  , s);
  263.     hrd->cpb_size_value_minus1[ SchedSelIdx ]             = ue_v  ( "VUI: cpb_size_value_minus1"                  , s);
  264.     hrd->cbr_flag[ SchedSelIdx ]                          = u_1   ( "VUI: cbr_flag"                               , s);
  265.   }
  266.   hrd->initial_cpb_removal_delay_length_minus1            = u_v  ( 5,"VUI: initial_cpb_removal_delay_length_minus1" , s);
  267.   hrd->cpb_removal_delay_length_minus1                    = u_v  ( 5,"VUI: cpb_removal_delay_length_minus1"         , s);
  268.   hrd->dpb_output_delay_length_minus1                     = u_v  ( 5,"VUI: dpb_output_delay_length_minus1"          , s);
  269.   hrd->time_offset_length                                 = u_v  ( 5,"VUI: time_offset_length"          , s);
  270.   return 0;
  271. }
  272. int InterpretPPS (DataPartition *p, pic_parameter_set_rbsp_t *pps)
  273. {
  274.   unsigned i;
  275.   unsigned n_ScalingList;
  276.   int chroma_format_idc;
  277.   int NumberBitsPerSliceGroupId;
  278.   Bitstream *s = p->bitstream;
  279.   assert (p != NULL);
  280.   assert (p->bitstream != NULL);
  281.   assert (p->bitstream->streamBuffer != 0);
  282.   assert (pps != NULL);
  283.   UsedBits = 0;
  284.   pps->pic_parameter_set_id                  = ue_v ("PPS: pic_parameter_set_id"                   , s);
  285.   pps->seq_parameter_set_id                  = ue_v ("PPS: seq_parameter_set_id"                   , s);
  286.   pps->entropy_coding_mode_flag              = u_1  ("PPS: entropy_coding_mode_flag"               , s);
  287.   //! Note: as per JVT-F078 the following bit is unconditional.  If F078 is not accepted, then
  288.   //! one has to fetch the correct SPS to check whether the bit is present (hopefully there is
  289.   //! no consistency problem :-(
  290.   //! The current encoder code handles this in the same way.  When you change this, don't forget
  291.   //! the encoder!  StW, 12/8/02
  292.   pps->pic_order_present_flag                = u_1  ("PPS: pic_order_present_flag"                 , s);
  293.   pps->num_slice_groups_minus1               = ue_v ("PPS: num_slice_groups_minus1"                , s);
  294.   // FMO stuff begins here
  295.   if (pps->num_slice_groups_minus1 > 0)
  296.   {
  297.     pps->slice_group_map_type               = ue_v ("PPS: slice_group_map_type"                , s);
  298.     if (pps->slice_group_map_type == 0)
  299.     {
  300.       for (i=0; i<=pps->num_slice_groups_minus1; i++)
  301.         pps->run_length_minus1 [i]                  = ue_v ("PPS: run_length_minus1 [i]"              , s);
  302.     }
  303.     else if (pps->slice_group_map_type == 2)
  304.     {
  305.       for (i=0; i<pps->num_slice_groups_minus1; i++)
  306.       {
  307.         //! JVT-F078: avoid reference of SPS by using ue(v) instead of u(v)
  308.         pps->top_left [i]                          = ue_v ("PPS: top_left [i]"                        , s);
  309.         pps->bottom_right [i]                      = ue_v ("PPS: bottom_right [i]"                    , s);
  310.       }
  311.     }
  312.     else if (pps->slice_group_map_type == 3 ||
  313.              pps->slice_group_map_type == 4 ||
  314.              pps->slice_group_map_type == 5)
  315.     {
  316.       pps->slice_group_change_direction_flag     = u_1  ("PPS: slice_group_change_direction_flag"      , s);
  317.       pps->slice_group_change_rate_minus1        = ue_v ("PPS: slice_group_change_rate_minus1"         , s);
  318.     }
  319.     else if (pps->slice_group_map_type == 6)
  320.     {
  321.       if (pps->num_slice_groups_minus1+1 >4)
  322.         NumberBitsPerSliceGroupId = 3;
  323.       else if (pps->num_slice_groups_minus1+1 > 2)
  324.         NumberBitsPerSliceGroupId = 2;
  325.       else
  326.         NumberBitsPerSliceGroupId = 1;
  327.       pps->num_slice_group_map_units_minus1      = ue_v ("PPS: num_slice_group_map_units_minus1"               , s);
  328.       if ((pps->slice_group_id = calloc (pps->num_slice_group_map_units_minus1+1, 1)) == NULL)
  329.         no_mem_exit ("InterpretPPS: slice_group_id");
  330.       for (i=0; i<=pps->num_slice_group_map_units_minus1; i++)
  331.         pps->slice_group_id[i] = u_v (NumberBitsPerSliceGroupId, "slice_group_id[i]", s);
  332.     }
  333.   }
  334.   // End of FMO stuff
  335.   pps->num_ref_idx_l0_active_minus1          = ue_v ("PPS: num_ref_idx_l0_active_minus1"           , s);
  336.   pps->num_ref_idx_l1_active_minus1          = ue_v ("PPS: num_ref_idx_l1_active_minus1"           , s);
  337.   pps->weighted_pred_flag                    = u_1  ("PPS: weighted_pred_flag"                     , s);
  338.   pps->weighted_bipred_idc                   = u_v  ( 2, "PPS: weighted_bipred_idc"                , s);
  339.   pps->pic_init_qp_minus26                   = se_v ("PPS: pic_init_qp_minus26"                    , s);
  340.   pps->pic_init_qs_minus26                   = se_v ("PPS: pic_init_qs_minus26"                    , s);
  341.   pps->chroma_qp_index_offset                = se_v ("PPS: chroma_qp_index_offset"                 , s);
  342.   pps->deblocking_filter_control_present_flag = u_1 ("PPS: deblocking_filter_control_present_flag" , s);
  343.   pps->constrained_intra_pred_flag           = u_1  ("PPS: constrained_intra_pred_flag"            , s);
  344.   pps->redundant_pic_cnt_present_flag        = u_1  ("PPS: redundant_pic_cnt_present_flag"         , s);
  345.   if(more_rbsp_data(s->streamBuffer, s->frame_bitoffset,s->bitstream_length)) // more_data_in_rbsp()
  346.   {
  347.     //Fidelity Range Extensions Stuff
  348.     pps->transform_8x8_mode_flag           = u_1  ("PPS: transform_8x8_mode_flag"                , s);
  349.     pps->pic_scaling_matrix_present_flag   =  u_1  ("PPS: pic_scaling_matrix_present_flag"        , s);
  350.     if(pps->pic_scaling_matrix_present_flag)
  351.     {
  352.       chroma_format_idc = SeqParSet[pps->seq_parameter_set_id].chroma_format_idc;
  353.       n_ScalingList = 6 + ((chroma_format_idc != YUV444) ? 2 : 6) * pps->transform_8x8_mode_flag;
  354.       for(i=0; i<n_ScalingList; i++)
  355.       {
  356.         pps->pic_scaling_list_present_flag[i]= u_1  ("PPS: pic_scaling_list_present_flag"          , s);
  357.         if(pps->pic_scaling_list_present_flag[i])
  358.         {
  359.           if(i<6)
  360.             Scaling_List(pps->ScalingList4x4[i], 16, &pps->UseDefaultScalingMatrix4x4Flag[i], s);
  361.           else
  362.             Scaling_List(pps->ScalingList8x8[i-6], 64, &pps->UseDefaultScalingMatrix8x8Flag[i-6], s);
  363.         }
  364.       }
  365.     }
  366.     pps->second_chroma_qp_index_offset      = se_v ("PPS: second_chroma_qp_index_offset"          , s);
  367.   }
  368.   else
  369.   {
  370.     pps->second_chroma_qp_index_offset      = pps->chroma_qp_index_offset;
  371.   }
  372.   pps->Valid = TRUE;
  373.   return UsedBits;
  374. }
  375. void PPSConsistencyCheck (pic_parameter_set_rbsp_t *pps)
  376. {
  377.   printf ("Consistency checking a picture parset, to be implementedn");
  378. //  if (pps->seq_parameter_set_id invalid then do something)
  379. }
  380. void SPSConsistencyCheck (seq_parameter_set_rbsp_t *sps)
  381. {
  382.   printf ("Consistency checking a sequence parset, to be implementedn");
  383. }
  384. void MakePPSavailable (int id, pic_parameter_set_rbsp_t *pps)
  385. {
  386.   assert (pps->Valid == TRUE);
  387.   if (PicParSet[id].Valid == TRUE && PicParSet[id].slice_group_id != NULL)
  388.     free (PicParSet[id].slice_group_id);
  389.   memcpy (&PicParSet[id], pps, sizeof (pic_parameter_set_rbsp_t));
  390.   // we can simply use the memory provided with the pps. the PPS is destroyed after this function
  391.   // call and will not try to free if pps->slice_group_id == NULL
  392.   PicParSet[id].slice_group_id = pps->slice_group_id;
  393.   pps->slice_group_id          = NULL;
  394. }
  395. void CleanUpPPS()
  396. {
  397.   int i;
  398.   for (i=0; i<MAXPPS; i++)
  399.   {
  400.     if (PicParSet[i].Valid == TRUE && PicParSet[i].slice_group_id != NULL)
  401.       free (PicParSet[i].slice_group_id);
  402.     PicParSet[i].Valid = FALSE;
  403.   }
  404. }
  405. void MakeSPSavailable (int id, seq_parameter_set_rbsp_t *sps)
  406. {
  407.   assert (sps->Valid == TRUE);
  408.   memcpy (&SeqParSet[id], sps, sizeof (seq_parameter_set_rbsp_t));
  409. }
  410. void ProcessSPS (NALU_t *nalu)
  411. {
  412.   DataPartition *dp = AllocPartition(1);
  413.   seq_parameter_set_rbsp_t *sps = AllocSPS();
  414.   memcpy (dp->bitstream->streamBuffer, &nalu->buf[1], nalu->len-1);
  415.   dp->bitstream->code_len = dp->bitstream->bitstream_length = RBSPtoSODB (dp->bitstream->streamBuffer, nalu->len-1);
  416.   dp->bitstream->ei_flag = 0;
  417.   dp->bitstream->read_len = dp->bitstream->frame_bitoffset = 0;
  418.   InterpretSPS (dp, sps);
  419.   if (sps->Valid)
  420.   {
  421.     if (active_sps)
  422.     {
  423.       if (sps->seq_parameter_set_id == active_sps->seq_parameter_set_id)
  424.       {
  425.         if (!sps_is_equal(sps, active_sps))
  426.         {
  427.           if (dec_picture)
  428.           {
  429.             // this may only happen on slice loss
  430.             exit_picture();
  431.           }
  432.           active_sps=NULL;
  433.         }
  434.       }
  435.     }
  436.     // SPSConsistencyCheck (pps);
  437.     MakeSPSavailable (sps->seq_parameter_set_id, sps);
  438.     img->profile_idc = sps->profile_idc; //ADD-VG
  439.     img->separate_colour_plane_flag = sps->separate_colour_plane_flag;
  440.     if( img->separate_colour_plane_flag )
  441.     {
  442.       img->ChromaArrayType = 0;
  443.     }
  444.     else
  445.     {
  446.       img->ChromaArrayType = sps->chroma_format_idc;
  447.     }
  448.   }
  449.   FreePartition (dp, 1);
  450.   FreeSPS (sps);
  451. }
  452. void ProcessPPS (NALU_t *nalu)
  453. {
  454.   DataPartition *dp;
  455.   pic_parameter_set_rbsp_t *pps;
  456.   dp = AllocPartition(1);
  457.   pps = AllocPPS();
  458.   memcpy (dp->bitstream->streamBuffer, &nalu->buf[1], nalu->len-1);
  459.   dp->bitstream->code_len = dp->bitstream->bitstream_length = RBSPtoSODB (dp->bitstream->streamBuffer, nalu->len-1);
  460.   dp->bitstream->ei_flag = 0;
  461.   dp->bitstream->read_len = dp->bitstream->frame_bitoffset = 0;
  462.   InterpretPPS (dp, pps);
  463.   // PPSConsistencyCheck (pps);
  464.   if (active_pps)
  465.   {
  466.     if (pps->pic_parameter_set_id == active_pps->pic_parameter_set_id)
  467.     {
  468.       if (!pps_is_equal(pps, active_pps))
  469.       {
  470.         if (dec_picture)
  471.         {
  472.           // this may only happen on slice loss
  473.           exit_picture();
  474.         }
  475.         active_pps = NULL;
  476.       }
  477.     }
  478.   }
  479.   MakePPSavailable (pps->pic_parameter_set_id, pps);
  480.   FreePartition (dp, 1);
  481.   FreePPS (pps);
  482. }
  483. void activate_sps (seq_parameter_set_rbsp_t *sps)
  484. {
  485.   int nplane;
  486.   if (active_sps != sps)
  487.   {
  488.     if (dec_picture)
  489.     {
  490.       // this may only happen on slice loss
  491.       exit_picture();
  492.     }
  493.     active_sps = sps;
  494.     img->bitdepth_chroma = 0;
  495.     img->width_cr        = 0;
  496.     img->height_cr       = 0;
  497.     // maximum vertical motion vector range in luma quarter pixel units
  498.     if (active_sps->level_idc <= 10)
  499.     {
  500.       img->max_vmv_r = 64 * 4;
  501.     }
  502.     else if (active_sps->level_idc <= 20)
  503.     {
  504.       img->max_vmv_r = 128 * 4;
  505.     }
  506.     else if (active_sps->level_idc <= 30)
  507.     {
  508.       img->max_vmv_r = 256 * 4;
  509.     }
  510.     else
  511.     {
  512.       img->max_vmv_r = 512 * 4; // 512 pixels in quarter pixels
  513.     }
  514.     // Fidelity Range Extensions stuff (part 1)
  515.     img->bitdepth_luma       = sps->bit_depth_luma_minus8 + 8;
  516.     img->bitdepth_scale[0]   = 1 << sps->bit_depth_luma_minus8;
  517.     if (sps->chroma_format_idc != YUV400)
  518.     {
  519.       img->bitdepth_chroma   = sps->bit_depth_chroma_minus8 + 8;
  520.       img->bitdepth_scale[1] = 1 << sps->bit_depth_chroma_minus8;
  521.     }
  522.     img->MaxFrameNum = 1<<(sps->log2_max_frame_num_minus4+4);
  523.     img->PicWidthInMbs = (sps->pic_width_in_mbs_minus1 +1);
  524.     img->PicHeightInMapUnits = (sps->pic_height_in_map_units_minus1 +1);
  525.     img->FrameHeightInMbs = ( 2 - sps->frame_mbs_only_flag ) * img->PicHeightInMapUnits;
  526.     img->FrameSizeInMbs = img->PicWidthInMbs * img->FrameHeightInMbs;
  527.     img->yuv_format=sps->chroma_format_idc;
  528.     img->width = img->PicWidthInMbs * MB_BLOCK_SIZE;
  529.     img->height = img->FrameHeightInMbs * MB_BLOCK_SIZE;
  530.     if (sps->chroma_format_idc == YUV420)
  531.     {
  532.       img->width_cr = img->width >>1;
  533.       img->height_cr = img->height >>1;
  534.     }
  535.     else if (sps->chroma_format_idc == YUV422)
  536.     {
  537.       img->width_cr = img->width >>1;
  538.       img->height_cr = img->height;
  539.     }
  540.     else if (sps->chroma_format_idc == YUV444)
  541.     {
  542.       //YUV444
  543.       img->width_cr = img->width;
  544.       img->height_cr = img->height;
  545.     }
  546.     img->width_cr_m1 = img->width_cr - 1;
  547.     init_frext(img);
  548.     init_global_buffers();
  549.     if (!img->no_output_of_prior_pics_flag)
  550.     {
  551.       flush_dpb();
  552.     }
  553.     init_dpb();
  554.     if( IS_INDEPENDENT(img) )
  555.     {
  556.       for( nplane=0; nplane<MAX_PLANE; nplane++ )
  557.       {
  558.         if( NULL != Co_located_JV[nplane] )
  559.         {
  560.           free_colocated(Co_located_JV[nplane]);        
  561.         }
  562.         Co_located_JV[nplane] = alloc_colocated (img->width, img->height,sps->mb_adaptive_frame_field_flag);      
  563.       }
  564.     }
  565.     else
  566.     {
  567.       if (NULL!=Co_located)
  568.       {
  569.         free_colocated(Co_located);
  570.       }
  571.       Co_located = alloc_colocated (img->width, img->height,sps->mb_adaptive_frame_field_flag);
  572.     }
  573.     ercInit(img->width, img->height, 1);
  574.   }
  575. }
  576. void activate_pps(pic_parameter_set_rbsp_t *pps)
  577. {
  578.   if (active_pps != pps)
  579.   {
  580.     if (dec_picture)
  581.     {
  582.       // this may only happen on slice loss
  583.       exit_picture();
  584.     }
  585.     active_pps = pps;
  586.     // Fidelity Range Extensions stuff (part 2)
  587.     img->Transform8x8Mode = pps->transform_8x8_mode_flag;
  588.   }
  589. }
  590. void UseParameterSet (int PicParsetId)
  591. {
  592.   seq_parameter_set_rbsp_t *sps = &SeqParSet[PicParSet[PicParsetId].seq_parameter_set_id];
  593.   pic_parameter_set_rbsp_t *pps = &PicParSet[PicParsetId];
  594.   int i;
  595.   if (PicParSet[PicParsetId].Valid != TRUE)
  596.     printf ("Trying to use an invalid (uninitialized) Picture Parameter Set with ID %d, expect the unexpected...n", PicParsetId);
  597.   if (SeqParSet[PicParSet[PicParsetId].seq_parameter_set_id].Valid != TRUE)
  598.     printf ("PicParset %d references an invalid (uninitialized) Sequence Parameter Set with ID %d, expect the unexpected...n", PicParsetId, (int) PicParSet[PicParsetId].seq_parameter_set_id);
  599.   sps =  &SeqParSet[PicParSet[PicParsetId].seq_parameter_set_id];
  600.   // In theory, and with a well-designed software, the lines above
  601.   // are everything necessary.  In practice, we need to patch many values
  602.   // in img-> (but no more in inp-> -- these have been taken care of)
  603.   // Sequence Parameter Set Stuff first
  604. //  printf ("Using Picture Parameter set %d and associated Sequence Parameter Set %dn", PicParsetId, PicParSet[PicParsetId].seq_parameter_set_id);
  605.   if ((int) sps->pic_order_cnt_type < 0 || sps->pic_order_cnt_type > 2)  // != 1
  606.   {
  607.     printf ("invalid sps->pic_order_cnt_type = %dn", (int) sps->pic_order_cnt_type);
  608.     error ("pic_order_cnt_type != 1", -1000);
  609.   }
  610.   if (sps->pic_order_cnt_type == 1)
  611.   {
  612.     if(sps->num_ref_frames_in_pic_order_cnt_cycle >= MAXnum_ref_frames_in_pic_order_cnt_cycle)
  613.     {
  614.       error("num_ref_frames_in_pic_order_cnt_cycle too large",-1011);
  615.     }
  616.   }
  617.   activate_sps(sps);
  618.   activate_pps(pps);
  619.   // currSlice->dp_mode is set by read_new_slice (NALU first byte available there)
  620.   if (pps->entropy_coding_mode_flag == UVLC)
  621.   {
  622.     nal_startcode_follows = uvlc_startcode_follows;
  623.     for (i=0; i<3; i++)
  624.     {
  625.       img->currentSlice->partArr[i].readSyntaxElement = readSyntaxElement_UVLC;
  626.     }
  627.   }
  628.   else
  629.   {
  630.     nal_startcode_follows = cabac_startcode_follows;
  631.     for (i=0; i<3; i++)
  632.     {
  633.       img->currentSlice->partArr[i].readSyntaxElement = readSyntaxElement_CABAC;
  634.     }
  635.   }
  636. }