mirror of https://github.com/SixLabors/ImageSharp
James Jackson-South
1 month ago
7 changed files with 473 additions and 81 deletions
File diff suppressed because one or more lines are too long
@ -0,0 +1,143 @@ |
|||
// Copyright (c) Six Labors.
|
|||
// Licensed under the Six Labors Split License.
|
|||
|
|||
namespace SixLabors.ImageSharp.Formats.Heif.Av1.Pipeline; |
|||
|
|||
/// <summary>
|
|||
/// Provides disjoint reusable buffers for intra-block-copy mode decisions.
|
|||
/// </summary>
|
|||
/// <typeparam name="TSample">The native sample type selected by the encoder pipeline.</typeparam>
|
|||
internal readonly ref struct Av1EncoderIntraBlockCopyWorkspace<TSample> |
|||
where TSample : unmanaged |
|||
{ |
|||
/// <summary>
|
|||
/// The number of samples in the fixed 8x8 intra-block-copy transform.
|
|||
/// </summary>
|
|||
public const int MaximumSampleCount = 8 * 8; |
|||
|
|||
/// <summary>
|
|||
/// The number of sample buffers retained by one mode decision.
|
|||
/// </summary>
|
|||
public const int SampleBufferCount = 10; |
|||
|
|||
/// <summary>
|
|||
/// The number of coefficient buffers retained by one mode decision.
|
|||
/// </summary>
|
|||
public const int CoefficientBufferCount = 7; |
|||
|
|||
private readonly Span<TSample> samples; |
|||
private readonly Span<short> residual; |
|||
private readonly Span<int> coefficients; |
|||
|
|||
/// <summary>
|
|||
/// Initializes a new instance of the <see cref="Av1EncoderIntraBlockCopyWorkspace{TSample}"/> struct.
|
|||
/// </summary>
|
|||
/// <param name="samples">The sample storage.</param>
|
|||
/// <param name="residual">The residual storage shared by sequential plane evaluations.</param>
|
|||
/// <param name="coefficients">The coefficient storage.</param>
|
|||
public Av1EncoderIntraBlockCopyWorkspace( |
|||
Span<TSample> samples, |
|||
Span<short> residual, |
|||
Span<int> coefficients) |
|||
{ |
|||
this.samples = samples; |
|||
this.residual = residual; |
|||
this.coefficients = coefficients; |
|||
} |
|||
|
|||
/// <summary>
|
|||
/// Gets the selected luma reconstruction.
|
|||
/// </summary>
|
|||
public Span<TSample> SelectedLumaReconstruction => this.GetSamples(0); |
|||
|
|||
/// <summary>
|
|||
/// Gets the selected blue-difference chroma reconstruction.
|
|||
/// </summary>
|
|||
public Span<TSample> SelectedBlueReconstruction => this.GetSamples(1); |
|||
|
|||
/// <summary>
|
|||
/// Gets the selected red-difference chroma reconstruction.
|
|||
/// </summary>
|
|||
public Span<TSample> SelectedRedReconstruction => this.GetSamples(2); |
|||
|
|||
/// <summary>
|
|||
/// Gets the current luma prediction.
|
|||
/// </summary>
|
|||
public Span<TSample> LumaPrediction => this.GetSamples(3); |
|||
|
|||
/// <summary>
|
|||
/// Gets the current blue-difference chroma prediction.
|
|||
/// </summary>
|
|||
public Span<TSample> BluePrediction => this.GetSamples(4); |
|||
|
|||
/// <summary>
|
|||
/// Gets the current red-difference chroma prediction.
|
|||
/// </summary>
|
|||
public Span<TSample> RedPrediction => this.GetSamples(5); |
|||
|
|||
/// <summary>
|
|||
/// Gets the current luma candidate reconstruction.
|
|||
/// </summary>
|
|||
public Span<TSample> LumaCandidateReconstruction => this.GetSamples(6); |
|||
|
|||
/// <summary>
|
|||
/// Gets the current blue-difference chroma candidate reconstruction.
|
|||
/// </summary>
|
|||
public Span<TSample> BlueCandidateReconstruction => this.GetSamples(7); |
|||
|
|||
/// <summary>
|
|||
/// Gets the current red-difference chroma candidate reconstruction.
|
|||
/// </summary>
|
|||
public Span<TSample> RedCandidateReconstruction => this.GetSamples(8); |
|||
|
|||
/// <summary>
|
|||
/// Gets the reconstruction scratch overwritten by each transform trial.
|
|||
/// </summary>
|
|||
public Span<TSample> TransformReconstruction => this.GetSamples(9); |
|||
|
|||
/// <summary>
|
|||
/// Gets the residual scratch shared by sequential plane evaluations.
|
|||
/// </summary>
|
|||
public Span<short> Residual => this.residual; |
|||
|
|||
/// <summary>
|
|||
/// Gets the selected luma coefficients.
|
|||
/// </summary>
|
|||
public Span<int> SelectedLumaCoefficients => this.GetCoefficients(0); |
|||
|
|||
/// <summary>
|
|||
/// Gets the selected blue-difference chroma coefficients.
|
|||
/// </summary>
|
|||
public Span<int> SelectedBlueCoefficients => this.GetCoefficients(1); |
|||
|
|||
/// <summary>
|
|||
/// Gets the selected red-difference chroma coefficients.
|
|||
/// </summary>
|
|||
public Span<int> SelectedRedCoefficients => this.GetCoefficients(2); |
|||
|
|||
/// <summary>
|
|||
/// Gets the current luma candidate coefficients.
|
|||
/// </summary>
|
|||
public Span<int> LumaCandidateCoefficients => this.GetCoefficients(3); |
|||
|
|||
/// <summary>
|
|||
/// Gets the current blue-difference chroma candidate coefficients.
|
|||
/// </summary>
|
|||
public Span<int> BlueCandidateCoefficients => this.GetCoefficients(4); |
|||
|
|||
/// <summary>
|
|||
/// Gets the current red-difference chroma candidate coefficients.
|
|||
/// </summary>
|
|||
public Span<int> RedCandidateCoefficients => this.GetCoefficients(5); |
|||
|
|||
/// <summary>
|
|||
/// Gets the coefficient scratch overwritten by each transform trial.
|
|||
/// </summary>
|
|||
public Span<int> TransformCoefficients => this.GetCoefficients(6); |
|||
|
|||
private Span<TSample> GetSamples(int index) |
|||
=> this.samples.Slice(index * MaximumSampleCount, MaximumSampleCount); |
|||
|
|||
private Span<int> GetCoefficients(int index) |
|||
=> this.coefficients.Slice(index * MaximumSampleCount, MaximumSampleCount); |
|||
} |
|||
Loading…
Reference in new issue