You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
340 lines
15 KiB
340 lines
15 KiB
4 months ago
|
/******************************************************************************
|
||
|
|
*
|
||
|
|
* Copyright (C) 2015 The Android Open Source Project
|
||
|
|
*
|
||
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
* you may not use this file except in compliance with the License.
|
||
|
|
* You may obtain a copy of the License at:
|
||
|
|
*
|
||
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
*
|
||
|
|
* Unless required by applicable law or agreed to in writing, software
|
||
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
* See the License for the specific language governing permissions and
|
||
|
|
* limitations under the License.
|
||
|
|
*
|
||
|
|
*****************************************************************************
|
||
|
|
* Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore
|
||
|
|
*/
|
||
|
|
/*****************************************************************************/
|
||
|
|
/* */
|
||
|
|
/* File Name : mpeg2dec_api_utils.c */
|
||
|
|
/* */
|
||
|
|
/* */
|
||
|
|
/* Description : This file defines the API interface for MPEG2 Decoder*/
|
||
|
|
/* */
|
||
|
|
/* List of Functions : <List the functions defined in this file> */
|
||
|
|
/* */
|
||
|
|
/* Issues / Problems : None */
|
||
|
|
/* */
|
||
|
|
/* Revision History : */
|
||
|
|
/* */
|
||
|
|
/* DD MM YYYY Author(s) Changes (Describe the changes made) */
|
||
|
|
/* 17 09 2007 Rajendra C Y Creation */
|
||
|
|
/* */
|
||
|
|
/*****************************************************************************/
|
||
|
|
|
||
|
|
/*****************************************************************************/
|
||
|
|
/* File Includes */
|
||
|
|
/*****************************************************************************/
|
||
|
|
/* System include files */
|
||
|
|
|
||
|
|
#include <stddef.h>
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <string.h>
|
||
|
|
|
||
|
|
/* User include files */
|
||
|
|
#include "iv_datatypedef.h"
|
||
|
|
#include "iv.h"
|
||
|
|
#include "ivd.h"
|
||
|
|
#include "ithread.h"
|
||
|
|
|
||
|
|
#include "impeg2_job_queue.h"
|
||
|
|
#include "impeg2_macros.h"
|
||
|
|
#include "impeg2_buf_mgr.h"
|
||
|
|
#include "impeg2_disp_mgr.h"
|
||
|
|
#include "impeg2_defs.h"
|
||
|
|
#include "impeg2_platform_macros.h"
|
||
|
|
#include "impeg2_inter_pred.h"
|
||
|
|
#include "impeg2_idct.h"
|
||
|
|
#include "impeg2_format_conv.h"
|
||
|
|
#include "impeg2_mem_func.h"
|
||
|
|
|
||
|
|
#include "impeg2d.h"
|
||
|
|
#include "impeg2d_api.h"
|
||
|
|
#include "impeg2d_bitstream.h"
|
||
|
|
#include "impeg2d_debug.h"
|
||
|
|
#include "impeg2d_structs.h"
|
||
|
|
#include "impeg2d_mc.h"
|
||
|
|
#include "impeg2d_pic_proc.h"
|
||
|
|
#include "impeg2d_dec_hdr.h"
|
||
|
|
|
||
|
|
void impeg2d_next_start_code(dec_state_t *ps_dec);
|
||
|
|
void impeg2d_next_code(dec_state_t *ps_dec, UWORD32 u4_start_code_val);
|
||
|
|
|
||
|
|
/*****************************************************************************/
|
||
|
|
/* */
|
||
|
|
/* Function Name : impeg2d_dec_hdr */
|
||
|
|
/* */
|
||
|
|
/* Description : */
|
||
|
|
/* Inputs : */
|
||
|
|
/* Globals : */
|
||
|
|
/* Processing : */
|
||
|
|
/* Outputs : */
|
||
|
|
/* Returns : */
|
||
|
|
/* */
|
||
|
|
/* Issues : */
|
||
|
|
/* */
|
||
|
|
/* Revision History: */
|
||
|
|
/* */
|
||
|
|
/* DD MM YYYY Author(s) Changes (Describe the changes made) */
|
||
|
|
/* 17 09 2007 Rajendra C Y Draft */
|
||
|
|
/* */
|
||
|
|
/*****************************************************************************/
|
||
|
|
void impeg2d_dec_hdr(void *pv_dec,impeg2d_video_decode_ip_t *ps_ip,
|
||
|
|
impeg2d_video_decode_op_t *ps_op)
|
||
|
|
{
|
||
|
|
|
||
|
|
UWORD32 u4_bits_read;
|
||
|
|
dec_state_t *ps_dec;
|
||
|
|
UWORD32 u4_size = ps_ip->s_ivd_video_decode_ip_t.u4_num_Bytes;
|
||
|
|
|
||
|
|
ps_dec = (dec_state_t *)pv_dec;
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_error_code = 0;
|
||
|
|
if (u4_size > MAX_BITSTREAM_BUFFER_SIZE)
|
||
|
|
{
|
||
|
|
u4_size = MAX_BITSTREAM_BUFFER_SIZE;
|
||
|
|
}
|
||
|
|
|
||
|
|
memcpy(ps_dec->pu1_input_buffer, ps_ip->s_ivd_video_decode_ip_t.pv_stream_buffer, u4_size);
|
||
|
|
|
||
|
|
impeg2d_bit_stream_init(&(ps_dec->s_bit_stream), ps_dec->pu1_input_buffer,
|
||
|
|
u4_size);
|
||
|
|
|
||
|
|
{
|
||
|
|
{
|
||
|
|
IMPEG2D_ERROR_CODES_T e_error;
|
||
|
|
e_error = impeg2d_process_video_header(ps_dec);
|
||
|
|
if ((IMPEG2D_ERROR_CODES_T)IVD_ERROR_NONE != e_error)
|
||
|
|
{
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_error_code = e_error;
|
||
|
|
|
||
|
|
u4_bits_read = impeg2d_bit_stream_num_bits_read(&ps_dec->s_bit_stream);
|
||
|
|
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_num_bytes_consumed = u4_bits_read>> 3;
|
||
|
|
if(ps_op->s_ivd_video_decode_op_t.u4_num_bytes_consumed > ps_ip->s_ivd_video_decode_ip_t.u4_num_Bytes)
|
||
|
|
{
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_num_bytes_consumed = ps_ip->s_ivd_video_decode_ip_t.u4_num_Bytes;
|
||
|
|
}
|
||
|
|
if(ps_op->s_ivd_video_decode_op_t.u4_error_code == 0)
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_error_code = e_error;
|
||
|
|
|
||
|
|
if (IMPEG2D_UNSUPPORTED_DIMENSIONS == e_error)
|
||
|
|
{
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_num_bytes_consumed = 0;
|
||
|
|
ps_dec->u2_header_done = 0;
|
||
|
|
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_pic_ht = ps_dec->u2_reinit_max_height;
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_pic_wd = ps_dec->u2_reinit_max_width;
|
||
|
|
}
|
||
|
|
impeg2d_next_code(ps_dec, SEQUENCE_HEADER_CODE);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_pic_ht = ps_dec->u2_vertical_size;
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_pic_wd = ps_dec->u2_horizontal_size;
|
||
|
|
|
||
|
|
ps_op->s_ivd_video_decode_op_t.e_pic_type = IV_NA_FRAME;
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_error_code = IV_SUCCESS;
|
||
|
|
|
||
|
|
u4_bits_read = impeg2d_bit_stream_num_bits_read(&ps_dec->s_bit_stream);
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_num_bytes_consumed = u4_bits_read>> 3;
|
||
|
|
if(ps_op->s_ivd_video_decode_op_t.u4_num_bytes_consumed > ps_ip->s_ivd_video_decode_ip_t.u4_num_Bytes)
|
||
|
|
{
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_num_bytes_consumed = ps_ip->s_ivd_video_decode_ip_t.u4_num_Bytes;
|
||
|
|
}
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_frame_decoded_flag = 0;
|
||
|
|
|
||
|
|
/* Set the stride */
|
||
|
|
if (0 == ps_dec->u4_frm_buf_stride)
|
||
|
|
{
|
||
|
|
ps_dec->u4_frm_buf_stride = ps_dec->u2_horizontal_size;
|
||
|
|
}
|
||
|
|
/* MOD */
|
||
|
|
ps_dec->u2_header_done = 1;
|
||
|
|
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/*****************************************************************************/
|
||
|
|
/* */
|
||
|
|
/* Function Name : impeg2d_dec_frm */
|
||
|
|
/* */
|
||
|
|
/* Description : */
|
||
|
|
/* Inputs : */
|
||
|
|
/* Globals : */
|
||
|
|
/* Processing : */
|
||
|
|
/* Outputs : */
|
||
|
|
/* Returns : */
|
||
|
|
/* */
|
||
|
|
/* Issues : */
|
||
|
|
/* */
|
||
|
|
/* Revision History: */
|
||
|
|
/* */
|
||
|
|
/* DD MM YYYY Author(s) Changes (Describe the changes made) */
|
||
|
|
/* 17 09 2007 Rajendra C Y Draft */
|
||
|
|
/* */
|
||
|
|
/*****************************************************************************/
|
||
|
|
void impeg2d_dec_frm(void *pv_dec,impeg2d_video_decode_ip_t *ps_ip,
|
||
|
|
impeg2d_video_decode_op_t *ps_op)
|
||
|
|
{
|
||
|
|
|
||
|
|
|
||
|
|
stream_t *ps_stream;
|
||
|
|
UWORD32 u4_size = ps_ip->s_ivd_video_decode_ip_t.u4_num_Bytes;
|
||
|
|
|
||
|
|
dec_state_t *ps_dec;
|
||
|
|
|
||
|
|
ps_dec = (dec_state_t *)pv_dec;
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_error_code = 0;
|
||
|
|
ps_dec->i4_bytes_consumed = 0;
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_num_bytes_consumed = 0;
|
||
|
|
|
||
|
|
IMPEG2D_FRM_NUM_SET();
|
||
|
|
if (u4_size > MAX_BITSTREAM_BUFFER_SIZE)
|
||
|
|
{
|
||
|
|
u4_size = MAX_BITSTREAM_BUFFER_SIZE;
|
||
|
|
}
|
||
|
|
|
||
|
|
memcpy(ps_dec->pu1_input_buffer, ps_ip->s_ivd_video_decode_ip_t.pv_stream_buffer, u4_size);
|
||
|
|
|
||
|
|
ps_dec->pu1_inp_bits_buf = ps_dec->pu1_input_buffer;
|
||
|
|
|
||
|
|
ps_dec->u4_num_inp_bytes = u4_size;
|
||
|
|
ps_stream = &ps_dec->s_bit_stream;
|
||
|
|
|
||
|
|
impeg2d_bit_stream_init(ps_stream, ps_dec->pu1_input_buffer, u4_size);
|
||
|
|
|
||
|
|
/* @ */ /* Updating the bufferID */
|
||
|
|
|
||
|
|
ps_dec->u4_xdmBufID = ps_ip->s_ivd_video_decode_ip_t.u4_ts;
|
||
|
|
|
||
|
|
{
|
||
|
|
IMPEG2D_ERROR_CODES_T e_error;
|
||
|
|
/* Process the Bitstream */
|
||
|
|
e_error = impeg2d_process_video_bit_stream(ps_dec);
|
||
|
|
if ((IMPEG2D_ERROR_CODES_T)IVD_ERROR_NONE != e_error)
|
||
|
|
{
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_error_code = e_error;
|
||
|
|
|
||
|
|
if ((IMPEG2D_ERROR_CODES_T) IVD_RES_CHANGED == e_error)
|
||
|
|
{
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_num_bytes_consumed = ps_dec->i4_bytes_consumed;
|
||
|
|
ps_dec->u2_header_done = 0;
|
||
|
|
}
|
||
|
|
else if (IMPEG2D_UNSUPPORTED_DIMENSIONS == e_error)
|
||
|
|
{
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_num_bytes_consumed = 0;
|
||
|
|
ps_dec->u2_header_done = 0;
|
||
|
|
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_pic_ht = ps_dec->u2_reinit_max_height;
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_pic_wd = ps_dec->u2_reinit_max_width;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
if(ps_dec->i4_num_cores > 1 && 0 != ps_dec->i4_bytes_consumed)
|
||
|
|
{
|
||
|
|
/* If the number of bytes consumed has been updated by
|
||
|
|
* get_slice_pos function, then use that. Else, the bytes consumed is
|
||
|
|
* calculated from the offset. The bytes consumed for multi-thread runs
|
||
|
|
* is updated only into ps_dec->i4_bytes_consumed if the get_slice_pos
|
||
|
|
* function has been called. If that function has not run, then we have
|
||
|
|
* encountered an error but still have to consume the bytes in header
|
||
|
|
* decode, etc.
|
||
|
|
*/
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_num_bytes_consumed = ps_dec->i4_bytes_consumed;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_num_bytes_consumed = (ps_dec->s_bit_stream.u4_offset + 7) >> 3;
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_num_bytes_consumed -= ((size_t)ps_dec->s_bit_stream.pv_bs_buf & 3);
|
||
|
|
}
|
||
|
|
|
||
|
|
if(ps_op->s_ivd_video_decode_op_t.u4_num_bytes_consumed
|
||
|
|
> ps_ip->s_ivd_video_decode_ip_t.u4_num_Bytes)
|
||
|
|
{
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_num_bytes_consumed =
|
||
|
|
ps_ip->s_ivd_video_decode_ip_t.u4_num_Bytes;
|
||
|
|
}
|
||
|
|
|
||
|
|
impeg2d_next_start_code(ps_dec);
|
||
|
|
}
|
||
|
|
|
||
|
|
if(ps_op->s_ivd_video_decode_op_t.u4_error_code == 0)
|
||
|
|
{
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_error_code = e_error;
|
||
|
|
}
|
||
|
|
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
/**************************************************************************/
|
||
|
|
/* Remove the bytes left till next start code is encountered */
|
||
|
|
/**************************************************************************/
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_error_code = IV_SUCCESS;
|
||
|
|
|
||
|
|
if(ps_dec->i4_num_cores > 1 && 0 != ps_dec->i4_bytes_consumed)
|
||
|
|
{
|
||
|
|
/* If the number of bytes consumed has been updated by
|
||
|
|
* get_slice_pos function, then use that. Else, the bytes consumed is
|
||
|
|
* calculated from the offset. The bytes consumed for multi-thread runs
|
||
|
|
* is updated only into ps_dec->i4_bytes_consumed if the get_slice_pos
|
||
|
|
* function has been called. If that function has not run, then we have
|
||
|
|
* encountered an error but still have to consume the bytes in header
|
||
|
|
* decode, etc.
|
||
|
|
*/
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_num_bytes_consumed = ps_dec->i4_bytes_consumed;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_num_bytes_consumed = (ps_dec->s_bit_stream.u4_offset + 7) >> 3;
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_num_bytes_consumed -= ((size_t)ps_dec->s_bit_stream.pv_bs_buf & 3);
|
||
|
|
}
|
||
|
|
if(ps_op->s_ivd_video_decode_op_t.u4_num_bytes_consumed > ps_ip->s_ivd_video_decode_ip_t.u4_num_Bytes)
|
||
|
|
{
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_num_bytes_consumed = ps_ip->s_ivd_video_decode_ip_t.u4_num_Bytes;
|
||
|
|
}
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_pic_ht = ps_dec->u2_vertical_size;
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_pic_wd = ps_dec->u2_horizontal_size;
|
||
|
|
|
||
|
|
switch(ps_dec->e_pic_type)
|
||
|
|
{
|
||
|
|
case I_PIC :
|
||
|
|
ps_op->s_ivd_video_decode_op_t.e_pic_type = IV_I_FRAME;
|
||
|
|
break;
|
||
|
|
|
||
|
|
case P_PIC:
|
||
|
|
ps_op->s_ivd_video_decode_op_t.e_pic_type = IV_P_FRAME;
|
||
|
|
break;
|
||
|
|
|
||
|
|
case B_PIC:
|
||
|
|
ps_op->s_ivd_video_decode_op_t.e_pic_type = IV_B_FRAME;
|
||
|
|
break;
|
||
|
|
|
||
|
|
case D_PIC:
|
||
|
|
ps_op->s_ivd_video_decode_op_t.e_pic_type = IV_I_FRAME;
|
||
|
|
break;
|
||
|
|
|
||
|
|
default :
|
||
|
|
ps_op->s_ivd_video_decode_op_t.e_pic_type = IV_FRAMETYPE_DEFAULT;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_frame_decoded_flag = ps_dec->i4_frame_decoded;
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_new_seq = 0;
|
||
|
|
ps_op->s_ivd_video_decode_op_t.u4_error_code = ps_dec->u4_error_code;
|
||
|
|
|
||
|
|
|
||
|
|
}
|