mirror of https://github.com/SixLabors/ImageSharp
James Jackson-South
5 years ago
committed by
GitHub
GitHub
18 changed files with 943 additions and 411 deletions
@ -0,0 +1,60 @@ |
|||
// Copyright (c) Six Labors.
|
|||
// Licensed under the Apache License, Version 2.0.
|
|||
|
|||
using System; |
|||
using System.Buffers; |
|||
|
|||
namespace SixLabors.ImageSharp.Memory |
|||
{ |
|||
/// <summary>
|
|||
/// A custom <see cref="MemoryManager{T}"/> that can wrap a rawpointer to a buffer of a specified type.
|
|||
/// </summary>
|
|||
/// <typeparam name="T">The value type to use when casting the wrapped <see cref="Memory{T}"/> instance.</typeparam>
|
|||
/// <remarks>This manager doesn't own the memory buffer that it points to.</remarks>
|
|||
internal sealed unsafe class UnmanagedMemoryManager<T> : MemoryManager<T> |
|||
where T : unmanaged |
|||
{ |
|||
/// <summary>
|
|||
/// The pointer to the memory buffer.
|
|||
/// </summary>
|
|||
private readonly void* pointer; |
|||
|
|||
/// <summary>
|
|||
/// The length of the memory area.
|
|||
/// </summary>
|
|||
private readonly int length; |
|||
|
|||
/// <summary>
|
|||
/// Initializes a new instance of the <see cref="UnmanagedMemoryManager{T}"/> class.
|
|||
/// </summary>
|
|||
/// <param name="pointer">The pointer to the memory buffer.</param>
|
|||
/// <param name="length">The length of the memory area.</param>
|
|||
public UnmanagedMemoryManager(void* pointer, int length) |
|||
{ |
|||
this.pointer = pointer; |
|||
this.length = length; |
|||
} |
|||
|
|||
/// <inheritdoc/>
|
|||
protected override void Dispose(bool disposing) |
|||
{ |
|||
} |
|||
|
|||
/// <inheritdoc/>
|
|||
public override Span<T> GetSpan() |
|||
{ |
|||
return new Span<T>(this.pointer, this.length); |
|||
} |
|||
|
|||
/// <inheritdoc/>
|
|||
public override MemoryHandle Pin(int elementIndex = 0) |
|||
{ |
|||
return new MemoryHandle(((T*)this.pointer) + elementIndex); |
|||
} |
|||
|
|||
/// <inheritdoc/>
|
|||
public override void Unpin() |
|||
{ |
|||
} |
|||
} |
|||
} |
|||
@ -1,163 +0,0 @@ |
|||
// Copyright (c) Six Labors.
|
|||
// Licensed under the Apache License, Version 2.0.
|
|||
|
|||
using System; |
|||
using System.Numerics; |
|||
using System.Runtime.CompilerServices; |
|||
using System.Runtime.InteropServices; |
|||
using SixLabors.ImageSharp.Advanced; |
|||
using SixLabors.ImageSharp.Memory; |
|||
using SixLabors.ImageSharp.PixelFormats; |
|||
|
|||
namespace SixLabors.ImageSharp.Processing.Processors.Convolution |
|||
{ |
|||
/// <summary>
|
|||
/// A <see langword="struct"/> implementing the logic for 1D convolution.
|
|||
/// </summary>
|
|||
internal readonly struct ConvolutionRowOperation<TPixel> : IRowOperation<Vector4> |
|||
where TPixel : unmanaged, IPixel<TPixel> |
|||
{ |
|||
private readonly Rectangle bounds; |
|||
private readonly Buffer2D<TPixel> targetPixels; |
|||
private readonly Buffer2D<TPixel> sourcePixels; |
|||
private readonly KernelSamplingMap map; |
|||
private readonly DenseMatrix<float> kernelMatrix; |
|||
private readonly Configuration configuration; |
|||
private readonly bool preserveAlpha; |
|||
|
|||
[MethodImpl(MethodImplOptions.AggressiveInlining)] |
|||
public ConvolutionRowOperation( |
|||
Rectangle bounds, |
|||
Buffer2D<TPixel> targetPixels, |
|||
Buffer2D<TPixel> sourcePixels, |
|||
KernelSamplingMap map, |
|||
DenseMatrix<float> kernelMatrix, |
|||
Configuration configuration, |
|||
bool preserveAlpha) |
|||
{ |
|||
this.bounds = bounds; |
|||
this.targetPixels = targetPixels; |
|||
this.sourcePixels = sourcePixels; |
|||
this.map = map; |
|||
this.kernelMatrix = kernelMatrix; |
|||
this.configuration = configuration; |
|||
this.preserveAlpha = preserveAlpha; |
|||
} |
|||
|
|||
/// <inheritdoc/>
|
|||
[MethodImpl(MethodImplOptions.AggressiveInlining)] |
|||
public void Invoke(int y, Span<Vector4> span) |
|||
{ |
|||
if (this.preserveAlpha) |
|||
{ |
|||
this.Convolve3(y, span); |
|||
} |
|||
else |
|||
{ |
|||
this.Convolve4(y, span); |
|||
} |
|||
} |
|||
|
|||
[MethodImpl(MethodImplOptions.AggressiveInlining)] |
|||
private void Convolve3(int y, Span<Vector4> span) |
|||
{ |
|||
// Span is 2x bounds.
|
|||
int boundsX = this.bounds.X; |
|||
int boundsWidth = this.bounds.Width; |
|||
Span<Vector4> sourceBuffer = span.Slice(0, this.bounds.Width); |
|||
Span<Vector4> targetBuffer = span.Slice(this.bounds.Width); |
|||
|
|||
var state = new ConvolutionState(in this.kernelMatrix, this.map); |
|||
ref int sampleRowBase = ref state.GetSampleRow(y - this.bounds.Y); |
|||
|
|||
// Clear the target buffer for each row run.
|
|||
targetBuffer.Clear(); |
|||
ref Vector4 targetBase = ref MemoryMarshal.GetReference(targetBuffer); |
|||
|
|||
ReadOnlyKernel kernel = state.Kernel; |
|||
Span<TPixel> sourceRow; |
|||
for (int kY = 0; kY < kernel.Rows; kY++) |
|||
{ |
|||
// Get the precalculated source sample row for this kernel row and copy to our buffer.
|
|||
int sampleY = Unsafe.Add(ref sampleRowBase, kY); |
|||
sourceRow = this.sourcePixels.GetRowSpan(sampleY).Slice(boundsX, boundsWidth); |
|||
PixelOperations<TPixel>.Instance.ToVector4(this.configuration, sourceRow, sourceBuffer); |
|||
|
|||
ref Vector4 sourceBase = ref MemoryMarshal.GetReference(sourceBuffer); |
|||
|
|||
for (int x = 0; x < sourceBuffer.Length; x++) |
|||
{ |
|||
ref int sampleColumnBase = ref state.GetSampleColumn(x); |
|||
ref Vector4 target = ref Unsafe.Add(ref targetBase, x); |
|||
|
|||
for (int kX = 0; kX < kernel.Columns; kX++) |
|||
{ |
|||
int sampleX = Unsafe.Add(ref sampleColumnBase, kX) - boundsX; |
|||
Vector4 sample = Unsafe.Add(ref sourceBase, sampleX); |
|||
target += kernel[kY, kX] * sample; |
|||
} |
|||
} |
|||
} |
|||
|
|||
// Now we need to copy the original alpha values from the source row.
|
|||
sourceRow = this.sourcePixels.GetRowSpan(y).Slice(boundsX, boundsWidth); |
|||
PixelOperations<TPixel>.Instance.ToVector4(this.configuration, sourceRow, sourceBuffer); |
|||
|
|||
for (int x = 0; x < sourceRow.Length; x++) |
|||
{ |
|||
ref Vector4 target = ref Unsafe.Add(ref targetBase, x); |
|||
target.W = Unsafe.Add(ref MemoryMarshal.GetReference(sourceBuffer), x).W; |
|||
} |
|||
|
|||
Span<TPixel> targetRow = this.targetPixels.GetRowSpan(y).Slice(boundsX, boundsWidth); |
|||
PixelOperations<TPixel>.Instance.FromVector4Destructive(this.configuration, targetBuffer, targetRow); |
|||
} |
|||
|
|||
[MethodImpl(MethodImplOptions.AggressiveInlining)] |
|||
private void Convolve4(int y, Span<Vector4> span) |
|||
{ |
|||
// Span is 2x bounds.
|
|||
int boundsX = this.bounds.X; |
|||
int boundsWidth = this.bounds.Width; |
|||
Span<Vector4> sourceBuffer = span.Slice(0, this.bounds.Width); |
|||
Span<Vector4> targetBuffer = span.Slice(this.bounds.Width); |
|||
|
|||
var state = new ConvolutionState(in this.kernelMatrix, this.map); |
|||
ref int sampleRowBase = ref state.GetSampleRow(y - this.bounds.Y); |
|||
|
|||
// Clear the target buffer for each row run.
|
|||
targetBuffer.Clear(); |
|||
ref Vector4 targetBase = ref MemoryMarshal.GetReference(targetBuffer); |
|||
|
|||
ReadOnlyKernel kernel = state.Kernel; |
|||
for (int kY = 0; kY < kernel.Rows; kY++) |
|||
{ |
|||
// Get the precalculated source sample row for this kernel row and copy to our buffer.
|
|||
int sampleY = Unsafe.Add(ref sampleRowBase, kY); |
|||
Span<TPixel> sourceRow = this.sourcePixels.GetRowSpan(sampleY).Slice(boundsX, boundsWidth); |
|||
PixelOperations<TPixel>.Instance.ToVector4(this.configuration, sourceRow, sourceBuffer); |
|||
|
|||
Numerics.Premultiply(sourceBuffer); |
|||
ref Vector4 sourceBase = ref MemoryMarshal.GetReference(sourceBuffer); |
|||
|
|||
for (int x = 0; x < sourceBuffer.Length; x++) |
|||
{ |
|||
ref int sampleColumnBase = ref state.GetSampleColumn(x); |
|||
ref Vector4 target = ref Unsafe.Add(ref targetBase, x); |
|||
|
|||
for (int kX = 0; kX < kernel.Columns; kX++) |
|||
{ |
|||
int sampleX = Unsafe.Add(ref sampleColumnBase, kX) - boundsX; |
|||
Vector4 sample = Unsafe.Add(ref sourceBase, sampleX); |
|||
target += kernel[kY, kX] * sample; |
|||
} |
|||
} |
|||
} |
|||
|
|||
Numerics.UnPremultiply(targetBuffer); |
|||
|
|||
Span<TPixel> targetRow = this.targetPixels.GetRowSpan(y).Slice(boundsX, boundsWidth); |
|||
PixelOperations<TPixel>.Instance.FromVector4Destructive(this.configuration, targetBuffer, targetRow); |
|||
} |
|||
} |
|||
} |
|||
Loading…
Reference in new issue