mirror of https://github.com/SixLabors/ImageSharp
James Jackson-South
7 years ago
10 changed files with 279 additions and 194 deletions
@ -0,0 +1,131 @@ |
|||
// Copyright (c) Six Labors and contributors.
|
|||
// Licensed under the Apache License, Version 2.0.
|
|||
|
|||
using System; |
|||
using System.Numerics; |
|||
using SixLabors.ImageSharp.Memory; |
|||
using SixLabors.ImageSharp.PixelFormats; |
|||
using SixLabors.ImageSharp.Primitives; |
|||
|
|||
namespace SixLabors.ImageSharp |
|||
{ |
|||
/// <summary>
|
|||
/// Extension methods for <see cref="DenseMatrix{T}"/>.
|
|||
/// </summary>
|
|||
internal static class DenseMatrixUtils |
|||
{ |
|||
/// <summary>
|
|||
/// Computes the sum of vectors in <paramref name="targetRow"/> weighted by the kernel weight values.
|
|||
/// </summary>
|
|||
/// <typeparam name="TPixel">The pixel format.</typeparam>
|
|||
/// <param name="matrix">The dense matrix.</param>
|
|||
/// <param name="sourcePixels">The source frame.</param>
|
|||
/// <param name="targetRow">The target row.</param>
|
|||
/// <param name="row">The current row.</param>
|
|||
/// <param name="column">The current column.</param>
|
|||
/// <param name="maxRow">The maximum working area row.</param>
|
|||
/// <param name="maxColumn">The maximum working area column.</param>
|
|||
/// <param name="offsetColumn">The column offset to apply to source sampling.</param>
|
|||
public static void Convolve<TPixel>( |
|||
in DenseMatrix<float> matrix, |
|||
Buffer2D<TPixel> sourcePixels, |
|||
Span<Vector4> targetRow, |
|||
int row, |
|||
int column, |
|||
int maxRow, |
|||
int maxColumn, |
|||
int offsetColumn) |
|||
where TPixel : struct, IPixel<TPixel> |
|||
{ |
|||
Vector4 vector = default; |
|||
int matrixHeight = matrix.Rows; |
|||
int matrixWidth = matrix.Columns; |
|||
int radiusY = matrixHeight >> 1; |
|||
int radiusX = matrixWidth >> 1; |
|||
|
|||
for (int y = 0; y < matrixHeight; y++) |
|||
{ |
|||
int offsetY = (row + y - radiusY).Clamp(0, maxRow); |
|||
Span<TPixel> sourceRowSpan = sourcePixels.GetRowSpan(offsetY); |
|||
|
|||
for (int x = 0; x < matrixWidth; x++) |
|||
{ |
|||
int offsetX = (column + offsetColumn + x - radiusX).Clamp(offsetColumn, maxColumn); |
|||
var currentColor = sourceRowSpan[offsetX].ToVector4(); |
|||
Vector4Utils.Premultiply(ref currentColor); |
|||
currentColor *= matrix[y, x]; |
|||
vector += currentColor; |
|||
} |
|||
} |
|||
|
|||
ref Vector4 target = ref targetRow[column]; |
|||
vector.W = target.W; |
|||
Vector4Utils.UnPremultiply(ref vector); |
|||
target = vector; |
|||
} |
|||
|
|||
/// <summary>
|
|||
/// Computes the sum of vectors in <paramref name="targetRow"/> weighted by the two kernel weight values.
|
|||
/// </summary>
|
|||
/// <typeparam name="TPixel">The pixel format.</typeparam>
|
|||
/// <param name="matrixY">The vertical dense matrix.</param>
|
|||
/// <param name="matrixX">The horizontal dense matrix.</param>
|
|||
/// <param name="sourcePixels">The source frame.</param>
|
|||
/// <param name="targetRow">The target row.</param>
|
|||
/// <param name="row">The current row.</param>
|
|||
/// <param name="column">The current column.</param>
|
|||
/// <param name="maxRow">The maximum working area row.</param>
|
|||
/// <param name="maxColumn">The maximum working area column.</param>
|
|||
/// <param name="offsetColumn">The column offset to apply to source sampling.</param>
|
|||
public static void Convolve2D<TPixel>( |
|||
in DenseMatrix<float> matrixY, |
|||
in DenseMatrix<float> matrixX, |
|||
Buffer2D<TPixel> sourcePixels, |
|||
Span<Vector4> targetRow, |
|||
int row, |
|||
int column, |
|||
int maxRow, |
|||
int maxColumn, |
|||
int offsetColumn) |
|||
where TPixel : struct, IPixel<TPixel> |
|||
{ |
|||
Vector4 vectorY = default; |
|||
Vector4 vectorX = default; |
|||
int matrixYHeight = matrixY.Rows; |
|||
int matrixYWidth = matrixY.Columns; |
|||
int matrixXHeight = matrixX.Rows; |
|||
int matrixXWidth = matrixX.Columns; |
|||
int radiusY = matrixYHeight >> 1; |
|||
int radiusX = matrixXWidth >> 1; |
|||
|
|||
for (int y = 0; y < matrixYHeight; y++) |
|||
{ |
|||
int offsetY = (row + y - radiusY).Clamp(0, maxRow); |
|||
Span<TPixel> sourceRowSpan = sourcePixels.GetRowSpan(offsetY); |
|||
|
|||
for (int x = 0; x < matrixXWidth; x++) |
|||
{ |
|||
int offsetX = (column + offsetColumn + x - radiusX).Clamp(offsetColumn, maxColumn); |
|||
var currentColor = sourceRowSpan[offsetX].ToVector4(); |
|||
Vector4Utils.Premultiply(ref currentColor); |
|||
|
|||
if (y < matrixXHeight) |
|||
{ |
|||
vectorX += matrixX[y, x] * currentColor; |
|||
} |
|||
|
|||
if (x < matrixYWidth) |
|||
{ |
|||
vectorY += matrixY[y, x] * currentColor; |
|||
} |
|||
} |
|||
} |
|||
|
|||
var vector = Vector4.SquareRoot((vectorX * vectorX) + (vectorY * vectorY)); |
|||
ref Vector4 target = ref targetRow[column]; |
|||
vector.W = target.W; |
|||
Vector4Utils.UnPremultiply(ref vector); |
|||
target = vector; |
|||
} |
|||
} |
|||
} |
|||
@ -0,0 +1,59 @@ |
|||
using System.Numerics; |
|||
using System.Runtime.CompilerServices; |
|||
using BenchmarkDotNet.Attributes; |
|||
|
|||
namespace SixLabors.ImageSharp.Benchmarks.General.Vectorization |
|||
{ |
|||
public class Premultiply |
|||
{ |
|||
[Benchmark(Baseline = true)] |
|||
public Vector4 PremultiplyByVal() |
|||
{ |
|||
var input = new Vector4(.5F); |
|||
return Vector4Utils.Premultiply(input); |
|||
} |
|||
|
|||
[Benchmark] |
|||
public Vector4 PremultiplyByRef() |
|||
{ |
|||
var input = new Vector4(.5F); |
|||
Vector4Utils.PremultiplyRef(ref input); |
|||
return input; |
|||
} |
|||
|
|||
[Benchmark] |
|||
public Vector4 PremultiplyRefWithPropertyAssign() |
|||
{ |
|||
var input = new Vector4(.5F); |
|||
Vector4Utils.PremultiplyRefWithPropertyAssign(ref input); |
|||
return input; |
|||
} |
|||
} |
|||
|
|||
internal static class Vector4Utils |
|||
{ |
|||
[MethodImpl(InliningOptions.ShortMethod)] |
|||
public static Vector4 Premultiply(Vector4 source) |
|||
{ |
|||
float w = source.W; |
|||
Vector4 premultiplied = source * w; |
|||
premultiplied.W = w; |
|||
return premultiplied; |
|||
} |
|||
|
|||
[MethodImpl(InliningOptions.ShortMethod)] |
|||
public static void PremultiplyRef(ref Vector4 source) |
|||
{ |
|||
float w = source.W; |
|||
source *= w; |
|||
source.W = w; |
|||
} |
|||
|
|||
[MethodImpl(InliningOptions.ShortMethod)] |
|||
public static void PremultiplyRefWithPropertyAssign(ref Vector4 source) |
|||
{ |
|||
float w = source.W; |
|||
source *= new Vector4(w) { W = 1 }; |
|||
} |
|||
} |
|||
} |
|||
Loading…
Reference in new issue