mirror of https://github.com/SixLabors/ImageSharp
James Jackson-South
6 years ago
21 changed files with 285 additions and 277 deletions
@ -0,0 +1,115 @@ |
|||||
|
// Copyright (c) Six Labors and contributors.
|
||||
|
// Licensed under the Apache License, Version 2.0.
|
||||
|
|
||||
|
using System; |
||||
|
using System.Buffers; |
||||
|
using System.Runtime.CompilerServices; |
||||
|
using SixLabors.ImageSharp.Memory; |
||||
|
using SixLabors.ImageSharp.PixelFormats; |
||||
|
|
||||
|
namespace SixLabors.ImageSharp.Processing.Processors.Quantization |
||||
|
{ |
||||
|
/// <summary>
|
||||
|
/// A pixel-specific image frame where each pixel buffer value represents an index in a color palette.
|
||||
|
/// </summary>
|
||||
|
/// <typeparam name="TPixel">The pixel format.</typeparam>
|
||||
|
public sealed class IndexedImageFrame<TPixel> : IDisposable |
||||
|
where TPixel : unmanaged, IPixel<TPixel> |
||||
|
{ |
||||
|
private IMemoryOwner<byte> pixelsOwner; |
||||
|
private IMemoryOwner<TPixel> paletteOwner; |
||||
|
private bool isDisposed; |
||||
|
|
||||
|
/// <summary>
|
||||
|
/// Initializes a new instance of the <see cref="IndexedImageFrame{TPixel}"/> class.
|
||||
|
/// </summary>
|
||||
|
/// <param name="configuration">
|
||||
|
/// The configuration which allows altering default behaviour or extending the library.
|
||||
|
/// </param>
|
||||
|
/// <param name="width">The frame width.</param>
|
||||
|
/// <param name="height">The frame height.</param>
|
||||
|
/// <param name="palette">The color palette.</param>
|
||||
|
internal IndexedImageFrame(Configuration configuration, int width, int height, ReadOnlyMemory<TPixel> palette) |
||||
|
{ |
||||
|
Guard.NotNull(configuration, nameof(configuration)); |
||||
|
Guard.MustBeGreaterThan(width, 0, nameof(width)); |
||||
|
Guard.MustBeGreaterThan(height, 0, nameof(height)); |
||||
|
|
||||
|
this.Configuration = configuration; |
||||
|
this.Width = width; |
||||
|
this.Height = height; |
||||
|
this.pixelsOwner = configuration.MemoryAllocator.AllocateManagedByteBuffer(width * height); |
||||
|
|
||||
|
// Copy the palette over. We want the lifetime of this frame to be independant of any palette source.
|
||||
|
this.paletteOwner = configuration.MemoryAllocator.Allocate<TPixel>(palette.Length); |
||||
|
palette.Span.CopyTo(this.paletteOwner.GetSpan()); |
||||
|
this.Palette = this.paletteOwner.Memory.Slice(0, palette.Length); |
||||
|
} |
||||
|
|
||||
|
/// <summary>
|
||||
|
/// Gets the configuration which allows altering default behaviour or extending the library.
|
||||
|
/// </summary>
|
||||
|
public Configuration Configuration { get; } |
||||
|
|
||||
|
/// <summary>
|
||||
|
/// Gets the width of this <see cref="IndexedImageFrame{TPixel}"/>.
|
||||
|
/// </summary>
|
||||
|
public int Width { get; } |
||||
|
|
||||
|
/// <summary>
|
||||
|
/// Gets the height of this <see cref="IndexedImageFrame{TPixel}"/>.
|
||||
|
/// </summary>
|
||||
|
public int Height { get; } |
||||
|
|
||||
|
/// <summary>
|
||||
|
/// Gets the color palette of this <see cref="IndexedImageFrame{TPixel}"/>.
|
||||
|
/// </summary>
|
||||
|
public ReadOnlyMemory<TPixel> Palette { get; } |
||||
|
|
||||
|
/// <summary>
|
||||
|
/// Gets the pixels of this <see cref="IndexedImageFrame{TPixel}"/>.
|
||||
|
/// </summary>
|
||||
|
/// <returns>The <see cref="ReadOnlySpan{T}"/></returns>
|
||||
|
[MethodImpl(InliningOptions.ShortMethod)] |
||||
|
public ReadOnlySpan<byte> GetPixelBufferSpan() => this.pixelsOwner.GetSpan(); // TODO: Buffer2D<byte>
|
||||
|
|
||||
|
/// <summary>
|
||||
|
/// Gets the representation of the pixels as a <see cref="ReadOnlySpan{T}"/> of contiguous memory
|
||||
|
/// at row <paramref name="rowIndex"/> beginning from the the first pixel on that row.
|
||||
|
/// </summary>
|
||||
|
/// <param name="rowIndex">The row index in the pixel buffer.</param>
|
||||
|
/// <returns>The pixel row as a <see cref="ReadOnlySpan{T}"/>.</returns>
|
||||
|
[MethodImpl(InliningOptions.ShortMethod)] |
||||
|
public ReadOnlySpan<byte> GetPixelRowSpan(int rowIndex) |
||||
|
=> this.GetWritablePixelRowSpanUnsafe(rowIndex); |
||||
|
|
||||
|
/// <summary>
|
||||
|
/// <para>
|
||||
|
/// Gets the representation of the pixels as a <see cref="Span{T}"/> of contiguous memory
|
||||
|
/// at row <paramref name="rowIndex"/> beginning from the the first pixel on that row.
|
||||
|
/// </para>
|
||||
|
/// <para>
|
||||
|
/// Note: Values written to this span are not sanitized against the palette length.
|
||||
|
/// Care should be taken during assignment to prevent out-of-bounds errors.
|
||||
|
/// </para>
|
||||
|
/// </summary>
|
||||
|
/// <param name="rowIndex">The row index in the pixel buffer.</param>
|
||||
|
/// <returns>The pixel row as a <see cref="Span{T}"/>.</returns>
|
||||
|
[MethodImpl(InliningOptions.ShortMethod)] |
||||
|
public Span<byte> GetWritablePixelRowSpanUnsafe(int rowIndex) |
||||
|
=> this.pixelsOwner.GetSpan().Slice(rowIndex * this.Width, this.Width); |
||||
|
|
||||
|
/// <inheritdoc/>
|
||||
|
public void Dispose() |
||||
|
{ |
||||
|
if (!this.isDisposed) |
||||
|
{ |
||||
|
this.isDisposed = true; |
||||
|
this.pixelsOwner.Dispose(); |
||||
|
this.paletteOwner.Dispose(); |
||||
|
this.pixelsOwner = null; |
||||
|
this.paletteOwner = null; |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} |
||||
@ -1,94 +0,0 @@ |
|||||
// Copyright (c) Six Labors and contributors.
|
|
||||
// Licensed under the Apache License, Version 2.0.
|
|
||||
|
|
||||
using System; |
|
||||
using System.Buffers; |
|
||||
using System.Runtime.CompilerServices; |
|
||||
using SixLabors.ImageSharp.Memory; |
|
||||
using SixLabors.ImageSharp.PixelFormats; |
|
||||
|
|
||||
namespace SixLabors.ImageSharp.Processing.Processors.Quantization |
|
||||
{ |
|
||||
/// <summary>
|
|
||||
/// Represents a quantized image frame where the pixels indexed by a color palette.
|
|
||||
/// </summary>
|
|
||||
/// <typeparam name="TPixel">The pixel format.</typeparam>
|
|
||||
public sealed class QuantizedFrame<TPixel> : IDisposable |
|
||||
where TPixel : unmanaged, IPixel<TPixel> |
|
||||
{ |
|
||||
private IMemoryOwner<TPixel> palette; |
|
||||
private IMemoryOwner<byte> pixels; |
|
||||
private bool isDisposed; |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// Initializes a new instance of the <see cref="QuantizedFrame{TPixel}"/> class.
|
|
||||
/// </summary>
|
|
||||
/// <param name="memoryAllocator">Used to allocated memory for image processing operations.</param>
|
|
||||
/// <param name="width">The image width.</param>
|
|
||||
/// <param name="height">The image height.</param>
|
|
||||
/// <param name="palette">The color palette.</param>
|
|
||||
internal QuantizedFrame(MemoryAllocator memoryAllocator, int width, int height, ReadOnlySpan<TPixel> palette) |
|
||||
{ |
|
||||
Guard.MustBeGreaterThan(width, 0, nameof(width)); |
|
||||
Guard.MustBeGreaterThan(height, 0, nameof(height)); |
|
||||
|
|
||||
this.Width = width; |
|
||||
this.Height = height; |
|
||||
this.pixels = memoryAllocator.AllocateManagedByteBuffer(width * height, AllocationOptions.Clean); |
|
||||
|
|
||||
this.palette = memoryAllocator.Allocate<TPixel>(palette.Length); |
|
||||
palette.CopyTo(this.palette.GetSpan()); |
|
||||
} |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// Gets the width of this <see cref="QuantizedFrame{TPixel}"/>.
|
|
||||
/// </summary>
|
|
||||
public int Width { get; } |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// Gets the height of this <see cref="QuantizedFrame{TPixel}"/>.
|
|
||||
/// </summary>
|
|
||||
public int Height { get; } |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// Gets the color palette of this <see cref="QuantizedFrame{TPixel}"/>.
|
|
||||
/// </summary>
|
|
||||
public ReadOnlyMemory<TPixel> Palette |
|
||||
{ |
|
||||
[MethodImpl(InliningOptions.ShortMethod)] |
|
||||
get { return this.palette.Memory; } |
|
||||
} |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// Gets the pixels of this <see cref="QuantizedFrame{TPixel}"/>.
|
|
||||
/// </summary>
|
|
||||
/// <returns>The <see cref="Span{T}"/></returns>
|
|
||||
[MethodImpl(InliningOptions.ShortMethod)] |
|
||||
public Span<byte> GetPixelSpan() => this.pixels.GetSpan(); |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// Gets the representation of the pixels as a <see cref="Span{T}"/> of contiguous memory
|
|
||||
/// at row <paramref name="rowIndex"/> beginning from the the first pixel on that row.
|
|
||||
/// </summary>
|
|
||||
/// <param name="rowIndex">The row.</param>
|
|
||||
/// <returns>The pixel row as a <see cref="Span{T}"/>.</returns>
|
|
||||
[MethodImpl(InliningOptions.ShortMethod)] |
|
||||
public Span<byte> GetPixelRowSpan(int rowIndex) |
|
||||
=> this.GetPixelSpan().Slice(rowIndex * this.Width, this.Width); |
|
||||
|
|
||||
/// <inheritdoc/>
|
|
||||
public void Dispose() |
|
||||
{ |
|
||||
if (!this.isDisposed) |
|
||||
{ |
|
||||
return; |
|
||||
} |
|
||||
|
|
||||
this.isDisposed = true; |
|
||||
this.pixels?.Dispose(); |
|
||||
this.palette?.Dispose(); |
|
||||
this.pixels = null; |
|
||||
this.palette = null; |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
Loading…
Reference in new issue