diff --git a/source/Lib/CommonLib/CodingStructure.h b/source/Lib/CommonLib/CodingStructure.h
index c06237b39c3c5c5978ffa11d6a9ae78973882f00..a82aa1a1e38107ed5538e8b11ebea0b46dc3354e 100644
--- a/source/Lib/CommonLib/CodingStructure.h
+++ b/source/Lib/CommonLib/CodingStructure.h
@@ -195,6 +195,10 @@ public:
LutMotionCand motionLut;
+#if JVET_O1170_CHECK_BV_AT_DECODER
+ bool resetIBCBuffer;
+#endif
+
void addMiToLut(static_vector<MotionInfo, MAX_NUM_HMVP_CANDS>& lut, const MotionInfo &mi);
private:
diff --git a/source/Lib/CommonLib/InterPrediction.cpp b/source/Lib/CommonLib/InterPrediction.cpp
index 92499e81b5631684dc5ff1e13f44c8248fa446b2..754f3db972ba51f2512a607bd3ec3760446d93a0 100644
--- a/source/Lib/CommonLib/InterPrediction.cpp
+++ b/source/Lib/CommonLib/InterPrediction.cpp
@@ -151,9 +151,16 @@ void InterPrediction::destroy()
xFree(m_cRefSamplesDMVRL1[ch]);
m_cRefSamplesDMVRL1[ch] = nullptr;
}
+#if JVET_O1170_IBC_VIRTUAL_BUFFER
+ m_IBCBuffer.destroy();
+#endif
}
+#if JVET_O1170_IBC_VIRTUAL_BUFFER
+void InterPrediction::init( RdCost* pcRdCost, ChromaFormat chromaFormatIDC, const int ctuSize )
+#else
void InterPrediction::init( RdCost* pcRdCost, ChromaFormat chromaFormatIDC )
+#endif
{
m_pcRdCost = pcRdCost;
@@ -219,6 +226,13 @@ void InterPrediction::init( RdCost* pcRdCost, ChromaFormat chromaFormatIDC )
const int MVBUFFER_SIZE = MAX_CU_SIZE / MIN_PU_SIZE;
m_storedMv = new Mv[MVBUFFER_SIZE*MVBUFFER_SIZE];
}
+#if JVET_O1170_IBC_VIRTUAL_BUFFER
+ if (m_IBCBuffer.bufs.empty())
+ {
+ m_IBCBufferWidth = 128 * 128 / ctuSize;
+ m_IBCBuffer.create(UnitArea(chromaFormatIDC, Area(0, 0, m_IBCBufferWidth, ctuSize)));
+ }
+#endif
}
// ====================================================================================================================
@@ -1642,6 +1656,22 @@ void InterPrediction::motionCompensation( PredictionUnit &pu, PelUnitBuf &predBu
CHECK(predBufWOBIO && pu.mhIntraFlag, "the case should not happen!");
#endif
+#if JVET_O1170_IBC_VIRTUAL_BUFFER
+ if (!pu.cs->pcv->isEncoder)
+ {
+ if (CU::isIBC(*pu.cu))
+ {
+ CHECK(!luma, "IBC only for Chroma is not allowed.");
+ xIntraBlockCopy(pu, predBuf, COMPONENT_Y);
+ if (chroma)
+ {
+ xIntraBlockCopy(pu, predBuf, COMPONENT_Cb);
+ xIntraBlockCopy(pu, predBuf, COMPONENT_Cr);
+ }
+ return;
+ }
+ }
+#endif
// dual tree handling for IBC as the only ref
if ((!luma || !chroma) && eRefPicList == REF_PIC_LIST_0)
{
@@ -2704,4 +2734,106 @@ void InterPrediction::cacheAssign( CacheModel *cache )
}
#endif
-//! \}
+#if JVET_O1170_IBC_VIRTUAL_BUFFER
+void InterPrediction::xFillIBCBuffer(CodingUnit &cu)
+{
+ for (auto &currPU : CU::traverseTUs(cu))
+ {
+ for (const CompArea &area : currPU.blocks)
+ {
+ if (!area.valid())
+ continue;
+
+ const unsigned int lcuWidth = cu.cs->slice->getSPS()->getMaxCUWidth();
+ const int shiftSample = ::getComponentScaleX(area.compID, cu.chromaFormat);
+ const int ctuSizeLog2 = g_aucLog2[lcuWidth] - shiftSample;
+ const int pux = area.x & ((m_IBCBufferWidth >> shiftSample) - 1);
+ const int puy = area.y & (( 1 << ctuSizeLog2 ) - 1);
+ const CompArea dstArea = CompArea(area.compID, cu.chromaFormat, Position(pux, puy), Size(area.width, area.height));
+ CPelBuf srcBuf = cu.cs->getRecoBuf(area);
+ PelBuf dstBuf = m_IBCBuffer.getBuf(dstArea);
+
+ dstBuf.copyFrom(srcBuf);
+ }
+ }
+}
+
+void InterPrediction::xIntraBlockCopy(PredictionUnit &pu, PelUnitBuf &predBuf, const ComponentID compID)
+{
+ const unsigned int lcuWidth = pu.cs->slice->getSPS()->getMaxCUWidth();
+ int shiftSample = ::getComponentScaleX(compID, pu.chromaFormat);
+ const int ctuSizeLog2 = g_aucLog2[lcuWidth] - shiftSample;
+ pu.bv = pu.mv[REF_PIC_LIST_0];
+ pu.bv.changePrecision(MV_PRECISION_INTERNAL, MV_PRECISION_INT);
+ int refx, refy;
+ if (compID == COMPONENT_Y)
+ {
+ refx = pu.Y().x + pu.bv.hor;
+ refy = pu.Y().y + pu.bv.ver;
+ }
+ else
+ {//Cb or Cr
+ refx = pu.Cb().x + (pu.bv.hor >> shiftSample);
+ refy = pu.Cb().y + (pu.bv.ver >> shiftSample);
+ }
+ refx &= ((m_IBCBufferWidth >> shiftSample) - 1);
+ refy &= ((1 << ctuSizeLog2) - 1);
+
+ if (refx + predBuf.bufs[compID].width <= (m_IBCBufferWidth >> shiftSample))
+ {
+ const CompArea srcArea = CompArea(compID, pu.chromaFormat, Position(refx, refy), Size(predBuf.bufs[compID].width, predBuf.bufs[compID].height));
+ const CPelBuf refBuf = m_IBCBuffer.getBuf(srcArea);
+ predBuf.bufs[compID].copyFrom(refBuf);
+ }
+ else
+ {//wrap around
+ int width = (m_IBCBufferWidth >> shiftSample) - refx;
+ CompArea srcArea = CompArea(compID, pu.chromaFormat, Position(refx, refy), Size(width, predBuf.bufs[compID].height));
+ CPelBuf srcBuf = m_IBCBuffer.getBuf(srcArea);
+ PelBuf dstBuf = PelBuf(predBuf.bufs[compID].bufAt(Position(0, 0)), predBuf.bufs[compID].stride, Size(width, predBuf.bufs[compID].height));
+ dstBuf.copyFrom(srcBuf);
+
+ width = refx + predBuf.bufs[compID].width - (m_IBCBufferWidth >> shiftSample);
+ srcArea = CompArea(compID, pu.chromaFormat, Position(0, refy), Size(width, predBuf.bufs[compID].height));
+ srcBuf = m_IBCBuffer.getBuf(srcArea);
+ dstBuf = PelBuf(predBuf.bufs[compID].bufAt(Position((m_IBCBufferWidth >> shiftSample) - refx, 0)), predBuf.bufs[compID].stride, Size(width, predBuf.bufs[compID].height));
+ dstBuf.copyFrom(srcBuf);
+ }
+}
+
+#if JVET_O1170_CHECK_BV_AT_DECODER
+void InterPrediction::resetIBCBuffer(const ChromaFormat chromaFormatIDC, const int ctuSize)
+{
+ const UnitArea area = UnitArea(chromaFormatIDC, Area(0, 0, m_IBCBufferWidth, ctuSize));
+ m_IBCBuffer.getBuf(area).fill(-1);
+}
+
+void InterPrediction::resetVPDUforIBC(const ChromaFormat chromaFormatIDC, const int ctuSize, const int vSize, const int xPos, const int yPos)
+{
+ const UnitArea area = UnitArea(chromaFormatIDC, Area(xPos & (m_IBCBufferWidth - 1), yPos & (ctuSize - 1), vSize, vSize));
+ m_IBCBuffer.getBuf(area).fill(-1);
+}
+
+bool InterPrediction::isLumaBvValid(const int ctuSize, const int xCb, const int yCb, const int width, const int height, const int xBv, const int yBv)
+{
+ if(((yCb + yBv) & (ctuSize - 1)) + height > ctuSize)
+ {
+ return false;
+ }
+ int refTLx = xCb + xBv;
+ int refTLy = (yCb + yBv) & (ctuSize - 1);
+ PelBuf buf = m_IBCBuffer.Y();
+ for(int x = 0; x < width; x += 4)
+ {
+ for(int y = 0; y < height; y += 4)
+ {
+ if(buf.at((x + refTLx) & (m_IBCBufferWidth - 1), y + refTLy) == -1) return false;
+ if(buf.at((x + 3 + refTLx) & (m_IBCBufferWidth - 1), y + refTLy) == -1) return false;
+ if(buf.at((x + refTLx) & (m_IBCBufferWidth - 1), y + 3 + refTLy) == -1) return false;
+ if(buf.at((x + 3 + refTLx) & (m_IBCBufferWidth - 1), y + 3 + refTLy) == -1) return false;
+ }
+ }
+ return true;
+}
+#endif
+#endif
diff --git a/source/Lib/CommonLib/InterPrediction.h b/source/Lib/CommonLib/InterPrediction.h
index 353d3c7d18ff052036e847a0411c4c6f3e658e33..fc9703d303aa1dae42bdc7ec7cddd1736aac2594 100644
--- a/source/Lib/CommonLib/InterPrediction.h
+++ b/source/Lib/CommonLib/InterPrediction.h
@@ -123,6 +123,11 @@ protected:
Pel* m_gradY1;
bool m_subPuMC;
+#if JVET_O1170_IBC_VIRTUAL_BUFFER
+ int m_IBCBufferWidth;
+ PelStorage m_IBCBuffer;
+ void xIntraBlockCopy (PredictionUnit &pu, PelUnitBuf &predBuf, const ComponentID compID);
+#endif
int rightShiftMSB(int numer, int denom);
void applyBiOptFlow(const PredictionUnit &pu, const CPelUnitBuf &yuvSrc0, const CPelUnitBuf &yuvSrc1, const int &refIdx0, const int &refIdx1, PelUnitBuf &yuvDst, const BitDepths &clipBitDepths);
bool xCalcBiPredSubBlkDist(const PredictionUnit &pu, const Pel* yuvSrc0, const int src0Stride, const Pel* yuvSrc1, const int src1Stride, const BitDepths &clipBitDepths);
@@ -183,7 +188,11 @@ public:
InterPrediction();
virtual ~InterPrediction();
+#if JVET_O1170_IBC_VIRTUAL_BUFFER
+ void init (RdCost* pcRdCost, ChromaFormat chromaFormatIDC, const int ctuSize);
+#else
void init (RdCost* pcRdCost, ChromaFormat chromaFormatIDC);
+#endif
// inter
void motionCompensation (PredictionUnit &pu, PelUnitBuf& predBuf, const RefPicList &eRefPicList = REF_PIC_LIST_X
@@ -226,6 +235,14 @@ public:
int getShareState() const { return m_shareState; }
#endif
static bool isSubblockVectorSpreadOverLimit( int a, int b, int c, int d, int predType );
+#if JVET_O1170_IBC_VIRTUAL_BUFFER
+ void xFillIBCBuffer(CodingUnit &cu);
+#if JVET_O1170_CHECK_BV_AT_DECODER
+ void resetIBCBuffer(const ChromaFormat chromaFormatIDC, const int ctuSize);
+ void resetVPDUforIBC(const ChromaFormat chromaFormatIDC, const int ctuSize, const int vSize, const int xPos, const int yPos);
+ bool isLumaBvValid(const int ctuSize, const int xCb, const int yCb, const int width, const int height, const int xBv, const int yBv);
+#endif
+#endif
};
//! \}
diff --git a/source/Lib/CommonLib/TypeDef.h b/source/Lib/CommonLib/TypeDef.h
index 113a820b3e9f569114db64a494fb8005fb2b8d3c..6beaa9dcaecb53bd978c367a87e34636d1d10f1f 100644
--- a/source/Lib/CommonLib/TypeDef.h
+++ b/source/Lib/CommonLib/TypeDef.h
@@ -50,8 +50,20 @@
#include <assert.h>
#include <cassert>
+
#define JVET_O0070_PROF 1 // JVET-O0070 method 4-2.1a: Prediction refinement with optical flow for affine mode
+#define JVET_O1170_IBC_VIRTUAL_BUFFER 1 // JVET-O1170/O1171: IBC virtual buffer
+#if JVET_O1170_IBC_VIRTUAL_BUFFER
+#define JVET_O1170_CHECK_BV_AT_DECODER 1 // For decoder to check if a BV is valid or not
+#endif
+
+#define JVET_O0213_RESTRICT_LFNST_TO_MAX_TB_SIZE 1 // JVET-O0213: Block size restriction of LFNST to maximum transform size
+
+#define JVET_O0244_DELTA_POC 1 // JVET-O0244: weighted prediction in SPS and delta POC
+
+#define JVET_O1153_INTRA_CHROMAMODE_CODING 1 //JVET-O1153: simplified intra chromamode coding
+
#define JVET_O0159_10BITTCTABLE_DEBLOCKING 1 // tc table for 10-bit video
#define JVET_O0061_MV_THR_DEBLOCKING 1 // a deblocking mv threshold of half pel
diff --git a/source/Lib/CommonLib/UnitTools.cpp b/source/Lib/CommonLib/UnitTools.cpp
index dff0633ad827dd95e778ccada07549091df11c87..413d6dacd3f201d13c6724f73b9cfe1f8554babe 100755
--- a/source/Lib/CommonLib/UnitTools.cpp
+++ b/source/Lib/CommonLib/UnitTools.cpp
@@ -1679,6 +1679,8 @@ bool PU::checkDMVRCondition(const PredictionUnit& pu)
return false;
}
}
+
+#if !JVET_O1170_IBC_VIRTUAL_BUFFER
// for ibc pu validation
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)
{
@@ -1756,6 +1758,7 @@ bool PU::isBlockVectorValid(PredictionUnit& pu, int xPos, int yPos, int width, i
return true;
}// for ibc pu validation
+#endif
static int xGetDistScaleFactor(const int &iCurrPOC, const int &iCurrRefPOC, const int &iColPOC, const int &iColRefPOC)
{
diff --git a/source/Lib/CommonLib/UnitTools.h b/source/Lib/CommonLib/UnitTools.h
index 94bfadc6bb88fb02e71c9c41d8681fd1ef0a94f5..0e8f46b1ab02a25eccee8b163c6537406338d464 100644
--- a/source/Lib/CommonLib/UnitTools.h
+++ b/source/Lib/CommonLib/UnitTools.h
@@ -204,7 +204,9 @@ namespace PU
bool isAddNeighborMv (const Mv& currMv, Mv* neighborMvs, int numNeighborMv);
void getIbcMVPsEncOnly(PredictionUnit &pu, Mv* mvPred, int& nbPred);
bool getDerivedBV(PredictionUnit &pu, const Mv& currentMv, Mv& derivedMv);
+#if !JVET_O1170_IBC_VIRTUAL_BUFFER
bool 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);
+#endif
bool checkDMVRCondition(const PredictionUnit& pu);
}
diff --git a/source/Lib/DecoderLib/DecCu.cpp b/source/Lib/DecoderLib/DecCu.cpp
index 13367e597e1da487143a8b5295d23d99bbe1eac2..ba1571a5286b893dd87ad14136ff280e3a478746 100644
--- a/source/Lib/DecoderLib/DecCu.cpp
+++ b/source/Lib/DecoderLib/DecCu.cpp
@@ -109,6 +109,13 @@ void DecCu::decompressCtu( CodingStructure& cs, const UnitArea& ctuArea )
}
bool sharePrepareCondition = ((!cs.pcv->isEncoder) && (!(cs.slice->isIntra()) || cs.slice->getSPS()->getIBCFlag()));
+#if JVET_O1170_CHECK_BV_AT_DECODER
+ if (cs.resetIBCBuffer)
+ {
+ m_pcInterPred->resetIBCBuffer(cs.pcv->chrFormat, cs.slice->getSPS()->getMaxCUHeight());
+ cs.resetIBCBuffer = false;
+ }
+#endif
for( int ch = 0; ch < maxNumChannelType; ch++ )
{
const ChannelType chType = ChannelType( ch );
@@ -117,6 +124,16 @@ void DecCu::decompressCtu( CodingStructure& cs, const UnitArea& ctuArea )
for( auto &currCU : cs.traverseCUs( CS::getArea( cs, ctuArea, chType ), chType ) )
{
+#if JVET_O1170_CHECK_BV_AT_DECODER
+ if(currCU.Y().valid())
+ {
+ const int vSize = cs.slice->getSPS()->getMaxCUHeight() > 64 ? 64 : cs.slice->getSPS()->getMaxCUHeight();
+ if((currCU.Y().x % vSize) == 0 && (currCU.Y().y % vSize) == 0)
+ {
+ m_pcInterPred->resetVPDUforIBC(cs.pcv->chrFormat, cs.slice->getSPS()->getMaxCUHeight(), vSize, currCU.Y().x, currCU.Y().y);
+ }
+ }
+#endif
if(sharePrepareCondition)
{
if ((currCU.shareParentPos.x >= 0) && (!(currCU.shareParentPos.x == prevTmpPos.x && currCU.shareParentPos.y == prevTmpPos.y)))
@@ -155,6 +172,9 @@ void DecCu::decompressCtu( CodingStructure& cs, const UnitArea& ctuArea )
{
xFillPCMBuffer( currCU );
}
+#if JVET_O1170_IBC_VIRTUAL_BUFFER
+ m_pcInterPred->xFillIBCBuffer(currCU);
+#endif
DTRACE_BLOCK_REC( cs.picture->getRecoBuf( currCU ), currCU, currCU.predMode );
}
@@ -900,12 +920,20 @@ void DecCu::xDeriveCUMV( CodingUnit &cu )
const int cuPelY = pu.Y().y;
int roiWidth = pu.lwidth();
int roiHeight = pu.lheight();
+#if !JVET_O1170_CHECK_BV_AT_DECODER
const int picWidth = pu.cs->slice->getSPS()->getPicWidthInLumaSamples();
const int picHeight = pu.cs->slice->getSPS()->getPicHeightInLumaSamples();
+#endif
const unsigned int lcuWidth = pu.cs->slice->getSPS()->getMaxCUWidth();
int xPred = pu.mv[0].getHor() >> MV_FRACTIONAL_BITS_INTERNAL;
int yPred = pu.mv[0].getVer() >> MV_FRACTIONAL_BITS_INTERNAL;
+#if JVET_O1170_CHECK_BV_AT_DECODER
+ CHECK(!m_pcInterPred->isLumaBvValid(lcuWidth, cuPelX, cuPelY, roiWidth, roiHeight, xPred, yPred), "invalid block vector for IBC detected.");
+#else
+#if !JVET_O1170_IBC_VIRTUAL_BUFFER
CHECK(!PU::isBlockVectorValid(pu, cuPelX, cuPelY, roiWidth, roiHeight, picWidth, picHeight, 0, 0, xPred, yPred, lcuWidth), "invalid block vector for IBC detected.");
+#endif
+#endif
}
}
}
diff --git a/source/Lib/DecoderLib/DecLib.cpp b/source/Lib/DecoderLib/DecLib.cpp
index 66a5f3cea2b2a743c65a2962c13d6f75ed5a4ad8..6bfde1cb5e2e09fa89f18e0020379ccc0571a1cf 100644
--- a/source/Lib/DecoderLib/DecLib.cpp
+++ b/source/Lib/DecoderLib/DecLib.cpp
@@ -850,7 +850,11 @@ void DecLib::xActivateParameterSets()
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 ) );
+#if JVET_O1170_IBC_VIRTUAL_BUFFER
+ m_cInterPred.init( &m_cRdCost, sps->getChromaFormatIdc(), sps->getMaxCUHeight() );
+#else
m_cInterPred.init( &m_cRdCost, sps->getChromaFormatIdc() );
+#endif
if (sps->getUseReshaper())
{
m_cReshaper.createDec(sps->getBitDepth(CHANNEL_TYPE_LUMA));
diff --git a/source/Lib/DecoderLib/DecSlice.cpp b/source/Lib/DecoderLib/DecSlice.cpp
index 5def5c887f3b9bb4677059532d536c172354b91a..2c6b68b2e4e351f5ff0af7ac971636dab5bdd45d 100644
--- a/source/Lib/DecoderLib/DecSlice.cpp
+++ b/source/Lib/DecoderLib/DecSlice.cpp
@@ -191,6 +191,9 @@ void DecSlice::decompressSlice( Slice* slice, InputBitstream* bitstream, int deb
{
cs.motionLut.lut.resize(0);
cs.motionLut.lutIbc.resize(0);
+#if JVET_O1170_CHECK_BV_AT_DECODER
+ cs.resetIBCBuffer = true;
+#endif
#if !JVET_O0078_SINGLE_HMVPLUT
cs.motionLut.lutShareIbc.resize(0);
#endif
diff --git a/source/Lib/EncoderLib/EncCu.cpp b/source/Lib/EncoderLib/EncCu.cpp
index 48a584c845faa95a6137f12dacba43f422b93c61..d81a940ffa7ff151ed4b474dc207ba7d06f905dc 100644
--- a/source/Lib/EncoderLib/EncCu.cpp
+++ b/source/Lib/EncoderLib/EncCu.cpp
@@ -3154,7 +3154,11 @@ void EncCu::xCheckRDCostIBCModeMerge2Nx2N(CodingStructure *&tempCS, CodingStruct
int xPred = pu.bv.getHor();
int yPred = pu.bv.getVer();
+#if JVET_O1170_IBC_VIRTUAL_BUFFER
+ if (!m_pcInterSearch->searchBv(pu, cuPelX, cuPelY, roiWidth, roiHeight, picWidth, picHeight, xPred, yPred, lcuWidth)) // not valid bv derived
+#else
if (!PU::isBlockVectorValid(pu, cuPelX, cuPelY, roiWidth, roiHeight, picWidth, picHeight, 0, 0, xPred, yPred, lcuWidth)) // not valid bv derived
+#endif
{
numValidBv--;
continue;
diff --git a/source/Lib/EncoderLib/InterSearch.cpp b/source/Lib/EncoderLib/InterSearch.cpp
index 648db4c17a90d00a5a90a7f53d78550139e87b66..aa185316bef82966e93cd0a1f7feed517fb2d049 100644
--- a/source/Lib/EncoderLib/InterSearch.cpp
+++ b/source/Lib/EncoderLib/InterSearch.cpp
@@ -252,7 +252,11 @@ void InterSearch::init( EncCfg* pcEncCfg,
}
const ChromaFormat cform = pcEncCfg->getChromaFormatIdc();
+#if JVET_O1170_IBC_VIRTUAL_BUFFER
+ InterPrediction::init( pcRdCost, cform, maxCUHeight );
+#else
InterPrediction::init( pcRdCost, cform );
+#endif
for( uint32_t i = 0; i < NUM_REF_PIC_LIST_01; i++ )
{
@@ -995,7 +999,11 @@ void InterSearch::xIntraPatternSearch(PredictionUnit& pu, IntTZSearchStruct& cS
&& !((yPred < srTop) || (yPred > srBottom))
&& !((xPred < srLeft) || (xPred > srRight)))
{
+#if JVET_O1170_IBC_VIRTUAL_BUFFER
+ bool validCand = searchBv(pu, cuPelX, cuPelY, roiWidth, roiHeight, picWidth, picHeight, xPred, yPred, lcuWidth);
+#else
bool validCand = PU::isBlockVectorValid(pu, cuPelX, cuPelY, roiWidth, roiHeight, picWidth, picHeight, 0, 0, xPred, yPred, lcuWidth);
+#endif
if (validCand)
{
@@ -1016,7 +1024,11 @@ void InterSearch::xIntraPatternSearch(PredictionUnit& pu, IntTZSearchStruct& cS
const int boundY = (0 - roiHeight - puPelOffsetY);
for (int y = std::max(srchRngVerTop, 0 - cuPelY); y <= boundY; ++y)
{
+#if JVET_O1170_IBC_VIRTUAL_BUFFER
+ if (!searchBv(pu, cuPelX, cuPelY, roiWidth, roiHeight, picWidth, picHeight, 0, y, lcuWidth))
+#else
if (!PU::isBlockVectorValid(pu, cuPelX, cuPelY, roiWidth, roiHeight, picWidth, picHeight, 0, 0, 0, y, lcuWidth))
+#endif
{
continue;
}
@@ -1041,7 +1053,11 @@ void InterSearch::xIntraPatternSearch(PredictionUnit& pu, IntTZSearchStruct& cS
const int boundX = std::max(srchRngHorLeft, -cuPelX);
for (int x = 0 - roiWidth - puPelOffsetX; x >= boundX; --x)
{
+#if JVET_O1170_IBC_VIRTUAL_BUFFER
+ if (!searchBv(pu, cuPelX, cuPelY, roiWidth, roiHeight, picWidth, picHeight, x, 0, lcuWidth))
+#else
if (!PU::isBlockVectorValid(pu, cuPelX, cuPelY, roiWidth, roiHeight, picWidth, picHeight, 0, 0, x, 0, lcuWidth))
+#endif
{
continue;
}
@@ -1092,7 +1108,11 @@ void InterSearch::xIntraPatternSearch(PredictionUnit& pu, IntTZSearchStruct& cS
if ((x == 0) || ((int)(cuPelX + x + roiWidth) >= picWidth))
continue;
+#if JVET_O1170_IBC_VIRTUAL_BUFFER
+ if (!searchBv(pu, cuPelX, cuPelY, roiWidth, roiHeight, picWidth, picHeight, x, y, lcuWidth))
+#else
if (!PU::isBlockVectorValid(pu, cuPelX, cuPelY, roiWidth, roiHeight, picWidth, picHeight, 0, 0, x, y, lcuWidth))
+#endif
{
continue;
}
@@ -1132,7 +1152,11 @@ void InterSearch::xIntraPatternSearch(PredictionUnit& pu, IntTZSearchStruct& cS
if ((x == 0) || ((int)(cuPelX + x + roiWidth) >= picWidth))
continue;
+#if JVET_O1170_IBC_VIRTUAL_BUFFER
+ if (!searchBv(pu, cuPelX, cuPelY, roiWidth, roiHeight, picWidth, picHeight, x, y, lcuWidth))
+#else
if (!PU::isBlockVectorValid(pu, cuPelX, cuPelY, roiWidth, roiHeight, picWidth, picHeight, 0, 0, x, y, lcuWidth))
+#endif
{
continue;
}
@@ -1189,7 +1213,11 @@ void InterSearch::xIntraPatternSearch(PredictionUnit& pu, IntTZSearchStruct& cS
if ((x == 0) || ((int)(cuPelX + x + roiWidth) >= picWidth))
continue;
+#if JVET_O1170_IBC_VIRTUAL_BUFFER
+ if (!searchBv(pu, cuPelX, cuPelY, roiWidth, roiHeight, picWidth, picHeight, x, y, lcuWidth))
+#else
if (!PU::isBlockVectorValid(pu, cuPelX, cuPelY, roiWidth, roiHeight, picWidth, picHeight, 0, 0, x, y, lcuWidth))
+#endif
{
continue;
}
@@ -1311,7 +1339,11 @@ void InterSearch::xIBCEstimation(PredictionUnit& pu, PelUnitBuf& origBuf,
int xBv = bv.hor;
int yBv = bv.ver;
+#if JVET_O1170_IBC_VIRTUAL_BUFFER
+ if (searchBv(pu, cuPelX, cuPelY, iRoiWidth, iRoiHeight, iPicWidth, iPicHeight, xBv, yBv, lcuWidth))
+#else
if (PU::isBlockVectorValid(pu, cuPelX, cuPelY, iRoiWidth, iRoiHeight, iPicWidth, iPicHeight, 0, 0, xBv, yBv, lcuWidth))
+#endif
{
buffered = true;
Distortion sad = m_pcRdCost->getBvCostMultiplePreds(xBv, yBv, pu.cs->sps->getAMVREnabledFlag());
@@ -1346,7 +1378,11 @@ void InterSearch::xIBCEstimation(PredictionUnit& pu, PelUnitBuf& origBuf,
int xPred = cMvPredEncOnly[cand].getHor();
int yPred = cMvPredEncOnly[cand].getVer();
+#if JVET_O1170_IBC_VIRTUAL_BUFFER
+ if (searchBv(pu, cuPelX, cuPelY, iRoiWidth, iRoiHeight, iPicWidth, iPicHeight, xPred, yPred, lcuWidth))
+#else
if (PU::isBlockVectorValid(pu, cuPelX, cuPelY, iRoiWidth, iRoiHeight, iPicWidth, iPicHeight, 0, 0, xPred, yPred, lcuWidth))
+#endif
{
Distortion sad = m_pcRdCost->getBvCostMultiplePreds(xPred, yPred, pu.cs->sps->getAMVREnabledFlag());
m_cDistParam.cur.buf = cStruct.piRefY + cStruct.iRefStride * yPred + xPred;
@@ -1584,7 +1620,11 @@ void InterSearch::xxIBCHashSearch(PredictionUnit& pu, Mv* mvPred, int numMvPred,
Mv candMv;
candMv.set(tmp.x, tmp.y);
+#if JVET_O1170_IBC_VIRTUAL_BUFFER
+ if (!searchBv(pu, cuPelX, cuPelY, roiWidth, roiHeight, picWidth, picHeight, candMv.getHor(), candMv.getVer(), lcuWidth))
+#else
if (!PU::isBlockVectorValid(pu, cuPelX, cuPelY, roiWidth, roiHeight, picWidth, picHeight, 0, 0, candMv.getHor(), candMv.getVer(), lcuWidth))
+#endif
{
continue;
}
@@ -8019,3 +8059,83 @@ uint64_t InterSearch::xCalcPuMeBits(PredictionUnit& pu)
}
return m_CABACEstimator->getEstFracBits();
}
+
+#if JVET_O1170_IBC_VIRTUAL_BUFFER
+bool InterSearch::searchBv(PredictionUnit& pu, int xPos, int yPos, int width, int height, int picWidth, int picHeight, int xBv, int yBv, int ctuSize)
+{
+ const int ctuSizeLog2 = g_aucLog2[ctuSize];
+
+ int refRightX = xPos + xBv + width - 1;
+ int refBottomY = yPos + yBv + height - 1;
+
+ 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;
+ }
+
+ // Don't search the above CTU row
+ if (refTopY >> ctuSizeLog2 < yPos >> ctuSizeLog2)
+ return false;
+
+ // Don't search the below CTU row
+ if (refBottomY >> ctuSizeLog2 > yPos >> ctuSizeLog2)
+ {
+ return false;
+ }
+
+ // in the same CTU line
+ int numLeftCTUs = (1 << ((7 - ctuSizeLog2) << 1)) - ((ctuSizeLog2 < 7) ? 1 : 0);
+ if ((refRightX >> ctuSizeLog2 <= xPos >> ctuSizeLog2) && (refLeftX >> ctuSizeLog2 >= (xPos >> ctuSizeLog2) - numLeftCTUs))
+ {
+
+ // 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 (((refLeftX >> ctuSizeLog2) == ((xPos >> ctuSizeLog2) - 1)) && (ctuSizeLog2 == 7))
+ {
+ // ref block's collocated block in current CTU
+ const Position refPosCol = pu.Y().topLeft().offset(xBv + ctuSize, yBv);
+ 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;
+ if (refPosCol64x64 == pu.Y().topLeft())
+ return false;
+ }
+ }
+ 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;
+}
+#endif
+
+//! \}
\ No newline at end of file
diff --git a/source/Lib/EncoderLib/InterSearch.h b/source/Lib/EncoderLib/InterSearch.h
index b708c938f9634f87ca45eced4f90c3b48e34b67f..4fcb540964b20087c5811d6a8700a97ccfb949b3 100644
--- a/source/Lib/EncoderLib/InterSearch.h
+++ b/source/Lib/EncoderLib/InterSearch.h
@@ -322,6 +322,9 @@ public:
#endif
void resetSavedAffineMotion();
void storeAffineMotion( Mv acAffineMv[2][3], int16_t affineRefIdx[2], EAffineModel affineType, int gbiIdx );
+#if JVET_O1170_IBC_VIRTUAL_BUFFER
+ bool searchBv(PredictionUnit& pu, int xPos, int yPos, int width, int height, int picWidth, int picHeight, int xBv, int yBv, int ctuSize);
+#endif
protected:
/// sub-function for motion vector refinement used in fractional-pel accuracy