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