mirror of https://github.com/SixLabors/ImageSharp
Ynse Hoornenborg
2 years ago
6 changed files with 349 additions and 40 deletions
@ -0,0 +1,75 @@ |
|||
// Copyright (c) Six Labors.
|
|||
// Licensed under the Six Labors Split License.
|
|||
|
|||
using SixLabors.ImageSharp.Formats.Heif.Av1.OpenBitstreamUnit; |
|||
using SixLabors.ImageSharp.Formats.Heif.Av1.Quantization; |
|||
|
|||
namespace SixLabors.ImageSharp.Formats.Heif.Av1.Symbol; |
|||
|
|||
internal class Av1ParseAboveNeighbor4x4Context |
|||
{ |
|||
/* Buffer holding the transform sizes of the previous 4x4 block row. */ |
|||
private readonly int[] aboveTransformWidth; |
|||
|
|||
/* Buffer holding the partition context of the previous 4x4 block row. */ |
|||
private int[] abovePartitionWidth; |
|||
|
|||
/* Buffer holding the sign of the DC coefficients and the |
|||
cumulative sum of the coefficient levels of the above 4x4 |
|||
blocks corresponding to the current super block row. */ |
|||
private int[][] aboveContext = new int[Av1Constants.MaxPlanes][]; |
|||
|
|||
/* Buffer holding the seg_id_predicted of the previous 4x4 block row. */ |
|||
private int[] aboveSegmentIdPredictionContext; |
|||
|
|||
/* Value of base colors for Y, U, and V */ |
|||
private int[][] abovePaletteColors = new int[Av1Constants.MaxPlanes][]; |
|||
|
|||
private int[] aboveCompGroupIndex; |
|||
|
|||
public Av1ParseAboveNeighbor4x4Context(int planesCount, int modeInfoColumnCount) |
|||
{ |
|||
int wide64x64Count = Av1BlockSize.Block64x64.Get4x4WideCount(); |
|||
this.aboveTransformWidth = new int[modeInfoColumnCount]; |
|||
this.abovePartitionWidth = new int[modeInfoColumnCount]; |
|||
for (int i = 0; i < planesCount; i++) |
|||
{ |
|||
this.aboveContext[i] = new int[modeInfoColumnCount]; |
|||
this.abovePaletteColors[i] = new int[wide64x64Count * Av1Constants.PaletteMaxSize]; |
|||
} |
|||
|
|||
this.aboveSegmentIdPredictionContext = new int[modeInfoColumnCount]; |
|||
this.aboveCompGroupIndex = new int[modeInfoColumnCount]; |
|||
} |
|||
|
|||
public int[] AbovePartitionWidth => this.abovePartitionWidth; |
|||
|
|||
public void Clear(ObuSequenceHeader sequenceHeader) |
|||
{ |
|||
int planeCount = sequenceHeader.ColorConfig.ChannelCount; |
|||
Array.Fill(this.aboveTransformWidth, Av1TransformSize.Size64x64.GetWidth()); |
|||
Array.Fill(this.abovePartitionWidth, 0); |
|||
for (int i = 0; i < planeCount; i++) |
|||
{ |
|||
Array.Fill(this.aboveContext[i], 0); |
|||
Array.Fill(this.abovePaletteColors[i], 0); |
|||
} |
|||
|
|||
Array.Fill(this.aboveSegmentIdPredictionContext, 0); |
|||
Array.Fill(this.aboveCompGroupIndex, 0); |
|||
} |
|||
|
|||
public void UpdatePartition(Point modeInfoLocation, Av1TileLocation tileLoc, Av1BlockSize subSize, Av1BlockSize blockSize) |
|||
{ |
|||
int startIndex = modeInfoLocation.X - tileLoc.ModeInfoColumnStart; |
|||
int bw = blockSize.Get4x4WideCount(); |
|||
int value = Av1PartitionContext.GetAboveContext(subSize); |
|||
for (int i = 0; i < bw; i++) |
|||
{ |
|||
this.abovePartitionWidth[startIndex + i] = value; |
|||
} |
|||
} |
|||
|
|||
internal void ClearContext(int plane, int offset, int length) |
|||
=> Array.Fill(this.aboveContext[plane], 0, offset, length); |
|||
} |
|||
@ -0,0 +1,76 @@ |
|||
// Copyright (c) Six Labors.
|
|||
// Licensed under the Six Labors Split License.
|
|||
|
|||
using SixLabors.ImageSharp.Formats.Heif.Av1.OpenBitstreamUnit; |
|||
using SixLabors.ImageSharp.Formats.Heif.Av1.Quantization; |
|||
|
|||
namespace SixLabors.ImageSharp.Formats.Heif.Av1.Symbol; |
|||
|
|||
internal class Av1ParseLeftNeighbor4x4Context |
|||
{ |
|||
/* Buffer holding the transform sizes of the left 4x4 blocks corresponding |
|||
to the current super block row. */ |
|||
private readonly int[] leftTransformHeight; |
|||
|
|||
/* Buffer holding the partition context of the left 4x4 blocks corresponding |
|||
to the current super block row. */ |
|||
private int[] leftPartitionHeight; |
|||
|
|||
/* Buffer holding the sign of the DC coefficients and the |
|||
cumulative sum of the coefficient levels of the left 4x4 |
|||
blocks corresponding to the current super block row. */ |
|||
private int[][] leftContext = new int[Av1Constants.MaxPlanes][]; |
|||
|
|||
/* Buffer holding the seg_id_predicted of the previous 4x4 block row. */ |
|||
private int[] leftSegmentIdPredictionContext; |
|||
|
|||
/* Value of base colors for Y, U, and V */ |
|||
private int[][] leftPaletteColors = new int[Av1Constants.MaxPlanes][]; |
|||
|
|||
private int[] leftCompGroupIndex; |
|||
|
|||
public Av1ParseLeftNeighbor4x4Context(int planesCount, int superblockModeInfoSize) |
|||
{ |
|||
this.leftTransformHeight = new int[superblockModeInfoSize]; |
|||
this.leftPartitionHeight = new int[superblockModeInfoSize]; |
|||
for (int i = 0; i < planesCount; i++) |
|||
{ |
|||
this.leftContext[i] = new int[superblockModeInfoSize]; |
|||
this.leftPaletteColors[i] = new int[superblockModeInfoSize * Av1Constants.PaletteMaxSize]; |
|||
} |
|||
|
|||
this.leftSegmentIdPredictionContext = new int[superblockModeInfoSize]; |
|||
this.leftCompGroupIndex = new int[superblockModeInfoSize]; |
|||
} |
|||
|
|||
public int[] LeftPartitionHeight => this.leftPartitionHeight; |
|||
|
|||
public void Clear(ObuSequenceHeader sequenceHeader) |
|||
{ |
|||
int planeCount = sequenceHeader.ColorConfig.ChannelCount; |
|||
Array.Fill(this.leftTransformHeight, Av1TransformSize.Size64x64.GetHeight()); |
|||
Array.Fill(this.leftPartitionHeight, 0); |
|||
for (int i = 0; i < planeCount; i++) |
|||
{ |
|||
Array.Fill(this.leftContext[i], 0); |
|||
Array.Fill(this.leftPaletteColors[i], 0); |
|||
} |
|||
|
|||
Array.Fill(this.leftSegmentIdPredictionContext, 0); |
|||
Array.Fill(this.leftCompGroupIndex, 0); |
|||
} |
|||
|
|||
public void UpdatePartition(Point modeInfoLocation, Av1SuperblockInfo superblockInfo, Av1BlockSize subSize, Av1BlockSize blockSize) |
|||
{ |
|||
int startIndex = (modeInfoLocation.Y - superblockInfo.Position.Y) & Av1PartitionContext.Mask; |
|||
int bh = blockSize.Get4x4HighCount(); |
|||
int value = Av1PartitionContext.GetLeftContext(subSize); |
|||
for (int i = 0; i < bh; i++) |
|||
{ |
|||
this.leftPartitionHeight[startIndex + i] = value; |
|||
} |
|||
} |
|||
|
|||
internal void ClearContext(int plane, int offset, int length) |
|||
=> Array.Fill(this.leftContext[plane], 0, offset, length); |
|||
} |
|||
@ -0,0 +1,23 @@ |
|||
// Copyright (c) Six Labors.
|
|||
// Licensed under the Six Labors Split License.
|
|||
|
|||
namespace SixLabors.ImageSharp.Formats.Heif.Av1.Symbol; |
|||
|
|||
// Generates 5 bit field in which each bit set to 1 represents
|
|||
// a BlockSize partition 11111 means we split 128x128, 64x64, 32x32, 16x16
|
|||
// and 8x8. 10000 means we just split the 128x128 to 64x64
|
|||
internal class Av1PartitionContext |
|||
{ |
|||
private static readonly int[] AboveLookup = |
|||
[31, 31, 30, 30, 30, 28, 28, 28, 24, 24, 24, 16, 16, 16, 0, 0, 31, 28, 30, 24, 28, 16]; |
|||
|
|||
private static readonly int[] LeftLookup = |
|||
[31, 30, 31, 30, 28, 30, 28, 24, 28, 24, 16, 24, 16, 0, 16, 0, 28, 31, 24, 30, 16, 28]; |
|||
|
|||
// Mask to extract ModeInfo offset within max ModeInfoBlock
|
|||
public const int Mask = (1 << (7 - 2)) - 1; |
|||
|
|||
public static int GetAboveContext(Av1BlockSize blockSize) => AboveLookup[(int)blockSize]; |
|||
|
|||
public static int GetLeftContext(Av1BlockSize blockSize) => LeftLookup[(int)blockSize]; |
|||
} |
|||
@ -0,0 +1,45 @@ |
|||
// Copyright (c) Six Labors.
|
|||
// Licensed under the Six Labors Split License.
|
|||
|
|||
using SixLabors.ImageSharp.Formats.Heif.Av1.OpenBitstreamUnit; |
|||
|
|||
namespace SixLabors.ImageSharp.Formats.Heif.Av1.Symbol; |
|||
|
|||
internal class Av1TileLocation |
|||
{ |
|||
public Av1TileLocation(int row, int column, ObuFrameHeader frameInfo) |
|||
{ |
|||
this.SetTileRow(frameInfo.TilesInfo, frameInfo.ModeInfoRowCount, row); |
|||
this.SetTileColumn(frameInfo.TilesInfo, frameInfo.ModeInfoColumnCount, column); |
|||
} |
|||
|
|||
public int ModeInfoRowStart { get; private set; } |
|||
|
|||
public int ModeInfoRowEnd { get; private set; } |
|||
|
|||
public int ModeInfoColumnStart { get; private set; } |
|||
|
|||
public int ModeInfoColumnEnd { get; private set; } |
|||
|
|||
public Point TileIndex { get; private set; } |
|||
|
|||
public int TileIndexInRasterOrder { get; } |
|||
|
|||
public void SetTileRow(ObuTileGroupHeader tileGroupHeader, int modeInfoRowCount, int row) |
|||
{ |
|||
this.ModeInfoRowStart = tileGroupHeader.TileRowStartModeInfo[row]; |
|||
this.ModeInfoRowEnd = Math.Min(tileGroupHeader.TileRowStartModeInfo[row + 1], modeInfoRowCount); |
|||
Point loc = this.TileIndex; |
|||
loc.Y = row; |
|||
this.TileIndex = loc; |
|||
} |
|||
|
|||
public void SetTileColumn(ObuTileGroupHeader tileGroupHeader, int modeInfoColumnCount, int column) |
|||
{ |
|||
this.ModeInfoColumnStart = tileGroupHeader.TileColumnStartModeInfo[column]; |
|||
this.ModeInfoColumnEnd = Math.Min(tileGroupHeader.TileColumnStartModeInfo[column + 1], modeInfoColumnCount); |
|||
Point loc = this.TileIndex; |
|||
loc.X = column; |
|||
this.TileIndex = loc; |
|||
} |
|||
} |
|||
Loading…
Reference in new issue