mirror of https://github.com/SixLabors/ImageSharp
James Jackson-South
1 month ago
18 changed files with 1235 additions and 1861 deletions
@ -1,163 +0,0 @@ |
|||
// Copyright (c) Six Labors.
|
|||
// Licensed under the Six Labors Split License.
|
|||
|
|||
using SixLabors.ImageSharp.Formats.Heif.Av1.Transform; |
|||
|
|||
namespace SixLabors.ImageSharp.Formats.Heif.Av1.ModeDecision; |
|||
|
|||
/// <summary>
|
|||
/// Describes the spatial, chroma, and transform layout of one AV1 block considered by mode-decision scanning.
|
|||
/// </summary>
|
|||
internal class Av1BlockGeometry |
|||
{ |
|||
/// <summary>
|
|||
/// The luma block size from which the cached luma dimensions are derived.
|
|||
/// </summary>
|
|||
private Av1BlockSize blockSize; |
|||
|
|||
/// <summary>
|
|||
/// The chroma block size from which the cached chroma dimensions are derived.
|
|||
/// </summary>
|
|||
private Av1BlockSize blockSizeUv; |
|||
|
|||
/// <summary>
|
|||
/// Initializes a new instance of the <see cref="Av1BlockGeometry"/> class with storage for every supported transform depth.
|
|||
/// </summary>
|
|||
public Av1BlockGeometry() |
|||
{ |
|||
this.RedunancyList = []; |
|||
this.TransformOrigin = new Point[Av1Constants.MaxVarTransform + 1][]; |
|||
for (int i = 0; i < this.TransformOrigin.Length; i++) |
|||
{ |
|||
this.TransformOrigin[i] = new Point[Av1Constants.MaxTransformBlockCount]; |
|||
} |
|||
} |
|||
|
|||
/// <summary>
|
|||
/// Gets or sets the luma block size and updates <see cref="BlockWidth"/> and <see cref="BlockHeight"/> to match.
|
|||
/// </summary>
|
|||
public Av1BlockSize BlockSize |
|||
{ |
|||
get => this.blockSize; |
|||
set |
|||
{ |
|||
this.blockSize = value; |
|||
this.BlockWidth = value.GetWidth(); |
|||
this.BlockHeight = value.GetHeight(); |
|||
} |
|||
} |
|||
|
|||
/// <summary>
|
|||
/// Gets or sets the chroma block size and updates <see cref="BlockWidthUv"/> and <see cref="BlockHeightUv"/> to match.
|
|||
/// </summary>
|
|||
public Av1BlockSize BlockSizeUv |
|||
{ |
|||
get => this.blockSizeUv; |
|||
set |
|||
{ |
|||
this.blockSizeUv = value; |
|||
this.BlockWidthUv = value.GetWidth(); |
|||
this.BlockHeightUv = value.GetHeight(); |
|||
} |
|||
} |
|||
|
|||
/// <summary>
|
|||
/// Gets or sets the block origin in pixels relative to the top-left corner of its superblock.
|
|||
/// </summary>
|
|||
public Point Origin { get; set; } |
|||
|
|||
/// <summary>
|
|||
/// Gets or sets a value indicating whether this luma block owns chroma samples in the mode-decision layout.
|
|||
/// </summary>
|
|||
public bool HasUv { get; set; } |
|||
|
|||
/// <summary>
|
|||
/// Gets the luma block width in pixels.
|
|||
/// </summary>
|
|||
public int BlockWidth { get; private set; } |
|||
|
|||
/// <summary>
|
|||
/// Gets the luma block height in pixels.
|
|||
/// </summary>
|
|||
public int BlockHeight { get; private set; } |
|||
|
|||
/// <summary>
|
|||
/// Gets the number of luma transform blocks at each transform depth.
|
|||
/// </summary>
|
|||
public int[] TransformBlockCount { get; } = new int[Av1Constants.MaxVarTransform + 1]; |
|||
|
|||
/// <summary>
|
|||
/// Gets the luma transform size selected at each transform depth.
|
|||
/// </summary>
|
|||
public Av1TransformSize[] TransformSize { get; } = new Av1TransformSize[Av1Constants.MaxVarTransform + 1]; |
|||
|
|||
/// <summary>
|
|||
/// Gets the chroma transform size selected at each transform depth.
|
|||
/// </summary>
|
|||
public Av1TransformSize[] TransformSizeUv { get; } = new Av1TransformSize[Av1Constants.MaxVarTransform + 1]; |
|||
|
|||
/// <summary>
|
|||
/// Gets the pixel origins of the transform blocks at each transform depth.
|
|||
/// </summary>
|
|||
public Point[][] TransformOrigin { get; private set; } |
|||
|
|||
/// <summary>
|
|||
/// Gets or sets the block index in mode-decision scan order.
|
|||
/// </summary>
|
|||
public int ModeDecisionIndex { get; set; } |
|||
|
|||
/// <summary>
|
|||
/// Gets or sets the scan offset from this square block to the next block at the same depth.
|
|||
/// </summary>
|
|||
public int NextDepthOffset { get; set; } |
|||
|
|||
/// <summary>
|
|||
/// Gets or sets the scan offset from this square block to its first child at the next depth.
|
|||
/// </summary>
|
|||
public int Depth1Offset { get; set; } |
|||
|
|||
/// <summary>
|
|||
/// Gets a value indicating whether this block is redundant to another.
|
|||
/// </summary>
|
|||
public bool IsRedundant => this.RedunancyList.Count > 0; |
|||
|
|||
/// <summary>
|
|||
/// Gets or sets the mode-decision indices of blocks with the same size and origin as this block.
|
|||
/// </summary>
|
|||
public List<int> RedunancyList { get; set; } |
|||
|
|||
/// <summary>
|
|||
/// Gets or sets the zero-based component index of this block within a non-square partition.
|
|||
/// </summary>
|
|||
public int NonSquareIndex { get; set; } |
|||
|
|||
/// <summary>
|
|||
/// Gets or sets the number of component blocks produced by this partition shape.
|
|||
/// </summary>
|
|||
public int TotalNonSuareCount { get; set; } |
|||
|
|||
/// <summary>
|
|||
/// Gets the chroma block width in pixels.
|
|||
/// </summary>
|
|||
public int BlockWidthUv { get; private set; } |
|||
|
|||
/// <summary>
|
|||
/// Gets the chroma block height in pixels.
|
|||
/// </summary>
|
|||
public int BlockHeightUv { get; private set; } |
|||
|
|||
/// <summary>
|
|||
/// Gets or sets the quadtree depth of this block within its superblock.
|
|||
/// </summary>
|
|||
public int Depth { get; set; } |
|||
|
|||
/// <summary>
|
|||
/// Gets or sets the width and height, in pixels, of the square sequence region that produced this block.
|
|||
/// </summary>
|
|||
public int SequenceSize { get; set; } |
|||
|
|||
/// <summary>
|
|||
/// Gets or sets a value indicating whether this block belongs to the last quadrant of its parent.
|
|||
/// </summary>
|
|||
public bool IsLastQuadrant { get; set; } |
|||
} |
|||
File diff suppressed because it is too large
@ -1,50 +0,0 @@ |
|||
// Copyright (c) Six Labors.
|
|||
// Licensed under the Six Labors Split License.
|
|||
|
|||
namespace SixLabors.ImageSharp.Formats.Heif.Av1.ModeDecision; |
|||
|
|||
/// <summary>
|
|||
/// Identifies a predefined AV1 mode-decision geometry with a fixed superblock size, search depth, and partition set.
|
|||
/// </summary>
|
|||
internal enum Av1GeometryIndex |
|||
{ |
|||
/// <summary>
|
|||
/// The 64-pixel, four-depth geometry limited to square partitions.
|
|||
/// </summary>
|
|||
Geometry0, |
|||
|
|||
/// <summary>
|
|||
/// The 64-pixel, four-depth geometry with horizontal and vertical binary partitions down to 16 pixels.
|
|||
/// </summary>
|
|||
Geometry1, |
|||
|
|||
/// <summary>
|
|||
/// The 64-pixel, four-depth geometry with horizontal and vertical binary partitions down to 8 pixels.
|
|||
/// </summary>
|
|||
Geometry2, |
|||
|
|||
/// <summary>
|
|||
/// The 64-pixel, four-depth geometry with binary partitions at every supported size.
|
|||
/// </summary>
|
|||
Geometry3, |
|||
|
|||
/// <summary>
|
|||
/// The 64-pixel, five-depth geometry with binary partitions at every supported size.
|
|||
/// </summary>
|
|||
Geometry4, |
|||
|
|||
/// <summary>
|
|||
/// The 64-pixel, five-depth geometry that also enables four-way horizontal and vertical partitions.
|
|||
/// </summary>
|
|||
Geometry5, |
|||
|
|||
/// <summary>
|
|||
/// The 64-pixel, five-depth geometry that enables all supported partition shapes.
|
|||
/// </summary>
|
|||
Geometry6, |
|||
|
|||
/// <summary>
|
|||
/// The 128-pixel, six-depth geometry that enables all supported partition shapes.
|
|||
/// </summary>
|
|||
Geometry7, |
|||
} |
|||
@ -0,0 +1,42 @@ |
|||
// Copyright (c) Six Labors.
|
|||
// Licensed under the Six Labors Split License.
|
|||
|
|||
using System.Runtime.InteropServices; |
|||
using SixLabors.ImageSharp.Formats.Heif.Av1.Transform; |
|||
|
|||
namespace SixLabors.ImageSharp.Formats.Heif.Av1.Tiling; |
|||
|
|||
/// <summary>
|
|||
/// Stores the entropy syntax retained for one AV1 transform block.
|
|||
/// </summary>
|
|||
[StructLayout(LayoutKind.Sequential, Size = sizeof(int))] |
|||
internal struct Av1EncoderTransformBlockState |
|||
{ |
|||
/// <summary>
|
|||
/// Stores the position after the final nonzero coefficient.
|
|||
/// </summary>
|
|||
private ushort endOfBlock; |
|||
|
|||
/// <summary>
|
|||
/// Stores the selected transform type.
|
|||
/// </summary>
|
|||
private Av1TransformType transformType; |
|||
|
|||
/// <summary>
|
|||
/// Gets or sets the position after the final nonzero coefficient.
|
|||
/// </summary>
|
|||
public ushort EndOfBlock |
|||
{ |
|||
readonly get => this.endOfBlock; |
|||
set => this.endOfBlock = value; |
|||
} |
|||
|
|||
/// <summary>
|
|||
/// Gets or sets the selected transform type.
|
|||
/// </summary>
|
|||
public Av1TransformType TransformType |
|||
{ |
|||
readonly get => this.transformType; |
|||
set => this.transformType = value; |
|||
} |
|||
} |
|||
@ -1,15 +0,0 @@ |
|||
// Copyright (c) Six Labors.
|
|||
// Licensed under the Six Labors Split License.
|
|||
|
|||
namespace SixLabors.ImageSharp.Formats.Heif.Av1.Tiling; |
|||
|
|||
/// <summary>
|
|||
/// Describes a superblock node's location, dimensions, and children in the encoder partition tree.
|
|||
/// </summary>
|
|||
internal class Av1SuperblockGeometry |
|||
{ |
|||
/// <summary>
|
|||
/// Gets or sets a value indicating whether the superblock lies completely within the coded frame.
|
|||
/// </summary>
|
|||
public bool IsComplete { get; set; } |
|||
} |
|||
File diff suppressed because it is too large
@ -1,54 +0,0 @@ |
|||
// Copyright (c) Six Labors.
|
|||
// Licensed under the Six Labors Split License.
|
|||
|
|||
using System.Diagnostics.CodeAnalysis; |
|||
using System.Runtime.CompilerServices; |
|||
using SixLabors.ImageSharp.Formats.Heif.Av1.Transform; |
|||
|
|||
namespace SixLabors.ImageSharp.Formats.Heif.Av1.Tiling; |
|||
|
|||
/// <summary>
|
|||
/// Stores the transform syntax and coefficient range for one AV1 transform unit.
|
|||
/// </summary>
|
|||
internal struct Av1TransformUnit |
|||
{ |
|||
/// <summary>
|
|||
/// Stores the luma, blue-difference, and red-difference end-of-block positions.
|
|||
/// </summary>
|
|||
private InlineArray3<ushort> nzCoefficientCount; |
|||
|
|||
/// <summary>
|
|||
/// Stores the luma and shared chroma transform types.
|
|||
/// </summary>
|
|||
private InlineArray2<Av1TransformType> transformType; |
|||
|
|||
/// <summary>
|
|||
/// Gets the nonzero-coefficient count for each color plane.
|
|||
/// </summary>
|
|||
[UnscopedRef] |
|||
public Span<ushort> NzCoefficientCount => this.nzCoefficientCount; |
|||
|
|||
/// <summary>
|
|||
/// Gets the transform type selected for each color plane.
|
|||
/// </summary>
|
|||
[UnscopedRef] |
|||
public Span<Av1TransformType> TransformType => this.transformType; |
|||
|
|||
/// <summary>
|
|||
/// Stores the three per-plane coefficient counts inline.
|
|||
/// </summary>
|
|||
[InlineArray(3)] |
|||
private struct InlineArray3<T> |
|||
{ |
|||
private T element; |
|||
} |
|||
|
|||
/// <summary>
|
|||
/// Stores the luma and shared chroma transform types inline.
|
|||
/// </summary>
|
|||
[InlineArray(Av1Constants.PlaneTypeCount)] |
|||
private struct InlineArray2<T> |
|||
{ |
|||
private T element; |
|||
} |
|||
} |
|||
Loading…
Reference in new issue