based on a patch by anonymous, with optimizations by me. Originally committed as revision 4660 to svn://svn.ffmpeg.org/ffmpeg/trunktags/v0.5
| @@ -90,6 +90,9 @@ typedef struct SPS{ | |||
| short offset_for_ref_frame[256]; //FIXME dyn aloc? | |||
| int bitstream_restriction_flag; | |||
| int num_reorder_frames; | |||
| int scaling_matrix_present; | |||
| uint8_t scaling_matrix4[6][16]; | |||
| uint8_t scaling_matrix8[2][64]; | |||
| }SPS; | |||
| /** | |||
| @@ -111,6 +114,8 @@ typedef struct PPS{ | |||
| int constrained_intra_pred; ///< constrained_intra_pred_flag | |||
| int redundant_pic_cnt_present; ///< redundant_pic_cnt_present_flag | |||
| int transform_8x8_mode; ///< transform_8x8_mode_flag | |||
| uint8_t scaling_matrix4[6][16]; | |||
| uint8_t scaling_matrix8[2][64]; | |||
| }PPS; | |||
| /** | |||
| @@ -237,8 +242,11 @@ typedef struct H264Context{ | |||
| */ | |||
| PPS pps; //FIXME move to Picture perhaps? (->no) do we need that? | |||
| uint16_t (*dequant4_coeff)[16]; // FIXME quant matrices should be per SPS or PPS | |||
| uint16_t (*dequant8_coeff)[64]; | |||
| uint32_t dequant4_buffer[6][52][16]; | |||
| uint32_t dequant8_buffer[2][52][64]; | |||
| uint32_t (*dequant4_coeff[6])[16]; | |||
| uint32_t (*dequant8_coeff[2])[64]; | |||
| int dequant_coeff_pps; ///< reinit tables when pps changes | |||
| int slice_num; | |||
| uint8_t *slice_table_base; | |||
| @@ -1598,8 +1606,7 @@ static int decode_rbsp_trailing(uint8_t *src){ | |||
| * idct tranforms the 16 dc values and dequantize them. | |||
| * @param qp quantization parameter | |||
| */ | |||
| static void h264_luma_dc_dequant_idct_c(DCTELEM *block, int qp){ | |||
| const int qmul= dequant_coeff[qp][0]; | |||
| static void h264_luma_dc_dequant_idct_c(DCTELEM *block, int qp, int qmul){ | |||
| #define stride 16 | |||
| int i; | |||
| int temp[16]; //FIXME check if this is a good idea | |||
| @@ -1628,10 +1635,10 @@ static void h264_luma_dc_dequant_idct_c(DCTELEM *block, int qp){ | |||
| const int z2= temp[4*1+i] - temp[4*3+i]; | |||
| const int z3= temp[4*1+i] + temp[4*3+i]; | |||
| block[stride*0 +offset]= ((z0 + z3)*qmul + 2)>>2; //FIXME think about merging this into decode_resdual | |||
| block[stride*2 +offset]= ((z1 + z2)*qmul + 2)>>2; | |||
| block[stride*8 +offset]= ((z1 - z2)*qmul + 2)>>2; | |||
| block[stride*10+offset]= ((z0 - z3)*qmul + 2)>>2; | |||
| block[stride*0 +offset]= ((((z0 + z3)*qmul + 128 ) >> 8)); //FIXME think about merging this into decode_resdual | |||
| block[stride*2 +offset]= ((((z1 + z2)*qmul + 128 ) >> 8)); | |||
| block[stride*8 +offset]= ((((z1 - z2)*qmul + 128 ) >> 8)); | |||
| block[stride*10+offset]= ((((z0 - z3)*qmul + 128 ) >> 8)); | |||
| } | |||
| } | |||
| @@ -1678,8 +1685,7 @@ static void h264_luma_dc_dct_c(DCTELEM *block/*, int qp*/){ | |||
| #undef xStride | |||
| #undef stride | |||
| static void chroma_dc_dequant_idct_c(DCTELEM *block, int qp){ | |||
| const int qmul= dequant_coeff[qp][0]; | |||
| static void chroma_dc_dequant_idct_c(DCTELEM *block, int qp, int qmul){ | |||
| const int stride= 16*2; | |||
| const int xStride= 16; | |||
| int a,b,c,d,e; | |||
| @@ -1694,10 +1700,10 @@ static void chroma_dc_dequant_idct_c(DCTELEM *block, int qp){ | |||
| b= c-d; | |||
| c= c+d; | |||
| block[stride*0 + xStride*0]= ((a+c)*qmul + 0)>>1; | |||
| block[stride*0 + xStride*1]= ((e+b)*qmul + 0)>>1; | |||
| block[stride*1 + xStride*0]= ((a-c)*qmul + 0)>>1; | |||
| block[stride*1 + xStride*1]= ((e-b)*qmul + 0)>>1; | |||
| block[stride*0 + xStride*0]= ((a+c)*qmul) >> 7; | |||
| block[stride*0 + xStride*1]= ((e+b)*qmul) >> 7; | |||
| block[stride*1 + xStride*0]= ((a-c)*qmul) >> 7; | |||
| block[stride*1 + xStride*1]= ((e-b)*qmul) >> 7; | |||
| } | |||
| #if 0 | |||
| @@ -2921,12 +2927,70 @@ static void free_tables(H264Context *h){ | |||
| av_freep(&h->mb2b_xy); | |||
| av_freep(&h->mb2b8_xy); | |||
| av_freep(&h->dequant4_coeff); | |||
| av_freep(&h->dequant8_coeff); | |||
| av_freep(&h->s.obmc_scratchpad); | |||
| } | |||
| static void init_dequant8_coeff_table(H264Context *h){ | |||
| int i,q,x; | |||
| h->dequant8_coeff[0] = h->dequant8_buffer[0]; | |||
| h->dequant8_coeff[1] = h->dequant8_buffer[1]; | |||
| for(i=0; i<2; i++ ){ | |||
| if(i && !memcmp(h->pps.scaling_matrix8[0], h->pps.scaling_matrix8[1], 64*sizeof(uint8_t))){ | |||
| h->dequant8_coeff[1] = h->dequant8_buffer[0]; | |||
| break; | |||
| } | |||
| for(q=0; q<52; q++){ | |||
| int shift = div6[q]; | |||
| int idx = rem6[q]; | |||
| for(x=0; x<64; x++) | |||
| h->dequant8_coeff[i][q][x] = ((uint32_t)dequant8_coeff_init[idx][ | |||
| dequant8_coeff_init_scan[((x>>1)&12) | (x&3)] ] * h->pps.scaling_matrix8[i][x]) << shift; | |||
| } | |||
| } | |||
| } | |||
| static void init_dequant4_coeff_table(H264Context *h){ | |||
| int i,j,q,x; | |||
| for(i=0; i<6; i++ ){ | |||
| h->dequant4_coeff[i] = h->dequant4_buffer[i]; | |||
| for(j=0; j<i; j++){ | |||
| if(!memcmp(h->pps.scaling_matrix4[j], h->pps.scaling_matrix4[i], 16*sizeof(uint8_t))){ | |||
| h->dequant4_coeff[i] = h->dequant4_buffer[j]; | |||
| break; | |||
| } | |||
| } | |||
| if(j<i) | |||
| continue; | |||
| for(q=0; q<52; q++){ | |||
| int shift = div6[q] + 2; | |||
| int idx = rem6[q]; | |||
| for(x=0; x<16; x++) | |||
| h->dequant4_coeff[i][q][x] = ((uint32_t)dequant4_coeff_init[idx][(x&1) + ((x>>2)&1)] * | |||
| h->pps.scaling_matrix4[i][x]) << shift; | |||
| } | |||
| } | |||
| } | |||
| static void init_dequant_tables(H264Context *h){ | |||
| int i,x; | |||
| init_dequant4_coeff_table(h); | |||
| if(h->pps.transform_8x8_mode) | |||
| init_dequant8_coeff_table(h); | |||
| if(h->sps.transform_bypass){ | |||
| for(i=0; i<6; i++) | |||
| for(x=0; x<16; x++) | |||
| h->dequant4_coeff[i][0][x] = 1<<6; | |||
| if(h->pps.transform_8x8_mode) | |||
| for(i=0; i<2; i++) | |||
| for(x=0; x<64; x++) | |||
| h->dequant8_coeff[i][0][x] = 1<<6; | |||
| } | |||
| } | |||
| /** | |||
| * allocates tables. | |||
| * needs width/height | |||
| @@ -2934,7 +2998,7 @@ static void free_tables(H264Context *h){ | |||
| static int alloc_tables(H264Context *h){ | |||
| MpegEncContext * const s = &h->s; | |||
| const int big_mb_num= s->mb_stride * (s->mb_height+1); | |||
| int x,y,q; | |||
| int x,y; | |||
| CHECKED_ALLOCZ(h->intra4x4_pred_mode, big_mb_num * 8 * sizeof(uint8_t)) | |||
| @@ -2967,25 +3031,6 @@ static int alloc_tables(H264Context *h){ | |||
| } | |||
| } | |||
| CHECKED_ALLOCZ(h->dequant4_coeff, 52*16 * sizeof(uint16_t)); | |||
| CHECKED_ALLOCZ(h->dequant8_coeff, 52*64 * sizeof(uint16_t)); | |||
| memcpy(h->dequant4_coeff, dequant_coeff, 52*16 * sizeof(uint16_t)); | |||
| for(q=0; q<52; q++){ | |||
| int shift = div6[q]; | |||
| int idx = rem6[q]; | |||
| if(shift >= 2) // qp<12 are shifted during dequant | |||
| shift -= 2; | |||
| for(x=0; x<64; x++) | |||
| h->dequant8_coeff[q][x] = dequant8_coeff_init[idx][ | |||
| dequant8_coeff_init_scan[((x>>1)&12) | (x&3)] ] << shift; | |||
| } | |||
| if(h->sps.transform_bypass){ | |||
| for(x=0; x<16; x++) | |||
| h->dequant4_coeff[0][x] = 1; | |||
| for(x=0; x<64; x++) | |||
| h->dequant8_coeff[0][x] = 1<<2; | |||
| } | |||
| s->obmc_scratchpad = NULL; | |||
| return 0; | |||
| @@ -3003,6 +3048,7 @@ static void common_init(H264Context *h){ | |||
| init_pred_ptrs(h); | |||
| h->dequant_coeff_pps= -1; | |||
| s->unrestricted_mv=1; | |||
| s->decode=1; //FIXME | |||
| } | |||
| @@ -3349,7 +3395,7 @@ static void hl_decode_mb(H264Context *h){ | |||
| h->pred16x16[ h->intra16x16_pred_mode ](dest_y , linesize); | |||
| if(s->codec_id == CODEC_ID_H264){ | |||
| if(!transform_bypass) | |||
| h264_luma_dc_dequant_idct_c(h->mb, s->qscale); | |||
| h264_luma_dc_dequant_idct_c(h->mb, s->qscale, h->dequant4_coeff[IS_INTRA(mb_type) ? 0:3][s->qscale][0]); | |||
| }else | |||
| svq3_luma_dc_dequant_idct_c(h->mb, s->qscale); | |||
| } | |||
| @@ -3397,8 +3443,8 @@ static void hl_decode_mb(H264Context *h){ | |||
| if(!(s->flags&CODEC_FLAG_GRAY)){ | |||
| idct_add = transform_bypass ? s->dsp.add_pixels4 : s->dsp.h264_idct_add; | |||
| if(!transform_bypass){ | |||
| chroma_dc_dequant_idct_c(h->mb + 16*16, h->chroma_qp); | |||
| chroma_dc_dequant_idct_c(h->mb + 16*16+4*16, h->chroma_qp); | |||
| chroma_dc_dequant_idct_c(h->mb + 16*16, h->chroma_qp, h->dequant4_coeff[IS_INTRA(mb_type) ? 1:4][h->chroma_qp][0]); | |||
| chroma_dc_dequant_idct_c(h->mb + 16*16+4*16, h->chroma_qp, h->dequant4_coeff[IS_INTRA(mb_type) ? 2:5][h->chroma_qp][0]); | |||
| } | |||
| if(s->codec_id == CODEC_ID_H264){ | |||
| for(i=16; i<16+4; i++){ | |||
| @@ -4178,6 +4224,11 @@ static int decode_slice_header(H264Context *h){ | |||
| av_log(h->s.avctx, AV_LOG_ERROR, "non existing SPS referenced\n"); | |||
| return -1; | |||
| } | |||
| if(h->dequant_coeff_pps != pps_id){ | |||
| h->dequant_coeff_pps = pps_id; | |||
| init_dequant_tables(h); | |||
| } | |||
| s->mb_width= h->sps.mb_width; | |||
| s->mb_height= h->sps.mb_height * (2 - h->sps.frame_mbs_only_flag); | |||
| @@ -4434,7 +4485,7 @@ static inline int get_dct8x8_allowed(H264Context *h){ | |||
| * @param max_coeff number of coefficients in the block | |||
| * @return <0 if an error occured | |||
| */ | |||
| static int decode_residual(H264Context *h, GetBitContext *gb, DCTELEM *block, int n, const uint8_t *scantable, const uint16_t *qmul, int max_coeff){ | |||
| static int decode_residual(H264Context *h, GetBitContext *gb, DCTELEM *block, int n, const uint8_t *scantable, const uint32_t *qmul, int max_coeff){ | |||
| MpegEncContext * const s = &h->s; | |||
| static const int coeff_token_table_index[17]= {0, 0, 1, 1, 2, 2, 2, 2, 3, 3, 3, 3, 3, 3, 3, 3, 3}; | |||
| int level[16]; | |||
| @@ -4551,7 +4602,7 @@ static int decode_residual(H264Context *h, GetBitContext *gb, DCTELEM *block, in | |||
| block[j]= level[i]; | |||
| } | |||
| }else{ | |||
| block[j] = level[0] * qmul[j]; | |||
| block[j] = (level[0] * qmul[j] + 32)>>6; | |||
| for(i=1;i<total_coeff;i++) { | |||
| if(zeros_left <= 0) | |||
| run_before = 0; | |||
| @@ -4564,8 +4615,7 @@ static int decode_residual(H264Context *h, GetBitContext *gb, DCTELEM *block, in | |||
| coeff_num -= 1 + run_before; | |||
| j= scantable[ coeff_num ]; | |||
| block[j]= level[i] * qmul[j]; | |||
| // printf("%d %d ", block[j], qmul[j]); | |||
| block[j]= (level[i] * qmul[j] + 32)>>6; | |||
| } | |||
| } | |||
| @@ -5008,7 +5058,7 @@ decode_intra_mb: | |||
| h->chroma_qp= chroma_qp= get_chroma_qp(h->pps.chroma_qp_index_offset, s->qscale); | |||
| if(IS_INTRA16x16(mb_type)){ | |||
| if( decode_residual(h, h->intra_gb_ptr, h->mb, LUMA_DC_BLOCK_INDEX, dc_scan, h->dequant4_coeff[s->qscale], 16) < 0){ | |||
| if( decode_residual(h, h->intra_gb_ptr, h->mb, LUMA_DC_BLOCK_INDEX, dc_scan, h->dequant4_coeff[0][s->qscale], 16) < 0){ | |||
| return -1; //FIXME continue if partitioned and other return -1 too | |||
| } | |||
| @@ -5018,7 +5068,7 @@ decode_intra_mb: | |||
| for(i8x8=0; i8x8<4; i8x8++){ | |||
| for(i4x4=0; i4x4<4; i4x4++){ | |||
| const int index= i4x4 + 4*i8x8; | |||
| if( decode_residual(h, h->intra_gb_ptr, h->mb + 16*index, index, scan + 1, h->dequant4_coeff[s->qscale], 15) < 0 ){ | |||
| if( decode_residual(h, h->intra_gb_ptr, h->mb + 16*index, index, scan + 1, h->dequant4_coeff[0][s->qscale], 15) < 0 ){ | |||
| return -1; | |||
| } | |||
| } | |||
| @@ -5034,21 +5084,16 @@ decode_intra_mb: | |||
| uint8_t *nnz; | |||
| for(i4x4=0; i4x4<4; i4x4++){ | |||
| if( decode_residual(h, gb, buf, i4x4+4*i8x8, zigzag_scan8x8_cavlc+16*i4x4, | |||
| h->dequant8_coeff[s->qscale], 16) <0 ) | |||
| h->dequant8_coeff[IS_INTRA( mb_type ) ? 0:1][s->qscale], 16) <0 ) | |||
| return -1; | |||
| } | |||
| if(s->qscale < 12){ | |||
| int i; | |||
| for(i=0; i<64; i++) | |||
| buf[i] = (buf[i] + 2) >> 2; | |||
| } | |||
| nnz= &h->non_zero_count_cache[ scan8[4*i8x8] ]; | |||
| nnz[0] |= nnz[1] | nnz[8] | nnz[9]; | |||
| }else{ | |||
| for(i4x4=0; i4x4<4; i4x4++){ | |||
| const int index= i4x4 + 4*i8x8; | |||
| if( decode_residual(h, gb, h->mb + 16*index, index, scan, h->dequant4_coeff[s->qscale], 16) <0 ){ | |||
| if( decode_residual(h, gb, h->mb + 16*index, index, scan, h->dequant4_coeff[IS_INTRA( mb_type ) ? 0:3][s->qscale], 16) <0 ){ | |||
| return -1; | |||
| } | |||
| } | |||
| @@ -5062,7 +5107,7 @@ decode_intra_mb: | |||
| if(cbp&0x30){ | |||
| for(chroma_idx=0; chroma_idx<2; chroma_idx++) | |||
| if( decode_residual(h, gb, h->mb + 256 + 16*4*chroma_idx, CHROMA_DC_BLOCK_INDEX, chroma_dc_scan, h->dequant4_coeff[chroma_qp], 4) < 0){ | |||
| if( decode_residual(h, gb, h->mb + 256 + 16*4*chroma_idx, CHROMA_DC_BLOCK_INDEX, chroma_dc_scan, NULL, 4) < 0){ | |||
| return -1; | |||
| } | |||
| } | |||
| @@ -5071,7 +5116,7 @@ decode_intra_mb: | |||
| for(chroma_idx=0; chroma_idx<2; chroma_idx++){ | |||
| for(i4x4=0; i4x4<4; i4x4++){ | |||
| const int index= 16 + 4*chroma_idx + i4x4; | |||
| if( decode_residual(h, gb, h->mb + 16*index, index, scan + 1, h->dequant4_coeff[chroma_qp], 15) < 0){ | |||
| if( decode_residual(h, gb, h->mb + 16*index, index, scan + 1, h->dequant4_coeff[chroma_idx+1+(IS_INTRA( mb_type ) ? 0:3)][chroma_qp], 15) < 0){ | |||
| return -1; | |||
| } | |||
| } | |||
| @@ -5510,7 +5555,7 @@ static int inline get_cabac_cbf_ctx( H264Context *h, int cat, int idx ) { | |||
| return ctx + 4 * cat; | |||
| } | |||
| static int inline decode_cabac_residual( H264Context *h, DCTELEM *block, int cat, int n, const uint8_t *scantable, const uint16_t *qmul, int max_coeff) { | |||
| static int inline decode_cabac_residual( H264Context *h, DCTELEM *block, int cat, int n, const uint8_t *scantable, const uint32_t *qmul, int max_coeff) { | |||
| const int mb_xy = h->s.mb_x + h->s.mb_y*h->s.mb_stride; | |||
| static const int significant_coeff_flag_field_offset[2] = { 105, 277 }; | |||
| static const int last_significant_coeff_flag_field_offset[2] = { 166, 338 }; | |||
| @@ -5616,8 +5661,8 @@ static int inline decode_cabac_residual( H264Context *h, DCTELEM *block, int cat | |||
| if( get_cabac_bypass( &h->cabac ) ) block[j] = -1; | |||
| else block[j] = 1; | |||
| }else{ | |||
| if( get_cabac_bypass( &h->cabac ) ) block[j] = -qmul[j]; | |||
| else block[j] = qmul[j]; | |||
| if( get_cabac_bypass( &h->cabac ) ) block[j] = (-qmul[j] + 32) >> 6; | |||
| else block[j] = ( qmul[j] + 32) >> 6; | |||
| } | |||
| abslevel1++; | |||
| @@ -5645,8 +5690,8 @@ static int inline decode_cabac_residual( H264Context *h, DCTELEM *block, int cat | |||
| if( get_cabac_bypass( &h->cabac ) ) block[j] = -coeff_abs; | |||
| else block[j] = coeff_abs; | |||
| }else{ | |||
| if( get_cabac_bypass( &h->cabac ) ) block[j] = -coeff_abs * qmul[j]; | |||
| else block[j] = coeff_abs * qmul[j]; | |||
| if( get_cabac_bypass( &h->cabac ) ) block[j] = (-coeff_abs * qmul[j] + 32) >> 6; | |||
| else block[j] = ( coeff_abs * qmul[j] + 32) >> 6; | |||
| } | |||
| abslevelgt1++; | |||
| @@ -6078,7 +6123,7 @@ decode_intra_mb: | |||
| if( cbp&15 ) { | |||
| for( i = 0; i < 16; i++ ) { | |||
| //av_log( s->avctx, AV_LOG_ERROR, "INTRA16x16 AC:%d\n", i ); | |||
| if( decode_cabac_residual(h, h->mb + 16*i, 1, i, scan + 1, h->dequant4_coeff[s->qscale], 15) < 0 ) | |||
| if( decode_cabac_residual(h, h->mb + 16*i, 1, i, scan + 1, h->dequant4_coeff[0][s->qscale], 15) < 0 ) | |||
| return -1; | |||
| } | |||
| } else { | |||
| @@ -6090,18 +6135,13 @@ decode_intra_mb: | |||
| if( cbp & (1<<i8x8) ) { | |||
| if( IS_8x8DCT(mb_type) ) { | |||
| if( decode_cabac_residual(h, h->mb + 64*i8x8, 5, 4*i8x8, | |||
| zigzag_scan8x8, h->dequant8_coeff[s->qscale], 64) < 0 ) | |||
| zigzag_scan8x8, h->dequant8_coeff[IS_INTRA( mb_type ) ? 0:1][s->qscale], 64) < 0 ) | |||
| return -1; | |||
| if(s->qscale < 12){ | |||
| int i; | |||
| for(i=0; i<64; i++) | |||
| h->mb[64*i8x8+i] = (h->mb[64*i8x8+i] + 2) >> 2; | |||
| } | |||
| } else | |||
| for( i4x4 = 0; i4x4 < 4; i4x4++ ) { | |||
| const int index = 4*i8x8 + i4x4; | |||
| //av_log( s->avctx, AV_LOG_ERROR, "Luma4x4: %d\n", index ); | |||
| if( decode_cabac_residual(h, h->mb + 16*index, 2, index, scan, h->dequant4_coeff[s->qscale], 16) < 0 ) | |||
| if( decode_cabac_residual(h, h->mb + 16*index, 2, index, scan, h->dequant4_coeff[IS_INTRA( mb_type ) ? 0:3][s->qscale], 16) < 0 ) | |||
| return -1; | |||
| } | |||
| } else { | |||
| @@ -6126,7 +6166,7 @@ decode_intra_mb: | |||
| for( i = 0; i < 4; i++ ) { | |||
| const int index = 16 + 4 * c + i; | |||
| //av_log( s->avctx, AV_LOG_ERROR, "INTRA C%d-AC %d\n",c, index - 16 ); | |||
| if( decode_cabac_residual(h, h->mb + 16*index, 4, index - 16, scan + 1, h->dequant4_coeff[h->chroma_qp], 15) < 0) | |||
| if( decode_cabac_residual(h, h->mb + 16*index, 4, index - 16, scan + 1, h->dequant4_coeff[c+1+(IS_INTRA( mb_type ) ? 0:3)][h->chroma_qp], 15) < 0) | |||
| return -1; | |||
| } | |||
| } | |||
| @@ -7006,6 +7046,52 @@ static inline int decode_vui_parameters(H264Context *h, SPS *sps){ | |||
| return 0; | |||
| } | |||
| static void decode_scaling_list(H264Context *h, uint8_t *factors, int size, const uint8_t *default_list){ | |||
| MpegEncContext * const s = &h->s; | |||
| int i, last = 8, next = 8; | |||
| const uint8_t *scan = size == 16 ? zigzag_scan : zigzag_scan8x8; | |||
| if(!get_bits1(&s->gb)) /* matrix not written, we use the default one */ | |||
| memcpy(factors, default_list, size*sizeof(uint8_t)); | |||
| else | |||
| for(i=0;i<size;i++){ | |||
| if(next) | |||
| next = (last + get_se_golomb(&s->gb)) & 0xff; | |||
| if(!i && !next){ /* matrix not written, we use the default one */ | |||
| memcpy(factors, default_list, size*sizeof(uint8_t)); | |||
| break; | |||
| } | |||
| last = factors[scan[i]] = next ? next : last; | |||
| } | |||
| } | |||
| static void decode_scaling_matrices(H264Context *h, SPS *sps, PPS *pps, int is_sps, | |||
| uint8_t (*scaling_matrix4)[16], uint8_t (*scaling_matrix8)[64]){ | |||
| MpegEncContext * const s = &h->s; | |||
| int fallback_sps = !is_sps && sps->scaling_matrix_present; | |||
| const uint8_t *fallback[4] = { | |||
| fallback_sps ? sps->scaling_matrix4[0] : default_scaling4[0], | |||
| fallback_sps ? sps->scaling_matrix4[3] : default_scaling4[1], | |||
| fallback_sps ? sps->scaling_matrix8[0] : default_scaling8[0], | |||
| fallback_sps ? sps->scaling_matrix8[1] : default_scaling8[1] | |||
| }; | |||
| if(get_bits1(&s->gb)){ | |||
| sps->scaling_matrix_present |= is_sps; | |||
| decode_scaling_list(h,scaling_matrix4[0],16,fallback[0]); // Intra, Y | |||
| decode_scaling_list(h,scaling_matrix4[1],16,scaling_matrix4[0]); // Intra, Cr | |||
| decode_scaling_list(h,scaling_matrix4[2],16,scaling_matrix4[1]); // Intra, Cb | |||
| decode_scaling_list(h,scaling_matrix4[3],16,fallback[1]); // Inter, Y | |||
| decode_scaling_list(h,scaling_matrix4[4],16,scaling_matrix4[3]); // Inter, Cr | |||
| decode_scaling_list(h,scaling_matrix4[5],16,scaling_matrix4[4]); // Inter, Cb | |||
| if(is_sps || pps->transform_8x8_mode){ | |||
| decode_scaling_list(h,scaling_matrix8[0],64,fallback[2]); // Intra, Y | |||
| decode_scaling_list(h,scaling_matrix8[1],64,fallback[3]); // Inter, Y | |||
| } | |||
| } else if(fallback_sps) { | |||
| memcpy(scaling_matrix4, sps->scaling_matrix4, 6*16*sizeof(uint8_t)); | |||
| memcpy(scaling_matrix8, sps->scaling_matrix8, 2*64*sizeof(uint8_t)); | |||
| } | |||
| } | |||
| static inline int decode_seq_parameter_set(H264Context *h){ | |||
| MpegEncContext * const s = &h->s; | |||
| int profile_idc, level_idc; | |||
| @@ -7031,11 +7117,9 @@ static inline int decode_seq_parameter_set(H264Context *h){ | |||
| get_ue_golomb(&s->gb); //bit_depth_luma_minus8 | |||
| get_ue_golomb(&s->gb); //bit_depth_chroma_minus8 | |||
| sps->transform_bypass = get_bits1(&s->gb); | |||
| if(get_bits1(&s->gb)){ //seq_scaling_matrix_present_flag | |||
| av_log(h->s.avctx, AV_LOG_ERROR, "custom scaling matrix not implemented\n"); | |||
| return -1; | |||
| } | |||
| } | |||
| decode_scaling_matrices(h, sps, NULL, 1, sps->scaling_matrix4, sps->scaling_matrix8); | |||
| }else | |||
| sps->scaling_matrix_present = 0; | |||
| sps->log2_max_frame_num= get_ue_golomb(&s->gb) + 4; | |||
| sps->poc_type= get_ue_golomb(&s->gb); | |||
| @@ -7172,13 +7256,13 @@ static inline int decode_picture_parameter_set(H264Context *h, int bit_length){ | |||
| pps->deblocking_filter_parameters_present= get_bits1(&s->gb); | |||
| pps->constrained_intra_pred= get_bits1(&s->gb); | |||
| pps->redundant_pic_cnt_present = get_bits1(&s->gb); | |||
| memset(pps->scaling_matrix4, 16, 6*16*sizeof(uint8_t)); | |||
| memset(pps->scaling_matrix8, 16, 2*64*sizeof(uint8_t)); | |||
| if(get_bits_count(&s->gb) < bit_length){ | |||
| pps->transform_8x8_mode= get_bits1(&s->gb); | |||
| if(get_bits1(&s->gb)){ //pic_scaling_matrix_present_flag | |||
| av_log(h->s.avctx, AV_LOG_ERROR, "custom scaling matrix not implemented\n"); | |||
| return -1; | |||
| } | |||
| decode_scaling_matrices(h, &h->sps_buffer[pps->sps_id], pps, 0, pps->scaling_matrix4, pps->scaling_matrix8); | |||
| get_se_golomb(&s->gb); //second_chroma_qp_index_offset | |||
| } | |||
| @@ -457,61 +457,45 @@ static const uint8_t div6[52]={ | |||
| 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 3, 3, 3, 3, 3, 3, 4, 4, 4, 4, 4, 4, 5, 5, 5, 5, 5, 5, 6, 6, 6, 6, 6, 6, 7, 7, 7, 7, 7, 7, 8, 8, 8, 8, | |||
| }; | |||
| static const uint16_t dequant_coeff[52][16]={ | |||
| { 10, 13, 10, 13, 13, 16, 13, 16, 10, 13, 10, 13, 13, 16, 13, 16, }, | |||
| { 11, 14, 11, 14, 14, 18, 14, 18, 11, 14, 11, 14, 14, 18, 14, 18, }, | |||
| { 13, 16, 13, 16, 16, 20, 16, 20, 13, 16, 13, 16, 16, 20, 16, 20, }, | |||
| { 14, 18, 14, 18, 18, 23, 18, 23, 14, 18, 14, 18, 18, 23, 18, 23, }, | |||
| { 16, 20, 16, 20, 20, 25, 20, 25, 16, 20, 16, 20, 20, 25, 20, 25, }, | |||
| { 18, 23, 18, 23, 23, 29, 23, 29, 18, 23, 18, 23, 23, 29, 23, 29, }, | |||
| { 20, 26, 20, 26, 26, 32, 26, 32, 20, 26, 20, 26, 26, 32, 26, 32, }, | |||
| { 22, 28, 22, 28, 28, 36, 28, 36, 22, 28, 22, 28, 28, 36, 28, 36, }, | |||
| { 26, 32, 26, 32, 32, 40, 32, 40, 26, 32, 26, 32, 32, 40, 32, 40, }, | |||
| { 28, 36, 28, 36, 36, 46, 36, 46, 28, 36, 28, 36, 36, 46, 36, 46, }, | |||
| { 32, 40, 32, 40, 40, 50, 40, 50, 32, 40, 32, 40, 40, 50, 40, 50, }, | |||
| { 36, 46, 36, 46, 46, 58, 46, 58, 36, 46, 36, 46, 46, 58, 46, 58, }, | |||
| { 40, 52, 40, 52, 52, 64, 52, 64, 40, 52, 40, 52, 52, 64, 52, 64, }, | |||
| { 44, 56, 44, 56, 56, 72, 56, 72, 44, 56, 44, 56, 56, 72, 56, 72, }, | |||
| { 52, 64, 52, 64, 64, 80, 64, 80, 52, 64, 52, 64, 64, 80, 64, 80, }, | |||
| { 56, 72, 56, 72, 72, 92, 72, 92, 56, 72, 56, 72, 72, 92, 72, 92, }, | |||
| { 64, 80, 64, 80, 80, 100, 80, 100, 64, 80, 64, 80, 80, 100, 80, 100, }, | |||
| { 72, 92, 72, 92, 92, 116, 92, 116, 72, 92, 72, 92, 92, 116, 92, 116, }, | |||
| { 80, 104, 80, 104, 104, 128, 104, 128, 80, 104, 80, 104, 104, 128, 104, 128, }, | |||
| { 88, 112, 88, 112, 112, 144, 112, 144, 88, 112, 88, 112, 112, 144, 112, 144, }, | |||
| { 104, 128, 104, 128, 128, 160, 128, 160, 104, 128, 104, 128, 128, 160, 128, 160, }, | |||
| { 112, 144, 112, 144, 144, 184, 144, 184, 112, 144, 112, 144, 144, 184, 144, 184, }, | |||
| { 128, 160, 128, 160, 160, 200, 160, 200, 128, 160, 128, 160, 160, 200, 160, 200, }, | |||
| { 144, 184, 144, 184, 184, 232, 184, 232, 144, 184, 144, 184, 184, 232, 184, 232, }, | |||
| { 160, 208, 160, 208, 208, 256, 208, 256, 160, 208, 160, 208, 208, 256, 208, 256, }, | |||
| { 176, 224, 176, 224, 224, 288, 224, 288, 176, 224, 176, 224, 224, 288, 224, 288, }, | |||
| { 208, 256, 208, 256, 256, 320, 256, 320, 208, 256, 208, 256, 256, 320, 256, 320, }, | |||
| { 224, 288, 224, 288, 288, 368, 288, 368, 224, 288, 224, 288, 288, 368, 288, 368, }, | |||
| { 256, 320, 256, 320, 320, 400, 320, 400, 256, 320, 256, 320, 320, 400, 320, 400, }, | |||
| { 288, 368, 288, 368, 368, 464, 368, 464, 288, 368, 288, 368, 368, 464, 368, 464, }, | |||
| { 320, 416, 320, 416, 416, 512, 416, 512, 320, 416, 320, 416, 416, 512, 416, 512, }, | |||
| { 352, 448, 352, 448, 448, 576, 448, 576, 352, 448, 352, 448, 448, 576, 448, 576, }, | |||
| { 416, 512, 416, 512, 512, 640, 512, 640, 416, 512, 416, 512, 512, 640, 512, 640, }, | |||
| { 448, 576, 448, 576, 576, 736, 576, 736, 448, 576, 448, 576, 576, 736, 576, 736, }, | |||
| { 512, 640, 512, 640, 640, 800, 640, 800, 512, 640, 512, 640, 640, 800, 640, 800, }, | |||
| { 576, 736, 576, 736, 736, 928, 736, 928, 576, 736, 576, 736, 736, 928, 736, 928, }, | |||
| { 640, 832, 640, 832, 832,1024, 832,1024, 640, 832, 640, 832, 832,1024, 832,1024, }, | |||
| { 704, 896, 704, 896, 896,1152, 896,1152, 704, 896, 704, 896, 896,1152, 896,1152, }, | |||
| { 832,1024, 832,1024, 1024,1280,1024,1280, 832,1024, 832,1024, 1024,1280,1024,1280, }, | |||
| { 896,1152, 896,1152, 1152,1472,1152,1472, 896,1152, 896,1152, 1152,1472,1152,1472, }, | |||
| {1024,1280,1024,1280, 1280,1600,1280,1600, 1024,1280,1024,1280, 1280,1600,1280,1600, }, | |||
| {1152,1472,1152,1472, 1472,1856,1472,1856, 1152,1472,1152,1472, 1472,1856,1472,1856, }, | |||
| {1280,1664,1280,1664, 1664,2048,1664,2048, 1280,1664,1280,1664, 1664,2048,1664,2048, }, | |||
| {1408,1792,1408,1792, 1792,2304,1792,2304, 1408,1792,1408,1792, 1792,2304,1792,2304, }, | |||
| {1664,2048,1664,2048, 2048,2560,2048,2560, 1664,2048,1664,2048, 2048,2560,2048,2560, }, | |||
| {1792,2304,1792,2304, 2304,2944,2304,2944, 1792,2304,1792,2304, 2304,2944,2304,2944, }, | |||
| {2048,2560,2048,2560, 2560,3200,2560,3200, 2048,2560,2048,2560, 2560,3200,2560,3200, }, | |||
| {2304,2944,2304,2944, 2944,3712,2944,3712, 2304,2944,2304,2944, 2944,3712,2944,3712, }, | |||
| {2560,3328,2560,3328, 3328,4096,3328,4096, 2560,3328,2560,3328, 3328,4096,3328,4096, }, | |||
| {2816,3584,2816,3584, 3584,4608,3584,4608, 2816,3584,2816,3584, 3584,4608,3584,4608, }, | |||
| {3328,4096,3328,4096, 4096,5120,4096,5120, 3328,4096,3328,4096, 4096,5120,4096,5120, }, | |||
| {3584,4608,3584,4608, 4608,5888,4608,5888, 3584,4608,3584,4608, 4608,5888,4608,5888, }, | |||
| //{4096,5120,4096,5120, 5120,6400,5120,6400, 4096,5120,4096,5120, 5120,6400,5120,6400, }, | |||
| //{4608,5888,4608,5888, 5888,7424,5888,7424, 4608,5888,4608,5888, 5888,7424,5888,7424, }, | |||
| static const uint8_t default_scaling4[2][16]={ | |||
| { 6,13,20,28, | |||
| 13,20,28,32, | |||
| 20,28,32,37, | |||
| 28,32,37,42 | |||
| },{ | |||
| 10,14,20,24, | |||
| 14,20,24,27, | |||
| 20,24,27,30, | |||
| 24,27,30,34 | |||
| }}; | |||
| static const uint8_t default_scaling8[2][64]={ | |||
| { 6,10,13,16,18,23,25,27, | |||
| 10,11,16,18,23,25,27,29, | |||
| 13,16,18,23,25,27,29,31, | |||
| 16,18,23,25,27,29,31,33, | |||
| 18,23,25,27,29,31,33,36, | |||
| 23,25,27,29,31,33,36,38, | |||
| 25,27,29,31,33,36,38,40, | |||
| 27,29,31,33,36,38,40,42 | |||
| },{ | |||
| 9,13,15,17,19,21,22,24, | |||
| 13,13,17,19,21,22,24,25, | |||
| 15,17,19,21,22,24,25,27, | |||
| 17,19,21,22,24,25,27,28, | |||
| 19,21,22,24,25,27,28,30, | |||
| 21,22,24,25,27,28,30,32, | |||
| 22,24,25,27,28,30,32,33, | |||
| 24,25,27,28,30,32,33,35 | |||
| }}; | |||
| static const int dequant4_coeff_init[6][3]={ | |||
| {10,13,16}, | |||
| {11,14,18}, | |||
| {13,16,20}, | |||
| {14,18,23}, | |||
| {16,20,25}, | |||
| {18,23,29}, | |||
| }; | |||
| static const int dequant8_coeff_init_scan[16] = { | |||