<div dir="ltr"># HG changeset patch<br># User Shazeb Nawaz Khan <<a href="mailto:shazeb@multicorewareinc.com">shazeb@multicorewareinc.com</a>><br># Date 1380713440 -19800<br># Node ID 09de1a4441d6b00c36c79c6ed90e296cbe144004<br>
# Parent  c5dc3e37d76772b5e26c6996ab0df5ce325f54a7<br>Templating weightUnidir primitive to support pixel inputs<br><br>diff -r c5dc3e37d767 -r 09de1a4441d6 source/Lib/TLibCommon/TComWeightPrediction.cpp<br>--- a/source/Lib/TLibCommon/TComWeightPrediction.cpp    Tue Oct 01 17:28:19 2013 +0530<br>
+++ b/source/Lib/TLibCommon/TComWeightPrediction.cpp    Wed Oct 02 17:00:40 2013 +0530<br>@@ -431,7 +431,7 @@<br>         srcStride = srcYuv0->m_width;<br>         dstStride  = outDstYuv->getStride();<br> <br>-        primitives.weightpUni(srcY0, dstY, srcStride, dstStride, width, height, w0, round, shift, offset);<br>
+        primitives.weightpUni((uint16_t *)srcY0, dstY, srcStride, dstStride, width, height, w0, round, shift, offset);<br>     }<br> <br>     // Chroma U : --------------------------------------------<br>@@ -447,7 +447,7 @@<br>
     width  >>= 1;<br>     height >>= 1;<br> <br>-    primitives.weightpUni(srcU0, dstU, srcStride, dstStride, width, height, w0, round, shift, offset);<br>+    primitives.weightpUni((uint16_t *)srcU0, dstU, srcStride, dstStride, width, height, w0, round, shift, offset);<br>
 <br>     // Chroma V : --------------------------------------------<br>     w0      = wp0[2].w;<br>@@ -455,7 +455,7 @@<br>     shift   = wp0[2].shift + shiftNum;<br>     round   = shift ? (1 << (shift - 1)) : 0;<br>
 <br>-    primitives.weightpUni(srcV0, dstV, srcStride, dstStride, width, height, w0, round, shift, offset);<br>+    primitives.weightpUni((uint16_t *)srcV0, dstV, srcStride, dstStride, width, height, w0, round, shift, offset);<br>
 }<br> <br> //=======================================================<br>diff -r c5dc3e37d767 -r 09de1a4441d6 source/common/pixel.cpp<br>--- a/source/common/pixel.cpp    Tue Oct 01 17:28:19 2013 +0530<br>+++ b/source/common/pixel.cpp    Wed Oct 02 17:00:40 2013 +0530<br>
@@ -514,7 +514,8 @@<br>     }<br> }<br> <br>-void weightUnidir(short *src, pixel *dst, intptr_t srcStride, intptr_t dstStride, int width, int height, int w0, int round, int shift, int offset)<br>+template <typename T><br>
+void weightUnidir(T *src, pixel *dst, intptr_t srcStride, intptr_t dstStride, int width, int height, int w0, int round, int shift, int offset)<br> {<br>     int x, y;<br>     for (y = height - 1; y >= 0; y--)<br>@@ -842,7 +843,8 @@<br>
     p.transpose[3] = transpose<32>;<br>     p.transpose[4] = transpose<64>;<br> <br>-    p.weightpUni = weightUnidir;<br>+    p.weightpUniPixel = weightUnidir<pixel>;<br>+    p.weightpUni = weightUnidir<uint16_t>;<br>
 <br>     p.pixelsub_sp = pixelsub_sp_c;<br>     p.pixeladd_pp = pixeladd_pp_c;<br>diff -r c5dc3e37d767 -r 09de1a4441d6 source/common/primitives.h<br>--- a/source/common/primitives.h    Tue Oct 01 17:28:19 2013 +0530<br>+++ b/source/common/primitives.h    Wed Oct 02 17:00:40 2013 +0530<br>
@@ -228,8 +228,8 @@<br> typedef void (*filterRowV_N_t)(short *midA, intptr_t midStride, pixel *dstA, pixel *dstB, pixel *dstC, intptr_t dstStride, int width, int height, int marginX, int marginY, int row, int isLastRow);<br>
 typedef void (*extendCURowBorder_t)(pixel* txt, intptr_t stride, int width, int height, int marginX);<br> <br>-<br>-typedef void (*weightpUni_t)(short *src, pixel *dst, intptr_t srcStride, intptr_t dstStride, int width, int height, int w0, int round, int shift, int offset);<br>
+typedef void (*weightpUniPixel_t)(pixel *src, pixel *dst, intptr_t srcStride, intptr_t dstStride, int width, int height, int w0, int round, int shift, int offset);<br>+typedef void (*weightpUni_t)(uint16_t *src, pixel *dst, intptr_t srcStride, intptr_t dstStride, int width, int height, int w0, int round, int shift, int offset);<br>
 typedef void (*scale_t)(pixel *dst, pixel *src, intptr_t stride);<br> typedef void (*downscale_t)(pixel *src0, pixel *dstf, pixel *dsth, pixel *dstv, pixel *dstc,<br>                             intptr_t src_stride, intptr_t dst_stride, int width, int height);<br>
@@ -286,6 +286,7 @@<br>     calcrecon_t     calcrecon[NUM_SQUARE_BLOCKS];<br>     transpose_t     transpose[NUM_SQUARE_BLOCKS];<br> <br>+    weightpUniPixel_t    weightpUniPixel;<br>     weightpUni_t    weightpUni;<br>     pixelsub_sp_t   pixelsub_sp;<br>
     pixeladd_ss_t   pixeladd_ss;<br>diff -r c5dc3e37d767 -r 09de1a4441d6 source/common/vec/pixel.inc<br>--- a/source/common/vec/pixel.inc    Tue Oct 01 17:28:19 2013 +0530<br>+++ b/source/common/vec/pixel.inc    Wed Oct 02 17:00:40 2013 +0530<br>
@@ -469,7 +469,9 @@<br>     p.transpose[2] = transpose16;<br>     p.transpose[3] = transpose32;<br>     p.transpose[4] = transpose<64>;<br>-    p.weightpUni = weightUnidir;<br>+    p.weightpUniPixel = weightUnidir<pixel>;<br>
+    p.weightpUni = weightUnidir<uint16_t>;<br>+<br> #endif<br> <br> #if !HIGH_BIT_DEPTH<br>diff -r c5dc3e37d767 -r 09de1a4441d6 source/common/vec/pixel8.inc<br>--- a/source/common/vec/pixel8.inc    Tue Oct 01 17:28:19 2013 +0530<br>
+++ b/source/common/vec/pixel8.inc    Wed Oct 02 17:00:40 2013 +0530<br>@@ -8573,7 +8573,8 @@<br>     }<br> }<br> <br>-void weightUnidir(short *src, pixel *dst, intptr_t srcStride, intptr_t dstStride, int width, int height, int w0, int round, int shift, int offset)<br>
+template <typename T><br>+void weightUnidir(T *src, pixel *dst, intptr_t srcStride, intptr_t dstStride, int width, int height, int w0, int round, int shift, int offset)<br> {<br>     int x, y;<br>     Vec8s tmp;<br>
diff -r c5dc3e37d767 -r 09de1a4441d6 source/test/pixelharness.cpp<br>--- a/source/test/pixelharness.cpp    Tue Oct 01 17:28:19 2013 +0530<br>+++ b/source/test/pixelharness.cpp    Wed Oct 02 17:00:40 2013 +0530<br>@@ -343,8 +343,8 @@<br>
     int offset = (rand() % 256) - 128;<br>     for (int i = 0; i < ITERS; i++)<br>     {<br>-        opt(sbuf1 + j, opt_dest, 64, 64, width, height, w0, round, shift, offset);<br>-        ref(sbuf1 + j, ref_dest, 64, 64, width, height, w0, round, shift, offset);<br>
+        opt((uint16_t*)sbuf1 + j, opt_dest, 64, 64, width, height, w0, round, shift, offset);<br>+        ref((uint16_t*)sbuf1 + j, ref_dest, 64, 64, width, height, w0, round, shift, offset);<br> <br>         if (memcmp(ref_dest, opt_dest, 64 * 64 * sizeof(pixel)))<br>
             return false;<br>@@ -355,6 +355,34 @@<br>     return true;<br> }<br> <br>+bool PixelHarness::check_weightpUni(weightpUniPixel_t ref, weightpUniPixel_t opt)<br>+{<br>+    ALIGN_VAR_16(pixel, ref_dest[64 * 64]);<br>
+    ALIGN_VAR_16(pixel, opt_dest[64 * 64]);<br>+<br>+    memset(ref_dest, 0, 64 * 64 * sizeof(pixel));<br>+    memset(opt_dest, 0, 64 * 64 * sizeof(pixel));<br>+    int j = 0;<br>+    int width = (2 * rand()) % 64;<br>+    int height = 8;<br>
+    int w0 = rand() % 256;<br>+    int shift = rand() % 12;<br>+    int round = shift ? (1 << (shift - 1)) : 0;<br>+    int offset = (rand() % 256) - 128;<br>+    for (int i = 0; i < ITERS; i++)<br>+    {<br>+        opt((pixel *)sbuf1 + j, opt_dest, 64, 64, width, height, w0, round, shift, offset);<br>
+        ref((pixel *)sbuf1 + j, ref_dest, 64, 64, width, height, w0, round, shift, offset);<br>+<br>+        if (memcmp(ref_dest, opt_dest, 64 * 64 * sizeof(pixel)))<br>+            return false;<br>+<br>+        j += INCR;<br>
+    }<br>+    <br>+    return true;<br>+}<br>+ <br> bool PixelHarness::check_pixelsub_sp(pixelsub_sp_t ref, pixelsub_sp_t opt)<br> {<br>     ALIGN_VAR_16(short, ref_dest[64 * 64]);<br>@@ -604,6 +632,24 @@<br>         }<br>
     }<br> <br>+    if (opt.weightpUniPixel)<br>+    {<br>+        if (!check_weightpUni(ref.weightpUniPixel, opt.weightpUniPixel))<br>+        {<br>+            printf("Weighted Prediction for Unidir failed!\n");<br>
+            return false;<br>+        }<br>+    }<br>+<br>+    if (opt.weightpUniPixel)<br>+    {<br>+        if (!check_weightpUni(ref.weightpUniPixel, opt.weightpUniPixel))<br>+        {<br>+            printf("Weighted Prediction for Unidir failed!\n");<br>
+            return false;<br>+        }<br>+    }<br>+<br>     if (opt.weightpUni)<br>     {<br>         if (!check_weightpUni(ref.weightpUni, opt.weightpUni))<br>@@ -751,10 +797,16 @@<br>         REPORT_SPEEDUP(opt.blockcpy_sc, ref.blockcpy_sc, 64, 64, (short*)pbuf1, FENC_STRIDE, (uint8_t*)pbuf2, STRIDE);<br>
     }<br> <br>+    if (opt.weightpUniPixel)<br>+    {<br>+        printf("WeightpUni");<br>+        REPORT_SPEEDUP(opt.weightpUniPixel, ref.weightpUniPixel, pbuf1, pbuf2, 64, 64, 32, 32, 128, 1 << 9, 10, 100);<br>
+    }<br>+<br>     if (opt.weightpUni)<br>     {<br>         printf("WeightpUni");<br>-        REPORT_SPEEDUP(opt.weightpUni, ref.weightpUni, sbuf1, pbuf1, 64, 64, 32, 32, 128, 1 << 9, 10, 100);<br>+        REPORT_SPEEDUP(opt.weightpUni, ref.weightpUni, (uint16_t*)sbuf1, pbuf1, 64, 64, 32, 32, 128, 1 << 9, 10, 100);<br>
     }<br> <br>     if (opt.pixelsub_sp)<br>diff -r c5dc3e37d767 -r 09de1a4441d6 source/test/pixelharness.h<br>--- a/source/test/pixelharness.h    Tue Oct 01 17:28:19 2013 +0530<br>+++ b/source/test/pixelharness.h    Wed Oct 02 17:00:40 2013 +0530<br>
@@ -46,6 +46,7 @@<br>     bool check_block_copy_s_c(blockcpy_sc_t ref, blockcpy_sc_t opt);<br>     bool check_calresidual(calcresidual_t ref, calcresidual_t opt);<br>     bool check_calcrecon(calcrecon_t ref, calcrecon_t opt);<br>
+    bool check_weightpUni(weightpUniPixel_t ref, weightpUniPixel_t opt);<br>     bool check_weightpUni(weightpUni_t ref, weightpUni_t opt);<br>     bool check_pixelsub_sp(pixelsub_sp_t ref, pixelsub_sp_t opt);<br>     bool check_pixeladd_ss(pixeladd_ss_t ref, pixeladd_ss_t opt);<br>
</div>