From 1421500a18cc51ce8f510c77f3586e07c79a7a5a Mon Sep 17 00:00:00 2001
From: Sergio Pedri <sergio0694@live.com>
Date: Fri, 20 Dec 2019 14:20:23 +0100
Subject: [PATCH] Added PositionAwarePixelShader types

---
 .../Delegates/PositionAwarePixelShader.cs     | 14 ++++
 .../DelegatePixelShaderProcessor{TPixel}.cs   |  2 +-
 .../PositionAwarePixelShaderProcessor.cs      | 38 +++++++++++
 ...sitionAwarePixelShaderProcessor{TPixel}.cs | 65 +++++++++++++++++++
 4 files changed, 118 insertions(+), 1 deletion(-)
 create mode 100644 src/ImageSharp/Processing/Delegates/PositionAwarePixelShader.cs
 create mode 100644 src/ImageSharp/Processing/Processors/PixelShading/PositionAwarePixelShaderProcessor.cs
 create mode 100644 src/ImageSharp/Processing/Processors/PixelShading/PositionAwarePixelShaderProcessor{TPixel}.cs

diff --git a/src/ImageSharp/Processing/Delegates/PositionAwarePixelShader.cs b/src/ImageSharp/Processing/Delegates/PositionAwarePixelShader.cs
new file mode 100644
index 000000000..08314600a
--- /dev/null
+++ b/src/ImageSharp/Processing/Delegates/PositionAwarePixelShader.cs
@@ -0,0 +1,14 @@
+using System;
+using System.Numerics;
+
+namespace SixLabors.ImageSharp.Processing.Delegates
+{
+    /// <summary>
+    /// A <see langword="delegate"/> representing a user defined pixel shader.
+    /// </summary>
+    /// <param name="span">The target row of <see cref="Vector4"/> pixels to process.</param>
+    /// <param name="offsetY">The initial vertical offset for the input pixels to process.</param>
+    /// <param name="offsetX">The initial horizontal offset for the input pixels to process.</param>
+    /// <remarks>The <see cref="Vector4.X"/>, <see cref="Vector4.Y"/>, <see cref="Vector4.Z"/>, and <see cref="Vector4.W"/> fields map the RGBA channels respectively.</remarks>
+    public delegate void PositionAwarePixelShader(Span<Vector4> span, int offsetY, int offsetX);
+}
diff --git a/src/ImageSharp/Processing/Processors/PixelShading/DelegatePixelShaderProcessor{TPixel}.cs b/src/ImageSharp/Processing/Processors/PixelShading/DelegatePixelShaderProcessor{TPixel}.cs
index c773e8b80..d15614cc4 100644
--- a/src/ImageSharp/Processing/Processors/PixelShading/DelegatePixelShaderProcessor{TPixel}.cs
+++ b/src/ImageSharp/Processing/Processors/PixelShading/DelegatePixelShaderProcessor{TPixel}.cs
@@ -13,7 +13,7 @@ using SixLabors.Primitives;
 namespace SixLabors.ImageSharp.Processing.Processors.PixelShading
 {
     /// <summary>
-    /// Performs simple binary threshold filtering against an image.
+    /// Applies a user defined pixel shader effect through a given delegate.
     /// </summary>
     /// <typeparam name="TPixel">The pixel format.</typeparam>
     internal class DelegatePixelShaderProcessor<TPixel> : ImageProcessor<TPixel>
diff --git a/src/ImageSharp/Processing/Processors/PixelShading/PositionAwarePixelShaderProcessor.cs b/src/ImageSharp/Processing/Processors/PixelShading/PositionAwarePixelShaderProcessor.cs
new file mode 100644
index 000000000..60125b2e2
--- /dev/null
+++ b/src/ImageSharp/Processing/Processors/PixelShading/PositionAwarePixelShaderProcessor.cs
@@ -0,0 +1,38 @@
+// Copyright (c) Six Labors and contributors.
+// Licensed under the Apache License, Version 2.0.
+
+using SixLabors.ImageSharp.PixelFormats;
+using SixLabors.ImageSharp.Processing.Delegates;
+using SixLabors.Primitives;
+
+namespace SixLabors.ImageSharp.Processing.Processors.PixelShading
+{
+    /// <summary>
+    /// Applies a user defined pixel shader effect through a given delegate.
+    /// </summary>
+    public sealed class PositionAwarePixelShaderProcessor : IImageProcessor
+    {
+        /// <summary>
+        /// Initializes a new instance of the <see cref="PositionAwarePixelShaderProcessor"/> class.
+        /// </summary>
+        /// <param name="pixelShader">
+        /// The user defined pixel shader to use to modify images.
+        /// </param>
+        public PositionAwarePixelShaderProcessor(PositionAwarePixelShader pixelShader)
+        {
+            this.PixelShader = pixelShader;
+        }
+
+        /// <summary>
+        /// Gets the user defined pixel shader.
+        /// </summary>
+        public PositionAwarePixelShader PixelShader { get; }
+
+        /// <inheritdoc />
+        public IImageProcessor<TPixel> CreatePixelSpecificProcessor<TPixel>(Image<TPixel> source, Rectangle sourceRectangle)
+            where TPixel : struct, IPixel<TPixel>
+        {
+            return new PositionAwarePixelShader<TPixel>(this, source, sourceRectangle);
+        }
+    }
+}
diff --git a/src/ImageSharp/Processing/Processors/PixelShading/PositionAwarePixelShaderProcessor{TPixel}.cs b/src/ImageSharp/Processing/Processors/PixelShading/PositionAwarePixelShaderProcessor{TPixel}.cs
new file mode 100644
index 000000000..55ad325ad
--- /dev/null
+++ b/src/ImageSharp/Processing/Processors/PixelShading/PositionAwarePixelShaderProcessor{TPixel}.cs
@@ -0,0 +1,65 @@
+// Copyright (c) Six Labors and contributors.
+// Licensed under the Apache License, Version 2.0.
+
+using System;
+using System.Numerics;
+
+using SixLabors.ImageSharp.Advanced;
+using SixLabors.ImageSharp.Advanced.ParallelUtils;
+using SixLabors.ImageSharp.PixelFormats;
+using SixLabors.ImageSharp.Processing.Delegates;
+using SixLabors.Primitives;
+
+namespace SixLabors.ImageSharp.Processing.Processors.PixelShading
+{
+    /// <summary>
+    /// Applies a user defined pixel shader effect through a given delegate.
+    /// </summary>
+    /// <typeparam name="TPixel">The pixel format.</typeparam>
+    internal class PositionAwarePixelShaderProcessor<TPixel> : ImageProcessor<TPixel>
+        where TPixel : struct, IPixel<TPixel>
+    {
+        /// <summary>
+        /// The user defined pixel shader.
+        /// </summary>
+        private readonly PositionAwarePixelShader pixelShader;
+
+        /// <summary>
+        /// Initializes a new instance of the <see cref="PositionAwarePixelShaderProcessor{TPixel}"/> class.
+        /// </summary>
+        /// <param name="definition">The <see cref="PositionAwarePixelShaderProcessor"/> defining the processor parameters.</param>
+        /// <param name="source">The source <see cref="Image{TPixel}"/> for the current processor instance.</param>
+        /// <param name="sourceRectangle">The source area to process for the current processor instance.</param>
+        public PositionAwarePixelShaderProcessor(PositionAwarePixelShaderProcessor definition, Image<TPixel> source, Rectangle sourceRectangle)
+            : base(source, sourceRectangle)
+        {
+            this.pixelShader = definition.PixelShader;
+        }
+
+        /// <inheritdoc/>
+        protected override void OnFrameApply(ImageFrame<TPixel> source)
+        {
+            var interest = Rectangle.Intersect(this.SourceRectangle, source.Bounds());
+            int startX = interest.X;
+            PositionAwarePixelShader pixelShader = this.pixelShader;
+
+            ParallelHelper.IterateRowsWithTempBuffer<Vector4>(
+                interest,
+                this.Configuration,
+                (rows, vectorBuffer) =>
+                {
+                    for (int y = rows.Min; y < rows.Max; y++)
+                    {
+                        Span<Vector4> vectorSpan = vectorBuffer.Span;
+                        int length = vectorSpan.Length;
+                        Span<TPixel> rowSpan = source.GetPixelRowSpan(y).Slice(startX, length);
+                        PixelOperations<TPixel>.Instance.ToVector4(this.Configuration, rowSpan, vectorSpan);
+
+                        pixelShader(vectorSpan, y, startX);
+
+                        PixelOperations<TPixel>.Instance.FromVector4Destructive(this.Configuration, vectorSpan, rowSpan);
+                    }
+                });
+        }
+    }
+}