Newer
Older

Karsten Suehring
committed
/* 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-2020, ITU/ISO/IEC

Karsten Suehring
committed
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
* 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.
*/
/** \file DecApp.cpp
\brief Decoder application class
*/
#include <list>
#include <vector>
#include <stdio.h>
#include <fcntl.h>
#include "DecApp.h"
#include "DecoderLib/AnnexBread.h"
#include "DecoderLib/NALread.h"
#if RExt__DECODER_DEBUG_STATISTICS
#include "CommonLib/CodingStatistics.h"
#endif
#include "CommonLib/dtrace_codingstruct.h"
//! \ingroup DecoderApp
//! \{
// ====================================================================================================================
// Constructor / destructor / initialization / destroy
// ====================================================================================================================
DecApp::DecApp()
: m_iPOCLastDisplay(-MAX_INT)
{
}
// ====================================================================================================================
// Public member functions
// ====================================================================================================================
/**
- create internal class
- initialize internal class
- until the end of the bitstream, call decoding function in DecApp class
- delete allocated buffers
- destroy internal class
- returns the number of mismatching pictures
*/
uint32_t DecApp::decode()
{
int poc;
PicList* pcListPic = NULL;
ifstream bitstreamFile(m_bitstreamFileName.c_str(), ifstream::in | ifstream::binary);
if (!bitstreamFile)
{
EXIT( "Failed to open bitstream file " << m_bitstreamFileName.c_str() << " for reading" ) ;
}
InputByteStream bytestream(bitstreamFile);
if (!m_outputDecodedSEIMessagesFilename.empty() && m_outputDecodedSEIMessagesFilename!="-")
{
m_seiMessageFileStream.open(m_outputDecodedSEIMessagesFilename.c_str(), std::ios::out);
if (!m_seiMessageFileStream.is_open() || !m_seiMessageFileStream.good())
{
EXIT( "Unable to open file "<< m_outputDecodedSEIMessagesFilename.c_str() << " for writing decoded SEI messages");
}
}
Alexey Filippov
committed
#if JVET_P2008_OUTPUT_LOG
if (!m_oplFilename.empty() && m_oplFilename!="-")
{
m_oplFileStream.open(m_oplFilename.c_str(), std::ios::out);
if (!m_oplFileStream.is_open() || !m_oplFileStream.good())
{
EXIT( "Unable to open file "<< m_oplFilename.c_str() << " to write an opl-file for conformance testing (see JVET-P2008 for details)");
}
}
#endif //JVET_P2008_OUTPUT_LOG

Karsten Suehring
committed
// create & initialize internal classes
xCreateDecLib();
m_iPOCLastDisplay += m_iSkipFrame; // set the last displayed POC correctly for skip forward.
// clear contents of colour-remap-information-SEI output file
if (!m_colourRemapSEIFileName.empty())
{
std::ofstream ofile(m_colourRemapSEIFileName.c_str());
if (!ofile.good() || !ofile.is_open())
{
EXIT( "Unable to open file " << m_colourRemapSEIFileName.c_str() << " for writing colour-remap-information-SEI video");
}
}
// main decoder loop
#if JVET_P0125_EOS_LAYER_SPECIFIC
bool loopFiltered[MAX_VPS_LAYERS] = { false };

Karsten Suehring
committed
bool loopFiltered = false;

Karsten Suehring
committed
bool bPicSkipped = false;

Karsten Suehring
committed
while (!!bitstreamFile)
{
InputNALUnit nalu;
nalu.m_nalUnitType = NAL_UNIT_INVALID;
// determine if next NAL unit will be the first one from a new picture
bool bNewPicture = isNewPicture(&bitstreamFile, &bytestream);
bool bNewAccessUnit = bNewPicture && isNewAccessUnit( bNewPicture, &bitstreamFile, &bytestream );
if(!bNewPicture)
{
AnnexBStats stats = AnnexBStats();
// find next NAL unit in stream
byteStreamNALUnit(bytestream, nalu.getBitstream().getFifo(), stats);
if (nalu.getBitstream().getFifo().empty())
{
/* this can happen if the following occur:
* - empty input file
* - two back-to-back start_code_prefixes
* - start_code_prefix immediately followed by EOF
*/
msg( ERROR, "Warning: Attempt to decode an empty NAL unit\n");
}
else
{
// read NAL unit header
read(nalu);
// flush output for first slice of an IDR picture
if(m_cDecLib.getFirstSliceInPicture() &&
(nalu.m_nalUnitType == NAL_UNIT_CODED_SLICE_IDR_W_RADL ||
nalu.m_nalUnitType == NAL_UNIT_CODED_SLICE_IDR_N_LP))
{
xFlushOutput(pcListPic, nalu.m_nuhLayerId);
}
// parse NAL unit syntax if within target decoding layer
#if JVET_Q0814_DPB
if( ( m_iMaxTemporalLayer < 0 || nalu.m_temporalId <= m_iMaxTemporalLayer ) && xIsNaluWithinTargetDecLayerIdSet( &nalu ) )
#else
if ((m_iMaxTemporalLayer < 0 || nalu.m_temporalId <= m_iMaxTemporalLayer) && isNaluWithinTargetDecLayerIdSet(&nalu))
#if JVET_P0115_LAYER_TID_CONSTRAINT
CHECK(nalu.m_temporalId > m_iMaxTemporalLayer, "bitstream shall not include any NAL unit with TemporalId greater than HighestTid");
if (m_targetDecLayerIdSet.size())
{
CHECK(std::find(m_targetDecLayerIdSet.begin(), m_targetDecLayerIdSet.end(), nalu.m_nuhLayerId) == m_targetDecLayerIdSet.end(), "bitstream shall not contain any other layers than included in the OLS with OlsIdx");
if (bPicSkipped)
{
if ((nalu.m_nalUnitType == NAL_UNIT_CODED_SLICE_TRAIL) || (nalu.m_nalUnitType == NAL_UNIT_CODED_SLICE_STSA) || (nalu.m_nalUnitType == NAL_UNIT_CODED_SLICE_RASL) || (nalu.m_nalUnitType == NAL_UNIT_CODED_SLICE_RADL) || (nalu.m_nalUnitType == NAL_UNIT_CODED_SLICE_IDR_W_RADL) || (nalu.m_nalUnitType == NAL_UNIT_CODED_SLICE_IDR_N_LP) || (nalu.m_nalUnitType == NAL_UNIT_CODED_SLICE_CRA) || (nalu.m_nalUnitType == NAL_UNIT_CODED_SLICE_GDR))
{
if (m_cDecLib.isSliceNaluFirstInAU(true, nalu))
{
m_cDecLib.resetAccessUnitNals();
m_cDecLib.resetAccessUnitApsNals();
#if JVET_P0101_POC_MULTILAYER
m_cDecLib.resetAccessUnitPicInfo();
#endif
}
bPicSkipped = false;
}
}
m_cDecLib.decode(nalu, m_iSkipFrame, m_iPOCLastDisplay, m_targetOlsIdx);
m_cDecLib.decode(nalu, m_iSkipFrame, m_iPOCLastDisplay);
if (nalu.m_nalUnitType == NAL_UNIT_VPS)
{
#if JVET_Q0814_DPB
m_cDecLib.deriveTargetOutputLayerSet( m_targetOlsIdx );
m_targetDecLayerIdSet = m_cDecLib.getVPS()->m_targetLayerIdSet;
m_targetOutputLayerIdSet = m_cDecLib.getVPS()->m_targetOutputLayerIdSet;
#else
else
{
bPicSkipped = true;
}

Karsten Suehring
committed
#if JVET_P0125_EOS_LAYER_SPECIFIC
if ((bNewPicture || !bitstreamFile || nalu.m_nalUnitType == NAL_UNIT_EOS) && !m_cDecLib.getFirstSliceInSequence(nalu.m_nuhLayerId) && !bPicSkipped)
#else
if ((bNewPicture || !bitstreamFile || nalu.m_nalUnitType == NAL_UNIT_EOS) && !m_cDecLib.getFirstSliceInSequence() && !bPicSkipped)

Karsten Suehring
committed
{
#if JVET_P0125_EOS_LAYER_SPECIFIC
if (!loopFiltered[nalu.m_nuhLayerId] || bitstreamFile)
#else

Karsten Suehring
committed
if (!loopFiltered || bitstreamFile)

Karsten Suehring
committed
{
m_cDecLib.executeLoopFilters();
m_cDecLib.finishPicture( poc, pcListPic );
}
#if JVET_P0125_EOS_LAYER_SPECIFIC
loopFiltered[nalu.m_nuhLayerId] = (nalu.m_nalUnitType == NAL_UNIT_EOS);
#else

Karsten Suehring
committed
loopFiltered = (nalu.m_nalUnitType == NAL_UNIT_EOS);

Karsten Suehring
committed
if (nalu.m_nalUnitType == NAL_UNIT_EOS)
{
#if JVET_P0125_EOS_LAYER_SPECIFIC
m_cDecLib.setFirstSliceInSequence(true, nalu.m_nuhLayerId);
#else

Karsten Suehring
committed
m_cDecLib.setFirstSliceInSequence(true);

Karsten Suehring
committed
}
}
else if ( (bNewPicture || !bitstreamFile || nalu.m_nalUnitType == NAL_UNIT_EOS ) &&
#if JVET_P0125_EOS_LAYER_SPECIFIC
m_cDecLib.getFirstSliceInSequence(nalu.m_nuhLayerId))
#else

Karsten Suehring
committed
m_cDecLib.getFirstSliceInSequence () )

Karsten Suehring
committed
{
m_cDecLib.setFirstSliceInPicture (true);
}
if( pcListPic )
{
if( !m_reconFileName.empty() && !m_cVideoIOYuvReconFile[nalu.m_nuhLayerId].isOpen() )

Karsten Suehring
committed
{
const BitDepths &bitDepths=pcListPic->front()->cs->sps->getBitDepths(); // use bit depths of first reconstructed picture.
for( uint32_t channelType = 0; channelType < MAX_NUM_CHANNEL_TYPE; channelType++ )
{
if( m_outputBitDepth[channelType] == 0 )
{
m_outputBitDepth[channelType] = bitDepths.recon[channelType];
}
}
if (m_packedYUVMode && (m_outputBitDepth[CH_L] != 10 && m_outputBitDepth[CH_L] != 12))
{
EXIT ("Invalid output bit-depth for packed YUV output, aborting\n");
}
std::string reconFileName = m_reconFileName;
#if JVET_Q0814_DPB
if( m_reconFileName.compare( "/dev/null" ) && m_cDecLib.getVPS() != nullptr && m_cDecLib.getVPS()->getMaxLayers() > 1 && xIsNaluWithinTargetOutputLayerIdSet( &nalu ) )
#else
if (m_reconFileName.compare("/dev/null") && (m_cDecLib.getVPS() != nullptr) && (m_cDecLib.getVPS()->getMaxLayers() > 1) && (isNaluWithinTargetOutputLayerIdSet(&nalu)))
size_t pos = reconFileName.find_last_of('.');
if (pos != string::npos)
reconFileName.insert( pos, std::to_string( nalu.m_nuhLayerId ) );
}
else
{
reconFileName.append( std::to_string( nalu.m_nuhLayerId ) );
#if JVET_Q0814_DPB
if( ( m_cDecLib.getVPS() != nullptr && ( m_cDecLib.getVPS()->getMaxLayers() == 1 || xIsNaluWithinTargetOutputLayerIdSet( &nalu ) ) ) || m_cDecLib.getVPS() == nullptr )
{
m_cVideoIOYuvReconFile[nalu.m_nuhLayerId].open( reconFileName, true, m_outputBitDepth, m_outputBitDepth, bitDepths.recon ); // write mode
}
#else
Remy Foray
committed
if(((m_cDecLib.getVPS() != nullptr) &&
((m_cDecLib.getVPS()->getMaxLayers() == 1) || (isNaluWithinTargetOutputLayerIdSet(&nalu)))) ||
(m_cDecLib.getVPS() == nullptr))
m_cVideoIOYuvReconFile[nalu.m_nuhLayerId].open(reconFileName, true, m_outputBitDepth, m_outputBitDepth, bitDepths.recon); // write mode

Karsten Suehring
committed
}
// write reconstruction to file
if( bNewPicture )
{
xWriteOutput( pcListPic, nalu.m_temporalId );
}
if (nalu.m_nalUnitType == NAL_UNIT_EOS)
{
xWriteOutput( pcListPic, nalu.m_temporalId );
m_cDecLib.setFirstSliceInPicture (false);
}
// write reconstruction to file -- for additional bumping as defined in C.5.2.3
if (!bNewPicture && ((nalu.m_nalUnitType >= NAL_UNIT_CODED_SLICE_TRAIL && nalu.m_nalUnitType <= NAL_UNIT_RESERVED_IRAP_VCL_12)
|| (nalu.m_nalUnitType >= NAL_UNIT_CODED_SLICE_IDR_W_RADL && nalu.m_nalUnitType <= NAL_UNIT_CODED_SLICE_GDR)))

Karsten Suehring
committed
{
xWriteOutput( pcListPic, nalu.m_temporalId );
}
}
#if JVET_P0125_ASPECT_TID_LAYER_ID_NUH
m_cDecLib.checkTidLayerIdInAccessUnit();
m_cDecLib.resetAccessUnitSeiTids();
#endif
m_cDecLib.resetAccessUnitNals();
m_cDecLib.resetAccessUnitApsNals();
#if JVET_P0101_POC_MULTILAYER
m_cDecLib.resetAccessUnitPicInfo();
#endif

Karsten Suehring
committed
}
xFlushOutput( pcListPic );
// get the number of checksum errors
uint32_t nRet = m_cDecLib.getNumberOfChecksumErrorsDetected();
// delete buffers
m_cDecLib.deletePicBuffer();
// destroy internal classes
xDestroyDecLib();
#if RExt__DECODER_DEBUG_STATISTICS
CodingStatistics::DestroyInstance();
#endif
destroyROM();
return nRet;
}
bool DecApp::deriveOutputLayerSet()
{
int vps_max_layers_minus1 = m_cDecLib.getVPS()->getMaxLayers() - 1;
if(m_targetOlsIdx == - 1 || vps_max_layers_minus1 == 0)
{
m_targetDecLayerIdSet.clear();
return true;
}
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
int TotalNumOlss = 0;
int each_layer_is_an_ols_flag = m_cDecLib.getVPS()->getEachLayerIsAnOlsFlag();
int ols_mode_idc = m_cDecLib.getVPS()->getOlsModeIdc();
int num_output_layer_sets_minus1 = m_cDecLib.getVPS()->getNumOutputLayerSets() - 1;
int i = 0, j = 0, k = 0, r = 0;
int* NumOutputLayersInOls;
int* NumLayersInOls;
int** OutputLayerIdInOls;
int** OutputLayerIdx;
int** layerIncludedInOlsFlag;
int** LayerIdInOls;
int** dependencyFlag;
int** RefLayerIdx;
int* NumRefLayers;
if (vps_max_layers_minus1 == 0)
TotalNumOlss = 1;
else if (each_layer_is_an_ols_flag || ols_mode_idc == 0 || ols_mode_idc == 1)
TotalNumOlss = vps_max_layers_minus1 + 1;
else if (ols_mode_idc == 2)
TotalNumOlss = num_output_layer_sets_minus1 + 1;
NumOutputLayersInOls = new int[m_cDecLib.getVPS()->getNumOutputLayerSets()];
NumLayersInOls = new int[m_cDecLib.getVPS()->getNumOutputLayerSets()];
OutputLayerIdInOls = new int*[TotalNumOlss];
OutputLayerIdx = new int*[TotalNumOlss];
layerIncludedInOlsFlag = new int*[TotalNumOlss];
LayerIdInOls = new int*[TotalNumOlss];
for (i = 0; i < TotalNumOlss; i++)
{
OutputLayerIdInOls[i] = new int[vps_max_layers_minus1 + 1];
OutputLayerIdx[i] = new int[vps_max_layers_minus1 + 1];
layerIncludedInOlsFlag[i] = new int[vps_max_layers_minus1 + 1];
LayerIdInOls[i] = new int[vps_max_layers_minus1 + 1];
}
dependencyFlag = new int*[vps_max_layers_minus1 + 1];
RefLayerIdx = new int*[vps_max_layers_minus1 + 1];
NumRefLayers = new int[vps_max_layers_minus1 + 1];
for (i = 0; i <= vps_max_layers_minus1; i++)
{
dependencyFlag[i] = new int[vps_max_layers_minus1 + 1];
RefLayerIdx[i] = new int[vps_max_layers_minus1 + 1];
}
for (i = 0; i <= vps_max_layers_minus1; i++) {
for (j = 0; j <= vps_max_layers_minus1; j++) {
dependencyFlag[i][j] = m_cDecLib.getVPS()->getDirectRefLayerFlag(i, j);
for (k = 0; k < i; k++)
if (m_cDecLib.getVPS()->getDirectRefLayerFlag(i, k) && dependencyFlag[k][j])
dependencyFlag[i][j] = 1;
}
}
for (i = 0; i <= vps_max_layers_minus1; i++)
{
for (j = 0, r = 0; j <= vps_max_layers_minus1; j++)
{
if (dependencyFlag[i][j])
RefLayerIdx[i][r++] = j;
}
NumRefLayers[i] = r;
}
NumOutputLayersInOls[0] = 1;
OutputLayerIdInOls[0][0] = m_cDecLib.getVPS()->getLayerId(0);
for (i = 1; i < TotalNumOlss; i++)
{
if (each_layer_is_an_ols_flag || ols_mode_idc == 0)
{
NumOutputLayersInOls[i] = 1;
OutputLayerIdInOls[i][0] = m_cDecLib.getVPS()->getLayerId(i);
}
else if (ols_mode_idc == 1) {
NumOutputLayersInOls[i] = i + 1;
for (j = 0; j < NumOutputLayersInOls[i]; j++)
OutputLayerIdInOls[i][j] = m_cDecLib.getVPS()->getLayerId(j);
}
else if (ols_mode_idc == 2) {
for (j = 0; j <= vps_max_layers_minus1; j++)
{
layerIncludedInOlsFlag[i][j] = 0;
}
for (k = 0, j = 0; k <= vps_max_layers_minus1; k++)
{
if (m_cDecLib.getVPS()->getOlsOutputLayerFlag(i, k))
{
layerIncludedInOlsFlag[i][k] = 1;
OutputLayerIdx[i][j] = k;
OutputLayerIdInOls[i][j++] = m_cDecLib.getVPS()->getLayerId(k);
}
}
NumOutputLayersInOls[i] = j;
for (j = 0; j < NumOutputLayersInOls[i]; j++)
{
int idx = OutputLayerIdx[i][j];
for (k = 0; k < NumRefLayers[idx]; k++)
layerIncludedInOlsFlag[i][RefLayerIdx[idx][k]] = 1;
}
}
}
m_targetOutputLayerIdSet.clear();
for (i = 0; i < NumOutputLayersInOls[m_targetOlsIdx]; i++)
m_targetOutputLayerIdSet.push_back(OutputLayerIdInOls[m_targetOlsIdx][i]);
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
NumLayersInOls[0] = 1;
LayerIdInOls[0][0] = m_cDecLib.getVPS()->getLayerId(0);
for (i = 1; i < TotalNumOlss; i++)
{
if (each_layer_is_an_ols_flag)
{
NumLayersInOls[i] = 1;
LayerIdInOls[i][0] = m_cDecLib.getVPS()->getLayerId(i);
}
else if (ols_mode_idc == 0 || ols_mode_idc == 1)
{
NumLayersInOls[i] = i + 1;
for (j = 0; j < NumLayersInOls[i]; j++)
LayerIdInOls[i][j] = m_cDecLib.getVPS()->getLayerId(j);
}
else if (ols_mode_idc == 2)
{
for (k = 0, j = 0; k <= vps_max_layers_minus1; k++)
if (layerIncludedInOlsFlag[i][k])
LayerIdInOls[i][j++] = m_cDecLib.getVPS()->getLayerId(k);
NumLayersInOls[i] = j;
}
}
m_targetDecLayerIdSet.clear();
for (i = 0; i < NumLayersInOls[m_targetOlsIdx]; i++)
m_targetDecLayerIdSet.push_back(LayerIdInOls[m_targetOlsIdx][i]);
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
delete[] NumOutputLayersInOls;
delete[] NumLayersInOls;
delete[] NumRefLayers;
for (i = 0; i < TotalNumOlss; i++)
{
delete[] OutputLayerIdInOls[i];
delete[] OutputLayerIdx[i];
delete[] layerIncludedInOlsFlag[i];
delete[] LayerIdInOls[i];
}
delete[] OutputLayerIdInOls;
delete[] OutputLayerIdx;
delete[] layerIncludedInOlsFlag;
delete[] LayerIdInOls;
for (i = 0; i <= vps_max_layers_minus1; i++)
{
delete[] dependencyFlag[i];
delete[] RefLayerIdx[i];
}
delete[] dependencyFlag;
delete[] RefLayerIdx;
return true;
}
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
/**
- lookahead through next NAL units to determine if current NAL unit is the first NAL unit in a new picture
*/
bool DecApp::isNewPicture(ifstream *bitstreamFile, class InputByteStream *bytestream)
{
bool ret = false;
bool finished = false;
// cannot be a new picture if there haven't been any slices yet
if(m_cDecLib.getFirstSliceInPicture())
{
return false;
}
// save stream position for backup
#if RExt__DECODER_DEBUG_STATISTICS
CodingStatistics::CodingStatisticsData* backupStats = new CodingStatistics::CodingStatisticsData(CodingStatistics::GetStatistics());
streampos location = bitstreamFile->tellg() - streampos(bytestream->GetNumBufferedBytes());
#else
streampos location = bitstreamFile->tellg();
#endif
// look ahead until picture start location is determined
while (!finished && !!(*bitstreamFile))
{
AnnexBStats stats = AnnexBStats();
InputNALUnit nalu;
byteStreamNALUnit(*bytestream, nalu.getBitstream().getFifo(), stats);
if (nalu.getBitstream().getFifo().empty())
{
msg( ERROR, "Warning: Attempt to decode an empty NAL unit\n");
}
else
{
// get next NAL unit type
read(nalu);
switch( nalu.m_nalUnitType ) {
// NUT that indicate the start of a new picture
case NAL_UNIT_ACCESS_UNIT_DELIMITER:
#if JVET_Q0117_PARAMETER_SETS_CLEANUP
case NAL_UNIT_DCI:
#else
case NAL_UNIT_VPS:
case NAL_UNIT_SPS:
case NAL_UNIT_PPS:
case NAL_UNIT_PH:
ret = true;
finished = true;
break;
#if JVET_Q0775_PH_IN_SH
// NUT that may be the start of a new picture - check first bit in slice header
#else
#endif
case NAL_UNIT_CODED_SLICE_TRAIL:
case NAL_UNIT_CODED_SLICE_STSA:
case NAL_UNIT_CODED_SLICE_RASL:
case NAL_UNIT_CODED_SLICE_RADL:
case NAL_UNIT_RESERVED_VCL_4:
case NAL_UNIT_RESERVED_VCL_5:
case NAL_UNIT_RESERVED_VCL_6:
case NAL_UNIT_CODED_SLICE_IDR_W_RADL:
case NAL_UNIT_CODED_SLICE_IDR_N_LP:
case NAL_UNIT_CODED_SLICE_CRA:
case NAL_UNIT_CODED_SLICE_GDR:
case NAL_UNIT_RESERVED_IRAP_VCL_11:
case NAL_UNIT_RESERVED_IRAP_VCL_12:
#if JVET_Q0775_PH_IN_SH
ret = checkPictureHeaderInSliceHeaderFlag(nalu);
finished = true;
break;
// NUT that are not the start of a new picture
#endif
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
case NAL_UNIT_EOS:
case NAL_UNIT_EOB:
case NAL_UNIT_SUFFIX_APS:
case NAL_UNIT_SUFFIX_SEI:
case NAL_UNIT_FD:
ret = false;
finished = true;
break;
// NUT that might indicate the start of a new picture - keep looking
case NAL_UNIT_PREFIX_APS:
case NAL_UNIT_PREFIX_SEI:
case NAL_UNIT_RESERVED_NVCL_26:
case NAL_UNIT_RESERVED_NVCL_27:
case NAL_UNIT_UNSPECIFIED_28:
case NAL_UNIT_UNSPECIFIED_29:
case NAL_UNIT_UNSPECIFIED_30:
case NAL_UNIT_UNSPECIFIED_31:
default:
break;
}
}
}
// restore previous stream location - minus 3 due to the need for the annexB parser to read three extra bytes
#if RExt__DECODER_DEBUG_BIT_STATISTICS
bitstreamFile->clear();
bitstreamFile->seekg(location);
bytestream->reset();
CodingStatistics::SetStatistics(*backupStats);
delete backupStats;
#else
bitstreamFile->clear();
bitstreamFile->seekg(location-streamoff(3));
bytestream->reset();
#endif
// return TRUE if next NAL unit is the start of a new picture
return ret;
}
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
/**
- lookahead through next NAL units to determine if current NAL unit is the first NAL unit in a new access unit
*/
bool DecApp::isNewAccessUnit( bool newPicture, ifstream *bitstreamFile, class InputByteStream *bytestream )
{
bool ret = false;
bool finished = false;
// can only be the start of an AU if this is the start of a new picture
if( newPicture == false )
{
return false;
}
// save stream position for backup
#if RExt__DECODER_DEBUG_STATISTICS
CodingStatistics::CodingStatisticsData* backupStats = new CodingStatistics::CodingStatisticsData(CodingStatistics::GetStatistics());
streampos location = bitstreamFile->tellg() - streampos(bytestream->GetNumBufferedBytes());
#else
streampos location = bitstreamFile->tellg();
#endif
// look ahead until access unit start location is determined
while (!finished && !!(*bitstreamFile))
{
AnnexBStats stats = AnnexBStats();
InputNALUnit nalu;
byteStreamNALUnit(*bytestream, nalu.getBitstream().getFifo(), stats);
if (nalu.getBitstream().getFifo().empty())
{
msg( ERROR, "Warning: Attempt to decode an empty NAL unit\n");
}
else
{
// get next NAL unit type
read(nalu);
switch( nalu.m_nalUnitType ) {
// AUD always indicates the start of a new access unit
case NAL_UNIT_ACCESS_UNIT_DELIMITER:
ret = true;
finished = true;
break;
// slice types - check layer ID and POC
case NAL_UNIT_CODED_SLICE_TRAIL:
case NAL_UNIT_CODED_SLICE_STSA:
case NAL_UNIT_CODED_SLICE_RASL:
case NAL_UNIT_CODED_SLICE_RADL:
case NAL_UNIT_CODED_SLICE_IDR_W_RADL:
case NAL_UNIT_CODED_SLICE_IDR_N_LP:
case NAL_UNIT_CODED_SLICE_CRA:
case NAL_UNIT_CODED_SLICE_GDR:
ret = m_cDecLib.isSliceNaluFirstInAU( newPicture, nalu );
finished = true;
break;
// NUT that are not the start of a new access unit
case NAL_UNIT_EOS:
case NAL_UNIT_EOB:
case NAL_UNIT_SUFFIX_APS:
case NAL_UNIT_SUFFIX_SEI:
case NAL_UNIT_FD:
ret = false;
finished = true;
break;
// all other NUT - keep looking to find first VCL
default:
break;
}
}
}
// restore previous stream location
#if RExt__DECODER_DEBUG_BIT_STATISTICS
bitstreamFile->clear();
bitstreamFile->seekg(location);
bytestream->reset();
CodingStatistics::SetStatistics(*backupStats);
delete backupStats;
#else
bitstreamFile->clear();
bitstreamFile->seekg(location);
bytestream->reset();
#endif
// return TRUE if next NAL unit is the start of a new picture
return ret;
}
Alexey Filippov
committed
#if JVET_P2008_OUTPUT_LOG
void DecApp::writeLineToOutputLog(Picture * pcPic)
{
if (m_oplFileStream.is_open() && m_oplFileStream.good())
{
const SPS* sps = pcPic->cs->sps;
PictureHash recon_digest;
auto numChar = calcMD5(((const Picture*)pcPic)->getRecoBuf(), recon_digest, sps->getBitDepths());
m_oplFileStream << std::setw(8) << pcPic->getPOC() << "," << std::setw(5) << pcPic->Y().width << "," << std::setw(5) << pcPic->Y().height << "," << hashToString(recon_digest, numChar) << "\n";
}
}
#endif // JVET_P2008_OUTPUT_LOG

Karsten Suehring
committed
// ====================================================================================================================
// Protected member functions
// ====================================================================================================================
void DecApp::xCreateDecLib()
{
initROM();
// create decoder class
m_cDecLib.create();
// initialize decoder class
m_cDecLib.init(

Karsten Suehring
committed
#if JVET_J0090_MEMORY_BANDWITH_MEASURE
m_cacheCfgFile

Karsten Suehring
committed
#endif
);
m_cDecLib.setDecodedPictureHashSEIEnabled(m_decodedPictureHashSEIEnabled);

Karsten Suehring
committed
if (!m_outputDecodedSEIMessagesFilename.empty())
{
std::ostream &os=m_seiMessageFileStream.is_open() ? m_seiMessageFileStream : std::cout;
m_cDecLib.setDecodedSEIMessageOutputStream(&os);
}
m_cDecLib.m_targetSubPicIdx = this->m_targetSubPicIdx;
#endif
m_cDecLib.initScalingList();

Karsten Suehring
committed
}
void DecApp::xDestroyDecLib()
{
if( !m_reconFileName.empty() )
{
for( auto & recFile : m_cVideoIOYuvReconFile )
{
recFile.second.close();
}
}

Karsten Suehring
committed
// destroy decoder class
m_cDecLib.destroy();
}
/** \param pcListPic list of pictures to be written to file
\param tId temporal sub-layer ID
*/
void DecApp::xWriteOutput( PicList* pcListPic, uint32_t tId )
{
if (pcListPic->empty())
{
return;
}
PicList::iterator iterPic = pcListPic->begin();
int numPicsNotYetDisplayed = 0;
int dpbFullness = 0;
const SPS* activeSPS = (pcListPic->front()->cs->sps);
uint32_t numReorderPicsHighestTid;
uint32_t maxDecPicBufferingHighestTid;
uint32_t maxNrSublayers = activeSPS->getMaxTLayers();
#if JVET_Q0814_DPB
const VPS* referredVPS = pcListPic->front()->cs->vps;
const int temporalId = ( m_iMaxTemporalLayer == -1 || m_iMaxTemporalLayer >= maxNrSublayers ) ? maxNrSublayers - 1 : m_iMaxTemporalLayer;
if( referredVPS == nullptr || referredVPS->m_numLayersInOls[referredVPS->m_targetOlsIdx] == 1 )
{
numReorderPicsHighestTid = activeSPS->getNumReorderPics( temporalId );
maxDecPicBufferingHighestTid = activeSPS->getMaxDecPicBuffering( temporalId );
}
else
{
numReorderPicsHighestTid = referredVPS->getNumReorderPics( temporalId );
maxDecPicBufferingHighestTid = referredVPS->getMaxDecPicBuffering( temporalId );
}
#else

Karsten Suehring
committed
if(m_iMaxTemporalLayer == -1 || m_iMaxTemporalLayer >= maxNrSublayers)
{
numReorderPicsHighestTid = activeSPS->getNumReorderPics(maxNrSublayers-1);
maxDecPicBufferingHighestTid = activeSPS->getMaxDecPicBuffering(maxNrSublayers-1);
}
else
{
numReorderPicsHighestTid = activeSPS->getNumReorderPics(m_iMaxTemporalLayer);
maxDecPicBufferingHighestTid = activeSPS->getMaxDecPicBuffering(m_iMaxTemporalLayer);
}

Karsten Suehring
committed
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
while (iterPic != pcListPic->end())
{
Picture* pcPic = *(iterPic);
if(pcPic->neededForOutput && pcPic->getPOC() > m_iPOCLastDisplay)
{
numPicsNotYetDisplayed++;
dpbFullness++;
}
else if(pcPic->referenced)
{
dpbFullness++;
}
iterPic++;
}
iterPic = pcListPic->begin();
if (numPicsNotYetDisplayed>2)
{
iterPic++;
}
Picture* pcPic = *(iterPic);
if( numPicsNotYetDisplayed>2 && pcPic->fieldPic ) //Field Decoding
{
PicList::iterator endPic = pcListPic->end();
endPic--;
iterPic = pcListPic->begin();
while (iterPic != endPic)
{
Picture* pcPicTop = *(iterPic);
iterPic++;
Picture* pcPicBottom = *(iterPic);
if ( pcPicTop->neededForOutput && pcPicBottom->neededForOutput &&
(numPicsNotYetDisplayed > numReorderPicsHighestTid || dpbFullness > maxDecPicBufferingHighestTid) &&
(!(pcPicTop->getPOC()%2) && pcPicBottom->getPOC() == pcPicTop->getPOC()+1) &&
(pcPicTop->getPOC() == m_iPOCLastDisplay+1 || m_iPOCLastDisplay < 0))
{
// write to file
numPicsNotYetDisplayed = numPicsNotYetDisplayed-2;
if ( !m_reconFileName.empty() )
{
const Window &conf = pcPicTop->cs->pps->getConformanceWindow();

Karsten Suehring
committed
const bool isTff = pcPicTop->topField;
bool display = true;
if (display)
{
m_cVideoIOYuvReconFile[pcPicTop->layerId].write( pcPicTop->getRecoBuf(), pcPicBottom->getRecoBuf(),
m_outputColourSpaceConvert,
false, // TODO: m_packedYUVMode,
conf.getWindowLeftOffset() * SPS::getWinUnitX( pcPicTop->cs->sps->getChromaFormatIdc() ),
conf.getWindowRightOffset() * SPS::getWinUnitX( pcPicTop->cs->sps->getChromaFormatIdc() ),
conf.getWindowTopOffset() * SPS::getWinUnitY( pcPicTop->cs->sps->getChromaFormatIdc() ),
conf.getWindowBottomOffset() * SPS::getWinUnitY( pcPicTop->cs->sps->getChromaFormatIdc() ),
NUM_CHROMA_FORMAT, isTff );

Karsten Suehring
committed
}
}
Alexey Filippov
committed
#if JVET_P2008_OUTPUT_LOG
writeLineToOutputLog(pcPicTop);
writeLineToOutputLog(pcPicBottom);
#endif

Karsten Suehring
committed
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
// update POC of display order
m_iPOCLastDisplay = pcPicBottom->getPOC();
// erase non-referenced picture in the reference picture list after display
if ( ! pcPicTop->referenced && pcPicTop->reconstructed )
{
pcPicTop->reconstructed = false;
}
if ( ! pcPicBottom->referenced && pcPicBottom->reconstructed )
{
pcPicBottom->reconstructed = false;
}
pcPicTop->neededForOutput = false;
pcPicBottom->neededForOutput = false;
}
}
}
else if( !pcPic->fieldPic ) //Frame Decoding
{
iterPic = pcListPic->begin();
while (iterPic != pcListPic->end())
{
pcPic = *(iterPic);
if(pcPic->neededForOutput && pcPic->getPOC() > m_iPOCLastDisplay &&
(numPicsNotYetDisplayed > numReorderPicsHighestTid || dpbFullness > maxDecPicBufferingHighestTid))
{
// write to file
numPicsNotYetDisplayed--;
if (!pcPic->referenced)
{
dpbFullness--;
}
if (!m_reconFileName.empty())
{
const Window &conf = pcPic->getConformanceWindow();
const SPS* sps = pcPic->cs->sps;
ChromaFormat chromaFormatIDC = sps->getChromaFormatIdc();
m_cVideoIOYuvReconFile[pcPic->layerId].writeUpscaledPicture( *sps, *pcPic->cs->pps, pcPic->getRecoBuf(), m_outputColourSpaceConvert, m_packedYUVMode, m_upscaledOutput, NUM_CHROMA_FORMAT, m_bClipOutputVideoToRec709Range );
{
m_cVideoIOYuvReconFile[pcPic->layerId].write( pcPic->getRecoBuf().get( COMPONENT_Y ).width, pcPic->getRecoBuf().get( COMPONENT_Y ).height, pcPic->getRecoBuf(),

Karsten Suehring
committed
m_outputColourSpaceConvert,
m_packedYUVMode,
conf.getWindowLeftOffset() * SPS::getWinUnitX( chromaFormatIDC ),
conf.getWindowRightOffset() * SPS::getWinUnitX( chromaFormatIDC ),
conf.getWindowTopOffset() * SPS::getWinUnitY( chromaFormatIDC ),
conf.getWindowBottomOffset() * SPS::getWinUnitY( chromaFormatIDC ),

Karsten Suehring
committed
NUM_CHROMA_FORMAT, m_bClipOutputVideoToRec709Range );

Karsten Suehring
committed
}
Alexey Filippov
committed
#if JVET_P2008_OUTPUT_LOG
writeLineToOutputLog(pcPic);
#endif

Karsten Suehring
committed
// update POC of display order
m_iPOCLastDisplay = pcPic->getPOC();
// erase non-referenced picture in the reference picture list after display
if (!pcPic->referenced && pcPic->reconstructed)
{
pcPic->reconstructed = false;
}
pcPic->neededForOutput = false;
}
iterPic++;
}
}
}
/** \param pcListPic list of pictures to be written to file
*/
void DecApp::xFlushOutput( PicList* pcListPic, const int layerId )

Karsten Suehring
committed
{
if(!pcListPic || pcListPic->empty())
{
return;
}
PicList::iterator iterPic = pcListPic->begin();
iterPic = pcListPic->begin();
Picture* pcPic = *(iterPic);