18d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S/****************************************************************************** 28d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * 38d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * Copyright (C) 2015 The Android Open Source Project 48d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * 58d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * Licensed under the Apache License, Version 2.0 (the "License"); 68d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * you may not use this file except in compliance with the License. 78d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * You may obtain a copy of the License at: 88d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * 98d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * http://www.apache.org/licenses/LICENSE-2.0 108d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * 118d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * Unless required by applicable law or agreed to in writing, software 128d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * distributed under the License is distributed on an "AS IS" BASIS, 138d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 148d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * See the License for the specific language governing permissions and 158d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * limitations under the License. 168d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * 178d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ***************************************************************************** 188d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore 198d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S*/ 208d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 218d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S/** 228d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S******************************************************************************* 238d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @file 248d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* ih264e_process.c 258d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 268d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @brief 278d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* Contains functions for codec thread 288d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 298d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @author 308d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* Harish 318d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 328d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @par List of Functions: 338d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* - ih264e_generate_sps_pps() 348d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* - ih264e_init_entropy_ctxt() 358d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* - ih264e_entropy() 368d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* - ih264e_pack_header_data() 378d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* - ih264e_update_proc_ctxt() 388d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* - ih264e_init_proc_ctxt() 398d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* - ih264e_pad_recon_buffer() 408d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* - ih264e_dblk_pad_hpel_processing_n_mbs() 418d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* - ih264e_process() 428d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* - ih264e_set_rc_pic_params() 438d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* - ih264e_update_rc_post_enc() 448d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* - ih264e_process_thread() 458d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 468d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @remarks 478d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* None 488d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 498d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S******************************************************************************* 508d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S*/ 518d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 528d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S/*****************************************************************************/ 538d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S/* File Includes */ 548d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S/*****************************************************************************/ 558d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 568d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S/* System include files */ 578d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include <stdio.h> 588d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include <stddef.h> 598d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include <stdlib.h> 608d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include <string.h> 618d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include <limits.h> 628d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include <assert.h> 638d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 648d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S/* User include files */ 658d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264_typedefs.h" 668d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "iv2.h" 678d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ive2.h" 688d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264_defs.h" 698d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264_debug.h" 708d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ime_distortion_metrics.h" 713749f6f435e79624f72841e866245d84195551cdHarinarayanan K K#include "ime_defs.h" 728d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ime_structs.h" 738d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264_error.h" 748d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264_structs.h" 758d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264_trans_quant_itrans_iquant.h" 768d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264_inter_pred_filters.h" 778d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264_mem_fns.h" 788d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264_padding.h" 798d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264_intra_pred_filters.h" 808d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264_deblk_edge_filters.h" 813749f6f435e79624f72841e866245d84195551cdHarinarayanan K K#include "ih264_cabac_tables.h" 828d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264_platform_macros.h" 838d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264_macros.h" 848d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264_buf_mgr.h" 858d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264e_error.h" 868d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264e_bitstream.h" 878d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264_common_tables.h" 888d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264_list.h" 898d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264e_defs.h" 908d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "irc_cntrl_param.h" 918d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "irc_frame_info_collector.h" 928d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264e_rate_control.h" 933749f6f435e79624f72841e866245d84195551cdHarinarayanan K K#include "ih264e_cabac_structs.h" 948d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264e_structs.h" 953749f6f435e79624f72841e866245d84195551cdHarinarayanan K K#include "ih264e_cabac.h" 968d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264e_process.h" 978d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ithread.h" 988d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264e_intra_modes_eval.h" 998d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264e_encode_header.h" 1008d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264e_globals.h" 1018d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264e_config.h" 1028d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264e_trace.h" 1038d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264e_statistics.h" 1048d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264_cavlc_tables.h" 1058d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264e_cavlc.h" 1068d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264e_deblk.h" 1078d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264e_me.h" 1088d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264e_debug.h" 1098d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264e_master.h" 1108d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264e_utils.h" 1118d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "irc_mem_req_and_acq.h" 1128d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "irc_rate_control_api.h" 1138d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ih264e_platform_macros.h" 1148d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#include "ime_statistics.h" 1158d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 1168d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 1178d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S/*****************************************************************************/ 1188d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S/* Function Definitions */ 1198d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S/*****************************************************************************/ 1208d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 1218d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S/** 1228d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S****************************************************************************** 1238d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 1248d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @brief This function generates sps, pps set on request 1258d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 1268d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @par Description 1278d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* When the encoder is set in header generation mode, the following function 1288d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* is called. This generates sps and pps headers and returns the control back 1298d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* to caller. 1308d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 1318d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @param[in] ps_codec 1328d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* pointer to codec context 1338d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 1348d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @return success or failure error code 1358d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 1368d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S****************************************************************************** 1378d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S*/ 1388d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha SIH264E_ERROR_T ih264e_generate_sps_pps(codec_t *ps_codec) 1398d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S{ 1408d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* choose between ping-pong process buffer set */ 141d3e18f0c8854c258021be1e603d8f50a241698a7Harinarayanan K K WORD32 ctxt_sel = ps_codec->i4_encode_api_call_cnt % MAX_CTXT_SETS; 1428d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 1438d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* entropy ctxt */ 1448d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S entropy_ctxt_t *ps_entropy = &ps_codec->as_process[ctxt_sel * MAX_PROCESS_THREADS].s_entropy; 1458d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 1468d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* Bitstream structure */ 1478d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S bitstrm_t *ps_bitstrm = ps_entropy->ps_bitstrm; 1488d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 1498d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* sps */ 1508d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S sps_t *ps_sps = NULL; 1518d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 1528d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* pps */ 1538d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pps_t *ps_pps = NULL; 1548d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 1558d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* output buff */ 1568d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S out_buf_t *ps_out_buf = &ps_codec->as_out_buf[ctxt_sel]; 1578d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 1588d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 1598d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /********************************************************************/ 1608d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* initialize the bit stream buffer */ 1618d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /********************************************************************/ 1628d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ih264e_bitstrm_init(ps_bitstrm, ps_out_buf->s_bits_buf.pv_buf, ps_out_buf->s_bits_buf.u4_bufsize); 1638d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 1648d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /********************************************************************/ 1658d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* BEGIN HEADER GENERATION */ 1668d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /********************************************************************/ 1678d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /*ps_codec->i4_pps_id ++;*/ 1688d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->i4_pps_id %= MAX_PPS_CNT; 1698d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 1708d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /*ps_codec->i4_sps_id ++;*/ 1718d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->i4_sps_id %= MAX_SPS_CNT; 1728d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 1738d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* populate sps header */ 1748d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_sps = ps_codec->ps_sps_base + ps_codec->i4_sps_id; 1758d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ih264e_populate_sps(ps_codec, ps_sps); 1768d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 1778d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* populate pps header */ 1788d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_pps = ps_codec->ps_pps_base + ps_codec->i4_pps_id; 1798d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ih264e_populate_pps(ps_codec, ps_pps); 1808d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 1818d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_entropy->i4_error_code = IH264E_SUCCESS; 1828d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 1838d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* generate sps */ 184126d84e0d3b6f3cb3a6e89f85da070e39ac51219Hamsalekha S ps_entropy->i4_error_code |= ih264e_generate_sps(ps_bitstrm, ps_sps, &ps_codec->s_vui); 1858d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 1868d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* generate pps */ 1878d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_entropy->i4_error_code |= ih264e_generate_pps(ps_bitstrm, ps_pps, ps_sps); 1888d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 1898d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* queue output buffer */ 1908d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_out_buf->s_bits_buf.u4_bytes = ps_bitstrm->u4_strm_buf_offset; 1918d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 1928d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S return ps_entropy->i4_error_code; 1938d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S} 1948d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 1958d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S/** 1968d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S******************************************************************************* 1978d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 1988d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @brief initialize entropy context. 1998d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 2008d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @par Description: 2018d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* Before invoking the call to perform to entropy coding the entropy context 2028d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* associated with the job needs to be initialized. This involves the start 2038d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* mb address, end mb address, slice index and the pointer to location at 2048d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* which the mb residue info and mb header info are packed. 2058d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 2068d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @param[in] ps_proc 2078d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* Pointer to the current process context 2088d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 2098d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @returns error status 2108d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 2118d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @remarks none 2128d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 2138d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S******************************************************************************* 2148d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S*/ 2158d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha SIH264E_ERROR_T ih264e_init_entropy_ctxt(process_ctxt_t *ps_proc) 2168d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S{ 2178d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* codec context */ 2188d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S codec_t *ps_codec = ps_proc->ps_codec; 2198d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 2208d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* entropy ctxt */ 2218d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S entropy_ctxt_t *ps_entropy = &ps_proc->s_entropy; 2228d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 2238d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* start address */ 2248d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_entropy->i4_mb_start_add = ps_entropy->i4_mb_y * ps_entropy->i4_wd_mbs + ps_entropy->i4_mb_x; 2258d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 2268d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* end address */ 2278d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_entropy->i4_mb_end_add = ps_entropy->i4_mb_start_add + ps_entropy->i4_mb_cnt; 2288d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 2298d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* slice index */ 2308d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_entropy->i4_cur_slice_idx = ps_proc->pu1_slice_idx[ps_entropy->i4_mb_start_add]; 2318d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 2328d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* sof */ 2338d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* @ start of frame or start of a new slice, set sof flag */ 2348d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_entropy->i4_mb_start_add == 0) 2358d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 2368d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_entropy->i4_sof = 1; 2378d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 2388d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 2398d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_entropy->i4_mb_x == 0) 2408d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 2418d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* packed mb coeff data */ 2428d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_entropy->pv_mb_coeff_data = ((UWORD8 *)ps_entropy->pv_pic_mb_coeff_data) + 2438d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_entropy->i4_mb_y * ps_codec->u4_size_coeff_data; 2448d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 2458d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* packed mb header data */ 2468d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_entropy->pv_mb_header_data = ((UWORD8 *)ps_entropy->pv_pic_mb_header_data) + 2478d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_entropy->i4_mb_y * ps_codec->u4_size_header_data; 2488d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 2498d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 2508d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S return IH264E_SUCCESS; 2518d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S} 2528d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 2538d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S/** 2548d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S******************************************************************************* 2558d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 2568d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @brief entry point for entropy coding 2578d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 2588d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @par Description 2598d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* This function calls lower level functions to perform entropy coding for a 2608d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* group (n rows) of mb's. After encoding 1 row of mb's, the function takes 2618d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* back the control, updates the ctxt and calls lower level functions again. 2628d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* This process is repeated till all the rows or group of mb's (which ever is 2638d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* minimum) are coded 2648d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 2658d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @param[in] ps_proc 2668d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* process context 2678d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 2688d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @returns error status 2698d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 2708d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @remarks 2718d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 2728d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S******************************************************************************* 2738d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S*/ 2748d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 2758d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha SIH264E_ERROR_T ih264e_entropy(process_ctxt_t *ps_proc) 2768d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S{ 2778d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* codec context */ 2788d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S codec_t *ps_codec = ps_proc->ps_codec; 2798d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 2808d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* entropy context */ 2818d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S entropy_ctxt_t *ps_entropy = &ps_proc->s_entropy; 2828d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 2833749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* cabac context */ 2843749f6f435e79624f72841e866245d84195551cdHarinarayanan K K cabac_ctxt_t *ps_cabac_ctxt = ps_entropy->ps_cabac; 2853749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 2868d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* sps */ 2878d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S sps_t *ps_sps = ps_entropy->ps_sps_base + (ps_entropy->u4_sps_id % MAX_SPS_CNT); 2888d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 2898d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* pps */ 2908d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pps_t *ps_pps = ps_entropy->ps_pps_base + (ps_entropy->u4_pps_id % MAX_PPS_CNT); 2918d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 2928d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* slice header */ 2938d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S slice_header_t *ps_slice_hdr = ps_entropy->ps_slice_hdr_base + (ps_entropy->i4_cur_slice_idx % MAX_SLICE_HDR_CNT); 2948d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 2958d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* slice type */ 2968d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_slice_type = ps_proc->i4_slice_type; 2978d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 2988d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* Bitstream structure */ 2998d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S bitstrm_t *ps_bitstrm = ps_entropy->ps_bitstrm; 3008d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 3018d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* output buff */ 3028d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S out_buf_t s_out_buf; 3038d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 3048d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* proc map */ 3058d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD8 *pu1_proc_map; 3068d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 3078d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* entropy map */ 3088d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD8 *pu1_entropy_map_curr; 3098d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 3108d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* proc base idx */ 311d3e18f0c8854c258021be1e603d8f50a241698a7Harinarayanan K K WORD32 ctxt_sel = ps_proc->i4_encode_api_call_cnt % MAX_CTXT_SETS; 3128d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 3138d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* temp var */ 3148d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_wd_mbs, i4_ht_mbs; 3158d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD32 u4_mb_cnt, u4_mb_idx, u4_mb_end_idx; 3163749f6f435e79624f72841e866245d84195551cdHarinarayanan K K WORD32 bitstream_start_offset, bitstream_end_offset; 3178d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /********************************************************************/ 3188d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* BEGIN INIT */ 3198d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /********************************************************************/ 3208d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 3218d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* entropy encode start address */ 3228d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S u4_mb_idx = ps_entropy->i4_mb_start_add; 3238d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 3248d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* entropy encode end address */ 3258d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S u4_mb_end_idx = ps_entropy->i4_mb_end_add; 3268d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 3278d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* width in mbs */ 3288d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S i4_wd_mbs = ps_entropy->i4_wd_mbs; 3298d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 3308d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* height in mbs */ 3318d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S i4_ht_mbs = ps_entropy->i4_ht_mbs; 3328d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 3338d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* total mb cnt */ 3348d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S u4_mb_cnt = i4_wd_mbs * i4_ht_mbs; 3358d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 3368d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* proc map */ 3378d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_proc_map = ps_proc->pu1_proc_map + ps_entropy->i4_mb_y * i4_wd_mbs; 3388d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 3398d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* entropy map */ 3408d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_entropy_map_curr = ps_entropy->pu1_entropy_map + ps_entropy->i4_mb_y * i4_wd_mbs; 3418d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 3428d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /********************************************************************/ 3438d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* @ start of frame / slice, */ 3448d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* initialize the output buffer, */ 3458d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* initialize the bit stream buffer, */ 3468d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* check if sps and pps headers have to be generated, */ 3478d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* populate and generate slice header */ 3488d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /********************************************************************/ 3498d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_entropy->i4_sof) 3508d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 3518d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /********************************************************************/ 3528d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* initialize the output buffer */ 3538d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /********************************************************************/ 3548d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S s_out_buf = ps_codec->as_out_buf[ctxt_sel]; 3558d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 3568d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* is last frame to encode */ 3578d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S s_out_buf.u4_is_last = ps_entropy->u4_is_last; 3588d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 3598d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* frame idx */ 3608d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S s_out_buf.u4_timestamp_high = ps_entropy->u4_timestamp_high; 3618d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S s_out_buf.u4_timestamp_low = ps_entropy->u4_timestamp_low; 3628d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 3638d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /********************************************************************/ 3648d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* initialize the bit stream buffer */ 3658d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /********************************************************************/ 3668d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ih264e_bitstrm_init(ps_bitstrm, s_out_buf.s_bits_buf.pv_buf, s_out_buf.s_bits_buf.u4_bufsize); 3678d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 3688d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /********************************************************************/ 3698d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* BEGIN HEADER GENERATION */ 3708d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /********************************************************************/ 3718d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (1 == ps_entropy->i4_gen_header) 3728d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 3738d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* generate sps */ 374126d84e0d3b6f3cb3a6e89f85da070e39ac51219Hamsalekha S ps_entropy->i4_error_code |= ih264e_generate_sps(ps_bitstrm, ps_sps, &ps_codec->s_vui); 3758d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 3768d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* generate pps */ 3778d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_entropy->i4_error_code |= ih264e_generate_pps(ps_bitstrm, ps_pps, ps_sps); 3788d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 3798d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* reset i4_gen_header */ 3808d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_entropy->i4_gen_header = 0; 3818d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 3828d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 3838d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* populate slice header */ 3848d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ih264e_populate_slice_header(ps_proc, ps_slice_hdr, ps_pps, ps_sps); 3858d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 3868d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* generate slice header */ 3878d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_entropy->i4_error_code |= ih264e_generate_slice_header(ps_bitstrm, ps_slice_hdr, 3888d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_pps, ps_sps); 3898d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 3908d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* once start of frame / slice is done, you can reset it */ 3918d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* it is the responsibility of the caller to set this flag */ 3928d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_entropy->i4_sof = 0; 3933749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 3943749f6f435e79624f72841e866245d84195551cdHarinarayanan K K if (CABAC == ps_entropy->u1_entropy_coding_mode_flag) 3953749f6f435e79624f72841e866245d84195551cdHarinarayanan K K { 3963749f6f435e79624f72841e866245d84195551cdHarinarayanan K K BITSTREAM_BYTE_ALIGN(ps_bitstrm); 3973749f6f435e79624f72841e866245d84195551cdHarinarayanan K K BITSTREAM_FLUSH(ps_bitstrm); 3983749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ih264e_init_cabac_ctxt(ps_entropy); 3993749f6f435e79624f72841e866245d84195551cdHarinarayanan K K } 4008d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 4018d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 4028d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* begin entropy coding for the mb set */ 4038d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S while (u4_mb_idx < u4_mb_end_idx) 4048d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 4058d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* init ptrs/indices */ 4068d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_entropy->i4_mb_x == i4_wd_mbs) 4078d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 4083749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_entropy->i4_mb_y++; 4098d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_entropy->i4_mb_x = 0; 4108d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 4118d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* packed mb coeff data */ 4128d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_entropy->pv_mb_coeff_data = ((UWORD8 *)ps_entropy->pv_pic_mb_coeff_data) + 4138d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_entropy->i4_mb_y * ps_codec->u4_size_coeff_data; 4148d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 4158d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* packed mb header data */ 4168d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_entropy->pv_mb_header_data = ((UWORD8 *)ps_entropy->pv_pic_mb_header_data) + 4178d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_entropy->i4_mb_y * ps_codec->u4_size_header_data; 4188d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 4198d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* proc map */ 4203749f6f435e79624f72841e866245d84195551cdHarinarayanan K K pu1_proc_map = ps_proc->pu1_proc_map + ps_entropy->i4_mb_y * i4_wd_mbs; 4218d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 4228d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* entropy map */ 4238d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_entropy_map_curr = ps_entropy->pu1_entropy_map + ps_entropy->i4_mb_y * i4_wd_mbs; 4248d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 4258d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 4268d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S DEBUG("\nmb indices x, y %d, %d", ps_entropy->i4_mb_x, ps_entropy->i4_mb_y); 4278d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ENTROPY_TRACE("mb index x %d", ps_entropy->i4_mb_x); 4288d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ENTROPY_TRACE("mb index y %d", ps_entropy->i4_mb_y); 4298d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 4308d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* wait until the curr mb is core coded */ 4318d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* The wait for curr mb to be core coded is essential when entropy is launched 4328d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * as a separate job 4338d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S */ 4348d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S while (1) 4358d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 4368d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S volatile UWORD8 *pu1_buf1; 4378d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 idx = ps_entropy->i4_mb_x; 4388d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 4393749f6f435e79624f72841e866245d84195551cdHarinarayanan K K pu1_buf1 = pu1_proc_map + idx; 4403749f6f435e79624f72841e866245d84195551cdHarinarayanan K K if (*pu1_buf1) 4418d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S break; 4428d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ithread_yield(); 4438d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 4448d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 4453749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 4468d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* write mb layer */ 4473749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_entropy->i4_error_code |= ps_codec->pf_write_mb_syntax_layer[ps_entropy->u1_entropy_coding_mode_flag][i4_slice_type](ps_entropy); 4483749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* Starting bitstream offset for header in bits */ 4493749f6f435e79624f72841e866245d84195551cdHarinarayanan K K bitstream_start_offset = GET_NUM_BITS(ps_bitstrm); 4508d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 4518d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* set entropy map */ 4528d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_entropy_map_curr[ps_entropy->i4_mb_x] = 1; 4538d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 4543749f6f435e79624f72841e866245d84195551cdHarinarayanan K K u4_mb_idx++; 4553749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_entropy->i4_mb_x++; 4563749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* check for eof */ 4573749f6f435e79624f72841e866245d84195551cdHarinarayanan K K if (CABAC == ps_entropy->u1_entropy_coding_mode_flag) 4583749f6f435e79624f72841e866245d84195551cdHarinarayanan K K { 4593749f6f435e79624f72841e866245d84195551cdHarinarayanan K K if (ps_entropy->i4_mb_x < i4_wd_mbs) 4603749f6f435e79624f72841e866245d84195551cdHarinarayanan K K { 4613749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ih264e_cabac_encode_terminate(ps_cabac_ctxt, 0); 4623749f6f435e79624f72841e866245d84195551cdHarinarayanan K K } 4633749f6f435e79624f72841e866245d84195551cdHarinarayanan K K } 4648d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 4658d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_entropy->i4_mb_x == i4_wd_mbs) 4668d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 4678d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* if slices are enabled */ 4688d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_codec->s_cfg.e_slice_mode == IVE_SLICE_MODE_BLOCKS) 4698d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 4708d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* current slice index */ 4718d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_curr_slice_idx = ps_entropy->i4_cur_slice_idx; 4728d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 4738d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* slice map */ 4748d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD8 *pu1_slice_idx = ps_entropy->pu1_slice_idx; 4758d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 4768d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* No need to open a slice at end of frame. The current slice can be closed at the time 4778d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * of signaling eof flag. 4788d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S */ 4793749f6f435e79624f72841e866245d84195551cdHarinarayanan K K if ((u4_mb_idx != u4_mb_cnt) && (i4_curr_slice_idx 4803749f6f435e79624f72841e866245d84195551cdHarinarayanan K K != pu1_slice_idx[u4_mb_idx])) 4818d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 4823749f6f435e79624f72841e866245d84195551cdHarinarayanan K K if (CAVLC == ps_entropy->u1_entropy_coding_mode_flag) 4833749f6f435e79624f72841e866245d84195551cdHarinarayanan K K { /* mb skip run */ 4843749f6f435e79624f72841e866245d84195551cdHarinarayanan K K if ((i4_slice_type != ISLICE) 4853749f6f435e79624f72841e866245d84195551cdHarinarayanan K K && *ps_entropy->pi4_mb_skip_run) 4868d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 4873749f6f435e79624f72841e866245d84195551cdHarinarayanan K K if (*ps_entropy->pi4_mb_skip_run) 4883749f6f435e79624f72841e866245d84195551cdHarinarayanan K K { 4898d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S PUT_BITS_UEV(ps_bitstrm, *ps_entropy->pi4_mb_skip_run, ps_entropy->i4_error_code, "mb skip run"); 4903749f6f435e79624f72841e866245d84195551cdHarinarayanan K K *ps_entropy->pi4_mb_skip_run = 0; 4913749f6f435e79624f72841e866245d84195551cdHarinarayanan K K } 4928d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 4933749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* put rbsp trailing bits for the previous slice */ 4943749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_entropy->i4_error_code |= ih264e_put_rbsp_trailing_bits(ps_bitstrm); 4953749f6f435e79624f72841e866245d84195551cdHarinarayanan K K } 4963749f6f435e79624f72841e866245d84195551cdHarinarayanan K K else 4973749f6f435e79624f72841e866245d84195551cdHarinarayanan K K { 4983749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ih264e_cabac_encode_terminate(ps_cabac_ctxt, 1); 4998d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 5008d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 5018d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* update slice header pointer */ 5028d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S i4_curr_slice_idx = pu1_slice_idx[u4_mb_idx]; 5038d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_entropy->i4_cur_slice_idx = i4_curr_slice_idx; 5043749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_slice_hdr = ps_entropy->ps_slice_hdr_base+ (i4_curr_slice_idx % MAX_SLICE_HDR_CNT); 5058d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 5068d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* populate slice header */ 5078d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_entropy->i4_mb_start_add = u4_mb_idx; 5083749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ih264e_populate_slice_header(ps_proc, ps_slice_hdr, ps_pps, 5093749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_sps); 5108d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 5118d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* generate slice header */ 5123749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_entropy->i4_error_code |= ih264e_generate_slice_header( 5133749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_bitstrm, ps_slice_hdr, ps_pps, ps_sps); 5143749f6f435e79624f72841e866245d84195551cdHarinarayanan K K if (CABAC == ps_entropy->u1_entropy_coding_mode_flag) 5153749f6f435e79624f72841e866245d84195551cdHarinarayanan K K { 5163749f6f435e79624f72841e866245d84195551cdHarinarayanan K K BITSTREAM_BYTE_ALIGN(ps_bitstrm); 5173749f6f435e79624f72841e866245d84195551cdHarinarayanan K K BITSTREAM_FLUSH(ps_bitstrm); 5183749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ih264e_init_cabac_ctxt(ps_entropy); 5193749f6f435e79624f72841e866245d84195551cdHarinarayanan K K } 5203749f6f435e79624f72841e866245d84195551cdHarinarayanan K K } 5213749f6f435e79624f72841e866245d84195551cdHarinarayanan K K else 5223749f6f435e79624f72841e866245d84195551cdHarinarayanan K K { 5233749f6f435e79624f72841e866245d84195551cdHarinarayanan K K if (CABAC == ps_entropy->u1_entropy_coding_mode_flag 5243749f6f435e79624f72841e866245d84195551cdHarinarayanan K K && u4_mb_idx != u4_mb_cnt) 5253749f6f435e79624f72841e866245d84195551cdHarinarayanan K K { 5263749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ih264e_cabac_encode_terminate(ps_cabac_ctxt, 0); 5273749f6f435e79624f72841e866245d84195551cdHarinarayanan K K } 5288d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 5298d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 5308d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* Dont execute any further instructions until store synchronization took place */ 5318d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S DATA_SYNC(); 5328d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 5333749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 5343749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* Ending bitstream offset for header in bits */ 5353749f6f435e79624f72841e866245d84195551cdHarinarayanan K K bitstream_end_offset = GET_NUM_BITS(ps_bitstrm); 5363749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_entropy->u4_header_bits[i4_slice_type == PSLICE] += 5373749f6f435e79624f72841e866245d84195551cdHarinarayanan K K bitstream_end_offset - bitstream_start_offset; 5388d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 5398d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 5408d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* check for eof */ 5418d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (u4_mb_idx == u4_mb_cnt) 5428d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 5438d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* set end of frame flag */ 5448d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_entropy->i4_eof = 1; 5458d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 5463749f6f435e79624f72841e866245d84195551cdHarinarayanan K K else 5473749f6f435e79624f72841e866245d84195551cdHarinarayanan K K { 5483749f6f435e79624f72841e866245d84195551cdHarinarayanan K K if (CABAC == ps_entropy->u1_entropy_coding_mode_flag 5493749f6f435e79624f72841e866245d84195551cdHarinarayanan K K && ps_codec->s_cfg.e_slice_mode 5503749f6f435e79624f72841e866245d84195551cdHarinarayanan K K != IVE_SLICE_MODE_BLOCKS) 5513749f6f435e79624f72841e866245d84195551cdHarinarayanan K K { 5523749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ih264e_cabac_encode_terminate(ps_cabac_ctxt, 0); 5533749f6f435e79624f72841e866245d84195551cdHarinarayanan K K } 5543749f6f435e79624f72841e866245d84195551cdHarinarayanan K K } 5558d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 5568d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_entropy->i4_eof) 5578d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 5583749f6f435e79624f72841e866245d84195551cdHarinarayanan K K if (CAVLC == ps_entropy->u1_entropy_coding_mode_flag) 5598d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 5603749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* mb skip run */ 5613749f6f435e79624f72841e866245d84195551cdHarinarayanan K K if ((i4_slice_type != ISLICE) && *ps_entropy->pi4_mb_skip_run) 5628d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 5633749f6f435e79624f72841e866245d84195551cdHarinarayanan K K if (*ps_entropy->pi4_mb_skip_run) 5643749f6f435e79624f72841e866245d84195551cdHarinarayanan K K { 5653749f6f435e79624f72841e866245d84195551cdHarinarayanan K K PUT_BITS_UEV(ps_bitstrm, *ps_entropy->pi4_mb_skip_run, 5663749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_entropy->i4_error_code, "mb skip run"); 5673749f6f435e79624f72841e866245d84195551cdHarinarayanan K K *ps_entropy->pi4_mb_skip_run = 0; 5683749f6f435e79624f72841e866245d84195551cdHarinarayanan K K } 5698d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 5703749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* put rbsp trailing bits */ 5713749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_entropy->i4_error_code |= ih264e_put_rbsp_trailing_bits(ps_bitstrm); 5723749f6f435e79624f72841e866245d84195551cdHarinarayanan K K } 5733749f6f435e79624f72841e866245d84195551cdHarinarayanan K K else 5743749f6f435e79624f72841e866245d84195551cdHarinarayanan K K { 5753749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ih264e_cabac_encode_terminate(ps_cabac_ctxt, 1); 5768d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 5778d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 5788d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* update current frame stats to rc library */ 5798d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 5808d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* number of bytes to stuff */ 5818d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_stuff_bytes; 5828d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 5838d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* update */ 5843749f6f435e79624f72841e866245d84195551cdHarinarayanan K K i4_stuff_bytes = ih264e_update_rc_post_enc( 5853749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_codec, ctxt_sel, 5863749f6f435e79624f72841e866245d84195551cdHarinarayanan K K (ps_proc->ps_codec->i4_poc == 0)); 5878d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 5888d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* cbr rc - house keeping */ 5898d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_codec->s_rate_control.post_encode_skip[ctxt_sel]) 5908d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 5918d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_entropy->ps_bitstrm->u4_strm_buf_offset = 0; 5928d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 5938d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S else if (i4_stuff_bytes) 5948d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 5958d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* add filler nal units */ 5968d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_entropy->i4_error_code |= ih264e_add_filler_nal_unit(ps_bitstrm, i4_stuff_bytes); 5978d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 5988d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 5998d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 6003749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* 6013749f6f435e79624f72841e866245d84195551cdHarinarayanan K K *Frame number is to be incremented only if the current frame is a 6023749f6f435e79624f72841e866245d84195551cdHarinarayanan K K * reference frame. After each successful frame encode, we increment 6033749f6f435e79624f72841e866245d84195551cdHarinarayanan K K * frame number by 1 6043749f6f435e79624f72841e866245d84195551cdHarinarayanan K K */ 6053749f6f435e79624f72841e866245d84195551cdHarinarayanan K K if (!ps_codec->s_rate_control.post_encode_skip[ctxt_sel] 6063749f6f435e79624f72841e866245d84195551cdHarinarayanan K K && ps_codec->u4_is_curr_frm_ref) 6073749f6f435e79624f72841e866245d84195551cdHarinarayanan K K { 6083749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_codec->i4_frame_num++; 6093749f6f435e79624f72841e866245d84195551cdHarinarayanan K K } 6108d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /********************************************************************/ 6118d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* signal the output */ 6128d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /********************************************************************/ 6133749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_codec->as_out_buf[ctxt_sel].s_bits_buf.u4_bytes = 6143749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_entropy->ps_bitstrm->u4_strm_buf_offset; 6158d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 6168d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S DEBUG("entropy status %x", ps_entropy->i4_error_code); 6178d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 6188d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 6198d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* allow threads to dequeue entropy jobs */ 6208d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->au4_entropy_thread_active[ctxt_sel] = 0; 6218d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 6228d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S return ps_entropy->i4_error_code; 6238d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S} 6248d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 6258d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S/** 6268d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S******************************************************************************* 6278d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 6288d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @brief Packs header information of a mb in to a buffer 6298d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 6308d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @par Description: 6318d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* After the deciding the mode info of a macroblock, the syntax elements 6328d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* associated with the mb are packed and stored. The entropy thread unpacks 6338d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* this buffer and generates the end bit stream. 6348d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 6358d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @param[in] ps_proc 6368d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* Pointer to the current process context 6378d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 6388d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @returns error status 6398d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 6408d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @remarks none 6418d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 6428d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S******************************************************************************* 6438d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S*/ 6448d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha SIH264E_ERROR_T ih264e_pack_header_data(process_ctxt_t *ps_proc) 6458d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S{ 6468d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* curr mb type */ 6478d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD32 u4_mb_type = ps_proc->u4_mb_type; 6488d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 6498d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* pack mb syntax layer of curr mb (used for entropy coding) */ 6508d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (u4_mb_type == I4x4) 6518d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 6528d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* pointer to mb header storage space */ 6538d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD8 *pu1_ptr = ps_proc->pv_mb_header_data; 6548d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 6558d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* temp var */ 6568d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4, byte; 6578d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 6588d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* mb type plus mode */ 6598d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S *pu1_ptr++ = (ps_proc->u1_c_i8_mode << 6) + u4_mb_type; 6608d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 6618d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* cbp */ 6628d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S *pu1_ptr++ = ps_proc->u4_cbp; 6638d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 6648d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* mb qp delta */ 6658d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S *pu1_ptr++ = ps_proc->u4_mb_qp - ps_proc->u4_mb_qp_prev; 6668d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 6678d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* sub mb modes */ 6688d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S for (i4 = 0; i4 < 16; i4 ++) 6698d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 6708d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S byte = 0; 6718d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 6728d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_proc->au1_predicted_intra_luma_mb_4x4_modes[i4] == 6738d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->au1_intra_luma_mb_4x4_modes[i4]) 6748d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 6758d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S byte |= 1; 6768d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 6778d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S else 6788d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 6798d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 6808d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_proc->au1_intra_luma_mb_4x4_modes[i4] < 6818d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->au1_predicted_intra_luma_mb_4x4_modes[i4]) 6828d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 6838d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S byte |= (ps_proc->au1_intra_luma_mb_4x4_modes[i4] << 1); 6848d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 6858d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S else 6868d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 6878d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S byte |= (ps_proc->au1_intra_luma_mb_4x4_modes[i4] - 1) << 1; 6888d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 6898d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 6908d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 6918d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S i4++; 6928d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 6938d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_proc->au1_predicted_intra_luma_mb_4x4_modes[i4] == 6948d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->au1_intra_luma_mb_4x4_modes[i4]) 6958d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 6968d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S byte |= 16; 6978d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 6988d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S else 6998d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 7008d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 7018d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_proc->au1_intra_luma_mb_4x4_modes[i4] < 7028d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->au1_predicted_intra_luma_mb_4x4_modes[i4]) 7038d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 7048d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S byte |= (ps_proc->au1_intra_luma_mb_4x4_modes[i4] << 5); 7058d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 7068d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S else 7078d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 7088d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S byte |= (ps_proc->au1_intra_luma_mb_4x4_modes[i4] - 1) << 5; 7098d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 7108d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 7118d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 7128d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S *pu1_ptr++ = byte; 7138d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 7148d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 7158d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* end of mb layer */ 7168d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->pv_mb_header_data = pu1_ptr; 7178d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 7188d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S else if (u4_mb_type == I16x16) 7198d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 7208d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* pointer to mb header storage space */ 7218d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD8 *pu1_ptr = ps_proc->pv_mb_header_data; 7228d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 7238d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* mb type plus mode */ 7248d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S *pu1_ptr++ = (ps_proc->u1_c_i8_mode << 6) + (ps_proc->u1_l_i16_mode << 4) + u4_mb_type; 7258d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 7268d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* cbp */ 7278d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S *pu1_ptr++ = ps_proc->u4_cbp; 7288d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 7298d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* mb qp delta */ 7308d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S *pu1_ptr++ = ps_proc->u4_mb_qp - ps_proc->u4_mb_qp_prev; 7318d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 7328d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* end of mb layer */ 7338d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->pv_mb_header_data = pu1_ptr; 7348d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 7358d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S else if (u4_mb_type == P16x16) 7368d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 7378d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* pointer to mb header storage space */ 7388d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD8 *pu1_ptr = ps_proc->pv_mb_header_data; 7398d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 7408d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD16 *i2_mv_ptr; 7418d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 7428d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* mb type plus mode */ 7438d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S *pu1_ptr++ = u4_mb_type; 7448d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 7458d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* cbp */ 7468d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S *pu1_ptr++ = ps_proc->u4_cbp; 7478d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 7488d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* mb qp delta */ 7498d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S *pu1_ptr++ = ps_proc->u4_mb_qp - ps_proc->u4_mb_qp_prev; 7508d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 7518d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S i2_mv_ptr = (WORD16 *)pu1_ptr; 7528d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 7533749f6f435e79624f72841e866245d84195551cdHarinarayanan K K *i2_mv_ptr++ = ps_proc->ps_pu->s_me_info[0].s_mv.i2_mvx - ps_proc->ps_pred_mv[0].s_mv.i2_mvx; 7548d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 7553749f6f435e79624f72841e866245d84195551cdHarinarayanan K K *i2_mv_ptr++ = ps_proc->ps_pu->s_me_info[0].s_mv.i2_mvy - ps_proc->ps_pred_mv[0].s_mv.i2_mvy; 7568d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 7578d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* end of mb layer */ 7588d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->pv_mb_header_data = i2_mv_ptr; 7598d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 7608d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S else if (u4_mb_type == PSKIP) 7618d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 7628d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* pointer to mb header storage space */ 7638d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD8 *pu1_ptr = ps_proc->pv_mb_header_data; 7648d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 7658d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* mb type plus mode */ 7668d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S *pu1_ptr++ = u4_mb_type; 7678d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 7688d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* end of mb layer */ 7698d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->pv_mb_header_data = pu1_ptr; 7708d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 7713749f6f435e79624f72841e866245d84195551cdHarinarayanan K K else if(u4_mb_type == B16x16) 7723749f6f435e79624f72841e866245d84195551cdHarinarayanan K K { 7733749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 7743749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* pointer to mb header storage space */ 7753749f6f435e79624f72841e866245d84195551cdHarinarayanan K K UWORD8 *pu1_ptr = ps_proc->pv_mb_header_data; 7763749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 7773749f6f435e79624f72841e866245d84195551cdHarinarayanan K K WORD16 *i2_mv_ptr; 7783749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 7793749f6f435e79624f72841e866245d84195551cdHarinarayanan K K UWORD32 u4_pred_mode = ps_proc->ps_pu->b2_pred_mode; 7803749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 7813749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* mb type plus mode */ 7823749f6f435e79624f72841e866245d84195551cdHarinarayanan K K *pu1_ptr++ = (u4_pred_mode << 4) + u4_mb_type; 7833749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 7843749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* cbp */ 7853749f6f435e79624f72841e866245d84195551cdHarinarayanan K K *pu1_ptr++ = ps_proc->u4_cbp; 7863749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 7873749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* mb qp delta */ 7883749f6f435e79624f72841e866245d84195551cdHarinarayanan K K *pu1_ptr++ = ps_proc->u4_mb_qp - ps_proc->u4_mb_qp_prev; 7893749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 7903749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* l0 & l1 me data */ 7913749f6f435e79624f72841e866245d84195551cdHarinarayanan K K i2_mv_ptr = (WORD16 *)pu1_ptr; 7923749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 7933749f6f435e79624f72841e866245d84195551cdHarinarayanan K K if (u4_pred_mode != PRED_L1) 7943749f6f435e79624f72841e866245d84195551cdHarinarayanan K K { 7953749f6f435e79624f72841e866245d84195551cdHarinarayanan K K *i2_mv_ptr++ = ps_proc->ps_pu->s_me_info[0].s_mv.i2_mvx 7963749f6f435e79624f72841e866245d84195551cdHarinarayanan K K - ps_proc->ps_pred_mv[0].s_mv.i2_mvx; 7973749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 7983749f6f435e79624f72841e866245d84195551cdHarinarayanan K K *i2_mv_ptr++ = ps_proc->ps_pu->s_me_info[0].s_mv.i2_mvy 7993749f6f435e79624f72841e866245d84195551cdHarinarayanan K K - ps_proc->ps_pred_mv[0].s_mv.i2_mvy; 8003749f6f435e79624f72841e866245d84195551cdHarinarayanan K K } 8013749f6f435e79624f72841e866245d84195551cdHarinarayanan K K if (u4_pred_mode != PRED_L0) 8023749f6f435e79624f72841e866245d84195551cdHarinarayanan K K { 8033749f6f435e79624f72841e866245d84195551cdHarinarayanan K K *i2_mv_ptr++ = ps_proc->ps_pu->s_me_info[1].s_mv.i2_mvx 8043749f6f435e79624f72841e866245d84195551cdHarinarayanan K K - ps_proc->ps_pred_mv[1].s_mv.i2_mvx; 8053749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 8063749f6f435e79624f72841e866245d84195551cdHarinarayanan K K *i2_mv_ptr++ = ps_proc->ps_pu->s_me_info[1].s_mv.i2_mvy 8073749f6f435e79624f72841e866245d84195551cdHarinarayanan K K - ps_proc->ps_pred_mv[1].s_mv.i2_mvy; 8083749f6f435e79624f72841e866245d84195551cdHarinarayanan K K } 8093749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 8103749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* end of mb layer */ 8113749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_proc->pv_mb_header_data = i2_mv_ptr; 8123749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 8133749f6f435e79624f72841e866245d84195551cdHarinarayanan K K } 8143749f6f435e79624f72841e866245d84195551cdHarinarayanan K K else if(u4_mb_type == BDIRECT) 8153749f6f435e79624f72841e866245d84195551cdHarinarayanan K K { 8163749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* pointer to mb header storage space */ 8173749f6f435e79624f72841e866245d84195551cdHarinarayanan K K UWORD8 *pu1_ptr = ps_proc->pv_mb_header_data; 8183749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 8193749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* mb type plus mode */ 8203749f6f435e79624f72841e866245d84195551cdHarinarayanan K K *pu1_ptr++ = u4_mb_type; 8213749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 8223749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* cbp */ 8233749f6f435e79624f72841e866245d84195551cdHarinarayanan K K *pu1_ptr++ = ps_proc->u4_cbp; 8243749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 8253749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* mb qp delta */ 8263749f6f435e79624f72841e866245d84195551cdHarinarayanan K K *pu1_ptr++ = ps_proc->u4_mb_qp - ps_proc->u4_mb_qp_prev; 8273749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 8283749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_proc->pv_mb_header_data = pu1_ptr; 8293749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 8303749f6f435e79624f72841e866245d84195551cdHarinarayanan K K } 8313749f6f435e79624f72841e866245d84195551cdHarinarayanan K K else if(u4_mb_type == BSKIP) 8323749f6f435e79624f72841e866245d84195551cdHarinarayanan K K { 8333749f6f435e79624f72841e866245d84195551cdHarinarayanan K K UWORD32 u4_pred_mode = ps_proc->ps_pu->b2_pred_mode; 8343749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 8353749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* pointer to mb header storage space */ 8363749f6f435e79624f72841e866245d84195551cdHarinarayanan K K UWORD8 *pu1_ptr = ps_proc->pv_mb_header_data; 8373749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 8383749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* mb type plus mode */ 8393749f6f435e79624f72841e866245d84195551cdHarinarayanan K K *pu1_ptr++ = (u4_pred_mode << 4) + u4_mb_type; 8403749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 8413749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* end of mb layer */ 8423749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_proc->pv_mb_header_data = pu1_ptr; 8433749f6f435e79624f72841e866245d84195551cdHarinarayanan K K } 8448d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 8458d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S return IH264E_SUCCESS; 8468d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S} 8478d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 8488d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S/** 8498d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S******************************************************************************* 8508d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 8518d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @brief update process context after encoding an mb. This involves preserving 8528d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* the current mb information for later use, initialize the proc ctxt elements to 8538d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* encode next mb. 8548d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 8558d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @par Description: 8568d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* This function performs house keeping tasks after encoding an mb. 8578d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* After encoding an mb, various elements of the process context needs to be 8588d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* updated to encode the next mb. For instance, the source, recon and reference 8598d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* pointers, mb indices have to be adjusted to the next mb. The slice index of 8608d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* the current mb needs to be updated. If mb qp modulation is enabled, then if 8618d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* the qp changes the quant param structure needs to be updated. Also to encoding 8628d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* the next mb, the current mb info is used as part of mode prediction or mv 8638d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* prediction. Hence the current mb info has to preserved at top/top left/left 8648d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* locations. 8658d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 8668d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @param[in] ps_proc 8678d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* Pointer to the current process context 8688d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 8698d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @returns none 8708d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 8718d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @remarks none 8728d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 8738d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S******************************************************************************* 8748d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S*/ 8758d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha SWORD32 ih264e_update_proc_ctxt(process_ctxt_t *ps_proc) 8768d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S{ 8778d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* error status */ 8788d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 error_status = IH264_SUCCESS; 8798d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 8808d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* codec context */ 8818d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S codec_t *ps_codec = ps_proc->ps_codec; 8828d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 8838d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* curr mb indices */ 8848d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_mb_x = ps_proc->i4_mb_x; 8858d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_mb_y = ps_proc->i4_mb_y; 8868d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 8878d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* mb syntax elements of neighbors */ 8888d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S mb_info_t *ps_left_syn = &ps_proc->s_left_mb_syntax_ele; 8898d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S mb_info_t *ps_top_syn = ps_proc->ps_top_row_mb_syntax_ele + i4_mb_x; 8908d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S mb_info_t *ps_top_left_syn = &ps_proc->s_top_left_mb_syntax_ele; 8918d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 8928d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* curr mb type */ 8938d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD32 u4_mb_type = ps_proc->u4_mb_type; 8948d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 8958d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* curr mb type */ 8968d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD32 u4_is_intra = ps_proc->u4_is_intra; 8978d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 8988d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* width in mbs */ 8998d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_wd_mbs = ps_proc->i4_wd_mbs; 9008d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 9018d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /*height in mbs*/ 9028d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_ht_mbs = ps_proc->i4_ht_mbs; 9038d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 9048d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* proc map */ 9058d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD8 *pu1_proc_map = ps_proc->pu1_proc_map + (i4_mb_y * i4_wd_mbs); 9068d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 9078d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* deblk context */ 9088d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S deblk_ctxt_t *ps_deblk = &ps_proc->s_deblk_ctxt; 9098d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 9108d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* deblk bs context */ 9118d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S bs_ctxt_t *ps_bs = &(ps_deblk->s_bs_ctxt); 9128d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 9138d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* top row motion vector info */ 9148d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S enc_pu_t *ps_top_row_pu = ps_proc->ps_top_row_pu + i4_mb_x; 9158d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 9168d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* top left mb motion vector */ 9178d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S enc_pu_t *ps_top_left_mb_pu = &ps_proc->s_top_left_mb_pu; 9188d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 9198d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* left mb motion vector */ 9208d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S enc_pu_t *ps_left_mb_pu = &ps_proc->s_left_mb_pu; 9218d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 9228d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* sub mb modes */ 9238d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD8 *pu1_top_mb_intra_modes = ps_proc->pu1_top_mb_intra_modes + (i4_mb_x << 4); 9248d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 9258d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /*************************************************************/ 9268d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* During MV prediction, when top right mb is not available, */ 9278d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* top left mb info. is used for prediction. Hence the curr */ 9288d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* top, which will be top left for the next mb needs to be */ 9298d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* preserved before updating it with curr mb info. */ 9308d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /*************************************************************/ 9318d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 9328d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* mb type, mb class, csbp */ 9338d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S *ps_top_left_syn = *ps_top_syn; 9348d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 9353749f6f435e79624f72841e866245d84195551cdHarinarayanan K K if (ps_proc->i4_slice_type != ISLICE) 9368d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 9378d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /*****************************************/ 9388d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* update top left with top info results */ 9398d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /*****************************************/ 9408d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* mv */ 9418d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S *ps_top_left_mb_pu = *ps_top_row_pu; 9428d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 9438d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 9448d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /*************************************************/ 9458d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* update top and left with curr mb info results */ 9468d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /*************************************************/ 9478d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 9488d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* mb type */ 9498d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_left_syn->u2_mb_type = ps_top_syn->u2_mb_type = u4_mb_type; 9508d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 9518d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* mb class */ 9528d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_left_syn->u2_is_intra = ps_top_syn->u2_is_intra = u4_is_intra; 9538d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 9548d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* csbp */ 9558d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_left_syn->u4_csbp = ps_top_syn->u4_csbp = ps_proc->u4_csbp; 9568d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 9578d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* distortion */ 9588d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_left_syn->i4_mb_distortion = ps_top_syn->i4_mb_distortion = ps_proc->i4_mb_distortion; 9598d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 9608d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (u4_is_intra) 9618d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 9628d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* mb / sub mb modes */ 9638d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (I16x16 == u4_mb_type) 9648d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 9658d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_top_mb_intra_modes[0] = ps_proc->au1_left_mb_intra_modes[0] = ps_proc->u1_l_i16_mode; 9668d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 9678d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S else if (I4x4 == u4_mb_type) 9688d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 9698d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_mem_cpy_mul8(ps_proc->au1_left_mb_intra_modes, ps_proc->au1_intra_luma_mb_4x4_modes, 16); 9708d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_mem_cpy_mul8(pu1_top_mb_intra_modes, ps_proc->au1_intra_luma_mb_4x4_modes, 16); 9718d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 9728d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S else if (I8x8 == u4_mb_type) 9738d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 9748d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S memcpy(ps_proc->au1_left_mb_intra_modes, ps_proc->au1_intra_luma_mb_8x8_modes, 4); 9758d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S memcpy(pu1_top_mb_intra_modes, ps_proc->au1_intra_luma_mb_8x8_modes, 4); 9768d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 9778d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 9783749f6f435e79624f72841e866245d84195551cdHarinarayanan K K if ((ps_proc->i4_slice_type == PSLICE) ||(ps_proc->i4_slice_type == BSLICE)) 9798d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 9808d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* mv */ 9818d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S *ps_left_mb_pu = *ps_top_row_pu = *(ps_proc->ps_pu); 9828d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 9833749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 9843749f6f435e79624f72841e866245d84195551cdHarinarayanan K K *ps_proc->pu4_mb_pu_cnt = 1; 9858d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 9868d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S else 9878d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 9888d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* mv */ 9898d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S *ps_left_mb_pu = *ps_top_row_pu = *(ps_proc->ps_pu); 9908d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 9918d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 9928d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* 9938d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * Mark that the MB has been coded intra 9948d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * So that future AIRs can skip it 9958d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S */ 9968d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->pu1_is_intra_coded[i4_mb_x + (i4_mb_y * i4_wd_mbs)] = u4_is_intra; 9978d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 9988d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /**************************************************/ 9998d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* pack mb header info. for entropy coding */ 10008d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /**************************************************/ 10018d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ih264e_pack_header_data(ps_proc); 10028d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 10038d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* update previous mb qp */ 10048d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->u4_mb_qp_prev = ps_proc->u4_mb_qp; 10058d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 10068d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* store qp */ 10078d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->s_deblk_ctxt.s_bs_ctxt.pu1_pic_qp[(i4_mb_y * i4_wd_mbs) + i4_mb_x] = ps_proc->u4_mb_qp; 10088d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 10098d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* 10108d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * We need to sync the cache to make sure that the nmv content of proc 10118d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * is updated to cache properly 10128d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S */ 10138d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S DATA_SYNC(); 10148d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 10158d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* Just before finishing the row, enqueue the job in to entropy queue. 10168d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * The master thread depending on its convenience shall dequeue it and 10178d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * performs entropy. 10188d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * 10198d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * WARN !! Placing this block post proc map update can cause queuing of 10208d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * entropy jobs in out of order. 10218d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S */ 10228d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (i4_mb_x == i4_wd_mbs - 1) 10238d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 10248d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* job structures */ 10258d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S job_t s_job; 10268d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 10278d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* job class */ 10288d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S s_job.i4_cmd = CMD_ENTROPY; 10298d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 10308d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* number of mbs to be processed in the current job */ 10318d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S s_job.i2_mb_cnt = ps_codec->s_cfg.i4_wd_mbs; 10328d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 10338d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* job start index x */ 10348d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S s_job.i2_mb_x = 0; 10358d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 10368d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* job start index y */ 10378d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S s_job.i2_mb_y = ps_proc->i4_mb_y; 10388d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 10398d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* proc base idx */ 1040d3e18f0c8854c258021be1e603d8f50a241698a7Harinarayanan K K s_job.i2_proc_base_idx = (ps_codec->i4_encode_api_call_cnt % MAX_CTXT_SETS) ? (MAX_PROCESS_CTXT / 2) : 0; 10418d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 10428d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* queue the job */ 10438d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S error_status |= ih264_list_queue(ps_proc->pv_entropy_jobq, &s_job, 1); 10448d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 10458d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if(ps_proc->i4_mb_y == (i4_ht_mbs - 1)) 10468d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ih264_list_terminate(ps_codec->pv_entropy_jobq); 10478d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 10488d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 10498d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* update proc map */ 10508d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_proc_map[i4_mb_x] = 1; 10518d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 10528d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /**************************************************/ 10538d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* update proc ctxt elements for encoding next mb */ 10548d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /**************************************************/ 10558d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* update indices */ 10568d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S i4_mb_x ++; 10578d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->i4_mb_x = i4_mb_x; 10588d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 10598d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_proc->i4_mb_x == i4_wd_mbs) 10608d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 10618d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->i4_mb_y++; 10628d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->i4_mb_x = 0; 10638d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 10648d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 10658d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* update slice index */ 10668d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->i4_cur_slice_idx = ps_proc->pu1_slice_idx[ps_proc->i4_mb_y * i4_wd_mbs + ps_proc->i4_mb_x]; 10678d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 10688d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* update buffers pointers */ 10698d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->pu1_src_buf_luma += MB_SIZE; 10708d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->pu1_rec_buf_luma += MB_SIZE; 10713749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_proc->apu1_ref_buf_luma[0] += MB_SIZE; 10723749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_proc->apu1_ref_buf_luma[1] += MB_SIZE; 10738d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 10748d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* 10758d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * Note: Although chroma mb size is 8, as the chroma buffers are interleaved, 10768d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * the stride per MB is MB_SIZE 10778d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S */ 10788d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->pu1_src_buf_chroma += MB_SIZE; 10798d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->pu1_rec_buf_chroma += MB_SIZE; 10803749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_proc->apu1_ref_buf_chroma[0] += MB_SIZE; 10813749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_proc->apu1_ref_buf_chroma[1] += MB_SIZE; 10823749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 10838d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 10848d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 10858d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* Reset cost, distortion params */ 10868d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->i4_mb_cost = INT_MAX; 10878d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->i4_mb_distortion = SHRT_MAX; 10888d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 10898d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->ps_pu += *ps_proc->pu4_mb_pu_cnt; 10908d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 10918d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->pu4_mb_pu_cnt += 1; 10928d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 10933749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* Update colocated pu */ 10943749f6f435e79624f72841e866245d84195551cdHarinarayanan K K if (ps_proc->i4_slice_type == BSLICE) 10953749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_proc->ps_colpu += *(ps_proc->aps_mv_buf[1]->pu4_mb_pu_cnt + (i4_mb_y * ps_proc->i4_wd_mbs) + i4_mb_x); 10963749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 10978d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* deblk ctxts */ 10988d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_proc->u4_disable_deblock_level != 1) 10998d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 11008d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* indices */ 11018d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_bs->i4_mb_x = ps_proc->i4_mb_x; 11028d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_bs->i4_mb_y = ps_proc->i4_mb_y; 11038d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 11048d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#ifndef N_MB_ENABLE /* For N MB processing update take place inside deblocking function */ 11058d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_deblk->i4_mb_x ++; 11068d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 11078d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_deblk->pu1_cur_pic_luma += MB_SIZE; 11088d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* 11098d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * Note: Although chroma mb size is 8, as the chroma buffers are interleaved, 11108d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * the stride per MB is MB_SIZE 11118d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S */ 11128d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_deblk->pu1_cur_pic_chroma += MB_SIZE; 11138d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S#endif 11148d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 11158d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 11168d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S return error_status; 11178d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S} 11188d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 11198d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S/** 11208d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S******************************************************************************* 11218d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 11228d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @brief initialize process context. 11238d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 11248d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @par Description: 11258d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* Before dispatching the current job to process thread, the process context 11268d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* associated with the job is initialized. Usually every job aims to encode one 11278d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* row of mb's. Basing on the row indices provided by the job, the process 11288d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* context's buffer ptrs, slice indices and other elements that are necessary 11298d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* during core-coding are initialized. 11308d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 11318d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @param[in] ps_proc 11328d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* Pointer to the current process context 11338d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 11348d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @returns error status 11358d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 11368d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @remarks none 11378d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 11388d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S******************************************************************************* 11398d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S*/ 11408d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha SIH264E_ERROR_T ih264e_init_proc_ctxt(process_ctxt_t *ps_proc) 11418d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S{ 11428d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* codec context */ 11438d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S codec_t *ps_codec = ps_proc->ps_codec; 11448d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 11458d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* nmb processing context*/ 11468d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S n_mb_process_ctxt_t *ps_n_mb_ctxt = &ps_proc->s_n_mb_ctxt; 11478d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 11488d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* indices */ 11498d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_mb_x, i4_mb_y; 11508d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 11518d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* strides */ 11528d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_src_strd = ps_proc->i4_src_strd; 11539113f5615cb4b2be179b288a59324de723d2ec9aMartin Storsjo WORD32 i4_src_chroma_strd = ps_proc->i4_src_chroma_strd; 11548d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_rec_strd = ps_proc->i4_rec_strd; 11558d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 11568d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* quant params */ 11578d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S quant_params_t *ps_qp_params = ps_proc->ps_qp_params[0]; 11588d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 11598d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* deblk ctxt */ 11608d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S deblk_ctxt_t *ps_deblk = &ps_proc->s_deblk_ctxt; 11618d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 11628d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* deblk bs context */ 11638d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S bs_ctxt_t *ps_bs = &(ps_deblk->s_bs_ctxt); 11648d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 11658d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* Pointer to mv_buffer of current frame */ 11668d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S mv_buf_t *ps_cur_mv_buf = ps_proc->ps_cur_mv_buf; 11678d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 11688d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* Pointers for color space conversion */ 11698d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD8 *pu1_y_buf_base, *pu1_u_buf_base, *pu1_v_buf_base; 11708d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 11718d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* Pad the MB to support non standard sizes */ 11723a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar UWORD32 u4_pad_right_sz = ps_codec->s_cfg.u4_wd - ps_codec->s_cfg.u4_disp_wd; 11738d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD32 u4_pad_bottom_sz = ps_codec->s_cfg.u4_ht - ps_codec->s_cfg.u4_disp_ht; 11743a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar UWORD16 u2_num_rows = MB_SIZE; 11753a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar WORD32 convert_uv_only; 11768d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 11778d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /********************************************************************/ 11788d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* BEGIN INIT */ 11798d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /********************************************************************/ 11808d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 11818d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S i4_mb_x = ps_proc->i4_mb_x; 11828d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S i4_mb_y = ps_proc->i4_mb_y; 11838d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 11848d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* Number of mbs processed in one loop of process function */ 1185d3e18f0c8854c258021be1e603d8f50a241698a7Harinarayanan K K ps_proc->i4_nmb_ntrpy = ps_proc->i4_wd_mbs; 1186d3e18f0c8854c258021be1e603d8f50a241698a7Harinarayanan K K ps_proc->u4_nmb_me = ps_proc->i4_wd_mbs; 11878d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 11883749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* init buffer pointers */ 11893a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar convert_uv_only = 1; 11904f5ceca078b8caab482bfd314bb24308ad872446Martin Storsjo if (u4_pad_bottom_sz || u4_pad_right_sz || 1191cb6a43532c0b863d46c82feb25b10dc8732a34f9Martin Storsjo ps_codec->s_cfg.e_inp_color_fmt == IV_YUV_422ILE) 11923a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar { 1193cb6a43532c0b863d46c82feb25b10dc8732a34f9Martin Storsjo if (ps_proc->i4_mb_y == ps_proc->i4_ht_mbs - 1) 1194cb6a43532c0b863d46c82feb25b10dc8732a34f9Martin Storsjo u2_num_rows = (UWORD16) MB_SIZE - u4_pad_bottom_sz; 11953a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar ps_proc->pu1_src_buf_luma_base = ps_codec->pu1_y_csc_buf_base; 1196cb6a43532c0b863d46c82feb25b10dc8732a34f9Martin Storsjo i4_src_strd = ps_proc->i4_src_strd = ps_codec->s_cfg.u4_max_wd; 11973a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar ps_proc->pu1_src_buf_luma = ps_proc->pu1_src_buf_luma_base + (i4_mb_x * MB_SIZE) + ps_codec->s_cfg.u4_max_wd * (i4_mb_y * MB_SIZE); 11983a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar convert_uv_only = 0; 11993a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar } 12003a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar else 1201cb6a43532c0b863d46c82feb25b10dc8732a34f9Martin Storsjo { 1202cb6a43532c0b863d46c82feb25b10dc8732a34f9Martin Storsjo i4_src_strd = ps_proc->i4_src_strd = ps_proc->s_inp_buf.s_raw_buf.au4_strd[0]; 12033a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar ps_proc->pu1_src_buf_luma = ps_proc->pu1_src_buf_luma_base + (i4_mb_x * MB_SIZE) + i4_src_strd * (i4_mb_y * MB_SIZE); 1204cb6a43532c0b863d46c82feb25b10dc8732a34f9Martin Storsjo } 12053a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar 12063a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar 1207cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar if (ps_codec->s_cfg.e_inp_color_fmt == IV_YUV_422ILE || 1208cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar ps_codec->s_cfg.e_inp_color_fmt == IV_YUV_420P || 12094f5ceca078b8caab482bfd314bb24308ad872446Martin Storsjo ps_proc->i4_mb_y == (ps_proc->i4_ht_mbs - 1) || 12104f5ceca078b8caab482bfd314bb24308ad872446Martin Storsjo u4_pad_bottom_sz || u4_pad_right_sz) 1211cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar { 1212cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar if ((ps_codec->s_cfg.e_inp_color_fmt == IV_YUV_420SP_UV) || 1213cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar (ps_codec->s_cfg.e_inp_color_fmt == IV_YUV_420SP_VU)) 1214cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar ps_proc->pu1_src_buf_chroma_base = ps_codec->pu1_uv_csc_buf_base; 1215cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar 1216cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar ps_proc->pu1_src_buf_chroma = ps_proc->pu1_src_buf_chroma_base + (i4_mb_x * MB_SIZE) + ps_codec->s_cfg.u4_max_wd * (i4_mb_y * BLK8x8SIZE); 1217cb6a43532c0b863d46c82feb25b10dc8732a34f9Martin Storsjo i4_src_chroma_strd = ps_proc->i4_src_chroma_strd = ps_codec->s_cfg.u4_max_wd; 1218cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar } 1219cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar else 1220cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar { 1221cb6a43532c0b863d46c82feb25b10dc8732a34f9Martin Storsjo i4_src_chroma_strd = ps_proc->i4_src_chroma_strd = ps_proc->s_inp_buf.s_raw_buf.au4_strd[1]; 12229113f5615cb4b2be179b288a59324de723d2ec9aMartin Storsjo ps_proc->pu1_src_buf_chroma = ps_proc->pu1_src_buf_chroma_base + (i4_mb_x * MB_SIZE) + i4_src_chroma_strd * (i4_mb_y * BLK8x8SIZE); 1223cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar } 1224cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar 12258d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->pu1_rec_buf_luma = ps_proc->pu1_rec_buf_luma_base + (i4_mb_x * MB_SIZE) + i4_rec_strd * (i4_mb_y * MB_SIZE); 12268d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->pu1_rec_buf_chroma = ps_proc->pu1_rec_buf_chroma_base + (i4_mb_x * MB_SIZE) + i4_rec_strd * (i4_mb_y * BLK8x8SIZE); 12278d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 12283749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* Tempral back and forward reference buffer */ 12293749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_proc->apu1_ref_buf_luma[0] = ps_proc->apu1_ref_buf_luma_base[0] + (i4_mb_x * MB_SIZE) + i4_rec_strd * (i4_mb_y * MB_SIZE); 12303749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_proc->apu1_ref_buf_chroma[0] = ps_proc->apu1_ref_buf_chroma_base[0] + (i4_mb_x * MB_SIZE) + i4_rec_strd * (i4_mb_y * BLK8x8SIZE); 12313749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_proc->apu1_ref_buf_luma[1] = ps_proc->apu1_ref_buf_luma_base[1] + (i4_mb_x * MB_SIZE) + i4_rec_strd * (i4_mb_y * MB_SIZE); 12323749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_proc->apu1_ref_buf_chroma[1] = ps_proc->apu1_ref_buf_chroma_base[1] + (i4_mb_x * MB_SIZE) + i4_rec_strd * (i4_mb_y * BLK8x8SIZE); 12333a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar 12348d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* 12358d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * Do color space conversion 12368d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * NOTE : We assume there that the number of MB's to process will not span multiple rows 12378d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S */ 12388d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S switch (ps_codec->s_cfg.e_inp_color_fmt) 12398d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 12408d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S case IV_YUV_420SP_UV: 12418d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S case IV_YUV_420SP_VU: 1242cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar /* In case of 420 semi-planar input, copy last few rows to intermediate 1243cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar buffer as chroma trans functions access one extra byte due to interleaved input. 1244cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar This data will be padded if required */ 12454f5ceca078b8caab482bfd314bb24308ad872446Martin Storsjo if (ps_proc->i4_mb_y == (ps_proc->i4_ht_mbs - 1) || u4_pad_bottom_sz || u4_pad_right_sz) 1246cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar { 12474f5ceca078b8caab482bfd314bb24308ad872446Martin Storsjo WORD32 num_rows = MB_SIZE; 1248cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar UWORD8 *pu1_src; 1249cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar UWORD8 *pu1_dst; 1250cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar WORD32 i; 1251cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar pu1_src = (UWORD8 *)ps_proc->s_inp_buf.s_raw_buf.apv_bufs[0] + (i4_mb_x * MB_SIZE) + 1252cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar ps_proc->s_inp_buf.s_raw_buf.au4_strd[0] * (i4_mb_y * MB_SIZE); 1253cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar 1254cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar pu1_dst = ps_proc->pu1_src_buf_luma; 1255cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar 12564f5ceca078b8caab482bfd314bb24308ad872446Martin Storsjo /* If padding is required, we always copy luma, if padding isn't required we never copy luma. */ 12574f5ceca078b8caab482bfd314bb24308ad872446Martin Storsjo if (u4_pad_bottom_sz || u4_pad_right_sz) { 12584f5ceca078b8caab482bfd314bb24308ad872446Martin Storsjo if (ps_proc->i4_mb_y == (ps_proc->i4_ht_mbs - 1)) 12594f5ceca078b8caab482bfd314bb24308ad872446Martin Storsjo num_rows = MB_SIZE - u4_pad_bottom_sz; 12604f5ceca078b8caab482bfd314bb24308ad872446Martin Storsjo for (i = 0; i < num_rows; i++) 12614f5ceca078b8caab482bfd314bb24308ad872446Martin Storsjo { 12624f5ceca078b8caab482bfd314bb24308ad872446Martin Storsjo memcpy(pu1_dst, pu1_src, ps_codec->s_cfg.u4_wd); 12634f5ceca078b8caab482bfd314bb24308ad872446Martin Storsjo pu1_src += ps_proc->s_inp_buf.s_raw_buf.au4_strd[0]; 12644f5ceca078b8caab482bfd314bb24308ad872446Martin Storsjo pu1_dst += ps_proc->i4_src_strd; 12654f5ceca078b8caab482bfd314bb24308ad872446Martin Storsjo } 1266cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar } 1267cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar pu1_src = (UWORD8 *)ps_proc->s_inp_buf.s_raw_buf.apv_bufs[1] + (i4_mb_x * BLK8x8SIZE) + 1268cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar ps_proc->s_inp_buf.s_raw_buf.au4_strd[1] * (i4_mb_y * BLK8x8SIZE); 1269cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar pu1_dst = ps_proc->pu1_src_buf_chroma; 1270cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar 1271cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar /* Last MB row of chroma is copied unconditionally, since trans functions access an extra byte 1272cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar * due to interleaved input 1273cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar */ 12744f5ceca078b8caab482bfd314bb24308ad872446Martin Storsjo if (ps_proc->i4_mb_y == (ps_proc->i4_ht_mbs - 1)) 12754f5ceca078b8caab482bfd314bb24308ad872446Martin Storsjo num_rows = (ps_codec->s_cfg.u4_disp_ht >> 1) - (ps_proc->i4_mb_y * BLK8x8SIZE); 12764f5ceca078b8caab482bfd314bb24308ad872446Martin Storsjo else 12774f5ceca078b8caab482bfd314bb24308ad872446Martin Storsjo num_rows = BLK8x8SIZE; 1278cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar for (i = 0; i < num_rows; i++) 1279cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar { 1280cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar memcpy(pu1_dst, pu1_src, ps_codec->s_cfg.u4_wd); 1281cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar pu1_src += ps_proc->s_inp_buf.s_raw_buf.au4_strd[1]; 12829113f5615cb4b2be179b288a59324de723d2ec9aMartin Storsjo pu1_dst += ps_proc->i4_src_chroma_strd; 1283cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar } 1284cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar 1285cf91c87b25ad49fc7e307932754d188e3ba2a479Harish Mahendrakar } 12868d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S break; 12878d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 12888d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S case IV_YUV_420P : 12898d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_y_buf_base = (UWORD8 *)ps_proc->s_inp_buf.s_raw_buf.apv_bufs[0] + (i4_mb_x * MB_SIZE) + 12908d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->s_inp_buf.s_raw_buf.au4_strd[0] * (i4_mb_y * MB_SIZE); 12918d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 12928d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_u_buf_base = (UWORD8 *)ps_proc->s_inp_buf.s_raw_buf.apv_bufs[1] + (i4_mb_x * BLK8x8SIZE) + 12938d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->s_inp_buf.s_raw_buf.au4_strd[1] * (i4_mb_y * BLK8x8SIZE); 12948d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 12958d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_v_buf_base = (UWORD8 *)ps_proc->s_inp_buf.s_raw_buf.apv_bufs[2] + (i4_mb_x * BLK8x8SIZE) + 12968d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->s_inp_buf.s_raw_buf.au4_strd[2] * (i4_mb_y * BLK8x8SIZE); 12978d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 12988d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_ih264e_conv_420p_to_420sp( 12998d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_y_buf_base, pu1_u_buf_base, pu1_v_buf_base, 13008d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->pu1_src_buf_luma, 13013a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar ps_proc->pu1_src_buf_chroma, u2_num_rows, 13023a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar ps_codec->s_cfg.u4_disp_wd, 13038d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->s_inp_buf.s_raw_buf.au4_strd[0], 13048d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->s_inp_buf.s_raw_buf.au4_strd[1], 13058d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->s_inp_buf.s_raw_buf.au4_strd[2], 13069113f5615cb4b2be179b288a59324de723d2ec9aMartin Storsjo ps_proc->i4_src_strd, ps_proc->i4_src_chroma_strd, 13073a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar convert_uv_only); 13088d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S break; 13098d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 13108d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S case IV_YUV_422ILE : 13118d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_y_buf_base = (UWORD8 *)ps_proc->s_inp_buf.s_raw_buf.apv_bufs[0] + (i4_mb_x * MB_SIZE * 2) 13128d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S + ps_proc->s_inp_buf.s_raw_buf.au4_strd[0] * (i4_mb_y * MB_SIZE); 13138d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 13148d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_ih264e_fmt_conv_422i_to_420sp( 13158d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->pu1_src_buf_luma, 13168d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->pu1_src_buf_chroma, 13178d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->pu1_src_buf_chroma + 1, pu1_y_buf_base, 13183a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar ps_codec->s_cfg.u4_disp_wd, u2_num_rows, 13199113f5615cb4b2be179b288a59324de723d2ec9aMartin Storsjo ps_proc->i4_src_strd, ps_proc->i4_src_chroma_strd, 13209113f5615cb4b2be179b288a59324de723d2ec9aMartin Storsjo ps_proc->i4_src_chroma_strd, 13218d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->s_inp_buf.s_raw_buf.au4_strd[0] >> 1); 13228d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S break; 13238d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 13248d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S default: 13258d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S break; 13268d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 13278d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 13284f5ceca078b8caab482bfd314bb24308ad872446Martin Storsjo if (u4_pad_right_sz && (ps_proc->i4_mb_x == 0)) 13293a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar { 13303a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar UWORD32 u4_pad_wd, u4_pad_ht; 13313a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar u4_pad_wd = (UWORD32)(ps_proc->i4_src_strd - ps_codec->s_cfg.u4_disp_wd); 13323a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar u4_pad_wd = MIN(u4_pad_right_sz, u4_pad_wd); 13333a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar u4_pad_ht = MB_SIZE; 13343a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar if(ps_proc->i4_mb_y == ps_proc->i4_ht_mbs - 1) 13353a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar u4_pad_ht = MIN(MB_SIZE, (MB_SIZE - u4_pad_bottom_sz)); 13363a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar 13373a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar ih264_pad_right_luma( 13383a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar ps_proc->pu1_src_buf_luma + ps_codec->s_cfg.u4_disp_wd, 13393a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar ps_proc->i4_src_strd, u4_pad_ht, u4_pad_wd); 13403a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar 13413a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar ih264_pad_right_chroma( 13423a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar ps_proc->pu1_src_buf_chroma + ps_codec->s_cfg.u4_disp_wd, 13439113f5615cb4b2be179b288a59324de723d2ec9aMartin Storsjo ps_proc->i4_src_chroma_strd, u4_pad_ht / 2, u4_pad_wd); 13443a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar } 13453a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar 13468d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* pad bottom edge */ 13478d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (u4_pad_bottom_sz && (ps_proc->i4_mb_y == ps_proc->i4_ht_mbs - 1) && ps_proc->i4_mb_x == 0) 13488d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 13498d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ih264_pad_bottom(ps_proc->pu1_src_buf_luma + (MB_SIZE - u4_pad_bottom_sz) * ps_proc->i4_src_strd, 13503a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar ps_proc->i4_src_strd, ps_proc->i4_src_strd, u4_pad_bottom_sz); 13518d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 13529113f5615cb4b2be179b288a59324de723d2ec9aMartin Storsjo ih264_pad_bottom(ps_proc->pu1_src_buf_chroma + (MB_SIZE - u4_pad_bottom_sz) * ps_proc->i4_src_chroma_strd / 2, 13539113f5615cb4b2be179b288a59324de723d2ec9aMartin Storsjo ps_proc->i4_src_chroma_strd, ps_proc->i4_src_chroma_strd, (u4_pad_bottom_sz / 2)); 13548d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 13558d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 13563a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar 13578d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* packed mb coeff data */ 13588d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->pv_mb_coeff_data = ((UWORD8 *)ps_proc->pv_pic_mb_coeff_data) + i4_mb_y * ps_codec->u4_size_coeff_data; 13598d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 13608d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* packed mb header data */ 13618d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->pv_mb_header_data = ((UWORD8 *)ps_proc->pv_pic_mb_header_data) + i4_mb_y * ps_codec->u4_size_header_data; 13628d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 13638d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* slice index */ 13648d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->i4_cur_slice_idx = ps_proc->pu1_slice_idx[i4_mb_y * ps_proc->i4_wd_mbs + i4_mb_x]; 13658d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 13668d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /*********************************************************************/ 13678d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* ih264e_init_quant_params() routine is called at the pic init level*/ 13688d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* this would have initialized the qp. */ 13698d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* TODO_LATER: currently it is assumed that quant params donot change*/ 13708d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* across mb's. When they do calculate update ps_qp_params accordingly*/ 13718d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /*********************************************************************/ 13728d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 13738d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* init mv buffer ptr */ 1374d3e18f0c8854c258021be1e603d8f50a241698a7Harinarayanan K K ps_proc->ps_pu = ps_cur_mv_buf->ps_pic_pu + (i4_mb_y * ps_proc->i4_wd_mbs * 1375d3e18f0c8854c258021be1e603d8f50a241698a7Harinarayanan K K ((MB_SIZE * MB_SIZE) / (ENC_MIN_PU_SIZE * ENC_MIN_PU_SIZE))); 13768d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 13773749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* Init co-located mv buffer */ 1378d3e18f0c8854c258021be1e603d8f50a241698a7Harinarayanan K K ps_proc->ps_colpu = ps_proc->aps_mv_buf[1]->ps_pic_pu + (i4_mb_y * ps_proc->i4_wd_mbs * 1379d3e18f0c8854c258021be1e603d8f50a241698a7Harinarayanan K K ((MB_SIZE * MB_SIZE) / (ENC_MIN_PU_SIZE * ENC_MIN_PU_SIZE))); 13803749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 13818d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (i4_mb_y == 0) 13828d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 13838d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->ps_top_row_pu_ME = ps_cur_mv_buf->ps_pic_pu; 13848d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 13858d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S else 13868d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 1387d3e18f0c8854c258021be1e603d8f50a241698a7Harinarayanan K K ps_proc->ps_top_row_pu_ME = ps_cur_mv_buf->ps_pic_pu + ((i4_mb_y - 1) * ps_proc->i4_wd_mbs * 1388d3e18f0c8854c258021be1e603d8f50a241698a7Harinarayanan K K ((MB_SIZE * MB_SIZE) / (ENC_MIN_PU_SIZE * ENC_MIN_PU_SIZE))); 13898d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 13908d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 13918d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->pu4_mb_pu_cnt = ps_cur_mv_buf->pu4_mb_pu_cnt + (i4_mb_y * ps_proc->i4_wd_mbs); 13928d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 13938d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* mb type */ 13948d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->u4_mb_type = I16x16; 13958d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 13968d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* lambda */ 13978d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->u4_lambda = gu1_qp0[ps_qp_params->u1_mb_qp]; 13988d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 13998d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* mb distortion */ 14008d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->i4_mb_distortion = SHRT_MAX; 14018d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 14028d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (i4_mb_x == 0) 14038d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 14048d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->s_left_mb_syntax_ele.i4_mb_distortion = 0; 14058d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 14068d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->s_top_left_mb_syntax_ele.i4_mb_distortion = 0; 14078d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 14088d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->s_top_left_mb_syntax_ME.i4_mb_distortion = 0; 14098d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 14108d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (i4_mb_y == 0) 14118d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 14128d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S memset(ps_proc->ps_top_row_mb_syntax_ele, 0, (ps_proc->i4_wd_mbs + 1)*sizeof(mb_info_t)); 14138d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 14148d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 14158d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 14168d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* mb cost */ 14178d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->i4_mb_cost = INT_MAX; 14188d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 14198d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /**********************/ 14208d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* init deblk context */ 14218d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /**********************/ 14228d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_deblk->i4_mb_x = ps_proc->i4_mb_x; 14238d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* deblk lags the current mb proc by 1 row */ 14248d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* NOTE: Intra prediction has to happen with non deblocked samples used as reference */ 14258d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* Hence to deblk MB 0 of row 0, you have wait till MB 0 of row 1 is encoded. */ 14268d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* For simplicity, we chose to lag deblking by 1 Row wrt to proc */ 14278d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_deblk->i4_mb_y = ps_proc->i4_mb_y - 1; 14288d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 14298d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* buffer ptrs */ 14308d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_deblk->pu1_cur_pic_luma = ps_proc->pu1_rec_buf_luma_base + i4_rec_strd * (ps_deblk->i4_mb_y * MB_SIZE); 14318d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_deblk->pu1_cur_pic_chroma = ps_proc->pu1_rec_buf_chroma_base + i4_rec_strd * (ps_deblk->i4_mb_y * BLK8x8SIZE); 14328d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 14338d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* init deblk bs context */ 14348d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* mb indices */ 14358d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_bs->i4_mb_x = ps_proc->i4_mb_x; 14368d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_bs->i4_mb_y = ps_proc->i4_mb_y; 14378d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 14388d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* init n_mb_process context */ 14398d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_n_mb_ctxt->i4_mb_x = 0; 14408d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_n_mb_ctxt->i4_mb_y = ps_deblk->i4_mb_y; 14418d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_n_mb_ctxt->i4_n_mbs = ps_proc->i4_nmb_ntrpy; 14428d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 14438d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S return IH264E_SUCCESS; 14448d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S} 14458d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 14468d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S/** 14478d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S******************************************************************************* 14488d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 14498d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @brief This function performs luma & chroma padding 14508d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 14518d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @par Description: 14528d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 14538d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @param[in] ps_proc 14548d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* Process context corresponding to the job 14558d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 14568d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @param[in] pu1_curr_pic_luma 14578d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* Pointer to luma buffer 14588d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 14598d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @param[in] pu1_curr_pic_chroma 14608d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* Pointer to chroma buffer 14618d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 14628d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @param[in] i4_mb_x 14638d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* mb index x 14648d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 14658d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @param[in] i4_mb_y 14668d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* mb index y 14678d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 14688d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @param[in] i4_pad_ht 14698d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* number of rows to be padded 14708d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 14718d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @returns error status 14728d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 14738d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @remarks none 14748d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 14758d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S******************************************************************************* 14768d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S*/ 14778d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha SIH264E_ERROR_T ih264e_pad_recon_buffer(process_ctxt_t *ps_proc, 14788d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD8 *pu1_curr_pic_luma, 14798d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD8 *pu1_curr_pic_chroma, 14808d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_mb_x, 14818d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_mb_y, 14828d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_pad_ht) 14838d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S{ 14848d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* codec context */ 14858d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S codec_t *ps_codec = ps_proc->ps_codec; 14868d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 14878d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* strides */ 14888d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_rec_strd = ps_proc->i4_rec_strd; 14898d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 14908d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (i4_mb_x == 0) 14918d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 14928d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding left luma */ 14938d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_left_luma(pu1_curr_pic_luma, i4_rec_strd, i4_pad_ht, PAD_LEFT); 14948d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 14958d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding left chroma */ 14968d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_left_chroma(pu1_curr_pic_chroma, i4_rec_strd, i4_pad_ht >> 1, PAD_LEFT); 14978d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 14983a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar if (i4_mb_x == ps_proc->i4_wd_mbs - 1) 14998d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 15008d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding right luma */ 15018d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_right_luma(pu1_curr_pic_luma + MB_SIZE, i4_rec_strd, i4_pad_ht, PAD_RIGHT); 15028d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 15038d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding right chroma */ 15048d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_right_chroma(pu1_curr_pic_chroma + MB_SIZE, i4_rec_strd, i4_pad_ht >> 1, PAD_RIGHT); 15058d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 15068d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (i4_mb_y == ps_proc->i4_ht_mbs - 1) 15078d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 15088d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD8 *pu1_rec_luma = pu1_curr_pic_luma + MB_SIZE + PAD_RIGHT + ((i4_pad_ht - 1) * i4_rec_strd); 15098d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD8 *pu1_rec_chroma = pu1_curr_pic_chroma + MB_SIZE + PAD_RIGHT + (((i4_pad_ht >> 1) - 1) * i4_rec_strd); 15108d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 15118d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding bottom luma */ 15128d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_bottom(pu1_rec_luma, i4_rec_strd, i4_rec_strd, PAD_BOT); 15138d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 15148d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding bottom chroma */ 15158d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_bottom(pu1_rec_chroma, i4_rec_strd, i4_rec_strd, (PAD_BOT >> 1)); 15168d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 15178d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 15188d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 15198d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (i4_mb_y == 0) 15208d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 15218d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD8 *pu1_rec_luma = pu1_curr_pic_luma; 15228d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD8 *pu1_rec_chroma = pu1_curr_pic_chroma; 15238d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 wd = MB_SIZE; 15248d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 15258d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (i4_mb_x == 0) 15268d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 15278d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_rec_luma -= PAD_LEFT; 15288d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_rec_chroma -= PAD_LEFT; 15298d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 15308d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S wd += PAD_LEFT; 15318d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 15323a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar if (i4_mb_x == ps_proc->i4_wd_mbs - 1) 15338d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 15348d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S wd += PAD_RIGHT; 15358d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 15368d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 15378d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding top luma */ 15388d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_top(pu1_rec_luma, i4_rec_strd, wd, PAD_TOP); 15398d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 15408d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding top chroma */ 15418d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_top(pu1_rec_chroma, i4_rec_strd, wd, (PAD_TOP >> 1)); 15428d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 15438d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 15448d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S return IH264E_SUCCESS; 15458d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S} 15468d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 15478d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 15488d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 15498d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 15508d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S/** 15518d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S******************************************************************************* 15528d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 15538d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @brief This function performs deblocking, padding and halfpel generation for 15548d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 'n' MBs 15558d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 15568d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @par Description: 15578d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 15588d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @param[in] ps_proc 15598d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* Process context corresponding to the job 15608d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 15618d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @param[in] pu1_curr_pic_luma 15628d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* Current MB being processed(Luma) 15638d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 15648d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @param[in] pu1_curr_pic_chroma 15658d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* Current MB being processed(Chroma) 15668d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 15678d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @param[in] i4_mb_x 15688d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* Column value of current MB processed 15698d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 15708d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @param[in] i4_mb_y 15718d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* Curent row processed 15728d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 15738d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @returns error status 15748d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 15758d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @remarks none 15768d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 15778d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S******************************************************************************* 15788d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S*/ 15798d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha SIH264E_ERROR_T ih264e_dblk_pad_hpel_processing_n_mbs(process_ctxt_t *ps_proc, 15808d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD8 *pu1_curr_pic_luma, 15818d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD8 *pu1_curr_pic_chroma, 15828d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_mb_x, 15838d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_mb_y) 15848d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S{ 15858d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* codec context */ 15868d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S codec_t *ps_codec = ps_proc->ps_codec; 15878d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 15888d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* n_mb processing context */ 15898d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S n_mb_process_ctxt_t *ps_n_mb_ctxt = &ps_proc->s_n_mb_ctxt; 15908d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 15918d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* deblk context */ 15928d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S deblk_ctxt_t *ps_deblk = &ps_proc->s_deblk_ctxt; 15938d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 15948d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* strides */ 15958d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_rec_strd = ps_proc->i4_rec_strd; 15968d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 15978d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* loop variables */ 15988d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 row, i, j, col; 15998d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 16008d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* Padding Width */ 16018d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD32 u4_pad_wd; 16028d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 16038d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* deblk_map of the row being deblocked */ 16048d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD8 *pu1_deblk_map = ps_proc->pu1_deblk_map + ps_deblk->i4_mb_y * ps_proc->i4_wd_mbs; 16058d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 16068d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* deblk_map_previous row */ 16078d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD8 *pu1_deblk_map_prev_row = pu1_deblk_map - ps_proc->i4_wd_mbs; 16088d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 16098d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 u4_pad_top = 0; 16108d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 16118d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 u4_deblk_prev_row = 0; 16128d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 16138d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* Number of mbs to be processed */ 16148d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_n_mbs = ps_n_mb_ctxt->i4_n_mbs; 16158d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 16168d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* Number of mbs actually processed 16178d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * (at the end of a row, when remaining number of MBs are less than i4_n_mbs) */ 16188d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_n_mb_process_count = 0; 16198d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 16208d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD8 *pu1_pad_bottom_src = NULL; 16218d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 16228d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD8 *pu1_pad_src_luma = NULL; 16238d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD8 *pu1_pad_src_chroma = NULL; 16248d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 16258d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_proc->u4_disable_deblock_level == 1) 16268d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 16278d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* If left most MB is processed, then pad left */ 16288d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (i4_mb_x == 0) 16298d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 16308d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding left luma */ 16318d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_left_luma(pu1_curr_pic_luma, i4_rec_strd, MB_SIZE, PAD_LEFT); 16328d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 16338d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding left chroma */ 16348d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_left_chroma(pu1_curr_pic_chroma, i4_rec_strd, MB_SIZE >> 1, PAD_LEFT); 16358d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 16368d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /*last col*/ 16378d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (i4_mb_x == (ps_proc->i4_wd_mbs - 1)) 16388d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 16398d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding right luma */ 16408d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_right_luma(pu1_curr_pic_luma + MB_SIZE, i4_rec_strd, MB_SIZE, PAD_RIGHT); 16418d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 16428d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding right chroma */ 16438d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_right_chroma(pu1_curr_pic_chroma + MB_SIZE, i4_rec_strd, MB_SIZE >> 1, PAD_RIGHT); 16448d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 16458d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 16468d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 16473a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar if ((i4_mb_y > 0) || (i4_mb_y == (ps_proc->i4_ht_mbs - 1))) 16488d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 16498d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* if number of mb's to be processed are less than 'N', go back. 16508d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * exception to the above clause is end of row */ 16518d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if ( ((i4_mb_x - (ps_n_mb_ctxt->i4_mb_x - 1)) < i4_n_mbs) && (i4_mb_x < (ps_proc->i4_wd_mbs - 1)) ) 16528d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 16538d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S return IH264E_SUCCESS; 16548d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 16558d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S else 16568d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 16578d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S i4_n_mb_process_count = MIN(i4_mb_x - (ps_n_mb_ctxt->i4_mb_x - 1), i4_n_mbs); 16588d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 16598d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* performing deblocking for required number of MBs */ 16603a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar if ((i4_mb_y > 0) && (ps_proc->u4_disable_deblock_level != 1)) 16618d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 16621ff75b8cc0e9c5bc56917c050c6d472fc67bc6f9Martin Storsjo u4_deblk_prev_row = 1; 16631ff75b8cc0e9c5bc56917c050c6d472fc67bc6f9Martin Storsjo 16641ff75b8cc0e9c5bc56917c050c6d472fc67bc6f9Martin Storsjo /* checking whether the top rows are deblocked */ 16651ff75b8cc0e9c5bc56917c050c6d472fc67bc6f9Martin Storsjo for (col = 0; col < i4_n_mb_process_count; col++) 16661ff75b8cc0e9c5bc56917c050c6d472fc67bc6f9Martin Storsjo { 16671ff75b8cc0e9c5bc56917c050c6d472fc67bc6f9Martin Storsjo u4_deblk_prev_row &= pu1_deblk_map_prev_row[ps_deblk->i4_mb_x + col]; 16681ff75b8cc0e9c5bc56917c050c6d472fc67bc6f9Martin Storsjo } 16691ff75b8cc0e9c5bc56917c050c6d472fc67bc6f9Martin Storsjo 16701ff75b8cc0e9c5bc56917c050c6d472fc67bc6f9Martin Storsjo /* checking whether the top right MB is deblocked */ 16711ff75b8cc0e9c5bc56917c050c6d472fc67bc6f9Martin Storsjo if ((ps_deblk->i4_mb_x + i4_n_mb_process_count) != ps_proc->i4_wd_mbs) 16721ff75b8cc0e9c5bc56917c050c6d472fc67bc6f9Martin Storsjo { 16731ff75b8cc0e9c5bc56917c050c6d472fc67bc6f9Martin Storsjo u4_deblk_prev_row &= pu1_deblk_map_prev_row[ps_deblk->i4_mb_x + i4_n_mb_process_count]; 16741ff75b8cc0e9c5bc56917c050c6d472fc67bc6f9Martin Storsjo } 16751ff75b8cc0e9c5bc56917c050c6d472fc67bc6f9Martin Storsjo 16768d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* Top or Top right MBs not deblocked */ 16773a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar if ((u4_deblk_prev_row != 1) && (i4_mb_y > 0)) 16788d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 16798d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S return IH264E_SUCCESS; 16808d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 16818d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 16828d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S for (row = 0; row < i4_n_mb_process_count; row++) 16838d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 16848d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ih264e_deblock_mb(ps_proc, ps_deblk); 16858d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 16868d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_deblk_map[ps_deblk->i4_mb_x] = 1; 16878d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 16888d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_deblk->i4_mb_y > 0) 16898d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 16908d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_deblk->i4_mb_x == 0)/* If left most MB is processed, then pad left*/ 16918d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 16928d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding left luma */ 16938d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_left_luma(ps_deblk->pu1_cur_pic_luma - i4_rec_strd * MB_SIZE, i4_rec_strd, MB_SIZE, PAD_LEFT); 16948d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 16958d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding left chroma */ 16968d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_left_chroma(ps_deblk->pu1_cur_pic_chroma - i4_rec_strd * BLK8x8SIZE, i4_rec_strd, MB_SIZE >> 1, PAD_LEFT); 16978d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 16988d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 16998d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_deblk->i4_mb_x == (ps_proc->i4_wd_mbs - 1))/*last column*/ 17008d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 17018d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding right luma */ 17028d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_right_luma(ps_deblk->pu1_cur_pic_luma - i4_rec_strd * MB_SIZE + MB_SIZE, i4_rec_strd, MB_SIZE, PAD_RIGHT); 17038d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 17048d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding right chroma */ 17058d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_right_chroma(ps_deblk->pu1_cur_pic_chroma - i4_rec_strd * BLK8x8SIZE + MB_SIZE, i4_rec_strd, MB_SIZE >> 1, PAD_RIGHT); 17068d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 17078d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 17088d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_deblk->i4_mb_x++; 17098d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 17108d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_deblk->pu1_cur_pic_luma += MB_SIZE; 17118d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_deblk->pu1_cur_pic_chroma += MB_SIZE; 17128d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 17138d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 17148d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 17153a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar else if(i4_mb_y > 0) 17168d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 17178d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_deblk->i4_mb_x += i4_n_mb_process_count; 17188d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 17198d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_deblk->pu1_cur_pic_luma += i4_n_mb_process_count * MB_SIZE; 17208d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_deblk->pu1_cur_pic_chroma += i4_n_mb_process_count * MB_SIZE; 17218d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 17228d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 17238d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (i4_mb_y == 2) 17248d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 17258d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S u4_pad_wd = i4_n_mb_process_count * MB_SIZE; 17268d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S u4_pad_top = ps_n_mb_ctxt->i4_mb_x * MB_SIZE; 17278d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 17288d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_n_mb_ctxt->i4_mb_x == 0) 17298d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 17308d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S u4_pad_wd += PAD_LEFT; 17318d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S u4_pad_top = -PAD_LEFT; 17328d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 17338d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 17348d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (i4_mb_x == ps_proc->i4_wd_mbs - 1) 17358d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 17368d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S u4_pad_wd += PAD_RIGHT; 17378d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 17388d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 17398d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding top luma */ 17408d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_top(ps_proc->pu1_rec_buf_luma_base + u4_pad_top, i4_rec_strd, u4_pad_wd, PAD_TOP); 17418d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 17428d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding top chroma */ 17438d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_top(ps_proc->pu1_rec_buf_chroma_base + u4_pad_top, i4_rec_strd, u4_pad_wd, (PAD_TOP >> 1)); 17448d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 17458d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 17468d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_n_mb_ctxt->i4_mb_x += i4_n_mb_process_count; 17478d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 17488d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (i4_mb_x == ps_proc->i4_wd_mbs - 1) 17498d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 17508d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_proc->i4_mb_y == ps_proc->i4_ht_mbs - 1) 17518d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 17528d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* Bottom Padding is done in one stretch for the entire width */ 17538d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_proc->u4_disable_deblock_level != 1) 17548d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 17558d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_deblk->pu1_cur_pic_luma = ps_proc->pu1_rec_buf_luma_base + (ps_proc->i4_ht_mbs - 1) * i4_rec_strd * MB_SIZE; 17568d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 17578d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_deblk->pu1_cur_pic_chroma = ps_proc->pu1_rec_buf_chroma_base + (ps_proc->i4_ht_mbs - 1) * i4_rec_strd * BLK8x8SIZE; 17588d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 17598d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_n_mb_ctxt->i4_mb_x = 0; 17608d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_n_mb_ctxt->i4_mb_y = ps_proc->i4_mb_y; 17618d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_deblk->i4_mb_x = 0; 17628d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_deblk->i4_mb_y = ps_proc->i4_mb_y; 17638d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 17648d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* update pic qp map (as update_proc_ctxt is still not called for the last MB) */ 17658d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->s_deblk_ctxt.s_bs_ctxt.pu1_pic_qp[(i4_mb_y * ps_proc->i4_wd_mbs) + i4_mb_x] = ps_proc->u4_mb_qp; 17668d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 17678d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S i4_n_mb_process_count = (ps_proc->i4_wd_mbs) % i4_n_mbs; 17688d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 17698d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S j = (ps_proc->i4_wd_mbs) / i4_n_mbs; 17708d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 17718d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S for (i = 0; i < j; i++) 17728d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 17738d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S for (col = 0; col < i4_n_mbs; col++) 17748d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 17758d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ih264e_deblock_mb(ps_proc, ps_deblk); 17768d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 17778d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_deblk_map[ps_deblk->i4_mb_x] = 1; 17788d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 17798d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_deblk->i4_mb_x++; 17808d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_deblk->pu1_cur_pic_luma += MB_SIZE; 17818d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_deblk->pu1_cur_pic_chroma += MB_SIZE; 17828d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_n_mb_ctxt->i4_mb_x++; 17838d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 17848d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 17858d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 17868d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S for (col = 0; col < i4_n_mb_process_count; col++) 17878d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 17888d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ih264e_deblock_mb(ps_proc, ps_deblk); 17898d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 17908d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_deblk_map[ps_deblk->i4_mb_x] = 1; 17918d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 17928d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_deblk->i4_mb_x++; 17938d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_deblk->pu1_cur_pic_luma += MB_SIZE; 17948d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_deblk->pu1_cur_pic_chroma += MB_SIZE; 17958d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_n_mb_ctxt->i4_mb_x++; 17968d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 17978d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 17988d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_pad_src_luma = ps_proc->pu1_rec_buf_luma_base + (ps_proc->i4_ht_mbs - 2) * MB_SIZE * i4_rec_strd; 17998d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 18008d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_pad_src_chroma = ps_proc->pu1_rec_buf_chroma_base + (ps_proc->i4_ht_mbs - 2) * BLK8x8SIZE * i4_rec_strd; 18018d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 18028d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding left luma */ 18038d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_left_luma(pu1_pad_src_luma, i4_rec_strd, MB_SIZE, PAD_LEFT); 18048d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 18058d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding left chroma */ 18068d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_left_chroma(pu1_pad_src_chroma, i4_rec_strd, BLK8x8SIZE, PAD_LEFT); 18078d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 18088d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_pad_src_luma += i4_rec_strd * MB_SIZE; 18098d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_pad_src_chroma += i4_rec_strd * BLK8x8SIZE; 18108d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 18118d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding left luma */ 18128d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_left_luma(pu1_pad_src_luma, i4_rec_strd, MB_SIZE, PAD_LEFT); 18138d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 18148d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding left chroma */ 18158d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_left_chroma(pu1_pad_src_chroma, i4_rec_strd, BLK8x8SIZE, PAD_LEFT); 18168d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 18178d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_pad_src_luma = ps_proc->pu1_rec_buf_luma_base + (ps_proc->i4_ht_mbs - 2) * MB_SIZE * i4_rec_strd + (ps_proc->i4_wd_mbs) * MB_SIZE; 18188d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 18198d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_pad_src_chroma = ps_proc->pu1_rec_buf_chroma_base + (ps_proc->i4_ht_mbs - 2) * BLK8x8SIZE * i4_rec_strd + (ps_proc->i4_wd_mbs) * MB_SIZE; 18208d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 18218d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding right luma */ 18228d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_right_luma(pu1_pad_src_luma, i4_rec_strd, MB_SIZE, PAD_RIGHT); 18238d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 18248d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding right chroma */ 18258d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_right_chroma(pu1_pad_src_chroma, i4_rec_strd, BLK8x8SIZE, PAD_RIGHT); 18268d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 18278d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_pad_src_luma += i4_rec_strd * MB_SIZE; 18288d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_pad_src_chroma += i4_rec_strd * BLK8x8SIZE; 18298d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 18308d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding right luma */ 18318d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_right_luma(pu1_pad_src_luma, i4_rec_strd, MB_SIZE, PAD_RIGHT); 18328d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 18338d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding right chroma */ 18348d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_right_chroma(pu1_pad_src_chroma, i4_rec_strd, BLK8x8SIZE, PAD_RIGHT); 18358d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 18368d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 18378d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 18383a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar /* In case height is less than 2 MBs pad top */ 18393a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar if (ps_proc->i4_ht_mbs <= 2) 18403a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar { 18413a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar UWORD8 *pu1_pad_top_src; 18423a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar /* padding top luma */ 18433a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar pu1_pad_top_src = ps_proc->pu1_rec_buf_luma_base - PAD_LEFT; 18443a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar ps_codec->pf_pad_top(pu1_pad_top_src, i4_rec_strd, i4_rec_strd, PAD_TOP); 18453a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar 18463a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar /* padding top chroma */ 18473a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar pu1_pad_top_src = ps_proc->pu1_rec_buf_chroma_base - PAD_LEFT; 18483a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar ps_codec->pf_pad_top(pu1_pad_top_src, i4_rec_strd, i4_rec_strd, (PAD_TOP >> 1)); 18493a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar } 18503a52efd36102c9a4cb8e8f23b6f9131b8b01eef5Harish Mahendrakar 18518d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding bottom luma */ 18528d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_pad_bottom_src = ps_proc->pu1_rec_buf_luma_base + ps_proc->i4_ht_mbs * MB_SIZE * i4_rec_strd - PAD_LEFT; 18538d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_bottom(pu1_pad_bottom_src, i4_rec_strd, i4_rec_strd, PAD_BOT); 18548d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 18558d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* padding bottom chroma */ 18568d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_pad_bottom_src = ps_proc->pu1_rec_buf_chroma_base + ps_proc->i4_ht_mbs * (MB_SIZE >> 1) * i4_rec_strd - PAD_LEFT; 18578d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->pf_pad_bottom(pu1_pad_bottom_src, i4_rec_strd, i4_rec_strd, (PAD_BOT >> 1)); 18588d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 18598d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 18608d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 18618d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 18628d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 18638d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S return IH264E_SUCCESS; 18648d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S} 18658d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 18668d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 18678d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S/** 18688d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S******************************************************************************* 18698d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 18708d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @brief This function performs luma & chroma core coding for a set of mb's. 18718d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 18728d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @par Description: 18738d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* The mb to be coded is taken and is evaluated over a predefined set of modes 18748d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* (intra (i16, i4, i8)/inter (mv, skip)) for best cost. The mode with least cost 18758d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* is selected and using intra/inter prediction filters, prediction is carried out. 18768d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* The deviation between src and pred signal constitutes error signal. This error 18778d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* signal is transformed (hierarchical transform if necessary) and quantized. The 18788d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* quantized residue is packed in to entropy buffer for entropy coding. This is 18798d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* repeated for all the mb's enlisted under the job. 18808d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 18818d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @param[in] ps_proc 18828d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* Process context corresponding to the job 18838d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 18848d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @returns error status 18858d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 18868d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @remarks none 18878d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 18888d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S******************************************************************************* 18898d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S*/ 18908d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha SWORD32 ih264e_process(process_ctxt_t *ps_proc) 18918d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S{ 18928d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* error status */ 18938d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 error_status = IH264_SUCCESS; 18948d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 18958d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* codec context */ 18968d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S codec_t *ps_codec = ps_proc->ps_codec; 18978d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 18988d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* cbp luma, chroma */ 18998d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD32 u4_cbp_l, u4_cbp_c; 19008d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 19018d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* width in mbs */ 19028d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_wd_mbs = ps_proc->i4_wd_mbs; 19038d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 19048d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* loop var */ 19058d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_mb_idx, i4_mb_cnt = ps_proc->i4_mb_cnt; 19068d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 19078d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* valid modes */ 19088d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD32 u4_valid_modes = 0; 19098d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 19108d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* gate threshold */ 19118d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_gate_threshold = 0; 19128d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 19138d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* is intra */ 19148d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 luma_idx, chroma_idx, is_intra; 19158d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 19168d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* temp variables */ 1917d3e18f0c8854c258021be1e603d8f50a241698a7Harinarayanan K K WORD32 ctxt_sel = ps_proc->i4_encode_api_call_cnt % MAX_CTXT_SETS; 19188d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 19197dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K /* 19207dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K * list of modes for evaluation 19217dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K * ------------------------------------------------------------------------- 19227dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K * Note on enabling I4x4 and I16x16 19237dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K * At very low QP's the hadamard transform in I16x16 will push up the maximum 19247dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K * coeff value very high. CAVLC may not be able to represent the value and 19257dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K * hence the stream may not be decodable in some clips. 19267dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K * Hence at low QPs, we will enable I4x4 and disable I16x16 irrespective of preset. 19277dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K */ 19288d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_proc->i4_slice_type == ISLICE) 19298d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 19307dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K if (ps_proc->u4_frame_qp > 10) 19317dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K { 19327dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K /* enable intra 16x16 */ 19337dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K u4_valid_modes |= ps_codec->s_cfg.u4_enable_intra_16x16 ? (1 << I16x16) : 0; 19348d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 19357dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K /* enable intra 8x8 */ 19367dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K u4_valid_modes |= ps_codec->s_cfg.u4_enable_intra_8x8 ? (1 << I8x8) : 0; 19377dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K } 19388d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 19398d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* enable intra 4x4 */ 19408d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S u4_valid_modes |= ps_codec->s_cfg.u4_enable_intra_4x4 ? (1 << I4x4) : 0; 19417dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K u4_valid_modes |= (ps_proc->u4_frame_qp <= 10) << I4x4; 19427dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K 19438d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 19448d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S else if (ps_proc->i4_slice_type == PSLICE) 19458d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 19467dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K if (ps_proc->u4_frame_qp > 10) 19477dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K { 19487dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K /* enable intra 16x16 */ 19497dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K u4_valid_modes |= ps_codec->s_cfg.u4_enable_intra_16x16 ? (1 << I16x16) : 0; 19507dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K } 19518d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 19528d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* enable intra 4x4 */ 19538d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_codec->s_cfg.u4_enc_speed_preset == IVE_SLOWEST) 19548d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 19558d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S u4_valid_modes |= ps_codec->s_cfg.u4_enable_intra_4x4 ? (1 << I4x4) : 0; 19568d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 19577dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K u4_valid_modes |= (ps_proc->u4_frame_qp <= 10) << I4x4; 19588d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 19593749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* enable inter P16x16 */ 19608d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S u4_valid_modes |= (1 << P16x16); 19618d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 19623749f6f435e79624f72841e866245d84195551cdHarinarayanan K K else if (ps_proc->i4_slice_type == BSLICE) 19633749f6f435e79624f72841e866245d84195551cdHarinarayanan K K { 19647dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K if (ps_proc->u4_frame_qp > 10) 19657dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K { 19667dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K /* enable intra 16x16 */ 19677dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K u4_valid_modes |= ps_codec->s_cfg.u4_enable_intra_16x16 ? (1 << I16x16) : 0; 19687dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K } 19693749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 19703749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* enable intra 4x4 */ 19713749f6f435e79624f72841e866245d84195551cdHarinarayanan K K if (ps_codec->s_cfg.u4_enc_speed_preset == IVE_SLOWEST) 19723749f6f435e79624f72841e866245d84195551cdHarinarayanan K K { 19733749f6f435e79624f72841e866245d84195551cdHarinarayanan K K u4_valid_modes |= ps_codec->s_cfg.u4_enable_intra_4x4 ? (1 << I4x4) : 0; 19743749f6f435e79624f72841e866245d84195551cdHarinarayanan K K } 19757dbacdf7651457dfc6244ed96b797e93a6944680Harinarayanan K K u4_valid_modes |= (ps_proc->u4_frame_qp <= 10) << I4x4; 19763749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 19773749f6f435e79624f72841e866245d84195551cdHarinarayanan K K /* enable inter B16x16 */ 19783749f6f435e79624f72841e866245d84195551cdHarinarayanan K K u4_valid_modes |= (1 << B16x16); 19793749f6f435e79624f72841e866245d84195551cdHarinarayanan K K } 19808d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 19818d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 19828d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* init entropy */ 19838d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->s_entropy.i4_mb_x = ps_proc->i4_mb_x; 19848d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->s_entropy.i4_mb_y = ps_proc->i4_mb_y; 19858d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->s_entropy.i4_mb_cnt = MIN(ps_proc->i4_nmb_ntrpy, i4_wd_mbs - ps_proc->i4_mb_x); 19868d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 19878d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* compute recon when : 19888d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * 1. current frame is to be used as a reference 19898d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * 2. dump recon for bit stream sanity check 19908d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S */ 19918d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->u4_compute_recon = ps_codec->u4_is_curr_frm_ref || 19928d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->s_cfg.u4_enable_recon; 19938d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 19948d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* Encode 'n' macroblocks, 19958d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * 'n' being the number of mbs dictated by current proc ctxt */ 19968d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S for (i4_mb_idx = 0; i4_mb_idx < i4_mb_cnt; i4_mb_idx ++) 19978d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 19988d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* since we have not yet found sad, we have not yet got min sad */ 19998d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* we need to initialize these variables for each MB */ 20008d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* TODO how to get the min sad into the codec */ 20018d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->u4_min_sad = ps_codec->s_cfg.i4_min_sad; 20028d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->u4_min_sad_reached = 0; 20038d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 20048d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* mb analysis */ 20058d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 20068d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* temp var */ 20078d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_mb_id = ps_proc->i4_mb_x + ps_proc->i4_mb_y * i4_wd_mbs; 20088d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 20098d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* force intra refresh ? */ 20108d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_air_enable_inter = (ps_codec->s_cfg.e_air_mode == IVE_AIR_MODE_NONE) || 20118d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S (ps_proc->pu1_is_intra_coded[i4_mb_id] != 0) || 20128d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S (ps_codec->pu2_intr_rfrsh_map[i4_mb_id] != ps_codec->i4_air_pic_cnt); 20138d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 20148d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* evaluate inter 16x16 modes */ 20153749f6f435e79624f72841e866245d84195551cdHarinarayanan K K if ((u4_valid_modes & (1 << P16x16)) || (u4_valid_modes & (1 << B16x16))) 20168d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 20178d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* compute nmb me */ 20188d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_proc->i4_mb_x % ps_proc->u4_nmb_me == 0) 20198d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 20208d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ih264e_compute_me_nmb(ps_proc, MIN((WORD32)ps_proc->u4_nmb_me, 20218d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S i4_wd_mbs - ps_proc->i4_mb_x)); 20228d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 20238d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 20248d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* set pointers to ME data appropriately for other modules to use */ 20258d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 20268d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD32 u4_mb_index = ps_proc->i4_mb_x % ps_proc->u4_nmb_me ; 20278d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 20288d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* get the min sad condition for current mb */ 20298d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->u4_min_sad_reached = ps_proc->ps_nmb_info[u4_mb_index].u4_min_sad_reached; 20308d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->u4_min_sad = ps_proc->ps_nmb_info[u4_mb_index].u4_min_sad; 20318d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 20323749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_proc->ps_skip_mv = &(ps_proc->ps_nmb_info[u4_mb_index].as_skip_mv[0]); 20338d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->ps_ngbr_avbl = &(ps_proc->ps_nmb_info[u4_mb_index].s_ngbr_avbl); 20343749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_proc->ps_pred_mv = &(ps_proc->ps_nmb_info[u4_mb_index].as_pred_mv[0]); 20358d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 20368d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->i4_mb_distortion = ps_proc->ps_nmb_info[u4_mb_index].i4_mb_distortion; 20378d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->i4_mb_cost = ps_proc->ps_nmb_info[u4_mb_index].i4_mb_cost; 20388d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->u4_min_sad = ps_proc->ps_nmb_info[u4_mb_index].u4_min_sad; 20398d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->u4_min_sad_reached = ps_proc->ps_nmb_info[u4_mb_index].u4_min_sad_reached; 20408d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->u4_mb_type = ps_proc->ps_nmb_info[u4_mb_index].u4_mb_type; 20418d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 20428d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* get the best sub pel buffer */ 20438d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->pu1_best_subpel_buf = ps_proc->ps_nmb_info[u4_mb_index].pu1_best_sub_pel_buf; 20448d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->u4_bst_spel_buf_strd = ps_proc->ps_nmb_info[u4_mb_index].u4_bst_spel_buf_strd; 20458d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 20468d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ih264e_derive_nghbr_avbl_of_mbs(ps_proc); 20478d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 20488d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S else 20498d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 20508d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* Derive neighbor availability for the current macroblock */ 20518d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->ps_ngbr_avbl = &ps_proc->s_ngbr_avbl; 20528d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 20538d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ih264e_derive_nghbr_avbl_of_mbs(ps_proc); 20548d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 20558d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 20568d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* 20578d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * If air says intra, we need to force the following code path to evaluate intra 20588d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * The easy way is just to say that the inter cost is too much 20598d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S */ 20608d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (!i4_air_enable_inter) 20618d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 20628d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->u4_min_sad_reached = 0; 20638d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->i4_mb_cost = INT_MAX; 20648d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->i4_mb_distortion = INT_MAX; 20658d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 20668d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S else if (ps_proc->u4_mb_type == PSKIP) 20678d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 20688d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S goto UPDATE_MB_INFO; 20698d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 20708d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 20718d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* wait until the proc of [top + 1] mb is computed. 20728d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * We wait till the proc dependencies are satisfied */ 20738d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if(ps_proc->i4_mb_y > 0) 20748d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 20758d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* proc map */ 20768d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD8 *pu1_proc_map_top; 20778d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 20788d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_proc_map_top = ps_proc->pu1_proc_map + ((ps_proc->i4_mb_y - 1) * i4_wd_mbs); 20798d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 20808d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S while (1) 20818d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 20828d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S volatile UWORD8 *pu1_buf; 20838d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 idx = i4_mb_idx + 1; 20848d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 20858d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S idx = MIN(idx, ((WORD32)ps_codec->s_cfg.i4_wd_mbs - 1)); 20868d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_buf = pu1_proc_map_top + idx; 20878d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if(*pu1_buf) 20888d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S break; 20898d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ithread_yield(); 20908d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 20918d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 20928d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 20938d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* If we already have the minimum sad, there is no point in searching for sad again */ 20948d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_proc->u4_min_sad_reached == 0) 20958d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 20968d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* intra gating in inter slices */ 20978d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* No need of gating if we want to force intra, we need to find the threshold only if inter is enabled by AIR*/ 20983749f6f435e79624f72841e866245d84195551cdHarinarayanan K K if (i4_air_enable_inter && ps_proc->i4_slice_type != ISLICE && ps_codec->u4_inter_gate) 20998d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 21008d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* distortion of neighboring blocks */ 21018d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_distortion[4]; 21028d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 21038d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S i4_distortion[0] = ps_proc->s_left_mb_syntax_ele.i4_mb_distortion; 21048d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 21058d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S i4_distortion[1] = ps_proc->ps_top_row_mb_syntax_ele[ps_proc->i4_mb_x].i4_mb_distortion; 21068d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 21078d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S i4_distortion[2] = ps_proc->ps_top_row_mb_syntax_ele[ps_proc->i4_mb_x + 1].i4_mb_distortion; 21088d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 21098d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S i4_distortion[3] = ps_proc->s_top_left_mb_syntax_ele.i4_mb_distortion; 21108d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 21118d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S i4_gate_threshold = (i4_distortion[0] + i4_distortion[1] + i4_distortion[2] + i4_distortion[3]) >> 2; 21128d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 21138d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 21148d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 21153749f6f435e79624f72841e866245d84195551cdHarinarayanan K K 21168d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* If we are going to force intra we need to evaluate intra irrespective of gating */ 21178d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if ( (!i4_air_enable_inter) || ((i4_gate_threshold + 16 *((WORD32) ps_proc->u4_lambda)) < ps_proc->i4_mb_distortion)) 21188d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 21198d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* evaluate intra 4x4 modes */ 21208d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (u4_valid_modes & (1 << I4x4)) 21218d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 21228d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_codec->s_cfg.u4_enc_speed_preset == IVE_SLOWEST) 21238d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 21248d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ih264e_evaluate_intra4x4_modes_for_least_cost_rdopton(ps_proc); 21258d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 21268d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S else 21278d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 21288d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ih264e_evaluate_intra4x4_modes_for_least_cost_rdoptoff(ps_proc); 21298d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 21308d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 21318d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 21328d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* evaluate intra 16x16 modes */ 21338d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (u4_valid_modes & (1 << I16x16)) 21348d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 21358d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ih264e_evaluate_intra16x16_modes_for_least_cost_rdoptoff(ps_proc); 21368d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 21378d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 21388d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* evaluate intra 8x8 modes */ 21398d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (u4_valid_modes & (1 << I8x8)) 21408d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 21418d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ih264e_evaluate_intra8x8_modes_for_least_cost_rdoptoff(ps_proc); 21428d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 21438d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 21443749f6f435e79624f72841e866245d84195551cdHarinarayanan K K } 21458d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 21463749f6f435e79624f72841e866245d84195551cdHarinarayanan K K } 21478d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 21488d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* is intra */ 21498d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_proc->u4_mb_type == I4x4 || ps_proc->u4_mb_type == I16x16 || ps_proc->u4_mb_type == I8x8) 21508d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 21518d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S luma_idx = ps_proc->u4_mb_type; 21528d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S chroma_idx = 0; 21538d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S is_intra = 1; 21548d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 21558d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* evaluate chroma blocks for intra */ 21568d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ih264e_evaluate_chroma_intra8x8_modes_for_least_cost_rdoptoff(ps_proc); 21578d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 21588d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S else 21598d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 21608d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S luma_idx = 3; 21618d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S chroma_idx = 1; 21628d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S is_intra = 0; 21638d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 21648d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->u4_is_intra = is_intra; 21653749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_proc->ps_pu->b1_intra_flag = is_intra; 21668d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 21678d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* redo MV pred of neighbors in the case intra mb */ 21688d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* TODO : currently called unconditionally, needs to be called only in the case of intra 21698d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * to modify neighbors */ 21708d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_proc->i4_slice_type != ISLICE) 21718d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 21723749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ih264e_mv_pred(ps_proc, ps_proc->i4_slice_type); 21738d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 21748d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 21758d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* Perform luma mb core coding */ 21768d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S u4_cbp_l = (ps_codec->luma_energy_compaction)[luma_idx](ps_proc); 21778d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 21788d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* Perform luma mb core coding */ 21798d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S u4_cbp_c = (ps_codec->chroma_energy_compaction)[chroma_idx](ps_proc); 21808d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 21818d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* coded block pattern */ 21828d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->u4_cbp = (u4_cbp_c << 4) | u4_cbp_l; 21838d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 21843749f6f435e79624f72841e866245d84195551cdHarinarayanan K K if (!ps_proc->u4_is_intra) 21858d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 21863749f6f435e79624f72841e866245d84195551cdHarinarayanan K K if (ps_proc->i4_slice_type == BSLICE) 21878d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 21883749f6f435e79624f72841e866245d84195551cdHarinarayanan K K if (ih264e_find_bskip_params(ps_proc, PRED_L0)) 21893749f6f435e79624f72841e866245d84195551cdHarinarayanan K K { 21903749f6f435e79624f72841e866245d84195551cdHarinarayanan K K ps_proc->u4_mb_type = (ps_proc->u4_cbp) ? BDIRECT : BSKIP; 21913749f6f435e79624f72841e866245d84195551cdHarinarayanan K K } 21923749f6f435e79624f72841e866245d84195551cdHarinarayanan K K } 21933749f6f435e79624f72841e866245d84195551cdHarinarayanan K K else if(!ps_proc->u4_cbp) 21943749f6f435e79624f72841e866245d84195551cdHarinarayanan K K { 21953749f6f435e79624f72841e866245d84195551cdHarinarayanan K K if (ih264e_find_pskip_params(ps_proc, PRED_L0)) 21968d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 21978d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->u4_mb_type = PSKIP; 21988d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 21998d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 22008d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 22018d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 22028d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha SUPDATE_MB_INFO: 22038d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 22048d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* Update mb sad, mb qp and intra mb cost. Will be used by rate control */ 22058d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ih264e_update_rc_mb_info(&ps_proc->s_frame_info, ps_proc); 22068d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 22078d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /**********************************************************************/ 22088d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* if disable deblock level is '0' this implies enable deblocking for */ 22098d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* all edges of all macroblocks with out any restrictions */ 22108d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* */ 22118d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* if disable deblock level is '1' this implies disable deblocking for*/ 22128d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* all edges of all macroblocks with out any restrictions */ 22138d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* */ 22148d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* if disable deblock level is '2' this implies enable deblocking for */ 22158d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* all edges of all macroblocks except edges overlapping with slice */ 22168d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* boundaries. This option is not currently supported by the encoder */ 22178d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* hence the slice map should be of no significance to perform debloc */ 22188d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* king */ 22198d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /**********************************************************************/ 22208d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 22218d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_proc->u4_compute_recon) 22228d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 22238d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* deblk context */ 22248d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* src pointers */ 22258d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD8 *pu1_cur_pic_luma = ps_proc->pu1_rec_buf_luma; 22268d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD8 *pu1_cur_pic_chroma = ps_proc->pu1_rec_buf_chroma; 22278d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 22288d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* src indices */ 22298d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD32 i4_mb_x = ps_proc->i4_mb_x; 22308d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD32 i4_mb_y = ps_proc->i4_mb_y; 22318d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 22328d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* compute blocking strength */ 22338d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_proc->u4_disable_deblock_level != 1) 22348d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 22358d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ih264e_compute_bs(ps_proc); 22368d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 22378d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 22388d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* nmb deblocking and hpel and padding */ 22398d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ih264e_dblk_pad_hpel_processing_n_mbs(ps_proc, pu1_cur_pic_luma, 22408d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pu1_cur_pic_chroma, i4_mb_x, 22418d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S i4_mb_y); 22428d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 22438d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 22448d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* update the context after for coding next mb */ 22458d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S error_status |= ih264e_update_proc_ctxt(ps_proc); 22468d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 22478d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* Once the last row is processed, mark the buffer status appropriately */ 22488d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_proc->i4_ht_mbs == ps_proc->i4_mb_y) 22498d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 22508d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* Pointer to current picture buffer structure */ 22518d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S pic_buf_t *ps_cur_pic = ps_proc->ps_cur_pic; 22528d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 22538d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* Pointer to current picture's mv buffer structure */ 22548d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S mv_buf_t *ps_cur_mv_buf = ps_proc->ps_cur_mv_buf; 22558d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 22568d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /**********************************************************************/ 22578d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* if disable deblock level is '0' this implies enable deblocking for */ 22588d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* all edges of all macroblocks with out any restrictions */ 22598d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* */ 22608d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* if disable deblock level is '1' this implies disable deblocking for*/ 22618d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* all edges of all macroblocks with out any restrictions */ 22628d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* */ 22638d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* if disable deblock level is '2' this implies enable deblocking for */ 22648d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* all edges of all macroblocks except edges overlapping with slice */ 22658d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* boundaries. This option is not currently supported by the encoder */ 22668d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* hence the slice map should be of no significance to perform debloc */ 22678d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* king */ 22688d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /**********************************************************************/ 22698d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S error_status |= ih264_buf_mgr_release(ps_codec->pv_mv_buf_mgr, ps_cur_mv_buf->i4_buf_id , BUF_MGR_CODEC); 22708d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 22718d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S error_status |= ih264_buf_mgr_release(ps_codec->pv_ref_buf_mgr, ps_cur_pic->i4_buf_id , BUF_MGR_CODEC); 22728d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 22738d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (ps_codec->s_cfg.u4_enable_recon) 22748d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 22758d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* pic cnt */ 22768d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->as_rec_buf[ctxt_sel].i4_pic_cnt = ps_proc->i4_pic_cnt; 22778d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 22788d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* rec buffers */ 22798d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->as_rec_buf[ctxt_sel].s_pic_buf = *ps_proc->ps_cur_pic; 22808d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 22818d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* is last? */ 22828d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->as_rec_buf[ctxt_sel].u4_is_last = ps_proc->s_entropy.u4_is_last; 22838d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 22848d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* frame time stamp */ 22858d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->as_rec_buf[ctxt_sel].u4_timestamp_high = ps_proc->s_entropy.u4_timestamp_high; 22868d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->as_rec_buf[ctxt_sel].u4_timestamp_low = ps_proc->s_entropy.u4_timestamp_low; 22878d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 22888d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 22898d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 22908d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 22918d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 22928d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S DEBUG_HISTOGRAM_DUMP(ps_codec->s_cfg.i4_ht_mbs == ps_proc->i4_mb_y); 22938d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 22948d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S return error_status; 22958d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S} 22968d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 22978d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S/** 22988d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S******************************************************************************* 22998d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 23008d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @brief 23018d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* Function to update rc context after encoding 23028d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 23038d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @par Description 23048d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* This function updates the rate control context after the frame is encoded. 23058d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* Number of bits consumed by the current frame, frame distortion, frame cost, 23068d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* number of intra/inter mb's, ... are passed on to rate control context for 23078d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* updating the rc model. 23088d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 23098d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @param[in] ps_codec 23108d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* Handle to codec context 23118d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 23128d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @param[in] ctxt_sel 23138d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* frame context selector 23148d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 23158d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @param[in] pic_cnt 23168d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* pic count 23178d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 23188d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @returns i4_stuffing_byte 23198d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* number of stuffing bytes (if necessary) 23208d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 23218d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @remarks 23228d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 23238d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S******************************************************************************* 23248d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S*/ 23253749f6f435e79624f72841e866245d84195551cdHarinarayanan K KWORD32 ih264e_update_rc_post_enc(codec_t *ps_codec, WORD32 ctxt_sel, WORD32 i4_is_first_frm) 23268d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S{ 23278d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* proc set base idx */ 23288d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_proc_ctxt_sel_base = ctxt_sel ? (MAX_PROCESS_CTXT / 2) : 0; 23298d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 23308d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* proc ctxt */ 23318d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S process_ctxt_t *ps_proc = &ps_codec->as_process[i4_proc_ctxt_sel_base]; 23328d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 23338d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* frame qp */ 23348d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S UWORD8 u1_frame_qp = ps_codec->u4_frame_qp; 23358d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 23368d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* cbr rc return status */ 23378d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i4_stuffing_byte = 0; 23388d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 23398d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* current frame stats */ 23408d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S frame_info_t s_frame_info; 23418d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S picture_type_e rc_pic_type; 23428d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 23438d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* temp var */ 23448d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 i, j; 23458d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 23468d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /********************************************************************/ 23478d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* BEGIN INIT */ 23488d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /********************************************************************/ 23498d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 23508d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* init frame info */ 23518d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S irc_init_frame_info(&s_frame_info); 23528d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 23538d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* get frame info */ 23548d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S for (i = 0; i < (WORD32)ps_codec->s_cfg.u4_num_cores; i++) 23558d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 23568d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /*****************************************************************/ 23578d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* One frame can be encoded by max of u4_num_cores threads */ 23588d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* Accumulating the num mbs, sad, qp and intra_mb_cost from */ 23598d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* u4_num_cores threads */ 23608d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /*****************************************************************/ 23618d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S for (j = 0; j< MAX_MB_TYPE; j++) 23628d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 23638d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S s_frame_info.num_mbs[j] += ps_proc[i].s_frame_info.num_mbs[j]; 23648d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 23658d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S s_frame_info.tot_mb_sad[j] += ps_proc[i].s_frame_info.tot_mb_sad[j]; 23668d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 23678d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S s_frame_info.qp_sum[j] += ps_proc[i].s_frame_info.qp_sum[j]; 23688d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 23698d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 23708d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S s_frame_info.intra_mb_cost_sum += ps_proc[i].s_frame_info.intra_mb_cost_sum; 23718d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 23728d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S s_frame_info.activity_sum += ps_proc[i].s_frame_info.activity_sum; 23738d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 23748d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /*****************************************************************/ 23758d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* gather number of residue and header bits consumed by the frame*/ 23768d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /*****************************************************************/ 23778d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ih264e_update_rc_bits_info(&s_frame_info, &ps_proc[i].s_entropy); 23788d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 23798d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 23808d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* get pic type */ 23818d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S switch (ps_codec->pic_type) 23828d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 23838d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S case PIC_I: 23848d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S case PIC_IDR: 23858d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S rc_pic_type = I_PIC; 23868d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S break; 23878d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S case PIC_P: 23888d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S rc_pic_type = P_PIC; 23898d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S break; 23908d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S case PIC_B: 23918d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S rc_pic_type = B_PIC; 23928d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S break; 23938d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S default: 23948d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S assert(0); 23958d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S break; 23968d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 23978d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 23988d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* update rc lib with current frame stats */ 23998d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S i4_stuffing_byte = ih264e_rc_post_enc(ps_codec->s_rate_control.pps_rate_control_api, 24008d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S &(s_frame_info), 24018d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->s_rate_control.pps_pd_frm_rate, 24028d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->s_rate_control.pps_time_stamp, 24038d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_codec->s_rate_control.pps_frame_time, 24048d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S (ps_proc->i4_wd_mbs * ps_proc->i4_ht_mbs), 24058d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S &rc_pic_type, 24063749f6f435e79624f72841e866245d84195551cdHarinarayanan K K i4_is_first_frm, 24078d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S &ps_codec->s_rate_control.post_encode_skip[ctxt_sel], 24088d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S u1_frame_qp, 24098d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S &ps_codec->s_rate_control.num_intra_in_prev_frame, 24108d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S &ps_codec->s_rate_control.i4_avg_activity); 24118d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S return i4_stuffing_byte; 24128d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S} 24138d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 24148d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S/** 24158d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S******************************************************************************* 24168d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 24178d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @brief 24188d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* entry point of a spawned encoder thread 24198d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 24208d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @par Description: 24218d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* The encoder thread dequeues a proc/entropy job from the encoder queue and 24228d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* calls necessary routines. 24238d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 24248d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @param[in] pv_proc 24258d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* Process context corresponding to the thread 24268d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 24278d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @returns error status 24288d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 24298d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* @remarks 24308d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S* 24318d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S******************************************************************************* 24328d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S*/ 24338d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha SWORD32 ih264e_process_thread(void *pv_proc) 24348d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S{ 24358d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* error status */ 24368d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S IH264_ERROR_T ret = IH264_SUCCESS; 24378d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 error_status = IH264_SUCCESS; 24388d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 24398d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* proc ctxt */ 24408d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S process_ctxt_t *ps_proc = pv_proc; 24418d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 24428d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* codec ctxt */ 24438d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S codec_t *ps_codec = ps_proc->ps_codec; 24448d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 24458d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* structure to represent a processing job entry */ 24468d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S job_t s_job; 24478d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 24488d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* blocking call : entropy dequeue is non-blocking till all 24498d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S * the proc jobs are processed */ 24508d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S WORD32 is_blocking = 0; 24518d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 24528d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* set affinity */ 24538d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ithread_set_affinity(ps_proc->i4_id); 24548d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 24558d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S while(1) 24568d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 24578d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* dequeue a job from the entropy queue */ 24588d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 24598d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S int error = ithread_mutex_lock(ps_codec->pv_entropy_mutex); 24608d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 24618d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* codec context selector */ 2462d3e18f0c8854c258021be1e603d8f50a241698a7Harinarayanan K K WORD32 ctxt_sel = ps_codec->i4_encode_api_call_cnt % MAX_CTXT_SETS; 24638d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 24648d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S volatile UWORD32 *pu4_buf = &ps_codec->au4_entropy_thread_active[ctxt_sel]; 24658d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 24668d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* have the lock */ 24678d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (error == 0) 24688d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 24698d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (*pu4_buf == 0) 24708d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 24718d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* no entropy threads are active, try dequeuing a job from the entropy queue */ 24728d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ret = ih264_list_dequeue(ps_proc->pv_entropy_jobq, &s_job, is_blocking); 24738d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (IH264_SUCCESS == ret) 24748d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 24758d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S *pu4_buf = 1; 24768d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ithread_mutex_unlock(ps_codec->pv_entropy_mutex); 24778d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S goto WORKER; 24788d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 24798d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S else if(is_blocking) 24808d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 24818d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ithread_mutex_unlock(ps_codec->pv_entropy_mutex); 24828d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S break; 24838d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 24848d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 24858d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ithread_mutex_unlock(ps_codec->pv_entropy_mutex); 24868d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 24878d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 24888d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 24898d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* dequeue a job from the process queue */ 24908d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ret = ih264_list_dequeue(ps_proc->pv_proc_jobq, &s_job, 1); 24918d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if (IH264_SUCCESS != ret) 24928d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 24938d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S if(ps_proc->i4_id) 24948d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S break; 24958d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S else 24968d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 24978d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S is_blocking = 1; 24988d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S continue; 24998d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 25008d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 25018d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 25028d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha SWORKER: 25038d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* choose appropriate proc context based on proc_base_idx */ 25048d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc = &ps_codec->as_process[ps_proc->i4_id + s_job.i2_proc_base_idx]; 25058d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 25068d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S switch (s_job.i4_cmd) 25078d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S { 25088d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S case CMD_PROCESS: 25098d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->i4_mb_cnt = s_job.i2_mb_cnt; 25108d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->i4_mb_x = s_job.i2_mb_x; 25118d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->i4_mb_y = s_job.i2_mb_y; 25128d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 25138d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* init process context */ 25148d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ih264e_init_proc_ctxt(ps_proc); 25158d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 25168d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* core code all mbs enlisted under the current job */ 25178d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S error_status |= ih264e_process(ps_proc); 25188d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S break; 25198d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 25208d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S case CMD_ENTROPY: 25218d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->s_entropy.i4_mb_x = s_job.i2_mb_x; 25228d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->s_entropy.i4_mb_y = s_job.i2_mb_y; 25238d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->s_entropy.i4_mb_cnt = s_job.i2_mb_cnt; 25248d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 25258d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* init entropy */ 25268d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ih264e_init_entropy_ctxt(ps_proc); 25278d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 25288d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* entropy code all mbs enlisted under the current job */ 25298d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S error_status |= ih264e_entropy(ps_proc); 25308d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S break; 25318d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 25328d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S default: 25338d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S error_status |= IH264_FAIL; 25348d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S break; 25358d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 25368d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S } 25378d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S 25388d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S /* send error code */ 25398d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S ps_proc->i4_error_code = error_status; 25408d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S return ret; 25418d3d303c7942ced6a987a52db8977d768dc3605fHamsalekha S} 2542