xref: /aosp_15_r20/external/libhevc/decoder/ihevcd_nal.c (revision c83a76b084498d55f252f48b2e3786804cdf24b7)
1*c83a76b0SSuyog Pawar /******************************************************************************
2*c83a76b0SSuyog Pawar *
3*c83a76b0SSuyog Pawar * Copyright (C) 2012 Ittiam Systems Pvt Ltd, Bangalore
4*c83a76b0SSuyog Pawar *
5*c83a76b0SSuyog Pawar * Licensed under the Apache License, Version 2.0 (the "License");
6*c83a76b0SSuyog Pawar * you may not use this file except in compliance with the License.
7*c83a76b0SSuyog Pawar * You may obtain a copy of the License at:
8*c83a76b0SSuyog Pawar *
9*c83a76b0SSuyog Pawar * http://www.apache.org/licenses/LICENSE-2.0
10*c83a76b0SSuyog Pawar *
11*c83a76b0SSuyog Pawar * Unless required by applicable law or agreed to in writing, software
12*c83a76b0SSuyog Pawar * distributed under the License is distributed on an "AS IS" BASIS,
13*c83a76b0SSuyog Pawar * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14*c83a76b0SSuyog Pawar * See the License for the specific language governing permissions and
15*c83a76b0SSuyog Pawar * limitations under the License.
16*c83a76b0SSuyog Pawar *
17*c83a76b0SSuyog Pawar ******************************************************************************/
18*c83a76b0SSuyog Pawar /**
19*c83a76b0SSuyog Pawar *******************************************************************************
20*c83a76b0SSuyog Pawar * @file
21*c83a76b0SSuyog Pawar *  ihevcd_nal.c
22*c83a76b0SSuyog Pawar *
23*c83a76b0SSuyog Pawar * @brief
24*c83a76b0SSuyog Pawar *  Contains functions for NAL level such as search start code etc
25*c83a76b0SSuyog Pawar *
26*c83a76b0SSuyog Pawar * @author
27*c83a76b0SSuyog Pawar *  Harish
28*c83a76b0SSuyog Pawar *
29*c83a76b0SSuyog Pawar * @par List of Functions:
30*c83a76b0SSuyog Pawar *
31*c83a76b0SSuyog Pawar * @remarks
32*c83a76b0SSuyog Pawar *  None
33*c83a76b0SSuyog Pawar *
34*c83a76b0SSuyog Pawar *******************************************************************************
35*c83a76b0SSuyog Pawar */
36*c83a76b0SSuyog Pawar /*****************************************************************************/
37*c83a76b0SSuyog Pawar /* File Includes                                                             */
38*c83a76b0SSuyog Pawar /*****************************************************************************/
39*c83a76b0SSuyog Pawar #include <stdio.h>
40*c83a76b0SSuyog Pawar #include <stddef.h>
41*c83a76b0SSuyog Pawar #include <stdlib.h>
42*c83a76b0SSuyog Pawar #include <string.h>
43*c83a76b0SSuyog Pawar #include <assert.h>
44*c83a76b0SSuyog Pawar 
45*c83a76b0SSuyog Pawar #include "ihevc_typedefs.h"
46*c83a76b0SSuyog Pawar #include "iv.h"
47*c83a76b0SSuyog Pawar #include "ivd.h"
48*c83a76b0SSuyog Pawar #include "ihevcd_cxa.h"
49*c83a76b0SSuyog Pawar 
50*c83a76b0SSuyog Pawar #include "ihevc_defs.h"
51*c83a76b0SSuyog Pawar #include "ihevc_debug.h"
52*c83a76b0SSuyog Pawar #include "ihevc_structs.h"
53*c83a76b0SSuyog Pawar #include "ihevc_macros.h"
54*c83a76b0SSuyog Pawar #include "ihevc_platform_macros.h"
55*c83a76b0SSuyog Pawar #include "ihevc_cabac_tables.h"
56*c83a76b0SSuyog Pawar 
57*c83a76b0SSuyog Pawar 
58*c83a76b0SSuyog Pawar #include "ihevcd_defs.h"
59*c83a76b0SSuyog Pawar #include "ihevcd_function_selector.h"
60*c83a76b0SSuyog Pawar #include "ihevcd_structs.h"
61*c83a76b0SSuyog Pawar #include "ihevcd_error.h"
62*c83a76b0SSuyog Pawar #include "ihevcd_nal.h"
63*c83a76b0SSuyog Pawar #include "ihevcd_bitstream.h"
64*c83a76b0SSuyog Pawar #include "ihevcd_parse_headers.h"
65*c83a76b0SSuyog Pawar #include "ihevcd_parse_slice.h"
66*c83a76b0SSuyog Pawar #include "ihevcd_debug.h"
67*c83a76b0SSuyog Pawar /*****************************************************************************/
68*c83a76b0SSuyog Pawar /* Function Prototypes                                                       */
69*c83a76b0SSuyog Pawar /*****************************************************************************/
70*c83a76b0SSuyog Pawar 
71*c83a76b0SSuyog Pawar /**
72*c83a76b0SSuyog Pawar *******************************************************************************
73*c83a76b0SSuyog Pawar *
74*c83a76b0SSuyog Pawar * @brief
75*c83a76b0SSuyog Pawar *  Search start code from the given buffer pointer
76*c83a76b0SSuyog Pawar *
77*c83a76b0SSuyog Pawar * @par Description:
78*c83a76b0SSuyog Pawar *  Search for start code  Return the offset of start code if start code is
79*c83a76b0SSuyog Pawar * found  If no start code is found till end of given bitstream  then treat
80*c83a76b0SSuyog Pawar * it as invalid NAL and return end of buffer as  offset
81*c83a76b0SSuyog Pawar *
82*c83a76b0SSuyog Pawar * @param[in] pu1_buf
83*c83a76b0SSuyog Pawar *  Pointer to bitstream
84*c83a76b0SSuyog Pawar *
85*c83a76b0SSuyog Pawar * @param[in] bytes_remaining
86*c83a76b0SSuyog Pawar *  Number of bytes remaining in the buffer
87*c83a76b0SSuyog Pawar *
88*c83a76b0SSuyog Pawar * @returns  Offset to the first byte in NAL after start code
89*c83a76b0SSuyog Pawar *
90*c83a76b0SSuyog Pawar * @remarks
91*c83a76b0SSuyog Pawar *  Incomplete start code at the end of input bitstream is  not handled. This
92*c83a76b0SSuyog Pawar * has to be taken care outside this func
93*c83a76b0SSuyog Pawar *
94*c83a76b0SSuyog Pawar *******************************************************************************
95*c83a76b0SSuyog Pawar */
ihevcd_nal_search_start_code(UWORD8 * pu1_buf,WORD32 bytes_remaining)96*c83a76b0SSuyog Pawar WORD32 ihevcd_nal_search_start_code(UWORD8 *pu1_buf, WORD32 bytes_remaining)
97*c83a76b0SSuyog Pawar {
98*c83a76b0SSuyog Pawar     WORD32 ofst;
99*c83a76b0SSuyog Pawar 
100*c83a76b0SSuyog Pawar     WORD32 zero_byte_cnt;
101*c83a76b0SSuyog Pawar     WORD32 start_code_found;
102*c83a76b0SSuyog Pawar 
103*c83a76b0SSuyog Pawar     ofst = -1;
104*c83a76b0SSuyog Pawar 
105*c83a76b0SSuyog Pawar     zero_byte_cnt = 0;
106*c83a76b0SSuyog Pawar     start_code_found = 0;
107*c83a76b0SSuyog Pawar     while(ofst < (bytes_remaining - 1))
108*c83a76b0SSuyog Pawar     {
109*c83a76b0SSuyog Pawar         ofst++;
110*c83a76b0SSuyog Pawar         if(pu1_buf[ofst] != 0)
111*c83a76b0SSuyog Pawar         {
112*c83a76b0SSuyog Pawar             zero_byte_cnt = 0;
113*c83a76b0SSuyog Pawar             continue;
114*c83a76b0SSuyog Pawar         }
115*c83a76b0SSuyog Pawar 
116*c83a76b0SSuyog Pawar         zero_byte_cnt++;
117*c83a76b0SSuyog Pawar         if((ofst < (bytes_remaining - 1)) &&
118*c83a76b0SSuyog Pawar            (pu1_buf[ofst + 1] == START_CODE_PREFIX_BYTE) &&
119*c83a76b0SSuyog Pawar            (zero_byte_cnt >= NUM_ZEROS_BEFORE_START_CODE))
120*c83a76b0SSuyog Pawar         {
121*c83a76b0SSuyog Pawar             /* Found the start code */
122*c83a76b0SSuyog Pawar             ofst++;
123*c83a76b0SSuyog Pawar             start_code_found = 1;
124*c83a76b0SSuyog Pawar             break;
125*c83a76b0SSuyog Pawar         }
126*c83a76b0SSuyog Pawar     }
127*c83a76b0SSuyog Pawar     if((0 == start_code_found) && (ofst < bytes_remaining))
128*c83a76b0SSuyog Pawar     {
129*c83a76b0SSuyog Pawar         if((START_CODE_PREFIX_BYTE == pu1_buf[ofst]) &&
130*c83a76b0SSuyog Pawar            (zero_byte_cnt >= NUM_ZEROS_BEFORE_START_CODE))
131*c83a76b0SSuyog Pawar         {
132*c83a76b0SSuyog Pawar             /* Found a start code at the end*/
133*c83a76b0SSuyog Pawar             ofst++;
134*c83a76b0SSuyog Pawar         }
135*c83a76b0SSuyog Pawar     }
136*c83a76b0SSuyog Pawar     /* Since ofst started at -1, increment it by 1 */
137*c83a76b0SSuyog Pawar     ofst++;
138*c83a76b0SSuyog Pawar 
139*c83a76b0SSuyog Pawar     return ofst;
140*c83a76b0SSuyog Pawar }
141*c83a76b0SSuyog Pawar 
142*c83a76b0SSuyog Pawar /**
143*c83a76b0SSuyog Pawar *******************************************************************************
144*c83a76b0SSuyog Pawar *
145*c83a76b0SSuyog Pawar * @brief
146*c83a76b0SSuyog Pawar *  Remove emulation prevention byte present in the bitstream till next start
147*c83a76b0SSuyog Pawar * code is found. Emulation prevention byte  removed data is stored in a
148*c83a76b0SSuyog Pawar * different buffer
149*c83a76b0SSuyog Pawar *
150*c83a76b0SSuyog Pawar * @par Description:
151*c83a76b0SSuyog Pawar *  Assumption is first start code is already found and  pu1_buf is pointing
152*c83a76b0SSuyog Pawar * to a byte after the start code  Search for Next NAL's start code  Return
153*c83a76b0SSuyog Pawar * if start code is found  Remove any emulation prevention byte present  Copy
154*c83a76b0SSuyog Pawar * data to new buffer  If no start code is found, then treat complete buffer
155*c83a76b0SSuyog Pawar * as  one nal.
156*c83a76b0SSuyog Pawar *
157*c83a76b0SSuyog Pawar * @param[in] pu1_src
158*c83a76b0SSuyog Pawar *  Pointer to bitstream (excludes the initial the start code)
159*c83a76b0SSuyog Pawar *
160*c83a76b0SSuyog Pawar * @param[in] pu1_dst
161*c83a76b0SSuyog Pawar *  Pointer to destination buffer
162*c83a76b0SSuyog Pawar *
163*c83a76b0SSuyog Pawar * @param[in] bytes_remaining
164*c83a76b0SSuyog Pawar *  Number of bytes remaining
165*c83a76b0SSuyog Pawar *
166*c83a76b0SSuyog Pawar * @param[out] pi4_nal_len
167*c83a76b0SSuyog Pawar *  NAL length (length of bitstream parsed)
168*c83a76b0SSuyog Pawar *
169*c83a76b0SSuyog Pawar * @param[out] pi4_dst_len
170*c83a76b0SSuyog Pawar *  Destination bitstream size (length of bitstream parsed with emulation bytes
171*c83a76b0SSuyog Pawar * removed)
172*c83a76b0SSuyog Pawar *
173*c83a76b0SSuyog Pawar * @returns Error code from IHEVCD_ERROR_T
174*c83a76b0SSuyog Pawar *
175*c83a76b0SSuyog Pawar * @remarks
176*c83a76b0SSuyog Pawar *  Incomplete start code at the end of input bitstream is  not handled. This
177*c83a76b0SSuyog Pawar * has to be taken care outside this func
178*c83a76b0SSuyog Pawar *
179*c83a76b0SSuyog Pawar *******************************************************************************
180*c83a76b0SSuyog Pawar */
ihevcd_nal_remv_emuln_bytes(UWORD8 * pu1_src,UWORD8 * pu1_dst,WORD32 bytes_remaining,WORD32 * pi4_nal_len,WORD32 * pi4_dst_len)181*c83a76b0SSuyog Pawar IHEVCD_ERROR_T ihevcd_nal_remv_emuln_bytes(UWORD8 *pu1_src,
182*c83a76b0SSuyog Pawar                                            UWORD8 *pu1_dst,
183*c83a76b0SSuyog Pawar                                            WORD32 bytes_remaining,
184*c83a76b0SSuyog Pawar                                            WORD32 *pi4_nal_len,
185*c83a76b0SSuyog Pawar                                            WORD32 *pi4_dst_len)
186*c83a76b0SSuyog Pawar {
187*c83a76b0SSuyog Pawar     WORD32 src_cnt;
188*c83a76b0SSuyog Pawar     WORD32 dst_cnt;
189*c83a76b0SSuyog Pawar     WORD32 zero_byte_cnt;
190*c83a76b0SSuyog Pawar     WORD32 start_code_found;
191*c83a76b0SSuyog Pawar     UWORD8 u1_src;
192*c83a76b0SSuyog Pawar     IHEVCD_ERROR_T ret = (IHEVCD_ERROR_T)IHEVCD_SUCCESS;
193*c83a76b0SSuyog Pawar 
194*c83a76b0SSuyog Pawar     src_cnt = 0;
195*c83a76b0SSuyog Pawar     dst_cnt = 0;
196*c83a76b0SSuyog Pawar     zero_byte_cnt = 0;
197*c83a76b0SSuyog Pawar     start_code_found = 0;
198*c83a76b0SSuyog Pawar     while(src_cnt < (bytes_remaining - 1))
199*c83a76b0SSuyog Pawar     {
200*c83a76b0SSuyog Pawar         u1_src = pu1_src[src_cnt++];
201*c83a76b0SSuyog Pawar 
202*c83a76b0SSuyog Pawar         pu1_dst[dst_cnt++] = u1_src;
203*c83a76b0SSuyog Pawar         if(u1_src != 0)
204*c83a76b0SSuyog Pawar         {
205*c83a76b0SSuyog Pawar             zero_byte_cnt = 0;
206*c83a76b0SSuyog Pawar             continue;
207*c83a76b0SSuyog Pawar         }
208*c83a76b0SSuyog Pawar 
209*c83a76b0SSuyog Pawar         zero_byte_cnt++;
210*c83a76b0SSuyog Pawar         if(zero_byte_cnt >= NUM_ZEROS_BEFORE_START_CODE)
211*c83a76b0SSuyog Pawar         {
212*c83a76b0SSuyog Pawar             u1_src = pu1_src[src_cnt];
213*c83a76b0SSuyog Pawar             if(START_CODE_PREFIX_BYTE == u1_src)
214*c83a76b0SSuyog Pawar             {
215*c83a76b0SSuyog Pawar                 /* Found the start code */
216*c83a76b0SSuyog Pawar                 src_cnt -= zero_byte_cnt;
217*c83a76b0SSuyog Pawar                 dst_cnt -= zero_byte_cnt;
218*c83a76b0SSuyog Pawar                 start_code_found = 1;
219*c83a76b0SSuyog Pawar                 break;
220*c83a76b0SSuyog Pawar             }
221*c83a76b0SSuyog Pawar             else if(EMULATION_PREVENT_BYTE == u1_src)
222*c83a76b0SSuyog Pawar             {
223*c83a76b0SSuyog Pawar                 /* Found the emulation prevention byte */
224*c83a76b0SSuyog Pawar                 src_cnt++;
225*c83a76b0SSuyog Pawar                 zero_byte_cnt = 0;
226*c83a76b0SSuyog Pawar 
227*c83a76b0SSuyog Pawar                 /* Decrement dst_cnt so that the next byte overwrites
228*c83a76b0SSuyog Pawar                  * the emulation prevention byte already copied to dst above
229*c83a76b0SSuyog Pawar                  */
230*c83a76b0SSuyog Pawar             }
231*c83a76b0SSuyog Pawar         }
232*c83a76b0SSuyog Pawar 
233*c83a76b0SSuyog Pawar     }
234*c83a76b0SSuyog Pawar 
235*c83a76b0SSuyog Pawar     if((0 == start_code_found) && (src_cnt < bytes_remaining))
236*c83a76b0SSuyog Pawar     {
237*c83a76b0SSuyog Pawar         u1_src = pu1_src[src_cnt++];
238*c83a76b0SSuyog Pawar         if(zero_byte_cnt >= NUM_ZEROS_BEFORE_START_CODE)
239*c83a76b0SSuyog Pawar         {
240*c83a76b0SSuyog Pawar 
241*c83a76b0SSuyog Pawar             if(START_CODE_PREFIX_BYTE == u1_src)
242*c83a76b0SSuyog Pawar             {
243*c83a76b0SSuyog Pawar                 /* Found a start code at the end*/
244*c83a76b0SSuyog Pawar                 src_cnt -= zero_byte_cnt;
245*c83a76b0SSuyog Pawar             }
246*c83a76b0SSuyog Pawar             else if(EMULATION_PREVENT_BYTE == u1_src)
247*c83a76b0SSuyog Pawar             {
248*c83a76b0SSuyog Pawar                 /* Found the emulation prevention byte at the end*/
249*c83a76b0SSuyog Pawar                 src_cnt++;
250*c83a76b0SSuyog Pawar                 /* Decrement dst_cnt so that the next byte overwrites
251*c83a76b0SSuyog Pawar                  * the emulation prevention byte already copied to dst above
252*c83a76b0SSuyog Pawar                  */
253*c83a76b0SSuyog Pawar                 dst_cnt--;
254*c83a76b0SSuyog Pawar             }
255*c83a76b0SSuyog Pawar         }
256*c83a76b0SSuyog Pawar         else
257*c83a76b0SSuyog Pawar         {
258*c83a76b0SSuyog Pawar             pu1_dst[dst_cnt++] = u1_src;
259*c83a76b0SSuyog Pawar         }
260*c83a76b0SSuyog Pawar 
261*c83a76b0SSuyog Pawar 
262*c83a76b0SSuyog Pawar     }
263*c83a76b0SSuyog Pawar     *pi4_nal_len = src_cnt;
264*c83a76b0SSuyog Pawar     *pi4_dst_len = dst_cnt;
265*c83a76b0SSuyog Pawar     return ret;
266*c83a76b0SSuyog Pawar }
267*c83a76b0SSuyog Pawar /**
268*c83a76b0SSuyog Pawar *******************************************************************************
269*c83a76b0SSuyog Pawar *
270*c83a76b0SSuyog Pawar * @brief
271*c83a76b0SSuyog Pawar *  Decode given NAL unit's header
272*c83a76b0SSuyog Pawar *
273*c83a76b0SSuyog Pawar * @par Description:
274*c83a76b0SSuyog Pawar *  Call NAL unit's header decode  Section: 7.3.1.2
275*c83a76b0SSuyog Pawar *
276*c83a76b0SSuyog Pawar * @param[in] ps_bitstrm
277*c83a76b0SSuyog Pawar *  Pointer to bitstream context
278*c83a76b0SSuyog Pawar *
279*c83a76b0SSuyog Pawar * @param[out] ps_nal
280*c83a76b0SSuyog Pawar *  Pointer to NAL header
281*c83a76b0SSuyog Pawar *
282*c83a76b0SSuyog Pawar * @returns Error code from IHEVCD_ERROR_T
283*c83a76b0SSuyog Pawar *
284*c83a76b0SSuyog Pawar * @remarks
285*c83a76b0SSuyog Pawar *
286*c83a76b0SSuyog Pawar *
287*c83a76b0SSuyog Pawar *******************************************************************************
288*c83a76b0SSuyog Pawar */
ihevcd_nal_unit_header(bitstrm_t * ps_bitstrm,nal_header_t * ps_nal)289*c83a76b0SSuyog Pawar IHEVCD_ERROR_T ihevcd_nal_unit_header(bitstrm_t *ps_bitstrm, nal_header_t *ps_nal)
290*c83a76b0SSuyog Pawar {
291*c83a76b0SSuyog Pawar     WORD32 unused;
292*c83a76b0SSuyog Pawar     IHEVCD_ERROR_T ret = (IHEVCD_ERROR_T)IHEVCD_SUCCESS;
293*c83a76b0SSuyog Pawar     UNUSED(unused);
294*c83a76b0SSuyog Pawar     /* Syntax : forbidden_zero_bit */
295*c83a76b0SSuyog Pawar     unused = ihevcd_bits_get(ps_bitstrm, 1);
296*c83a76b0SSuyog Pawar 
297*c83a76b0SSuyog Pawar     /* Syntax : nal_unit_type */
298*c83a76b0SSuyog Pawar     ps_nal->i1_nal_unit_type = ihevcd_bits_get(ps_bitstrm, 6);
299*c83a76b0SSuyog Pawar 
300*c83a76b0SSuyog Pawar     /* Syntax : nuh_reserved_zero_6bits */
301*c83a76b0SSuyog Pawar     unused = ihevcd_bits_get(ps_bitstrm, 6);
302*c83a76b0SSuyog Pawar 
303*c83a76b0SSuyog Pawar     /* Syntax : nuh_temporal_id_plus1 */
304*c83a76b0SSuyog Pawar     ps_nal->i1_nuh_temporal_id = (WORD32)ihevcd_bits_get(ps_bitstrm, 3) - 1;
305*c83a76b0SSuyog Pawar 
306*c83a76b0SSuyog Pawar     return ret;
307*c83a76b0SSuyog Pawar 
308*c83a76b0SSuyog Pawar }
309*c83a76b0SSuyog Pawar 
310*c83a76b0SSuyog Pawar /**
311*c83a76b0SSuyog Pawar *******************************************************************************
312*c83a76b0SSuyog Pawar *
313*c83a76b0SSuyog Pawar * @brief
314*c83a76b0SSuyog Pawar *  Decode given NAL
315*c83a76b0SSuyog Pawar *
316*c83a76b0SSuyog Pawar * @par Description:
317*c83a76b0SSuyog Pawar *  Based on the NAL type call appropriate decode function  Section: 7.3.1.1
318*c83a76b0SSuyog Pawar *
319*c83a76b0SSuyog Pawar *
320*c83a76b0SSuyog Pawar * @param[in,out] ps_codec
321*c83a76b0SSuyog Pawar *  Pointer to codec context (Functions called within will modify contents of
322*c83a76b0SSuyog Pawar * ps_codec)
323*c83a76b0SSuyog Pawar *
324*c83a76b0SSuyog Pawar * @returns Error code from IHEVCD_ERROR_T
325*c83a76b0SSuyog Pawar *
326*c83a76b0SSuyog Pawar * @remarks
327*c83a76b0SSuyog Pawar *
328*c83a76b0SSuyog Pawar *
329*c83a76b0SSuyog Pawar *******************************************************************************
330*c83a76b0SSuyog Pawar */
ihevcd_nal_unit(codec_t * ps_codec)331*c83a76b0SSuyog Pawar IHEVCD_ERROR_T ihevcd_nal_unit(codec_t *ps_codec)
332*c83a76b0SSuyog Pawar {
333*c83a76b0SSuyog Pawar     IHEVCD_ERROR_T ret = (IHEVCD_ERROR_T)IHEVCD_SUCCESS;
334*c83a76b0SSuyog Pawar 
335*c83a76b0SSuyog Pawar     /* NAL Header */
336*c83a76b0SSuyog Pawar     nal_header_t s_nal;
337*c83a76b0SSuyog Pawar 
338*c83a76b0SSuyog Pawar     ret = ihevcd_nal_unit_header(&ps_codec->s_parse.s_bitstrm, &s_nal);
339*c83a76b0SSuyog Pawar     RETURN_IF((ret != (IHEVCD_ERROR_T)IHEVCD_SUCCESS), ret);
340*c83a76b0SSuyog Pawar 
341*c83a76b0SSuyog Pawar     if(ps_codec->i4_slice_error)
342*c83a76b0SSuyog Pawar         s_nal.i1_nal_unit_type = ps_codec->s_parse.ps_slice_hdr->i1_nal_unit_type;
343*c83a76b0SSuyog Pawar 
344*c83a76b0SSuyog Pawar     /* Setting RASL Output flag */
345*c83a76b0SSuyog Pawar     switch(s_nal.i1_nal_unit_type)
346*c83a76b0SSuyog Pawar     {
347*c83a76b0SSuyog Pawar         case NAL_BLA_W_LP    :
348*c83a76b0SSuyog Pawar         case NAL_BLA_W_DLP   :
349*c83a76b0SSuyog Pawar         case NAL_BLA_N_LP    :
350*c83a76b0SSuyog Pawar             ps_codec->i4_rasl_output_flag = 0;
351*c83a76b0SSuyog Pawar             break;
352*c83a76b0SSuyog Pawar 
353*c83a76b0SSuyog Pawar         //TODO: After IDR, there is no case of open GOP
354*c83a76b0SSuyog Pawar         //To be fixed appropriately by ignoring RASL only if the
355*c83a76b0SSuyog Pawar         // required references are not found
356*c83a76b0SSuyog Pawar         case NAL_IDR_W_LP    :
357*c83a76b0SSuyog Pawar         case NAL_IDR_N_LP    :
358*c83a76b0SSuyog Pawar             ps_codec->i4_rasl_output_flag = 1;
359*c83a76b0SSuyog Pawar             break;
360*c83a76b0SSuyog Pawar 
361*c83a76b0SSuyog Pawar         case NAL_CRA         :
362*c83a76b0SSuyog Pawar             ps_codec->i4_rasl_output_flag = (0 != ps_codec->i4_cra_as_first_pic) ? 0 : 1;
363*c83a76b0SSuyog Pawar             break;
364*c83a76b0SSuyog Pawar 
365*c83a76b0SSuyog Pawar         default:
366*c83a76b0SSuyog Pawar             break;
367*c83a76b0SSuyog Pawar     }
368*c83a76b0SSuyog Pawar 
369*c83a76b0SSuyog Pawar     switch(s_nal.i1_nal_unit_type)
370*c83a76b0SSuyog Pawar     {
371*c83a76b0SSuyog Pawar         case NAL_BLA_W_LP    :
372*c83a76b0SSuyog Pawar         case NAL_BLA_W_DLP   :
373*c83a76b0SSuyog Pawar         case NAL_BLA_N_LP    :
374*c83a76b0SSuyog Pawar         case NAL_IDR_W_LP    :
375*c83a76b0SSuyog Pawar         case NAL_IDR_N_LP    :
376*c83a76b0SSuyog Pawar         case NAL_CRA         :
377*c83a76b0SSuyog Pawar         case NAL_TRAIL_N     :
378*c83a76b0SSuyog Pawar         case NAL_TRAIL_R     :
379*c83a76b0SSuyog Pawar         case NAL_TSA_N       :
380*c83a76b0SSuyog Pawar         case NAL_TSA_R       :
381*c83a76b0SSuyog Pawar         case NAL_STSA_N      :
382*c83a76b0SSuyog Pawar         case NAL_STSA_R      :
383*c83a76b0SSuyog Pawar         case NAL_RADL_N      :
384*c83a76b0SSuyog Pawar         case NAL_RADL_R      :
385*c83a76b0SSuyog Pawar         case NAL_RASL_N      :
386*c83a76b0SSuyog Pawar         case NAL_RASL_R      :
387*c83a76b0SSuyog Pawar             if(ps_codec->i4_header_mode)
388*c83a76b0SSuyog Pawar                 return IHEVCD_SLICE_IN_HEADER_MODE;
389*c83a76b0SSuyog Pawar 
390*c83a76b0SSuyog Pawar             if((0 == ps_codec->i4_sps_done) ||
391*c83a76b0SSuyog Pawar                             (0 == ps_codec->i4_pps_done))
392*c83a76b0SSuyog Pawar             {
393*c83a76b0SSuyog Pawar                 return IHEVCD_INVALID_HEADER;
394*c83a76b0SSuyog Pawar             }
395*c83a76b0SSuyog Pawar 
396*c83a76b0SSuyog Pawar             ps_codec->i4_header_in_slice_mode = 0;
397*c83a76b0SSuyog Pawar             ps_codec->i4_cra_as_first_pic = 0;
398*c83a76b0SSuyog Pawar 
399*c83a76b0SSuyog Pawar             ret = ihevcd_parse_slice_header(ps_codec, &s_nal);
400*c83a76b0SSuyog Pawar             DEBUG_PRINT_NAL_INFO(ps_codec, s_nal.i1_nal_unit_type);
401*c83a76b0SSuyog Pawar             if(ret == (IHEVCD_ERROR_T)IHEVCD_SUCCESS)
402*c83a76b0SSuyog Pawar             {
403*c83a76b0SSuyog Pawar                 if((s_nal.i1_nal_unit_type != NAL_RASL_N && s_nal.i1_nal_unit_type != NAL_RASL_R) ||
404*c83a76b0SSuyog Pawar                                 ps_codec->i4_rasl_output_flag ||
405*c83a76b0SSuyog Pawar                                 ps_codec->i4_slice_error)
406*c83a76b0SSuyog Pawar                     ret = ihevcd_parse_slice_data(ps_codec);
407*c83a76b0SSuyog Pawar             }
408*c83a76b0SSuyog Pawar             break;
409*c83a76b0SSuyog Pawar 
410*c83a76b0SSuyog Pawar         case NAL_VPS        :
411*c83a76b0SSuyog Pawar             // ret = ihevcd_parse_vps(ps_codec);
412*c83a76b0SSuyog Pawar             DEBUG_PRINT_NAL_INFO(ps_codec, s_nal.i1_nal_unit_type);
413*c83a76b0SSuyog Pawar             break;
414*c83a76b0SSuyog Pawar 
415*c83a76b0SSuyog Pawar         case NAL_SPS        :
416*c83a76b0SSuyog Pawar             if(0 == ps_codec->i4_header_mode)
417*c83a76b0SSuyog Pawar             {
418*c83a76b0SSuyog Pawar                 ps_codec->i4_header_in_slice_mode = 1;
419*c83a76b0SSuyog Pawar                 if(ps_codec->i4_sps_done &&
420*c83a76b0SSuyog Pawar                                 ps_codec->i4_pic_present)
421*c83a76b0SSuyog Pawar                     break;
422*c83a76b0SSuyog Pawar             }
423*c83a76b0SSuyog Pawar 
424*c83a76b0SSuyog Pawar             ret = ihevcd_parse_sps(ps_codec);
425*c83a76b0SSuyog Pawar             if(ret == (IHEVCD_ERROR_T)IHEVCD_SUCCESS)
426*c83a76b0SSuyog Pawar             {
427*c83a76b0SSuyog Pawar                 sps_t *ps_sps = ps_codec->ps_sps_base + MAX_SPS_CNT - 1;
428*c83a76b0SSuyog Pawar                 ihevcd_copy_sps(ps_codec, ps_sps->i1_sps_id, MAX_SPS_CNT - 1);
429*c83a76b0SSuyog Pawar             }
430*c83a76b0SSuyog Pawar             ps_codec->i4_error_code = ret;
431*c83a76b0SSuyog Pawar 
432*c83a76b0SSuyog Pawar             DEBUG_PRINT_NAL_INFO(ps_codec, s_nal.i1_nal_unit_type);
433*c83a76b0SSuyog Pawar             break;
434*c83a76b0SSuyog Pawar 
435*c83a76b0SSuyog Pawar         case NAL_PPS        :
436*c83a76b0SSuyog Pawar             if(0 == ps_codec->i4_header_mode)
437*c83a76b0SSuyog Pawar             {
438*c83a76b0SSuyog Pawar                 ps_codec->i4_header_in_slice_mode = 1;
439*c83a76b0SSuyog Pawar                 if(ps_codec->i4_pps_done &&
440*c83a76b0SSuyog Pawar                                 ps_codec->i4_pic_present)
441*c83a76b0SSuyog Pawar                     break;
442*c83a76b0SSuyog Pawar             }
443*c83a76b0SSuyog Pawar 
444*c83a76b0SSuyog Pawar             ret = ihevcd_parse_pps(ps_codec);
445*c83a76b0SSuyog Pawar             if(ret == (IHEVCD_ERROR_T)IHEVCD_SUCCESS)
446*c83a76b0SSuyog Pawar             {
447*c83a76b0SSuyog Pawar                 pps_t *ps_pps = ps_codec->ps_pps_base + MAX_PPS_CNT - 1;
448*c83a76b0SSuyog Pawar                 ihevcd_copy_pps(ps_codec, ps_pps->i1_pps_id, MAX_PPS_CNT - 1);
449*c83a76b0SSuyog Pawar             }
450*c83a76b0SSuyog Pawar             ps_codec->i4_error_code = ret;
451*c83a76b0SSuyog Pawar             DEBUG_PRINT_NAL_INFO(ps_codec, s_nal.i1_nal_unit_type);
452*c83a76b0SSuyog Pawar             break;
453*c83a76b0SSuyog Pawar 
454*c83a76b0SSuyog Pawar #ifndef DISABLE_SEI
455*c83a76b0SSuyog Pawar         case NAL_PREFIX_SEI:
456*c83a76b0SSuyog Pawar         case NAL_SUFFIX_SEI:
457*c83a76b0SSuyog Pawar             if(IVD_DECODE_HEADER == ps_codec->i4_header_mode)
458*c83a76b0SSuyog Pawar             {
459*c83a76b0SSuyog Pawar                 return IHEVCD_SLICE_IN_HEADER_MODE;
460*c83a76b0SSuyog Pawar             }
461*c83a76b0SSuyog Pawar 
462*c83a76b0SSuyog Pawar             ret = ihevcd_parse_sei(ps_codec, &s_nal);
463*c83a76b0SSuyog Pawar             break;
464*c83a76b0SSuyog Pawar 
465*c83a76b0SSuyog Pawar #endif
466*c83a76b0SSuyog Pawar         case NAL_EOS        :
467*c83a76b0SSuyog Pawar             ps_codec->i4_cra_as_first_pic = 1;
468*c83a76b0SSuyog Pawar             break;
469*c83a76b0SSuyog Pawar 
470*c83a76b0SSuyog Pawar         default:
471*c83a76b0SSuyog Pawar             DEBUG_PRINT_NAL_INFO(ps_codec, s_nal.i1_nal_unit_type);
472*c83a76b0SSuyog Pawar             break;
473*c83a76b0SSuyog Pawar     }
474*c83a76b0SSuyog Pawar 
475*c83a76b0SSuyog Pawar     return ret;
476*c83a76b0SSuyog Pawar }
477*c83a76b0SSuyog Pawar 
478