src/gallium/state_trackers/va/picture_h264.c

   1  /**************************************************************************
   2  *
   3  * Copyright 2010 Thomas Balling Sørensen & Orasanu Lucian.
   4  * Copyright 2014 Advanced Micro Devices, Inc.
   5  * All Rights Reserved.
   6  *
   7  * Permission is hereby granted, free of charge, to any person obtaining a
   8  * copy of this software and associated documentation files (the
   9  * "Software"), to deal in the Software without restriction, including
  10  * without limitation the rights to use, copy, modify, merge, publish,
  11  * distribute, sub license, and/or sell copies of the Software, and to
  12  * permit persons to whom the Software is furnished to do so, subject to
  13  * the following conditions:
  14  *
  15  * The above copyright notice and this permission notice (including the
  16  * next paragraph) shall be included in all copies or substantial portions
  17  * of the Software.
  18  *
  19  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  20  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  21  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
  22  * IN NO EVENT SHALL THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR
  23  * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
  24  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
  25  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  26  *
  27  **************************************************************************/
  28
  29 #include "util/u_video.h"
  30 #include "va_private.h"
  31
  32 static void resetReferencePictureDesc(struct pipe_h264_picture_desc *h264,
  33                                       unsigned int i)
  34 {
  35    h264->ref[i] = NULL;
  36    h264->frame_num_list[i] = 0;
  37    h264->is_long_term[i] = 0;
  38    h264->top_is_reference[i] = 0;
  39    h264->bottom_is_reference[i] = 0;
  40    h264->field_order_cnt_list[i][0] = 0;
  41    h264->field_order_cnt_list[i][1] = 0;
  42 }
  43
  44 void vlVaHandlePictureParameterBufferH264(vlVaDriver *drv, vlVaContext *context, vlVaBuffer *buf)
  45 {
  46    VAPictureParameterBufferH264 *h264 = buf->data;
  47    unsigned int top_or_bottom_field;
  48    unsigned i;
  49
  50    assert(buf->size >= sizeof(VAPictureParameterBufferH264) && buf->num_elements == 1);
  51    /*CurrPic*/
  52    context->desc.h264.field_order_cnt[0] = h264->CurrPic.TopFieldOrderCnt;
  53    context->desc.h264.field_order_cnt[1] = h264->CurrPic.BottomFieldOrderCnt;
  54    /*ReferenceFrames[16]*/
  55    /*picture_width_in_mbs_minus1*/
  56    /*picture_height_in_mbs_minus1*/
  57    /*bit_depth_luma_minus8*/
  58    /*bit_depth_chroma_minus8*/
  59    context->desc.h264.num_ref_frames = h264->num_ref_frames;
  60    /*chroma_format_idc*/
  61    /*residual_colour_transform_flag*/
  62    /*gaps_in_frame_num_value_allowed_flag*/
  63    context->desc.h264.pps->sps->frame_mbs_only_flag =
  64       h264->seq_fields.bits.frame_mbs_only_flag;
  65    context->desc.h264.pps->sps->mb_adaptive_frame_field_flag =
  66       h264->seq_fields.bits.mb_adaptive_frame_field_flag;
  67    context->desc.h264.pps->sps->direct_8x8_inference_flag =
  68       h264->seq_fields.bits.direct_8x8_inference_flag;
  69    /*MinLumaBiPredSize8x8*/
  70    context->desc.h264.pps->sps->log2_max_frame_num_minus4 =
  71       h264->seq_fields.bits.log2_max_frame_num_minus4;
  72    context->desc.h264.pps->sps->pic_order_cnt_type =
  73       h264->seq_fields.bits.pic_order_cnt_type;
  74    context->desc.h264.pps->sps->log2_max_pic_order_cnt_lsb_minus4 =
  75       h264->seq_fields.bits.log2_max_pic_order_cnt_lsb_minus4;
  76    context->desc.h264.pps->sps->delta_pic_order_always_zero_flag =
  77       h264->seq_fields.bits.delta_pic_order_always_zero_flag;
  78    /*num_slice_groups_minus1*/
  79    /*slice_group_map_type*/
  80    /*slice_group_change_rate_minus1*/
  81    context->desc.h264.pps->pic_init_qp_minus26 =
  82       h264->pic_init_qp_minus26;
  83    /*pic_init_qs_minus26*/
  84    context->desc.h264.pps->chroma_qp_index_offset =
  85       h264->chroma_qp_index_offset;
  86    context->desc.h264.pps->second_chroma_qp_index_offset =
  87       h264->second_chroma_qp_index_offset;
  88    context->desc.h264.pps->entropy_coding_mode_flag =
  89       h264->pic_fields.bits.entropy_coding_mode_flag;
  90    context->desc.h264.pps->weighted_pred_flag =
  91       h264->pic_fields.bits.weighted_pred_flag;
  92    context->desc.h264.pps->weighted_bipred_idc =
  93       h264->pic_fields.bits.weighted_bipred_idc;
  94    context->desc.h264.pps->transform_8x8_mode_flag =
  95       h264->pic_fields.bits.transform_8x8_mode_flag;
  96    context->desc.h264.field_pic_flag =
  97       h264->pic_fields.bits.field_pic_flag;
  98    context->desc.h264.pps->constrained_intra_pred_flag =
  99       h264->pic_fields.bits.constrained_intra_pred_flag;
 100    context->desc.h264.pps->bottom_field_pic_order_in_frame_present_flag =
 101       h264->pic_fields.bits.pic_order_present_flag;
 102    context->desc.h264.pps->deblocking_filter_control_present_flag =
 103       h264->pic_fields.bits.deblocking_filter_control_present_flag;
 104    context->desc.h264.pps->redundant_pic_cnt_present_flag =
 105       h264->pic_fields.bits.redundant_pic_cnt_present_flag;
 106    /*reference_pic_flag*/
 107    context->desc.h264.frame_num = h264->frame_num;
 108    context->desc.h264.is_reference = h264->pic_fields.bits.reference_pic_flag;
 109    context->desc.h264.bottom_field_flag =
 110       h264->pic_fields.bits.field_pic_flag &&
 111       (h264->CurrPic.flags & VA_PICTURE_H264_BOTTOM_FIELD) != 0;
 112
 113    if (!context->decoder && context->desc.h264.num_ref_frames > 0)
 114       context->templat.max_references = MIN2(context->desc.h264.num_ref_frames, 16);
 115
 116    for (i = 0; i < context->templat.max_references; ++i) {
 117       if ((h264->ReferenceFrames[i].flags & VA_PICTURE_H264_INVALID) ||
 118           (h264->ReferenceFrames[i].picture_id == VA_INVALID_SURFACE)) {
 119          resetReferencePictureDesc(&context->desc.h264, i);
 120          break;
 121       }
 122
 123       vlVaGetReferenceFrame(drv, h264->ReferenceFrames[i].picture_id, &context->desc.h264.ref[i]);
 124       context->desc.h264.frame_num_list[i] = h264->ReferenceFrames[i].frame_idx;
 125
 126       top_or_bottom_field = h264->ReferenceFrames[i].flags &
 127          (VA_PICTURE_H264_TOP_FIELD | VA_PICTURE_H264_BOTTOM_FIELD);
 128       context->desc.h264.is_long_term[i] = (h264->ReferenceFrames[i].flags &
 129          (VA_PICTURE_H264_SHORT_TERM_REFERENCE |
 130          VA_PICTURE_H264_LONG_TERM_REFERENCE)) !=
 131          VA_PICTURE_H264_SHORT_TERM_REFERENCE;
 132       context->desc.h264.top_is_reference[i] =
 133          !context->desc.h264.is_long_term[i] ||
 134          !!(h264->ReferenceFrames[i].flags & VA_PICTURE_H264_TOP_FIELD);
 135       context->desc.h264.bottom_is_reference[i] =
 136          !context->desc.h264.is_long_term[i] ||
 137          !!(h264->ReferenceFrames[i].flags & VA_PICTURE_H264_BOTTOM_FIELD);
 138       context->desc.h264.field_order_cnt_list[i][0] =
 139          top_or_bottom_field != VA_PICTURE_H264_BOTTOM_FIELD ?
 140          h264->ReferenceFrames[i].TopFieldOrderCnt: INT_MAX;
 141       context->desc.h264.field_order_cnt_list[i][1] =
 142          top_or_bottom_field != VA_PICTURE_H264_TOP_FIELD ?
 143          h264->ReferenceFrames[i].BottomFieldOrderCnt: INT_MAX;
 144    }
 145
 146    /* Make sure remaining elements are clean */
 147    for (; i < 16; ++i)
 148       resetReferencePictureDesc(&context->desc.h264, i);
 149 }
 150
 151 void vlVaHandleIQMatrixBufferH264(vlVaContext *context, vlVaBuffer *buf)
 152 {
 153    VAIQMatrixBufferH264 *h264 = buf->data;
 154
 155    assert(buf->size >= sizeof(VAIQMatrixBufferH264) && buf->num_elements == 1);
 156    memcpy(&context->desc.h264.pps->ScalingList4x4, h264->ScalingList4x4, 6 * 16);
 157    memcpy(&context->desc.h264.pps->ScalingList8x8, h264->ScalingList8x8, 2 * 64);
 158 }
 159
 160 void vlVaHandleSliceParameterBufferH264(vlVaContext *context, vlVaBuffer *buf)
 161 {
 162    VASliceParameterBufferH264 *h264 = buf->data;
 163
 164    assert(buf->size >= sizeof(VASliceParameterBufferH264) && buf->num_elements == 1);
 165    context->desc.h264.num_ref_idx_l0_active_minus1 =
 166       h264->num_ref_idx_l0_active_minus1;
 167    context->desc.h264.num_ref_idx_l1_active_minus1 =
 168       h264->num_ref_idx_l1_active_minus1;
 169 }