Skip to content
Snippets Groups Projects
UnitTools.cpp 172 KiB
Newer Older
  • Learn to ignore specific revisions
  •     if (mergeCandList.interDirNeighbours[ui] == mergeCandList.interDirNeighbours[mergeCandIndex])
    
        {
          if (mergeCandList.interDirNeighbours[ui] == 3)
          {
    
    Li's avatar
    Li committed
            int offset0 = (ui * 2);
            int offset1 = (mergeCandIndex * 2);
    
            if (mergeCandList.mvFieldNeighbours[offset0].refIdx == mergeCandList.mvFieldNeighbours[offset1].refIdx &&
                mergeCandList.mvFieldNeighbours[offset0 + 1].refIdx == mergeCandList.mvFieldNeighbours[offset1 + 1].refIdx &&
                mergeCandList.mvFieldNeighbours[offset0].mv == mergeCandList.mvFieldNeighbours[offset1].mv &&
                mergeCandList.mvFieldNeighbours[offset0 + 1].mv == mergeCandList.mvFieldNeighbours[offset1 + 1].mv
              )
            {
              hasPruned[ui] = true;
              return true;
            }
          }
          else
          {
    
    Li's avatar
    Li committed
            int offset0 = (ui * 2) + mergeCandList.interDirNeighbours[ui] - 1;
            int offset1 = (mergeCandIndex * 2) + mergeCandList.interDirNeighbours[ui] - 1;
    
            if (mergeCandList.mvFieldNeighbours[offset0].refIdx == mergeCandList.mvFieldNeighbours[offset1].refIdx &&
    
                mergeCandList.mvFieldNeighbours[offset0].mv == mergeCandList.mvFieldNeighbours[offset1].mv
    
    #if JVET_L0090_PAIR_AVG
    
    Xiaozhong Xu's avatar
    Xiaozhong Xu committed
    
    
    bool PU::addMergeHMVPCand(const CodingStructure &cs, MergeCtx& mrgCtx, bool canFastExit, const int& mrgCandIdx, const uint32_t maxNumMergeCandMin1, int &cnt, const int prevCnt, bool isAvailableSubPu, unsigned subPuMvpPos
    
    Yu Han's avatar
    Yu Han committed
      , bool ibcFlag
    
      , bool isShared
    
    Xiaozhong Xu's avatar
    Xiaozhong Xu committed
    )
    
    Xiaozhong Xu's avatar
    Xiaozhong Xu committed
    
    
    bool PU::addMergeHMVPCand(const CodingStructure &cs, MergeCtx& mrgCtx, bool isCandInter[MRG_MAX_NUM_CANDS], bool canFastExit, const int& mrgCandIdx, const uint32_t maxNumMergeCandMin1, int &cnt, const int prevCnt, bool isAvailableSubPu, unsigned subPuMvpPos
    
    Xiaozhong Xu's avatar
    Xiaozhong Xu committed
      , int mmvdList
    )
    
      const Slice& slice = *cs.slice;
    
      MotionInfo miNeighbor;
      bool hasPruned[MRG_MAX_NUM_CANDS];
    
    Li's avatar
    Li committed
      memset(hasPruned, 0, MRG_MAX_NUM_CANDS * sizeof(bool));
    
      if (isAvailableSubPu)
      {
        hasPruned[subPuMvpPos] = true;
      }
    
      auto &lut = ibcFlag ? ( isShared ? cs.motionLut.lutShareIbc : cs.motionLut.lutIbc ) : ( isShared ? cs.motionLut.lutShare : cs.motionLut.lut );
      int num_avai_candInLUT = (int) lut.size();
    
      for (int mrgIdx = 1; mrgIdx <= num_avai_candInLUT; mrgIdx++)
      {
    
        miNeighbor = lut[num_avai_candInLUT - mrgIdx];
    
        mrgCtx.interDirNeighbours[cnt] = miNeighbor.interDir;
        mrgCtx.mvFieldNeighbours[cnt << 1].setMvField(miNeighbor.mv[0], miNeighbor.refIdx[0]);
        if (slice.isInterB())
        {
          mrgCtx.mvFieldNeighbours[(cnt << 1) + 1].setMvField(miNeighbor.mv[1], miNeighbor.refIdx[1]);
        }
    
    Yu Han's avatar
    Yu Han committed
    #if JVET_N0843_BVP_SIMPLIFICATION
        if (mrgIdx > 2 || (mrgIdx > 1 && ibcFlag) || !xCheckSimilarMotion(cnt, prevCnt, mrgCtx, hasPruned))
    #else
    
    Yu Han's avatar
    Yu Han committed
        if (mrgIdx > 2 || !xCheckSimilarMotion(cnt, prevCnt, mrgCtx, hasPruned))
    
    Yu Han's avatar
    Yu Han committed
    #endif
    
        {
    #if !JVET_L0090_PAIR_AVG
          isCandInter[cnt] = true;
    #endif
    
          mrgCtx.GBiIdx[cnt] = (mrgCtx.interDirNeighbours[cnt] == 3) ? miNeighbor.GBiIdx : GBI_DEFAULT;
    
    Yu Han's avatar
    Yu Han committed
          if (mrgCandIdx == cnt && canFastExit)
    
          {
            return true;
          }
          cnt ++;
          if (cnt  == maxNumMergeCandMin1)
          {
            break;
          }
        }
      }
      return false;
    }
    
    Yu Han's avatar
    Yu Han committed
    void PU::getIBCMergeCandidates(const PredictionUnit &pu, MergeCtx& mrgCtx, const int& mrgCandIdx)
    {
      const CodingStructure &cs = *pu.cs;
      const Slice &slice = *pu.cs->slice;
      const uint32_t maxNumMergeCand = slice.getMaxNumMergeCand();
      const bool canFastExit = pu.cs->pps->getLog2ParallelMergeLevelMinus2() == 0;
    
      for (uint32_t ui = 0; ui < maxNumMergeCand; ++ui)
      {
        mrgCtx.GBiIdx[ui] = GBI_DEFAULT;
        mrgCtx.interDirNeighbours[ui] = 0;
        mrgCtx.mrgTypeNeighbours[ui] = MRG_TYPE_IBC;
    
    Yu Han's avatar
    Yu Han committed
        mrgCtx.mvFieldNeighbours[ui * 2].refIdx = NOT_VALID;
        mrgCtx.mvFieldNeighbours[ui * 2 + 1].refIdx = NOT_VALID;
    
    Yu Han's avatar
    Yu Han committed
      }
    
      mrgCtx.numValidMergeCand = maxNumMergeCand;
      // compute the location of the current PU
    
      int cnt = 0;
    
    
    Yu Han's avatar
    Yu Han committed
    #if JVET_N0843_BVP_SIMPLIFICATION==0
    
      const Position posLT = pu.shareParentPos;
    
    Yu Han's avatar
    Yu Han committed
    #endif
    
      const Position posRT = pu.shareParentPos.offset(pu.shareParentSize.width - 1, 0);
      const Position posLB = pu.shareParentPos.offset(0, pu.shareParentSize.height - 1);
    
    Yu Han's avatar
    Yu Han committed
    
      MotionInfo miAbove, miLeft, miAboveLeft, miAboveRight, miBelowLeft;
    
      //left
      const PredictionUnit* puLeft = cs.getPURestricted(posLB.offset(-1, 0), pu, pu.chType);
      const bool isAvailableA1 = puLeft && isDiffMER(pu, *puLeft) && pu.cu != puLeft->cu && CU::isIBC(*puLeft->cu);
      if (isAvailableA1)
      {
        miLeft = puLeft->getMotionInfo(posLB.offset(-1, 0));
    
        // get Inter Dir
        mrgCtx.interDirNeighbours[cnt] = miLeft.interDir;
        // get Mv from Left
        mrgCtx.mvFieldNeighbours[cnt << 1].setMvField(miLeft.mv[0], miLeft.refIdx[0]);
        if (mrgCandIdx == cnt && canFastExit)
        {
          return;
        }
        cnt++;
      }
    
      // early termination
      if (cnt == maxNumMergeCand)
      {
        return;
      }
    
    
      // above
      const PredictionUnit *puAbove = cs.getPURestricted(posRT.offset(0, -1), pu, pu.chType);
      bool isAvailableB1 = puAbove && isDiffMER(pu, *puAbove) && pu.cu != puAbove->cu && CU::isIBC(*puAbove->cu);
      if (isAvailableB1)
      {
        miAbove = puAbove->getMotionInfo(posRT.offset(0, -1));
    
        if (!isAvailableA1 || (miAbove != miLeft))
        {
          // get Inter Dir
          mrgCtx.interDirNeighbours[cnt] = miAbove.interDir;
          // get Mv from Above
          mrgCtx.mvFieldNeighbours[cnt << 1].setMvField(miAbove.mv[0], miAbove.refIdx[0]);
          if (mrgCandIdx == cnt && canFastExit)
          {
            return;
          }
    
          cnt++;
        }
      }
    
      // early termination
      if (cnt == maxNumMergeCand)
      {
        return;
      }
    
    
    Yu Han's avatar
    Yu Han committed
      int spatialCandPos = cnt;
    
    
    Yu Han's avatar
    Yu Han committed
    #if JVET_N0843_BVP_SIMPLIFICATION==0
    
    Yu Han's avatar
    Yu Han committed
      // above right
      const PredictionUnit *puAboveRight = cs.getPURestricted(posRT.offset(1, -1), pu, pu.chType);
      bool isAvailableB0 = puAboveRight && isDiffMER(pu, *puAboveRight) && CU::isIBC(*puAboveRight->cu);
      if (isAvailableB0)
      {
        miAboveRight = puAboveRight->getMotionInfo(posRT.offset(1, -1));
    
    #if HM_JEM_MERGE_CANDS
        if ((!isAvailableB1 || (miAbove != miAboveRight)) && (!isAvailableA1 || (miLeft != miAboveRight)))
    #else
        if (!isAvailableB1 || (miAbove != miAboveRight))
    #endif
        {
          // get Inter Dir
          mrgCtx.interDirNeighbours[cnt] = miAboveRight.interDir;
          // get Mv from Above-right
          mrgCtx.mvFieldNeighbours[cnt << 1].setMvField(miAboveRight.mv[0], miAboveRight.refIdx[0]);
    
          if (mrgCandIdx == cnt && canFastExit)
          {
            return;
          }
    
          cnt++;
        }
      }
      // early termination
      if (cnt == maxNumMergeCand)
      {
        return;
      }
    
      //left bottom
      const PredictionUnit *puLeftBottom = cs.getPURestricted(posLB.offset(-1, 1), pu, pu.chType);
      bool isAvailableA0 = puLeftBottom && isDiffMER(pu, *puLeftBottom) && CU::isIBC(*puLeftBottom->cu);
      if (isAvailableA0)
      {
        miBelowLeft = puLeftBottom->getMotionInfo(posLB.offset(-1, 1));
    
    #if HM_JEM_MERGE_CANDS
        if ((!isAvailableA1 || (miBelowLeft != miLeft)) && (!isAvailableB1 || (miBelowLeft != miAbove)) && (!isAvailableB0 || (miBelowLeft != miAboveRight)))
    #else
        if (!isAvailableA1 || (miBelowLeft != miLeft))
    #endif
        {
          // get Inter Dir
          mrgCtx.interDirNeighbours[cnt] = miBelowLeft.interDir;
          mrgCtx.mvFieldNeighbours[cnt << 1].setMvField(miBelowLeft.mv[0], miBelowLeft.refIdx[0]);
          if (mrgCandIdx == cnt && canFastExit)
          {
            return;
          }
    
          cnt++;
        }
      }
      // early termination
      if (cnt == maxNumMergeCand)
      {
        return;
      }
    
      // above left
      if (cnt < 4)
      {
        const PredictionUnit *puAboveLeft = cs.getPURestricted(posLT.offset(-1, -1), pu, pu.chType);
        bool isAvailableB2 = puAboveLeft && isDiffMER(pu, *puAboveLeft) && CU::isIBC(*puAboveLeft->cu);
        if (isAvailableB2)
        {
          miAboveLeft = puAboveLeft->getMotionInfo(posLT.offset(-1, -1));
    
    #if HM_JEM_MERGE_CANDS
          if ((!isAvailableA1 || (miLeft != miAboveLeft)) && (!isAvailableB1 || (miAbove != miAboveLeft)) && (!isAvailableA0 || (miBelowLeft != miAboveLeft)) && (!isAvailableB0 || (miAboveRight != miAboveLeft)))
    #else
          if ((!isAvailableA1 || (miLeft != miAboveLeft)) && (!isAvailableB1 || (miAbove != miAboveLeft)))
    #endif
          {
            // get Inter Dir
            mrgCtx.interDirNeighbours[cnt] = miAboveLeft.interDir;
            mrgCtx.mvFieldNeighbours[cnt << 1].setMvField(miAboveLeft.mv[0], miAboveLeft.refIdx[0]);
            if (mrgCandIdx == cnt && canFastExit)
            {
              return;
            }
    
            cnt++;
          }
        }
      }
      // early termination
      if (cnt == maxNumMergeCand)
      {
        return;
      }
    
    Yu Han's avatar
    Yu Han committed
    #endif
    
    Yu Han's avatar
    Yu Han committed
    #if JVET_N0843_BVP_SIMPLIFICATION
      int maxNumMergeCandMin1 = maxNumMergeCand;
    #else
    
    Yu Han's avatar
    Yu Han committed
      int maxNumMergeCandMin1 = maxNumMergeCand - 1;
    
    Yu Han's avatar
    Yu Han committed
    #endif
    
    Yu Han's avatar
    Yu Han committed
      if (cnt != maxNumMergeCandMin1)
      {
        bool isAvailableSubPu = false;
        unsigned subPuMvpPos = 0;
    
    
        bool  isShared = ((pu.Y().lumaSize().width != pu.shareParentSize.width) || (pu.Y().lumaSize().height != pu.shareParentSize.height));
    
    
    Yu Han's avatar
    Yu Han committed
    #if JVET_L0090_PAIR_AVG
    
        bool bFound = addMergeHMVPCand(cs, mrgCtx, canFastExit
    
    Yu Han's avatar
    Yu Han committed
          , mrgCandIdx
    
    Yu Han's avatar
    Yu Han committed
          , maxNumMergeCandMin1, cnt
          , spatialCandPos
          , isAvailableSubPu, subPuMvpPos
    
    Yu Han's avatar
    Yu Han committed
          , true
    
    Yu Han's avatar
    Yu Han committed
        );
    #else
        bool bFound = addMergeHMVPCand(slice, mrgCtx, isCandInter, canFastExit
          , mrgCandIdx
          , maxNumMergeCandMin1, cnt, cnt, isAvailableSubPu, subPuMvpPos
        );
    #endif
        if (bFound)
        {
          return;
        }
      }
    
    
    Yu Han's avatar
    Yu Han committed
    #if JVET_L0090_PAIR_AVG && JVET_N0843_BVP_SIMPLIFICATION==0
    
    Yu Han's avatar
    Yu Han committed
      // pairwise-average candidates
    
    Yu Han's avatar
    Yu Han committed
        if (cnt>1 && cnt <maxNumMergeCand)
        {
           mrgCtx.mvFieldNeighbours[cnt * 2    ].setMvField(Mv(0, 0), NOT_VALID);
           mrgCtx.mvFieldNeighbours[cnt * 2 + 1].setMvField(Mv(0, 0), NOT_VALID);
    
    Yu Han's avatar
    Yu Han committed
           const Mv& MvI = mrgCtx.mvFieldNeighbours[0 * 2].mv;
           const Mv& MvJ = mrgCtx.mvFieldNeighbours[1 * 2].mv;
           // average two MVs
           Mv avgMv = MvI;
    
    Yu Han's avatar
    Yu Han committed
           avgMv += MvJ;
           mrgCtx.mrgTypeNeighbours[cnt] = MRG_TYPE_IBC;
           roundAffineMv(avgMv.hor, avgMv.ver, 1);
    
    Yu Han's avatar
    Yu Han committed
           avgMv.roundToPrecision(MV_PRECISION_INTERNAL, MV_PRECISION_INT);
    
    Yu Han's avatar
    Yu Han committed
          mrgCtx.mvFieldNeighbours[cnt * 2 ].setMvField(avgMv, MAX_NUM_REF);
          mrgCtx.interDirNeighbours[cnt] = 1;
          cnt++;
    
    Yu Han's avatar
    Yu Han committed
        }
    
        // early termination
        if (cnt == maxNumMergeCand)
        {
          return;
        }
    #endif
    
    
    Yu Han's avatar
    Yu Han committed
    #if JVET_N0317_ADD_ZERO_BV
        while (cnt < maxNumMergeCand)
        {
          mrgCtx.mvFieldNeighbours[cnt * 2].setMvField(Mv(0, 0), MAX_NUM_REF);
          mrgCtx.interDirNeighbours[cnt] = 1;
          cnt++;
          if (mrgCandIdx == cnt && canFastExit)
          {
            return;
          }
        }
    #endif
    
    
    Yu Han's avatar
    Yu Han committed
      mrgCtx.numValidMergeCand = cnt;
    
    }
    
    
    void PU::getInterMergeCandidates( const PredictionUnit &pu, MergeCtx& mrgCtx,
    
                                     int mmvdList,
                                     const int& mrgCandIdx )
    
    {
      const CodingStructure &cs  = *pu.cs;
      const Slice &slice         = *pu.cs->slice;
      const uint32_t maxNumMergeCand = slice.getMaxNumMergeCand();
      const bool canFastExit     = pu.cs->pps->getLog2ParallelMergeLevelMinus2() == 0;
    
    
    #if !JVET_L0090_PAIR_AVG
      // this variable is unused if remove HEVC combined candidates
    
    Yu-Chi Su's avatar
    Yu-Chi Su committed
    #endif
    
        mrgCtx.GBiIdx[ui] = GBI_DEFAULT;
    
        mrgCtx.interDirNeighbours[ui] = 0;
        mrgCtx.mrgTypeNeighbours [ui] = MRG_TYPE_DEFAULT_N;
        mrgCtx.mvFieldNeighbours[(ui << 1)    ].refIdx = NOT_VALID;
        mrgCtx.mvFieldNeighbours[(ui << 1) + 1].refIdx = NOT_VALID;
      }
    
      mrgCtx.numValidMergeCand = maxNumMergeCand;
      // compute the location of the current PU
    
      int cnt = 0;
    
      const Position posLT = pu.shareParentPos;
      const Position posRT = pu.shareParentPos.offset(pu.shareParentSize.width - 1, 0);
      const Position posLB = pu.shareParentPos.offset(0, pu.shareParentSize.height - 1);
    
      MotionInfo miAbove, miLeft, miAboveLeft, miAboveRight, miBelowLeft;
    
      //left
      const PredictionUnit* puLeft = cs.getPURestricted( posLB.offset( -1, 0 ), pu, pu.chType );
    
      const bool isAvailableA1 = puLeft && isDiffMER( pu, *puLeft ) && pu.cu != puLeft->cu && CU::isInter( *puLeft->cu );
    
      if( isAvailableA1 )
      {
        miLeft = puLeft->getMotionInfo( posLB.offset(-1, 0) );
    
    
    
        // get Inter Dir
        mrgCtx.interDirNeighbours[cnt] = miLeft.interDir;
    
        mrgCtx.GBiIdx[cnt] = (mrgCtx.interDirNeighbours[cnt] == 3) ? puLeft->cu->GBiIdx : GBI_DEFAULT;
    
        // get Mv from Left
        mrgCtx.mvFieldNeighbours[cnt << 1].setMvField(miLeft.mv[0], miLeft.refIdx[0]);
    
        if (slice.isInterB())
        {
          mrgCtx.mvFieldNeighbours[(cnt << 1) + 1].setMvField(miLeft.mv[1], miLeft.refIdx[1]);
        }
    
    Yu Han's avatar
    Yu Han committed
        if (mrgCandIdx == cnt && canFastExit)
    
        {
          return;
        }
    
        cnt++;
      }
    
      // early termination
      if (cnt == maxNumMergeCand)
      {
        return;
      }
    
    
      // above
      const PredictionUnit *puAbove = cs.getPURestricted( posRT.offset( 0, -1 ), pu, pu.chType );
    
      bool isAvailableB1 = puAbove && isDiffMER( pu, *puAbove ) && pu.cu != puAbove->cu && CU::isInter( *puAbove->cu );
    
      if( isAvailableB1 )
      {
        miAbove = puAbove->getMotionInfo( posRT.offset( 0, -1 ) );
    
        if( !isAvailableA1 || ( miAbove != miLeft ) )
        {
    
    
          // get Inter Dir
          mrgCtx.interDirNeighbours[cnt] = miAbove.interDir;
    
          mrgCtx.GBiIdx[cnt] = (mrgCtx.interDirNeighbours[cnt] == 3) ? puAbove->cu->GBiIdx : GBI_DEFAULT;
    
          mrgCtx.mvFieldNeighbours[cnt << 1].setMvField( miAbove.mv[0], miAbove.refIdx[0] );
    
    
          if( slice.isInterB() )
          {
            mrgCtx.mvFieldNeighbours[( cnt << 1 ) + 1].setMvField( miAbove.mv[1], miAbove.refIdx[1] );
          }
    
    Yu Han's avatar
    Yu Han committed
          if (mrgCandIdx == cnt && canFastExit)
    
    Yu Han's avatar
    Yu Han committed
      int spatialCandPos = cnt;
    
    
      // above right
      const PredictionUnit *puAboveRight = cs.getPURestricted( posRT.offset( 1, -1 ), pu, pu.chType );
    
      bool isAvailableB0 = puAboveRight && isDiffMER( pu, *puAboveRight ) && CU::isInter( *puAboveRight->cu );
    
      if( isAvailableB0 )
      {
        miAboveRight = puAboveRight->getMotionInfo( posRT.offset( 1, -1 ) );
    
    #if HM_JEM_MERGE_CANDS
        if( ( !isAvailableB1 || ( miAbove != miAboveRight ) ) && ( !isAvailableA1 || ( miLeft != miAboveRight ) ) )
    #else
        if( !isAvailableB1 || ( miAbove != miAboveRight ) )
    #endif
        {
    
    
          // get Inter Dir
          mrgCtx.interDirNeighbours[cnt] = miAboveRight.interDir;
    
          mrgCtx.GBiIdx[cnt] = (mrgCtx.interDirNeighbours[cnt] == 3) ? puAboveRight->cu->GBiIdx : GBI_DEFAULT;
    
          mrgCtx.mvFieldNeighbours[cnt << 1].setMvField( miAboveRight.mv[0], miAboveRight.refIdx[0] );
    
    
          if( slice.isInterB() )
          {
            mrgCtx.mvFieldNeighbours[( cnt << 1 ) + 1].setMvField( miAboveRight.mv[1], miAboveRight.refIdx[1] );
          }
    
    
    Yu Han's avatar
    Yu Han committed
          if (mrgCandIdx == cnt && canFastExit)
    
          {
            return;
          }
    
          cnt++;
        }
      }
      // early termination
      if( cnt == maxNumMergeCand )
      {
        return;
      }
    
      //left bottom
      const PredictionUnit *puLeftBottom = cs.getPURestricted( posLB.offset( -1, 1 ), pu, pu.chType );
    
      bool isAvailableA0 = puLeftBottom && isDiffMER( pu, *puLeftBottom ) && CU::isInter( *puLeftBottom->cu );
    
      if( isAvailableA0 )
      {
        miBelowLeft = puLeftBottom->getMotionInfo( posLB.offset( -1, 1 ) );
    
    #if HM_JEM_MERGE_CANDS
        if( ( !isAvailableA1 || ( miBelowLeft != miLeft ) ) && ( !isAvailableB1 || ( miBelowLeft != miAbove ) ) && ( !isAvailableB0 || ( miBelowLeft != miAboveRight ) ) )
    #else
        if( !isAvailableA1 || ( miBelowLeft != miLeft ) )
    #endif
        {
    
    
          // get Inter Dir
          mrgCtx.interDirNeighbours[cnt] = miBelowLeft.interDir;
    
          mrgCtx.GBiIdx[cnt] = (mrgCtx.interDirNeighbours[cnt] == 3) ? puLeftBottom->cu->GBiIdx : GBI_DEFAULT;
    
          // get Mv from Bottom-Left
          mrgCtx.mvFieldNeighbours[cnt << 1].setMvField( miBelowLeft.mv[0], miBelowLeft.refIdx[0] );
    
          if( slice.isInterB() )
          {
            mrgCtx.mvFieldNeighbours[( cnt << 1 ) + 1].setMvField( miBelowLeft.mv[1], miBelowLeft.refIdx[1] );
          }
    
    
    Yu Han's avatar
    Yu Han committed
          if (mrgCandIdx == cnt && canFastExit)
    
      {
        const PredictionUnit *puAboveLeft = cs.getPURestricted( posLT.offset( -1, -1 ), pu, pu.chType );
    
        bool isAvailableB2 = puAboveLeft && isDiffMER( pu, *puAboveLeft ) && CU::isInter( *puAboveLeft->cu );
    
        if( isAvailableB2 )
        {
          miAboveLeft = puAboveLeft->getMotionInfo( posLT.offset( -1, -1 ) );
    
    #if HM_JEM_MERGE_CANDS
          if( ( !isAvailableA1 || ( miLeft != miAboveLeft ) ) && ( !isAvailableB1 || ( miAbove != miAboveLeft ) ) && ( !isAvailableA0 || ( miBelowLeft != miAboveLeft ) ) && ( !isAvailableB0 || ( miAboveRight != miAboveLeft ) ) )
    #else
          if( ( !isAvailableA1 || ( miLeft != miAboveLeft ) ) && ( !isAvailableB1 || ( miAbove != miAboveLeft ) ) )
    #endif
          {
    
    
            // get Inter Dir
            mrgCtx.interDirNeighbours[cnt] = miAboveLeft.interDir;
    
            mrgCtx.GBiIdx[cnt] = (mrgCtx.interDirNeighbours[cnt] == 3) ? puAboveLeft->cu->GBiIdx : GBI_DEFAULT;
    
            // get Mv from Above-Left
            mrgCtx.mvFieldNeighbours[cnt << 1].setMvField( miAboveLeft.mv[0], miAboveLeft.refIdx[0] );
    
            if( slice.isInterB() )
            {
              mrgCtx.mvFieldNeighbours[( cnt << 1 ) + 1].setMvField( miAboveLeft.mv[1], miAboveLeft.refIdx[1] );
            }
    
    
    Yu Han's avatar
    Yu Han committed
            if (mrgCandIdx == cnt && canFastExit)
    
            {
              return;
            }
    
            cnt++;
          }
        }
      }
      // early termination
      if (cnt == maxNumMergeCand)
      {
        return;
      }
    
      if (slice.getEnableTMVPFlag())
      {
        //>> MTK colocated-RightBottom
        // offset the pos to be sure to "point" to the same position the uiAbsPartIdx would've pointed to
    
        Position posRB = pu.shareParentPos.offset(pu.shareParentSize.width-3, pu.shareParentSize.height - 3);
    
        Position posC1 = pu.shareParentPos.offset((pu.shareParentSize.width/2), (pu.shareParentSize.height/2));
    
    
        bool C1Avail = (posC1.x < pcv.lumaWidth) && (posC1.y  < pcv.lumaHeight);
    
    
        if (((posRB.x + pcv.minCUWidth) < pcv.lumaWidth) && ((posRB.y + pcv.minCUHeight) < pcv.lumaHeight))
        {
          {
            Position posInCtu( posRB.x & pcv.maxCUWidthMask, posRB.y & pcv.maxCUHeightMask );
    
            if( ( posInCtu.x + 4 < pcv.maxCUWidth ) &&           // is not at the last column of CTU
                ( posInCtu.y + 4 < pcv.maxCUHeight ) )           // is not at the last row    of CTU
            {
              posC0 = posRB.offset( 4, 4 );
              C0Avail = true;
            }
            else if( posInCtu.x + 4 < pcv.maxCUWidth )           // is not at the last column of CTU But is last row of CTU
            {
              posC0 = posRB.offset( 4, 4 );
              // in the reference the CTU address is not set - thus probably resulting in no using this C0 possibility
            }
            else if( posInCtu.y + 4 < pcv.maxCUHeight )          // is not at the last row of CTU But is last column of CTU
            {
              posC0 = posRB.offset( 4, 4 );
              C0Avail = true;
            }
            else //is the right bottom corner of CTU
            {
              posC0 = posRB.offset( 4, 4 );
              // same as for last column but not last row
            }
          }
        }
    
        Mv        cColMv;
        int       iRefIdx     = 0;
        int       dir         = 0;
        unsigned  uiArrayAddr = cnt;
        bool      bExistMV    = ( C0Avail && getColocatedMVP(pu, REF_PIC_LIST_0, posC0, cColMv, iRefIdx ) )
    
                                          || ( C1Avail && getColocatedMVP(pu, REF_PIC_LIST_0, posC1, cColMv, iRefIdx ));
    
    
        if (bExistMV)
        {
          dir     |= 1;
          mrgCtx.mvFieldNeighbours[2 * uiArrayAddr].setMvField(cColMv, iRefIdx);
        }
    
        if (slice.isInterB())
        {
          bExistMV = ( C0Avail && getColocatedMVP(pu, REF_PIC_LIST_1, posC0, cColMv, iRefIdx ) )
    
                               || (C1Avail &&  getColocatedMVP(pu, REF_PIC_LIST_1, posC1, cColMv, iRefIdx ) );
    
          if (bExistMV)
          {
            dir     |= 2;
            mrgCtx.mvFieldNeighbours[2 * uiArrayAddr + 1].setMvField(cColMv, iRefIdx);
          }
        }
    
        if( dir != 0 )
        {
    
          for( int i = 0; i < iSpanCand; i++ )
          {
            if( mrgCtx.interDirNeighbours[  i           ] == dir &&
                mrgCtx.mvFieldNeighbours [  i << 1      ] == mrgCtx.mvFieldNeighbours[  uiArrayAddr << 1      ] &&
                mrgCtx.mvFieldNeighbours [( i << 1 ) + 1] == mrgCtx.mvFieldNeighbours[( uiArrayAddr << 1 ) + 1] )
            {
              addTMvp = false;
            }
          }
    #endif
          if( addTMvp )
          {
            mrgCtx.interDirNeighbours[uiArrayAddr] = dir;
    
            mrgCtx.GBiIdx[uiArrayAddr] = GBI_DEFAULT;
    
    Yu Han's avatar
    Yu Han committed
            if (mrgCandIdx == cnt && canFastExit)
    
      int maxNumMergeCandMin1 = maxNumMergeCand - 1;
      if (cnt != maxNumMergeCandMin1)
      {
    
        bool isAvailableSubPu = false;
        unsigned subPuMvpPos = 0;
    
    #if JVET_L0090_PAIR_AVG
    
        bool  isShared = ((pu.Y().lumaSize().width != pu.shareParentSize.width) || (pu.Y().lumaSize().height != pu.shareParentSize.height));
    
        bool bFound = addMergeHMVPCand(cs, mrgCtx, canFastExit
    
    Yu Han's avatar
    Yu Han committed
          , mrgCandIdx
    
    Yu Han's avatar
    Yu Han committed
          , maxNumMergeCandMin1, cnt
          , spatialCandPos
          , isAvailableSubPu, subPuMvpPos
    
    Yu Han's avatar
    Yu Han committed
          , CU::isIBC(*pu.cu)
    
    Xiaozhong Xu's avatar
    Xiaozhong Xu committed
        );
    #else
        bool bFound = addMergeHMVPCand(slice, mrgCtx, isCandInter, canFastExit
    
    Yu Han's avatar
    Yu Han committed
          , (mmvdList != 0 && mrgCandIdx != -1) ? (const int)mrgCandIdxIBC : mrgCandIdx
    
    Xiaozhong Xu's avatar
    Xiaozhong Xu committed
          , maxNumMergeCandMin1, cnt, cnt, isAvailableSubPu, subPuMvpPos
          , mmvdList
        );
    
    Yu Han's avatar
    Yu Han committed
        if (cnt > 1 && cnt < maxNumMergeCand)
    
        {
    
          mrgCtx.mvFieldNeighbours[cnt * 2].setMvField( Mv( 0, 0 ), NOT_VALID );
          mrgCtx.mvFieldNeighbours[cnt * 2 + 1].setMvField( Mv( 0, 0 ), NOT_VALID );
          // calculate average MV for L0 and L1 seperately
          unsigned char interDir = 0;
    
          for( int refListId = 0; refListId < (slice.isInterB() ? 2 : 1); refListId++ )
          {
    
            const short refIdxI = mrgCtx.mvFieldNeighbours[0 * 2 + refListId].refIdx;
            const short refIdxJ = mrgCtx.mvFieldNeighbours[1 * 2 + refListId].refIdx;
    
    
            // both MVs are invalid, skip
            if( (refIdxI == NOT_VALID) && (refIdxJ == NOT_VALID) )
            {
              continue;
            }
    
            interDir += 1 << refListId;
            // both MVs are valid, average these two MVs
            if( (refIdxI != NOT_VALID) && (refIdxJ != NOT_VALID) )
            {
    
              const Mv& MvI = mrgCtx.mvFieldNeighbours[0 * 2 + refListId].mv;
              const Mv& MvJ = mrgCtx.mvFieldNeighbours[1 * 2 + refListId].mv;
    
              roundAffineMv(avgMv.hor, avgMv.ver, 1);
    
    Xiaozhong Xu's avatar
    Xiaozhong Xu committed
    
    
              mrgCtx.mvFieldNeighbours[cnt * 2 + refListId].setMvField( avgMv, refIdxI );
            }
            // only one MV is valid, take the only one MV
            else if( refIdxI != NOT_VALID )
            {
    
              Mv singleMv = mrgCtx.mvFieldNeighbours[0 * 2 + refListId].mv;
    
              mrgCtx.mvFieldNeighbours[cnt * 2 + refListId].setMvField( singleMv, refIdxI );
            }
            else if( refIdxJ != NOT_VALID )
            {
    
              Mv singleMv = mrgCtx.mvFieldNeighbours[1 * 2 + refListId].mv;
    
              mrgCtx.mvFieldNeighbours[cnt * 2 + refListId].setMvField( singleMv, refIdxJ );
            }
          }
    
          mrgCtx.interDirNeighbours[cnt] = interDir;
          if( interDir > 0 )
          {
            cnt++;
          }
        }
    
        // early termination
        if( cnt == maxNumMergeCand )
        {
          return;
        }
      }
    #endif
    
    
      uint32_t uiCutoff    = std::min( uiArrayAddr, 3u );
    
      if (slice.isInterB())
      {
        static const uint32_t NUM_PRIORITY_LIST = 12;
        static const uint32_t uiPriorityList0[NUM_PRIORITY_LIST] = { 0 , 1, 0, 2, 1, 2, 0, 3, 1, 3, 2, 3 };
        static const uint32_t uiPriorityList1[NUM_PRIORITY_LIST] = { 1 , 0, 2, 0, 2, 1, 3, 0, 3, 1, 3, 2 };
    
        for (int idx = 0; idx < uiCutoff * (uiCutoff - 1) && uiArrayAddr != maxNumMergeCand; idx++)
        {
          CHECK( idx >= NUM_PRIORITY_LIST, "Invalid priority list number" );
          int i = uiPriorityList0[idx];
          int j = uiPriorityList1[idx];
          if (isCandInter[i] && isCandInter[j] && (mrgCtx.interDirNeighbours[i] & 0x1) && (mrgCtx.interDirNeighbours[j] & 0x2))
          {
            isCandInter[uiArrayAddr] = true;
            mrgCtx.interDirNeighbours[uiArrayAddr] = 3;
    
            mrgCtx.GBiIdx[uiArrayAddr] = ((mrgCtx.interDirNeighbours[uiArrayAddr] == 3)) ? CU::deriveGbiIdx(mrgCtx.GBiIdx[i], mrgCtx.GBiIdx[j]) : GBI_DEFAULT;
    
    
            // get Mv from cand[i] and cand[j]
            mrgCtx.mvFieldNeighbours[ uiArrayAddr << 1     ].setMvField(mrgCtx.mvFieldNeighbours[ i << 1     ].mv, mrgCtx.mvFieldNeighbours[ i << 1     ].refIdx);
            mrgCtx.mvFieldNeighbours[(uiArrayAddr << 1) + 1].setMvField(mrgCtx.mvFieldNeighbours[(j << 1) + 1].mv, mrgCtx.mvFieldNeighbours[(j << 1) + 1].refIdx);
    
            int iRefPOCL0 = slice.getRefPOC(REF_PIC_LIST_0, mrgCtx.mvFieldNeighbours[(uiArrayAddr << 1)    ].refIdx);
            int iRefPOCL1 = slice.getRefPOC(REF_PIC_LIST_1, mrgCtx.mvFieldNeighbours[(uiArrayAddr << 1) + 1].refIdx);
    
            if( iRefPOCL0 == iRefPOCL1 && mrgCtx.mvFieldNeighbours[( uiArrayAddr << 1 )].mv == mrgCtx.mvFieldNeighbours[( uiArrayAddr << 1 ) + 1].mv )
            {
              isCandInter[uiArrayAddr] = false;
            }
            else
            {
              uiArrayAddr++;
            }
          }
        }
      }
    
      // early termination
      if (uiArrayAddr == maxNumMergeCand)
      {
        return;
      }
    
    
      int iNumRefIdx = slice.isInterB() ? std::min(slice.getNumRefIdx(REF_PIC_LIST_0), slice.getNumRefIdx(REF_PIC_LIST_1)) : slice.getNumRefIdx(REF_PIC_LIST_0);
    
      int r = 0;
      int refcnt = 0;
      while (uiArrayAddr < maxNumMergeCand)
      {
    
        mrgCtx.interDirNeighbours [uiArrayAddr     ] = 1;
    
        mrgCtx.GBiIdx             [uiArrayAddr     ] = GBI_DEFAULT;
    
        mrgCtx.mvFieldNeighbours  [uiArrayAddr << 1].setMvField(Mv(0, 0), r);
    
        if (slice.isInterB())
        {
          mrgCtx.interDirNeighbours [ uiArrayAddr          ] = 3;
          mrgCtx.mvFieldNeighbours  [(uiArrayAddr << 1) + 1].setMvField(Mv(0, 0), r);
        }
    
    
    Xiaozhong Xu's avatar
    Xiaozhong Xu committed
        if ( mrgCtx.interDirNeighbours[uiArrayAddr] == 1 && pu.cs->slice->getRefPic(REF_PIC_LIST_0, mrgCtx.mvFieldNeighbours[uiArrayAddr << 1].refIdx)->getPOC() == pu.cs->slice->getPOC())
        {
    
    Yu Han's avatar
    Yu Han committed
          mrgCtx.mrgTypeNeighbours[uiArrayAddr] = MRG_TYPE_IBC;
    
        uiArrayAddr++;
    
        if (refcnt == iNumRefIdx - 1)
        {
          r = 0;
        }
        else
        {
          ++r;
          ++refcnt;
        }
      }
      mrgCtx.numValidMergeCand = uiArrayAddr;
    }
    
    bool PU::checkDMVRCondition(const PredictionUnit& pu)
    {
    
      if (pu.cs->sps->getUseDMVR())
    
      {
        return pu.mergeFlag
          && pu.mergeType == MRG_TYPE_DEFAULT_N
          && !pu.cu->affine
          && !pu.mmvdMergeFlag
          && !pu.cu->mmvdSkip
          && PU::isBiPredFromDifferentDirEqDistPoc(pu)
          && (pu.lheight() >= 8)
          && ((pu.lheight() * pu.lwidth()) >= 64)
          ;
      }
      else
      {
        return false;
      }
    }
    
    Yu Han's avatar
    Yu Han committed
    // for ibc pu validation
    
    Xiaozhong Xu's avatar
    Xiaozhong Xu committed
    bool PU::isBlockVectorValid(PredictionUnit& pu, int xPos, int yPos, int width, int height, int picWidth, int picHeight, int xStartInCU, int yStartInCU, int xBv, int yBv, int ctuSize)
    {
      const int ctuSizeLog2 = g_aucLog2[ctuSize];
    
      int refRightX = xPos + xBv + width - 1;
      int refBottomY = yPos + yBv + height - 1;
    
    Xiaozhong Xu's avatar
    Xiaozhong Xu committed
      int refLeftX = xPos + xBv;
      int refTopY = yPos + yBv;
    
      if ((xPos + xBv) < 0)
      {
        return false;
      }
      if (refRightX >= picWidth)
      {
        return false;
      }
    
      if ((yPos + yBv) < 0)
      {
        return false;
      }
      if (refBottomY >= picHeight)
      {
        return false;
      }
      if ((xBv + width) > 0 && (yBv + height) > 0)
      {
        return false;
      }
    
      // cannot be in the above CTU row
      if (refTopY >> ctuSizeLog2 < yPos >> ctuSizeLog2)
        return false;
    
      // cannot be in the below CTU row
      if (refBottomY >> ctuSizeLog2 > yPos >> ctuSizeLog2)
      {
        return false;
      }
    
      // in the same CTU line
    
    #if JVET_N0175_N0251_N0384_IBC_SMALL_CTU
      int numLeftCTUs = (1 << ((7 - ctuSizeLog2) << 1)) - ((ctuSizeLog2 < 7) ? 1 : 0);
      if ((refRightX >> ctuSizeLog2 <= xPos >> ctuSizeLog2) && (refLeftX >> ctuSizeLog2 >= (xPos >> ctuSizeLog2) - numLeftCTUs))
    #else
    
      if ((refRightX >> ctuSizeLog2 <= xPos >> ctuSizeLog2) && (refLeftX >> ctuSizeLog2 >= (xPos >> ctuSizeLog2) - 1))
    
    Xiaozhong Xu's avatar
    Xiaozhong Xu committed
      {
    
    
        // in the same CTU, or left CTU
        // if part of ref block is in the left CTU, some area can be referred from the not-yet updated local CTU buffer
    
    #if JVET_N0175_N0251_N0384_IBC_SMALL_CTU
        if (((refLeftX >> ctuSizeLog2) == ((xPos >> ctuSizeLog2) - 1)) && (ctuSizeLog2 == 7))
    #else
    
        if ((refLeftX >> ctuSizeLog2) == ((xPos >> ctuSizeLog2) - 1))
    
        {
          // ref block's collocated block in current CTU
          const Position refPosCol = pu.Y().topLeft().offset(xBv + ctuSize, yBv);
    
    Xiaozhong Xu's avatar
    Xiaozhong Xu committed
          int offset64x = (refPosCol.x >> (ctuSizeLog2 - 1)) << (ctuSizeLog2 - 1);
          int offset64y = (refPosCol.y >> (ctuSizeLog2 - 1)) << (ctuSizeLog2 - 1);
          const Position refPosCol64x64 = {offset64x, offset64y};
    
          if (pu.cs->isDecomp(refPosCol64x64, toChannelType(COMPONENT_Y)))
            return false;
    
    Xiaozhong Xu's avatar
    Xiaozhong Xu committed
    #if JVET_N0383_N0251_IBC_COL_VPDU_REMOVE
    
          if (refPosCol64x64 == pu.Y().topLeft())
            return false;
    #endif
    
    Xiaozhong Xu's avatar
    Xiaozhong Xu committed
      }
      else
        return false;
    
    
      // in the same CTU, or valid area from left CTU. Check if the reference block is already coded
      const Position refPosLT = pu.Y().topLeft().offset(xBv, yBv);
      const Position refPosBR = pu.Y().bottomRight().offset(xBv, yBv);
      const ChannelType      chType = toChannelType(COMPONENT_Y);
      if (!pu.cs->isDecomp(refPosBR, chType))
        return false;
      if (!pu.cs->isDecomp(refPosLT, chType))
        return false;
      return true;
    
    
    Yu Han's avatar
    Yu Han committed
    }// for ibc pu validation
    
    
    static int xGetDistScaleFactor(const int &iCurrPOC, const int &iCurrRefPOC, const int &iColPOC, const int &iColRefPOC)
    {
      int iDiffPocD = iColPOC - iColRefPOC;
      int iDiffPocB = iCurrPOC - iCurrRefPOC;