/*****************************************************************************

 * Copyright © 2010-2014 VideoLAN

 *

 * Authors: Jean-Baptiste Kempf <jb@videolan.org>

 *

 * This program is free software; you can redistribute it and/or modify it

 * under the terms of the GNU Lesser General Public License as published by

 * the Free Software Foundation; either version 2.1 of the License, or

 * (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

 * GNU Lesser General Public License for more details.

 *

 * You should have received a copy of the GNU Lesser General Public License

 * along with this program; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.

 *****************************************************************************/

#ifndef H264_NAL_H

# define H264_NAL_H

# include <vlc_common.h>

# include <vlc_es.h>

# include <vlc_bits.h>

#ifdef __cplusplus

extern "C" {

#endif

#define PROFILE_H264_BASELINE             66

#define PROFILE_H264_MAIN                 77

#define PROFILE_H264_EXTENDED             88

#define PROFILE_H264_HIGH                 100

#define PROFILE_H264_HIGH_10              110

#define PROFILE_H264_HIGH_422             122

#define PROFILE_H264_HIGH_444             144

#define PROFILE_H264_HIGH_444_PREDICTIVE  244

#define PROFILE_H264_CAVLC_INTRA          44

#define PROFILE_H264_SVC_BASELINE         83

#define PROFILE_H264_SVC_HIGH             86

#define PROFILE_H264_MVC_STEREO_HIGH      128

#define PROFILE_H264_MVC_MULTIVIEW_HIGH   118

#define PROFILE_H264_MFC_HIGH                          134

#define PROFILE_H264_MVC_MULTIVIEW_DEPTH_HIGH          138

#define PROFILE_H264_MVC_ENHANCED_MULTIVIEW_DEPTH_HIGH 139

#define H264_SPS_ID_MAX (31)

#define H264_PPS_ID_MAX (255)

#define H264_SPSEXT_ID_MAX H264_SPS_ID_MAX

#define H264_MAX_NUM_SPS (H264_SPS_ID_MAX + 1)

#define H264_MAX_NUM_PPS (H264_PPS_ID_MAX + 1)

#define H264_MAX_NUM_SPSEXT (H264_SPSEXT_ID_MAX + 1)

enum h264_nal_unit_type_e

{

    H264_NAL_UNKNOWN = 0,

    H264_NAL_SLICE   = 1,

    H264_NAL_SLICE_DPA   = 2,

    H264_NAL_SLICE_DPB   = 3,

    H264_NAL_SLICE_DPC   = 4,

    H264_NAL_SLICE_IDR   = 5,    /* ref_idc != 0 */

    H264_NAL_SEI         = 6,    /* ref_idc == 0 */

    H264_NAL_SPS         = 7,

    H264_NAL_PPS         = 8,

    H264_NAL_AU_DELIMITER= 9,

    /* ref_idc == 0 for 6,9,10,11,12 */

    H264_NAL_END_OF_SEQ  = 10,

    H264_NAL_END_OF_STREAM = 11,

    H264_NAL_FILLER_DATA = 12,

    H264_NAL_SPS_EXT     = 13,

    H264_NAL_PREFIX      = 14,

    H264_NAL_SUBSET_SPS  = 15,

    H264_NAL_DEPTH_PS    = 16,

    H264_NAL_RESERVED_17 = 17,

    H264_NAL_RESERVED_18 = 18,

    H264_NAL_SLICE_WP    = 19,

    H264_NAL_SLICE_EXT   = 20,

    H264_NAL_SLICE_3D_EXT= 21,

    H264_NAL_RESERVED_22 = 22,

    H264_NAL_RESERVED_23 = 23,

};

static inline enum h264_nal_unit_type_e h264_getNALType(const uint8_t *p)

{

    return (enum h264_nal_unit_type_e) (*p & 0x1f);

}

avi.c:h264_getNALType

Line

Count

Source

88

25

{

89

25

    return (enum h264_nal_unit_type_e) (*p & 0x1f);

90

25

}

Unexecuted instantiation: h26x.c:h264_getNALType

h264.c:h264_getNALType

Line

Count

Source

88

3.11M

{

89

3.11M

    return (enum h264_nal_unit_type_e) (*p & 0x1f);

90

3.11M

}

h264_nal.c:h264_getNALType

Line

Count

Source

88

695k

{

89

695k

    return (enum h264_nal_unit_type_e) (*p & 0x1f);

90

695k

}

Unexecuted instantiation: h264_slice.c:h264_getNALType

typedef struct h264_sequence_parameter_set_t h264_sequence_parameter_set_t;

typedef struct h264_picture_parameter_set_t h264_picture_parameter_set_t;

typedef struct h264_sequence_parameter_set_extension_t h264_sequence_parameter_set_extension_t;

bool h264_get_xps_id(const uint8_t *p_nalbuf, size_t i_nalbuf, uint8_t *pi_id);

h264_sequence_parameter_set_t * h264_decode_sps( const uint8_t *, size_t, bool );

h264_picture_parameter_set_t *  h264_decode_pps( const uint8_t *, size_t, bool );

h264_sequence_parameter_set_extension_t * h264_decode_sps_extension( const uint8_t *, size_t, bool );

void h264_release_sps( h264_sequence_parameter_set_t * );

void h264_release_pps( h264_picture_parameter_set_t * );

void h264_release_sps_extension( h264_sequence_parameter_set_extension_t * );

uint8_t h264_get_sps_id( const h264_sequence_parameter_set_t * );

uint8_t h264_get_pps_sps_id( const h264_picture_parameter_set_t * );

struct h264_sequence_parameter_set_t

{

    uint8_t i_id;

    uint8_t i_profile, i_level;

    uint8_t i_constraint_set_flags;

    /* according to avcC, 3 bits max for those */

    uint8_t i_chroma_idc;

    uint8_t i_bit_depth_luma;

    uint8_t i_bit_depth_chroma;

    uint8_t b_separate_colour_planes_flag;

    uint32_t pic_width_in_mbs_minus1;

    uint32_t pic_height_in_map_units_minus1;

    struct

    {

        uint32_t left_offset;

        uint32_t right_offset;

        uint32_t top_offset;

        uint32_t bottom_offset;

    } frame_crop;

    uint8_t frame_mbs_only_flag;

    uint8_t mb_adaptive_frame_field_flag;

    int i_log2_max_frame_num;

    int i_pic_order_cnt_type;

    int i_delta_pic_order_always_zero_flag;

    int32_t offset_for_non_ref_pic;

    int32_t offset_for_top_to_bottom_field;

    int i_num_ref_frames_in_pic_order_cnt_cycle;

    int32_t offset_for_ref_frame[255];

    int i_log2_max_pic_order_cnt_lsb;

    bool vui_parameters_present_flag;

    struct {

        int i_sar_num, i_sar_den;

        struct {

            bool b_full_range;

            uint8_t i_colour_primaries;

            uint8_t i_transfer_characteristics;

            uint8_t i_matrix_coefficients;

        } colour;

        bool b_timing_info_present_flag;

        uint32_t i_num_units_in_tick;

        uint32_t i_time_scale;

        bool b_fixed_frame_rate;

        bool b_pic_struct_present_flag;

        bool b_hrd_parameters_present_flag; /* CpbDpbDelaysPresentFlag */

        uint8_t i_cpb_removal_delay_length_minus1;

        uint8_t i_dpb_output_delay_length_minus1;

        /* restrictions */

        uint8_t b_bitstream_restriction_flag;

        uint8_t i_max_num_reorder_frames;

        uint8_t i_max_dec_frame_buffering;

    } vui;

};

struct h264_picture_parameter_set_t

{

    uint8_t i_id;

    uint8_t i_sps_id;

    uint8_t i_pic_order_present_flag;

    uint8_t i_redundant_pic_present_flag;

    uint8_t weighted_pred_flag;

    uint8_t weighted_bipred_idc;

    uint32_t num_ref_idx_l01_default_active_minus1[2];

};

struct h264_sequence_parameter_set_extension_t

{

    uint8_t i_sps_id;

};

/*

    AnnexB : [\x00] \x00 \x00 \x01 Prefixed NAL

    AVC Sample format : NalLengthSize encoded size prefixed NAL

    avcC: AVCDecoderConfigurationRecord combining SPS & PPS in AVC Sample Format

*/

#define H264_MIN_AVCC_SIZE 7

bool h264_isavcC( const uint8_t *, size_t );

/* Convert AVC Sample format to Annex B in-place */

void h264_AVC_to_AnnexB( uint8_t *p_buf, uint32_t i_len,

                         uint8_t i_nal_length_size );

/* Get the First SPS/PPS NAL pointers from an Annex B buffer

 * Returns TRUE if a SPS and/or a PPS is found */

bool h264_AnnexB_get_spspps( const uint8_t *p_buf, size_t i_buf,

                             const uint8_t **pp_sps, size_t *p_sps_size,

                             const uint8_t **pp_pps, size_t *p_pps_size,

                             const uint8_t **pp_ext, size_t *p_ext_size );

/* Create a AVCDecoderConfigurationRecord from non prefixed SPS/PPS

 * Returns a valid block_t on success, must be freed with block_Release */

block_t *h264_NAL_to_avcC( uint8_t i_nal_length_size,

                           const uint8_t **pp_sps_buf,

                           const size_t *p_sps_size, uint8_t i_sps_count,

                           const uint8_t **pp_pps_buf,

                           const size_t *p_pps_size, uint8_t i_pps_count,

                           const uint8_t **pp_sps_ext_buf,

                           const size_t *p_sps_ext_size, uint8_t i_sps_ext_count);

/* Convert AVCDecoderConfigurationRecord SPS/PPS to Annex B format */

bool h264_avcC_to_AnnexB_NAL( const uint8_t *p_buf, size_t i_buf,

                              uint8_t **pp_result, size_t *pi_result,

                              uint8_t *pi_nal_length_size );

bool h264_get_dpb_values( const h264_sequence_parameter_set_t *,

                          uint8_t *pi_max_num_reorder, uint8_t *pi_max_dec_buffering );

unsigned h264_get_max_frame_num( const h264_sequence_parameter_set_t * );

bool h264_is_frames_only( const h264_sequence_parameter_set_t * );

bool h264_using_adaptive_frames( const h264_sequence_parameter_set_t * );

bool h264_get_sps_profile_tier_level( const h264_sequence_parameter_set_t *,

                                     uint8_t *pi_profile, uint8_t *pi_level );

bool h264_get_picture_size( const h264_sequence_parameter_set_t *,

                            unsigned *p_ox, unsigned *p_oy,

                            unsigned *p_w, unsigned *p_h,

                            unsigned *p_vw, unsigned *p_vh );

bool h264_get_frame_rate( const h264_sequence_parameter_set_t *,

                          unsigned *pi_num, unsigned *pi_den );

bool h264_get_aspect_ratio( const h264_sequence_parameter_set_t *,

                           unsigned *pi_num, unsigned *pi_den );

bool h264_get_chroma_luma( const h264_sequence_parameter_set_t *, uint8_t *pi_chroma_format,

                           uint8_t *pi_depth_luma, uint8_t *pi_depth_chroma );

bool h264_get_colorimetry( const h264_sequence_parameter_set_t *p_sps,

                           video_color_primaries_t *p_primaries,

                           video_transfer_func_t *p_transfer,

                           video_color_space_t *p_colorspace,

                           video_color_range_t *p_full_range );

/* Get level and Profile from DecoderConfigurationRecord */

bool h264_get_profile_level(const es_format_t *p_fmt, uint8_t *pi_profile,

                            uint8_t *pi_level, uint8_t *p_nal_length_size);

typedef struct

{

    unsigned i_frames;

} h264_sei_recovery_point_t;

bool h264_decode_sei_recovery_point( bs_t *, h264_sei_recovery_point_t * );

bool h264_decode_sei_pic_timing(  bs_t *, const h264_sequence_parameter_set_t *,

                                  uint8_t *, uint8_t *  );

#ifdef __cplusplus

}

#endif

#endif /* H264_NAL_H */