1 /**************************************************************************
2  *
3  * Copyright 2014 Advanced Micro Devices, Inc.
4  * All Rights Reserved.
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining a
7  * copy of this software and associated documentation files (the
8  * "Software"), to deal in the Software without restriction, including
9  * without limitation the rights to use, copy, modify, merge, publish,
10  * distribute, sub license, and/or sell copies of the Software, and to
11  * permit persons to whom the Software is furnished to do so, subject to
12  * the following conditions:
13  *
14  * The above copyright notice and this permission notice (including the
15  * next paragraph) shall be included in all copies or substantial portions
16  * of the Software.
17  *
18  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
19  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
20  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
21  * IN NO EVENT SHALL THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR
22  * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
23  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
24  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
25  *
26  **************************************************************************/
27 
28 #include "va_private.h"
29 
vlVaHandlePictureParameterBufferMPEG4(vlVaDriver * drv,vlVaContext * context,vlVaBuffer * buf)30 void vlVaHandlePictureParameterBufferMPEG4(vlVaDriver *drv, vlVaContext *context, vlVaBuffer *buf)
31 {
32    static const uint8_t default_intra_quant_matrix[64] = { 0 };
33    static const uint8_t default_non_intra_quant_matrix[64] = { 0 };
34 
35    VAPictureParameterBufferMPEG4 *mpeg4 = buf->data;
36    unsigned i;
37 
38    assert(buf->size >= sizeof(VAPictureParameterBufferMPEG4) && buf->num_elements == 1);
39 
40    context->mpeg4.pps = *mpeg4;
41 
42    /* vop_width */
43    /* vop_height */
44    /* forward_reference_picture */
45    /* backward_reference_picture */
46    context->desc.mpeg4.short_video_header =
47          mpeg4->vol_fields.bits.short_video_header;
48    /* chroma_format */
49    context->desc.mpeg4.interlaced = mpeg4->vol_fields.bits.interlaced;
50    /* obmc_disable */
51    /* sprite_enable */
52    /* sprite_warping_accuracy */
53    context->desc.mpeg4.quant_type = mpeg4->vol_fields.bits.quant_type;
54    context->desc.mpeg4.quarter_sample = mpeg4->vol_fields.bits.quarter_sample;
55    /* data_partitioned */
56    /* reversible_vlc */
57    context->desc.mpeg4.resync_marker_disable =
58          mpeg4->vol_fields.bits.resync_marker_disable;
59    /* no_of_sprite_warping_points */
60    /* sprite_trajectory_du */
61    /* sprite_trajectory_dv */
62    /* quant_precision */
63    context->desc.mpeg4.vop_coding_type = mpeg4->vop_fields.bits.vop_coding_type;
64    /* backward_reference_vop_coding_type */
65    /* vop_rounding_type */
66    /* intra_dc_vlc_thr */
67    context->desc.mpeg4.top_field_first =
68          mpeg4->vop_fields.bits.top_field_first;
69    context->desc.mpeg4.alternate_vertical_scan_flag =
70          mpeg4->vop_fields.bits.alternate_vertical_scan_flag;
71    context->desc.mpeg4.vop_fcode_forward = mpeg4->vop_fcode_forward;
72    context->desc.mpeg4.vop_fcode_backward = mpeg4->vop_fcode_backward;
73    context->desc.mpeg4.vop_time_increment_resolution =
74          mpeg4->vop_time_increment_resolution;
75    /* num_gobs_in_vop */
76    /* num_macroblocks_in_gob */
77    context->desc.mpeg4.trb[0] = mpeg4->TRB;
78    context->desc.mpeg4.trb[1] = mpeg4->TRB;
79    context->desc.mpeg4.trd[0] = mpeg4->TRD;
80    context->desc.mpeg4.trd[1] = mpeg4->TRD;
81 
82    /* default [non-]intra quant matrix because mpv does not set these
83       matrices */
84    if (!context->desc.mpeg4.intra_matrix)
85       context->desc.mpeg4.intra_matrix = default_intra_quant_matrix;
86    if (!context->desc.mpeg4.non_intra_matrix)
87       context->desc.mpeg4.non_intra_matrix = default_non_intra_quant_matrix;
88 
89    vlVaGetReferenceFrame(drv, mpeg4->forward_reference_picture, &context->desc.mpeg4.ref[0]);
90    vlVaGetReferenceFrame(drv, mpeg4->backward_reference_picture, &context->desc.mpeg4.ref[1]);
91 
92    context->mpeg4.vti_bits = 0;
93    for (i = context->desc.mpeg4.vop_time_increment_resolution; i > 0; i /= 2)
94       ++context->mpeg4.vti_bits;
95 }
96 
vlVaHandleIQMatrixBufferMPEG4(vlVaContext * context,vlVaBuffer * buf)97 void vlVaHandleIQMatrixBufferMPEG4(vlVaContext *context, vlVaBuffer *buf)
98 {
99    VAIQMatrixBufferMPEG4 *mpeg4 = buf->data;
100 
101    assert(buf->size >= sizeof(VAIQMatrixBufferMPEG4) && buf->num_elements == 1);
102    if (mpeg4->load_intra_quant_mat)
103       context->desc.mpeg4.intra_matrix = mpeg4->intra_quant_mat;
104    else
105       context->desc.mpeg4.intra_matrix = NULL;
106 
107    if (mpeg4->load_non_intra_quant_mat)
108       context->desc.mpeg4.non_intra_matrix = mpeg4->non_intra_quant_mat;
109    else
110       context->desc.mpeg4.non_intra_matrix = NULL;
111 }
112 
vlVaHandleSliceParameterBufferMPEG4(vlVaContext * context,vlVaBuffer * buf)113 void vlVaHandleSliceParameterBufferMPEG4(vlVaContext *context, vlVaBuffer *buf)
114 {
115    VASliceParameterBufferMPEG4 *mpeg4 = buf->data;
116 
117    assert(buf->size >= sizeof(VASliceParameterBufferMPEG4) && buf->num_elements == 1);
118    context->mpeg4.quant_scale = mpeg4->quant_scale;
119 }
120 
121 struct bit_stream
122 {
123    uint8_t *data;
124    unsigned int length; /* bits */
125    unsigned int pos;    /* bits */
126 };
127 
128 static inline void
write_bit(struct bit_stream * writer,unsigned int bit)129 write_bit(struct bit_stream *writer, unsigned int bit)
130 {
131    assert(writer->length > (writer)->pos);
132    writer->data[writer->pos>>3] |= ((bit & 1)<<(7 - (writer->pos & 7)));
133    writer->pos++;
134 }
135 
136 static inline void
write_bits(struct bit_stream * writer,unsigned int bits,unsigned int len)137 write_bits(struct bit_stream *writer, unsigned int bits, unsigned int len)
138 {
139    int i;
140    assert(len <= sizeof(bits)*8);
141    for (i = len - 1; i >= 0; i--)
142       write_bit(writer, bits>>i);
143 }
144 
vlVaDecoderFixMPEG4Startcode(vlVaContext * context)145 void vlVaDecoderFixMPEG4Startcode(vlVaContext *context)
146 {
147    uint8_t vop[] = { 0x00, 0x00, 0x01, 0xb6, 0x00, 0x00, 0x00, 0x00, 0x00 };
148    struct bit_stream bs_vop = {vop, sizeof(vop)*8, 32};
149    unsigned int vop_time_inc;
150    int mod_time;
151    unsigned int vop_size;
152    unsigned int vop_coding_type = context->desc.mpeg4.vop_coding_type;
153 
154    context->mpeg4.start_code_size = 0;
155    memset(context->mpeg4.start_code, 0, sizeof(context->mpeg4.start_code));
156    if (vop_coding_type+1 == PIPE_MPEG12_PICTURE_CODING_TYPE_I) {
157       unsigned int vop_time = context->mpeg4.frame_num/
158             context->desc.mpeg4.vop_time_increment_resolution;
159       unsigned int vop_hour = vop_time / 3600;
160       unsigned int vop_minute = (vop_time / 60) % 60;
161       unsigned int vop_second = vop_time % 60;
162       uint8_t group_of_vop[] = { 0x00, 0x00, 0x01, 0xb3, 0x00, 0x00, 0x00 };
163       struct bit_stream bs_gvop = {group_of_vop, sizeof(group_of_vop)*8, 32};
164 
165       write_bits(&bs_gvop, vop_hour, 5);
166       write_bits(&bs_gvop, vop_minute, 6);
167       write_bit(&bs_gvop, 1); /* marker_bit */
168       write_bits(&bs_gvop, vop_second, 6);
169       write_bit(&bs_gvop, 0); /* closed_gov */ /* TODO replace magic */
170       write_bit(&bs_gvop, 0); /* broken_link */
171       write_bit(&bs_gvop, 0); /* padding */
172       write_bits(&bs_gvop, 7, 3); /* padding */
173 
174       memcpy(context->mpeg4.start_code, group_of_vop, sizeof(group_of_vop));
175       context->mpeg4.start_code_size += sizeof(group_of_vop);
176    }
177 
178    write_bits(&bs_vop, vop_coding_type, 2);
179    mod_time = context->mpeg4.frame_num %
180          context->desc.mpeg4.vop_time_increment_resolution == 0 &&
181          vop_coding_type+1 != PIPE_MPEG12_PICTURE_CODING_TYPE_I;
182    while (mod_time--)
183       write_bit(&bs_vop, 1); /* modulo_time_base */
184    write_bit(&bs_vop, 0); /* modulo_time_base */
185 
186    write_bit(&bs_vop, 1); /* marker_bit */
187    vop_time_inc = context->mpeg4.frame_num %
188          context->desc.mpeg4.vop_time_increment_resolution;
189    write_bits(&bs_vop, vop_time_inc, context->mpeg4.vti_bits);
190    write_bit(&bs_vop, 1); /* marker_bit */
191    write_bit(&bs_vop, 1); /* vop_coded */
192    if (vop_coding_type+1 == PIPE_MPEG12_PICTURE_CODING_TYPE_P)
193       write_bit(&bs_vop, context->mpeg4.pps.vop_fields.bits.vop_rounding_type);
194    write_bits(&bs_vop, context->mpeg4.pps.vop_fields.bits.intra_dc_vlc_thr, 3);
195    if (context->mpeg4.pps.vol_fields.bits.interlaced) {
196       write_bit(&bs_vop, context->mpeg4.pps.vop_fields.bits.top_field_first);
197       write_bit(&bs_vop, context->mpeg4.pps.vop_fields.bits.alternate_vertical_scan_flag);
198    }
199 
200    write_bits(&bs_vop, context->mpeg4.quant_scale, context->mpeg4.pps.quant_precision);
201    if (vop_coding_type+1 != PIPE_MPEG12_PICTURE_CODING_TYPE_I)
202       write_bits(&bs_vop, context->desc.mpeg4.vop_fcode_forward, 3);
203    if (vop_coding_type+1 == PIPE_MPEG12_PICTURE_CODING_TYPE_B)
204       write_bits(&bs_vop, context->desc.mpeg4.vop_fcode_backward, 3);
205 
206    vop_size = bs_vop.pos/8;
207    memcpy(context->mpeg4.start_code + context->mpeg4.start_code_size, vop, vop_size);
208    context->mpeg4.start_code_size += vop_size;
209 }
210