[x265] [PATCH] cleanup: simplify slice ref lists; moving shared offset buffers into SPS
Pradeep
pradeep at multicorewareinc.com
Wed Aug 12 11:24:06 CEST 2015
# HG changeset patch
# User Pradeep
# Date 1439332317 0
# Tue Aug 11 22:31:57 2015 +0000
# Node ID 1b2e5e8ccb850d83bfae57b5e2dec623330f558d
# Parent faecefdb2ae90c031dfda5c11d14b9ba7005257b
cleanup: simplify slice ref lists; moving shared offset buffers into SPS
Stripping down Steve's experimental patches into separate patches
Part 1:
Rename slice.m_refPicList to m_refFrameList since it is an array of Frame
pointers, and make a new slice.m_refReconPicList array which points directly
to the motion reference PicYuv buffers (bypassing the Frame structure)
The shared offset buffers were moved from the top-level encoder into the SPS
structure so the FrameData functions could use them directly (avoiding a major
layering violations). The offset buffers are computed based on SPS values, so it
seems minimally ugly to keep them there.
diff -r faecefdb2ae9 -r 1b2e5e8ccb85 source/common/cudata.cpp
--- a/source/common/cudata.cpp Tue Aug 11 14:28:23 2015 +0000
+++ b/source/common/cudata.cpp Tue Aug 11 22:31:57 2015 +0000
@@ -1676,7 +1676,7 @@
if (tempRefIdx != -1)
{
uint32_t cuAddr = neighbours[MD_COLLOCATED].cuAddr[picList];
- const Frame* colPic = m_slice->m_refPicList[m_slice->isInterB() && !m_slice->m_colFromL0Flag][m_slice->m_colRefIdx];
+ const Frame* colPic = m_slice->m_refFrameList[m_slice->isInterB() && !m_slice->m_colFromL0Flag][m_slice->m_colRefIdx];
const CUData* colCU = colPic->m_encData->getPicCTU(cuAddr);
// Scale the vector
@@ -1857,7 +1857,7 @@
bool CUData::getColMVP(MV& outMV, int& outRefIdx, int picList, int cuAddr, int partUnitIdx) const
{
- const Frame* colPic = m_slice->m_refPicList[m_slice->isInterB() && !m_slice->m_colFromL0Flag][m_slice->m_colRefIdx];
+ const Frame* colPic = m_slice->m_refFrameList[m_slice->isInterB() && !m_slice->m_colFromL0Flag][m_slice->m_colRefIdx];
const CUData* colCU = colPic->m_encData->getPicCTU(cuAddr);
uint32_t absPartAddr = partUnitIdx & TMVP_UNIT_MASK;
@@ -1892,7 +1892,7 @@
// Cache the collocated MV.
bool CUData::getCollocatedMV(int cuAddr, int partUnitIdx, InterNeighbourMV *neighbour) const
{
- const Frame* colPic = m_slice->m_refPicList[m_slice->isInterB() && !m_slice->m_colFromL0Flag][m_slice->m_colRefIdx];
+ const Frame* colPic = m_slice->m_refFrameList[m_slice->isInterB() && !m_slice->m_colFromL0Flag][m_slice->m_colRefIdx];
const CUData* colCU = colPic->m_encData->getPicCTU(cuAddr);
uint32_t absPartAddr = partUnitIdx & TMVP_UNIT_MASK;
diff -r faecefdb2ae9 -r 1b2e5e8ccb85 source/common/deblock.cpp
--- a/source/common/deblock.cpp Tue Aug 11 14:28:23 2015 +0000
+++ b/source/common/deblock.cpp Tue Aug 11 22:31:57 2015 +0000
@@ -209,8 +209,8 @@
const Slice* const sliceQ = cuQ->m_slice;
const Slice* const sliceP = cuP->m_slice;
- const Frame* refP0 = sliceP->getRefPic(0, cuP->m_refIdx[0][partP]);
- const Frame* refQ0 = sliceQ->getRefPic(0, cuQ->m_refIdx[0][partQ]);
+ const Frame* refP0 = sliceP->m_refFrameList[0][cuP->m_refIdx[0][partP]];
+ const Frame* refQ0 = sliceQ->m_refFrameList[0][cuQ->m_refIdx[0][partQ]];
const MV& mvP0 = refP0 ? cuP->m_mv[0][partP] : zeroMv;
const MV& mvQ0 = refQ0 ? cuQ->m_mv[0][partQ] : zeroMv;
@@ -221,8 +221,8 @@
}
// (sliceQ->isInterB() || sliceP->isInterB())
- const Frame* refP1 = sliceP->getRefPic(1, cuP->m_refIdx[1][partP]);
- const Frame* refQ1 = sliceQ->getRefPic(1, cuQ->m_refIdx[1][partQ]);
+ const Frame* refP1 = sliceP->m_refFrameList[1][cuP->m_refIdx[1][partP]];
+ const Frame* refQ1 = sliceQ->m_refFrameList[1][cuQ->m_refIdx[1][partQ]];
const MV& mvP1 = refP1 ? cuP->m_mv[1][partP] : zeroMv;
const MV& mvQ1 = refQ1 ? cuQ->m_mv[1][partQ] : zeroMv;
diff -r faecefdb2ae9 -r 1b2e5e8ccb85 source/common/frame.cpp
--- a/source/common/frame.cpp Tue Aug 11 14:28:23 2015 +0000
+++ b/source/common/frame.cpp Tue Aug 11 22:31:57 2015 +0000
@@ -56,7 +56,7 @@
m_encData = new FrameData;
m_reconPic = new PicYuv;
m_encData->m_reconPic = m_reconPic;
- bool ok = m_encData->create(param, sps) && m_reconPic->create(param->sourceWidth, param->sourceHeight, param->internalCsp);
+ bool ok = m_encData->create(*param, sps) && m_reconPic->create(param->sourceWidth, param->sourceHeight, param->internalCsp);
if (ok)
{
/* initialize right border of m_reconpicYuv as SAO may read beyond the
@@ -65,6 +65,12 @@
memset(m_reconPic->m_picOrg[0], 0, sizeof(pixel) * m_reconPic->m_stride * maxHeight);
memset(m_reconPic->m_picOrg[1], 0, sizeof(pixel) * m_reconPic->m_strideC * (maxHeight >> m_reconPic->m_vChromaShift));
memset(m_reconPic->m_picOrg[2], 0, sizeof(pixel) * m_reconPic->m_strideC * (maxHeight >> m_reconPic->m_vChromaShift));
+
+ /* use pre-calculated cu/pu offsets cached in the SPS structure */
+ m_reconPic->m_cuOffsetC = sps.cuOffsetC;
+ m_reconPic->m_cuOffsetY = sps.cuOffsetY;
+ m_reconPic->m_buOffsetC = sps.buOffsetC;
+ m_reconPic->m_buOffsetY = sps.buOffsetY;
}
return ok;
}
diff -r faecefdb2ae9 -r 1b2e5e8ccb85 source/common/framedata.cpp
--- a/source/common/framedata.cpp Tue Aug 11 14:28:23 2015 +0000
+++ b/source/common/framedata.cpp Tue Aug 11 22:31:57 2015 +0000
@@ -31,15 +31,15 @@
memset(this, 0, sizeof(*this));
}
-bool FrameData::create(x265_param *param, const SPS& sps)
+bool FrameData::create(const x265_param& param, const SPS& sps)
{
- m_param = param;
+ m_param = ¶m;
m_slice = new Slice;
m_picCTU = new CUData[sps.numCUsInFrame];
- m_cuMemPool.create(0, param->internalCsp, sps.numCUsInFrame);
+ m_cuMemPool.create(0, param.internalCsp, sps.numCUsInFrame);
for (uint32_t ctuAddr = 0; ctuAddr < sps.numCUsInFrame; ctuAddr++)
- m_picCTU[ctuAddr].initialize(m_cuMemPool, 0, param->internalCsp, ctuAddr);
+ m_picCTU[ctuAddr].initialize(m_cuMemPool, 0, param.internalCsp, ctuAddr);
CHECKED_MALLOC(m_cuStat, RCStatCU, sps.numCUsInFrame);
CHECKED_MALLOC(m_rowStat, RCStatRow, sps.numCuInHeight);
diff -r faecefdb2ae9 -r 1b2e5e8ccb85 source/common/framedata.h
--- a/source/common/framedata.h Tue Aug 11 14:28:23 2015 +0000
+++ b/source/common/framedata.h Tue Aug 11 22:31:57 2015 +0000
@@ -96,7 +96,7 @@
Slice* m_slice;
SAOParam* m_saoParam;
- x265_param* m_param;
+ const x265_param* m_param;
FrameData* m_freeListNext;
PicYuv* m_reconPic;
@@ -142,11 +142,11 @@
FrameData();
- bool create(x265_param *param, const SPS& sps);
+ bool create(const x265_param& param, const SPS& sps);
void reinit(const SPS& sps);
void destroy();
- CUData* getPicCTU(uint32_t ctuAddr) { return &m_picCTU[ctuAddr]; }
+ inline CUData* getPicCTU(uint32_t ctuAddr) { return &m_picCTU[ctuAddr]; }
};
}
diff -r faecefdb2ae9 -r 1b2e5e8ccb85 source/common/predict.cpp
--- a/source/common/predict.cpp Tue Aug 11 14:28:23 2015 +0000
+++ b/source/common/predict.cpp Tue Aug 11 22:31:57 2015 +0000
@@ -109,18 +109,18 @@
ShortYuv& shortYuv = m_predShortYuv[0];
if (bLuma)
- predInterLumaShort(pu, shortYuv, *cu.m_slice->m_refPicList[0][refIdx0]->m_reconPic, mv0);
+ predInterLumaShort(pu, shortYuv, *cu.m_slice->m_refReconPicList[0][refIdx0], mv0);
if (bChroma)
- predInterChromaShort(pu, shortYuv, *cu.m_slice->m_refPicList[0][refIdx0]->m_reconPic, mv0);
+ predInterChromaShort(pu, shortYuv, *cu.m_slice->m_refReconPicList[0][refIdx0], mv0);
addWeightUni(pu, predYuv, shortYuv, wv0, bLuma, bChroma);
}
else
{
if (bLuma)
- predInterLumaPixel(pu, predYuv, *cu.m_slice->m_refPicList[0][refIdx0]->m_reconPic, mv0);
+ predInterLumaPixel(pu, predYuv, *cu.m_slice->m_refReconPicList[0][refIdx0], mv0);
if (bChroma)
- predInterChromaPixel(pu, predYuv, *cu.m_slice->m_refPicList[0][refIdx0]->m_reconPic, mv0);
+ predInterChromaPixel(pu, predYuv, *cu.m_slice->m_refReconPicList[0][refIdx0], mv0);
}
}
else
@@ -179,13 +179,13 @@
if (bLuma)
{
- predInterLumaShort(pu, m_predShortYuv[0], *cu.m_slice->m_refPicList[0][refIdx0]->m_reconPic, mv0);
- predInterLumaShort(pu, m_predShortYuv[1], *cu.m_slice->m_refPicList[1][refIdx1]->m_reconPic, mv1);
+ predInterLumaShort(pu, m_predShortYuv[0], *cu.m_slice->m_refReconPicList[0][refIdx0], mv0);
+ predInterLumaShort(pu, m_predShortYuv[1], *cu.m_slice->m_refReconPicList[1][refIdx1], mv1);
}
if (bChroma)
{
- predInterChromaShort(pu, m_predShortYuv[0], *cu.m_slice->m_refPicList[0][refIdx0]->m_reconPic, mv0);
- predInterChromaShort(pu, m_predShortYuv[1], *cu.m_slice->m_refPicList[1][refIdx1]->m_reconPic, mv1);
+ predInterChromaShort(pu, m_predShortYuv[0], *cu.m_slice->m_refReconPicList[0][refIdx0], mv0);
+ predInterChromaShort(pu, m_predShortYuv[1], *cu.m_slice->m_refReconPicList[1][refIdx1], mv1);
}
if (pwp0 && pwp1 && (pwp0->bPresentFlag || pwp1->bPresentFlag))
@@ -203,18 +203,18 @@
ShortYuv& shortYuv = m_predShortYuv[0];
if (bLuma)
- predInterLumaShort(pu, shortYuv, *cu.m_slice->m_refPicList[0][refIdx0]->m_reconPic, mv0);
+ predInterLumaShort(pu, shortYuv, *cu.m_slice->m_refReconPicList[0][refIdx0], mv0);
if (bChroma)
- predInterChromaShort(pu, shortYuv, *cu.m_slice->m_refPicList[0][refIdx0]->m_reconPic, mv0);
+ predInterChromaShort(pu, shortYuv, *cu.m_slice->m_refReconPicList[0][refIdx0], mv0);
addWeightUni(pu, predYuv, shortYuv, wv0, bLuma, bChroma);
}
else
{
if (bLuma)
- predInterLumaPixel(pu, predYuv, *cu.m_slice->m_refPicList[0][refIdx0]->m_reconPic, mv0);
+ predInterLumaPixel(pu, predYuv, *cu.m_slice->m_refReconPicList[0][refIdx0], mv0);
if (bChroma)
- predInterChromaPixel(pu, predYuv, *cu.m_slice->m_refPicList[0][refIdx0]->m_reconPic, mv0);
+ predInterChromaPixel(pu, predYuv, *cu.m_slice->m_refReconPicList[0][refIdx0], mv0);
}
}
else
@@ -230,18 +230,18 @@
ShortYuv& shortYuv = m_predShortYuv[0];
if (bLuma)
- predInterLumaShort(pu, shortYuv, *cu.m_slice->m_refPicList[1][refIdx1]->m_reconPic, mv1);
+ predInterLumaShort(pu, shortYuv, *cu.m_slice->m_refReconPicList[1][refIdx1], mv1);
if (bChroma)
- predInterChromaShort(pu, shortYuv, *cu.m_slice->m_refPicList[1][refIdx1]->m_reconPic, mv1);
+ predInterChromaShort(pu, shortYuv, *cu.m_slice->m_refReconPicList[1][refIdx1], mv1);
addWeightUni(pu, predYuv, shortYuv, wv0, bLuma, bChroma);
}
else
{
if (bLuma)
- predInterLumaPixel(pu, predYuv, *cu.m_slice->m_refPicList[1][refIdx1]->m_reconPic, mv1);
+ predInterLumaPixel(pu, predYuv, *cu.m_slice->m_refReconPicList[1][refIdx1], mv1);
if (bChroma)
- predInterChromaPixel(pu, predYuv, *cu.m_slice->m_refPicList[1][refIdx1]->m_reconPic, mv1);
+ predInterChromaPixel(pu, predYuv, *cu.m_slice->m_refReconPicList[1][refIdx1], mv1);
}
}
}
@@ -600,8 +600,9 @@
int tuSize = 1 << intraNeighbors.log2TrSize;
int tuSize2 = tuSize << 1;
- pixel* adiOrigin = cu.m_encData->m_reconPic->getLumaAddr(cu.m_cuAddr, cuGeom.absPartIdx + puAbsPartIdx);
- intptr_t picStride = cu.m_encData->m_reconPic->m_stride;
+ PicYuv* reconPic = cu.m_encData->m_reconPic;
+ pixel* adiOrigin = reconPic->getLumaAddr(cu.m_cuAddr, cuGeom.absPartIdx + puAbsPartIdx);
+ intptr_t picStride = reconPic->m_stride;
fillReferenceSamples(adiOrigin, picStride, intraNeighbors, intraNeighbourBuf[0]);
@@ -648,8 +649,9 @@
void Predict::initAdiPatternChroma(const CUData& cu, const CUGeom& cuGeom, uint32_t puAbsPartIdx, const IntraNeighbors& intraNeighbors, uint32_t chromaId)
{
- const pixel* adiOrigin = cu.m_encData->m_reconPic->getChromaAddr(chromaId, cu.m_cuAddr, cuGeom.absPartIdx + puAbsPartIdx);
- intptr_t picStride = cu.m_encData->m_reconPic->m_strideC;
+ PicYuv* reconPic = cu.m_encData->m_reconPic;
+ const pixel* adiOrigin = reconPic->getChromaAddr(chromaId, cu.m_cuAddr, cuGeom.absPartIdx + puAbsPartIdx);
+ intptr_t picStride = reconPic->m_strideC;
fillReferenceSamples(adiOrigin, picStride, intraNeighbors, intraNeighbourBuf[0]);
diff -r faecefdb2ae9 -r 1b2e5e8ccb85 source/common/slice.cpp
--- a/source/common/slice.cpp Tue Aug 11 14:28:23 2015 +0000
+++ b/source/common/slice.cpp Tue Aug 11 22:31:57 2015 +0000
@@ -33,7 +33,9 @@
{
if (m_sliceType == I_SLICE)
{
- memset(m_refPicList, 0, sizeof(m_refPicList));
+ memset(m_refFrameList, 0, sizeof(m_refFrameList));
+ memset(m_refReconPicList, 0, sizeof(m_refReconPicList));
+ memset(m_refPOCList, 0, sizeof(m_refPOCList));
m_numRefIdx[1] = m_numRefIdx[0] = 0;
return;
}
@@ -106,13 +108,13 @@
{
cIdx = rIdx % numPocTotalCurr;
X265_CHECK(cIdx >= 0 && cIdx < numPocTotalCurr, "RPS index check fail\n");
- m_refPicList[0][rIdx] = rpsCurrList0[cIdx];
+ m_refFrameList[0][rIdx] = rpsCurrList0[cIdx];
}
if (m_sliceType != B_SLICE)
{
m_numRefIdx[1] = 0;
- memset(m_refPicList[1], 0, sizeof(m_refPicList[1]));
+ memset(m_refFrameList[1], 0, sizeof(m_refFrameList[1]));
}
else
{
@@ -120,13 +122,13 @@
{
cIdx = rIdx % numPocTotalCurr;
X265_CHECK(cIdx >= 0 && cIdx < numPocTotalCurr, "RPS index check fail\n");
- m_refPicList[1][rIdx] = rpsCurrList1[cIdx];
+ m_refFrameList[1][rIdx] = rpsCurrList1[cIdx];
}
}
for (int dir = 0; dir < 2; dir++)
for (int numRefIdx = 0; numRefIdx < m_numRefIdx[dir]; numRefIdx++)
- m_refPOCList[dir][numRefIdx] = m_refPicList[dir][numRefIdx]->m_poc;
+ m_refPOCList[dir][numRefIdx] = m_refFrameList[dir][numRefIdx]->m_poc;
}
void Slice::disableWeights()
diff -r faecefdb2ae9 -r 1b2e5e8ccb85 source/common/slice.h
--- a/source/common/slice.h Tue Aug 11 14:28:23 2015 +0000
+++ b/source/common/slice.h Tue Aug 11 22:31:57 2015 +0000
@@ -31,6 +31,7 @@
class Frame;
class PicList;
+class PicYuv;
class MotionReference;
enum SliceType
@@ -209,6 +210,13 @@
struct SPS
{
+ /* cached PicYuv offset arrays, shared by all instances of
+ * PicYuv created by this encoder */
+ intptr_t* cuOffsetY;
+ intptr_t* cuOffsetC;
+ intptr_t* buOffsetY;
+ intptr_t* buOffsetC;
+
int chromaFormatIdc; // use param
uint32_t picWidthInLumaSamples; // use param
uint32_t picHeightInLumaSamples; // use param
@@ -242,6 +250,11 @@
Window conformanceWindow;
VUI vuiParameters;
+
+ SPS()
+ {
+ memset(this, 0, sizeof(*this));
+ }
};
struct PPS
@@ -321,7 +334,8 @@
uint32_t m_colRefIdx; // never modified
int m_numRefIdx[2];
- Frame* m_refPicList[2][MAX_NUM_REF + 1];
+ Frame* m_refFrameList[2][MAX_NUM_REF + 1];
+ PicYuv* m_refReconPicList[2][MAX_NUM_REF + 1];
int m_refPOCList[2][MAX_NUM_REF + 1];
uint32_t m_maxNumMergeCand; // use param
@@ -332,14 +346,9 @@
m_lastIDR = 0;
m_sLFaseFlag = true;
m_numRefIdx[0] = m_numRefIdx[1] = 0;
- for (int i = 0; i < MAX_NUM_REF; i++)
- {
- m_refPicList[0][i] = NULL;
- m_refPicList[1][i] = NULL;
- m_refPOCList[0][i] = 0;
- m_refPOCList[1][i] = 0;
- }
-
+ memset(m_refFrameList, 0, sizeof(m_refFrameList));
+ memset(m_refReconPicList, 0, sizeof(m_refReconPicList));
+ memset(m_refPOCList, 0, sizeof(m_refPOCList));
disableWeights();
}
@@ -347,8 +356,6 @@
void setRefPicList(PicList& picList);
- const Frame* getRefPic(int list, int refIdx) const { return refIdx >= 0 ? m_refPicList[list][refIdx] : NULL; }
-
bool getRapPicFlag() const
{
return m_nalUnitType == NAL_UNIT_CODED_SLICE_IDR_W_RADL
diff -r faecefdb2ae9 -r 1b2e5e8ccb85 source/encoder/analysis.cpp
--- a/source/encoder/analysis.cpp Tue Aug 11 14:28:23 2015 +0000
+++ b/source/encoder/analysis.cpp Tue Aug 11 22:31:57 2015 +0000
@@ -757,6 +757,8 @@
ModeDepth& md = m_modeDepth[depth];
md.bestMode = NULL;
+ PicYuv& reconPic = *m_frame->m_reconPic;
+
bool mightSplit = !(cuGeom.flags & CUGeom::LEAF);
bool mightNotSplit = !(cuGeom.flags & CUGeom::SPLIT_MANDATORY);
uint32_t minDepth = topSkipMinDepth(parentCTU, cuGeom);
@@ -1051,7 +1053,7 @@
residualTransformQuantIntra(*md.bestMode, cuGeom, 0, 0, tuDepthRange);
getBestIntraModeChroma(*md.bestMode, cuGeom);
residualQTIntraChroma(*md.bestMode, cuGeom, 0, 0);
- md.bestMode->reconYuv.copyFromPicYuv(*m_frame->m_reconPic, cu.m_cuAddr, cuGeom.absPartIdx); // TODO:
+ md.bestMode->reconYuv.copyFromPicYuv(reconPic, cu.m_cuAddr, cuGeom.absPartIdx); // TODO:
}
}
}
@@ -1107,7 +1109,7 @@
X265_CHECK(md.bestMode->ok(), "best mode is not ok");
md.bestMode->cu.copyToPic(depth);
if (m_param->rdLevel)
- md.bestMode->reconYuv.copyToPicYuv(*m_frame->m_reconPic, cuAddr, cuGeom.absPartIdx);
+ md.bestMode->reconYuv.copyToPicYuv(reconPic, cuAddr, cuGeom.absPartIdx);
return refMask;
}
@@ -1851,6 +1853,8 @@
cu.copyFromPic(ctu, cuGeom);
+ PicYuv& reconPic = *m_frame->m_reconPic;
+
Yuv& fencYuv = m_modeDepth[cuGeom.depth].fencYuv;
if (cuGeom.depth)
m_modeDepth[0].fencYuv.copyPartToYuv(fencYuv, absPartIdx);
@@ -1906,7 +1910,6 @@
/* residualTransformQuantInter() wrote transformed residual back into
* resiYuv. Generate the recon pixels by adding it to the prediction */
- PicYuv& reconPic = *m_frame->m_reconPic;
if (cu.m_cbf[0][0])
primitives.cu[sizeIdx].add_ps(reconPic.getLumaAddr(cu.m_cuAddr, absPartIdx), reconPic.m_stride,
predY, resiYuv.m_buf[0], predYuv.m_size, resiYuv.m_size);
@@ -1969,7 +1972,7 @@
if (m_slice->m_numRefIdx[0])
{
numRefs++;
- const CUData& cu = *m_slice->m_refPicList[0][0]->m_encData->getPicCTU(parentCTU.m_cuAddr);
+ const CUData& cu = *m_slice->m_refFrameList[0][0]->m_encData->getPicCTU(parentCTU.m_cuAddr);
previousQP = cu.m_qp[0];
if (!cu.m_cuDepth[cuGeom.absPartIdx])
return 0;
@@ -1983,7 +1986,7 @@
if (m_slice->m_numRefIdx[1])
{
numRefs++;
- const CUData& cu = *m_slice->m_refPicList[1][0]->m_encData->getPicCTU(parentCTU.m_cuAddr);
+ const CUData& cu = *m_slice->m_refFrameList[1][0]->m_encData->getPicCTU(parentCTU.m_cuAddr);
if (!cu.m_cuDepth[cuGeom.absPartIdx])
return 0;
for (uint32_t i = 0; i < cuGeom.numPartitions; i += 4)
diff -r faecefdb2ae9 -r 1b2e5e8ccb85 source/encoder/dpb.cpp
--- a/source/encoder/dpb.cpp Tue Aug 11 14:28:23 2015 +0000
+++ b/source/encoder/dpb.cpp Tue Aug 11 22:31:57 2015 +0000
@@ -47,16 +47,16 @@
delete curFrame;
}
- while (m_picSymFreeList)
+ while (m_frameDataFreeList)
{
- FrameData* next = m_picSymFreeList->m_freeListNext;
- m_picSymFreeList->destroy();
+ FrameData* next = m_frameDataFreeList->m_freeListNext;
+ m_frameDataFreeList->destroy();
- m_picSymFreeList->m_reconPic->destroy();
- delete m_picSymFreeList->m_reconPic;
+ m_frameDataFreeList->m_reconPic->destroy();
+ delete m_frameDataFreeList->m_reconPic;
- delete m_picSymFreeList;
- m_picSymFreeList = next;
+ delete m_frameDataFreeList;
+ m_frameDataFreeList = next;
}
}
@@ -79,8 +79,8 @@
iterFrame = m_picList.first();
m_freeList.pushBack(*curFrame);
- curFrame->m_encData->m_freeListNext = m_picSymFreeList;
- m_picSymFreeList = curFrame->m_encData;
+ curFrame->m_encData->m_freeListNext = m_frameDataFreeList;
+ m_frameDataFreeList = curFrame->m_encData;
curFrame->m_encData = NULL;
curFrame->m_reconPic = NULL;
}
@@ -171,7 +171,7 @@
{
for (int ref = 0; ref < slice->m_numRefIdx[l]; ref++)
{
- Frame *refpic = slice->m_refPicList[l][ref];
+ Frame *refpic = slice->m_refFrameList[l][ref];
ATOMIC_INC(&refpic->m_countRefEncoders);
}
}
diff -r faecefdb2ae9 -r 1b2e5e8ccb85 source/encoder/dpb.h
--- a/source/encoder/dpb.h Tue Aug 11 14:28:23 2015 +0000
+++ b/source/encoder/dpb.h Tue Aug 11 22:31:57 2015 +0000
@@ -46,14 +46,14 @@
bool m_bTemporalSublayer;
PicList m_picList;
PicList m_freeList;
- FrameData* m_picSymFreeList;
+ FrameData* m_frameDataFreeList;
DPB(x265_param *param)
{
m_lastIDR = 0;
m_pocCRA = 0;
m_bRefreshPending = false;
- m_picSymFreeList = NULL;
+ m_frameDataFreeList = NULL;
m_maxRefL0 = param->maxNumReferences;
m_maxRefL1 = param->bBPyramid ? 2 : 1;
m_bOpenGOP = param->bOpenGOP;
diff -r faecefdb2ae9 -r 1b2e5e8ccb85 source/encoder/encoder.cpp
--- a/source/encoder/encoder.cpp Tue Aug 11 14:28:23 2015 +0000
+++ b/source/encoder/encoder.cpp Tue Aug 11 22:31:57 2015 +0000
@@ -66,10 +66,6 @@
m_outputCount = 0;
m_param = NULL;
m_latestParam = NULL;
- m_cuOffsetY = NULL;
- m_cuOffsetC = NULL;
- m_buOffsetY = NULL;
- m_buOffsetC = NULL;
m_threadPool = NULL;
m_analysisFile = NULL;
for (int i = 0; i < X265_MAX_FRAME_THREADS; i++)
@@ -318,10 +314,10 @@
delete m_rateControl;
}
- X265_FREE(m_cuOffsetY);
- X265_FREE(m_cuOffsetC);
- X265_FREE(m_buOffsetY);
- X265_FREE(m_buOffsetC);
+ X265_FREE(m_sps.cuOffsetY);
+ X265_FREE(m_sps.cuOffsetC);
+ X265_FREE(m_sps.buOffsetY);
+ X265_FREE(m_sps.buOffsetC);
if (m_analysisFile)
fclose(m_analysisFile);
@@ -416,12 +412,12 @@
/* the first PicYuv created is asked to generate the CU and block unit offset
* arrays which are then shared with all subsequent PicYuv (orig and recon)
* allocated by this top level encoder */
- if (m_cuOffsetY)
+ if (m_sps.cuOffsetY)
{
- inFrame->m_fencPic->m_cuOffsetC = m_cuOffsetC;
- inFrame->m_fencPic->m_cuOffsetY = m_cuOffsetY;
- inFrame->m_fencPic->m_buOffsetC = m_buOffsetC;
- inFrame->m_fencPic->m_buOffsetY = m_buOffsetY;
+ inFrame->m_fencPic->m_cuOffsetC = m_sps.cuOffsetC;
+ inFrame->m_fencPic->m_cuOffsetY = m_sps.cuOffsetY;
+ inFrame->m_fencPic->m_buOffsetC = m_sps.buOffsetC;
+ inFrame->m_fencPic->m_buOffsetY = m_sps.buOffsetY;
}
else
{
@@ -435,10 +431,10 @@
}
else
{
- m_cuOffsetC = inFrame->m_fencPic->m_cuOffsetC;
- m_cuOffsetY = inFrame->m_fencPic->m_cuOffsetY;
- m_buOffsetC = inFrame->m_fencPic->m_buOffsetC;
- m_buOffsetY = inFrame->m_fencPic->m_buOffsetY;
+ m_sps.cuOffsetC = inFrame->m_fencPic->m_cuOffsetC;
+ m_sps.cuOffsetY = inFrame->m_fencPic->m_cuOffsetY;
+ m_sps.buOffsetC = inFrame->m_fencPic->m_buOffsetC;
+ m_sps.buOffsetY = inFrame->m_fencPic->m_buOffsetY;
}
}
}
@@ -633,10 +629,10 @@
if (frameEnc && !pass)
{
/* give this frame a FrameData instance before encoding */
- if (m_dpb->m_picSymFreeList)
+ if (m_dpb->m_frameDataFreeList)
{
- frameEnc->m_encData = m_dpb->m_picSymFreeList;
- m_dpb->m_picSymFreeList = m_dpb->m_picSymFreeList->m_freeListNext;
+ frameEnc->m_encData = m_dpb->m_frameDataFreeList;
+ m_dpb->m_frameDataFreeList = m_dpb->m_frameDataFreeList->m_freeListNext;
frameEnc->reinit(m_sps);
}
else
@@ -647,10 +643,6 @@
slice->m_pps = &m_pps;
slice->m_maxNumMergeCand = m_param->maxNumMergeCand;
slice->m_endCUAddr = slice->realEndAddress(m_sps.numCUsInFrame * NUM_4x4_PARTITIONS);
- frameEnc->m_reconPic->m_cuOffsetC = m_cuOffsetC;
- frameEnc->m_reconPic->m_cuOffsetY = m_cuOffsetY;
- frameEnc->m_reconPic->m_buOffsetC = m_buOffsetC;
- frameEnc->m_reconPic->m_buOffsetY = m_buOffsetY;
}
curEncoder->m_rce.encodeOrder = m_encodedFrameNum++;
diff -r faecefdb2ae9 -r 1b2e5e8ccb85 source/encoder/encoder.h
--- a/source/encoder/encoder.h Tue Aug 11 14:28:23 2015 +0000
+++ b/source/encoder/encoder.h Tue Aug 11 22:31:57 2015 +0000
@@ -93,13 +93,6 @@
int m_numPools;
int m_curEncoder;
- /* cached PicYuv offset arrays, shared by all instances of
- * PicYuv created by this encoder */
- intptr_t* m_cuOffsetY;
- intptr_t* m_cuOffsetC;
- intptr_t* m_buOffsetY;
- intptr_t* m_buOffsetC;
-
/* Collect statistics globally */
EncStats m_analyzeAll;
EncStats m_analyzeI;
diff -r faecefdb2ae9 -r 1b2e5e8ccb85 source/encoder/frameencoder.cpp
--- a/source/encoder/frameencoder.cpp Tue Aug 11 14:28:23 2015 +0000
+++ b/source/encoder/frameencoder.cpp Tue Aug 11 22:31:57 2015 +0000
@@ -35,10 +35,6 @@
#include "slicetype.h"
#include "nal.h"
-#if HAVE_LIBNUMA
-#include <numa.h>
-#endif
-
namespace X265_NS {
void weightAnalyse(Slice& slice, Frame& frame, x265_param& param);
@@ -333,28 +329,6 @@
if (m_frame->m_lowres.bKeyframe && m_param->bRepeatHeaders)
m_top->getStreamHeaders(m_nalList, m_entropyCoder, m_bs);
-<<<<<<< local
-#if (defined(_WIN32_WINNT) && _WIN32_WINNT >= _WIN32_WINNT_WIN7) || HAVE_LIBNUMA
- int numaNode = m_pool ? m_pool->m_numaNode : 0;
-#else
- int numaNode = 0;
-#endif
-
- /* Claim this frame as being encoded by this NUMA node */
- // FIXME - Trying to see if copying recon to all NUMA nodes is better
- // m_frame->m_encData->allocRecon(m_top->m_sps, numaNode); /* TODO: bail if failure */
- for (int i = 0; i < ThreadPool::getNumaNodeCount(); i++)
- {
- // TODO - Fix for non-numa and windows
- numa_set_preferred(i) ;
- m_frame->m_encData->allocRecon(m_top->m_sps, i); /* TODO: bail if failure */
- }
- // TODO - Fix for non-numa and windows
- numa_set_preferred(numaNode) ;
- m_frame->m_encData->m_ownerNode = numaNode;
-
-=======
->>>>>>> other
// Weighted Prediction parameters estimation.
bool bUseWeightP = slice->m_sliceType == P_SLICE && slice->m_pps->bUseWeightPred;
bool bUseWeightB = slice->m_sliceType == B_SLICE && slice->m_pps->bUseWeightedBiPred;
@@ -383,7 +357,8 @@
WeightParam *w = NULL;
if ((bUseWeightP || bUseWeightB) && slice->m_weightPredTable[l][ref][0].bPresentFlag)
w = slice->m_weightPredTable[l][ref];
- m_mref[l][ref].init(slice->m_refPicList[l][ref]->m_reconPic, w, *m_param);
+ slice->m_refReconPicList[l][ref] = slice->m_refFrameList[l][ref]->m_reconPic;
+ m_mref[l][ref].init(slice->m_refReconPicList[l][ref], w, *m_param);
}
}
@@ -503,7 +478,7 @@
/* CQP and CRF (without capped VBV) doesn't use mid-frame statistics to
* tune RateControl parameters for other frames.
* Hence, for these modes, update m_startEndOrder and unlock RC for previous threads waiting in
- * RateControlEnd here, after the slicecontexts are initialized. For the rest - ABR
+ * RateControlEnd here, after the slice contexts are initialized. For the rest - ABR
* and VBV, unlock only after rateControlUpdateStats of this frame is called */
if (m_param->rc.rateControlMode != X265_RC_ABR && !m_top->m_rateControl->m_isVbv)
{
@@ -527,7 +502,7 @@
{
for (int ref = 0; ref < slice->m_numRefIdx[l]; ref++)
{
- Frame *refpic = slice->m_refPicList[l][ref];
+ Frame *refpic = slice->m_refFrameList[l][ref];
uint32_t reconRowCount = refpic->m_reconRowCount.get();
while ((reconRowCount != m_numRows) && (reconRowCount < row + m_refLagRows))
@@ -566,7 +541,7 @@
int list = l;
for (int ref = 0; ref < slice->m_numRefIdx[list]; ref++)
{
- Frame *refpic = slice->m_refPicList[list][ref];
+ Frame *refpic = slice->m_refFrameList[list][ref];
uint32_t reconRowCount = refpic->m_reconRowCount.get();
while ((reconRowCount != m_numRows) && (reconRowCount < i + m_refLagRows))
@@ -723,7 +698,7 @@
{
for (int ref = 0; ref < slice->m_numRefIdx[l]; ref++)
{
- Frame *refpic = slice->m_refPicList[l][ref];
+ Frame *refpic = slice->m_refFrameList[l][ref];
ATOMIC_DEC(&refpic->m_countRefEncoders);
}
}
diff -r faecefdb2ae9 -r 1b2e5e8ccb85 source/encoder/framefilter.cpp
--- a/source/encoder/framefilter.cpp Tue Aug 11 14:28:23 2015 +0000
+++ b/source/encoder/framefilter.cpp Tue Aug 11 22:31:57 2015 +0000
@@ -233,10 +233,10 @@
}
if (m_param->bEnableSsim && m_ssimBuf)
{
- pixel *rec = m_frame->m_reconPic->m_picOrg[0];
+ pixel *rec = reconPic->m_picOrg[0];
pixel *fenc = m_frame->m_fencPic->m_picOrg[0];
intptr_t stride1 = m_frame->m_fencPic->m_stride;
- intptr_t stride2 = m_frame->m_reconPic->m_stride;
+ intptr_t stride2 = reconPic->m_stride;
uint32_t bEnd = ((row + 1) == (this->m_numRows - 1));
uint32_t bStart = (row == 0);
uint32_t minPixY = row * g_maxCUSize - 4 * !bStart;
diff -r faecefdb2ae9 -r 1b2e5e8ccb85 source/encoder/ratecontrol.cpp
--- a/source/encoder/ratecontrol.cpp Tue Aug 11 14:28:23 2015 +0000
+++ b/source/encoder/ratecontrol.cpp Tue Aug 11 22:31:57 2015 +0000
@@ -1345,10 +1345,10 @@
{
/* B-frames don't have independent rate control, but rather get the
* average QP of the two adjacent P-frames + an offset */
- Slice* prevRefSlice = m_curSlice->m_refPicList[0][0]->m_encData->m_slice;
- Slice* nextRefSlice = m_curSlice->m_refPicList[1][0]->m_encData->m_slice;
- double q0 = m_curSlice->m_refPicList[0][0]->m_encData->m_avgQpRc;
- double q1 = m_curSlice->m_refPicList[1][0]->m_encData->m_avgQpRc;
+ Slice* prevRefSlice = m_curSlice->m_refFrameList[0][0]->m_encData->m_slice;
+ Slice* nextRefSlice = m_curSlice->m_refFrameList[1][0]->m_encData->m_slice;
+ double q0 = m_curSlice->m_refFrameList[0][0]->m_encData->m_avgQpRc;
+ double q1 = m_curSlice->m_refFrameList[1][0]->m_encData->m_avgQpRc;
bool i0 = prevRefSlice->m_sliceType == I_SLICE;
bool i1 = nextRefSlice->m_sliceType == I_SLICE;
int dt0 = abs(m_curSlice->m_poc - prevRefSlice->m_poc);
@@ -1364,9 +1364,9 @@
q0 = q1;
}
}
- if (prevRefSlice->m_sliceType == B_SLICE && IS_REFERENCED(m_curSlice->m_refPicList[0][0]))
+ if (prevRefSlice->m_sliceType == B_SLICE && IS_REFERENCED(m_curSlice->m_refFrameList[0][0]))
q0 -= m_pbOffset / 2;
- if (nextRefSlice->m_sliceType == B_SLICE && IS_REFERENCED(m_curSlice->m_refPicList[1][0]))
+ if (nextRefSlice->m_sliceType == B_SLICE && IS_REFERENCED(m_curSlice->m_refFrameList[1][0]))
q1 -= m_pbOffset / 2;
if (i0 && i1)
q = (q0 + q1) / 2 + m_ipOffset;
@@ -1483,7 +1483,7 @@
* Then bias the quant up or down if total size so far was far from
* the target.
* Result: Depending on the value of rate_tolerance, there is a
- * tradeoff between quality and bitrate precision. But at large
+ * trade-off between quality and bitrate precision. But at large
* tolerances, the bit distribution approaches that of 2pass. */
double overflow = 1;
@@ -1832,7 +1832,7 @@
double qScale = x265_qp2qScale(qpVbv);
FrameData& curEncData = *curFrame->m_encData;
int picType = curEncData.m_slice->m_sliceType;
- Frame* refFrame = curEncData.m_slice->m_refPicList[0][0];
+ Frame* refFrame = curEncData.m_slice->m_refFrameList[0][0];
uint32_t maxRows = curEncData.m_slice->m_sps->numCuInHeight;
uint32_t maxCols = curEncData.m_slice->m_sps->numCuInWidth;
@@ -1921,7 +1921,7 @@
updatePredictor(rce->rowPred[0], qScaleVbv, (double)rowSatdCost, encodedBits);
if (curEncData.m_slice->m_sliceType == P_SLICE)
{
- Frame* refFrame = curEncData.m_slice->m_refPicList[0][0];
+ Frame* refFrame = curEncData.m_slice->m_refFrameList[0][0];
if (qpVbv < refFrame->m_encData->m_rowStat[row].diagQp)
{
uint64_t intraRowSatdCost = curEncData.m_rowStat[row].diagIntraSatd;
diff -r faecefdb2ae9 -r 1b2e5e8ccb85 source/encoder/sao.cpp
--- a/source/encoder/sao.cpp Tue Aug 11 14:28:23 2015 +0000
+++ b/source/encoder/sao.cpp Tue Aug 11 22:31:57 2015 +0000
@@ -243,12 +243,13 @@
void SAO::processSaoCu(int addr, int typeIdx, int plane)
{
int x, y;
- const CUData* cu = m_frame->m_encData->getPicCTU(addr);
- pixel* rec = m_frame->m_reconPic->getPlaneAddr(plane, addr);
- intptr_t stride = plane ? m_frame->m_reconPic->m_strideC : m_frame->m_reconPic->m_stride;
+ PicYuv* reconPic = m_frame->m_reconPic;
+ pixel* rec = reconPic->getPlaneAddr(plane, addr);
+ intptr_t stride = plane ? reconPic->m_strideC : reconPic->m_stride;
uint32_t picWidth = m_param->sourceWidth;
uint32_t picHeight = m_param->sourceHeight;
- int ctuWidth = g_maxCUSize;
+ const CUData* cu = m_frame->m_encData->getPicCTU(addr);
+ int ctuWidth = g_maxCUSize;
int ctuHeight = g_maxCUSize;
uint32_t lpelx = cu->m_cuPelX;
uint32_t tpely = cu->m_cuPelY;
@@ -572,7 +573,8 @@
/* Process SAO all units */
void SAO::processSaoUnitRow(SaoCtuParam* ctuParam, int idxY, int plane)
{
- intptr_t stride = plane ? m_frame->m_reconPic->m_strideC : m_frame->m_reconPic->m_stride;
+ PicYuv* reconPic = m_frame->m_reconPic;
+ intptr_t stride = plane ? reconPic->m_strideC : reconPic->m_stride;
uint32_t picWidth = m_param->sourceWidth;
int ctuWidth = g_maxCUSize;
int ctuHeight = g_maxCUSize;
@@ -585,12 +587,12 @@
if (!idxY)
{
- pixel* rec = m_frame->m_reconPic->m_picOrg[plane];
+ pixel* rec = reconPic->m_picOrg[plane];
memcpy(m_tmpU1[plane], rec, sizeof(pixel) * picWidth);
}
int addr = idxY * m_numCuInWidth;
- pixel* rec = plane ? m_frame->m_reconPic->getChromaAddr(plane, addr) : m_frame->m_reconPic->getLumaAddr(addr);
+ pixel* rec = plane ? reconPic->getChromaAddr(plane, addr) : reconPic->getLumaAddr(addr);
for (int i = 0; i < ctuHeight + 1; i++)
{
@@ -635,7 +637,7 @@
}
else if (idxX != (m_numCuInWidth - 1))
{
- rec = plane ? m_frame->m_reconPic->getChromaAddr(plane, addr) : m_frame->m_reconPic->getLumaAddr(addr);
+ rec = plane ? reconPic->getChromaAddr(plane, addr) : reconPic->getLumaAddr(addr);
for (int i = 0; i < ctuHeight + 1; i++)
{
@@ -671,12 +673,13 @@
/* Calculate SAO statistics for current CTU without non-crossing slice */
void SAO::calcSaoStatsCu(int addr, int plane)
{
+ const PicYuv* reconPic = m_frame->m_reconPic;
const CUData* cu = m_frame->m_encData->getPicCTU(addr);
const pixel* fenc0 = m_frame->m_fencPic->getPlaneAddr(plane, addr);
- const pixel* rec0 = m_frame->m_reconPic->getPlaneAddr(plane, addr);
+ const pixel* rec0 = reconPic->getPlaneAddr(plane, addr);
const pixel* fenc;
const pixel* rec;
- intptr_t stride = plane ? m_frame->m_reconPic->m_strideC : m_frame->m_reconPic->m_stride;
+ intptr_t stride = plane ? reconPic->m_strideC : reconPic->m_stride;
uint32_t picWidth = m_param->sourceWidth;
uint32_t picHeight = m_param->sourceHeight;
int ctuWidth = g_maxCUSize;
@@ -825,9 +828,10 @@
int x, y;
const CUData* cu = frame->m_encData->getPicCTU(addr);
+ const PicYuv* reconPic = m_frame->m_reconPic;
const pixel* fenc;
const pixel* rec;
- intptr_t stride = m_frame->m_reconPic->m_stride;
+ intptr_t stride = reconPic->m_stride;
uint32_t picWidth = m_param->sourceWidth;
uint32_t picHeight = m_param->sourceHeight;
int ctuWidth = g_maxCUSize;
@@ -861,7 +865,7 @@
{
if (plane == 1)
{
- stride = frame->m_reconPic->m_strideC;
+ stride = reconPic->m_strideC;
picWidth >>= m_hChromaShift;
picHeight >>= m_vChromaShift;
ctuWidth >>= m_hChromaShift;
@@ -881,7 +885,7 @@
count = m_countPreDblk[addr][plane][SAO_BO];
const pixel* fenc0 = m_frame->m_fencPic->getPlaneAddr(plane, addr);
- const pixel* rec0 = m_frame->m_reconPic->getPlaneAddr(plane, addr);
+ const pixel* rec0 = reconPic->getPlaneAddr(plane, addr);
fenc = fenc0;
rec = rec0;
diff -r faecefdb2ae9 -r 1b2e5e8ccb85 source/encoder/search.cpp
--- a/source/encoder/search.cpp Tue Aug 11 14:28:23 2015 +0000
+++ b/source/encoder/search.cpp Tue Aug 11 22:31:57 2015 +0000
@@ -446,8 +446,9 @@
}
// set reconstruction for next intra prediction blocks if full TU prediction won
- pixel* picReconY = m_frame->m_reconPic->getLumaAddr(cu.m_cuAddr, cuGeom.absPartIdx + absPartIdx);
- intptr_t picStride = m_frame->m_reconPic->m_stride;
+ PicYuv* reconPic = m_frame->m_reconPic;
+ pixel* picReconY = reconPic->getLumaAddr(cu.m_cuAddr, cuGeom.absPartIdx + absPartIdx);
+ intptr_t picStride = reconPic->m_stride;
primitives.cu[sizeIdx].copy_pp(picReconY, picStride, reconQt, reconQtStride);
outCost.rdcost += fullCost.rdcost;
@@ -611,8 +612,9 @@
}
// set reconstruction for next intra prediction blocks
- pixel* picReconY = m_frame->m_reconPic->getLumaAddr(cu.m_cuAddr, cuGeom.absPartIdx + absPartIdx);
- intptr_t picStride = m_frame->m_reconPic->m_stride;
+ PicYuv* reconPic = m_frame->m_reconPic;
+ pixel* picReconY = reconPic->getLumaAddr(cu.m_cuAddr, cuGeom.absPartIdx + absPartIdx);
+ intptr_t picStride = reconPic->m_stride;
primitives.cu[sizeIdx].copy_pp(picReconY, picStride, reconQt, reconQtStride);
outCost.rdcost += fullCost.rdcost;
@@ -661,8 +663,9 @@
uint32_t sizeIdx = log2TrSize - 2;
primitives.cu[sizeIdx].calcresidual(fenc, pred, residual, stride);
- pixel* picReconY = m_frame->m_reconPic->getLumaAddr(cu.m_cuAddr, cuGeom.absPartIdx + absPartIdx);
- intptr_t picStride = m_frame->m_reconPic->m_stride;
+ PicYuv* reconPic = m_frame->m_reconPic;
+ pixel* picReconY = reconPic->getLumaAddr(cu.m_cuAddr, cuGeom.absPartIdx + absPartIdx);
+ intptr_t picStride = reconPic->m_stride;
uint32_t numSig = m_quant.transformNxN(cu, fenc, stride, residual, stride, coeffY, log2TrSize, TEXT_LUMA, absPartIdx, false);
if (numSig)
@@ -821,8 +824,9 @@
coeff_t* coeffC = m_rqt[qtLayer].coeffRQT[chromaId] + coeffOffsetC;
pixel* reconQt = m_rqt[qtLayer].reconQtYuv.getChromaAddr(chromaId, absPartIdxC);
uint32_t reconQtStride = m_rqt[qtLayer].reconQtYuv.m_csize;
- pixel* picReconC = m_frame->m_reconPic->getChromaAddr(chromaId, cu.m_cuAddr, cuGeom.absPartIdx + absPartIdxC);
- intptr_t picStride = m_frame->m_reconPic->m_strideC;
+ PicYuv* reconPic = m_frame->m_reconPic;
+ pixel* picReconC = reconPic->getChromaAddr(chromaId, cu.m_cuAddr, cuGeom.absPartIdx + absPartIdxC);
+ intptr_t picStride = reconPic->m_strideC;
uint32_t chromaPredMode = cu.m_chromaIntraDir[absPartIdxC];
if (chromaPredMode == DM_CHROMA_IDX)
@@ -998,8 +1002,9 @@
cu.setCbfPartRange(bCbf << tuDepth, ttype, absPartIdxC, tuIterator.absPartIdxStep);
cu.setTransformSkipPartRange(bTSkip, ttype, absPartIdxC, tuIterator.absPartIdxStep);
- pixel* reconPicC = m_frame->m_reconPic->getChromaAddr(chromaId, cu.m_cuAddr, cuGeom.absPartIdx + absPartIdxC);
- intptr_t picStride = m_frame->m_reconPic->m_strideC;
+ PicYuv* reconPic = m_frame->m_reconPic;
+ pixel* reconPicC = reconPic->getChromaAddr(chromaId, cu.m_cuAddr, cuGeom.absPartIdx + absPartIdxC);
+ intptr_t picStride = reconPic->m_strideC;
primitives.cu[sizeIdxC].copy_pp(reconPicC, picStride, reconQt, reconQtStride);
outDist += bDist;
@@ -1108,8 +1113,9 @@
int16_t* residual = resiYuv.getChromaAddr(chromaId, absPartIdxC);
uint32_t coeffOffsetC = absPartIdxC << (LOG2_UNIT_SIZE * 2 - (m_hChromaShift + m_vChromaShift));
coeff_t* coeffC = cu.m_trCoeff[ttype] + coeffOffsetC;
- pixel* picReconC = m_frame->m_reconPic->getChromaAddr(chromaId, cu.m_cuAddr, cuGeom.absPartIdx + absPartIdxC);
- intptr_t picStride = m_frame->m_reconPic->m_strideC;
+ PicYuv* reconPic = m_frame->m_reconPic;
+ pixel* picReconC = reconPic->getChromaAddr(chromaId, cu.m_cuAddr, cuGeom.absPartIdx + absPartIdxC);
+ intptr_t picStride = reconPic->m_strideC;
uint32_t chromaPredMode = cu.m_chromaIntraDir[absPartIdxC];
if (chromaPredMode == DM_CHROMA_IDX)
@@ -1591,10 +1597,11 @@
* output recon picture, so it cannot proceed in parallel with anything else when doing INTRA_NXN. Also
* it is not updating m_rdContexts[depth].cur for the later PUs which I suspect is slightly wrong. I think
* that the contexts should be tracked through each PU */
- pixel* dst = m_frame->m_reconPic->getLumaAddr(cu.m_cuAddr, cuGeom.absPartIdx + absPartIdx);
- uint32_t dststride = m_frame->m_reconPic->m_stride;
- const pixel* src = reconYuv->getLumaAddr(absPartIdx);
- uint32_t srcstride = reconYuv->m_size;
+ PicYuv* reconPic = m_frame->m_reconPic;
+ pixel* dst = reconPic->getLumaAddr(cu.m_cuAddr, cuGeom.absPartIdx + absPartIdx);
+ uint32_t dststride = reconPic->m_stride;
+ const pixel* src = reconYuv->getLumaAddr(absPartIdx);
+ uint32_t srcstride = reconYuv->m_size;
primitives.cu[log2TrSize - 2].copy_pp(dst, dststride, src, srcstride);
}
}
@@ -1757,15 +1764,16 @@
if (!tuIterator.isLastSection())
{
uint32_t zorder = cuGeom.absPartIdx + absPartIdxC;
- uint32_t dststride = m_frame->m_reconPic->m_strideC;
+ PicYuv* reconPic = m_frame->m_reconPic;
+ uint32_t dststride = reconPic->m_strideC;
const pixel* src;
pixel* dst;
- dst = m_frame->m_reconPic->getCbAddr(cu.m_cuAddr, zorder);
+ dst = reconPic->getCbAddr(cu.m_cuAddr, zorder);
src = reconYuv.getCbAddr(absPartIdxC);
primitives.chroma[m_csp].cu[size].copy_pp(dst, dststride, src, reconYuv.m_csize);
- dst = m_frame->m_reconPic->getCrAddr(cu.m_cuAddr, zorder);
+ dst = reconPic->getCrAddr(cu.m_cuAddr, zorder);
src = reconYuv.getCrAddr(absPartIdxC);
primitives.chroma[m_csp].cu[size].copy_pp(dst, dststride, src, reconYuv.m_csize);
}
@@ -1866,7 +1874,7 @@
/* find the lowres motion vector from lookahead in middle of current PU */
MV Search::getLowresMV(const CUData& cu, const PredictionUnit& pu, int list, int ref)
{
- int diffPoc = abs(m_slice->m_poc - m_slice->m_refPicList[list][ref]->m_poc);
+ int diffPoc = abs(m_slice->m_poc - m_slice->m_refPOCList[list][ref]);
if (diffPoc > m_param->bframes + 1)
/* poc difference is out of range for lookahead */
return 0;
@@ -1906,7 +1914,7 @@
else
{
cu.clipMv(mvCand);
- predInterLumaPixel(pu, tmpPredYuv, *m_slice->m_refPicList[list][ref]->m_reconPic, mvCand);
+ predInterLumaPixel(pu, tmpPredYuv, *m_slice->m_refReconPicList[list][ref], mvCand);
costs[i] = m_me.bufSAD(tmpPredYuv.getLumaAddr(pu.puAbsPartIdx), tmpPredYuv.m_size);
}
}
@@ -2197,8 +2205,8 @@
}
else
{
- PicYuv* refPic0 = slice->m_refPicList[0][bestME[0].ref]->m_reconPic;
- PicYuv* refPic1 = slice->m_refPicList[1][bestME[1].ref]->m_reconPic;
+ PicYuv* refPic0 = slice->m_refReconPicList[0][bestME[0].ref];
+ PicYuv* refPic1 = slice->m_refReconPicList[1][bestME[1].ref];
Yuv* bidirYuv = m_rqt[cuGeom.depth].bidirPredYuv;
/* Generate reference subpels */
diff -r faecefdb2ae9 -r 1b2e5e8ccb85 source/encoder/slicetype.cpp
--- a/source/encoder/slicetype.cpp Tue Aug 11 14:28:23 2015 +0000
+++ b/source/encoder/slicetype.cpp Tue Aug 11 22:31:57 2015 +0000
@@ -714,16 +714,16 @@
case P_SLICE:
b = p1 = poc - l0poc;
- frames[p0] = &slice->m_refPicList[0][0]->m_lowres;
+ frames[p0] = &slice->m_refFrameList[0][0]->m_lowres;
frames[b] = &curFrame->m_lowres;
break;
case B_SLICE:
b = poc - l0poc;
p1 = b + l1poc - poc;
- frames[p0] = &slice->m_refPicList[0][0]->m_lowres;
+ frames[p0] = &slice->m_refFrameList[0][0]->m_lowres;
frames[b] = &curFrame->m_lowres;
- frames[p1] = &slice->m_refPicList[1][0]->m_lowres;
+ frames[p1] = &slice->m_refFrameList[1][0]->m_lowres;
break;
default:
diff -r faecefdb2ae9 -r 1b2e5e8ccb85 source/encoder/weightPrediction.cpp
--- a/source/encoder/weightPrediction.cpp Tue Aug 11 14:28:23 2015 +0000
+++ b/source/encoder/weightPrediction.cpp Tue Aug 11 22:31:57 2015 +0000
@@ -259,7 +259,7 @@
for (int list = 0; list < cache.numPredDir; list++)
{
WeightParam *weights = wp[list][0];
- Frame *refFrame = slice.m_refPicList[list][0];
+ Frame *refFrame = slice.m_refFrameList[list][0];
Lowres& refLowres = refFrame->m_lowres;
int diffPoc = abs(curPoc - refFrame->m_poc);
More information about the x265-devel
mailing list