/****************************************************************************** * * Copyright (C) 2012 Ittiam Systems Pvt Ltd, Bangalore * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at: * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. * ******************************************************************************/ /** ******************************************************************************* * @file * ihevcd_nal.c * * @brief * Contains functions for NAL level such as search start code etc * * @author * Harish * * @par List of Functions: * * @remarks * None * ******************************************************************************* */ /*****************************************************************************/ /* File Includes */ /*****************************************************************************/ #include #include #include #include #include #include "ihevc_typedefs.h" #include "iv.h" #include "ivd.h" #include "ihevcd_cxa.h" #include "ihevc_defs.h" #include "ihevc_debug.h" #include "ihevc_structs.h" #include "ihevc_macros.h" #include "ihevc_platform_macros.h" #include "ihevc_cabac_tables.h" #include "ihevcd_defs.h" #include "ihevcd_function_selector.h" #include "ihevcd_structs.h" #include "ihevcd_error.h" #include "ihevcd_nal.h" #include "ihevcd_bitstream.h" #include "ihevcd_parse_headers.h" #include "ihevcd_parse_slice.h" #include "ihevcd_debug.h" /*****************************************************************************/ /* Function Prototypes */ /*****************************************************************************/ /** ******************************************************************************* * * @brief * Search start code from the given buffer pointer * * @par Description: * Search for start code Return the offset of start code if start code is * found If no start code is found till end of given bitstream then treat * it as invalid NAL and return end of buffer as offset * * @param[in] pu1_buf * Pointer to bitstream * * @param[in] bytes_remaining * Number of bytes remaining in the buffer * * @returns Offset to the first byte in NAL after start code * * @remarks * Incomplete start code at the end of input bitstream is not handled. This * has to be taken care outside this func * ******************************************************************************* */ WORD32 ihevcd_nal_search_start_code(UWORD8 *pu1_buf, WORD32 bytes_remaining) { WORD32 ofst; WORD32 zero_byte_cnt; WORD32 start_code_found; ofst = -1; zero_byte_cnt = 0; start_code_found = 0; while(ofst < (bytes_remaining - 1)) { ofst++; if(pu1_buf[ofst] != 0) { zero_byte_cnt = 0; continue; } zero_byte_cnt++; if((ofst < (bytes_remaining - 1)) && (pu1_buf[ofst + 1] == START_CODE_PREFIX_BYTE) && (zero_byte_cnt >= NUM_ZEROS_BEFORE_START_CODE)) { /* Found the start code */ ofst++; start_code_found = 1; break; } } if((0 == start_code_found) && (ofst < bytes_remaining)) { if((START_CODE_PREFIX_BYTE == pu1_buf[ofst]) && (zero_byte_cnt >= NUM_ZEROS_BEFORE_START_CODE)) { /* Found a start code at the end*/ ofst++; } } /* Since ofst started at -1, increment it by 1 */ ofst++; return ofst; } /** ******************************************************************************* * * @brief * Remove emulation prevention byte present in the bitstream till next start * code is found. Emulation prevention byte removed data is stored in a * different buffer * * @par Description: * Assumption is first start code is already found and pu1_buf is pointing * to a byte after the start code Search for Next NAL's start code Return * if start code is found Remove any emulation prevention byte present Copy * data to new buffer If no start code is found, then treat complete buffer * as one nal. * * @param[in] pu1_src * Pointer to bitstream (excludes the initial the start code) * * @param[in] pu1_dst * Pointer to destination buffer * * @param[in] bytes_remaining * Number of bytes remaining * * @param[out] pi4_nal_len * NAL length (length of bitstream parsed) * * @param[out] pi4_dst_len * Destination bitstream size (length of bitstream parsed with emulation bytes * removed) * * @returns Error code from IHEVCD_ERROR_T * * @remarks * Incomplete start code at the end of input bitstream is not handled. This * has to be taken care outside this func * ******************************************************************************* */ IHEVCD_ERROR_T ihevcd_nal_remv_emuln_bytes(UWORD8 *pu1_src, UWORD8 *pu1_dst, WORD32 bytes_remaining, WORD32 *pi4_nal_len, WORD32 *pi4_dst_len) { WORD32 src_cnt; WORD32 dst_cnt; WORD32 zero_byte_cnt; WORD32 start_code_found; UWORD8 u1_src; IHEVCD_ERROR_T ret = (IHEVCD_ERROR_T)IHEVCD_SUCCESS; src_cnt = 0; dst_cnt = 0; zero_byte_cnt = 0; start_code_found = 0; while(src_cnt < (bytes_remaining - 1)) { u1_src = pu1_src[src_cnt++]; pu1_dst[dst_cnt++] = u1_src; if(u1_src != 0) { zero_byte_cnt = 0; continue; } zero_byte_cnt++; if(zero_byte_cnt >= NUM_ZEROS_BEFORE_START_CODE) { u1_src = pu1_src[src_cnt]; if(START_CODE_PREFIX_BYTE == u1_src) { /* Found the start code */ src_cnt -= zero_byte_cnt; dst_cnt -= zero_byte_cnt; start_code_found = 1; break; } else if(EMULATION_PREVENT_BYTE == u1_src) { /* Found the emulation prevention byte */ src_cnt++; zero_byte_cnt = 0; /* Decrement dst_cnt so that the next byte overwrites * the emulation prevention byte already copied to dst above */ } } } if((0 == start_code_found) && (src_cnt < bytes_remaining)) { u1_src = pu1_src[src_cnt++]; if(zero_byte_cnt >= NUM_ZEROS_BEFORE_START_CODE) { if(START_CODE_PREFIX_BYTE == u1_src) { /* Found a start code at the end*/ src_cnt -= zero_byte_cnt; } else if(EMULATION_PREVENT_BYTE == u1_src) { /* Found the emulation prevention byte at the end*/ src_cnt++; /* Decrement dst_cnt so that the next byte overwrites * the emulation prevention byte already copied to dst above */ dst_cnt--; } } else { pu1_dst[dst_cnt++] = u1_src; } } *pi4_nal_len = src_cnt; *pi4_dst_len = dst_cnt; return ret; } /** ******************************************************************************* * * @brief * Decode given NAL unit's header * * @par Description: * Call NAL unit's header decode Section: 7.3.1.2 * * @param[in] ps_bitstrm * Pointer to bitstream context * * @param[out] ps_nal * Pointer to NAL header * * @returns Error code from IHEVCD_ERROR_T * * @remarks * * ******************************************************************************* */ IHEVCD_ERROR_T ihevcd_nal_unit_header(bitstrm_t *ps_bitstrm, nal_header_t *ps_nal) { WORD32 unused; IHEVCD_ERROR_T ret = (IHEVCD_ERROR_T)IHEVCD_SUCCESS; UNUSED(unused); /* Syntax : forbidden_zero_bit */ unused = ihevcd_bits_get(ps_bitstrm, 1); /* Syntax : nal_unit_type */ ps_nal->i1_nal_unit_type = ihevcd_bits_get(ps_bitstrm, 6); /* Syntax : nuh_reserved_zero_6bits */ unused = ihevcd_bits_get(ps_bitstrm, 6); /* Syntax : nuh_temporal_id_plus1 */ ps_nal->i1_nuh_temporal_id = (WORD32)ihevcd_bits_get(ps_bitstrm, 3) - 1; return ret; } /** ******************************************************************************* * * @brief * Decode given NAL * * @par Description: * Based on the NAL type call appropriate decode function Section: 7.3.1.1 * * * @param[in,out] ps_codec * Pointer to codec context (Functions called within will modify contents of * ps_codec) * * @returns Error code from IHEVCD_ERROR_T * * @remarks * * ******************************************************************************* */ IHEVCD_ERROR_T ihevcd_nal_unit(codec_t *ps_codec) { IHEVCD_ERROR_T ret = (IHEVCD_ERROR_T)IHEVCD_SUCCESS; /* NAL Header */ nal_header_t s_nal; ret = ihevcd_nal_unit_header(&ps_codec->s_parse.s_bitstrm, &s_nal); RETURN_IF((ret != (IHEVCD_ERROR_T)IHEVCD_SUCCESS), ret); if(ps_codec->i4_slice_error) s_nal.i1_nal_unit_type = ps_codec->s_parse.ps_slice_hdr->i1_nal_unit_type; /* Setting RASL Output flag */ switch(s_nal.i1_nal_unit_type) { case NAL_BLA_W_LP : case NAL_BLA_W_DLP : case NAL_BLA_N_LP : ps_codec->i4_rasl_output_flag = 0; break; //TODO: After IDR, there is no case of open GOP //To be fixed appropriately by ignoring RASL only if the // required references are not found case NAL_IDR_W_LP : case NAL_IDR_N_LP : ps_codec->i4_rasl_output_flag = 1; break; case NAL_CRA : ps_codec->i4_rasl_output_flag = (0 != ps_codec->i4_cra_as_first_pic) ? 0 : 1; break; default: break; } switch(s_nal.i1_nal_unit_type) { case NAL_BLA_W_LP : case NAL_BLA_W_DLP : case NAL_BLA_N_LP : case NAL_IDR_W_LP : case NAL_IDR_N_LP : case NAL_CRA : case NAL_TRAIL_N : case NAL_TRAIL_R : case NAL_TSA_N : case NAL_TSA_R : case NAL_STSA_N : case NAL_STSA_R : case NAL_RADL_N : case NAL_RADL_R : case NAL_RASL_N : case NAL_RASL_R : if(ps_codec->i4_header_mode) return IHEVCD_SLICE_IN_HEADER_MODE; if((0 == ps_codec->i4_sps_done) || (0 == ps_codec->i4_pps_done)) { return IHEVCD_INVALID_HEADER; } ps_codec->i4_header_in_slice_mode = 0; ps_codec->i4_cra_as_first_pic = 0; ret = ihevcd_parse_slice_header(ps_codec, &s_nal); DEBUG_PRINT_NAL_INFO(ps_codec, s_nal.i1_nal_unit_type); if(ret == (IHEVCD_ERROR_T)IHEVCD_SUCCESS) { if((s_nal.i1_nal_unit_type != NAL_RASL_N && s_nal.i1_nal_unit_type != NAL_RASL_R) || ps_codec->i4_rasl_output_flag || ps_codec->i4_slice_error) ret = ihevcd_parse_slice_data(ps_codec); } break; case NAL_VPS : // ret = ihevcd_parse_vps(ps_codec); DEBUG_PRINT_NAL_INFO(ps_codec, s_nal.i1_nal_unit_type); break; case NAL_SPS : if(0 == ps_codec->i4_header_mode) { ps_codec->i4_header_in_slice_mode = 1; if(ps_codec->i4_sps_done && ps_codec->i4_pic_present) break; } ret = ihevcd_parse_sps(ps_codec); if(ret == (IHEVCD_ERROR_T)IHEVCD_SUCCESS) { sps_t *ps_sps = ps_codec->ps_sps_base + MAX_SPS_CNT - 1; ihevcd_copy_sps(ps_codec, ps_sps->i1_sps_id, MAX_SPS_CNT - 1); } ps_codec->i4_error_code = ret; DEBUG_PRINT_NAL_INFO(ps_codec, s_nal.i1_nal_unit_type); break; case NAL_PPS : if(0 == ps_codec->i4_header_mode) { ps_codec->i4_header_in_slice_mode = 1; if(ps_codec->i4_pps_done && ps_codec->i4_pic_present) break; } ret = ihevcd_parse_pps(ps_codec); if(ret == (IHEVCD_ERROR_T)IHEVCD_SUCCESS) { pps_t *ps_pps = ps_codec->ps_pps_base + MAX_PPS_CNT - 1; ihevcd_copy_pps(ps_codec, ps_pps->i1_pps_id, MAX_PPS_CNT - 1); } ps_codec->i4_error_code = ret; DEBUG_PRINT_NAL_INFO(ps_codec, s_nal.i1_nal_unit_type); break; case NAL_PREFIX_SEI: case NAL_SUFFIX_SEI: if(IVD_DECODE_HEADER == ps_codec->i4_header_mode) { return IHEVCD_SLICE_IN_HEADER_MODE; } ret = ihevcd_parse_sei(ps_codec, &s_nal); break; case NAL_EOS : ps_codec->i4_cra_as_first_pic = 1; break; default: DEBUG_PRINT_NAL_INFO(ps_codec, s_nal.i1_nal_unit_type); break; } return ret; }