Skip to content
Snippets Groups Projects
EncLib.cpp 94.4 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     EncLib.cpp
        \brief    encoder class
    */
    #include "EncLib.h"
    
    #include "EncModeCtrl.h"
    #include "AQp.h"
    #include "EncCu.h"
    
    #include "CommonLib/Picture.h"
    #include "CommonLib/CommonDef.h"
    #include "CommonLib/ChromaFormat.h"
    #if ENABLE_SPLIT_PARALLELISM
    #include <omp.h>
    #endif
    
    #include "EncLibCommon.h"
    
    
    //! \ingroup EncoderLib
    //! \{
    
    // ====================================================================================================================
    // Constructor / destructor / create / destroy
    // ====================================================================================================================
    
    
    Vadim Seregin's avatar
    Vadim Seregin committed
    #if JVET_N0278_FIXES
    
    EncLib::EncLib( EncLibCommon* encLibCommon )
      : m_cListPic( encLibCommon->getPictureBuffer() )
      , m_cEncALF( encLibCommon->getApsIdStart() )
      , m_spsMap( encLibCommon->getSpsMap() )
      , m_ppsMap( encLibCommon->getPpsMap() )
      , m_apsMap( encLibCommon->getApsMap() )
    
    Vadim Seregin's avatar
    Vadim Seregin committed
    #else
    
    EncLib::EncLib()
      : m_spsMap( MAX_NUM_SPS )
      , m_ppsMap( MAX_NUM_PPS )
    
      , m_apsMap(MAX_NUM_APS * MAX_NUM_APS_TYPE)
    
    Vadim Seregin's avatar
    Vadim Seregin committed
    #endif
    
    #if JVET_J0090_MEMORY_BANDWITH_MEASURE
      , m_cacheModel()
    #endif
    
    {
      m_iPOCLast          = -1;
      m_iNumPicRcvd       =  0;
      m_uiNumAllPicCoded  =  0;
    
      m_iMaxRefPicNum     = 0;
    
    #if ENABLE_SIMD_OPT_BUFFER
      g_pelBufOP.initPelBufOpsX86();
    #endif
    
    #if JVET_O0756_CALCULATE_HDRMETRICS
      m_metricTime = std::chrono::milliseconds(0);
    #endif
    
    Vadim Seregin's avatar
    Vadim Seregin committed
    
    #if JVET_N0278_FIXES
    
    Vadim Seregin's avatar
    Vadim Seregin committed
      m_layerId = NOT_VALID;
    
      m_picIdInGOP = NOT_VALID;
    
    Vadim Seregin's avatar
    Vadim Seregin committed
    #endif
    
    Vadim Seregin's avatar
    Vadim Seregin committed
    #if JVET_N0278_FIXES
    
    Vadim Seregin's avatar
    Vadim Seregin committed
    void EncLib::create( const int layerId )
    
    Vadim Seregin's avatar
    Vadim Seregin committed
    #else
    
    Vadim Seregin's avatar
    Vadim Seregin committed
    #endif
    
    Vadim Seregin's avatar
    Vadim Seregin committed
    #if JVET_N0278_FIXES
    
    Vadim Seregin's avatar
    Vadim Seregin committed
      m_layerId = layerId;
    
    Vadim Seregin's avatar
    Vadim Seregin committed
    #else
    
      TComHash::initBlockSizeToIndex();
    
    Vadim Seregin's avatar
    Vadim Seregin committed
    #endif
    
      m_iPOCLast = m_compositeRefEnabled ? -2 : -1;
    
      // create processing unit classes
      m_cGOPEncoder.        create( );
    #if ENABLE_SPLIT_PARALLELISM || ENABLE_WPP_PARALLELISM
    #if ENABLE_SPLIT_PARALLELISM
      m_numCuEncStacks  = m_numSplitThreads == 1 ? 1 : NUM_RESERVERD_SPLIT_JOBS;
    #else
      m_numCuEncStacks  = 1;
    #endif
    #if ENABLE_WPP_PARALLELISM
      m_numCuEncStacks *= ( m_numWppThreads + m_numWppExtraLines );
    #endif
    
      m_cCuEncoder      = new EncCu              [m_numCuEncStacks];
      m_cInterSearch    = new InterSearch        [m_numCuEncStacks];
      m_cIntraSearch    = new IntraSearch        [m_numCuEncStacks];
      m_cTrQuant        = new TrQuant            [m_numCuEncStacks];
      m_CABACEncoder    = new CABACEncoder       [m_numCuEncStacks];
      m_cRdCost         = new RdCost             [m_numCuEncStacks];
      m_CtxCache        = new CtxCache           [m_numCuEncStacks];
    
      for( int jId = 0; jId < m_numCuEncStacks; jId++ )
      {
        m_cCuEncoder[jId].         create( this );
      }
    #else
      m_cCuEncoder.         create( this );
    #endif
    #if JVET_J0090_MEMORY_BANDWITH_MEASURE
      m_cInterSearch.cacheAssign( &m_cacheModel );
    #endif
    
    #if ENABLE_SPLIT_PARALLELISM || ENABLE_WPP_PARALLELISM
      m_cReshaper = new EncReshape[m_numCuEncStacks];
    #endif
    
    Taoran Lu's avatar
    Taoran Lu committed
      if (m_lumaReshapeEnable)
    
    #if ENABLE_SPLIT_PARALLELISM || ENABLE_WPP_PARALLELISM
        for (int jId = 0; jId < m_numCuEncStacks; jId++)
        {
          m_cReshaper[jId].createEnc(getSourceWidth(), getSourceHeight(), m_maxCUWidth, m_maxCUHeight, m_bitDepth[COMPONENT_Y]);
        }
    #else
    
        m_cReshaper.createEnc( getSourceWidth(), getSourceHeight(), m_maxCUWidth, m_maxCUHeight, m_bitDepth[COMPONENT_Y]);
    
    #endif
    
        m_cRateCtrl.init(m_framesToBeEncoded, m_RCTargetBitrate, (int)((double)m_iFrameRate / m_temporalSubsampleRatio + 0.5), m_iGOPSize, m_iSourceWidth, m_iSourceHeight,
          m_maxCUWidth, m_maxCUHeight, getBitDepth(CHANNEL_TYPE_LUMA), m_RCKeepHierarchicalBit, m_RCUseLCUSeparateModel, m_GOPList);
    
      }
    
    }
    
    void EncLib::destroy ()
    {
      // destroy processing unit classes
      m_cGOPEncoder.        destroy();
      m_cSliceEncoder.      destroy();
    #if ENABLE_SPLIT_PARALLELISM || ENABLE_WPP_PARALLELISM
      for( int jId = 0; jId < m_numCuEncStacks; jId++ )
      {
        m_cCuEncoder[jId].destroy();
      }
    #else
      m_cCuEncoder.         destroy();
    #endif
      if( m_alf )
      {
        m_cEncALF.destroy();
      }
      m_cEncSAO.            destroyEncData();
      m_cEncSAO.            destroy();
      m_cLoopFilter.        destroy();
      m_cRateCtrl.          destroy();
    
    #if ENABLE_SPLIT_PARALLELISM || ENABLE_WPP_PARALLELISM
      for (int jId = 0; jId < m_numCuEncStacks; jId++)
      {
        m_cReshaper[jId].   destroy();
      }
    #else
    
    Taoran Lu's avatar
    Taoran Lu committed
      m_cReshaper.          destroy();
    #endif
    
    #if ENABLE_SPLIT_PARALLELISM || ENABLE_WPP_PARALLELISM
      for( int jId = 0; jId < m_numCuEncStacks; jId++ )
      {
        m_cInterSearch[jId].   destroy();
        m_cIntraSearch[jId].   destroy();
      }
    #else
      m_cInterSearch.       destroy();
      m_cIntraSearch.       destroy();
    #endif
    
    #if ENABLE_SPLIT_PARALLELISM || ENABLE_WPP_PARALLELISM
      delete[] m_cCuEncoder;
      delete[] m_cInterSearch;
      delete[] m_cIntraSearch;
      delete[] m_cTrQuant;
      delete[] m_CABACEncoder;
      delete[] m_cRdCost;
      delete[] m_CtxCache;
    #endif
    
    
    Vadim Seregin's avatar
    Vadim Seregin committed
    #if !JVET_N0278_FIXES
    
    Vadim Seregin's avatar
    Vadim Seregin committed
    #endif
    
      return;
    }
    
    void EncLib::init( bool isFieldCoding, AUWriterIf* auWriterIf )
    {
      m_AUWriterIf = auWriterIf;
    
      SPS &sps0=*(m_spsMap.allocatePS(0)); // NOTE: implementations that use more than 1 SPS need to be aware of activation issues.
    
    #if JVET_N0278_FIXES
      PPS &pps0 = *( m_ppsMap.allocatePS( m_layerId ) );
    #else
    
      APS &aps0 = *( m_apsMap.allocatePS( SCALING_LIST_APS ) );
      aps0.setAPSId( 0 );
      aps0.setAPSType( SCALING_LIST_APS );
    
      int dpsId = getDecodingParameterSetEnabled() ? 1 : 0;
      xInitDPS(m_dps, sps0, dpsId);
      sps0.setDecodingParameterSetId(m_dps.getDecodingParameterSetId());
    
    #if ENABLE_SPLIT_PARALLELISM
      if( omp_get_dynamic() )
      {
        omp_set_dynamic( false );
      }
      omp_set_nested( true );
    #endif
    
    
      if (getUseCompositeRef() || getDependentRAPIndicationSEIEnabled())
    
    
    #if U0132_TARGET_BITS_SATURATION
      if (m_RCCpbSaturationEnabled)
      {
    
    Virginie Drugeon's avatar
    Virginie Drugeon committed
        m_cRateCtrl.initHrdParam(sps0.getHrdParameters(), m_iFrameRate, m_RCInitialCpbFullness);
    
      }
    #endif
    #if ENABLE_SPLIT_PARALLELISM || ENABLE_WPP_PARALLELISM
      for( int jId = 0; jId < m_numCuEncStacks; jId++ )
      {
        m_cRdCost[jId].setCostMode ( m_costMode );
      }
    #else
      m_cRdCost.setCostMode ( m_costMode );
    #endif
    
      // initialize PPS
    
      pps0.setPicWidthInLumaSamples( m_iSourceWidth );
      pps0.setPicHeightInLumaSamples( m_iSourceHeight );
      pps0.setConformanceWindow( m_conformanceWindow );
    
    Hendry's avatar
    Hendry committed
      // initialize APS
    
    Hendry's avatar
    Hendry committed
      xInitRPL(sps0, isFieldCoding);
    
    Vadim Seregin's avatar
    Vadim Seregin committed
        PPS &pps = *( m_ppsMap.allocatePS( ENC_PPS_ID_RPR ) );
    
    #if JVET_P0590_SCALING_WINDOW
        Window& inputScalingWindow = pps0.getScalingWindow();
        int scaledWidth = int( ( pps0.getPicWidthInLumaSamples() - inputScalingWindow.getWindowLeftOffset() - inputScalingWindow.getWindowRightOffset() ) / m_scalingRatioHor );
    #else
    
    Vadim Seregin's avatar
    Vadim Seregin committed
        Window& inputConfWindow = pps0.getConformanceWindow();
    
        int scaledWidth = int((pps0.getPicWidthInLumaSamples() - (inputConfWindow.getWindowLeftOffset() + inputConfWindow.getWindowRightOffset()) * SPS::getWinUnitX(sps0.getChromaFormatIdc())) / m_scalingRatioHor);
    
    Vadim Seregin's avatar
    Vadim Seregin committed
        int minSizeUnit = std::max(8, (int)(sps0.getMaxCUHeight() >> (sps0.getMaxCodingDepth() - 1)));
        int temp = scaledWidth / minSizeUnit;
        int width = ( scaledWidth - ( temp * minSizeUnit) > 0 ? temp + 1 : temp ) * minSizeUnit;
    
    #if JVET_P0590_SCALING_WINDOW
        int scaledHeight = int( ( pps0.getPicHeightInLumaSamples() - inputScalingWindow.getWindowTopOffset() - inputScalingWindow.getWindowBottomOffset() ) / m_scalingRatioVer );
    #else
    
        int scaledHeight = int((pps0.getPicHeightInLumaSamples() - (inputConfWindow.getWindowTopOffset() + inputConfWindow.getWindowBottomOffset()) * SPS::getWinUnitY(sps0.getChromaFormatIdc())) / m_scalingRatioVer);
    
    Vadim Seregin's avatar
    Vadim Seregin committed
        temp = scaledHeight / minSizeUnit;
        int height = ( scaledHeight - ( temp * minSizeUnit) > 0 ? temp + 1 : temp ) * minSizeUnit;
    
    
        pps.setPicWidthInLumaSamples( width );
        pps.setPicHeightInLumaSamples( height );
    
        Window conformanceWindow;
        conformanceWindow.setWindow( 0, ( width - scaledWidth ) / SPS::getWinUnitX( sps0.getChromaFormatIdc() ), 0, ( height - scaledHeight ) / SPS::getWinUnitY( sps0.getChromaFormatIdc() ) );
        pps.setConformanceWindow( conformanceWindow );
    
    
    #if JVET_P0590_SCALING_WINDOW
        Window scalingWindow;
        scalingWindow.setWindow( 0, width - scaledWidth, 0, height - scaledHeight );
        pps.setScalingWindow( scalingWindow );
    #endif
    
    
    #if JVET_P1004_REMOVE_BRICKS
        // disable picture partitioning for scaled RPR pictures (slice/tile config only provided for the original resolution)
        m_noPicPartitionFlag = true;
    #endif
    
    
        xInitPPS( pps, sps0 ); // will allocate memory for and initialize pps.pcv inside
      }
    
    
    #if ER_CHROMA_QP_WCG_PPS
      if (m_wcgChromaQpControl.isEnabled())
      {
        PPS &pps1=*(m_ppsMap.allocatePS(1));
        xInitPPS(pps1, sps0);
      }
    #endif
    
      if (getUseCompositeRef())
    
      {
        PPS &pps2 = *(m_ppsMap.allocatePS(2));
        xInitPPS(pps2, sps0);
        xInitPPSforLT(pps2);
      }
    
    Brian Heng's avatar
    Brian Heng committed
    #if JVET_P1006_PICTURE_HEADER
      xInitPicHeader(m_picHeader, sps0, pps0);
    #endif
    
    
      // initialize processing unit classes
      m_cGOPEncoder.  init( this );
      m_cSliceEncoder.init( this, sps0 );
    #if ENABLE_SPLIT_PARALLELISM || ENABLE_WPP_PARALLELISM
      for( int jId = 0; jId < m_numCuEncStacks; jId++ )
      {
        // precache a few objects
        for( int i = 0; i < 10; i++ )
        {
          auto x = m_CtxCache[jId].get();
          m_CtxCache[jId].cache( x );
        }
    
        m_cCuEncoder[jId].init( this, sps0, jId );
    
        // initialize transform & quantization class
        m_cTrQuant[jId].init( jId == 0 ? nullptr : m_cTrQuant[0].getQuant(),
    
                              m_useRDOQ,
                              m_useRDOQTS,
    #if T0196_SELECTIVE_RDOQ
                              m_useSelectiveRDOQ,
    #endif
    
        );
    
        // initialize encoder search class
        CABACWriter* cabacEstimator = m_CABACEncoder[jId].getCABACEstimator( &sps0 );
        m_cIntraSearch[jId].init( this,
                                  &m_cTrQuant[jId],
                                  &m_cRdCost[jId],
                                  cabacEstimator,
    
                                  getCtxCache( jId ), m_maxCUWidth, m_maxCUHeight, m_maxTotalCUDepth
                                , &m_cReshaper[jId]
    
    #if JVET_P0077_LINE_CG_PALETTE
                                , sps0.getBitDepth(CHANNEL_TYPE_LUMA)
    #endif
    
        m_cInterSearch[jId].init( this,
                                  &m_cTrQuant[jId],
                                  m_iSearchRange,
                                  m_bipredSearchRange,
                                  m_motionEstimationSearchMethod,
    
                                  getUseCompositeRef(),
                                  m_maxCUWidth, m_maxCUHeight, m_maxTotalCUDepth, &m_cRdCost[jId], cabacEstimator, getCtxCache( jId )
                               , &m_cReshaper[jId]
        );
    
    
        // link temporary buffets from intra search with inter search to avoid unnecessary memory overhead
        m_cInterSearch[jId].setTempBuffers( m_cIntraSearch[jId].getSplitCSBuf(), m_cIntraSearch[jId].getFullCSBuf(), m_cIntraSearch[jId].getSaveCSBuf() );
      }
    #else  // ENABLE_SPLIT_PARALLELISM || ENABLE_WPP_PARALLELISM
    
      m_cCuEncoder.   init( this, sps0 );
    
    
      // initialize transform & quantization class
      m_cTrQuant.init( nullptr,
    
                       1 << m_log2MaxTbSize,
    
                       m_useRDOQ,
                       m_useRDOQTS,
    #if T0196_SELECTIVE_RDOQ
                       m_useSelectiveRDOQ,
    #endif
    
      );
    
      // initialize encoder search class
      CABACWriter* cabacEstimator = m_CABACEncoder.getCABACEstimator(&sps0);
      m_cIntraSearch.init( this,
                           &m_cTrQuant,
                           &m_cRdCost,
                           cabacEstimator,
    
                           getCtxCache(), m_maxCUWidth, m_maxCUHeight, m_maxTotalCUDepth
    
    Taoran Lu's avatar
    Taoran Lu committed
                         , &m_cReshaper
    
    #if JVET_P0077_LINE_CG_PALETTE
                         , sps0.getBitDepth(CHANNEL_TYPE_LUMA)
    #endif
    
      m_cInterSearch.init( this,
                           &m_cTrQuant,
                           m_iSearchRange,
                           m_bipredSearchRange,
                           m_motionEstimationSearchMethod,
    
                           getUseCompositeRef(),
    
    Taoran Lu's avatar
    Taoran Lu committed
        m_maxCUWidth, m_maxCUHeight, m_maxTotalCUDepth, &m_cRdCost, cabacEstimator, getCtxCache()
                         , &m_cReshaper
      );
    
    
      // link temporary buffets from intra search with inter search to avoid unneccessary memory overhead
      m_cInterSearch.setTempBuffers( m_cIntraSearch.getSplitCSBuf(), m_cIntraSearch.getFullCSBuf(), m_cIntraSearch.getSaveCSBuf() );
    #endif // ENABLE_SPLIT_PARALLELISM || ENABLE_WPP_PARALLELISM
    
      m_iMaxRefPicNum = 0;
    
    #if ER_CHROMA_QP_WCG_PPS
      if( m_wcgChromaQpControl.isEnabled() )
      {
    
        xInitScalingLists( sps0, *m_apsMap.getPS( 1 ) );
        xInitScalingLists( sps0, aps0 );
    
        xInitScalingLists( sps0, aps0 );
    
        xInitScalingLists( sps0, *m_apsMap.getPS( ENC_PPS_ID_RPR ) );
    
    #if ENABLE_WPP_PARALLELISM
      m_entropyCodingSyncContextStateVec.resize( pps0.pcv->heightInCtus );
    #endif
    
      if (getUseCompositeRef())
    
    Vadim Seregin's avatar
    Vadim Seregin committed
    #if JVET_N0278_FIXES
    
    Vadim Seregin's avatar
    Vadim Seregin committed
        picBg->create( sps0.getChromaFormatIdc(), Size( pps0.getPicWidthInLumaSamples(), pps0.getPicHeightInLumaSamples() ), sps0.getMaxCUWidth(), sps0.getMaxCUWidth() + 16, false, m_layerId );
    
    Vadim Seregin's avatar
    Vadim Seregin committed
    #else
    
        picBg->create( sps0.getChromaFormatIdc(), Size( pps0.getPicWidthInLumaSamples(), pps0.getPicHeightInLumaSamples() ), sps0.getMaxCUWidth(), sps0.getMaxCUWidth() + 16, false );
    
    Vadim Seregin's avatar
    Vadim Seregin committed
    #endif
    
    Brian Heng's avatar
    Brian Heng committed
    #if JVET_P1006_PICTURE_HEADER
        picBg->finalInit( sps0, pps0, &m_picHeader, m_apss, m_lmcsAPS, m_scalinglistAPS );
    #else
    
        picBg->finalInit( sps0, pps0, m_apss, m_lmcsAPS, m_scalinglistAPS );
    
    Brian Heng's avatar
    Brian Heng committed
    #endif
    
    #if !JVET_P1004_REMOVE_BRICKS
    
        pps0.setNumBricksInPic((int)picBg->brickMap->bricks.size());
    
    #endif
    
        picBg->allocateNewSlice();
        picBg->createSpliceIdx(pps0.pcv->sizeInCtus);
        m_cGOPEncoder.setPicBg(picBg);
        Picture *picOrig = new Picture;
    
    Vadim Seregin's avatar
    Vadim Seregin committed
    #if JVET_N0278_FIXES
    
    Vadim Seregin's avatar
    Vadim Seregin committed
        picOrig->create( sps0.getChromaFormatIdc(), Size( pps0.getPicWidthInLumaSamples(), pps0.getPicHeightInLumaSamples() ), sps0.getMaxCUWidth(), sps0.getMaxCUWidth() + 16, false, m_layerId );
    
    Vadim Seregin's avatar
    Vadim Seregin committed
    #else
    
        picOrig->create( sps0.getChromaFormatIdc(), Size( pps0.getPicWidthInLumaSamples(), pps0.getPicHeightInLumaSamples() ), sps0.getMaxCUWidth(), sps0.getMaxCUWidth() + 16, false );
    
    Vadim Seregin's avatar
    Vadim Seregin committed
    #endif
    
        picOrig->getOrigBuf().fill(0);
        m_cGOPEncoder.setPicOrig(picOrig);
      }
    
    void EncLib::xInitScalingLists( SPS &sps, APS &aps )
    
    {
      // Initialise scaling lists
      // The encoder will only use the SPS scaling lists. The PPS will never be marked present.
      const int maxLog2TrDynamicRange[MAX_NUM_CHANNEL_TYPE] =
      {
    
        sps.getMaxLog2TrDynamicRange(CHANNEL_TYPE_LUMA),
        sps.getMaxLog2TrDynamicRange(CHANNEL_TYPE_CHROMA)
    
      };
    
      Quant* quant = getTrQuant()->getQuant();
    
      if(getUseScalingListId() == SCALING_LIST_OFF)
      {
        quant->setFlatScalingList(maxLog2TrDynamicRange, sps.getBitDepths());
        quant->setUseScalingList(false);
    #if ENABLE_SPLIT_PARALLELISM || ENABLE_WPP_PARALLELISM
        for( int jId = 1; jId < m_numCuEncStacks; jId++ )
        {
          getTrQuant( jId )->getQuant()->setFlatScalingList( maxLog2TrDynamicRange, sps.getBitDepths() );
          getTrQuant( jId )->getQuant()->setUseScalingList( false );
        }
    
      }
      else if(getUseScalingListId() == SCALING_LIST_DEFAULT)
      {
    
        aps.getScalingList().setDefaultScalingList ();
        quant->setScalingList( &( aps.getScalingList() ), maxLog2TrDynamicRange, sps.getBitDepths() );
    
        quant->setUseScalingList(true);
    #if ENABLE_SPLIT_PARALLELISM || ENABLE_WPP_PARALLELISM
        for( int jId = 1; jId < m_numCuEncStacks; jId++ )
        {
          getTrQuant( jId )->getQuant()->setUseScalingList( true );
        }
    
    #if JVET_P0365_SCALING_MATRIX_LFNST
        aps.getScalingList().setDisableScalingMatrixForLfnstBlks(getDisableScalingMatrixForLfnstBlks());
    #endif
    
    #endif
      }
      else if(getUseScalingListId() == SCALING_LIST_FILE_READ)
      {
    
        aps.getScalingList().setDefaultScalingList();
    
        CHECK( aps.getScalingList().xParseScalingList( getScalingListFileName() ), "Error Parsing Scaling List Input File" );
    
        aps.getScalingList().checkDcOfMatrix();
    
    Kiyofumi Abe's avatar
    Kiyofumi Abe committed
        if( aps.getScalingList().isNotDefaultScalingList() == false )
    
        {
          setUseScalingListId( SCALING_LIST_DEFAULT );
        }
        quant->setScalingList( &( aps.getScalingList() ), maxLog2TrDynamicRange, sps.getBitDepths() );
    
        quant->setUseScalingList(true);
    #if ENABLE_SPLIT_PARALLELISM || ENABLE_WPP_PARALLELISM
        for( int jId = 1; jId < m_numCuEncStacks; jId++ )
        {
          getTrQuant( jId )->getQuant()->setUseScalingList( true );
        }
    
    #endif
    #if JVET_P0365_SCALING_MATRIX_LFNST
        aps.getScalingList().setDisableScalingMatrixForLfnstBlks(getDisableScalingMatrixForLfnstBlks());
    
    #endif
      }
      else
      {
        THROW("error : ScalingList == " << getUseScalingListId() << " not supported\n");
      }
    
    
      if( getUseScalingListId() == SCALING_LIST_FILE_READ )
    
    #if JVET_P01034_PRED_1D_SCALING_LIST
        for (uint32_t scalingListId = 0; scalingListId < 28; scalingListId++)
        {
            aps.getScalingList().checkPredMode(scalingListId);
        }
    #else
    
        for (uint32_t sizeId = SCALING_LIST_2x2; sizeId <= SCALING_LIST_64x64; sizeId++)
    
          for (uint32_t listId = 0; listId < SCALING_LIST_NUM; listId++)
          {
    
            if (((sizeId == SCALING_LIST_64x64) && (listId % (SCALING_LIST_NUM / SCALING_LIST_PRED_MODES) != 0))
             || ((sizeId == SCALING_LIST_2x2) && (listId % (SCALING_LIST_NUM / SCALING_LIST_PRED_MODES) == 0)))
    
            aps.getScalingList().checkPredMode( sizeId, listId );
    
    void EncLib::xInitPPSforLT(PPS& pps)
    {
      pps.setOutputFlagPresentFlag(true);
      pps.setDeblockingFilterControlPresentFlag(true);
      pps.setPPSDeblockingFilterDisabledFlag(true);
    }
    
    
    // ====================================================================================================================
    // Public member functions
    // ====================================================================================================================
    
    void EncLib::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();
    
        // get rid of the qpadaption layer
        while( pcPic->aqlayer.size() )
        {
          delete pcPic->aqlayer.back(); pcPic->aqlayer.pop_back();
        }
    
        delete pcPic;
        pcPic = NULL;
      }
    
    Vadim Seregin's avatar
    Vadim Seregin committed
    
    #if JVET_N0278_FIXES
      m_cListPic.clear();
    #endif
    
    #if JVET_N0278_FIXES
    bool EncLib::encodePrep( bool flush, PelStorage* pcPicYuvOrg, PelStorage* cPicYuvTrueOrg, const InputColourSpaceConversion snrCSC, std::list<PelUnitBuf*>& rcListPicYuvRecOut, int& iNumEncoded )
    {
      if( m_compositeRefEnabled && m_cGOPEncoder.getPicBg()->getSpliceFull() && m_iPOCLast >= 10 && m_iNumPicRcvd == 0 && m_cGOPEncoder.getEncodedLTRef() == false )
      {
        Picture* picCurr = NULL;
        xGetNewPicBuffer( rcListPicYuvRecOut, picCurr, 2 );
        const PPS *pps = m_ppsMap.getPS( 2 );
        const SPS *sps = m_spsMap.getPS( pps->getSPSId() );
    
        picCurr->M_BUFS( 0, PIC_ORIGINAL ).copyFrom( m_cGOPEncoder.getPicBg()->getRecoBuf() );
    
    Brian Heng's avatar
    Brian Heng committed
    #if JVET_P1006_PICTURE_HEADER
        picCurr->finalInit( *sps, *pps, &m_picHeader, m_apss, m_lmcsAPS, m_scalinglistAPS );
    #else
    
        picCurr->finalInit( *sps, *pps, m_apss, m_lmcsAPS, m_scalinglistAPS );
    
    Brian Heng's avatar
    Brian Heng committed
    #endif
    
        picCurr->poc = m_iPOCLast - 1;
        m_iPOCLast -= 2;
        if( getUseAdaptiveQP() )
        {
          AQpPreanalyzer::preanalyze( picCurr );
        }
        if( m_RCEnableRateControl )
        {
          m_cRateCtrl.initRCGOP( m_iNumPicRcvd );
        }
    
    
        m_cGOPEncoder.compressGOP( m_iPOCLast, m_iNumPicRcvd, m_cListPic, rcListPicYuvRecOut, false, false, snrCSC, m_printFrameMSE, true, 0 );
    
    
    #if JVET_O0756_CALCULATE_HDRMETRICS
        m_metricTime = m_cGOPEncoder.getMetricTime();
    #endif
        m_cGOPEncoder.setEncodedLTRef( true );
        if( m_RCEnableRateControl )
        {
          m_cRateCtrl.destroyRCGOP();
        }
    
        iNumEncoded = 0;
        m_iNumPicRcvd = 0;
      }
    
      //PROF_ACCUM_AND_START_NEW_SET( getProfilerPic(), P_GOP_LEVEL );
      if( pcPicYuvOrg != NULL )
      {
        // get original YUV
        Picture* pcPicCurr = NULL;
    
        int ppsID = -1; // Use default PPS ID
    #if ER_CHROMA_QP_WCG_PPS
        if( getWCGChromaQPControl().isEnabled() )
        {
          ppsID = getdQPs()[m_iPOCLast / ( m_compositeRefEnabled ? 2 : 1 ) + 1];
          ppsID += ( getSwitchPOC() != -1 && ( m_iPOCLast + 1 >= getSwitchPOC() ) ? 1 : 0 );
        }
    #endif
    
        if( m_rprEnabled && m_uiIntraPeriod == -1 )
        {
          const int poc = m_iPOCLast + ( m_compositeRefEnabled ? 2 : 1 );
    
          if( poc / m_switchPocPeriod % 2 )
          {
            ppsID = ENC_PPS_ID_RPR;
          }
          else
          {
            ppsID = 0;
          }
        }
    
    #if JVET_N0278_FIXES
        if( m_cVPS.getMaxLayers() > 1 )
        {
          ppsID = m_layerId;
        }
    #endif
    
        xGetNewPicBuffer( rcListPicYuvRecOut, pcPicCurr, ppsID );
    
        const PPS *pPPS = ( ppsID < 0 ) ? m_ppsMap.getFirstPS() : m_ppsMap.getPS( ppsID );
        const SPS *pSPS = m_spsMap.getPS( pPPS->getSPSId() );
    
        if( m_rprEnabled )
        {
          pcPicCurr->M_BUFS( 0, PIC_ORIGINAL_INPUT ).getBuf( COMPONENT_Y ).copyFrom( pcPicYuvOrg->getBuf( COMPONENT_Y ) );
          pcPicCurr->M_BUFS( 0, PIC_ORIGINAL_INPUT ).getBuf( COMPONENT_Cb ).copyFrom( pcPicYuvOrg->getBuf( COMPONENT_Cb ) );
          pcPicCurr->M_BUFS( 0, PIC_ORIGINAL_INPUT ).getBuf( COMPONENT_Cr ).copyFrom( pcPicYuvOrg->getBuf( COMPONENT_Cr ) );
    
          pcPicCurr->M_BUFS( 0, PIC_TRUE_ORIGINAL_INPUT ).getBuf( COMPONENT_Y ).copyFrom( cPicYuvTrueOrg->getBuf( COMPONENT_Y ) );
          pcPicCurr->M_BUFS( 0, PIC_TRUE_ORIGINAL_INPUT ).getBuf( COMPONENT_Cb ).copyFrom( cPicYuvTrueOrg->getBuf( COMPONENT_Cb ) );
          pcPicCurr->M_BUFS( 0, PIC_TRUE_ORIGINAL_INPUT ).getBuf( COMPONENT_Cr ).copyFrom( cPicYuvTrueOrg->getBuf( COMPONENT_Cr ) );
    
          const ChromaFormat chromaFormatIDC = pSPS->getChromaFormatIdc();
    
          const PPS *refPPS = m_ppsMap.getPS( 0 );
    
    #if JVET_P0590_SCALING_WINDOW
          const Window& curScalingWindow = pPPS->getScalingWindow();
          int curPicWidth = pPPS->getPicWidthInLumaSamples() - curScalingWindow.getWindowLeftOffset() - curScalingWindow.getWindowRightOffset();
          int curPicHeight = pPPS->getPicHeightInLumaSamples() - curScalingWindow.getWindowTopOffset() - curScalingWindow.getWindowBottomOffset();
    
          const Window& refScalingWindow = refPPS->getScalingWindow();
          int refPicWidth = refPPS->getPicWidthInLumaSamples() - refScalingWindow.getWindowLeftOffset() - refScalingWindow.getWindowRightOffset();
          int refPicHeight = refPPS->getPicHeightInLumaSamples() - refScalingWindow.getWindowTopOffset() - refScalingWindow.getWindowBottomOffset();
    
          int xScale = ( ( refPicWidth << SCALE_RATIO_BITS ) + ( curPicWidth >> 1 ) ) / curPicWidth;
          int yScale = ( ( refPicHeight << SCALE_RATIO_BITS ) + ( curPicHeight >> 1 ) ) / curPicHeight;
          std::pair<int, int> scalingRatio = std::pair<int, int>( xScale, yScale );
    
    #if JVET_P0592_CHROMA_PHASE
          Picture::rescalePicture( scalingRatio, *pcPicYuvOrg, refPPS->getScalingWindow(), pcPicCurr->getOrigBuf(), pPPS->getScalingWindow(), chromaFormatIDC, pSPS->getBitDepths(), true, true,
            pSPS->getHorCollocatedChromaFlag(), pSPS->getVerCollocatedChromaFlag() );
          Picture::rescalePicture( scalingRatio, *cPicYuvTrueOrg, refPPS->getScalingWindow(), pcPicCurr->getTrueOrigBuf(), pPPS->getScalingWindow(), chromaFormatIDC, pSPS->getBitDepths(), true, true,
            pSPS->getHorCollocatedChromaFlag(), pSPS->getVerCollocatedChromaFlag() );
    #else
          Picture::rescalePicture( scalingRatio, *pcPicYuvOrg, refPPS->getScalingWindow(), pcPicCurr->getOrigBuf(), pPPS->getScalingWindow(), chromaFormatIDC, pSPS->getBitDepths(), true, true );
          Picture::rescalePicture( scalingRatio, *cPicYuvTrueOrg, refPPS->getScalingWindow(), pcPicCurr->getTrueOrigBuf(), pPPS->getScalingWindow(), chromaFormatIDC, pSPS->getBitDepths(), true, true );
    #endif
    #elif JVET_P0592_CHROMA_PHASE
          const Window& curWindow = pPPS->getConformanceWindow();
          int curPicWidth = pPPS->getPicWidthInLumaSamples() - ( curWindow.getWindowLeftOffset() + curWindow.getWindowRightOffset() ) * SPS::getWinUnitX( chromaFormatIDC );
          int curPicHeight = pPPS->getPicHeightInLumaSamples() - ( curWindow.getWindowTopOffset() + curWindow.getWindowBottomOffset() ) * SPS::getWinUnitY( chromaFormatIDC );
    
          const Window& refWindow = refPPS->getConformanceWindow();
          int refPicWidth = refPPS->getPicWidthInLumaSamples() - ( refWindow.getWindowLeftOffset() + refWindow.getWindowRightOffset() ) * SPS::getWinUnitX( chromaFormatIDC );
          int refPicHeight = refPPS->getPicHeightInLumaSamples() - ( refWindow.getWindowTopOffset() + refWindow.getWindowBottomOffset() ) * SPS::getWinUnitY( chromaFormatIDC );
    
          int xScale = ( ( refPicWidth << SCALE_RATIO_BITS ) + ( curPicWidth >> 1 ) ) / curPicWidth;
          int yScale = ( ( refPicHeight << SCALE_RATIO_BITS ) + ( curPicHeight >> 1 ) ) / curPicHeight;
          std::pair<int, int> scalingRatio = std::pair<int, int>( xScale, yScale );
    
          Picture::rescalePicture( scalingRatio, *pcPicYuvOrg, refPPS->getConformanceWindow(), pcPicCurr->getOrigBuf(), pPPS->getConformanceWindow(), chromaFormatIDC, pSPS->getBitDepths(), true, true,
            pSPS->getHorCollocatedChromaFlag(), pSPS->getVerCollocatedChromaFlag() );
          Picture::rescalePicture( scalingRatio, *cPicYuvTrueOrg, refPPS->getConformanceWindow(), pcPicCurr->getTrueOrigBuf(), pPPS->getConformanceWindow(), chromaFormatIDC, pSPS->getBitDepths(), true, true,
            pSPS->getHorCollocatedChromaFlag(), pSPS->getVerCollocatedChromaFlag() );
    #else
    
          Picture::rescalePicture( *pcPicYuvOrg, refPPS->getConformanceWindow(), pcPicCurr->getOrigBuf(), pPPS->getConformanceWindow(), chromaFormatIDC, pSPS->getBitDepths(), true, true );
          Picture::rescalePicture( *cPicYuvTrueOrg, refPPS->getConformanceWindow(), pcPicCurr->getTrueOrigBuf(), pPPS->getConformanceWindow(), chromaFormatIDC, pSPS->getBitDepths(), true, true );
    
        }
        else
        {
          pcPicCurr->M_BUFS( 0, PIC_ORIGINAL ).swap( *pcPicYuvOrg );
          pcPicCurr->M_BUFS( 0, PIC_TRUE_ORIGINAL ).swap( *cPicYuvTrueOrg );
        }
    
    Brian Heng's avatar
    Brian Heng committed
    #if JVET_P1006_PICTURE_HEADER    
        pcPicCurr->finalInit( *pSPS, *pPPS, &m_picHeader, m_apss, m_lmcsAPS, m_scalinglistAPS );
    #else
    
        pcPicCurr->finalInit( *pSPS, *pPPS, m_apss, m_lmcsAPS, m_scalinglistAPS );
    
    Brian Heng's avatar
    Brian Heng committed
    #endif
    
    #if !JVET_P1004_REMOVE_BRICKS
    
        PPS *ptrPPS = ( ppsID < 0 ) ? m_ppsMap.getFirstPS() : m_ppsMap.getPS( ppsID );
        ptrPPS->setNumBricksInPic( (int)pcPicCurr->brickMap->bricks.size() );
    
    #endif
    
    
        pcPicCurr->poc = m_iPOCLast;
    
        // compute image characteristics
        if( getUseAdaptiveQP() )
        {
          AQpPreanalyzer::preanalyze( pcPicCurr );
        }
      }
    
      if( ( m_iNumPicRcvd == 0 ) || ( !flush && ( m_iPOCLast != 0 ) && ( m_iNumPicRcvd != m_iGOPSize ) && ( m_iGOPSize != 0 ) ) )
      {
        iNumEncoded = 0;
        return true;
      }
    
      if( m_RCEnableRateControl )
      {
        m_cRateCtrl.initRCGOP( m_iNumPicRcvd );
      }
    
      m_picIdInGOP = 0;
    
      return false;
    
    /**
     - Application has picture buffer list with size of GOP + 1
     - Picture buffer list acts like as ring buffer
     - End of the list has the latest picture
     .
     \param   flush               cause encoder to encode a partial GOP
     \param   pcPicYuvOrg         original YUV picture
     \param   pcPicYuvTrueOrg
     \param   snrCSC
     \retval  rcListPicYuvRecOut  list of reconstruction YUV pictures
     \retval  accessUnitsOut      list of output access units
     \retval  iNumEncoded         number of encoded pictures
     */
    
    bool EncLib::encode( const InputColourSpaceConversion snrCSC, std::list<PelUnitBuf*>& rcListPicYuvRecOut, int& iNumEncoded )
    { 
      // compress GOP
      m_cGOPEncoder.compressGOP( m_iPOCLast, m_iNumPicRcvd, m_cListPic, rcListPicYuvRecOut,
        false, false, snrCSC, m_printFrameMSE, false, m_picIdInGOP );
    
      m_picIdInGOP++;
    
      // go over all pictures in a GOP excluding the first IRAP
      if( m_picIdInGOP != m_iGOPSize && m_iPOCLast )
      {
        return true;
      }
    
    #if JVET_O0756_CALCULATE_HDRMETRICS
      m_metricTime = m_cGOPEncoder.getMetricTime();
    #endif
    
      if( m_RCEnableRateControl )
      {
        m_cRateCtrl.destroyRCGOP();
      }
    
      iNumEncoded = m_iNumPicRcvd;
      m_iNumPicRcvd = 0;
      m_uiNumAllPicCoded += iNumEncoded;
    
      return false;
    }
    #else
    
    /**
     - Application has picture buffer list with size of GOP + 1
     - Picture buffer list acts like as ring buffer
     - End of the list has the latest picture
     .
     \param   flush               cause encoder to encode a partial GOP
     \param   pcPicYuvOrg         original YUV picture
     \param   pcPicYuvTrueOrg
     \param   snrCSC
     \retval  rcListPicYuvRecOut  list of reconstruction YUV pictures
     \retval  accessUnitsOut      list of output access units
     \retval  iNumEncoded         number of encoded pictures
     */
    void EncLib::encode( bool flush, PelStorage* pcPicYuvOrg, PelStorage* cPicYuvTrueOrg, const InputColourSpaceConversion snrCSC, std::list<PelUnitBuf*>& rcListPicYuvRecOut,
                         int& iNumEncoded )
    {
    
      if (m_compositeRefEnabled && m_cGOPEncoder.getPicBg()->getSpliceFull() && m_iPOCLast >= 10 && m_iNumPicRcvd == 0 && m_cGOPEncoder.getEncodedLTRef() == false)
      {
        Picture* picCurr = NULL;
        xGetNewPicBuffer(rcListPicYuvRecOut, picCurr, 2);
        const PPS *pps = m_ppsMap.getPS(2);
        const SPS *sps = m_spsMap.getPS(pps->getSPSId());
    
        picCurr->M_BUFS(0, PIC_ORIGINAL).copyFrom(m_cGOPEncoder.getPicBg()->getRecoBuf());
    
    Brian Heng's avatar
    Brian Heng committed
    #if JVET_P1006_PICTURE_HEADER
        picCurr->finalInit( *sps, *pps, &m_picHeader, m_apss, m_lmcsAPS, m_scalinglistAPS );
    #else
    
        picCurr->finalInit( *sps, *pps, m_apss, m_lmcsAPS, m_scalinglistAPS );
    
    Brian Heng's avatar
    Brian Heng committed
    #endif
    
        picCurr->poc = m_iPOCLast - 1;
        m_iPOCLast -= 2;
        if (getUseAdaptiveQP())
        {
          AQpPreanalyzer::preanalyze(picCurr);
        }
        if (m_RCEnableRateControl)
        {
          m_cRateCtrl.initRCGOP(m_iNumPicRcvd);
        }
    
        m_cGOPEncoder.compressGOP(m_iPOCLast, m_iNumPicRcvd, m_cListPic, rcListPicYuvRecOut,
          false, false, snrCSC, m_printFrameMSE, true);
    
    #if JVET_O0756_CALCULATE_HDRMETRICS
    
        m_metricTime = m_cGOPEncoder.getMetricTime();
    
        m_cGOPEncoder.setEncodedLTRef(true);
        if (m_RCEnableRateControl)
        {
          m_cRateCtrl.destroyRCGOP();
        }
    
        iNumEncoded = 0;
        m_iNumPicRcvd = 0;
      }
    
      //PROF_ACCUM_AND_START_NEW_SET( getProfilerPic(), P_GOP_LEVEL );
      if (pcPicYuvOrg != NULL)
      {
        // get original YUV
        Picture* pcPicCurr = NULL;
    
        int ppsID=-1; // Use default PPS ID
    
    #if ER_CHROMA_QP_WCG_PPS
    
          ppsID = getdQPs()[m_iPOCLast / (m_compositeRefEnabled ? 2 : 1) + 1];
    
          ppsID+=(getSwitchPOC() != -1 && (m_iPOCLast+1 >= getSwitchPOC())?1:0);
        }
    
    
        if( m_rprEnabled && m_uiIntraPeriod == -1 )
        {
          const int poc = m_iPOCLast + ( m_compositeRefEnabled ? 2 : 1 );
    
          if( poc / m_switchPocPeriod % 2 )
          {
    
    Vadim Seregin's avatar
    Vadim Seregin committed
            ppsID = ENC_PPS_ID_RPR;
    
        xGetNewPicBuffer( rcListPicYuvRecOut, pcPicCurr, ppsID );
    
        const PPS *pPPS = ( ppsID < 0 ) ? m_ppsMap.getFirstPS() : m_ppsMap.getPS( ppsID );
        const SPS *pSPS = m_spsMap.getPS( pPPS->getSPSId() );
    
        if( m_rprEnabled )
        {
          pcPicCurr->M_BUFS( 0, PIC_ORIGINAL_INPUT ).getBuf( COMPONENT_Y ).copyFrom( pcPicYuvOrg->getBuf( COMPONENT_Y ) );
          pcPicCurr->M_BUFS( 0, PIC_ORIGINAL_INPUT ).getBuf( COMPONENT_Cb ).copyFrom( pcPicYuvOrg->getBuf( COMPONENT_Cb ) );
          pcPicCurr->M_BUFS( 0, PIC_ORIGINAL_INPUT ).getBuf( COMPONENT_Cr ).copyFrom( pcPicYuvOrg->getBuf( COMPONENT_Cr ) );
    
          pcPicCurr->M_BUFS( 0, PIC_TRUE_ORIGINAL_INPUT ).getBuf( COMPONENT_Y ).copyFrom( cPicYuvTrueOrg->getBuf( COMPONENT_Y ) );
          pcPicCurr->M_BUFS( 0, PIC_TRUE_ORIGINAL_INPUT ).getBuf( COMPONENT_Cb ).copyFrom( cPicYuvTrueOrg->getBuf( COMPONENT_Cb ) );
          pcPicCurr->M_BUFS( 0, PIC_TRUE_ORIGINAL_INPUT ).getBuf( COMPONENT_Cr ).copyFrom( cPicYuvTrueOrg->getBuf( COMPONENT_Cr ) );
    
          const ChromaFormat chromaFormatIDC = pSPS->getChromaFormatIdc();
    
          const PPS *refPPS = m_ppsMap.getPS( 0 );
    
          const Window& curScalingWindow = pPPS->getScalingWindow();
          int curPicWidth = pPPS->getPicWidthInLumaSamples() - curScalingWindow.getWindowLeftOffset() - curScalingWindow.getWindowRightOffset();
          int curPicHeight = pPPS->getPicHeightInLumaSamples() - curScalingWindow.getWindowTopOffset() - curScalingWindow.getWindowBottomOffset();
    
          const Window& refScalingWindow = refPPS->getScalingWindow();
          int refPicWidth = refPPS->getPicWidthInLumaSamples() - refScalingWindow.getWindowLeftOffset() - refScalingWindow.getWindowRightOffset();
          int refPicHeight = refPPS->getPicHeightInLumaSamples() - refScalingWindow.getWindowTopOffset() - refScalingWindow.getWindowBottomOffset();
    
          int xScale = ( ( refPicWidth << SCALE_RATIO_BITS ) + ( curPicWidth >> 1 ) ) / curPicWidth;
          int yScale = ( ( refPicHeight << SCALE_RATIO_BITS ) + ( curPicHeight >> 1 ) ) / curPicHeight;
          std::pair<int, int> scalingRatio = std::pair<int, int>( xScale, yScale );
    
          Picture::rescalePicture( scalingRatio, *pcPicYuvOrg, refPPS->getScalingWindow(), pcPicCurr->getOrigBuf(), pPPS->getScalingWindow(), chromaFormatIDC, pSPS->getBitDepths(), true, true,
    
              pSPS->getHorCollocatedChromaFlag(), pSPS->getVerCollocatedChromaFlag() );
    
          Picture::rescalePicture( scalingRatio, *cPicYuvTrueOrg, refPPS->getScalingWindow(), pcPicCurr->getTrueOrigBuf(), pPPS->getScalingWindow(), chromaFormatIDC, pSPS->getBitDepths(), true, true,
    
              pSPS->getHorCollocatedChromaFlag(), pSPS->getVerCollocatedChromaFlag() );
    #else
    
          Picture::rescalePicture( scalingRatio, *pcPicYuvOrg, refPPS->getScalingWindow(), pcPicCurr->getOrigBuf(), pPPS->getScalingWindow(), chromaFormatIDC, pSPS->getBitDepths(), true, true );
          Picture::rescalePicture( scalingRatio, *cPicYuvTrueOrg, refPPS->getScalingWindow(), pcPicCurr->getTrueOrigBuf(), pPPS->getScalingWindow(), chromaFormatIDC, pSPS->getBitDepths(), true, true );
    
          const Window& curWindow = pPPS->getConformanceWindow();
          int curPicWidth = pPPS->getPicWidthInLumaSamples() - ( curWindow.getWindowLeftOffset() + curWindow.getWindowRightOffset() ) * SPS::getWinUnitX( chromaFormatIDC );
          int curPicHeight = pPPS->getPicHeightInLumaSamples() - ( curWindow.getWindowTopOffset() + curWindow.getWindowBottomOffset() ) * SPS::getWinUnitY( chromaFormatIDC );
    
          const Window& refWindow = refPPS->getConformanceWindow();
          int refPicWidth = refPPS->getPicWidthInLumaSamples() - ( refWindow.getWindowLeftOffset() + refWindow.getWindowRightOffset() ) * SPS::getWinUnitX( chromaFormatIDC );
          int refPicHeight = refPPS->getPicHeightInLumaSamples() - ( refWindow.getWindowTopOffset() + refWindow.getWindowBottomOffset() ) * SPS::getWinUnitY( chromaFormatIDC );
    
          int xScale = ( ( refPicWidth << SCALE_RATIO_BITS ) + ( curPicWidth >> 1 ) ) / curPicWidth;
          int yScale = ( ( refPicHeight << SCALE_RATIO_BITS ) + ( curPicHeight >> 1 ) ) / curPicHeight;
          std::pair<int, int> scalingRatio = std::pair<int, int>( xScale, yScale );
    
          Picture::rescalePicture( scalingRatio, *pcPicYuvOrg, refPPS->getConformanceWindow(), pcPicCurr->getOrigBuf(), pPPS->getConformanceWindow(), chromaFormatIDC, pSPS->getBitDepths(), true, true,
    
              pSPS->getHorCollocatedChromaFlag(), pSPS->getVerCollocatedChromaFlag() );
    
          Picture::rescalePicture( scalingRatio, *cPicYuvTrueOrg, refPPS->getConformanceWindow(), pcPicCurr->getTrueOrigBuf(), pPPS->getConformanceWindow(), chromaFormatIDC, pSPS->getBitDepths(), true, true,
    
              pSPS->getHorCollocatedChromaFlag(), pSPS->getVerCollocatedChromaFlag() );
    #else
    
          Picture::rescalePicture( *pcPicYuvOrg, refPPS->getConformanceWindow(), pcPicCurr->getOrigBuf(), pPPS->getConformanceWindow(), chromaFormatIDC, pSPS->getBitDepths(), true, true );
          Picture::rescalePicture( *cPicYuvTrueOrg, refPPS->getConformanceWindow(), pcPicCurr->getTrueOrigBuf(), pPPS->getConformanceWindow(), chromaFormatIDC, pSPS->getBitDepths(), true, true );
    
        else
        {
          pcPicCurr->M_BUFS( 0, PIC_ORIGINAL ).swap( *pcPicYuvOrg );
          pcPicCurr->M_BUFS( 0, PIC_TRUE_ORIGINAL ).swap( *cPicYuvTrueOrg );
    
    Brian Heng's avatar
    Brian Heng committed
    #if JVET_P1006_PICTURE_HEADER
        pcPicCurr->finalInit( *pSPS, *pPPS, &m_picHeader, m_apss, m_lmcsAPS, m_scalinglistAPS );
    #else
    
        pcPicCurr->finalInit( *pSPS, *pPPS, m_apss, m_lmcsAPS, m_scalinglistAPS );