<div dir="ltr"><br><div class="gmail_extra"><br><br><div class="gmail_quote">On Thu, Oct 17, 2013 at 2:22 AM, <span dir="ltr"><<a href="mailto:aarthi@multicorewareinc.com" target="_blank">aarthi@multicorewareinc.com</a>></span> wrote:<br>
<blockquote class="gmail_quote" style="margin:0 0 0 .8ex;border-left:1px #ccc solid;padding-left:1ex"># HG changeset patch<br>
# User Aarthi Thirumalai<<a href="mailto:aarthi@multicorewareinc.com">aarthi@multicorewareinc.com</a>><br>
# Date 1381994486 -19800<br>
# Thu Oct 17 12:51:26 2013 +0530<br>
# Node ID 4e56bc9c1fbbc0ed9cbdb13de6af14d55eb2715e<br>
# Parent aad785763f1d38473177e3f1293d3012d79239ca<br>
rc: implement Adaptive Quantization.<br>
<br>
added functions to compute AC Energy per CU for all planes,<br>
calculate qpAqOffset for each CU<br></blockquote><div><br></div><div>Looks good; I'll queue this one so you only need to resend the first</div><div> </div><blockquote class="gmail_quote" style="margin:0 0 0 .8ex;border-left:1px #ccc solid;padding-left:1ex">
<br>
diff -r aad785763f1d -r 4e56bc9c1fbb source/encoder/encoder.cpp<br>
--- a/source/encoder/encoder.cpp Thu Oct 17 11:31:33 2013 +0530<br>
+++ b/source/encoder/encoder.cpp Thu Oct 17 12:51:26 2013 +0530<br>
@@ -185,6 +185,8 @@<br>
<br>
// Encoder holds a reference count until collecting stats<br>
ATOMIC_INC(&pic->m_countRefEncoders);<br>
+ if (param.rc.aqMode)<br>
+ m_rateControl->calcAdaptiveQuantFrame(pic);<br>
m_lookahead->addPicture(pic, pic_in->sliceType);<br>
}<br>
<br>
diff -r aad785763f1d -r 4e56bc9c1fbb source/encoder/ratecontrol.cpp<br>
--- a/source/encoder/ratecontrol.cpp Thu Oct 17 11:31:33 2013 +0530<br>
+++ b/source/encoder/ratecontrol.cpp Thu Oct 17 12:51:26 2013 +0530<br>
@@ -50,6 +50,85 @@<br>
return 0.85 * pow(2.0, (qp - 12.0) / 6.0);<br>
}<br>
<br>
+/* Compute variance to derive AC energy of each block */<br>
+static inline uint32_t acEnergyVar(uint64_t sum_ssd, int shift)<br>
+{<br>
+ uint32_t sum = (uint32_t)sum_ssd;<br>
+ uint32_t ssd = (uint32_t)(sum_ssd >> 32);<br>
+<br>
+ return ssd - ((uint64_t)sum * sum >> shift);<br>
+}<br>
+<br>
+/* Find the energy of each block in Y/Cb/Cr plane */<br>
+static inline uint32_t acEnergyPlane(pixel* src, int srcStride, int bChroma)<br>
+{<br>
+ int blockStride = FENC_STRIDE >> 3;<br>
+<br>
+ if (bChroma)<br>
+ {<br>
+ ALIGN_VAR_8(pixel, pix[8 * 8]);<br>
+ primitives.blockcpy_pp(8, 8, pix, blockStride, src, srcStride);<br>
+ return acEnergyVar(primitives.var[PARTITION_8x8](pix, blockStride), 6);<br>
+ }<br>
+ else<br>
+ return acEnergyVar(primitives.var[PARTITION_16x16](src, srcStride), 8);<br>
+}<br>
+<br>
+/* Find the total AC energy of each CU in all planes */<br>
+double RateControl::acEnergyCu(TComPic* pic, uint32_t cuAddr)<br>
+{<br>
+ uint32_t var = 0;<br>
+ double avgQp = 0;<br>
+ pixel* srcLuma = pic->getPicYuvOrg()->getLumaAddr(cuAddr);<br>
+ pixel* srcCb = pic->getPicYuvOrg()->getCbAddr(cuAddr);<br>
+ pixel* srcCr = pic->getPicYuvOrg()->getCrAddr(cuAddr);<br>
+ UInt blockWidth = g_maxCUWidth >> 2;<br>
+ UInt blockHeight = g_maxCUHeight >> 2;<br>
+ UInt frameStride = pic->getPicYuvOrg()->getStride();<br>
+ UInt cStride = pic->getPicYuvOrg()->getCStride();<br>
+<br>
+ // Calculate Qp offset for each 16x16 block in the CU and average them over entire CU<br>
+ for (UInt h = 0, cnt = 0; h < g_maxCUHeight; h += blockHeight)<br>
+ {<br>
+ for (UInt w = 0; w < g_maxCUWidth; w += blockWidth, cnt++)<br>
+ {<br>
+ UInt blockOffsetLuma = w + (h * frameStride);<br>
+ UInt blockOffsetChroma = (w >> 1) + ((h >> 1) * cStride);<br>
+ var = acEnergyPlane(srcLuma + blockOffsetLuma, frameStride, 0);<br>
+ var += acEnergyPlane(srcCb + blockOffsetChroma, cStride, 1);<br>
+ var += acEnergyPlane(srcCr + blockOffsetChroma, cStride, 1);<br>
+ avgQp += cfg->param.rc.aqStrength * (X265_LOG2(X265_MAX(var, 1)) - (14.427f));<br>
+ }<br>
+ }<br>
+<br>
+ avgQp /= 16;<br>
+ x265_emms();<br>
+ return avgQp;<br>
+}<br>
+<br>
+void RateControl::calcAdaptiveQuantFrame(TComPic *pic)<br>
+{<br>
+ double strength;<br>
+<br>
+ /* Actual adaptive quantization */<br>
+ if (cfg->param.rc.aqMode)<br>
+ {<br>
+ strength = cfg->param.rc.aqStrength * 1.0397f;<br>
+ int maxRows = pic->getPicSym()->getFrameHeightInCU();<br>
+ int maxCols = pic->getPicSym()->getFrameWidthInCU();<br>
+ for (int cu_y = 0; cu_y < maxRows; cu_y++)<br>
+ {<br>
+ for (int cu_x = 0; cu_x < maxCols; cu_x++)<br>
+ {<br>
+ double qp_adj;<br>
+ int cu_xy = maxCols * cu_y + cu_x;<br>
+ qp_adj = acEnergyCu(pic, cu_xy);<br>
+ pic->m_qpAqOffset[cu_xy] = qp_adj;<br>
+ }<br>
+ }<br>
+ }<br>
+}<br>
+<br>
RateControl::RateControl(TEncCfg * _cfg)<br>
{<br>
this->cfg = _cfg;<br>
diff -r aad785763f1d -r 4e56bc9c1fbb source/encoder/ratecontrol.h<br>
--- a/source/encoder/ratecontrol.h Thu Oct 17 11:31:33 2013 +0530<br>
+++ b/source/encoder/ratecontrol.h Thu Oct 17 12:51:26 2013 +0530<br>
@@ -80,7 +80,7 @@<br>
<br>
// to be called for each frame to process RateCOntrol and set QP<br>
void rateControlStart(TComPic* pic, Lookahead *, RateControlEntry* rce);<br>
-<br>
+ void calcAdaptiveQuantFrame(TComPic *pic);<br>
int rateControlEnd(int64_t bits, RateControlEntry* rce);<br>
<br>
protected:<br>
@@ -88,6 +88,7 @@<br>
double getQScale(RateControlEntry *rce, double rateFactor);<br>
double rateEstimateQscale(RateControlEntry *rce); // main logic for calculating QP based on ABR<br>
void accumPQpUpdate();<br>
+ double acEnergyCu(TComPic* pic, uint32_t cuAddr);<br>
};<br>
}<br>
<br>
_______________________________________________<br>
x265-devel mailing list<br>
<a href="mailto:x265-devel@videolan.org">x265-devel@videolan.org</a><br>
<a href="https://mailman.videolan.org/listinfo/x265-devel" target="_blank">https://mailman.videolan.org/listinfo/x265-devel</a><br>
</blockquote></div><br><br clear="all"><div><br></div>-- <br>Steve Borho
</div></div>