Void SEIWriter::xWriteSEIRecoveryPoint(const SEIRecoveryPoint& sei)
{
WRITE_SVLC( sei.m_recoveryPocCnt, "recovery_poc_cnt" );
WRITE_FLAG( sei.m_exactMatchingFlag, "exact_matching_flag" );
WRITE_FLAG( sei.m_brokenLinkFlag, "broken_link_flag" );
xWriteByteAlign();
}
Void SEIWriter::xWriteSEIFramePacking(const SEIFramePacking& sei)
{
WRITE_UVLC( sei.m_arrangementId, "frame_packing_arrangement_id" );
WRITE_FLAG( sei.m_arrangementCancelFlag, "frame_packing_arrangement_cancel_flag" );
if( sei.m_arrangementCancelFlag == 0 ) {
WRITE_CODE( sei.m_arrangementType, 7, "frame_packing_arrangement_type" );
WRITE_FLAG( sei.m_quincunxSamplingFlag, "quincunx_sampling_flag" );
WRITE_CODE( sei.m_contentInterpretationType, 6, "content_interpretation_type" );
WRITE_FLAG( sei.m_spatialFlippingFlag, "spatial_flipping_flag" );
WRITE_FLAG( sei.m_frame0FlippedFlag, "frame0_flipped_flag" );
WRITE_FLAG( sei.m_fieldViewsFlag, "field_views_flag" );
WRITE_FLAG( sei.m_currentFrameIsFrame0Flag, "current_frame_is_frame0_flag" );
WRITE_FLAG( sei.m_frame0SelfContainedFlag, "frame0_self_contained_flag" );
WRITE_FLAG( sei.m_frame1SelfContainedFlag, "frame1_self_contained_flag" );
if(sei.m_quincunxSamplingFlag == 0 && sei.m_arrangementType != 5)
{
WRITE_CODE( sei.m_frame0GridPositionX, 4, "frame0_grid_position_x" );
WRITE_CODE( sei.m_frame0GridPositionY, 4, "frame0_grid_position_y" );
WRITE_CODE( sei.m_frame1GridPositionX, 4, "frame1_grid_position_x" );
WRITE_CODE( sei.m_frame1GridPositionY, 4, "frame1_grid_position_y" );
}
WRITE_CODE( sei.m_arrangementReservedByte, 8, "frame_packing_arrangement_reserved_byte" );
WRITE_FLAG( sei.m_arrangementPersistenceFlag, "frame_packing_arrangement_persistence_flag" );
}
WRITE_FLAG( sei.m_upsampledAspectRatio, "upsampled_aspect_ratio" );
xWriteByteAlign();
}
Void SEIWriter::xWriteSEIActiveParameterSets(const SEIActiveParameterSets& sei)
{
WRITE_CODE(sei.activeVPSId, 4, "active_vps_id");
#if L0047_APS_FLAGS
WRITE_FLAG(sei.m_fullRandomAccessFlag, "full_random_access_flag");
WRITE_FLAG(sei.m_noParamSetUpdateFlag, "no_param_set_update_flag");
#endif
WRITE_UVLC(sei.numSpsIdsMinus1, "num_sps_ids_minus1");
assert (sei.activeSeqParamSetId.size() == (sei.numSpsIdsMinus1 + 1));
for (Int i = 0; i < sei.activeSeqParamSetId.size(); i++)
{
WRITE_UVLC(sei.activeSeqParamSetId[i], "active_seq_param_set_id");
}
UInt uiBits = m_pcBitIf->getNumberOfWrittenBits();
UInt uiAlignedBits = ( 8 - (uiBits&7) ) % 8;
if(uiAlignedBits)
{
WRITE_FLAG(1, "alignment_bit" );
uiAlignedBits--;
while(uiAlignedBits--)
{
WRITE_FLAG(0, "alignment_bit" );
}
}
}
Void SEIWriter::xWriteSEISegmentedRectFramePacking(const SEISegmentedRectFramePacking& sei)
{
WRITE_FLAG( sei.m_arrangementCancelFlag, "segmented_rect_frame_packing_arrangement_cancel_flag" );
if( sei.m_arrangementCancelFlag == 0 )
{
WRITE_CODE( sei.m_contentInterpretationType, 2, "segmented_rect_content_interpretation_type" );
WRITE_FLAG( sei.m_arrangementPersistenceFlag, "segmented_rect_frame_packing_arrangement_persistence" );
}
}
Void SEIWriter::xWriteSEIPictureTiming(const SEIPictureTiming& sei, TComSPS *sps)
{
Int i;
TComVUI *vui = sps->getVuiParameters();
TComHRD *hrd = vui->getHrdParameters();
#if !L0045_CONDITION_SIGNALLING
// This condition was probably OK before the pic_struct, progressive_source_idc, duplicate_flag were added
if( !hrd->getNalHrdParametersPresentFlag() && !hrd->getVclHrdParametersPresentFlag() )
return;
#endif
if( vui->getFrameFieldInfoPresentFlag() )
{
WRITE_CODE( sei.m_picStruct, 4, "pic_struct" );
#if L0046_RENAME_PROG_SRC_IDC
WRITE_CODE( sei.m_sourceScanType, 2, "source_scan_type" );
#else
WRITE_CODE( sei.m_progressiveSourceIdc, 2, "progressive_source_idc" );
#endif
WRITE_FLAG( sei.m_duplicateFlag ? 1 : 0, "duplicate_flag" );
}
#if L0045_CONDITION_SIGNALLING
if( hrd->getCpbDpbDelaysPresentFlag() )
{
#endif
WRITE_CODE( sei.m_auCpbRemovalDelay - 1, ( hrd->getCpbRemovalDelayLengthMinus1() + 1 ), "au_cpb_removal_delay_minus1" );
WRITE_CODE( sei.m_picDpbOutputDelay, ( hrd->getDpbOutputDelayLengthMinus1() + 1 ), "pic_dpb_output_delay" );
#if L0044_DU_DPB_OUTPUT_DELAY_HRD
if(hrd->getSubPicCpbParamsPresentFlag())
{
WRITE_CODE(sei.m_picDpbOutputDuDelay, hrd->getDpbOutputDelayDuLengthMinus1()+1, "pic_dpb_output_du_delay" );
}
#endif
if( hrd->getSubPicCpbParamsPresentFlag() && hrd->getSubPicCpbParamsInPicTimingSEIFlag() )
{
WRITE_UVLC( sei.m_numDecodingUnitsMinus1, "num_decoding_units_minus1" );
WRITE_FLAG( sei.m_duCommonCpbRemovalDelayFlag, "du_common_cpb_removal_delay_flag" );
if( sei.m_duCommonCpbRemovalDelayFlag )
{
WRITE_CODE( sei.m_duCommonCpbRemovalDelayMinus1, ( hrd->getDuCpbRemovalDelayLengthMinus1() + 1 ), "du_common_cpb_removal_delay_minus1" );
}
for( i = 0; i <= sei.m_numDecodingUnitsMinus1; i ++ )
{
WRITE_UVLC( sei.m_numNalusInDuMinus1[ i ], "num_nalus_in_du_minus1");
if( ( !sei.m_duCommonCpbRemovalDelayFlag ) && ( i < sei.m_numDecodingUnitsMinus1 ) )
{
WRITE_CODE( sei.m_duCpbRemovalDelayMinus1[ i ], ( hrd->getDuCpbRemovalDelayLengthMinus1() + 1 ), "du_cpb_removal_delay_minus1" );
}
}
}
#if L0045_CONDITION_SIGNALLING
}
#endif
xWriteByteAlign();
}
Void SEIWriter::xWriteByteAlign()
{
if( m_pcBitIf->getNumberOfWrittenBits() % 8 != 0)
{
WRITE_FLAG( 1, "bit_equal_to_one" );
while( m_pcBitIf->getNumberOfWrittenBits() % 8 != 0 )
{
WRITE_FLAG( 0, "bit_equal_to_zero" );
}
}
};
Void SEIWriter::xWriteSEIActiveParameterSets(const SEIActiveParameterSets& sei)
{
WRITE_CODE(sei.activeVPSId, 4, "active_video_parameter_set_id");
WRITE_FLAG(sei.m_selfContainedCvsFlag, "self_contained_cvs_flag");
WRITE_FLAG(sei.m_noParameterSetUpdateFlag, "no_parameter_set_update_flag");
WRITE_UVLC(sei.numSpsIdsMinus1, "num_sps_ids_minus1");
assert (sei.activeSeqParameterSetId.size() == (sei.numSpsIdsMinus1 + 1));
for (Int i = 0; i < sei.activeSeqParameterSetId.size(); i++)
{
WRITE_UVLC(sei.activeSeqParameterSetId[i], "active_seq_parameter_set_id");
}
}
Void SEIWriter::xWriteSEIPictureTiming(const SEIPictureTiming& sei)
{
Int i;
TComVUI *vui = sei.m_sps->getVuiParameters();
if( !vui->getNalHrdParametersPresentFlag() && !vui->getVclHrdParametersPresentFlag() )
return;
WRITE_CODE( sei.m_auCpbRemovalDelay, ( vui->getCpbRemovalDelayLengthMinus1() + 1 ), "au_cpb_removal_delay" );
WRITE_CODE( sei.m_picDpbOutputDelay, ( vui->getDpbOutputDelayLengthMinus1() + 1 ), "pic_dpb_output_delay" );
if( sei.m_sps->getVuiParameters()->getSubPicCpbParamsPresentFlag() )
{
WRITE_UVLC( sei.m_numDecodingUnitsMinus1, "num_decoding_units_minus1" );
WRITE_FLAG( sei.m_duCommonCpbRemovalDelayFlag, "du_common_cpb_removal_delay_flag" );
if( sei.m_duCommonCpbRemovalDelayFlag )
{
WRITE_CODE( sei.m_duCommonCpbRemovalDelayMinus1, ( vui->getDuCpbRemovalDelayLengthMinus1() + 1 ), "du_common_cpb_removal_delay_minus1" );
}
else
{
for( i = 0; i < ( sei.m_numDecodingUnitsMinus1 + 1 ); i ++ )
{
WRITE_UVLC( sei.m_numNalusInDuMinus1[ i ], "num_nalus_in_du_minus1");
WRITE_CODE( sei.m_duCpbRemovalDelayMinus1[ i ], ( vui->getDuCpbRemovalDelayLengthMinus1() + 1 ), "du_cpb_removal_delay_minus1" );
}
}
}
xWriteByteAlign();
}
Void SEIWriter::xWriteSEIBufferingPeriod(const SEIBufferingPeriod& sei)
{
Int i, nalOrVcl;
TComVUI *vui = sei.m_sps->getVuiParameters();
WRITE_UVLC( sei.m_seqParameterSetId, "seq_parameter_set_id" );
if( !vui->getSubPicCpbParamsPresentFlag() )
{
WRITE_FLAG( sei.m_altCpbParamsPresentFlag, "alt_cpb_params_present_flag" );
}
for( nalOrVcl = 0; nalOrVcl < 2; nalOrVcl ++ )
{
if( ( ( nalOrVcl == 0 ) && ( vui->getNalHrdParametersPresentFlag() ) ) ||
( ( nalOrVcl == 1 ) && ( vui->getVclHrdParametersPresentFlag() ) ) )
{
for( i = 0; i < ( vui->getCpbCntMinus1( 0 ) + 1 ); i ++ )
{
WRITE_CODE( sei.m_initialCpbRemovalDelay[i][nalOrVcl],( vui->getInitialCpbRemovalDelayLengthMinus1() + 1 ) , "initial_cpb_removal_delay" );
WRITE_CODE( sei.m_initialCpbRemovalDelayOffset[i][nalOrVcl],( vui->getInitialCpbRemovalDelayLengthMinus1() + 1 ), "initial_cpb_removal_delay_offset" );
if( vui->getSubPicCpbParamsPresentFlag() || sei.m_altCpbParamsPresentFlag )
{
WRITE_CODE( sei.m_initialAltCpbRemovalDelay[i][nalOrVcl], ( vui->getInitialCpbRemovalDelayLengthMinus1() + 1 ) , "initial_alt_cpb_removal_delay" );
WRITE_CODE( sei.m_initialAltCpbRemovalDelayOffset[i][nalOrVcl], ( vui->getInitialCpbRemovalDelayLengthMinus1() + 1 ),"initial_alt_cpb_removal_delay_offset" );
}
}
}
}
xWriteByteAlign();
}
Void SEIWriter::xWriteSEIDisplayOrientation(const SEIDisplayOrientation &sei)
{
WRITE_FLAG( sei.cancelFlag, "display_orientation_cancel_flag" );
if( !sei.cancelFlag )
{
WRITE_FLAG( sei.horFlip, "hor_flip" );
WRITE_FLAG( sei.verFlip, "ver_flip" );
WRITE_CODE( sei.anticlockwiseRotation, 16, "anticlockwise_rotation" );
#if L0045_PERSISTENCE_FLAGS
WRITE_FLAG( sei.persistenceFlag, "display_orientation_persistence_flag" );
#else
WRITE_UVLC( sei.repetitionPeriod, "display_orientation_repetition_period" );
#endif
#if !REMOVE_SINGLE_SEI_EXTENSION_FLAGS
WRITE_FLAG( sei.extensionFlag, "display_orientation_extension_flag" );
assert( !sei.extensionFlag );
#endif
}
xWriteByteAlign();
}
Void SEIWriter::xWriteSEIPictureTiming(const SEIPictureTiming& sei, TComSPS *sps)
{
Int i;
TComVUI *vui = sps->getVuiParameters();
TComHRD *hrd = vui->getHrdParameters();
if( vui->getFrameFieldInfoPresentFlag() )
{
WRITE_CODE( sei.m_picStruct, 4, "pic_struct" );
WRITE_CODE( sei.m_sourceScanType, 2, "source_scan_type" );
WRITE_FLAG( sei.m_duplicateFlag ? 1 : 0, "duplicate_flag" );
}
if( hrd->getCpbDpbDelaysPresentFlag() )
{
WRITE_CODE( sei.m_auCpbRemovalDelay - 1, ( hrd->getCpbRemovalDelayLengthMinus1() + 1 ), "au_cpb_removal_delay_minus1" );
WRITE_CODE( sei.m_picDpbOutputDelay, ( hrd->getDpbOutputDelayLengthMinus1() + 1 ), "pic_dpb_output_delay" );
if(hrd->getSubPicCpbParamsPresentFlag())
{
WRITE_CODE(sei.m_picDpbOutputDuDelay, hrd->getDpbOutputDelayDuLengthMinus1()+1, "pic_dpb_output_du_delay" );
}
if( hrd->getSubPicCpbParamsPresentFlag() && hrd->getSubPicCpbParamsInPicTimingSEIFlag() )
{
WRITE_UVLC( sei.m_numDecodingUnitsMinus1, "num_decoding_units_minus1" );
WRITE_FLAG( sei.m_duCommonCpbRemovalDelayFlag, "du_common_cpb_removal_delay_flag" );
if( sei.m_duCommonCpbRemovalDelayFlag )
{
WRITE_CODE( sei.m_duCommonCpbRemovalDelayMinus1, ( hrd->getDuCpbRemovalDelayLengthMinus1() + 1 ), "du_common_cpb_removal_delay_minus1" );
}
for( i = 0; i <= sei.m_numDecodingUnitsMinus1; i ++ )
{
WRITE_UVLC( sei.m_numNalusInDuMinus1[ i ], "num_nalus_in_du_minus1");
if( ( !sei.m_duCommonCpbRemovalDelayFlag ) && ( i < sei.m_numDecodingUnitsMinus1 ) )
{
WRITE_CODE( sei.m_duCpbRemovalDelayMinus1[ i ], ( hrd->getDuCpbRemovalDelayLengthMinus1() + 1 ), "du_cpb_removal_delay_minus1" );
}
}
}
}
xWriteByteAlign();
}
Void SEIWriter::xWriteSEIBufferingPeriod(const SEIBufferingPeriod& sei, TComSPS *sps)
{
Int i, nalOrVcl;
TComVUI *vui = sps->getVuiParameters();
TComHRD *hrd = vui->getHrdParameters();
WRITE_UVLC( sei.m_bpSeqParameterSetId, "bp_seq_parameter_set_id" );
if( !hrd->getSubPicCpbParamsPresentFlag() )
{
WRITE_FLAG( sei.m_rapCpbParamsPresentFlag, "rap_cpb_params_present_flag" );
}
#if L0328_SPLICING
WRITE_FLAG( sei.m_concatenationFlag, "concatenation_flag");
WRITE_CODE( sei.m_auCpbRemovalDelayDelta - 1, ( hrd->getCpbRemovalDelayLengthMinus1() + 1 ), "au_cpb_removal_delay_delta_minus1" );
#endif
#if L0044_CPB_DPB_DELAY_OFFSET
if( sei.m_rapCpbParamsPresentFlag )
{
WRITE_CODE( sei.m_cpbDelayOffset, hrd->getCpbRemovalDelayLengthMinus1() + 1, "cpb_delay_offset" );
WRITE_CODE( sei.m_dpbDelayOffset, hrd->getDpbOutputDelayLengthMinus1() + 1, "dpb_delay_offset" );
}
#endif
for( nalOrVcl = 0; nalOrVcl < 2; nalOrVcl ++ )
{
if( ( ( nalOrVcl == 0 ) && ( hrd->getNalHrdParametersPresentFlag() ) ) ||
( ( nalOrVcl == 1 ) && ( hrd->getVclHrdParametersPresentFlag() ) ) )
{
for( i = 0; i < ( hrd->getCpbCntMinus1( 0 ) + 1 ); i ++ )
{
WRITE_CODE( sei.m_initialCpbRemovalDelay[i][nalOrVcl],( hrd->getInitialCpbRemovalDelayLengthMinus1() + 1 ) , "initial_cpb_removal_delay" );
WRITE_CODE( sei.m_initialCpbRemovalDelayOffset[i][nalOrVcl],( hrd->getInitialCpbRemovalDelayLengthMinus1() + 1 ), "initial_cpb_removal_delay_offset" );
if( hrd->getSubPicCpbParamsPresentFlag() || sei.m_rapCpbParamsPresentFlag )
{
WRITE_CODE( sei.m_initialAltCpbRemovalDelay[i][nalOrVcl], ( hrd->getInitialCpbRemovalDelayLengthMinus1() + 1 ) , "initial_alt_cpb_removal_delay" );
WRITE_CODE( sei.m_initialAltCpbRemovalDelayOffset[i][nalOrVcl], ( hrd->getInitialCpbRemovalDelayLengthMinus1() + 1 ),"initial_alt_cpb_removal_delay_offset" );
}
}
}
}
xWriteByteAlign();
}
Void SEIWriter::xWriteSEIDecodingUnitInfo(const SEIDecodingUnitInfo& sei, const TComSPS *sps)
{
const TComVUI *vui = sps->getVuiParameters();
WRITE_UVLC(sei.m_decodingUnitIdx, "decoding_unit_idx");
if(vui->getHrdParameters()->getSubPicCpbParamsInPicTimingSEIFlag())
{
WRITE_CODE( sei.m_duSptCpbRemovalDelay, (vui->getHrdParameters()->getDuCpbRemovalDelayLengthMinus1() + 1), "du_spt_cpb_removal_delay");
}
WRITE_FLAG( sei.m_dpbOutputDuDelayPresentFlag, "dpb_output_du_delay_present_flag");
if(sei.m_dpbOutputDuDelayPresentFlag)
{
WRITE_CODE(sei.m_picSptDpbOutputDuDelay, vui->getHrdParameters()->getDpbOutputDelayDuLengthMinus1() + 1, "pic_spt_dpb_output_du_delay");
}
}
Void SEIWriter::xWriteSEIActiveParameterSets(const SEIActiveParameterSets& sei)
{
WRITE_CODE(sei.activeVPSId, 4, "active_vps_id");
WRITE_CODE(sei.activeSPSIdPresentFlag, 1, "active_sps_id_present_flag");
if (sei.activeSPSIdPresentFlag)
{
WRITE_UVLC(sei.activeSeqParamSetId, "active_seq_param_set_id");
}
WRITE_CODE(sei.activeParamSetSEIExtensionFlag, 1, "active_param_set_sei_extension_flag");
UInt uiBits = m_pcBitIf->getNumberOfWrittenBits();
UInt uiAlignedBits = ( 8 - (uiBits&7) ) % 8;
if(uiAlignedBits)
{
WRITE_FLAG(1, "alignment_bit" );
uiAlignedBits--;
while(uiAlignedBits--)
{
WRITE_FLAG(0, "alignment_bit" );
}
}
}
Void SEIWriter::xWriteSEIToneMappingInfo(const SEIToneMappingInfo& sei)
{
Int i;
WRITE_UVLC( sei.m_toneMapId, "tone_map_id" );
WRITE_FLAG( sei.m_toneMapCancelFlag, "tone_map_cancel_flag" );
if( !sei.m_toneMapCancelFlag )
{
WRITE_FLAG( sei.m_toneMapPersistenceFlag, "tone_map_persistence_flag" );
WRITE_CODE( sei.m_codedDataBitDepth, 8, "coded_data_bit_depth" );
WRITE_CODE( sei.m_targetBitDepth, 8, "target_bit_depth" );
WRITE_UVLC( sei.m_modelId, "model_id" );
switch(sei.m_modelId)
{
case 0:
{
WRITE_CODE( sei.m_minValue, 32, "min_value" );
WRITE_CODE( sei.m_maxValue, 32, "max_value" );
break;
}
case 1:
{
WRITE_CODE( sei.m_sigmoidMidpoint, 32, "sigmoid_midpoint" );
WRITE_CODE( sei.m_sigmoidWidth, 32, "sigmoid_width" );
break;
}
case 2:
{
UInt num = 1u << sei.m_targetBitDepth;
for(i = 0; i < num; i++)
{
WRITE_CODE( sei.m_startOfCodedInterval[i], (( sei.m_codedDataBitDepth + 7 ) >> 3 ) << 3, "start_of_coded_interval" );
}
break;
}
case 3:
{
WRITE_CODE( sei.m_numPivots, 16, "num_pivots" );
for(i = 0; i < sei.m_numPivots; i++ )
{
WRITE_CODE( sei.m_codedPivotValue[i], (( sei.m_codedDataBitDepth + 7 ) >> 3 ) << 3, "coded_pivot_value" );
WRITE_CODE( sei.m_targetPivotValue[i], (( sei.m_targetBitDepth + 7 ) >> 3 ) << 3, "target_pivot_value");
}
break;
}
case 4:
{
WRITE_CODE( sei.m_cameraIsoSpeedIdc, 8, "camera_iso_speed_idc" );
if( sei.m_cameraIsoSpeedIdc == 255) //Extended_ISO
{
WRITE_CODE( sei.m_cameraIsoSpeedValue, 32, "camera_iso_speed_value" );
}
WRITE_CODE( sei.m_exposureIndexIdc, 8, "exposure_index_idc" );
if( sei.m_exposureIndexIdc == 255) //Extended_ISO
{
WRITE_CODE( sei.m_exposureIndexValue, 32, "exposure_index_value" );
}
WRITE_FLAG( sei.m_exposureCompensationValueSignFlag, "exposure_compensation_value_sign_flag" );
WRITE_CODE( sei.m_exposureCompensationValueNumerator, 16, "exposure_compensation_value_numerator" );
WRITE_CODE( sei.m_exposureCompensationValueDenomIdc, 16, "exposure_compensation_value_denom_idc" );
WRITE_CODE( sei.m_refScreenLuminanceWhite, 32, "ref_screen_luminance_white" );
WRITE_CODE( sei.m_extendedRangeWhiteLevel, 32, "extended_range_white_level" );
WRITE_CODE( sei.m_nominalBlackLevelLumaCodeValue, 16, "nominal_black_level_luma_code_value" );
WRITE_CODE( sei.m_nominalWhiteLevelLumaCodeValue, 16, "nominal_white_level_luma_code_value" );
WRITE_CODE( sei.m_extendedWhiteLevelLumaCodeValue, 16, "extended_white_level_luma_code_value" );
break;
}
default:
{
assert(!"Undefined SEIToneMapModelId");
break;
}
}//switch m_modelId
}//if(!sei.m_toneMapCancelFlag)
Void SEIWriter::xWriteSEIGradualDecodingRefreshInfo(const SEIGradualDecodingRefreshInfo &sei)
{
WRITE_FLAG( sei.m_gdrForegroundFlag, "gdr_foreground_flag");
xWriteByteAlign();
}
