mirror of
https://github.com/ultravideo/uvg266.git
synced 2024-12-12 08:54:06 +00:00
HM10.0 compatible PCM-I coding
This commit is contained in:
parent
a7855f8103
commit
26a2c71e3a
|
@ -6,7 +6,7 @@
|
||||||
/*! \file bitstream.c
|
/*! \file bitstream.c
|
||||||
\brief Bitstream related functions
|
\brief Bitstream related functions
|
||||||
\author Marko Viitanen
|
\author Marko Viitanen
|
||||||
\date 2012-05
|
\date 2013-02
|
||||||
|
|
||||||
This file has all bitstream functions
|
This file has all bitstream functions
|
||||||
*/
|
*/
|
||||||
|
@ -108,7 +108,7 @@ void bitstream_clear_buffer(bitstream* stream)
|
||||||
* bits = number of bits to write
|
* bits = number of bits to write
|
||||||
*/
|
*/
|
||||||
|
|
||||||
INLINE void bitstream_put(bitstream* stream, uint32_t data, uint8_t bits)
|
void bitstream_put(bitstream* stream, uint32_t data, uint8_t bits)
|
||||||
{
|
{
|
||||||
uint32_t bitsleft=32-stream->cur_bit;
|
uint32_t bitsleft=32-stream->cur_bit;
|
||||||
#ifdef VERBOSE
|
#ifdef VERBOSE
|
||||||
|
@ -161,7 +161,7 @@ INLINE void bitstream_put(bitstream* stream, uint32_t data, uint8_t bits)
|
||||||
/*
|
/*
|
||||||
* \brief Align the bitstream
|
* \brief Align the bitstream
|
||||||
*/
|
*/
|
||||||
INLINE void bitstream_align(bitstream* stream)
|
void bitstream_align(bitstream* stream)
|
||||||
{
|
{
|
||||||
if((stream->cur_bit&7) != 0)
|
if((stream->cur_bit&7) != 0)
|
||||||
{
|
{
|
||||||
|
@ -169,7 +169,7 @@ INLINE void bitstream_align(bitstream* stream)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
INLINE void bitstream_flush(bitstream* stream)
|
void bitstream_flush(bitstream* stream)
|
||||||
{
|
{
|
||||||
/*
|
/*
|
||||||
* SAVE DATA TO OUTPUT
|
* SAVE DATA TO OUTPUT
|
||||||
|
|
|
@ -6,7 +6,7 @@
|
||||||
/*! \file encoder.c
|
/*! \file encoder.c
|
||||||
\brief Encoding related functions
|
\brief Encoding related functions
|
||||||
\author Marko Viitanen
|
\author Marko Viitanen
|
||||||
\date 2012-06
|
\date 2013-02
|
||||||
|
|
||||||
Encoder main level
|
Encoder main level
|
||||||
*/
|
*/
|
||||||
|
@ -244,14 +244,14 @@ void encode_one_frame(encoder_control* encoder)
|
||||||
encode_vid_parameter_set(encoder);
|
encode_vid_parameter_set(encoder);
|
||||||
bitstream_align(encoder->stream);
|
bitstream_align(encoder->stream);
|
||||||
bitstream_flush(encoder->stream);
|
bitstream_flush(encoder->stream);
|
||||||
nal_write(encoder->output, encoder->stream->buffer, encoder->stream->buffer_pos, 1, NAL_VID_PARAMETER_SET, 0);
|
nal_write(encoder->output, encoder->stream->buffer, encoder->stream->buffer_pos, 1, NAL_VID_PARAMETER_SET, 1);
|
||||||
bitstream_clear_buffer(encoder->stream);
|
bitstream_clear_buffer(encoder->stream);
|
||||||
|
|
||||||
/* Picture Parameter Set (PPS) */
|
/* Picture Parameter Set (PPS) */
|
||||||
encode_pic_parameter_set(encoder);
|
encode_pic_parameter_set(encoder);
|
||||||
bitstream_align(encoder->stream);
|
bitstream_align(encoder->stream);
|
||||||
bitstream_flush(encoder->stream);
|
bitstream_flush(encoder->stream);
|
||||||
nal_write(encoder->output, encoder->stream->buffer, encoder->stream->buffer_pos, 1, NAL_PIC_PARAMETER_SET, 0);
|
nal_write(encoder->output, encoder->stream->buffer, encoder->stream->buffer_pos, 1, NAL_PIC_PARAMETER_SET, 1);
|
||||||
bitstream_clear_buffer(encoder->stream);
|
bitstream_clear_buffer(encoder->stream);
|
||||||
|
|
||||||
/* First slice is IDR */
|
/* First slice is IDR */
|
||||||
|
@ -269,6 +269,7 @@ void encode_one_frame(encoder_control* encoder)
|
||||||
else if(encoder->frame < 3)
|
else if(encoder->frame < 3)
|
||||||
{
|
{
|
||||||
/* Non-IDR slice */
|
/* Non-IDR slice */
|
||||||
|
/*
|
||||||
cabac_start(&cabac);
|
cabac_start(&cabac);
|
||||||
encoder->in.cur_pic.type = NAL_NONIDR_SLICE;
|
encoder->in.cur_pic.type = NAL_NONIDR_SLICE;
|
||||||
encode_slice_header(encoder);
|
encode_slice_header(encoder);
|
||||||
|
@ -279,6 +280,7 @@ void encode_one_frame(encoder_control* encoder)
|
||||||
bitstream_flush(encoder->stream);
|
bitstream_flush(encoder->stream);
|
||||||
nal_write(encoder->output, encoder->stream->buffer, encoder->stream->buffer_pos, 0, NAL_NONIDR_SLICE, encoder->frame+1);
|
nal_write(encoder->output, encoder->stream->buffer, encoder->stream->buffer_pos, 0, NAL_NONIDR_SLICE, encoder->frame+1);
|
||||||
bitstream_clear_buffer(encoder->stream);
|
bitstream_clear_buffer(encoder->stream);
|
||||||
|
*/
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -313,7 +315,8 @@ void encode_pic_parameter_set(encoder_control* encoder)
|
||||||
|
|
||||||
//WRITE_U(encoder->stream, 0, 1, "dependent_slices_enabled_flag");
|
//WRITE_U(encoder->stream, 0, 1, "dependent_slices_enabled_flag");
|
||||||
WRITE_U(encoder->stream, 0, 1, "transquant_bypass_enable_flag");
|
WRITE_U(encoder->stream, 0, 1, "transquant_bypass_enable_flag");
|
||||||
WRITE_U(encoder->stream, 0, 2, "tiles_enabled_flag");
|
WRITE_U(encoder->stream, 0, 1, "tiles_enabled_flag");
|
||||||
|
WRITE_U(encoder->stream, 0, 1, "entropy_coding_sync_enabled_flag");
|
||||||
//ToDo: enable tiles for concurrency
|
//ToDo: enable tiles for concurrency
|
||||||
//IF tiles
|
//IF tiles
|
||||||
//ENDIF
|
//ENDIF
|
||||||
|
@ -324,14 +327,37 @@ void encode_pic_parameter_set(encoder_control* encoder)
|
||||||
WRITE_U(encoder->stream, 0, 1, "pps_scaling_list_data_present_flag");
|
WRITE_U(encoder->stream, 0, 1, "pps_scaling_list_data_present_flag");
|
||||||
//IF scaling_list
|
//IF scaling_list
|
||||||
//ENDIF
|
//ENDIF
|
||||||
|
WRITE_U(encoder->stream, 0, 1, "lists_modification_present_flag");
|
||||||
WRITE_UE(encoder->stream, 0, "log2_parallel_merge_level_minus2");
|
WRITE_UE(encoder->stream, 0, "log2_parallel_merge_level_minus2");
|
||||||
WRITE_U(encoder->stream, 0, 1, "slice_segment_header_extension_present_flag");
|
WRITE_U(encoder->stream, 0, 1, "slice_segment_header_extension_present_flag");
|
||||||
WRITE_U(encoder->stream, 0, 1, "pps_extension_flag");
|
WRITE_U(encoder->stream, 0, 1, "pps_extension_flag");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void encode_PTL(encoder_control *encoder)
|
||||||
|
{
|
||||||
|
/*PTL*/
|
||||||
|
/*Profile Tier*/
|
||||||
|
WRITE_U(encoder->stream, 0, 2, "XXX_profile_space[]");
|
||||||
|
WRITE_U(encoder->stream, 0, 1, "XXX_tier_flag[]");
|
||||||
|
WRITE_U(encoder->stream, 0, 5, "XXX_profile_idc[]");
|
||||||
|
WRITE_U(encoder->stream, 0, 32, "XXX_profile_compatibility_flag[][j]");
|
||||||
|
|
||||||
|
WRITE_U(encoder->stream, 0, 1, "general_progressive_source_flag");
|
||||||
|
WRITE_U(encoder->stream, 0, 1, "general_interlaced_source_flag");
|
||||||
|
WRITE_U(encoder->stream, 0, 1, "general_non_packed_constraint_flag");
|
||||||
|
WRITE_U(encoder->stream, 0, 1, "general_frame_only_constraint_flag");
|
||||||
|
|
||||||
|
WRITE_U(encoder->stream, 0, 32, "XXX_reserved_zero_44bits[0..31]");
|
||||||
|
WRITE_U(encoder->stream, 0, 12, "XXX_reserved_zero_44bits[32..43]");
|
||||||
|
|
||||||
|
/*end Profile Tier */
|
||||||
|
|
||||||
|
WRITE_U(encoder->stream, 0, 8, "general_level_idc");
|
||||||
|
/*end PTL*/
|
||||||
|
}
|
||||||
|
|
||||||
void encode_seq_parameter_set(encoder_control* encoder)
|
void encode_seq_parameter_set(encoder_control* encoder)
|
||||||
{
|
{
|
||||||
int i;
|
|
||||||
#ifdef _DEBUG
|
#ifdef _DEBUG
|
||||||
printf("=========== Sequence Parameter Set ID: 0 ===========\n");
|
printf("=========== Sequence Parameter Set ID: 0 ===========\n");
|
||||||
#endif
|
#endif
|
||||||
|
@ -339,7 +365,9 @@ void encode_seq_parameter_set(encoder_control* encoder)
|
||||||
WRITE_U(encoder->stream, 0, 4, "sps_video_parameter_set_id");
|
WRITE_U(encoder->stream, 0, 4, "sps_video_parameter_set_id");
|
||||||
WRITE_U(encoder->stream, 0, 3, "sps_max_sub_layers_minus1");
|
WRITE_U(encoder->stream, 0, 3, "sps_max_sub_layers_minus1");
|
||||||
|
|
||||||
WRITE_U(encoder->stream, 0, 1, "sps_temporal_id_nesting_flag");
|
WRITE_U(encoder->stream, 1, 1, "sps_temporal_id_nesting_flag");
|
||||||
|
|
||||||
|
encode_PTL(encoder);
|
||||||
|
|
||||||
WRITE_UE(encoder->stream, 0, "sps_seq_parameter_set_id");
|
WRITE_UE(encoder->stream, 0, "sps_seq_parameter_set_id");
|
||||||
WRITE_UE(encoder->stream, encoder->in.video_format, "chroma_format_idc"); /* 0 = 4:0:0, 1 = 4:2:0, 2 = 4:2:2, 3 = 4:4:4 */
|
WRITE_UE(encoder->stream, encoder->in.video_format, "chroma_format_idc"); /* 0 = 4:0:0, 1 = 4:2:0, 2 = 4:2:2, 3 = 4:4:4 */
|
||||||
|
@ -398,6 +426,8 @@ void encode_seq_parameter_set(encoder_control* encoder)
|
||||||
WRITE_U(encoder->stream, 0, 1, "sps_strong_intra_smoothing_enable_flag");
|
WRITE_U(encoder->stream, 0, 1, "sps_strong_intra_smoothing_enable_flag");
|
||||||
|
|
||||||
WRITE_U(encoder->stream, 0, 1, "vui_parameters_present_flag");
|
WRITE_U(encoder->stream, 0, 1, "vui_parameters_present_flag");
|
||||||
|
//ToDo: VUI?
|
||||||
|
//encode_VUI(encoder);
|
||||||
|
|
||||||
WRITE_U(encoder->stream, 0, 1, "sps_extension_flag");
|
WRITE_U(encoder->stream, 0, 1, "sps_extension_flag");
|
||||||
}
|
}
|
||||||
|
@ -411,10 +441,12 @@ void encode_vid_parameter_set(encoder_control* encoder)
|
||||||
WRITE_U(encoder->stream, 0, 4, "vps_video_parameter_set_id");
|
WRITE_U(encoder->stream, 0, 4, "vps_video_parameter_set_id");
|
||||||
WRITE_U(encoder->stream, 3, 2, "vps_reserved_three_2bits" );
|
WRITE_U(encoder->stream, 3, 2, "vps_reserved_three_2bits" );
|
||||||
WRITE_U(encoder->stream, 0, 6, "vps_reserved_zero_6bits" );
|
WRITE_U(encoder->stream, 0, 6, "vps_reserved_zero_6bits" );
|
||||||
|
|
||||||
WRITE_U(encoder->stream, 0, 3, "vps_max_sub_layers_minus1");
|
WRITE_U(encoder->stream, 0, 3, "vps_max_sub_layers_minus1");
|
||||||
WRITE_U(encoder->stream, 0, 1, "vps_temporal_id_nesting_flag");
|
WRITE_U(encoder->stream, 1, 1, "vps_temporal_id_nesting_flag");
|
||||||
WRITE_U(encoder->stream, 0xffff, 16, "vps_reserved_ffff_16bits");
|
WRITE_U(encoder->stream, 0xffff, 16, "vps_reserved_ffff_16bits");
|
||||||
|
|
||||||
|
encode_PTL(encoder);
|
||||||
|
|
||||||
WRITE_U(encoder->stream, 0, 1, "vps_sub_layer_ordering_info_present_flag");
|
WRITE_U(encoder->stream, 0, 1, "vps_sub_layer_ordering_info_present_flag");
|
||||||
//for each layer
|
//for each layer
|
||||||
WRITE_UE(encoder->stream, 0, "vps_max_dec_pic_buffering");
|
WRITE_UE(encoder->stream, 0, "vps_max_dec_pic_buffering");
|
||||||
|
@ -473,21 +505,24 @@ void encode_slice_header(encoder_control* encoder)
|
||||||
printf("=========== Slice ===========\n");
|
printf("=========== Slice ===========\n");
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
WRITE_U(encoder->stream, 1, 1, "first_slice_in_pic_flag");
|
WRITE_U(encoder->stream, 1, 1, "first_slice_segment_in_pic_flag");
|
||||||
if(encoder->in.cur_pic.type == NAL_IDR_SLICE)
|
if(encoder->in.cur_pic.type == NAL_IDR_SLICE)
|
||||||
{
|
{
|
||||||
WRITE_U(encoder->stream, 0, 1, "no_output_of_prior_pics_flag");
|
WRITE_U(encoder->stream, 0, 1, "no_output_of_prior_pics_flag");
|
||||||
}
|
}
|
||||||
WRITE_UE(encoder->stream, 0, "pic_parameter_set_id");
|
WRITE_UE(encoder->stream, 0, "slice_pic_parameter_set_id");
|
||||||
|
|
||||||
|
//WRITE_U(encoder->stream, 0, 1, "dependent_slice_segment_flag");
|
||||||
|
|
||||||
|
/* ToDo: add more slice types */
|
||||||
WRITE_UE(encoder->stream, SLICE_I, "slice_type");
|
WRITE_UE(encoder->stream, SLICE_I, "slice_type");
|
||||||
|
|
||||||
WRITE_U(encoder->stream, 0, 1, "dependent_slice_flag");
|
|
||||||
|
|
||||||
// if !entropy_slice_flag
|
// if !entropy_slice_flag
|
||||||
|
|
||||||
//if output_flag_present_flag
|
//if output_flag_present_flag
|
||||||
WRITE_U(encoder->stream, 1, 1, "pic_output_flag");
|
//WRITE_U(encoder->stream, 1, 1, "pic_output_flag");
|
||||||
//end if
|
//end if
|
||||||
//if( IdrPicFlag ) <- nal_unit_type == 5
|
//if( IdrPicFlag ) <- nal_unit_type == 5
|
||||||
if(encoder->in.cur_pic.type == NAL_IDR_SLICE)
|
if(encoder->in.cur_pic.type == NAL_IDR_SLICE)
|
||||||
|
@ -496,19 +531,28 @@ void encode_slice_header(encoder_control* encoder)
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
WRITE_U(encoder->stream, encoder->frame+1, 8, "pic_order_cnt_lsb");
|
WRITE_U(encoder->stream, encoder->frame+1, 4, "pic_order_cnt_lsb");
|
||||||
WRITE_U(encoder->stream, 1, 1, "short_term_ref_pic_set_sps_flag");
|
WRITE_U(encoder->stream, 1, 1, "short_term_ref_pic_set_sps_flag");
|
||||||
//WRITE_U(encoder->stream, 1, 1, "inter_ref_pic_set_prediction_flag");
|
//WRITE_U(encoder->stream, 1, 1, "inter_ref_pic_set_prediction_flag");
|
||||||
WRITE_UE(encoder->stream, 0, "short_term_ref_pic_set_idx");
|
WRITE_UE(encoder->stream, 0, "short_term_ref_pic_set_idx");
|
||||||
}
|
}
|
||||||
//end if
|
//end if
|
||||||
//end if
|
//end if
|
||||||
|
//IF sao
|
||||||
|
/*
|
||||||
|
WRITE_U(encoder->stream, 0,1, "slice_sao_luma_flag" );
|
||||||
|
WRITE_U(encoder->stream, 0,1, "slice_sao_chroma_flag" );
|
||||||
|
*/
|
||||||
|
//ENDIF
|
||||||
/* Skip flags that are not present */
|
/* Skip flags that are not present */
|
||||||
// if !entropy_slice_flag
|
// if !entropy_slice_flag
|
||||||
WRITE_SE(encoder->stream, 0, "slice_qp_delta");
|
WRITE_SE(encoder->stream, 0, "slice_qp_delta");
|
||||||
|
/*
|
||||||
WRITE_UE(encoder->stream, 0, "5_minus_max_num_merge_cand");
|
WRITE_UE(encoder->stream, 0, "5_minus_max_num_merge_cand");
|
||||||
|
|
||||||
WRITE_U(encoder->stream, 1, 1, "alignment");
|
WRITE_U(encoder->stream, 1, 1, "alignment");
|
||||||
|
*/
|
||||||
|
WRITE_U(encoder->stream, 1, 1, "alignment");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@ -596,11 +640,12 @@ void encode_coding_tree(encoder_control* encoder,uint16_t xCtb,uint16_t yCtb, ui
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
/*
|
||||||
if(yCtb > 20 && xCtb > 20)
|
if(yCtb > 20 && xCtb > 20)
|
||||||
{
|
{
|
||||||
cur_CU->type = CU_INTRA;
|
cur_CU->type = CU_INTRA;
|
||||||
}
|
}
|
||||||
|
*/
|
||||||
|
|
||||||
/* coding_unit( x0, y0, log2CbSize ) */
|
/* coding_unit( x0, y0, log2CbSize ) */
|
||||||
/* prediction_unit 2Nx2N*/
|
/* prediction_unit 2Nx2N*/
|
||||||
|
|
|
@ -45,7 +45,7 @@ void nal_write(FILE* output, uint8_t* buffer, uint32_t buffer_len, uint8_t nal_r
|
||||||
byte = nal_type<<1;
|
byte = nal_type<<1;
|
||||||
fwrite(&byte, 1, 1, output);
|
fwrite(&byte, 1, 1, output);
|
||||||
|
|
||||||
/* 5bits of reserved_one_6bits + Temporal_id(3) */
|
/* 5bits of reserved_one_6bits + nuh_temporal_id_plus1(3) */
|
||||||
byte = temporal_id&7;
|
byte = temporal_id&7;
|
||||||
fwrite(&byte, 1, 1, output);
|
fwrite(&byte, 1, 1, output);
|
||||||
|
|
||||||
|
|
|
@ -6,14 +6,14 @@
|
||||||
/*! \file nal.h
|
/*! \file nal.h
|
||||||
\brief NAL
|
\brief NAL
|
||||||
\author Marko Viitanen
|
\author Marko Viitanen
|
||||||
\date 2012-06
|
\date 2013-02
|
||||||
|
|
||||||
NAL function headers
|
NAL function headers
|
||||||
*/
|
*/
|
||||||
#ifndef __NAL_H
|
#ifndef __NAL_H
|
||||||
#define __NAL_H
|
#define __NAL_H
|
||||||
|
|
||||||
enum { NAL_NONIDR_SLICE = 0x1,NAL_IDR_SLICE = 0x8, NAL_VID_PARAMETER_SET = 25,NAL_SEQ_PARAMETER_SET = 26, NAL_PIC_PARAMETER_SET = 27 };
|
enum { NAL_NONIDR_SLICE = 0x1,NAL_IDR_SLICE = 19/*0x8*/, NAL_VID_PARAMETER_SET = 32,NAL_SEQ_PARAMETER_SET = 33, NAL_PIC_PARAMETER_SET = 34 };
|
||||||
|
|
||||||
void nal_write(FILE* output, uint8_t* buffer, uint32_t buffer_len, uint8_t nal_ref, uint8_t nal_type, uint8_t temporal_id);
|
void nal_write(FILE* output, uint8_t* buffer, uint32_t buffer_len, uint8_t nal_ref, uint8_t nal_type, uint8_t temporal_id);
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue