791 lines
31 KiB
C++
791 lines
31 KiB
C++
/* The copyright in this software is being made available under the BSD
|
|
* License, included below. This software may be subject to other third party
|
|
* and contributor rights, including patent rights, and no such rights are
|
|
* granted under this license.
|
|
*
|
|
* Copyright (c) 2010-2014, ITU/ISO/IEC
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions are met:
|
|
*
|
|
* * Redistributions of source code must retain the above copyright notice,
|
|
* this list of conditions and the following disclaimer.
|
|
* * Redistributions in binary form must reproduce the above copyright notice,
|
|
* this list of conditions and the following disclaimer in the documentation
|
|
* and/or other materials provided with the distribution.
|
|
* * Neither the name of the ITU/ISO/IEC nor the names of its contributors may
|
|
* be used to endorse or promote products derived from this software without
|
|
* specific prior written permission.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS
|
|
* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
|
|
* THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#include "TLibCommon/TComBitCounter.h"
|
|
#include "TLibCommon/TComBitStream.h"
|
|
#include "TLibCommon/SEI.h"
|
|
#include "TLibCommon/TComSlice.h"
|
|
#include "TLibCommon/TComPicYuv.h"
|
|
#include "SEIwrite.h"
|
|
|
|
//! \ingroup TLibEncoder
|
|
//! \{
|
|
|
|
#if ENC_DEC_TRACE
|
|
Void xTraceSEIHeader()
|
|
{
|
|
fprintf( g_hTrace, "=========== SEI message ===========\n");
|
|
}
|
|
|
|
Void xTraceSEIMessageType(SEI::PayloadType payloadType)
|
|
{
|
|
fprintf( g_hTrace, "=========== %s SEI message ===========\n", SEI::getSEIMessageString(payloadType));
|
|
}
|
|
#endif
|
|
|
|
Void SEIWriter::xWriteSEIpayloadData(TComBitIf& bs, const SEI& sei, TComSPS *sps)
|
|
{
|
|
switch (sei.payloadType())
|
|
{
|
|
case SEI::USER_DATA_UNREGISTERED:
|
|
xWriteSEIuserDataUnregistered(*static_cast<const SEIuserDataUnregistered*>(&sei));
|
|
break;
|
|
case SEI::ACTIVE_PARAMETER_SETS:
|
|
xWriteSEIActiveParameterSets(*static_cast<const SEIActiveParameterSets*>(& sei));
|
|
break;
|
|
case SEI::DECODING_UNIT_INFO:
|
|
xWriteSEIDecodingUnitInfo(*static_cast<const SEIDecodingUnitInfo*>(& sei), sps);
|
|
break;
|
|
case SEI::DECODED_PICTURE_HASH:
|
|
xWriteSEIDecodedPictureHash(*static_cast<const SEIDecodedPictureHash*>(&sei));
|
|
break;
|
|
case SEI::BUFFERING_PERIOD:
|
|
xWriteSEIBufferingPeriod(*static_cast<const SEIBufferingPeriod*>(&sei), sps);
|
|
break;
|
|
case SEI::PICTURE_TIMING:
|
|
xWriteSEIPictureTiming(*static_cast<const SEIPictureTiming*>(&sei), sps);
|
|
break;
|
|
case SEI::RECOVERY_POINT:
|
|
xWriteSEIRecoveryPoint(*static_cast<const SEIRecoveryPoint*>(&sei));
|
|
break;
|
|
case SEI::FRAME_PACKING:
|
|
xWriteSEIFramePacking(*static_cast<const SEIFramePacking*>(&sei));
|
|
break;
|
|
case SEI::SEGM_RECT_FRAME_PACKING:
|
|
xWriteSEISegmentedRectFramePacking(*static_cast<const SEISegmentedRectFramePacking*>(&sei));
|
|
break;
|
|
case SEI::DISPLAY_ORIENTATION:
|
|
xWriteSEIDisplayOrientation(*static_cast<const SEIDisplayOrientation*>(&sei));
|
|
break;
|
|
case SEI::TEMPORAL_LEVEL0_INDEX:
|
|
xWriteSEITemporalLevel0Index(*static_cast<const SEITemporalLevel0Index*>(&sei));
|
|
break;
|
|
case SEI::REGION_REFRESH_INFO:
|
|
xWriteSEIGradualDecodingRefreshInfo(*static_cast<const SEIGradualDecodingRefreshInfo*>(&sei));
|
|
break;
|
|
case SEI::NO_DISPLAY:
|
|
xWriteSEINoDisplay(*static_cast<const SEINoDisplay*>(&sei));
|
|
break;
|
|
case SEI::TONE_MAPPING_INFO:
|
|
xWriteSEIToneMappingInfo(*static_cast<const SEIToneMappingInfo*>(&sei));
|
|
break;
|
|
case SEI::SOP_DESCRIPTION:
|
|
xWriteSEISOPDescription(*static_cast<const SEISOPDescription*>(&sei));
|
|
break;
|
|
case SEI::SCALABLE_NESTING:
|
|
xWriteSEIScalableNesting(bs, *static_cast<const SEIScalableNesting*>(&sei), sps);
|
|
break;
|
|
case SEI::CHROMA_SAMPLING_FILTER_HINT:
|
|
xWriteSEIChromaSamplingFilterHint(*static_cast<const SEIChromaSamplingFilterHint*>(&sei)/*, sps*/);
|
|
break;
|
|
case SEI::TEMP_MOTION_CONSTRAINED_TILE_SETS:
|
|
xWriteSEITempMotionConstrainedTileSets(bs, *static_cast<const SEITempMotionConstrainedTileSets*>(&sei));
|
|
break;
|
|
case SEI::TIME_CODE:
|
|
xWriteSEITimeCode(*static_cast<const SEITimeCode*>(&sei));
|
|
break;
|
|
case SEI::KNEE_FUNCTION_INFO:
|
|
xWriteSEIKneeFunctionInfo(*static_cast<const SEIKneeFunctionInfo*>(&sei));
|
|
break;
|
|
case SEI::MASTERING_DISPLAY_COLOUR_VOLUME:
|
|
xWriteSEIMasteringDisplayColourVolume(*static_cast<const SEIMasteringDisplayColourVolume*>(&sei));
|
|
break;
|
|
default:
|
|
assert(!"Unhandled SEI message");
|
|
break;
|
|
}
|
|
xWriteByteAlign();
|
|
}
|
|
|
|
/**
|
|
* marshal a single SEI message sei, storing the marshalled representation
|
|
* in bitstream bs.
|
|
*/
|
|
Void SEIWriter::writeSEImessage(TComBitIf& bs, const SEI& sei, TComSPS *sps)
|
|
{
|
|
/* calculate how large the payload data is */
|
|
/* TODO: this would be far nicer if it used vectored buffers */
|
|
TComBitCounter bs_count;
|
|
bs_count.resetBits();
|
|
setBitstream(&bs_count);
|
|
|
|
|
|
#if ENC_DEC_TRACE
|
|
Bool traceEnable = g_HLSTraceEnable;
|
|
g_HLSTraceEnable = false;
|
|
#endif
|
|
xWriteSEIpayloadData(bs_count, sei, sps);
|
|
#if ENC_DEC_TRACE
|
|
g_HLSTraceEnable = traceEnable;
|
|
#endif
|
|
|
|
UInt payload_data_num_bits = bs_count.getNumberOfWrittenBits();
|
|
assert(0 == payload_data_num_bits % 8);
|
|
|
|
setBitstream(&bs);
|
|
|
|
#if ENC_DEC_TRACE
|
|
if (g_HLSTraceEnable)
|
|
xTraceSEIHeader();
|
|
#endif
|
|
|
|
UInt payloadType = sei.payloadType();
|
|
for (; payloadType >= 0xff; payloadType -= 0xff)
|
|
{
|
|
WRITE_CODE(0xff, 8, "payload_type");
|
|
}
|
|
WRITE_CODE(payloadType, 8, "payload_type");
|
|
|
|
UInt payloadSize = payload_data_num_bits/8;
|
|
for (; payloadSize >= 0xff; payloadSize -= 0xff)
|
|
{
|
|
WRITE_CODE(0xff, 8, "payload_size");
|
|
}
|
|
WRITE_CODE(payloadSize, 8, "payload_size");
|
|
|
|
/* payloadData */
|
|
#if ENC_DEC_TRACE
|
|
if (g_HLSTraceEnable)
|
|
xTraceSEIMessageType(sei.payloadType());
|
|
#endif
|
|
|
|
xWriteSEIpayloadData(bs, sei, sps);
|
|
}
|
|
|
|
/**
|
|
* marshal a user_data_unregistered SEI message sei, storing the marshalled
|
|
* representation in bitstream bs.
|
|
*/
|
|
Void SEIWriter::xWriteSEIuserDataUnregistered(const SEIuserDataUnregistered &sei)
|
|
{
|
|
for (UInt i = 0; i < ISO_IEC_11578_LEN; i++)
|
|
{
|
|
WRITE_CODE(sei.uuid_iso_iec_11578[i], 8 , "sei.uuid_iso_iec_11578[i]");
|
|
}
|
|
|
|
for (UInt i = 0; i < sei.userDataLength; i++)
|
|
{
|
|
WRITE_CODE(sei.userData[i], 8 , "user_data");
|
|
}
|
|
}
|
|
|
|
/**
|
|
* marshal a decoded picture hash SEI message, storing the marshalled
|
|
* representation in bitstream bs.
|
|
*/
|
|
Void SEIWriter::xWriteSEIDecodedPictureHash(const SEIDecodedPictureHash& sei)
|
|
{
|
|
const Char *traceString="\0";
|
|
switch (sei.method)
|
|
{
|
|
case SEIDecodedPictureHash::MD5: traceString="picture_md5"; break;
|
|
case SEIDecodedPictureHash::CRC: traceString="picture_crc"; break;
|
|
case SEIDecodedPictureHash::CHECKSUM: traceString="picture_checksum"; break;
|
|
default: assert(false); break;
|
|
}
|
|
|
|
if (traceString != 0) //use of this variable is needed to avoid a compiler error with G++ 4.6.1
|
|
{
|
|
WRITE_CODE(sei.method, 8, "hash_type");
|
|
for(UInt i=0; i<UInt(sei.m_digest.hash.size()); i++)
|
|
{
|
|
WRITE_CODE(sei.m_digest.hash[i], 8, traceString);
|
|
}
|
|
}
|
|
}
|
|
|
|
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::xWriteSEIDecodingUnitInfo(const SEIDecodingUnitInfo& sei, TComSPS *sps)
|
|
{
|
|
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::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, "irap_cpb_params_present_flag" );
|
|
}
|
|
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" );
|
|
}
|
|
WRITE_FLAG( sei.m_concatenationFlag, "concatenation_flag");
|
|
WRITE_CODE( sei.m_auCpbRemovalDelayDelta - 1, ( hrd->getCpbRemovalDelayLengthMinus1() + 1 ), "au_cpb_removal_delay_delta_minus1" );
|
|
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" );
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
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" );
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
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" );
|
|
}
|
|
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" );
|
|
}
|
|
|
|
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::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::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" );
|
|
WRITE_FLAG( sei.persistenceFlag, "display_orientation_persistence_flag" );
|
|
}
|
|
}
|
|
|
|
Void SEIWriter::xWriteSEITemporalLevel0Index(const SEITemporalLevel0Index &sei)
|
|
{
|
|
WRITE_CODE( sei.tl0Idx, 8 , "tl0_idx" );
|
|
WRITE_CODE( sei.rapIdx, 8 , "rap_idx" );
|
|
}
|
|
|
|
Void SEIWriter::xWriteSEIGradualDecodingRefreshInfo(const SEIGradualDecodingRefreshInfo &sei)
|
|
{
|
|
WRITE_FLAG( sei.m_gdrForegroundFlag, "gdr_foreground_flag");
|
|
}
|
|
|
|
Void SEIWriter::xWriteSEINoDisplay(const SEINoDisplay &sei)
|
|
{
|
|
}
|
|
|
|
Void SEIWriter::xWriteSEISOPDescription(const SEISOPDescription& sei)
|
|
{
|
|
WRITE_UVLC( sei.m_sopSeqParameterSetId, "sop_seq_parameter_set_id" );
|
|
WRITE_UVLC( sei.m_numPicsInSopMinus1, "num_pics_in_sop_minus1" );
|
|
for (UInt i = 0; i <= sei.m_numPicsInSopMinus1; i++)
|
|
{
|
|
WRITE_CODE( sei.m_sopDescVclNaluType[i], 6, "sop_desc_vcl_nalu_type" );
|
|
WRITE_CODE( sei.m_sopDescTemporalId[i], 3, "sop_desc_temporal_id" );
|
|
if (sei.m_sopDescVclNaluType[i] != NAL_UNIT_CODED_SLICE_IDR_W_RADL && sei.m_sopDescVclNaluType[i] != NAL_UNIT_CODED_SLICE_IDR_N_LP)
|
|
{
|
|
WRITE_UVLC( sei.m_sopDescStRpsIdx[i], "sop_desc_st_rps_idx" );
|
|
}
|
|
if (i > 0)
|
|
{
|
|
WRITE_SVLC( sei.m_sopDescPocDelta[i], "sop_desc_poc_delta" );
|
|
}
|
|
}
|
|
}
|
|
|
|
Void SEIWriter::xWriteSEIScalableNesting(TComBitIf& bs, const SEIScalableNesting& sei, TComSPS *sps)
|
|
{
|
|
WRITE_FLAG( sei.m_bitStreamSubsetFlag, "bitstream_subset_flag" );
|
|
WRITE_FLAG( sei.m_nestingOpFlag, "nesting_op_flag " );
|
|
if (sei.m_nestingOpFlag)
|
|
{
|
|
WRITE_FLAG( sei.m_defaultOpFlag, "default_op_flag" );
|
|
WRITE_UVLC( sei.m_nestingNumOpsMinus1, "nesting_num_ops_minus1" );
|
|
for (UInt i = (sei.m_defaultOpFlag ? 1 : 0); i <= sei.m_nestingNumOpsMinus1; i++)
|
|
{
|
|
WRITE_CODE( sei.m_nestingMaxTemporalIdPlus1[i], 3, "nesting_max_temporal_id_plus1" );
|
|
WRITE_UVLC( sei.m_nestingOpIdx[i], "nesting_op_idx" );
|
|
}
|
|
}
|
|
else
|
|
{
|
|
WRITE_FLAG( sei.m_allLayersFlag, "all_layers_flag" );
|
|
if (!sei.m_allLayersFlag)
|
|
{
|
|
WRITE_CODE( sei.m_nestingNoOpMaxTemporalIdPlus1, 3, "nesting_no_op_max_temporal_id_plus1" );
|
|
WRITE_UVLC( sei.m_nestingNumLayersMinus1, "nesting_num_layers" );
|
|
for (UInt i = 0; i <= sei.m_nestingNumLayersMinus1; i++)
|
|
{
|
|
WRITE_CODE( sei.m_nestingLayerId[i], 6, "nesting_layer_id" );
|
|
}
|
|
}
|
|
}
|
|
|
|
// byte alignment
|
|
while ( m_pcBitIf->getNumberOfWrittenBits() % 8 != 0 )
|
|
{
|
|
WRITE_FLAG( 0, "nesting_zero_bit" );
|
|
}
|
|
|
|
// write nested SEI messages
|
|
for (SEIMessages::const_iterator it = sei.m_nestedSEIs.begin(); it != sei.m_nestedSEIs.end(); it++)
|
|
{
|
|
writeSEImessage(bs, *(*it), sps);
|
|
}
|
|
}
|
|
|
|
Void SEIWriter::xWriteSEITempMotionConstrainedTileSets(TComBitIf& bs, const SEITempMotionConstrainedTileSets& sei)
|
|
{
|
|
//UInt code;
|
|
WRITE_FLAG((sei.m_mc_all_tiles_exact_sample_value_match_flag ? 1 : 0), "mc_all_tiles_exact_sample_value_match_flag");
|
|
WRITE_FLAG((sei.m_each_tile_one_tile_set_flag ? 1 : 0), "each_tile_one_tile_set_flag" );
|
|
|
|
if(!sei.m_each_tile_one_tile_set_flag)
|
|
{
|
|
WRITE_FLAG((sei.m_limited_tile_set_display_flag ? 1 : 0), "limited_tile_set_display_flag");
|
|
WRITE_UVLC((sei.getNumberOfTileSets() - 1), "num_sets_in_message_minus1" );
|
|
|
|
if(sei.getNumberOfTileSets() > 0)
|
|
{
|
|
for(Int i = 0; i < sei.getNumberOfTileSets(); i++)
|
|
{
|
|
WRITE_UVLC(sei.tileSetData(i).m_mcts_id, "mcts_id");
|
|
|
|
if(sei.m_limited_tile_set_display_flag)
|
|
{
|
|
WRITE_FLAG((sei.tileSetData(i).m_display_tile_set_flag ? 1 : 0), "display_tile_set_flag");
|
|
}
|
|
|
|
WRITE_UVLC((sei.tileSetData(i).getNumberOfTileRects() - 1), "num_tile_rects_in_set_minus1");
|
|
|
|
for(Int j = 0; j < sei.tileSetData(i).getNumberOfTileRects(); j++)
|
|
{
|
|
WRITE_UVLC(sei.tileSetData(i).topLeftTileIndex (j), "top_left_tile_index");
|
|
WRITE_UVLC(sei.tileSetData(i).bottomRightTileIndex(j), "bottom_right_tile_index");
|
|
}
|
|
|
|
if(!sei.m_mc_all_tiles_exact_sample_value_match_flag)
|
|
{
|
|
WRITE_FLAG((sei.tileSetData(i).m_exact_sample_value_match_flag ? 1 : 0), "exact_sample_value_match_flag");
|
|
}
|
|
|
|
WRITE_FLAG((sei.tileSetData(i).m_mcts_tier_level_idc_present_flag ? 1 : 0), "mcts_tier_level_idc_present_flag");
|
|
|
|
if(sei.tileSetData(i).m_mcts_tier_level_idc_present_flag)
|
|
{
|
|
WRITE_FLAG((sei.tileSetData(i).m_mcts_tier_flag ? 1 : 0), "mcts_tier_flag");
|
|
WRITE_CODE( sei.tileSetData(i).m_mcts_level_idc, 8, "mcts_level_idc");
|
|
}
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
WRITE_FLAG((sei.m_max_mcs_tier_level_idc_present_flag ? 1 : 0), "max_mcs_tier_level_idc_present_flag");
|
|
|
|
if(sei.m_max_mcs_tier_level_idc_present_flag)
|
|
{
|
|
WRITE_FLAG((sei.m_max_mcts_tier_flag ? 1 : 0), "max_mcts_tier_flag");
|
|
WRITE_CODE( sei.m_max_mcts_level_idc, 8, "max_mcts_level_idc");
|
|
}
|
|
}
|
|
}
|
|
|
|
Void SEIWriter::xWriteSEITimeCode(const SEITimeCode& sei)
|
|
{
|
|
WRITE_CODE(sei.numClockTs, 2, "num_clock_ts");
|
|
for(Int i = 0; i < sei.numClockTs; i++)
|
|
{
|
|
const TComSEITimeSet ¤tTimeSet = sei.timeSetArray[i];
|
|
WRITE_FLAG(currentTimeSet.clockTimeStampFlag, "clock_time_stamp_flag");
|
|
if(currentTimeSet.clockTimeStampFlag)
|
|
{
|
|
WRITE_FLAG(currentTimeSet.numUnitFieldBasedFlag, "units_field_based_flag");
|
|
WRITE_CODE(currentTimeSet.countingType, 5, "counting_type");
|
|
WRITE_FLAG(currentTimeSet.fullTimeStampFlag, "full_timestamp_flag");
|
|
WRITE_FLAG(currentTimeSet.discontinuityFlag, "discontinuity_flag");
|
|
WRITE_FLAG(currentTimeSet.cntDroppedFlag, "cnt_dropped_flag");
|
|
WRITE_CODE(currentTimeSet.numberOfFrames, 9, "n_frames");
|
|
if(currentTimeSet.fullTimeStampFlag)
|
|
{
|
|
WRITE_CODE(currentTimeSet.secondsValue, 6, "seconds_value");
|
|
WRITE_CODE(currentTimeSet.minutesValue, 6, "minutes_value");
|
|
WRITE_CODE(currentTimeSet.hoursValue, 5, "hours_value");
|
|
}
|
|
else
|
|
{
|
|
WRITE_FLAG(currentTimeSet.secondsFlag, "seconds_flag");
|
|
if(currentTimeSet.secondsFlag)
|
|
{
|
|
WRITE_CODE(currentTimeSet.secondsValue, 6, "seconds_value");
|
|
WRITE_FLAG(currentTimeSet.minutesFlag, "minutes_flag");
|
|
if(currentTimeSet.minutesFlag)
|
|
{
|
|
WRITE_CODE(currentTimeSet.minutesValue, 6, "minutes_value");
|
|
WRITE_FLAG(currentTimeSet.hoursFlag, "hours_flag");
|
|
if(currentTimeSet.hoursFlag)
|
|
WRITE_CODE(currentTimeSet.hoursValue, 5, "hours_value");
|
|
}
|
|
}
|
|
}
|
|
WRITE_CODE(currentTimeSet.timeOffsetLength, 5, "time_offset_length");
|
|
if(currentTimeSet.timeOffsetLength > 0)
|
|
{
|
|
if(currentTimeSet.timeOffsetValue >= 0)
|
|
{
|
|
WRITE_CODE((UInt)currentTimeSet.timeOffsetValue, currentTimeSet.timeOffsetLength, "time_offset_value");
|
|
}
|
|
else
|
|
{
|
|
// Two's complement conversion
|
|
UInt offsetValue = ~(currentTimeSet.timeOffsetValue) + 1;
|
|
offsetValue |= (1 << (currentTimeSet.timeOffsetLength-1));
|
|
WRITE_CODE(offsetValue, currentTimeSet.timeOffsetLength, "time_offset_value");
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
Void SEIWriter::xWriteSEIChromaSamplingFilterHint(const SEIChromaSamplingFilterHint &sei/*, TComSPS* sps*/)
|
|
{
|
|
WRITE_CODE(sei.m_verChromaFilterIdc, 8, "ver_chroma_filter_idc");
|
|
WRITE_CODE(sei.m_horChromaFilterIdc, 8, "hor_chroma_filter_idc");
|
|
WRITE_FLAG(sei.m_verFilteringProcessFlag, "ver_filtering_process_flag");
|
|
if(sei.m_verChromaFilterIdc == 1 || sei.m_horChromaFilterIdc == 1)
|
|
{
|
|
writeUserDefinedCoefficients(sei);
|
|
}
|
|
}
|
|
|
|
// write hardcoded chroma filter coefficients in the SEI messages
|
|
Void SEIWriter::writeUserDefinedCoefficients(const SEIChromaSamplingFilterHint &sei)
|
|
{
|
|
Int const iNumVerticalFilters = 3;
|
|
Int verticalTapLength_minus1[iNumVerticalFilters] = {5,3,3};
|
|
Int* userVerticalCoefficients[iNumVerticalFilters];
|
|
for(Int i = 0; i < iNumVerticalFilters; i ++)
|
|
{
|
|
userVerticalCoefficients[i] = (Int*)malloc( (verticalTapLength_minus1[i]+1) * sizeof(Int));
|
|
}
|
|
userVerticalCoefficients[0][0] = -3;
|
|
userVerticalCoefficients[0][1] = 13;
|
|
userVerticalCoefficients[0][2] = 31;
|
|
userVerticalCoefficients[0][3] = 23;
|
|
userVerticalCoefficients[0][4] = 3;
|
|
userVerticalCoefficients[0][5] = -3;
|
|
|
|
userVerticalCoefficients[1][0] = -1;
|
|
userVerticalCoefficients[1][1] = 25;
|
|
userVerticalCoefficients[1][2] = 247;
|
|
userVerticalCoefficients[1][3] = -15;
|
|
|
|
userVerticalCoefficients[2][0] = -20;
|
|
userVerticalCoefficients[2][1] = 186;
|
|
userVerticalCoefficients[2][2] = 100;
|
|
userVerticalCoefficients[2][3] = -10;
|
|
|
|
Int const iNumHorizontalFilters = 1;
|
|
Int horizontalTapLength_minus1[iNumHorizontalFilters] = {3};
|
|
Int* userHorizontalCoefficients[iNumHorizontalFilters];
|
|
for(Int i = 0; i < iNumHorizontalFilters; i ++)
|
|
{
|
|
userHorizontalCoefficients[i] = (Int*)malloc( (horizontalTapLength_minus1[i]+1) * sizeof(Int));
|
|
}
|
|
userHorizontalCoefficients[0][0] = 1;
|
|
userHorizontalCoefficients[0][1] = 6;
|
|
userHorizontalCoefficients[0][2] = 1;
|
|
|
|
WRITE_UVLC(3, "target_format_idc");
|
|
if(sei.m_verChromaFilterIdc == 1)
|
|
{
|
|
WRITE_UVLC(iNumVerticalFilters, "num_vertical_filters");
|
|
if(iNumVerticalFilters > 0)
|
|
{
|
|
for(Int i = 0; i < iNumVerticalFilters; i ++)
|
|
{
|
|
WRITE_UVLC(verticalTapLength_minus1[i], "ver_tap_length_minus_1");
|
|
for(Int j = 0; j < verticalTapLength_minus1[i]; j ++)
|
|
{
|
|
WRITE_SVLC(userVerticalCoefficients[i][j], "ver_filter_coeff");
|
|
}
|
|
}
|
|
}
|
|
}
|
|
if(sei.m_horChromaFilterIdc == 1)
|
|
{
|
|
WRITE_UVLC(iNumHorizontalFilters, "num_horizontal_filters");
|
|
if(iNumHorizontalFilters > 0)
|
|
{
|
|
for(Int i = 0; i < iNumHorizontalFilters; i ++)
|
|
{
|
|
WRITE_UVLC(horizontalTapLength_minus1[i], "hor_tap_length_minus_1");
|
|
for(Int j = 0; j < horizontalTapLength_minus1[i]; j ++)
|
|
{
|
|
WRITE_SVLC(userHorizontalCoefficients[i][j], "hor_filter_coeff");
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
Void SEIWriter::xWriteSEIKneeFunctionInfo(const SEIKneeFunctionInfo &sei)
|
|
{
|
|
WRITE_UVLC( sei.m_kneeId, "knee_function_id" );
|
|
WRITE_FLAG( sei.m_kneeCancelFlag, "knee_function_cancel_flag" );
|
|
if ( !sei.m_kneeCancelFlag )
|
|
{
|
|
WRITE_FLAG( sei.m_kneePersistenceFlag, "knee_function_persistence_flag" );
|
|
WRITE_CODE( (UInt)sei.m_kneeInputDrange , 32, "input_d_range" );
|
|
WRITE_CODE( (UInt)sei.m_kneeInputDispLuminance, 32, "input_disp_luminance" );
|
|
WRITE_CODE( (UInt)sei.m_kneeOutputDrange, 32, "output_d_range" );
|
|
WRITE_CODE( (UInt)sei.m_kneeOutputDispLuminance, 32, "output_disp_luminance" );
|
|
WRITE_UVLC( sei.m_kneeNumKneePointsMinus1, "num_knee_points_minus1" );
|
|
for(Int i = 0; i <= sei.m_kneeNumKneePointsMinus1; i++ )
|
|
{
|
|
WRITE_CODE( (UInt)sei.m_kneeInputKneePoint[i], 10,"input_knee_point" );
|
|
WRITE_CODE( (UInt)sei.m_kneeOutputKneePoint[i], 10, "output_knee_point" );
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
Void SEIWriter::xWriteSEIMasteringDisplayColourVolume(const SEIMasteringDisplayColourVolume& sei)
|
|
{
|
|
WRITE_CODE( sei.values.primaries[0][0], 16, "display_primaries_x[0]" );
|
|
WRITE_CODE( sei.values.primaries[0][1], 16, "display_primaries_y[0]" );
|
|
|
|
WRITE_CODE( sei.values.primaries[1][0], 16, "display_primaries_x[1]" );
|
|
WRITE_CODE( sei.values.primaries[1][1], 16, "display_primaries_y[1]" );
|
|
|
|
WRITE_CODE( sei.values.primaries[2][0], 16, "display_primaries_x[2]" );
|
|
WRITE_CODE( sei.values.primaries[2][1], 16, "display_primaries_y[2]" );
|
|
|
|
WRITE_CODE( sei.values.whitePoint[0], 16, "white_point_x" );
|
|
WRITE_CODE( sei.values.whitePoint[1], 16, "white_point_y" );
|
|
|
|
WRITE_CODE( sei.values.maxLuminance, 32, "max_display_mastering_luminance" );
|
|
WRITE_CODE( sei.values.minLuminance, 32, "min_display_mastering_luminance" );
|
|
}
|
|
|
|
|
|
Void SEIWriter::xWriteByteAlign()
|
|
{
|
|
if( m_pcBitIf->getNumberOfWrittenBits() % 8 != 0)
|
|
{
|
|
WRITE_FLAG( 1, "payload_bit_equal_to_one" );
|
|
while( m_pcBitIf->getNumberOfWrittenBits() % 8 != 0 )
|
|
{
|
|
WRITE_FLAG( 0, "payload_bit_equal_to_zero" );
|
|
}
|
|
}
|
|
}
|
|
|
|
//! \}
|