Void SEIReader::xReadSEImessage(SEImessages& seis)
{
#if ENC_DEC_TRACE
xTraceSEIHeader();
#endif
Int payloadType = 0;
UInt val = 0;
do
{
READ_CODE (8, val, "payload_type");
payloadType += val;
} while (val==0xFF);
UInt payloadSize = 0;
do
{
READ_CODE (8, val, "payload_size");
payloadSize += val;
} while (val==0xFF);
#if ENC_DEC_TRACE
xTraceSEIMessageType((SEI::PayloadType)payloadType);
#endif
switch (payloadType)
{
case SEI::USER_DATA_UNREGISTERED:
seis.user_data_unregistered = new SEIuserDataUnregistered;
xParseSEIuserDataUnregistered(*seis.user_data_unregistered, payloadSize);
break;
case SEI::ACTIVE_PARAMETER_SETS:
seis.active_parameter_sets = new SEIActiveParameterSets;
xParseSEIActiveParameterSets(*seis.active_parameter_sets, payloadSize);
break;
case SEI::DECODED_PICTURE_HASH:
seis.picture_digest = new SEIDecodedPictureHash;
xParseSEIDecodedPictureHash(*seis.picture_digest, payloadSize);
break;
case SEI::BUFFERING_PERIOD:
seis.buffering_period = new SEIBufferingPeriod;
seis.buffering_period->m_sps = seis.m_pSPS;
xParseSEIBufferingPeriod(*seis.buffering_period, payloadSize);
break;
case SEI::PICTURE_TIMING:
seis.picture_timing = new SEIPictureTiming;
seis.picture_timing->m_sps = seis.m_pSPS;
xParseSEIPictureTiming(*seis.picture_timing, payloadSize);
break;
case SEI::RECOVERY_POINT:
seis.recovery_point = new SEIRecoveryPoint;
xParseSEIRecoveryPoint(*seis.recovery_point, payloadSize);
break;
default:
assert(!"Unhandled SEI message");
}
}
/**
* marshal all SEI messages in provided list into one bitstream bs
*/
Void SEIWriter::writeSEImessages(TComBitIf& bs, const SEIMessages &seiList, const TComSPS *sps, Bool isNested)
{
#if ENC_DEC_TRACE
if (g_HLSTraceEnable)
xTraceSEIHeader();
#endif
TComBitCounter bs_count;
for (SEIMessages::const_iterator sei=seiList.begin(); sei!=seiList.end(); sei++)
{
// calculate how large the payload data is
// TODO: this would be far nicer if it used vectored buffers
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);
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);
}
if (!isNested)
{
xWriteRbspTrailingBits();
}
}
/**
* 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);
}
Void SEIReader::xReadSEImessage(SEIMessages& seis, const NalUnitType nalUnitType, TComSPS *sps)
{
#if ENC_DEC_TRACE
xTraceSEIHeader();
#endif
Int payloadType = 0;
UInt val = 0;
do
{
READ_CODE (8, val, "payload_type");
payloadType += val;
} while (val==0xFF);
UInt payloadSize = 0;
do
{
READ_CODE (8, val, "payload_size");
payloadSize += val;
} while (val==0xFF);
#if ENC_DEC_TRACE
xTraceSEIMessageType((SEI::PayloadType)payloadType);
#endif
/* extract the payload for this single SEI message.
* This allows greater safety in erroneous parsing of an SEI message
* from affecting subsequent messages.
* After parsing the payload, bs needs to be restored as the primary
* bitstream.
*/
TComInputBitstream *bs = getBitstream();
setBitstream(bs->extractSubstream(payloadSize * 8));
SEI *sei = NULL;
if(nalUnitType == NAL_UNIT_SEI)
{
switch (payloadType)
{
case SEI::USER_DATA_UNREGISTERED:
sei = new SEIuserDataUnregistered;
xParseSEIuserDataUnregistered((SEIuserDataUnregistered&) *sei, payloadSize);
break;
case SEI::ACTIVE_PARAMETER_SETS:
sei = new SEIActiveParameterSets;
xParseSEIActiveParameterSets((SEIActiveParameterSets&) *sei, payloadSize);
break;
case SEI::DECODING_UNIT_INFO:
if (!sps)
{
printf ("Warning: Found Decoding unit SEI message, but no active SPS is available. Ignoring.");
}
else
{
sei = new SEIDecodingUnitInfo;
xParseSEIDecodingUnitInfo((SEIDecodingUnitInfo&) *sei, payloadSize, sps);
}
break;
case SEI::BUFFERING_PERIOD:
if (!sps)
{
printf ("Warning: Found Buffering period SEI message, but no active SPS is available. Ignoring.");
}
else
{
sei = new SEIBufferingPeriod;
xParseSEIBufferingPeriod((SEIBufferingPeriod&) *sei, payloadSize, sps);
}
break;
case SEI::PICTURE_TIMING:
if (!sps)
{
printf ("Warning: Found Picture timing SEI message, but no active SPS is available. Ignoring.");
}
else
{
sei = new SEIPictureTiming;
xParseSEIPictureTiming((SEIPictureTiming&)*sei, payloadSize, sps);
}
break;
case SEI::RECOVERY_POINT:
sei = new SEIRecoveryPoint;
xParseSEIRecoveryPoint((SEIRecoveryPoint&) *sei, payloadSize);
break;
case SEI::FRAME_PACKING:
sei = new SEIFramePacking;
xParseSEIFramePacking((SEIFramePacking&) *sei, payloadSize);
break;
case SEI::DISPLAY_ORIENTATION:
sei = new SEIDisplayOrientation;
xParseSEIDisplayOrientation((SEIDisplayOrientation&) *sei, payloadSize);
break;
case SEI::TEMPORAL_LEVEL0_INDEX:
sei = new SEITemporalLevel0Index;
xParseSEITemporalLevel0Index((SEITemporalLevel0Index&) *sei, payloadSize);
break;
case SEI::REGION_REFRESH_INFO:
sei = new SEIGradualDecodingRefreshInfo;
xParseSEIGradualDecodingRefreshInfo((SEIGradualDecodingRefreshInfo&) *sei, payloadSize);
break;
default:
for (UInt i = 0; i < payloadSize; i++)
{
UInt seiByte;
READ_CODE (8, seiByte, "unknown prefix SEI payload byte");
}
printf ("Unknown prefix SEI message (payloadType = %d) was found!\n", payloadType);
}
}
else
{
switch (payloadType)
{
#if L0363_SEI_ALLOW_SUFFIX
case SEI::USER_DATA_UNREGISTERED:
sei = new SEIuserDataUnregistered;
xParseSEIuserDataUnregistered((SEIuserDataUnregistered&) *sei, payloadSize);
break;
#endif
case SEI::DECODED_PICTURE_HASH:
sei = new SEIDecodedPictureHash;
xParseSEIDecodedPictureHash((SEIDecodedPictureHash&) *sei, payloadSize);
break;
default:
for (UInt i = 0; i < payloadSize; i++)
{
UInt seiByte;
READ_CODE (8, seiByte, "unknown suffix SEI payload byte");
}
printf ("Unknown suffix SEI message (payloadType = %d) was found!\n", payloadType);
}
}
if (sei != NULL)
{
seis.push_back(sei);
}
/* By definition the underlying bitstream terminates in a byte-aligned manner.
* 1. Extract all bar the last MIN(bitsremaining,nine) bits as reserved_payload_extension_data
* 2. Examine the final 8 bits to determine the payload_bit_equal_to_one marker
* 3. Extract the remainingreserved_payload_extension_data bits.
*
* If there are fewer than 9 bits available, extract them.
*/
Int payloadBitsRemaining = getBitstream()->getNumBitsLeft();
if (payloadBitsRemaining) /* more_data_in_payload() */
{
for (; payloadBitsRemaining > 9; payloadBitsRemaining--)
{
UInt reservedPayloadExtensionData;
READ_CODE (1, reservedPayloadExtensionData, "reserved_payload_extension_data");
}
/* 2 */
Int finalBits = getBitstream()->peekBits(payloadBitsRemaining);
Int finalPayloadBits = 0;
for (Int mask = 0xff; finalBits & (mask >> finalPayloadBits); finalPayloadBits++)
{
continue;
}
/* 3 */
for (; payloadBitsRemaining > 9 - finalPayloadBits; payloadBitsRemaining--)
{
UInt reservedPayloadExtensionData;
READ_CODE (1, reservedPayloadExtensionData, "reserved_payload_extension_data");
}
UInt dummy;
READ_CODE (1, dummy, "payload_bit_equal_to_one");
READ_CODE (payloadBitsRemaining-1, dummy, "payload_bit_equal_to_zero");
}
/* restore primary bitstream for sei_message */
delete getBitstream();
setBitstream(bs);
}