Implement plane_copy_sse2, optimized plain copying of frames.

1 files changed, 32 insertions, 0 deletions

diff --git a/decoder/LAVVideo/pixconv/pixconv.cpp b/decoder/LAVVideo/pixconv/pixconv.cpp
index 68a5e4a0..b900c1e9 100644
--- a/decoder/LAVVideo/pixconv/pixconv.cpp
+++ b/decoder/LAVVideo/pixconv/pixconv.cpp
@@ -19,6 +19,7 @@
 
 #include "stdafx.h"
 #include "pixconv_internal.h"
+#include "pixconv_sse2_templates.h"
 
 // 8x8 Bayes ordered dithering table, scaled to the 0-255 range for 16->8 conversion
 // stored as 16-bit unsigned for optimized SIMD access
@@ -56,3 +57,34 @@ DECLARE_CONV_FUNC_IMPL(plane_copy)
 
   return S_OK;
 }
+
+DECLARE_CONV_FUNC_IMPL(plane_copy_sse2)
+{
+  LAVOutPixFmtDesc desc = lav_pixfmt_desc[outputFormat];
+
+  const int widthBytes = width * desc.codedbytes;
+  const int planes = max(desc.planes, 1);
+
+  ptrdiff_t line, plane;
+
+  for (plane = 0; plane < planes; plane++) {
+    const int planeWidth = widthBytes / desc.planeWidth[plane];
+    const int planeHeight = height / desc.planeHeight[plane];
+    const ptrdiff_t srcPlaneStride = srcStride[plane];
+    const ptrdiff_t dstPlaneStride = dstStride[plane];
+    const uint8_t * const srcBuf = src[plane];
+          uint8_t * const dstBuf = dst[plane];
+
+    if ((dstPlaneStride % 16) == 0 && ((intptr_t)dstBuf % 16u) == 0) {
+      for (line = 0; line < planeHeight; ++line) {
+        PIXCONV_MEMCPY_ALIGNED(dstBuf + line * dstPlaneStride, srcBuf + line * srcPlaneStride, planeWidth);
+      }
+    } else {
+      for (line = 0; line < planeHeight; ++line) {
+        memcpy(dstBuf + line * dstPlaneStride, srcBuf + line * srcPlaneStride, planeWidth);
+      }
+    }
+  }
+
+  return S_OK;
+}