/****************************************************************************** * * Copyright (C) 2015 The Android Open Source Project * * 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. * ***************************************************************************** * Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore */ /** ****************************************************************************** * @file * ih264e_encode_header.h * * @brief * This file contains structures and interface prototypes for h264 bitstream * header encoding * * @author * ittiam * * @remarks * None * ******************************************************************************* */ #ifndef IH264E_ENCODE_HEADER_H_ #define IH264E_ENCODE_HEADER_H_ /*****************************************************************************/ /* Function Macros */ /*****************************************************************************/ /** ****************************************************************************** * @brief Macro to put a code with specified number of bits into the * bitstream ****************************************************************************** */ #define PUT_BITS(ps_bitstrm, code_val, code_len, ret_val, syntax_string) \ { \ ENTROPY_TRACE(syntax_string, code_val); \ ret_val = ih264e_put_bits((ps_bitstrm), (code_val), (code_len)); \ if(ret_val != IH264E_SUCCESS) \ { \ return ret_val; \ } \ } /** ****************************************************************************** * @brief Macro to put a code with specified number of bits into the * bitstream using 0th order exponential Golomb encoding for * signed numbers ****************************************************************************** */ #define PUT_BITS_UEV(ps_bitstrm, code_val, ret_val, syntax_string) \ { \ ENTROPY_TRACE(syntax_string, code_val); \ ret_val = ih264e_put_uev((ps_bitstrm), (code_val)); \ if(ret_val != IH264E_SUCCESS) \ { \ return ret_val; \ } \ } /** ****************************************************************************** * @brief Macro to put a code with specified number of bits into the * bitstream using 0th order exponential Golomb encoding for * signed numbers ****************************************************************************** */ #define PUT_BITS_SEV(ps_bitstrm, code_val, ret_val, syntax_string) \ { \ ENTROPY_TRACE(syntax_string, code_val); \ ret_val = ih264e_put_sev((ps_bitstrm), (code_val)); \ if(ret_val != IH264E_SUCCESS) \ { \ return ret_val; \ } \ } /** ****************************************************************************** * @brief Macro to set active entropy threads to zero and return * in case of errors ****************************************************************************** */ #define RETURN_ENTROPY_IF_ERROR(ps_codec, ps_entropy, ctxt_sel) \ if(ps_entropy->i4_error_code != IH264E_SUCCESS) \ { \ DATA_SYNC(); \ ps_codec->au4_entropy_thread_active[ctxt_sel] = 0; \ return ps_entropy->i4_error_code; \ } /*****************************************************************************/ /* Extern Function Declarations */ /*****************************************************************************/ /** ****************************************************************************** * * @brief Generates SPS (Sequence Parameter Set) * * @par Description * This function generates Sequence Parameter Set header as per the spec * * @param[in] ps_bitstrm * pointer to bitstream context (handle) * * @param[in] ps_sps * pointer to structure containing SPS data * * @return success or failure error code * ****************************************************************************** */ WORD32 ih264e_generate_sps ( bitstrm_t *ps_bitstrm, sps_t *ps_sps, vui_t *ps_vui ); /** ****************************************************************************** * * @brief Generates PPS (Picture Parameter Set) * * @par Description * Generate Picture Parameter Set as per Section 7.3.2.2 * * @param[in] ps_bitstrm * pointer to bitstream context (handle) * * @param[in] ps_pps * pointer to structure containing PPS data * * @return success or failure error code * ****************************************************************************** */ WORD32 ih264e_generate_pps ( bitstrm_t *ps_bitstrm, pps_t *ps_pps, sps_t *ps_sps ); /** ****************************************************************************** * * @brief Generates SEI (Supplemental Enhancement Information) * * @par Description * This function generates Supplemental Enhancement Information header as per the spec * * @param[in] ps_bitstrm * pointer to bitstream context (handle) * * @param[in] ps_sei * pointer to structure containing SEI data * * @return success or failure error code * ****************************************************************************** */ IH264E_ERROR_T ih264e_generate_sei ( bitstrm_t *ps_bitstrm, sei_params_t *ps_sei, UWORD32 u4_insert_per_idr ); /** ****************************************************************************** * * @brief Generates Slice Header * * @par Description * Generate Slice Header as per Section 7.3.5.1 * * @param[inout] ps_bitstrm * pointer to bitstream context for generating slice header * * @param[in] ps_slice_hdr * pointer to slice header params * * @param[in] ps_pps * pointer to pps params referred by slice * * @param[in] ps_sps * pointer to sps params referred by slice * * @param[out] ps_dup_bit_strm_ent_offset * Bitstream struct to store bitstream state * * @param[out] pu4_first_slice_start_offset * first slice offset is returned * * @return success or failure error code * ****************************************************************************** */ WORD32 ih264e_generate_slice_header ( bitstrm_t *ps_bitstrm, slice_header_t *ps_slice_hdr, pps_t *ps_pps, sps_t *ps_sps ); /** ****************************************************************************** * * @brief Populates sps structure * * @par Description * Populates sps structure for its use in header generation * * @param[in] ps_codec * pointer to encoder context * * @param[out] ps_sps * pointer to sps params that needs to be populated * * @return success or failure error code * ****************************************************************************** */ IH264E_ERROR_T ih264e_populate_sps ( codec_t *ps_codec, sps_t *ps_sps ); /** ****************************************************************************** * * @brief Populates pps structure * * @par Description * Populates pps structure for its use in header generation * * @param[in] ps_codec * pointer to encoder context * * @param[out] ps_pps * pointer to pps params that needs to be populated * * @return success or failure error code * ****************************************************************************** */ IH264E_ERROR_T ih264e_populate_pps ( codec_t *ps_codec, pps_t *ps_pps ); /** ****************************************************************************** * * @brief Populates slice header structure * * @par Description * Populates slice header structure for its use in header generation * * @param[in] ps_proc * pointer to proc context * * @param[out] ps_slice_hdr * pointer to slice header structure that needs to be populated * * @param[in] ps_pps * pointer to pps params structure referred by the slice * * @param[in] ps_sps * pointer to sps params referred by the pps * * @return success or failure error code * ****************************************************************************** */ WORD32 ih264e_populate_slice_header ( process_ctxt_t *ps_proc, slice_header_t *ps_slice_hdr, pps_t *ps_pps, sps_t *ps_sps ); /** ****************************************************************************** * * @brief inserts FILLER Nal Unit. * * @par Description * In constant bit rate rc mode, when the bits generated by the codec is * underflowing the target bit rate, the encoder library inserts filler nal unit. * * @param[in] ps_bitstrm * pointer to bitstream context (handle) * * @param[in] insert_fill_bytes * Number of fill bytes to be inserted * * @return success or failure error code * ****************************************************************************** */ IH264E_ERROR_T ih264e_add_filler_nal_unit ( bitstrm_t *ps_bitstrm, WORD32 insert_fill_bytes ); #endif //IH264E_ENCODE_HEADER_H_