xref: /aosp_15_r20/external/libavc/encoder/svc/isvce_cabac.c (revision 495ae853bb871d1e5a258cb02c2cc13cde8ddb9a)
1*495ae853SAndroid Build Coastguard Worker /******************************************************************************
2*495ae853SAndroid Build Coastguard Worker  *
3*495ae853SAndroid Build Coastguard Worker  * Copyright (C) 2022 The Android Open Source Project
4*495ae853SAndroid Build Coastguard Worker  *
5*495ae853SAndroid Build Coastguard Worker  * Licensed under the Apache License, Version 2.0 (the "License");
6*495ae853SAndroid Build Coastguard Worker  * you may not use this file except in compliance with the License.
7*495ae853SAndroid Build Coastguard Worker  * You may obtain a copy of the License at:
8*495ae853SAndroid Build Coastguard Worker  *
9*495ae853SAndroid Build Coastguard Worker  * http://www.apache.org/licenses/LICENSE-2.0
10*495ae853SAndroid Build Coastguard Worker  *
11*495ae853SAndroid Build Coastguard Worker  * Unless required by applicable law or agreed to in writing, software
12*495ae853SAndroid Build Coastguard Worker  * distributed under the License is distributed on an "AS IS" BASIS,
13*495ae853SAndroid Build Coastguard Worker  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14*495ae853SAndroid Build Coastguard Worker  * See the License for the specific language governing permissions and
15*495ae853SAndroid Build Coastguard Worker  * limitations under the License.
16*495ae853SAndroid Build Coastguard Worker  *
17*495ae853SAndroid Build Coastguard Worker  *****************************************************************************
18*495ae853SAndroid Build Coastguard Worker  * Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore
19*495ae853SAndroid Build Coastguard Worker  */
20*495ae853SAndroid Build Coastguard Worker 
21*495ae853SAndroid Build Coastguard Worker /**
22*495ae853SAndroid Build Coastguard Worker *******************************************************************************
23*495ae853SAndroid Build Coastguard Worker * @file
24*495ae853SAndroid Build Coastguard Worker *  isvce_cabac.c
25*495ae853SAndroid Build Coastguard Worker *
26*495ae853SAndroid Build Coastguard Worker * @brief
27*495ae853SAndroid Build Coastguard Worker *  Contains all leaf level functions for CABAC entropy coding.
28*495ae853SAndroid Build Coastguard Worker *
29*495ae853SAndroid Build Coastguard Worker *
30*495ae853SAndroid Build Coastguard Worker * @author
31*495ae853SAndroid Build Coastguard Worker * Doney Alex
32*495ae853SAndroid Build Coastguard Worker *
33*495ae853SAndroid Build Coastguard Worker * @par List of Functions:
34*495ae853SAndroid Build Coastguard Worker *
35*495ae853SAndroid Build Coastguard Worker *
36*495ae853SAndroid Build Coastguard Worker * @remarks
37*495ae853SAndroid Build Coastguard Worker *  None
38*495ae853SAndroid Build Coastguard Worker *
39*495ae853SAndroid Build Coastguard Worker *******************************************************************************
40*495ae853SAndroid Build Coastguard Worker */
41*495ae853SAndroid Build Coastguard Worker 
42*495ae853SAndroid Build Coastguard Worker /*****************************************************************************/
43*495ae853SAndroid Build Coastguard Worker /* File Includes                                                             */
44*495ae853SAndroid Build Coastguard Worker /*****************************************************************************/
45*495ae853SAndroid Build Coastguard Worker 
46*495ae853SAndroid Build Coastguard Worker /* System include files */
47*495ae853SAndroid Build Coastguard Worker #include <stdio.h>
48*495ae853SAndroid Build Coastguard Worker #include <assert.h>
49*495ae853SAndroid Build Coastguard Worker #include <limits.h>
50*495ae853SAndroid Build Coastguard Worker #include <string.h>
51*495ae853SAndroid Build Coastguard Worker 
52*495ae853SAndroid Build Coastguard Worker /* User include files */
53*495ae853SAndroid Build Coastguard Worker #include "ih264e_config.h"
54*495ae853SAndroid Build Coastguard Worker #include "ih264_typedefs.h"
55*495ae853SAndroid Build Coastguard Worker #include "iv2.h"
56*495ae853SAndroid Build Coastguard Worker #include "ive2.h"
57*495ae853SAndroid Build Coastguard Worker #include "ih264_debug.h"
58*495ae853SAndroid Build Coastguard Worker #include "ih264_macros.h"
59*495ae853SAndroid Build Coastguard Worker #include "isvc_defs.h"
60*495ae853SAndroid Build Coastguard Worker #include "isvce_defs.h"
61*495ae853SAndroid Build Coastguard Worker #include "isvc_macros.h"
62*495ae853SAndroid Build Coastguard Worker #include "ih264e_error.h"
63*495ae853SAndroid Build Coastguard Worker #include "ih264e_bitstream.h"
64*495ae853SAndroid Build Coastguard Worker #include "ime_distortion_metrics.h"
65*495ae853SAndroid Build Coastguard Worker #include "ime_defs.h"
66*495ae853SAndroid Build Coastguard Worker #include "ime_structs.h"
67*495ae853SAndroid Build Coastguard Worker #include "ih264_error.h"
68*495ae853SAndroid Build Coastguard Worker #include "isvc_structs.h"
69*495ae853SAndroid Build Coastguard Worker #include "isvc_trans_quant_itrans_iquant.h"
70*495ae853SAndroid Build Coastguard Worker #include "isvc_inter_pred_filters.h"
71*495ae853SAndroid Build Coastguard Worker #include "isvc_mem_fns.h"
72*495ae853SAndroid Build Coastguard Worker #include "ih264_padding.h"
73*495ae853SAndroid Build Coastguard Worker #include "ih264_platform_macros.h"
74*495ae853SAndroid Build Coastguard Worker #include "ih264_intra_pred_filters.h"
75*495ae853SAndroid Build Coastguard Worker #include "ih264_deblk_edge_filters.h"
76*495ae853SAndroid Build Coastguard Worker #include "isvc_cabac_tables.h"
77*495ae853SAndroid Build Coastguard Worker #include "irc_cntrl_param.h"
78*495ae853SAndroid Build Coastguard Worker #include "irc_frame_info_collector.h"
79*495ae853SAndroid Build Coastguard Worker #include "isvce_rate_control.h"
80*495ae853SAndroid Build Coastguard Worker #include "isvce_cabac_structs.h"
81*495ae853SAndroid Build Coastguard Worker #include "isvce_structs.h"
82*495ae853SAndroid Build Coastguard Worker #include "isvce_cabac.h"
83*495ae853SAndroid Build Coastguard Worker #include "isvce_encode_header.h"
84*495ae853SAndroid Build Coastguard Worker #include "ih264_cavlc_tables.h"
85*495ae853SAndroid Build Coastguard Worker #include "ih264e_statistics.h"
86*495ae853SAndroid Build Coastguard Worker #include "ih264e_trace.h"
87*495ae853SAndroid Build Coastguard Worker 
88*495ae853SAndroid Build Coastguard Worker /*****************************************************************************/
89*495ae853SAndroid Build Coastguard Worker /* Function Definitions                                                      */
90*495ae853SAndroid Build Coastguard Worker /*****************************************************************************/
91*495ae853SAndroid Build Coastguard Worker 
92*495ae853SAndroid Build Coastguard Worker /**
93*495ae853SAndroid Build Coastguard Worker  *******************************************************************************
94*495ae853SAndroid Build Coastguard Worker  *
95*495ae853SAndroid Build Coastguard Worker  * @brief
96*495ae853SAndroid Build Coastguard Worker  *  k-th order Exp-Golomb (UEGk) binarization process: Implements concatenated
97*495ae853SAndroid Build Coastguard Worker  *   unary/ k-th order Exp-Golomb  (UEGk) binarization process,
98*495ae853SAndroid Build Coastguard Worker  *   where k = 0 as defined in 9.3.2.3 of  ITU_T_H264-201402
99*495ae853SAndroid Build Coastguard Worker  *
100*495ae853SAndroid Build Coastguard Worker  * @param[in] i2_sufs
101*495ae853SAndroid Build Coastguard Worker  *  Suffix bit string
102*495ae853SAndroid Build Coastguard Worker  *
103*495ae853SAndroid Build Coastguard Worker  * @param[in] pi1_bins_len
104*495ae853SAndroid Build Coastguard Worker  *  Pointer to length of tthe string
105*495ae853SAndroid Build Coastguard Worker  *
106*495ae853SAndroid Build Coastguard Worker  * @returns Binarized value
107*495ae853SAndroid Build Coastguard Worker  *
108*495ae853SAndroid Build Coastguard Worker  * @remarks
109*495ae853SAndroid Build Coastguard Worker  *  None
110*495ae853SAndroid Build Coastguard Worker  *
111*495ae853SAndroid Build Coastguard Worker  *******************************************************************************
112*495ae853SAndroid Build Coastguard Worker  */
113*495ae853SAndroid Build Coastguard Worker 
isvce_cabac_UEGk0_binarization(WORD16 i2_sufs,WORD8 * pi1_bins_len)114*495ae853SAndroid Build Coastguard Worker UWORD32 isvce_cabac_UEGk0_binarization(WORD16 i2_sufs, WORD8 *pi1_bins_len)
115*495ae853SAndroid Build Coastguard Worker {
116*495ae853SAndroid Build Coastguard Worker     WORD32 unary_length;
117*495ae853SAndroid Build Coastguard Worker     UWORD32 u4_sufs_shiftk_plus1, u4_egk, u4_unary_bins;
118*495ae853SAndroid Build Coastguard Worker 
119*495ae853SAndroid Build Coastguard Worker     u4_sufs_shiftk_plus1 = i2_sufs + 1;
120*495ae853SAndroid Build Coastguard Worker 
121*495ae853SAndroid Build Coastguard Worker     unary_length = (32 - CLZ(u4_sufs_shiftk_plus1) + (0 == u4_sufs_shiftk_plus1));
122*495ae853SAndroid Build Coastguard Worker 
123*495ae853SAndroid Build Coastguard Worker     /* unary code with (unary_length-1) '1's and terminating '0' bin */
124*495ae853SAndroid Build Coastguard Worker     u4_unary_bins = (1 << unary_length) - 2;
125*495ae853SAndroid Build Coastguard Worker 
126*495ae853SAndroid Build Coastguard Worker     /* insert the symbol prefix of (unary length - 1)  bins */
127*495ae853SAndroid Build Coastguard Worker     u4_egk = (u4_unary_bins << (unary_length - 1)) |
128*495ae853SAndroid Build Coastguard Worker              (u4_sufs_shiftk_plus1 & ((1 << (unary_length - 1)) - 1));
129*495ae853SAndroid Build Coastguard Worker 
130*495ae853SAndroid Build Coastguard Worker     /* length of the code = 2 *(unary_length - 1) + 1 + k */
131*495ae853SAndroid Build Coastguard Worker     *pi1_bins_len = (2 * unary_length) - 1;
132*495ae853SAndroid Build Coastguard Worker 
133*495ae853SAndroid Build Coastguard Worker     return (u4_egk);
134*495ae853SAndroid Build Coastguard Worker }
135*495ae853SAndroid Build Coastguard Worker 
136*495ae853SAndroid Build Coastguard Worker /**
137*495ae853SAndroid Build Coastguard Worker  *******************************************************************************
138*495ae853SAndroid Build Coastguard Worker  *
139*495ae853SAndroid Build Coastguard Worker  * @brief
140*495ae853SAndroid Build Coastguard Worker  *  Get cabac context for the MB :calculates the pointers to Top and   left
141*495ae853SAndroid Build Coastguard Worker  *          cabac neighbor context depending upon neighbor  availability.
142*495ae853SAndroid Build Coastguard Worker  *
143*495ae853SAndroid Build Coastguard Worker  * @param[in] ps_ent_ctxt
144*495ae853SAndroid Build Coastguard Worker  *  Pointer to entropy context structure
145*495ae853SAndroid Build Coastguard Worker  *
146*495ae853SAndroid Build Coastguard Worker  * @param[in] u4_mb_type
147*495ae853SAndroid Build Coastguard Worker  *  Type of MB
148*495ae853SAndroid Build Coastguard Worker  *
149*495ae853SAndroid Build Coastguard Worker  * @returns
150*495ae853SAndroid Build Coastguard Worker  *
151*495ae853SAndroid Build Coastguard Worker  * @remarks
152*495ae853SAndroid Build Coastguard Worker  *  None
153*495ae853SAndroid Build Coastguard Worker  *
154*495ae853SAndroid Build Coastguard Worker  *******************************************************************************
155*495ae853SAndroid Build Coastguard Worker  */
isvce_get_cabac_context(isvce_entropy_ctxt_t * ps_ent_ctxt,WORD32 u4_mb_type)156*495ae853SAndroid Build Coastguard Worker void isvce_get_cabac_context(isvce_entropy_ctxt_t *ps_ent_ctxt, WORD32 u4_mb_type)
157*495ae853SAndroid Build Coastguard Worker {
158*495ae853SAndroid Build Coastguard Worker     /* CABAC context */
159*495ae853SAndroid Build Coastguard Worker     isvce_cabac_ctxt_t *ps_cabac_ctxt = ps_ent_ctxt->ps_cabac;
160*495ae853SAndroid Build Coastguard Worker     isvce_mb_info_ctxt_t *ps_ctx_inc_mb_map;
161*495ae853SAndroid Build Coastguard Worker     cab_csbp_t *ps_lft_csbp;
162*495ae853SAndroid Build Coastguard Worker 
163*495ae853SAndroid Build Coastguard Worker     WORD32 i4_lft_avail, i4_top_avail, i4_is_intra;
164*495ae853SAndroid Build Coastguard Worker     WORD32 i4_mb_x, i4_mb_y;
165*495ae853SAndroid Build Coastguard Worker     UWORD8 *pu1_slice_idx = ps_ent_ctxt->pu1_slice_idx;
166*495ae853SAndroid Build Coastguard Worker 
167*495ae853SAndroid Build Coastguard Worker     i4_is_intra = ((u4_mb_type == I16x16) || (u4_mb_type == I8x8) || (u4_mb_type == I4x4));
168*495ae853SAndroid Build Coastguard Worker 
169*495ae853SAndroid Build Coastguard Worker     /* derive neighbor availability */
170*495ae853SAndroid Build Coastguard Worker     i4_mb_x = ps_ent_ctxt->i4_mb_x;
171*495ae853SAndroid Build Coastguard Worker     i4_mb_y = ps_ent_ctxt->i4_mb_y;
172*495ae853SAndroid Build Coastguard Worker     pu1_slice_idx += (i4_mb_y * ps_ent_ctxt->i4_wd_mbs);
173*495ae853SAndroid Build Coastguard Worker     /* left macroblock availability */
174*495ae853SAndroid Build Coastguard Worker     i4_lft_avail = (i4_mb_x == 0 || (pu1_slice_idx[i4_mb_x - 1] != pu1_slice_idx[i4_mb_x])) ? 0 : 1;
175*495ae853SAndroid Build Coastguard Worker     /* top macroblock availability */
176*495ae853SAndroid Build Coastguard Worker     i4_top_avail = (i4_mb_y == 0 ||
177*495ae853SAndroid Build Coastguard Worker                     (pu1_slice_idx[i4_mb_x - ps_ent_ctxt->i4_wd_mbs] != pu1_slice_idx[i4_mb_x]))
178*495ae853SAndroid Build Coastguard Worker                        ? 0
179*495ae853SAndroid Build Coastguard Worker                        : 1;
180*495ae853SAndroid Build Coastguard Worker     i4_mb_x = ps_ent_ctxt->i4_mb_x;
181*495ae853SAndroid Build Coastguard Worker     ps_ctx_inc_mb_map = ps_cabac_ctxt->ps_mb_map_ctxt_inc;
182*495ae853SAndroid Build Coastguard Worker     ps_cabac_ctxt->ps_curr_ctxt_mb_info = ps_ctx_inc_mb_map + i4_mb_x;
183*495ae853SAndroid Build Coastguard Worker     ps_cabac_ctxt->ps_left_ctxt_mb_info = ps_cabac_ctxt->ps_def_ctxt_mb_info;
184*495ae853SAndroid Build Coastguard Worker     ps_cabac_ctxt->ps_top_ctxt_mb_info = ps_cabac_ctxt->ps_def_ctxt_mb_info;
185*495ae853SAndroid Build Coastguard Worker     ps_lft_csbp = ps_cabac_ctxt->ps_lft_csbp;
186*495ae853SAndroid Build Coastguard Worker     ps_cabac_ctxt->pu1_left_y_ac_csbp = &ps_lft_csbp->u1_y_ac_csbp_top_mb;
187*495ae853SAndroid Build Coastguard Worker     ps_cabac_ctxt->pu1_left_uv_ac_csbp = &ps_lft_csbp->u1_uv_ac_csbp_top_mb;
188*495ae853SAndroid Build Coastguard Worker     ps_cabac_ctxt->pu1_left_yuv_dc_csbp = &ps_lft_csbp->u1_yuv_dc_csbp_top_mb;
189*495ae853SAndroid Build Coastguard Worker     ps_cabac_ctxt->pi1_left_ref_idx_ctxt_inc = &ps_cabac_ctxt->i1_left_ref_idx_ctx_inc_arr[0][0];
190*495ae853SAndroid Build Coastguard Worker     ps_cabac_ctxt->pu1_left_mv_ctxt_inc = ps_cabac_ctxt->u1_left_mv_ctxt_inc_arr[0];
191*495ae853SAndroid Build Coastguard Worker 
192*495ae853SAndroid Build Coastguard Worker     if(i4_lft_avail) ps_cabac_ctxt->ps_left_ctxt_mb_info = ps_cabac_ctxt->ps_curr_ctxt_mb_info - 1;
193*495ae853SAndroid Build Coastguard Worker     if(i4_top_avail) ps_cabac_ctxt->ps_top_ctxt_mb_info = ps_cabac_ctxt->ps_curr_ctxt_mb_info;
194*495ae853SAndroid Build Coastguard Worker 
195*495ae853SAndroid Build Coastguard Worker     if(!i4_lft_avail)
196*495ae853SAndroid Build Coastguard Worker     {
197*495ae853SAndroid Build Coastguard Worker         UWORD8 u1_def_csbp = i4_is_intra ? 0xf : 0;
198*495ae853SAndroid Build Coastguard Worker         *(ps_cabac_ctxt->pu1_left_y_ac_csbp) = u1_def_csbp;
199*495ae853SAndroid Build Coastguard Worker         *(ps_cabac_ctxt->pu1_left_uv_ac_csbp) = u1_def_csbp;
200*495ae853SAndroid Build Coastguard Worker         *(ps_cabac_ctxt->pu1_left_yuv_dc_csbp) = u1_def_csbp;
201*495ae853SAndroid Build Coastguard Worker         *((UWORD32 *) ps_cabac_ctxt->pi1_left_ref_idx_ctxt_inc) = 0;
202*495ae853SAndroid Build Coastguard Worker         memset(ps_cabac_ctxt->pu1_left_mv_ctxt_inc, 0, 16);
203*495ae853SAndroid Build Coastguard Worker     }
204*495ae853SAndroid Build Coastguard Worker     if(!i4_top_avail)
205*495ae853SAndroid Build Coastguard Worker     {
206*495ae853SAndroid Build Coastguard Worker         UWORD8 u1_def_csbp = i4_is_intra ? 0xff : 0;
207*495ae853SAndroid Build Coastguard Worker         ps_cabac_ctxt->ps_top_ctxt_mb_info->u1_yuv_ac_csbp = u1_def_csbp;
208*495ae853SAndroid Build Coastguard Worker         ps_cabac_ctxt->ps_top_ctxt_mb_info->u1_yuv_dc_csbp = u1_def_csbp;
209*495ae853SAndroid Build Coastguard Worker         ps_cabac_ctxt->ps_curr_ctxt_mb_info->i1_ref_idx[0] =
210*495ae853SAndroid Build Coastguard Worker             ps_cabac_ctxt->ps_curr_ctxt_mb_info->i1_ref_idx[1] =
211*495ae853SAndroid Build Coastguard Worker                 ps_cabac_ctxt->ps_curr_ctxt_mb_info->i1_ref_idx[2] =
212*495ae853SAndroid Build Coastguard Worker                     ps_cabac_ctxt->ps_curr_ctxt_mb_info->i1_ref_idx[3] = 0;
213*495ae853SAndroid Build Coastguard Worker         memset(ps_cabac_ctxt->ps_curr_ctxt_mb_info->u1_mv, 0, 16);
214*495ae853SAndroid Build Coastguard Worker     }
215*495ae853SAndroid Build Coastguard Worker }
216*495ae853SAndroid Build Coastguard Worker 
217*495ae853SAndroid Build Coastguard Worker /**
218*495ae853SAndroid Build Coastguard Worker  *******************************************************************************
219*495ae853SAndroid Build Coastguard Worker  * @brief
220*495ae853SAndroid Build Coastguard Worker  *  flushing at termination: Explained in flowchart 9-12(ITU_T_H264-201402).
221*495ae853SAndroid Build Coastguard Worker  *
222*495ae853SAndroid Build Coastguard Worker  *  @param[in]   ps_cabac_ctxt
223*495ae853SAndroid Build Coastguard Worker  *  pointer to cabac context (handle)
224*495ae853SAndroid Build Coastguard Worker  *
225*495ae853SAndroid Build Coastguard Worker  * @returns  none
226*495ae853SAndroid Build Coastguard Worker  *
227*495ae853SAndroid Build Coastguard Worker  * @remarks
228*495ae853SAndroid Build Coastguard Worker  *  None
229*495ae853SAndroid Build Coastguard Worker  *
230*495ae853SAndroid Build Coastguard Worker  *******************************************************************************
231*495ae853SAndroid Build Coastguard Worker  */
isvce_cabac_flush(isvce_cabac_ctxt_t * ps_cabac_ctxt)232*495ae853SAndroid Build Coastguard Worker void isvce_cabac_flush(isvce_cabac_ctxt_t *ps_cabac_ctxt)
233*495ae853SAndroid Build Coastguard Worker {
234*495ae853SAndroid Build Coastguard Worker     /* bit stream ptr */
235*495ae853SAndroid Build Coastguard Worker     bitstrm_t *ps_stream = ps_cabac_ctxt->ps_bitstrm;
236*495ae853SAndroid Build Coastguard Worker     encoding_envirnoment_t *ps_cab_enc_env = &(ps_cabac_ctxt->s_cab_enc_env);
237*495ae853SAndroid Build Coastguard Worker     UWORD32 u4_low = ps_cab_enc_env->u4_code_int_low;
238*495ae853SAndroid Build Coastguard Worker     UWORD32 u4_bits_gen = ps_cab_enc_env->u4_bits_gen;
239*495ae853SAndroid Build Coastguard Worker     UWORD8 *pu1_strm_buf = ps_stream->pu1_strm_buffer;
240*495ae853SAndroid Build Coastguard Worker     UWORD32 u4_strm_buf_offset = ps_stream->u4_strm_buf_offset;
241*495ae853SAndroid Build Coastguard Worker     WORD32 zero_run = ps_stream->i4_zero_bytes_run;
242*495ae853SAndroid Build Coastguard Worker     UWORD32 u4_out_standing_bytes = ps_cab_enc_env->u4_out_standing_bytes;
243*495ae853SAndroid Build Coastguard Worker 
244*495ae853SAndroid Build Coastguard Worker     /************************************************************************/
245*495ae853SAndroid Build Coastguard Worker     /* Insert the carry (propogated in previous byte) along with            */
246*495ae853SAndroid Build Coastguard Worker     /* outstanding bytes (if any) and flush remaining bits                  */
247*495ae853SAndroid Build Coastguard Worker     /************************************************************************/
248*495ae853SAndroid Build Coastguard Worker     {
249*495ae853SAndroid Build Coastguard Worker         /* carry = 1 => putbit(1); carry propogated due to L renorm */
250*495ae853SAndroid Build Coastguard Worker         WORD32 carry = (u4_low >> (u4_bits_gen + CABAC_BITS)) & 0x1;
251*495ae853SAndroid Build Coastguard Worker         WORD32 last_byte;
252*495ae853SAndroid Build Coastguard Worker         WORD32 bits_left;
253*495ae853SAndroid Build Coastguard Worker         WORD32 rem_bits;
254*495ae853SAndroid Build Coastguard Worker 
255*495ae853SAndroid Build Coastguard Worker         /* carry exists only if pu1_strm_buf has at least 1 byte of data */
256*495ae853SAndroid Build Coastguard Worker         carry = carry && (u4_strm_buf_offset > 0);
257*495ae853SAndroid Build Coastguard Worker 
258*495ae853SAndroid Build Coastguard Worker         if(carry)
259*495ae853SAndroid Build Coastguard Worker         {
260*495ae853SAndroid Build Coastguard Worker             /* CORNER CASE: if the previous data is 0x000003, then EPB will be
261*495ae853SAndroid Build Coastguard Worker              inserted and the data will become 0x00000303 and if the carry is present,
262*495ae853SAndroid Build Coastguard Worker              it will be added with the last byte and it will become 0x00000304 which
263*495ae853SAndroid Build Coastguard Worker              is not correct as per standard */
264*495ae853SAndroid Build Coastguard Worker             /* so check for previous four bytes and if it is equal to 0x00000303
265*495ae853SAndroid Build Coastguard Worker              then subtract u4_strm_buf_offset by 1 */
266*495ae853SAndroid Build Coastguard Worker             if((u4_strm_buf_offset >= 4) && pu1_strm_buf[u4_strm_buf_offset - 1] == 0x03 &&
267*495ae853SAndroid Build Coastguard Worker                pu1_strm_buf[u4_strm_buf_offset - 2] == 0x03 &&
268*495ae853SAndroid Build Coastguard Worker                pu1_strm_buf[u4_strm_buf_offset - 3] == 0x00 &&
269*495ae853SAndroid Build Coastguard Worker                pu1_strm_buf[u4_strm_buf_offset - 4] == 0x00)
270*495ae853SAndroid Build Coastguard Worker             {
271*495ae853SAndroid Build Coastguard Worker                 u4_strm_buf_offset -= 1;
272*495ae853SAndroid Build Coastguard Worker             }
273*495ae853SAndroid Build Coastguard Worker             /* previous byte carry add will not result in overflow to        */
274*495ae853SAndroid Build Coastguard Worker             /* u4_strm_buf_offset - 2 as we track 0xff as outstanding bytes  */
275*495ae853SAndroid Build Coastguard Worker             pu1_strm_buf[u4_strm_buf_offset - 1] += carry;
276*495ae853SAndroid Build Coastguard Worker             zero_run = 0;
277*495ae853SAndroid Build Coastguard Worker         }
278*495ae853SAndroid Build Coastguard Worker 
279*495ae853SAndroid Build Coastguard Worker         /*        Insert outstanding bytes (if any)         */
280*495ae853SAndroid Build Coastguard Worker         while(u4_out_standing_bytes)
281*495ae853SAndroid Build Coastguard Worker         {
282*495ae853SAndroid Build Coastguard Worker             UWORD8 u1_0_or_ff = carry ? 0 : 0xFF;
283*495ae853SAndroid Build Coastguard Worker 
284*495ae853SAndroid Build Coastguard Worker             PUTBYTE_EPB(pu1_strm_buf, u4_strm_buf_offset, u1_0_or_ff, zero_run);
285*495ae853SAndroid Build Coastguard Worker             u4_out_standing_bytes--;
286*495ae853SAndroid Build Coastguard Worker         }
287*495ae853SAndroid Build Coastguard Worker 
288*495ae853SAndroid Build Coastguard Worker         /*  clear the carry in low */
289*495ae853SAndroid Build Coastguard Worker         if(carry)
290*495ae853SAndroid Build Coastguard Worker         {
291*495ae853SAndroid Build Coastguard Worker             u4_low &= ((1 << (u4_bits_gen + CABAC_BITS)) - 1);
292*495ae853SAndroid Build Coastguard Worker         }
293*495ae853SAndroid Build Coastguard Worker 
294*495ae853SAndroid Build Coastguard Worker         /* extract the remaining bits;                                   */
295*495ae853SAndroid Build Coastguard Worker         /* includes additional msb bit of low as per Figure 9-12      */
296*495ae853SAndroid Build Coastguard Worker         bits_left = u4_bits_gen + 1;
297*495ae853SAndroid Build Coastguard Worker         rem_bits = (u4_low >> (u4_bits_gen + CABAC_BITS - bits_left));
298*495ae853SAndroid Build Coastguard Worker 
299*495ae853SAndroid Build Coastguard Worker         if(bits_left >= 8)
300*495ae853SAndroid Build Coastguard Worker         {
301*495ae853SAndroid Build Coastguard Worker             last_byte = (rem_bits >> (bits_left - 8)) & 0xFF;
302*495ae853SAndroid Build Coastguard Worker             PUTBYTE_EPB(pu1_strm_buf, u4_strm_buf_offset, last_byte, zero_run);
303*495ae853SAndroid Build Coastguard Worker             bits_left -= 8;
304*495ae853SAndroid Build Coastguard Worker         }
305*495ae853SAndroid Build Coastguard Worker 
306*495ae853SAndroid Build Coastguard Worker         /* insert last byte along with rbsp stop bit(1) and 0's in the end */
307*495ae853SAndroid Build Coastguard Worker         last_byte =
308*495ae853SAndroid Build Coastguard Worker             (rem_bits << (8 - bits_left)) | (1 << (7 - bits_left) | (1 << (7 - bits_left - 1)));
309*495ae853SAndroid Build Coastguard Worker         last_byte &= 0xFF;
310*495ae853SAndroid Build Coastguard Worker         PUTBYTE_EPB(pu1_strm_buf, u4_strm_buf_offset, last_byte, zero_run);
311*495ae853SAndroid Build Coastguard Worker 
312*495ae853SAndroid Build Coastguard Worker         /* update the state variables and return success */
313*495ae853SAndroid Build Coastguard Worker         ps_stream->u4_strm_buf_offset = u4_strm_buf_offset;
314*495ae853SAndroid Build Coastguard Worker         ps_stream->i4_zero_bytes_run = 0;
315*495ae853SAndroid Build Coastguard Worker         /* Default init values for scratch variables of bitstream context */
316*495ae853SAndroid Build Coastguard Worker         ps_stream->u4_cur_word = 0;
317*495ae853SAndroid Build Coastguard Worker         ps_stream->i4_bits_left_in_cw = WORD_SIZE;
318*495ae853SAndroid Build Coastguard Worker     }
319*495ae853SAndroid Build Coastguard Worker }
320*495ae853SAndroid Build Coastguard Worker 
321*495ae853SAndroid Build Coastguard Worker /**
322*495ae853SAndroid Build Coastguard Worker  ******************************************************************************
323*495ae853SAndroid Build Coastguard Worker  *
324*495ae853SAndroid Build Coastguard Worker  *  @brief Puts new byte (and outstanding bytes) into bitstream after cabac
325*495ae853SAndroid Build Coastguard Worker  *         renormalization
326*495ae853SAndroid Build Coastguard Worker  *
327*495ae853SAndroid Build Coastguard Worker  *  @par   Description
328*495ae853SAndroid Build Coastguard Worker  *  1. Extract the leading byte of low(L)
329*495ae853SAndroid Build Coastguard Worker  *  2. If leading byte=0xff increment outstanding bytes and return
330*495ae853SAndroid Build Coastguard Worker  *     (as the actual bits depend on carry propogation later)
331*495ae853SAndroid Build Coastguard Worker  *  3. If leading byte is not 0xff check for any carry propogation
332*495ae853SAndroid Build Coastguard Worker  *  4. Insert the carry (propogated in previous byte) along with outstanding
333*495ae853SAndroid Build Coastguard Worker  *     bytes (if any) and leading byte
334*495ae853SAndroid Build Coastguard Worker  *
335*495ae853SAndroid Build Coastguard Worker  *
336*495ae853SAndroid Build Coastguard Worker  *  @param[in]   ps_cabac_ctxt
337*495ae853SAndroid Build Coastguard Worker  *  pointer to cabac context (handle)
338*495ae853SAndroid Build Coastguard Worker  *
339*495ae853SAndroid Build Coastguard Worker  *  @return
340*495ae853SAndroid Build Coastguard Worker  *
341*495ae853SAndroid Build Coastguard Worker  ******************************************************************************
342*495ae853SAndroid Build Coastguard Worker  */
isvce_cabac_put_byte(isvce_cabac_ctxt_t * ps_cabac_ctxt)343*495ae853SAndroid Build Coastguard Worker void isvce_cabac_put_byte(isvce_cabac_ctxt_t *ps_cabac_ctxt)
344*495ae853SAndroid Build Coastguard Worker {
345*495ae853SAndroid Build Coastguard Worker     /* bit stream ptr */
346*495ae853SAndroid Build Coastguard Worker     bitstrm_t *ps_stream = ps_cabac_ctxt->ps_bitstrm;
347*495ae853SAndroid Build Coastguard Worker     encoding_envirnoment_t *ps_cab_enc_env = &(ps_cabac_ctxt->s_cab_enc_env);
348*495ae853SAndroid Build Coastguard Worker     UWORD32 u4_low = ps_cab_enc_env->u4_code_int_low;
349*495ae853SAndroid Build Coastguard Worker     UWORD32 u4_bits_gen = ps_cab_enc_env->u4_bits_gen;
350*495ae853SAndroid Build Coastguard Worker     WORD32 lead_byte = u4_low >> (u4_bits_gen + CABAC_BITS - 8);
351*495ae853SAndroid Build Coastguard Worker 
352*495ae853SAndroid Build Coastguard Worker     /* Sanity checks */
353*495ae853SAndroid Build Coastguard Worker     ASSERT((ps_cab_enc_env->u4_code_int_range >= 256) && (ps_cab_enc_env->u4_code_int_range < 512));
354*495ae853SAndroid Build Coastguard Worker     ASSERT((u4_bits_gen >= 8));
355*495ae853SAndroid Build Coastguard Worker 
356*495ae853SAndroid Build Coastguard Worker     /* update bits generated and low after extracting leading byte */
357*495ae853SAndroid Build Coastguard Worker     u4_bits_gen -= 8;
358*495ae853SAndroid Build Coastguard Worker     ps_cab_enc_env->u4_code_int_low &= ((1 << (CABAC_BITS + u4_bits_gen)) - 1);
359*495ae853SAndroid Build Coastguard Worker     ps_cab_enc_env->u4_bits_gen = u4_bits_gen;
360*495ae853SAndroid Build Coastguard Worker 
361*495ae853SAndroid Build Coastguard Worker     /************************************************************************/
362*495ae853SAndroid Build Coastguard Worker     /* 1. Extract the leading byte of low(L)                                */
363*495ae853SAndroid Build Coastguard Worker     /* 2. If leading byte=0xff increment outstanding bytes and return       */
364*495ae853SAndroid Build Coastguard Worker     /*      (as the actual bits depend on carry propogation later)          */
365*495ae853SAndroid Build Coastguard Worker     /* 3. If leading byte is not 0xff check for any carry propogation       */
366*495ae853SAndroid Build Coastguard Worker     /* 4. Insert the carry (propogated in previous byte) along with         */
367*495ae853SAndroid Build Coastguard Worker     /*    outstanding bytes (if any) and leading byte                       */
368*495ae853SAndroid Build Coastguard Worker     /************************************************************************/
369*495ae853SAndroid Build Coastguard Worker     if(lead_byte == 0xff)
370*495ae853SAndroid Build Coastguard Worker     {
371*495ae853SAndroid Build Coastguard Worker         /* actual bits depend on carry propogration     */
372*495ae853SAndroid Build Coastguard Worker         ps_cab_enc_env->u4_out_standing_bytes++;
373*495ae853SAndroid Build Coastguard Worker         return;
374*495ae853SAndroid Build Coastguard Worker     }
375*495ae853SAndroid Build Coastguard Worker     else
376*495ae853SAndroid Build Coastguard Worker     {
377*495ae853SAndroid Build Coastguard Worker         UWORD8 *pu1_strm_buf = ps_stream->pu1_strm_buffer;
378*495ae853SAndroid Build Coastguard Worker         UWORD32 u4_strm_buf_offset = ps_stream->u4_strm_buf_offset;
379*495ae853SAndroid Build Coastguard Worker         /* carry = 1 => putbit(1); carry propogated due to L renorm */
380*495ae853SAndroid Build Coastguard Worker         WORD32 carry = (lead_byte >> 8) & 0x1;
381*495ae853SAndroid Build Coastguard Worker         WORD32 zero_run = ps_stream->i4_zero_bytes_run;
382*495ae853SAndroid Build Coastguard Worker         UWORD32 u4_out_standing_bytes = ps_cab_enc_env->u4_out_standing_bytes;
383*495ae853SAndroid Build Coastguard Worker 
384*495ae853SAndroid Build Coastguard Worker         /*********************************************************************/
385*495ae853SAndroid Build Coastguard Worker         /*        Insert the carry propogated in previous byte               */
386*495ae853SAndroid Build Coastguard Worker         /*                                                                   */
387*495ae853SAndroid Build Coastguard Worker         /* Note : Do not worry about corruption into slice header align byte */
388*495ae853SAndroid Build Coastguard Worker         /*        This is because the first bin cannot result in overflow    */
389*495ae853SAndroid Build Coastguard Worker         /*********************************************************************/
390*495ae853SAndroid Build Coastguard Worker         if(carry)
391*495ae853SAndroid Build Coastguard Worker         {
392*495ae853SAndroid Build Coastguard Worker             /* CORNER CASE: if the previous data is 0x000003, then EPB will be
393*495ae853SAndroid Build Coastguard Worker              inserted and the data will become 0x00000303 and if the carry is present,
394*495ae853SAndroid Build Coastguard Worker              it will be added with the last byte and it will become 0x00000304 which
395*495ae853SAndroid Build Coastguard Worker              is not correct as per standard */
396*495ae853SAndroid Build Coastguard Worker             /* so check for previous four bytes and if it is equal to 0x00000303
397*495ae853SAndroid Build Coastguard Worker              then subtract u4_strm_buf_offset by 1 */
398*495ae853SAndroid Build Coastguard Worker             if((u4_strm_buf_offset > 3) && (pu1_strm_buf[u4_strm_buf_offset - 1] == 0x03) &&
399*495ae853SAndroid Build Coastguard Worker                (pu1_strm_buf[u4_strm_buf_offset - 2] == 0x03) &&
400*495ae853SAndroid Build Coastguard Worker                (pu1_strm_buf[u4_strm_buf_offset - 3] == 0x00) &&
401*495ae853SAndroid Build Coastguard Worker                (pu1_strm_buf[u4_strm_buf_offset - 4] == 0x00))
402*495ae853SAndroid Build Coastguard Worker             {
403*495ae853SAndroid Build Coastguard Worker                 u4_strm_buf_offset -= 1;
404*495ae853SAndroid Build Coastguard Worker             }
405*495ae853SAndroid Build Coastguard Worker 
406*495ae853SAndroid Build Coastguard Worker             /* previous byte carry add will not result in overflow to        */
407*495ae853SAndroid Build Coastguard Worker             /* u4_strm_buf_offset - 2 as we track 0xff as outstanding bytes  */
408*495ae853SAndroid Build Coastguard Worker             if(u4_strm_buf_offset > 0)
409*495ae853SAndroid Build Coastguard Worker             {
410*495ae853SAndroid Build Coastguard Worker                 pu1_strm_buf[u4_strm_buf_offset - 1] += carry;
411*495ae853SAndroid Build Coastguard Worker                 zero_run = 0;
412*495ae853SAndroid Build Coastguard Worker             }
413*495ae853SAndroid Build Coastguard Worker         }
414*495ae853SAndroid Build Coastguard Worker 
415*495ae853SAndroid Build Coastguard Worker         /*        Insert outstanding bytes (if any)         */
416*495ae853SAndroid Build Coastguard Worker         while(u4_out_standing_bytes)
417*495ae853SAndroid Build Coastguard Worker         {
418*495ae853SAndroid Build Coastguard Worker             UWORD8 u1_0_or_ff = carry ? 0 : 0xFF;
419*495ae853SAndroid Build Coastguard Worker 
420*495ae853SAndroid Build Coastguard Worker             PUTBYTE_EPB(pu1_strm_buf, u4_strm_buf_offset, u1_0_or_ff, zero_run);
421*495ae853SAndroid Build Coastguard Worker 
422*495ae853SAndroid Build Coastguard Worker             u4_out_standing_bytes--;
423*495ae853SAndroid Build Coastguard Worker         }
424*495ae853SAndroid Build Coastguard Worker         ps_cab_enc_env->u4_out_standing_bytes = 0;
425*495ae853SAndroid Build Coastguard Worker 
426*495ae853SAndroid Build Coastguard Worker         /*        Insert the leading byte                   */
427*495ae853SAndroid Build Coastguard Worker         lead_byte &= 0xFF;
428*495ae853SAndroid Build Coastguard Worker         PUTBYTE_EPB(pu1_strm_buf, u4_strm_buf_offset, lead_byte, zero_run);
429*495ae853SAndroid Build Coastguard Worker 
430*495ae853SAndroid Build Coastguard Worker         /* update the state variables and return success */
431*495ae853SAndroid Build Coastguard Worker         ps_stream->u4_strm_buf_offset = u4_strm_buf_offset;
432*495ae853SAndroid Build Coastguard Worker         ps_stream->i4_zero_bytes_run = zero_run;
433*495ae853SAndroid Build Coastguard Worker     }
434*495ae853SAndroid Build Coastguard Worker }
435*495ae853SAndroid Build Coastguard Worker 
436*495ae853SAndroid Build Coastguard Worker /**
437*495ae853SAndroid Build Coastguard Worker ******************************************************************************
438*495ae853SAndroid Build Coastguard Worker *
439*495ae853SAndroid Build Coastguard Worker *  @brief Codes a bin based on probablilty and mps packed context model
440*495ae853SAndroid Build Coastguard Worker *
441*495ae853SAndroid Build Coastguard Worker *  @par   Description
442*495ae853SAndroid Build Coastguard Worker *  1. Apart from encoding bin, context model is updated as per state transition
443*495ae853SAndroid Build Coastguard Worker *  2. Range and Low renormalization is done based on bin and original state
444*495ae853SAndroid Build Coastguard Worker *  3. After renorm bistream is updated (if required)
445*495ae853SAndroid Build Coastguard Worker *
446*495ae853SAndroid Build Coastguard Worker *  @param[in]   ps_cabac
447*495ae853SAndroid Build Coastguard Worker *  pointer to cabac context (handle)
448*495ae853SAndroid Build Coastguard Worker *
449*495ae853SAndroid Build Coastguard Worker *  @param[in]   bin
450*495ae853SAndroid Build Coastguard Worker *  bin(boolean) to be encoded
451*495ae853SAndroid Build Coastguard Worker *
452*495ae853SAndroid Build Coastguard Worker *  @param[in]  pu1_bin_ctxts
453*495ae853SAndroid Build Coastguard Worker *  index of cabac context model containing pState[bits 5-0] | MPS[bit6]
454*495ae853SAndroid Build Coastguard Worker *
455*495ae853SAndroid Build Coastguard Worker *  @return
456*495ae853SAndroid Build Coastguard Worker *
457*495ae853SAndroid Build Coastguard Worker ******************************************************************************
458*495ae853SAndroid Build Coastguard Worker */
isvce_cabac_encode_bin(isvce_cabac_ctxt_t * ps_cabac,WORD32 bin,bin_ctxt_model * pu1_bin_ctxts)459*495ae853SAndroid Build Coastguard Worker void isvce_cabac_encode_bin(isvce_cabac_ctxt_t *ps_cabac, WORD32 bin, bin_ctxt_model *pu1_bin_ctxts)
460*495ae853SAndroid Build Coastguard Worker {
461*495ae853SAndroid Build Coastguard Worker     encoding_envirnoment_t *ps_cab_enc_env = &(ps_cabac->s_cab_enc_env);
462*495ae853SAndroid Build Coastguard Worker     UWORD32 u4_range = ps_cab_enc_env->u4_code_int_range;
463*495ae853SAndroid Build Coastguard Worker     UWORD32 u4_low = ps_cab_enc_env->u4_code_int_low;
464*495ae853SAndroid Build Coastguard Worker     UWORD32 u4_rlps;
465*495ae853SAndroid Build Coastguard Worker     UWORD8 state_mps = (*pu1_bin_ctxts) & 0x3F;
466*495ae853SAndroid Build Coastguard Worker     UWORD8 u1_mps = !!((*pu1_bin_ctxts) & (0x40));
467*495ae853SAndroid Build Coastguard Worker     WORD32 shift;
468*495ae853SAndroid Build Coastguard Worker     UWORD32 u4_table_val;
469*495ae853SAndroid Build Coastguard Worker     /* Sanity checks */
470*495ae853SAndroid Build Coastguard Worker     ASSERT((bin == 0) || (bin == 1));
471*495ae853SAndroid Build Coastguard Worker     ASSERT((u4_range >= 256) && (u4_range < 512));
472*495ae853SAndroid Build Coastguard Worker 
473*495ae853SAndroid Build Coastguard Worker     /* Get the lps range from LUT based on quantized range and state */
474*495ae853SAndroid Build Coastguard Worker     u4_table_val = gau4_isvc_cabac_table[state_mps][(u4_range >> 6) & 0x3];
475*495ae853SAndroid Build Coastguard Worker     u4_rlps = u4_table_val & 0xFF;
476*495ae853SAndroid Build Coastguard Worker     u4_range -= u4_rlps;
477*495ae853SAndroid Build Coastguard Worker 
478*495ae853SAndroid Build Coastguard Worker     /* check if bin is mps or lps */
479*495ae853SAndroid Build Coastguard Worker     if(u1_mps ^ bin)
480*495ae853SAndroid Build Coastguard Worker     {
481*495ae853SAndroid Build Coastguard Worker         /* lps path;  L= L + R; R = RLPS */
482*495ae853SAndroid Build Coastguard Worker         u4_low += u4_range;
483*495ae853SAndroid Build Coastguard Worker         u4_range = u4_rlps;
484*495ae853SAndroid Build Coastguard Worker         if(state_mps == 0)
485*495ae853SAndroid Build Coastguard Worker         {
486*495ae853SAndroid Build Coastguard Worker             /* MPS(CtxIdx) = 1 - MPS(CtxIdx) */
487*495ae853SAndroid Build Coastguard Worker             u1_mps = 1 - u1_mps;
488*495ae853SAndroid Build Coastguard Worker         } /* update the context model from state transition LUT */
489*495ae853SAndroid Build Coastguard Worker 
490*495ae853SAndroid Build Coastguard Worker         state_mps = (u4_table_val >> 15) & 0x3F;
491*495ae853SAndroid Build Coastguard Worker     }
492*495ae853SAndroid Build Coastguard Worker     else
493*495ae853SAndroid Build Coastguard Worker     { /* update the context model from state transition LUT */
494*495ae853SAndroid Build Coastguard Worker         state_mps = (u4_table_val >> 8) & 0x3F;
495*495ae853SAndroid Build Coastguard Worker     }
496*495ae853SAndroid Build Coastguard Worker 
497*495ae853SAndroid Build Coastguard Worker     (*pu1_bin_ctxts) = (u1_mps << 6) | state_mps;
498*495ae853SAndroid Build Coastguard Worker 
499*495ae853SAndroid Build Coastguard Worker     /*****************************************************************/
500*495ae853SAndroid Build Coastguard Worker     /* Renormalization; calculate bits generated based on range(R)   */
501*495ae853SAndroid Build Coastguard Worker     /* Note : 6 <= R < 512; R is 2 only for terminating encode       */
502*495ae853SAndroid Build Coastguard Worker     /*****************************************************************/
503*495ae853SAndroid Build Coastguard Worker     GETRANGE(shift, u4_range);
504*495ae853SAndroid Build Coastguard Worker     shift = 9 - shift;
505*495ae853SAndroid Build Coastguard Worker     u4_low <<= shift;
506*495ae853SAndroid Build Coastguard Worker     u4_range <<= shift;
507*495ae853SAndroid Build Coastguard Worker 
508*495ae853SAndroid Build Coastguard Worker     /* bits to be inserted in the bitstream */
509*495ae853SAndroid Build Coastguard Worker     ps_cab_enc_env->u4_bits_gen += shift;
510*495ae853SAndroid Build Coastguard Worker     ps_cab_enc_env->u4_code_int_range = u4_range;
511*495ae853SAndroid Build Coastguard Worker     ps_cab_enc_env->u4_code_int_low = u4_low;
512*495ae853SAndroid Build Coastguard Worker 
513*495ae853SAndroid Build Coastguard Worker     /* generate stream when a byte is ready */
514*495ae853SAndroid Build Coastguard Worker     if(ps_cab_enc_env->u4_bits_gen > CABAC_BITS)
515*495ae853SAndroid Build Coastguard Worker     {
516*495ae853SAndroid Build Coastguard Worker         isvce_cabac_put_byte(ps_cabac);
517*495ae853SAndroid Build Coastguard Worker     }
518*495ae853SAndroid Build Coastguard Worker }
519*495ae853SAndroid Build Coastguard Worker 
520*495ae853SAndroid Build Coastguard Worker /**
521*495ae853SAndroid Build Coastguard Worker *******************************************************************************
522*495ae853SAndroid Build Coastguard Worker *
523*495ae853SAndroid Build Coastguard Worker * @brief
524*495ae853SAndroid Build Coastguard Worker *  Encoding process for a binary decision :implements encoding process of a
525*495ae853SAndroid Build Coastguard Worker decision
526*495ae853SAndroid Build Coastguard Worker *  as defined in 9.3.4.2 . This function encodes multiple bins, of a symbol.
527*495ae853SAndroid Build Coastguard Worker Implements
528*495ae853SAndroid Build Coastguard Worker *  flowchart Figure 9-7( ITU_T_H264-201402)
529*495ae853SAndroid Build Coastguard Worker *
530*495ae853SAndroid Build Coastguard Worker * @param[in] u4_bins
531*495ae853SAndroid Build Coastguard Worker * array of bin values
532*495ae853SAndroid Build Coastguard Worker *
533*495ae853SAndroid Build Coastguard Worker * @param[in] i1_bins_len
534*495ae853SAndroid Build Coastguard Worker *  Length of bins, maximum 32
535*495ae853SAndroid Build Coastguard Worker *
536*495ae853SAndroid Build Coastguard Worker * @param[in] u4_ctx_inc
537*495ae853SAndroid Build Coastguard Worker *  CtxInc, byte0- bin0, byte1-bin1 ..
538*495ae853SAndroid Build Coastguard Worker *
539*495ae853SAndroid Build Coastguard Worker * @param[in] i1_valid_len
540*495ae853SAndroid Build Coastguard Worker *  valid length of bins, after that CtxInc is constant
541*495ae853SAndroid Build Coastguard Worker *
542*495ae853SAndroid Build Coastguard Worker * @param[in] pu1_bin_ctxt_type
543*495ae853SAndroid Build Coastguard Worker *  Pointer to binary contexts
544*495ae853SAndroid Build Coastguard Worker 
545*495ae853SAndroid Build Coastguard Worker * @param[in] ps_cabac
546*495ae853SAndroid Build Coastguard Worker *  Pointer to cabac_context_structure
547*495ae853SAndroid Build Coastguard Worker *
548*495ae853SAndroid Build Coastguard Worker * @returns
549*495ae853SAndroid Build Coastguard Worker *
550*495ae853SAndroid Build Coastguard Worker * @remarks
551*495ae853SAndroid Build Coastguard Worker *  None
552*495ae853SAndroid Build Coastguard Worker *
553*495ae853SAndroid Build Coastguard Worker *******************************************************************************
554*495ae853SAndroid Build Coastguard Worker */
isvce_encode_decision_bins(UWORD32 u4_bins,WORD8 i1_bins_len,UWORD32 u4_ctx_inc,WORD8 i1_valid_len,bin_ctxt_model * pu1_bin_ctxt_type,isvce_cabac_ctxt_t * ps_cabac)555*495ae853SAndroid Build Coastguard Worker void isvce_encode_decision_bins(UWORD32 u4_bins, WORD8 i1_bins_len, UWORD32 u4_ctx_inc,
556*495ae853SAndroid Build Coastguard Worker                                 WORD8 i1_valid_len, bin_ctxt_model *pu1_bin_ctxt_type,
557*495ae853SAndroid Build Coastguard Worker                                 isvce_cabac_ctxt_t *ps_cabac)
558*495ae853SAndroid Build Coastguard Worker {
559*495ae853SAndroid Build Coastguard Worker     WORD8 i;
560*495ae853SAndroid Build Coastguard Worker     UWORD8 u1_ctx_inc, u1_bin;
561*495ae853SAndroid Build Coastguard Worker 
562*495ae853SAndroid Build Coastguard Worker     for(i = 0; i < i1_bins_len; i++)
563*495ae853SAndroid Build Coastguard Worker     {
564*495ae853SAndroid Build Coastguard Worker         u1_bin = (u4_bins & 0x01);
565*495ae853SAndroid Build Coastguard Worker         u4_bins = u4_bins >> 1;
566*495ae853SAndroid Build Coastguard Worker         u1_ctx_inc = u4_ctx_inc & 0x0f;
567*495ae853SAndroid Build Coastguard Worker         if(i < i1_valid_len) u4_ctx_inc = u4_ctx_inc >> 4;
568*495ae853SAndroid Build Coastguard Worker         /* Encode the bin */
569*495ae853SAndroid Build Coastguard Worker         isvce_cabac_encode_bin(ps_cabac, u1_bin, pu1_bin_ctxt_type + u1_ctx_inc);
570*495ae853SAndroid Build Coastguard Worker     }
571*495ae853SAndroid Build Coastguard Worker }
572*495ae853SAndroid Build Coastguard Worker 
573*495ae853SAndroid Build Coastguard Worker /**
574*495ae853SAndroid Build Coastguard Worker  *******************************************************************************
575*495ae853SAndroid Build Coastguard Worker  * @brief
576*495ae853SAndroid Build Coastguard Worker  *  Encoding process for a binary decision before termination:Encoding process
577*495ae853SAndroid Build Coastguard Worker  *  of a termination(9.3.4.5 :ITU_T_H264-201402) . Explained in flowchart 9-11.
578*495ae853SAndroid Build Coastguard Worker  *
579*495ae853SAndroid Build Coastguard Worker  * @param[in] ps_cabac
580*495ae853SAndroid Build Coastguard Worker  *  Pointer to cabac structure
581*495ae853SAndroid Build Coastguard Worker  *
582*495ae853SAndroid Build Coastguard Worker  * @param[in] term_bin
583*495ae853SAndroid Build Coastguard Worker  *  Symbol value, end of slice or not, term_bin is binary
584*495ae853SAndroid Build Coastguard Worker  *
585*495ae853SAndroid Build Coastguard Worker  * @returns
586*495ae853SAndroid Build Coastguard Worker  *
587*495ae853SAndroid Build Coastguard Worker  * @remarks
588*495ae853SAndroid Build Coastguard Worker  *  None
589*495ae853SAndroid Build Coastguard Worker  *
590*495ae853SAndroid Build Coastguard Worker  *******************************************************************************
591*495ae853SAndroid Build Coastguard Worker  */
isvce_cabac_encode_terminate(isvce_cabac_ctxt_t * ps_cabac,WORD32 term_bin)592*495ae853SAndroid Build Coastguard Worker void isvce_cabac_encode_terminate(isvce_cabac_ctxt_t *ps_cabac, WORD32 term_bin)
593*495ae853SAndroid Build Coastguard Worker {
594*495ae853SAndroid Build Coastguard Worker     encoding_envirnoment_t *ps_cab_enc_env = &(ps_cabac->s_cab_enc_env);
595*495ae853SAndroid Build Coastguard Worker 
596*495ae853SAndroid Build Coastguard Worker     UWORD32 u4_range = ps_cab_enc_env->u4_code_int_range;
597*495ae853SAndroid Build Coastguard Worker     UWORD32 u4_low = ps_cab_enc_env->u4_code_int_low;
598*495ae853SAndroid Build Coastguard Worker     UWORD32 u4_rlps;
599*495ae853SAndroid Build Coastguard Worker     WORD32 shift;
600*495ae853SAndroid Build Coastguard Worker 
601*495ae853SAndroid Build Coastguard Worker     /* Sanity checks */
602*495ae853SAndroid Build Coastguard Worker     ASSERT((u4_range >= 256) && (u4_range < 512));
603*495ae853SAndroid Build Coastguard Worker     ASSERT((term_bin == 0) || (term_bin == 1));
604*495ae853SAndroid Build Coastguard Worker 
605*495ae853SAndroid Build Coastguard Worker     /*  term_bin = 1 has lps range = 2 */
606*495ae853SAndroid Build Coastguard Worker     u4_rlps = 2;
607*495ae853SAndroid Build Coastguard Worker     u4_range -= u4_rlps;
608*495ae853SAndroid Build Coastguard Worker 
609*495ae853SAndroid Build Coastguard Worker     /* if terminate L is incremented by curR and R=2 */
610*495ae853SAndroid Build Coastguard Worker     if(term_bin)
611*495ae853SAndroid Build Coastguard Worker     {
612*495ae853SAndroid Build Coastguard Worker         /* lps path;  L= L + R; R = RLPS */
613*495ae853SAndroid Build Coastguard Worker         u4_low += u4_range;
614*495ae853SAndroid Build Coastguard Worker         u4_range = u4_rlps;
615*495ae853SAndroid Build Coastguard Worker     }
616*495ae853SAndroid Build Coastguard Worker 
617*495ae853SAndroid Build Coastguard Worker     /*****************************************************************/
618*495ae853SAndroid Build Coastguard Worker     /* Renormalization; calculate bits generated based on range(R)   */
619*495ae853SAndroid Build Coastguard Worker     /* Note : 6 <= R < 512; R is 2 only for terminating encode       */
620*495ae853SAndroid Build Coastguard Worker     /*****************************************************************/
621*495ae853SAndroid Build Coastguard Worker     GETRANGE(shift, u4_range);
622*495ae853SAndroid Build Coastguard Worker     shift = 9 - shift;
623*495ae853SAndroid Build Coastguard Worker     u4_low <<= shift;
624*495ae853SAndroid Build Coastguard Worker     u4_range <<= shift;
625*495ae853SAndroid Build Coastguard Worker 
626*495ae853SAndroid Build Coastguard Worker     /* bits to be inserted in the bitstream */
627*495ae853SAndroid Build Coastguard Worker     ps_cab_enc_env->u4_bits_gen += shift;
628*495ae853SAndroid Build Coastguard Worker     ps_cab_enc_env->u4_code_int_range = u4_range;
629*495ae853SAndroid Build Coastguard Worker     ps_cab_enc_env->u4_code_int_low = u4_low;
630*495ae853SAndroid Build Coastguard Worker 
631*495ae853SAndroid Build Coastguard Worker     /* generate stream when a byte is ready */
632*495ae853SAndroid Build Coastguard Worker     if(ps_cab_enc_env->u4_bits_gen > CABAC_BITS)
633*495ae853SAndroid Build Coastguard Worker     {
634*495ae853SAndroid Build Coastguard Worker         isvce_cabac_put_byte(ps_cabac);
635*495ae853SAndroid Build Coastguard Worker     }
636*495ae853SAndroid Build Coastguard Worker 
637*495ae853SAndroid Build Coastguard Worker     if(term_bin)
638*495ae853SAndroid Build Coastguard Worker     {
639*495ae853SAndroid Build Coastguard Worker         isvce_cabac_flush(ps_cabac);
640*495ae853SAndroid Build Coastguard Worker     }
641*495ae853SAndroid Build Coastguard Worker }
642*495ae853SAndroid Build Coastguard Worker 
643*495ae853SAndroid Build Coastguard Worker /**
644*495ae853SAndroid Build Coastguard Worker  *******************************************************************************
645*495ae853SAndroid Build Coastguard Worker  * @brief
646*495ae853SAndroid Build Coastguard Worker  * Bypass encoding process for binary decisions:  Explained (9.3.4.4
647*495ae853SAndroid Build Coastguard Worker  *:ITU_T_H264-201402) , flowchart 9-10.
648*495ae853SAndroid Build Coastguard Worker  *
649*495ae853SAndroid Build Coastguard Worker  *  @param[ino]  ps_cabac : pointer to cabac context (handle)
650*495ae853SAndroid Build Coastguard Worker  *
651*495ae853SAndroid Build Coastguard Worker  *  @param[in]   bin :  bypass bin(0/1) to be encoded
652*495ae853SAndroid Build Coastguard Worker  *
653*495ae853SAndroid Build Coastguard Worker  *  @returns
654*495ae853SAndroid Build Coastguard Worker  *
655*495ae853SAndroid Build Coastguard Worker  *  @remarks
656*495ae853SAndroid Build Coastguard Worker  *  None
657*495ae853SAndroid Build Coastguard Worker  *
658*495ae853SAndroid Build Coastguard Worker  *******************************************************************************
659*495ae853SAndroid Build Coastguard Worker  */
660*495ae853SAndroid Build Coastguard Worker 
isvce_cabac_encode_bypass_bin(isvce_cabac_ctxt_t * ps_cabac,WORD32 bin)661*495ae853SAndroid Build Coastguard Worker void isvce_cabac_encode_bypass_bin(isvce_cabac_ctxt_t *ps_cabac, WORD32 bin)
662*495ae853SAndroid Build Coastguard Worker {
663*495ae853SAndroid Build Coastguard Worker     encoding_envirnoment_t *ps_cab_enc_env = &(ps_cabac->s_cab_enc_env);
664*495ae853SAndroid Build Coastguard Worker 
665*495ae853SAndroid Build Coastguard Worker     UWORD32 u4_range = ps_cab_enc_env->u4_code_int_range;
666*495ae853SAndroid Build Coastguard Worker     UWORD32 u4_low = ps_cab_enc_env->u4_code_int_low;
667*495ae853SAndroid Build Coastguard Worker 
668*495ae853SAndroid Build Coastguard Worker     /* Sanity checks */
669*495ae853SAndroid Build Coastguard Worker     ASSERT((u4_range >= 256) && (u4_range < 512));
670*495ae853SAndroid Build Coastguard Worker     ASSERT((bin == 0) || (bin == 1));
671*495ae853SAndroid Build Coastguard Worker 
672*495ae853SAndroid Build Coastguard Worker     u4_low <<= 1;
673*495ae853SAndroid Build Coastguard Worker     /* add range if bin is 1 */
674*495ae853SAndroid Build Coastguard Worker     if(bin)
675*495ae853SAndroid Build Coastguard Worker     {
676*495ae853SAndroid Build Coastguard Worker         u4_low += u4_range;
677*495ae853SAndroid Build Coastguard Worker     }
678*495ae853SAndroid Build Coastguard Worker 
679*495ae853SAndroid Build Coastguard Worker     /* 1 bit to be inserted in the bitstream */
680*495ae853SAndroid Build Coastguard Worker     ps_cab_enc_env->u4_bits_gen++;
681*495ae853SAndroid Build Coastguard Worker     ps_cab_enc_env->u4_code_int_low = u4_low;
682*495ae853SAndroid Build Coastguard Worker 
683*495ae853SAndroid Build Coastguard Worker     /* generate stream when a byte is ready */
684*495ae853SAndroid Build Coastguard Worker     if(ps_cab_enc_env->u4_bits_gen > CABAC_BITS)
685*495ae853SAndroid Build Coastguard Worker     {
686*495ae853SAndroid Build Coastguard Worker         isvce_cabac_put_byte(ps_cabac);
687*495ae853SAndroid Build Coastguard Worker     }
688*495ae853SAndroid Build Coastguard Worker }
689*495ae853SAndroid Build Coastguard Worker 
690*495ae853SAndroid Build Coastguard Worker /**
691*495ae853SAndroid Build Coastguard Worker ******************************************************************************
692*495ae853SAndroid Build Coastguard Worker *
693*495ae853SAndroid Build Coastguard Worker *  @brief Encodes a series of bypass bins (FLC bypass bins)
694*495ae853SAndroid Build Coastguard Worker *
695*495ae853SAndroid Build Coastguard Worker *  @par   Description
696*495ae853SAndroid Build Coastguard Worker *  This function is more optimal than calling isvce_cabac_encode_bypass_bin()
697*495ae853SAndroid Build Coastguard Worker *  in a loop as cabac low, renorm and generating the stream (8bins at a time)
698*495ae853SAndroid Build Coastguard Worker *  can be done in one operation
699*495ae853SAndroid Build Coastguard Worker *
700*495ae853SAndroid Build Coastguard Worker *  @param[inout]ps_cabac
701*495ae853SAndroid Build Coastguard Worker *   pointer to cabac context (handle)
702*495ae853SAndroid Build Coastguard Worker *
703*495ae853SAndroid Build Coastguard Worker *  @param[in]   u4_bins
704*495ae853SAndroid Build Coastguard Worker *   syntax element to be coded (as FLC bins)
705*495ae853SAndroid Build Coastguard Worker *
706*495ae853SAndroid Build Coastguard Worker *  @param[in]   num_bins
707*495ae853SAndroid Build Coastguard Worker *   This is the FLC length for u4_sym
708*495ae853SAndroid Build Coastguard Worker *
709*495ae853SAndroid Build Coastguard Worker *  @return
710*495ae853SAndroid Build Coastguard Worker *
711*495ae853SAndroid Build Coastguard Worker ******************************************************************************
712*495ae853SAndroid Build Coastguard Worker */
713*495ae853SAndroid Build Coastguard Worker 
isvce_cabac_encode_bypass_bins(isvce_cabac_ctxt_t * ps_cabac,UWORD32 u4_bins,WORD32 num_bins)714*495ae853SAndroid Build Coastguard Worker void isvce_cabac_encode_bypass_bins(isvce_cabac_ctxt_t *ps_cabac, UWORD32 u4_bins, WORD32 num_bins)
715*495ae853SAndroid Build Coastguard Worker {
716*495ae853SAndroid Build Coastguard Worker     encoding_envirnoment_t *ps_cab_enc_env = &(ps_cabac->s_cab_enc_env);
717*495ae853SAndroid Build Coastguard Worker 
718*495ae853SAndroid Build Coastguard Worker     UWORD32 u4_range = ps_cab_enc_env->u4_code_int_range;
719*495ae853SAndroid Build Coastguard Worker     WORD32 next_byte;
720*495ae853SAndroid Build Coastguard Worker 
721*495ae853SAndroid Build Coastguard Worker     /* Sanity checks */
722*495ae853SAndroid Build Coastguard Worker     ASSERT((num_bins < 33) && (num_bins > 0));
723*495ae853SAndroid Build Coastguard Worker     ASSERT((u4_range >= 256) && (u4_range < 512));
724*495ae853SAndroid Build Coastguard Worker 
725*495ae853SAndroid Build Coastguard Worker     /* Compute bit always to populate the trace */
726*495ae853SAndroid Build Coastguard Worker     /* increment bits generated by num_bins */
727*495ae853SAndroid Build Coastguard Worker 
728*495ae853SAndroid Build Coastguard Worker     /* Encode 8bins at a time and put in the bit-stream */
729*495ae853SAndroid Build Coastguard Worker     while(num_bins > 8)
730*495ae853SAndroid Build Coastguard Worker     {
731*495ae853SAndroid Build Coastguard Worker         num_bins -= 8;
732*495ae853SAndroid Build Coastguard Worker 
733*495ae853SAndroid Build Coastguard Worker         next_byte = (u4_bins >> (num_bins)) & 0xff;
734*495ae853SAndroid Build Coastguard Worker 
735*495ae853SAndroid Build Coastguard Worker         /*  L = (L << 8) +  (R * next_byte) */
736*495ae853SAndroid Build Coastguard Worker         ps_cab_enc_env->u4_code_int_low <<= 8;
737*495ae853SAndroid Build Coastguard Worker         ps_cab_enc_env->u4_code_int_low += (next_byte * u4_range);
738*495ae853SAndroid Build Coastguard Worker         ps_cab_enc_env->u4_bits_gen += 8;
739*495ae853SAndroid Build Coastguard Worker 
740*495ae853SAndroid Build Coastguard Worker         if(ps_cab_enc_env->u4_bits_gen > CABAC_BITS)
741*495ae853SAndroid Build Coastguard Worker         {
742*495ae853SAndroid Build Coastguard Worker             /*  insert the leading byte of low into stream */
743*495ae853SAndroid Build Coastguard Worker             isvce_cabac_put_byte(ps_cabac);
744*495ae853SAndroid Build Coastguard Worker         }
745*495ae853SAndroid Build Coastguard Worker     }
746*495ae853SAndroid Build Coastguard Worker 
747*495ae853SAndroid Build Coastguard Worker     /* Update low with remaining bins and return */
748*495ae853SAndroid Build Coastguard Worker     next_byte = (u4_bins & ((1 << num_bins) - 1));
749*495ae853SAndroid Build Coastguard Worker 
750*495ae853SAndroid Build Coastguard Worker     ps_cab_enc_env->u4_code_int_low <<= num_bins;
751*495ae853SAndroid Build Coastguard Worker     ps_cab_enc_env->u4_code_int_low += (next_byte * u4_range);
752*495ae853SAndroid Build Coastguard Worker     ps_cab_enc_env->u4_bits_gen += num_bins;
753*495ae853SAndroid Build Coastguard Worker 
754*495ae853SAndroid Build Coastguard Worker     if(ps_cab_enc_env->u4_bits_gen > CABAC_BITS)
755*495ae853SAndroid Build Coastguard Worker     {
756*495ae853SAndroid Build Coastguard Worker         /*  insert the leading byte of low into stream */
757*495ae853SAndroid Build Coastguard Worker         isvce_cabac_put_byte(ps_cabac);
758*495ae853SAndroid Build Coastguard Worker     }
759*495ae853SAndroid Build Coastguard Worker }
760