Home | History | Annotate | Download | only in va
      1  /**************************************************************************
      2  *
      3  * Copyright 2010 Thomas Balling Srensen & 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    context->desc.h264.slice_count = 0;
     52    /*CurrPic*/
     53    context->desc.h264.field_order_cnt[0] = h264->CurrPic.TopFieldOrderCnt;
     54    context->desc.h264.field_order_cnt[1] = h264->CurrPic.BottomFieldOrderCnt;
     55    /*ReferenceFrames[16]*/
     56    /*picture_width_in_mbs_minus1*/
     57    /*picture_height_in_mbs_minus1*/
     58    /*bit_depth_luma_minus8*/
     59    /*bit_depth_chroma_minus8*/
     60    context->desc.h264.num_ref_frames = h264->num_ref_frames;
     61    /*chroma_format_idc*/
     62    /*residual_colour_transform_flag*/
     63    /*gaps_in_frame_num_value_allowed_flag*/
     64    context->desc.h264.pps->sps->frame_mbs_only_flag =
     65       h264->seq_fields.bits.frame_mbs_only_flag;
     66    context->desc.h264.pps->sps->mb_adaptive_frame_field_flag =
     67       h264->seq_fields.bits.mb_adaptive_frame_field_flag;
     68    context->desc.h264.pps->sps->direct_8x8_inference_flag =
     69       h264->seq_fields.bits.direct_8x8_inference_flag;
     70    /*MinLumaBiPredSize8x8*/
     71    context->desc.h264.pps->sps->log2_max_frame_num_minus4 =
     72       h264->seq_fields.bits.log2_max_frame_num_minus4;
     73    context->desc.h264.pps->sps->pic_order_cnt_type =
     74       h264->seq_fields.bits.pic_order_cnt_type;
     75    context->desc.h264.pps->sps->log2_max_pic_order_cnt_lsb_minus4 =
     76       h264->seq_fields.bits.log2_max_pic_order_cnt_lsb_minus4;
     77    context->desc.h264.pps->sps->delta_pic_order_always_zero_flag =
     78       h264->seq_fields.bits.delta_pic_order_always_zero_flag;
     79    /*num_slice_groups_minus1*/
     80    /*slice_group_map_type*/
     81    /*slice_group_change_rate_minus1*/
     82    context->desc.h264.pps->pic_init_qp_minus26 =
     83       h264->pic_init_qp_minus26;
     84    /*pic_init_qs_minus26*/
     85    context->desc.h264.pps->chroma_qp_index_offset =
     86       h264->chroma_qp_index_offset;
     87    context->desc.h264.pps->second_chroma_qp_index_offset =
     88       h264->second_chroma_qp_index_offset;
     89    context->desc.h264.pps->entropy_coding_mode_flag =
     90       h264->pic_fields.bits.entropy_coding_mode_flag;
     91    context->desc.h264.pps->weighted_pred_flag =
     92       h264->pic_fields.bits.weighted_pred_flag;
     93    context->desc.h264.pps->weighted_bipred_idc =
     94       h264->pic_fields.bits.weighted_bipred_idc;
     95    context->desc.h264.pps->transform_8x8_mode_flag =
     96       h264->pic_fields.bits.transform_8x8_mode_flag;
     97    context->desc.h264.field_pic_flag =
     98       h264->pic_fields.bits.field_pic_flag;
     99    context->desc.h264.pps->constrained_intra_pred_flag =
    100       h264->pic_fields.bits.constrained_intra_pred_flag;
    101    context->desc.h264.pps->bottom_field_pic_order_in_frame_present_flag =
    102       h264->pic_fields.bits.pic_order_present_flag;
    103    context->desc.h264.pps->deblocking_filter_control_present_flag =
    104       h264->pic_fields.bits.deblocking_filter_control_present_flag;
    105    context->desc.h264.pps->redundant_pic_cnt_present_flag =
    106       h264->pic_fields.bits.redundant_pic_cnt_present_flag;
    107    /*reference_pic_flag*/
    108    context->desc.h264.frame_num = h264->frame_num;
    109    context->desc.h264.is_reference = h264->pic_fields.bits.reference_pic_flag;
    110    context->desc.h264.bottom_field_flag =
    111       h264->pic_fields.bits.field_pic_flag &&
    112       (h264->CurrPic.flags & VA_PICTURE_H264_BOTTOM_FIELD) != 0;
    113 
    114    if (!context->decoder && context->desc.h264.num_ref_frames > 0)
    115       context->templat.max_references = MIN2(context->desc.h264.num_ref_frames, 16);
    116 
    117    for (i = 0; i < context->templat.max_references; ++i) {
    118       if ((h264->ReferenceFrames[i].flags & VA_PICTURE_H264_INVALID) ||
    119           (h264->ReferenceFrames[i].picture_id == VA_INVALID_SURFACE)) {
    120          resetReferencePictureDesc(&context->desc.h264, i);
    121          break;
    122       }
    123 
    124       vlVaGetReferenceFrame(drv, h264->ReferenceFrames[i].picture_id, &context->desc.h264.ref[i]);
    125       context->desc.h264.frame_num_list[i] = h264->ReferenceFrames[i].frame_idx;
    126 
    127       top_or_bottom_field = h264->ReferenceFrames[i].flags &
    128          (VA_PICTURE_H264_TOP_FIELD | VA_PICTURE_H264_BOTTOM_FIELD);
    129       context->desc.h264.is_long_term[i] = (h264->ReferenceFrames[i].flags &
    130          (VA_PICTURE_H264_SHORT_TERM_REFERENCE |
    131          VA_PICTURE_H264_LONG_TERM_REFERENCE)) !=
    132          VA_PICTURE_H264_SHORT_TERM_REFERENCE;
    133       context->desc.h264.top_is_reference[i] =
    134          !context->desc.h264.is_long_term[i] ||
    135          !!(h264->ReferenceFrames[i].flags & VA_PICTURE_H264_TOP_FIELD);
    136       context->desc.h264.bottom_is_reference[i] =
    137          !context->desc.h264.is_long_term[i] ||
    138          !!(h264->ReferenceFrames[i].flags & VA_PICTURE_H264_BOTTOM_FIELD);
    139       context->desc.h264.field_order_cnt_list[i][0] =
    140          top_or_bottom_field != VA_PICTURE_H264_BOTTOM_FIELD ?
    141          h264->ReferenceFrames[i].TopFieldOrderCnt: INT_MAX;
    142       context->desc.h264.field_order_cnt_list[i][1] =
    143          top_or_bottom_field != VA_PICTURE_H264_TOP_FIELD ?
    144          h264->ReferenceFrames[i].BottomFieldOrderCnt: INT_MAX;
    145    }
    146 
    147    /* Make sure remaining elements are clean */
    148    for (; i < 16; ++i)
    149       resetReferencePictureDesc(&context->desc.h264, i);
    150 }
    151 
    152 void vlVaHandleIQMatrixBufferH264(vlVaContext *context, vlVaBuffer *buf)
    153 {
    154    VAIQMatrixBufferH264 *h264 = buf->data;
    155 
    156    assert(buf->size >= sizeof(VAIQMatrixBufferH264) && buf->num_elements == 1);
    157    memcpy(&context->desc.h264.pps->ScalingList4x4, h264->ScalingList4x4, 6 * 16);
    158    memcpy(&context->desc.h264.pps->ScalingList8x8, h264->ScalingList8x8, 2 * 64);
    159 }
    160 
    161 void vlVaHandleSliceParameterBufferH264(vlVaContext *context, vlVaBuffer *buf)
    162 {
    163    VASliceParameterBufferH264 *h264 = buf->data;
    164 
    165    assert(buf->size >= sizeof(VASliceParameterBufferH264) && buf->num_elements == 1);
    166    context->desc.h264.slice_count += buf->num_elements;
    167    context->desc.h264.num_ref_idx_l0_active_minus1 =
    168       h264->num_ref_idx_l0_active_minus1;
    169    context->desc.h264.num_ref_idx_l1_active_minus1 =
    170       h264->num_ref_idx_l1_active_minus1;
    171 }
    172