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