Skip to content
Snippets Groups Projects
DecLib.cpp 45.3 KiB
Newer Older
  • Learn to ignore specific revisions
  • /* 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-2019, 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.
     */
    
    /** \file     DecLib.cpp
        \brief    decoder class
    */
    
    #include "NALread.h"
    #include "DecLib.h"
    
    #include "CommonLib/dtrace_next.h"
    #include "CommonLib/dtrace_buffer.h"
    #include "CommonLib/Buffer.h"
    #include "CommonLib/UnitTools.h"
    
    #include <fstream>
    #include <stdio.h>
    #include <fcntl.h>
    #include "AnnexBread.h"
    #include "NALread.h"
    
    #if K0149_BLOCK_STATISTICS
    #include "CommonLib/dtrace_blockstatistics.h"
    #endif
    
    
    #if RExt__DECODER_DEBUG_TOOL_STATISTICS
    #include "CommonLib/CodingStatistics.h"
    #endif
    
    bool tryDecodePicture( Picture* pcEncPic, const int expectedPoc, const std::string& bitstreamFileName, bool bDecodeUntilPocFound /* = false */ )
    {
      int      poc;
      PicList* pcListPic = NULL;
    
      static bool bFirstCall      = true;             /* TODO: MT */
      static bool loopFiltered    = false;            /* TODO: MT */
      static int  iPOCLastDisplay = -MAX_INT;         /* TODO: MT */
    
      static std::ifstream* bitstreamFile = nullptr;  /* TODO: MT */
      static InputByteStream* bytestream  = nullptr;  /* TODO: MT */
      bool bRet = false;
    
      // create & initialize internal classes
      static DecLib *pcDecLib = nullptr;              /* TODO: MT */
    
      if( pcEncPic )
      {
        if( bFirstCall )
        {
          bitstreamFile = new std::ifstream( bitstreamFileName.c_str(), std::ifstream::in | std::ifstream::binary );
          bytestream    = new InputByteStream( *bitstreamFile );
    
          CHECK( !*bitstreamFile, "failed to open bitstream file " << bitstreamFileName.c_str() << " for reading" ) ;
          // create decoder class
          pcDecLib = new DecLib;
          pcDecLib->create();
    
          // initialize decoder class
          pcDecLib->init(
    #if  JVET_J0090_MEMORY_BANDWITH_MEASURE
            ""
    #endif
          );
    
          pcDecLib->setDecodedPictureHashSEIEnabled( true );
    
          bFirstCall = false;
          msg( INFO, "start to decode %s \n", bitstreamFileName.c_str() );
        }
    
        bool goOn = true;
    
        // main decoder loop
        while( !!*bitstreamFile && goOn )
        {
          /* location serves to work around a design fault in the decoder, whereby
           * the process of reading a new slice that is the first slice of a new frame
           * requires the DecApp::decode() method to be called again with the same
           * nal unit. */
          std::streampos location = bitstreamFile->tellg();
          AnnexBStats stats       = AnnexBStats();
    
          InputNALUnit nalu;
          byteStreamNALUnit( *bytestream, nalu.getBitstream().getFifo(), stats );
    
          // call actual decoding function
          bool bNewPicture = false;
          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( nalu );
            int iSkipFrame = 0;
            bNewPicture = pcDecLib->decode( nalu, iSkipFrame, iPOCLastDisplay );
            if( bNewPicture )
            {
              bitstreamFile->clear();
              /* location points to the current nalunit payload[1] due to the
                * need for the annexB parser to read three extra bytes.
                * [1] except for the first NAL unit in the file
                *     (but bNewPicture doesn't happen then) */
              bitstreamFile->seekg( location - std::streamoff( 3 ) );
              bytestream->reset();
            }
          }
    
          if( ( bNewPicture || !*bitstreamFile || nalu.m_nalUnitType == NAL_UNIT_EOS ) && !pcDecLib->getFirstSliceInSequence() )
          {
            if( !loopFiltered || *bitstreamFile )
            {
              pcDecLib->finishPictureLight( poc, pcListPic );
    
              if( pcListPic )
              {
                for( auto & pic : *pcListPic )
                {
                  if( pic->poc == poc && (!bDecodeUntilPocFound || expectedPoc == poc ) )
                  {
                    CHECK( pcEncPic->slices.size() == 0, "at least one slice should be available" );
    
                    CHECK( expectedPoc != poc, "mismatch in POC - check encoder configuration" );
    
                    for( int i = 0; i < pic->slices.size(); i++ )
                    {
                      if( pcEncPic->slices.size() <= i )
                      {
                        pcEncPic->slices.push_back( new Slice );
                        pcEncPic->slices.back()->initSlice();
                      }
                      pcEncPic->slices[i]->copySliceInfo( pic->slices[i], false );
                    }
    
                    pcEncPic->cs->slice = pcEncPic->slices.back();
    
                    if ( pic->cs->sps->getUseSAO() )
                    {
                      pcEncPic->copySAO( *pic, 0 );
                    }
    
                    if( pic->cs->sps->getUseALF() )
                    {
                      for( int compIdx = 0; compIdx < MAX_NUM_COMPONENT; compIdx++ )
                      {
                        std::copy( pic->getAlfCtuEnableFlag()[compIdx].begin(), pic->getAlfCtuEnableFlag()[compIdx].end(), pcEncPic->getAlfCtuEnableFlag()[compIdx].begin() );
                      }
    
                      for( int i = 0; i < pic->slices.size(); i++ )
                      {
                        pcEncPic->slices[i]->getAlfSliceParam() = pic->slices[i]->getAlfSliceParam();
                      }
                    }
    
    
                    pcDecLib->executeLoopFilters();
                    if ( pic->cs->sps->getUseSAO() )
                    {
                      pcEncPic->copySAO( *pic, 1 );
                    }
    
    
    195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 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 472 473 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 500 501 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 529 530 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 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 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 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 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767
                    pcEncPic->cs->copyStructure( *pic->cs, CH_L, true, true );
    
                    if( CS::isDualITree( *pcEncPic->cs ) )
                    {
                      pcEncPic->cs->copyStructure( *pic->cs, CH_C, true, true );
                    }
                    goOn = false; // exit the loop return
                    bRet = true;
                    break;
                  }
                }
              }
              // postpone loop filters
              if (!bRet)
              {
                pcDecLib->executeLoopFilters();
              }
    
              pcDecLib->finishPicture( poc, pcListPic, DETAILS );
    
              // write output
              if( ! pcListPic->empty())
              {
                PicList::iterator iterPic   = pcListPic->begin();
                int numPicsNotYetDisplayed = 0;
                int dpbFullness = 0;
                const SPS* activeSPS = (pcListPic->front()->cs->sps);
                uint32_t maxNrSublayers = activeSPS->getMaxTLayers();
                uint32_t numReorderPicsHighestTid = activeSPS->getNumReorderPics(maxNrSublayers-1);
                uint32_t maxDecPicBufferingHighestTid =  activeSPS->getMaxDecPicBuffering(maxNrSublayers-1);
    
                while (iterPic != pcListPic->end())
                {
                  Picture* pcCurPic = *(iterPic);
                  if(pcCurPic->neededForOutput && pcCurPic->getPOC() > iPOCLastDisplay)
                  {
                    numPicsNotYetDisplayed++;
                    dpbFullness++;
                  }
                  else if(pcCurPic->referenced)
                  {
                    dpbFullness++;
                  }
                  iterPic++;
                }
    
                iterPic = pcListPic->begin();
    
                if (numPicsNotYetDisplayed>2)
                {
                  iterPic++;
                }
    
                Picture* pcCurPic = *(iterPic);
                if( numPicsNotYetDisplayed>2 && pcCurPic->fieldPic ) //Field Decoding
                {
                  THROW( "no field coding support ");
                }
                else if( !pcCurPic->fieldPic ) //Frame Decoding
                {
                  iterPic = pcListPic->begin();
    
                  while (iterPic != pcListPic->end())
                  {
                    pcCurPic = *(iterPic);
    
                    if(pcCurPic->neededForOutput && pcCurPic->getPOC() > iPOCLastDisplay &&
                      (numPicsNotYetDisplayed >  numReorderPicsHighestTid || dpbFullness > maxDecPicBufferingHighestTid))
                    {
                        numPicsNotYetDisplayed--;
                      if( ! pcCurPic->referenced )
                      {
                        dpbFullness--;
                      }
                      // update POC of display order
                      iPOCLastDisplay = pcCurPic->getPOC();
    
                      // erase non-referenced picture in the reference picture list after display
                      if( ! pcCurPic->referenced && pcCurPic->reconstructed )
                      {
                        pcCurPic->reconstructed = false;
                      }
                      pcCurPic->neededForOutput = false;
                    }
    
                    iterPic++;
                  }
                }
              }
    
    
            }
            loopFiltered = ( nalu.m_nalUnitType == NAL_UNIT_EOS );
            if( nalu.m_nalUnitType == NAL_UNIT_EOS )
            {
              pcDecLib->setFirstSliceInSequence( true );
            }
    
          }
          else if( ( bNewPicture || !*bitstreamFile || nalu.m_nalUnitType == NAL_UNIT_EOS ) && pcDecLib->getFirstSliceInSequence() )
          {
            pcDecLib->setFirstSliceInPicture( true );
          }
        }
      }
    
      if( !bRet )
      {
        CHECK( bDecodeUntilPocFound, " decoding failed - check decodeBitstream2 parameter File: " << bitstreamFileName.c_str() );
        if( pcDecLib )
        {
          pcDecLib->destroy();
          pcDecLib->deletePicBuffer();
          delete pcDecLib;
          pcDecLib = nullptr;
        }
        bFirstCall   = true;
        loopFiltered = false;
        iPOCLastDisplay = -MAX_INT;
    
        if( bytestream )
        {
          delete bytestream;
          bytestream = nullptr;
        }
    
        if( bitstreamFile )
        {
          delete bitstreamFile;
          bitstreamFile = nullptr;
        }
      }
    
      return bRet;
    }
    
    
    //! \ingroup DecoderLib
    //! \{
    
    DecLib::DecLib()
      : m_iMaxRefPicNum(0)
      , m_associatedIRAPType(NAL_UNIT_INVALID)
      , m_pocCRA(0)
      , m_pocRandomAccess(MAX_INT)
      , m_lastRasPoc(MAX_INT)
      , m_cListPic()
      , m_parameterSetManager()
      , m_apcSlicePilot(NULL)
      , m_SEIs()
      , m_cIntraPred()
      , m_cInterPred()
      , m_cTrQuant()
      , m_cSliceDecoder()
      , m_cCuDecoder()
      , m_HLSReader()
      , m_seiReader()
      , m_cLoopFilter()
      , m_cSAO()
    #if JVET_J0090_MEMORY_BANDWITH_MEASURE
      , m_cacheModel()
    #endif
      , m_pcPic(NULL)
      , m_prevPOC(MAX_INT)
      , m_prevTid0POC(0)
      , m_bFirstSliceInPicture(true)
      , m_bFirstSliceInSequence(true)
      , m_prevSliceSkipped(false)
      , m_skippedPOC(0)
      , m_bFirstSliceInBitstream(true)
      , m_lastPOCNoOutputPriorPics(-1)
      , m_isNoOutputPriorPics(false)
      , m_craNoRaslOutputFlag(false)
      , m_pDecodedSEIOutputStream(NULL)
      , m_decodedPictureHashSEIEnabled(false)
      , m_numberOfChecksumErrorsDetected(0)
      , m_warningMessageSkipPicture(false)
      , m_prefixSEINALUs()
    {
    #if ENABLE_SIMD_OPT_BUFFER
      g_pelBufOP.initPelBufOpsX86();
    #endif
    }
    
    DecLib::~DecLib()
    {
      while (!m_prefixSEINALUs.empty())
      {
        delete m_prefixSEINALUs.front();
        m_prefixSEINALUs.pop_front();
      }
    }
    
    void DecLib::create()
    {
      m_apcSlicePilot = new Slice;
      m_uiSliceSegmentIdx = 0;
    }
    
    void DecLib::destroy()
    {
      delete m_apcSlicePilot;
      m_apcSlicePilot = NULL;
    
      m_cSliceDecoder.destroy();
    }
    
    void DecLib::init(
    #if JVET_J0090_MEMORY_BANDWITH_MEASURE
      const std::string& cacheCfgFileName 
    #endif
    )
    {
      m_cSliceDecoder.init( &m_CABACDecoder, &m_cCuDecoder );
    #if JVET_J0090_MEMORY_BANDWITH_MEASURE
      m_cacheModel.create( cacheCfgFileName );
      m_cacheModel.clear( );
      m_cInterPred.cacheAssign( &m_cacheModel );
    #endif
      DTRACE_UPDATE( g_trace_ctx, std::make_pair( "final", 1 ) );
    }
    
    void DecLib::deletePicBuffer ( )
    {
      PicList::iterator  iterPic   = m_cListPic.begin();
      int iSize = int( m_cListPic.size() );
    
      for (int i = 0; i < iSize; i++ )
      {
        Picture* pcPic = *(iterPic++);
        pcPic->destroy();
    
        delete pcPic;
        pcPic = NULL;
      }
      m_cALF.destroy();
      m_cSAO.destroy();
      m_cLoopFilter.destroy();
    #if JVET_J0090_MEMORY_BANDWITH_MEASURE
      m_cacheModel.reportSequence( );
      m_cacheModel.destroy( );
    #endif
    }
    
    Picture* DecLib::xGetNewPicBuffer ( const SPS &sps, const PPS &pps, const uint32_t temporalLayer )
    {
      Picture * pcPic = nullptr;
      m_iMaxRefPicNum = sps.getMaxDecPicBuffering(temporalLayer);     // m_uiMaxDecPicBuffering has the space for the picture currently being decoded
      if (m_cListPic.size() < (uint32_t)m_iMaxRefPicNum)
      {
        pcPic = new Picture();
    
        pcPic->create( sps.getChromaFormatIdc(), Size( sps.getPicWidthInLumaSamples(), sps.getPicHeightInLumaSamples() ), sps.getMaxCUWidth(), sps.getMaxCUWidth() + 16, true );
    
        m_cListPic.push_back( pcPic );
    
        return pcPic;
      }
    
      bool bBufferIsAvailable = false;
      for(auto * p: m_cListPic)
      {
        pcPic = p;  // workaround because range-based for-loops don't work with existing variables
        if ( pcPic->reconstructed == false && ! pcPic->neededForOutput )
        {
          pcPic->neededForOutput = false;
          bBufferIsAvailable = true;
          break;
        }
    
        if( ! pcPic->referenced  && ! pcPic->neededForOutput )
        {
          pcPic->neededForOutput = false;
          pcPic->reconstructed = false;
          bBufferIsAvailable = true;
          break;
        }
      }
    
      if( ! bBufferIsAvailable )
      {
        //There is no room for this picture, either because of faulty encoder or dropped NAL. Extend the buffer.
        m_iMaxRefPicNum++;
    
        pcPic = new Picture();
    
        m_cListPic.push_back( pcPic );
    
        pcPic->create( sps.getChromaFormatIdc(), Size( sps.getPicWidthInLumaSamples(), sps.getPicHeightInLumaSamples() ), sps.getMaxCUWidth(), sps.getMaxCUWidth() + 16, true );
      }
      else
      {
        if( !pcPic->Y().Size::operator==( Size( sps.getPicWidthInLumaSamples(), sps.getPicHeightInLumaSamples() ) ) || pcPic->cs->pcv->maxCUWidth != sps.getMaxCUWidth() || pcPic->cs->pcv->maxCUHeight != sps.getMaxCUHeight() )
        {
          pcPic->destroy();
          pcPic->create( sps.getChromaFormatIdc(), Size( sps.getPicWidthInLumaSamples(), sps.getPicHeightInLumaSamples() ), sps.getMaxCUWidth(), sps.getMaxCUWidth() + 16, true );
        }
      }
    
      pcPic->setBorderExtension( false );
      pcPic->neededForOutput = false;
      pcPic->reconstructed = false;
    
      return pcPic;
    }
    
    
    void DecLib::executeLoopFilters()
    {
      if( !m_pcPic )
      {
        return; // nothing to deblock
      }
    
      CodingStructure& cs = *m_pcPic->cs;
    
      // deblocking filter
      m_cLoopFilter.loopFilterPic( cs );
    
      if( cs.sps->getUseSAO() )
      {
        m_cSAO.SAOProcess( cs, cs.picture->getSAO() );
      }
    
      if( cs.sps->getUseALF() )
      {
        m_cALF.ALFProcess( cs, cs.slice->getAlfSliceParam() );
      }
    }
    
    void DecLib::finishPictureLight(int& poc, PicList*& rpcListPic )
    {
      Slice*  pcSlice = m_pcPic->cs->slice;
    
      m_pcPic->neededForOutput = (pcSlice->getPicOutputFlag() ? true : false);
      m_pcPic->reconstructed = true;
    
      Slice::sortPicList( m_cListPic ); // sorting for application output
      poc                 = pcSlice->getPOC();
      rpcListPic          = &m_cListPic;
    }
    
    void DecLib::finishPicture(int& poc, PicList*& rpcListPic, MsgLevel msgl )
    {
    #if RExt__DECODER_DEBUG_TOOL_STATISTICS
      CodingStatistics::StatTool& s = CodingStatistics::GetStatisticTool( STATS__TOOL_TOTAL_FRAME );
      s.count++;
      s.pixels = s.count * m_pcPic->Y().width * m_pcPic->Y().height;
    #endif
    
      Slice*  pcSlice = m_pcPic->cs->slice;
    
      char c = (pcSlice->isIntra() ? 'I' : pcSlice->isInterP() ? 'P' : 'B');
      if (!m_pcPic->referenced)
      {
        c += 32;  // tolower
      }
    
      //-- For time output for each slice
      msg( msgl, "POC %4d TId: %1d ( %c-SLICE, QP%3d ) ", pcSlice->getPOC(),
             pcSlice->getTLayer(),
             c,
             pcSlice->getSliceQp() );
      msg( msgl, "[DT %6.3f] ", pcSlice->getProcessingTime() );
    
      for (int iRefList = 0; iRefList < 2; iRefList++)
      {
        msg( msgl, "[L%d ", iRefList);
        for (int iRefIndex = 0; iRefIndex < pcSlice->getNumRefIdx(RefPicList(iRefList)); iRefIndex++)
        {
          msg( msgl, "%d ", pcSlice->getRefPOC(RefPicList(iRefList), iRefIndex));
        }
        msg( msgl, "] ");
      }
      if (m_decodedPictureHashSEIEnabled)
      {
        SEIMessages pictureHashes = getSeisByType(m_pcPic->SEIs, SEI::DECODED_PICTURE_HASH );
        const SEIDecodedPictureHash *hash = ( pictureHashes.size() > 0 ) ? (SEIDecodedPictureHash*) *(pictureHashes.begin()) : NULL;
        if (pictureHashes.size() > 1)
        {
          msg( WARNING, "Warning: Got multiple decoded picture hash SEI messages. Using first.");
        }
        m_numberOfChecksumErrorsDetected += calcAndPrintHashStatus(((const Picture*) m_pcPic)->getRecoBuf(), hash, pcSlice->getSPS()->getBitDepths(), msgl);
      }
    
      msg( msgl, "\n");
    
      m_pcPic->neededForOutput = (pcSlice->getPicOutputFlag() ? true : false);
      m_pcPic->reconstructed = true;
    
    
      Slice::sortPicList( m_cListPic ); // sorting for application output
      poc                 = pcSlice->getPOC();
      rpcListPic          = &m_cListPic;
      m_bFirstSliceInPicture  = true; // TODO: immer true? hier ist irgendwas faul
    
      m_pcPic->destroyTempBuffers();
      m_pcPic->cs->destroyCoeffs();
      m_pcPic->cs->releaseIntermediateData();
    }
    
    void DecLib::checkNoOutputPriorPics (PicList* pcListPic)
    {
      if (!pcListPic || !m_isNoOutputPriorPics)
      {
        return;
      }
    
      PicList::iterator  iterPic   = pcListPic->begin();
    
      while (iterPic != pcListPic->end())
      {
        Picture* pcPicTmp = *(iterPic++);
        if (m_lastPOCNoOutputPriorPics != pcPicTmp->getPOC())
        {
          pcPicTmp->neededForOutput = false;
        }
      }
    }
    
    void DecLib::xUpdateRasInit(Slice* slice)
    {
      slice->setPendingRasInit( false );
      if ( slice->getPOC() > m_lastRasPoc )
      {
        m_lastRasPoc = MAX_INT;
        slice->setPendingRasInit( true );
      }
      if ( slice->isIRAP() )
      {
        m_lastRasPoc = slice->getPOC();
      }
    }
    
    void DecLib::xCreateLostPicture(int iLostPoc)
    {
      msg( INFO, "\ninserting lost poc : %d\n",iLostPoc);
      Picture *cFillPic = xGetNewPicBuffer(*(m_parameterSetManager.getFirstSPS()), *(m_parameterSetManager.getFirstPPS()), 0);
    
      CHECK( !cFillPic->slices.size(), "No slices in picture" );
    
      cFillPic->slices[0]->initSlice();
    
      PicList::iterator iterPic = m_cListPic.begin();
      int closestPoc = 1000000;
      while ( iterPic != m_cListPic.end())
      {
        Picture * rpcPic = *(iterPic++);
        if(abs(rpcPic->getPOC() -iLostPoc)<closestPoc&&abs(rpcPic->getPOC() -iLostPoc)!=0&&rpcPic->getPOC()!=m_apcSlicePilot->getPOC())
        {
          closestPoc=abs(rpcPic->getPOC() -iLostPoc);
        }
      }
      iterPic = m_cListPic.begin();
      while ( iterPic != m_cListPic.end())
      {
        Picture *rpcPic = *(iterPic++);
        if(abs(rpcPic->getPOC() -iLostPoc)==closestPoc&&rpcPic->getPOC()!=m_apcSlicePilot->getPOC())
        {
          msg( INFO, "copying picture %d to %d (%d)\n",rpcPic->getPOC() ,iLostPoc,m_apcSlicePilot->getPOC());
          cFillPic->getRecoBuf().copyFrom( rpcPic->getRecoBuf() );
          break;
        }
      }
    
    //  for(int ctuRsAddr=0; ctuRsAddr<cFillPic->getNumberOfCtusInFrame(); ctuRsAddr++)  { cFillPic->getCtu(ctuRsAddr)->initCtu(cFillPic, ctuRsAddr); }
      cFillPic->referenced = true;
      cFillPic->slices[0]->setPOC(iLostPoc);
      xUpdatePreviousTid0POC(cFillPic->slices[0]);
      cFillPic->reconstructed = true;
      cFillPic->neededForOutput = true;
      if(m_pocRandomAccess == MAX_INT)
      {
        m_pocRandomAccess = iLostPoc;
      }
    
    }
    
    
    void DecLib::xActivateParameterSets()
    {
      if (m_bFirstSliceInPicture)
      {
        const PPS *pps = m_parameterSetManager.getPPS(m_apcSlicePilot->getPPSId()); // this is a temporary PPS object. Do not store this value
        CHECK(pps == 0, "No PPS present");
    
        const SPS *sps = m_parameterSetManager.getSPS(pps->getSPSId());             // this is a temporary SPS object. Do not store this value
        CHECK(sps == 0, "No SPS present");
    
        if (NULL == pps->pcv)
        {
          m_parameterSetManager.getPPS( m_apcSlicePilot->getPPSId() )->pcv = new PreCalcValues( *sps, *pps, false );
        }
        m_parameterSetManager.clearSPSChangedFlag(sps->getSPSId());
        m_parameterSetManager.clearPPSChangedFlag(pps->getPPSId());
    
        if (false == m_parameterSetManager.activatePPS(m_apcSlicePilot->getPPSId(),m_apcSlicePilot->isIRAP()))
        {
          THROW("Parameter set activation failed!");
        }
    
        xParsePrefixSEImessages();
    
    #if RExt__HIGH_BIT_DEPTH_SUPPORT==0
        if (sps->getSpsRangeExtension().getExtendedPrecisionProcessingFlag() || sps->getBitDepth(CHANNEL_TYPE_LUMA)>12 || sps->getBitDepth(CHANNEL_TYPE_CHROMA)>12 )
        {
          THROW("High bit depth support must be enabled at compile-time in order to decode this bitstream\n");
        }
    #endif
    
        //  Get a new picture buffer. This will also set up m_pcPic, and therefore give us a SPS and PPS pointer that we can use.
        m_pcPic = xGetNewPicBuffer (*sps, *pps, m_apcSlicePilot->getTLayer());
    
        m_apcSlicePilot->applyReferencePictureSet(m_cListPic, m_apcSlicePilot->getRPS());
    
        m_pcPic->finalInit( *sps, *pps );
    
        m_pcPic->createTempBuffers( m_pcPic->cs->pps->pcv->maxCUWidth );
        m_pcPic->cs->createCoeffs();
    
        m_pcPic->allocateNewSlice();
        // make the slice-pilot a real slice, and set up the slice-pilot for the next slice
        CHECK(m_pcPic->slices.size() != (m_uiSliceSegmentIdx + 1), "Invalid number of slices");
        m_apcSlicePilot = m_pcPic->swapSliceObject(m_apcSlicePilot, m_uiSliceSegmentIdx);
    
        // we now have a real slice:
        Slice *pSlice = m_pcPic->slices[m_uiSliceSegmentIdx];
    
        // Update the PPS and SPS pointers with the ones of the picture.
        pps=pSlice->getPPS();
        sps=pSlice->getSPS();
    
        // fix Parameter Sets, now that we have the real slice
        m_pcPic->cs->slice = pSlice;
        m_pcPic->cs->sps   = sps;
        m_pcPic->cs->pps   = pps;
    #if HEVC_VPS
        m_pcPic->cs->vps   = pSlice->getVPS();
    #endif
        m_pcPic->cs->pcv   = pps->pcv;
    
        // Initialise the various objects for the new set of settings
        m_cSAO.create( sps->getPicWidthInLumaSamples(), sps->getPicHeightInLumaSamples(), sps->getChromaFormatIdc(), sps->getMaxCUWidth(), sps->getMaxCUHeight(), sps->getMaxCodingDepth(), pps->getPpsRangeExtension().getLog2SaoOffsetScale(CHANNEL_TYPE_LUMA), pps->getPpsRangeExtension().getLog2SaoOffsetScale(CHANNEL_TYPE_CHROMA) );
        m_cLoopFilter.create( sps->getMaxCodingDepth() );
        m_cIntraPred.init( sps->getChromaFormatIdc(), sps->getBitDepth( CHANNEL_TYPE_LUMA ) );
        m_cInterPred.init( &m_cRdCost, sps->getChromaFormatIdc() );
    
    
        bool isField = false;
        bool isTopField = false;
    
        if(!m_SEIs.empty())
        {
          // Check if any new Picture Timing SEI has arrived
          SEIMessages pictureTimingSEIs = getSeisByType(m_SEIs, SEI::PICTURE_TIMING);
          if (pictureTimingSEIs.size()>0)
          {
            SEIPictureTiming* pictureTiming = (SEIPictureTiming*) *(pictureTimingSEIs.begin());
            isField    = (pictureTiming->m_picStruct == 1) || (pictureTiming->m_picStruct == 2) || (pictureTiming->m_picStruct == 9) || (pictureTiming->m_picStruct == 10) || (pictureTiming->m_picStruct == 11) || (pictureTiming->m_picStruct == 12);
            isTopField = (pictureTiming->m_picStruct == 1) || (pictureTiming->m_picStruct == 9) || (pictureTiming->m_picStruct == 11);
          }
        }
    
        //Set Field/Frame coding mode
        m_pcPic->fieldPic = isField;
        m_pcPic->topField = isTopField;
    
        // transfer any SEI messages that have been received to the picture
        m_pcPic->SEIs = m_SEIs;
        m_SEIs.clear();
    
        // Recursive structure
        m_cCuDecoder.init( &m_cTrQuant, &m_cIntraPred, &m_cInterPred );
    
    Karsten Suehring's avatar
    Karsten Suehring committed
        m_cTrQuant.init( nullptr, sps->getMaxTrSize(), false, false, false, false, false );
    
    
        // RdCost
        m_cRdCost.setCostMode ( COST_STANDARD_LOSSY ); // not used in decoder side RdCost stuff -> set to default
    
        m_cSliceDecoder.create();
    
        if( sps->getUseALF() )
        {
          m_cALF.create( sps->getPicWidthInLumaSamples(), sps->getPicHeightInLumaSamples(), sps->getChromaFormatIdc(), sps->getMaxCUWidth(), sps->getMaxCUHeight(), sps->getMaxCodingDepth(), sps->getBitDepths().recon );
        }
      }
      else
      {
        // make the slice-pilot a real slice, and set up the slice-pilot for the next slice
        m_pcPic->allocateNewSlice();
        CHECK(m_pcPic->slices.size() != (size_t)(m_uiSliceSegmentIdx + 1), "Invalid number of slices");
        m_apcSlicePilot = m_pcPic->swapSliceObject(m_apcSlicePilot, m_uiSliceSegmentIdx);
    
        Slice *pSlice = m_pcPic->slices[m_uiSliceSegmentIdx]; // we now have a real slice.
    
        const SPS *sps = pSlice->getSPS();
        const PPS *pps = pSlice->getPPS();
    
        // fix Parameter Sets, now that we have the real slice
        m_pcPic->cs->slice = pSlice;
        m_pcPic->cs->sps   = sps;
        m_pcPic->cs->pps   = pps;
    #if HEVC_VPS
        m_pcPic->cs->vps   = pSlice->getVPS();
    #endif
        m_pcPic->cs->pcv   = pps->pcv;
    
        // check that the current active PPS has not changed...
        if (m_parameterSetManager.getSPSChangedFlag(sps->getSPSId()) )
        {
          EXIT("Error - a new SPS has been decoded while processing a picture");
        }
        if (m_parameterSetManager.getPPSChangedFlag(pps->getPPSId()) )
        {
          EXIT("Error - a new PPS has been decoded while processing a picture");
        }
    
        xParsePrefixSEImessages();
    
        // Check if any new SEI has arrived
         if(!m_SEIs.empty())
         {
           // Currently only decoding Unit SEI message occurring between VCL NALUs copied
           SEIMessages& picSEI = m_pcPic->SEIs;
           SEIMessages decodingUnitInfos = extractSeisByType( picSEI, SEI::DECODING_UNIT_INFO);
           picSEI.insert(picSEI.end(), decodingUnitInfos.begin(), decodingUnitInfos.end());
           deleteSEIs(m_SEIs);
         }
      }
    }
    
    
    void DecLib::xParsePrefixSEIsForUnknownVCLNal()
    {
      while (!m_prefixSEINALUs.empty())
      {
        // do nothing?
        msg( NOTICE, "Discarding Prefix SEI associated with unknown VCL NAL unit.\n");
        delete m_prefixSEINALUs.front();
      }
      // TODO: discard following suffix SEIs as well?
    }
    
    
    void DecLib::xParsePrefixSEImessages()
    {
      while (!m_prefixSEINALUs.empty())
      {
        InputNALUnit &nalu=*m_prefixSEINALUs.front();
        m_seiReader.parseSEImessage( &(nalu.getBitstream()), m_SEIs, nalu.m_nalUnitType, m_parameterSetManager.getActiveSPS(), m_pDecodedSEIOutputStream );
        delete m_prefixSEINALUs.front();
        m_prefixSEINALUs.pop_front();
      }
    }
    
    
    bool DecLib::xDecodeSlice(InputNALUnit &nalu, int &iSkipFrame, int iPOCLastDisplay )
    {
      m_apcSlicePilot->initSlice(); // the slice pilot is an object to prepare for a new slice
                                    // it is not associated with picture, sps or pps structures.
    
      if (m_bFirstSliceInPicture)
      {
        m_uiSliceSegmentIdx = 0;
      }
      else
      {
        m_apcSlicePilot->copySliceInfo( m_pcPic->slices[m_uiSliceSegmentIdx-1] );
      }
    #if HEVC_DEPENDENT_SLICES
      m_apcSlicePilot->setSliceSegmentIdx(m_uiSliceSegmentIdx);
    #endif
    
      m_apcSlicePilot->setNalUnitType(nalu.m_nalUnitType);
      bool nonReferenceFlag = (m_apcSlicePilot->getNalUnitType() == NAL_UNIT_CODED_SLICE_TRAIL_N ||
                               m_apcSlicePilot->getNalUnitType() == NAL_UNIT_CODED_SLICE_TSA_N   ||
                               m_apcSlicePilot->getNalUnitType() == NAL_UNIT_CODED_SLICE_STSA_N  ||
                               m_apcSlicePilot->getNalUnitType() == NAL_UNIT_CODED_SLICE_RADL_N  ||
                               m_apcSlicePilot->getNalUnitType() == NAL_UNIT_CODED_SLICE_RASL_N);
      m_apcSlicePilot->setTemporalLayerNonReferenceFlag(nonReferenceFlag);
      m_apcSlicePilot->setTLayer(nalu.m_temporalId);
    
      m_HLSReader.setBitstream( &nalu.getBitstream() );
      m_HLSReader.parseSliceHeader( m_apcSlicePilot, &m_parameterSetManager, m_prevTid0POC );
    
      // update independent slice index
      uint32_t uiIndependentSliceIdx = 0;
      if (!m_bFirstSliceInPicture)
      {
        uiIndependentSliceIdx = m_pcPic->slices[m_uiSliceSegmentIdx-1]->getIndependentSliceIdx();
    #if HEVC_DEPENDENT_SLICES
        if (!m_apcSlicePilot->getDependentSliceSegmentFlag())
        {
    #endif
          uiIndependentSliceIdx++;
    #if HEVC_DEPENDENT_SLICES
        }
    #endif
      }
      m_apcSlicePilot->setIndependentSliceIdx(uiIndependentSliceIdx);
    
    
    #if K0149_BLOCK_STATISTICS
      PPS *pps = m_parameterSetManager.getPPS(m_apcSlicePilot->getPPSId());
      CHECK(pps == 0, "No PPS present");
      SPS *sps = m_parameterSetManager.getSPS(pps->getSPSId());
      CHECK(sps == 0, "No SPS present");
    
      writeBlockStatisticsHeader(sps);
    #endif
    
    
      DTRACE_UPDATE( g_trace_ctx, std::make_pair( "poc", m_apcSlicePilot->getPOC() ) );
    
    #if HEVC_DEPENDENT_SLICES
      // set POC for dependent slices in skipped pictures
      if(m_apcSlicePilot->getDependentSliceSegmentFlag() && m_prevSliceSkipped)
      {
        m_apcSlicePilot->setPOC(m_skippedPOC);
      }
    #endif
    
      xUpdatePreviousTid0POC(m_apcSlicePilot);
    
      m_apcSlicePilot->setAssociatedIRAPPOC(m_pocCRA);
      m_apcSlicePilot->setAssociatedIRAPType(m_associatedIRAPType);
    
      //For inference of NoOutputOfPriorPicsFlag
      if (m_apcSlicePilot->getRapPicFlag())
      {
        if ((m_apcSlicePilot->getNalUnitType() >= NAL_UNIT_CODED_SLICE_BLA_W_LP && m_apcSlicePilot->getNalUnitType() <= NAL_UNIT_CODED_SLICE_IDR_N_LP) ||
            (m_apcSlicePilot->getNalUnitType() == NAL_UNIT_CODED_SLICE_CRA && m_bFirstSliceInSequence) ||
            (m_apcSlicePilot->getNalUnitType() == NAL_UNIT_CODED_SLICE_CRA && m_apcSlicePilot->getHandleCraAsBlaFlag()))
        {
          m_apcSlicePilot->setNoRaslOutputFlag(true);
        }
        //the inference for NoOutputPriorPicsFlag
        if (!m_bFirstSliceInBitstream && m_apcSlicePilot->getRapPicFlag() && m_apcSlicePilot->getNoRaslOutputFlag())
        {
          if (m_apcSlicePilot->getNalUnitType() == NAL_UNIT_CODED_SLICE_CRA)
          {
            m_apcSlicePilot->setNoOutputPriorPicsFlag(true);
          }
        }
        else
        {
          m_apcSlicePilot->setNoOutputPriorPicsFlag(false);
        }
    
        if(m_apcSlicePilot->getNalUnitType() == NAL_UNIT_CODED_SLICE_CRA)
        {
          m_craNoRaslOutputFlag = m_apcSlicePilot->getNoRaslOutputFlag();
        }
      }
      if (m_apcSlicePilot->getRapPicFlag() && m_apcSlicePilot->getNoOutputPriorPicsFlag())
      {
        m_lastPOCNoOutputPriorPics = m_apcSlicePilot->getPOC();
        m_isNoOutputPriorPics = true;
      }
      else
      {
        m_isNoOutputPriorPics = false;
      }
    
      //For inference of PicOutputFlag
      if (m_apcSlicePilot->getNalUnitType() == NAL_UNIT_CODED_SLICE_RASL_N || m_apcSlicePilot->getNalUnitType() == NAL_UNIT_CODED_SLICE_RASL_R)
      {
        if ( m_craNoRaslOutputFlag )
        {
          m_apcSlicePilot->setPicOutputFlag(false);
        }
      }
    
      if (m_apcSlicePilot->getNalUnitType() == NAL_UNIT_CODED_SLICE_CRA && m_craNoRaslOutputFlag) //Reset POC MSB when CRA has NoRaslOutputFlag equal to 1
      {
        PPS *pps = m_parameterSetManager.getPPS(m_apcSlicePilot->getPPSId());
        CHECK(pps == 0, "No PPS present");
        SPS *sps = m_parameterSetManager.getSPS(pps->getSPSId());
        CHECK(sps == 0, "No SPS present");
        int iMaxPOClsb = 1 << sps->getBitsForPOC();
        m_apcSlicePilot->setPOC( m_apcSlicePilot->getPOC() & (iMaxPOClsb - 1) );
        xUpdatePreviousTid0POC(m_apcSlicePilot);
      }
    
      // Skip pictures due to random access
    
      if (isRandomAccessSkipPicture(iSkipFrame, iPOCLastDisplay))
      {
        m_prevSliceSkipped = true;
        m_skippedPOC = m_apcSlicePilot->getPOC();
        return false;
      }
      // Skip TFD pictures associated with BLA/BLANT pictures
      if (isSkipPictureForBLA(iPOCLastDisplay))
      {
        m_prevSliceSkipped = true;
        m_skippedPOC = m_apcSlicePilot->getPOC();
        return false;
      }
    
      // clear previous slice skipped flag
      m_prevSliceSkipped = false;
    
      //we should only get a different poc for a new picture (with CTU address==0)
    #if HEVC_DEPENDENT_SLICES
      if (!m_apcSlicePilot->getDependentSliceSegmentFlag() && m_apcSlicePilot->getPOC()!=m_prevPOC && !m_bFirstSliceInSequence && (m_apcSlicePilot->getSliceCurStartCtuTsAddr() != 0))
    #else
      if(m_apcSlicePilot->getPOC() != m_prevPOC && !m_bFirstSliceInSequence && (m_apcSlicePilot->getSliceCurStartCtuTsAddr() != 0))
    #endif