mirror of
https://git.ryujinx.app/ryubing/ryujinx.git
synced 2026-03-09 08:01:07 +00:00
f91cd05260
Unmerged PR from OG Ryujinx (#4367). From @gdkchan: > The main goal of this change is porting the loop filtering from libvpx, which should fix the block artifacts on some VP9 videos on games using NVDEC to decode them. In addition to that, there are two other changes: > > - The remaining decoder code required to decode a VP9 video (with headers included) has been added. That was done because it's much better to test the decoder standalone with a video file. I decided to keep that code on the emulator, even if some of it is unused, since it makes standalone testing easier in the future too, and we can include unit tests with video files. > - Large refactoring of both new and existing code to conform with our conding [sic] styles, done by @TSRBerry (thanks!) Some of it has been automated. > > Since we had no loop filtering before, this change will make video decoding slower. That may cause frame drop etc if the decoder is not fast enough in some games. I plan to optimize the decoder more in the future to make up for that, but if possible I'd prefer to not do it as part of this PR, but if the perf loss is too severe I might consider. > > This will need to be tested on games that had the block artifacts, it would be nice to confirm if they match hardware now, and get some before/after screenshots etc. Comment from @Bjorn29512: > Significantly improves the block artifacts in FE: Engage. > > Before: >  > > After: >  --------- Co-authored-by: gdkchan <gab.dark.100@gmail.com> Co-authored-by: TSR Berry <20988865+TSRBerry@users.noreply.github.com>
655 lines
24 KiB
C#
655 lines
24 KiB
C#
using Ryujinx.Graphics.Nvdec.Vp9.Common;
|
|
using Ryujinx.Graphics.Nvdec.Vp9.Types;
|
|
using System;
|
|
using static Ryujinx.Graphics.Nvdec.Vp9.Dsp.IntraPred;
|
|
|
|
namespace Ryujinx.Graphics.Nvdec.Vp9
|
|
{
|
|
internal static class ReconIntra
|
|
{
|
|
public static readonly TxType[] IntraModeToTxTypeLookup =
|
|
{
|
|
TxType.DctDct, // DC
|
|
TxType.AdstDct, // V
|
|
TxType.DctAdst, // H
|
|
TxType.DctDct, // D45
|
|
TxType.AdstAdst, // D135
|
|
TxType.AdstDct, // D117
|
|
TxType.DctAdst, // D153
|
|
TxType.DctAdst, // D207
|
|
TxType.AdstDct, // D63
|
|
TxType.AdstAdst // TM
|
|
};
|
|
|
|
private const int NeedLeft = 1 << 1;
|
|
private const int NeedAbove = 1 << 2;
|
|
private const int NeedAboveRight = 1 << 3;
|
|
|
|
private static ReadOnlySpan<byte> ExtendModes => new byte[]
|
|
{
|
|
NeedAbove | NeedLeft, // DC
|
|
NeedAbove, // V
|
|
NeedLeft, // H
|
|
NeedAboveRight, // D45
|
|
NeedLeft | NeedAbove, // D135
|
|
NeedLeft | NeedAbove, // D117
|
|
NeedLeft | NeedAbove, // D153
|
|
NeedLeft, // D207
|
|
NeedAboveRight, // D63
|
|
NeedLeft | NeedAbove // TM
|
|
};
|
|
|
|
private unsafe delegate void IntraPredFn(byte* dst, int stride, byte* above, byte* left);
|
|
|
|
private static readonly unsafe IntraPredFn[][] Pred =
|
|
{
|
|
new IntraPredFn[] { null, null, null, null },
|
|
new IntraPredFn[] { VPredictor4x4, VPredictor8x8, VPredictor16x16, VPredictor32x32 },
|
|
new IntraPredFn[] { HPredictor4x4, HPredictor8x8, HPredictor16x16, HPredictor32x32 },
|
|
new IntraPredFn[] { D45Predictor4x4, D45Predictor8x8, D45Predictor16x16, D45Predictor32x32 },
|
|
new IntraPredFn[] { D135Predictor4x4, D135Predictor8x8, D135Predictor16x16, D135Predictor32x32 },
|
|
new IntraPredFn[] { D117Predictor4x4, D117Predictor8x8, D117Predictor16x16, D117Predictor32x32 },
|
|
new IntraPredFn[] { D153Predictor4x4, D153Predictor8x8, D153Predictor16x16, D153Predictor32x32 },
|
|
new IntraPredFn[] { D207Predictor4x4, D207Predictor8x8, D207Predictor16x16, D207Predictor32x32 },
|
|
new IntraPredFn[] { D63Predictor4x4, D63Predictor8x8, D63Predictor16x16, D63Predictor32x32 },
|
|
new IntraPredFn[] { TmPredictor4x4, TmPredictor8x8, TmPredictor16x16, TmPredictor32x32 }
|
|
};
|
|
|
|
private static readonly unsafe IntraPredFn[][][] DcPred =
|
|
{
|
|
new[]
|
|
{
|
|
new IntraPredFn[]
|
|
{
|
|
Dc128Predictor4x4, Dc128Predictor8x8, Dc128Predictor16x16, Dc128Predictor32x32
|
|
},
|
|
new IntraPredFn[]
|
|
{
|
|
DcTopPredictor4x4, DcTopPredictor8x8, DcTopPredictor16x16, DcTopPredictor32x32
|
|
}
|
|
},
|
|
new[]
|
|
{
|
|
new IntraPredFn[]
|
|
{
|
|
DcLeftPredictor4x4, DcLeftPredictor8x8, DcLeftPredictor16x16, DcLeftPredictor32x32
|
|
},
|
|
new IntraPredFn[] { DcPredictor4x4, DcPredictor8x8, DcPredictor16x16, DcPredictor32x32 }
|
|
}
|
|
};
|
|
|
|
private unsafe delegate void IntraHighPredFn(ushort* dst, int stride, ushort* above, ushort* left, int bd);
|
|
|
|
private static readonly unsafe IntraHighPredFn[][] PredHigh =
|
|
{
|
|
new IntraHighPredFn[] { null, null, null, null },
|
|
new IntraHighPredFn[]
|
|
{
|
|
HighbdVPredictor4x4, HighbdVPredictor8x8, HighbdVPredictor16x16, HighbdVPredictor32x32
|
|
},
|
|
new IntraHighPredFn[]
|
|
{
|
|
HighbdHPredictor4x4, HighbdHPredictor8x8, HighbdHPredictor16x16, HighbdHPredictor32x32
|
|
},
|
|
new IntraHighPredFn[]
|
|
{
|
|
HighbdD45Predictor4x4, HighbdD45Predictor8x8, HighbdD45Predictor16x16, HighbdD45Predictor32x32
|
|
},
|
|
new IntraHighPredFn[]
|
|
{
|
|
HighbdD135Predictor4x4, HighbdD135Predictor8x8, HighbdD135Predictor16x16,
|
|
HighbdD135Predictor32x32
|
|
},
|
|
new IntraHighPredFn[]
|
|
{
|
|
HighbdD117Predictor4x4, HighbdD117Predictor8x8, HighbdD117Predictor16x16,
|
|
HighbdD117Predictor32x32
|
|
},
|
|
new IntraHighPredFn[]
|
|
{
|
|
HighbdD153Predictor4x4, HighbdD153Predictor8x8, HighbdD153Predictor16x16,
|
|
HighbdD153Predictor32x32
|
|
},
|
|
new IntraHighPredFn[]
|
|
{
|
|
HighbdD207Predictor4x4, HighbdD207Predictor8x8, HighbdD207Predictor16x16,
|
|
HighbdD207Predictor32x32
|
|
},
|
|
new IntraHighPredFn[]
|
|
{
|
|
HighbdD63Predictor4x4, HighbdD63Predictor8x8, HighbdD63Predictor16x16, HighbdD63Predictor32x32
|
|
},
|
|
new IntraHighPredFn[]
|
|
{
|
|
HighbdTmPredictor4x4, HighbdTmPredictor8x8, HighbdTmPredictor16x16, HighbdTmPredictor32x32
|
|
}
|
|
};
|
|
|
|
private static readonly unsafe IntraHighPredFn[][][] DcPredHigh =
|
|
{
|
|
new[]
|
|
{
|
|
new IntraHighPredFn[]
|
|
{
|
|
HighbdDc128Predictor4x4, HighbdDc128Predictor8x8, HighbdDc128Predictor16x16,
|
|
HighbdDc128Predictor32x32
|
|
},
|
|
new IntraHighPredFn[]
|
|
{
|
|
HighbdDcTopPredictor4x4, HighbdDcTopPredictor8x8, HighbdDcTopPredictor16x16,
|
|
HighbdDcTopPredictor32x32
|
|
}
|
|
},
|
|
new[]
|
|
{
|
|
new IntraHighPredFn[]
|
|
{
|
|
HighbdDcLeftPredictor4x4, HighbdDcLeftPredictor8x8, HighbdDcLeftPredictor16x16,
|
|
HighbdDcLeftPredictor32x32
|
|
},
|
|
new IntraHighPredFn[]
|
|
{
|
|
HighbdDcPredictor4x4, HighbdDcPredictor8x8, HighbdDcPredictor16x16,
|
|
HighbdDcPredictor32x32
|
|
}
|
|
}
|
|
};
|
|
|
|
private static unsafe void BuildIntraPredictorsHigh(
|
|
ref MacroBlockD xd,
|
|
byte* ref8,
|
|
int refStride,
|
|
byte* dst8,
|
|
int dstStride,
|
|
PredictionMode mode,
|
|
TxSize txSize,
|
|
int upAvailable,
|
|
int leftAvailable,
|
|
int rightAvailable,
|
|
int x,
|
|
int y,
|
|
int plane)
|
|
{
|
|
int i;
|
|
ushort* dst = (ushort*)dst8;
|
|
ushort* refr = (ushort*)ref8;
|
|
ushort* leftCol = stackalloc ushort[32];
|
|
ushort* aboveData = stackalloc ushort[64 + 16];
|
|
ushort* aboveRow = aboveData + 16;
|
|
ushort* constAboveRow = aboveRow;
|
|
int bs = 4 << (int)txSize;
|
|
int frameWidth, frameHeight;
|
|
int x0, y0;
|
|
ref MacroBlockDPlane pd = ref xd.Plane[plane];
|
|
int needLeft = ExtendModes[(int)mode] & NeedLeft;
|
|
int needAbove = ExtendModes[(int)mode] & NeedAbove;
|
|
int needAboveRight = ExtendModes[(int)mode] & NeedAboveRight;
|
|
int baseVal = 128 << (xd.Bd - 8);
|
|
// 127 127 127 .. 127 127 127 127 127 127
|
|
// 129 A B .. Y Z
|
|
// 129 C D .. W X
|
|
// 129 E F .. U V
|
|
// 129 G H .. S T T T T T
|
|
// For 10 bit and 12 bit, 127 and 129 are replaced by base -1 and base + 1.
|
|
|
|
// Get current frame pointer, width and height.
|
|
if (plane == 0)
|
|
{
|
|
frameWidth = xd.CurBuf.Width;
|
|
frameHeight = xd.CurBuf.Height;
|
|
}
|
|
else
|
|
{
|
|
frameWidth = xd.CurBuf.UvWidth;
|
|
frameHeight = xd.CurBuf.UvHeight;
|
|
}
|
|
|
|
// Get block position in current frame.
|
|
x0 = (-xd.MbToLeftEdge >> (3 + pd.SubsamplingX)) + x;
|
|
y0 = (-xd.MbToTopEdge >> (3 + pd.SubsamplingY)) + y;
|
|
|
|
// NEED_LEFT
|
|
if (needLeft != 0)
|
|
{
|
|
if (leftAvailable != 0)
|
|
{
|
|
if (xd.MbToBottomEdge < 0)
|
|
{
|
|
/* slower path if the block needs border extension */
|
|
if (y0 + bs <= frameHeight)
|
|
{
|
|
for (i = 0; i < bs; ++i)
|
|
{
|
|
leftCol[i] = refr[(i * refStride) - 1];
|
|
}
|
|
}
|
|
else
|
|
{
|
|
int extendBottom = frameHeight - y0;
|
|
for (i = 0; i < extendBottom; ++i)
|
|
{
|
|
leftCol[i] = refr[(i * refStride) - 1];
|
|
}
|
|
|
|
for (; i < bs; ++i)
|
|
{
|
|
leftCol[i] = refr[((extendBottom - 1) * refStride) - 1];
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
/* faster path if the block does not need extension */
|
|
for (i = 0; i < bs; ++i)
|
|
{
|
|
leftCol[i] = refr[(i * refStride) - 1];
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
MemoryUtil.Fill(leftCol, (ushort)(baseVal + 1), bs);
|
|
}
|
|
}
|
|
|
|
// NEED_ABOVE
|
|
if (needAbove != 0)
|
|
{
|
|
if (upAvailable != 0)
|
|
{
|
|
ushort* aboveRef = refr - refStride;
|
|
if (xd.MbToRightEdge < 0)
|
|
{
|
|
/* slower path if the block needs border extension */
|
|
if (x0 + bs <= frameWidth)
|
|
{
|
|
MemoryUtil.Copy(aboveRow, aboveRef, bs);
|
|
}
|
|
else if (x0 <= frameWidth)
|
|
{
|
|
int r = frameWidth - x0;
|
|
MemoryUtil.Copy(aboveRow, aboveRef, r);
|
|
MemoryUtil.Fill(aboveRow + r, aboveRow[r - 1], x0 + bs - frameWidth);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
/* faster path if the block does not need extension */
|
|
if (bs == 4 && rightAvailable != 0 && leftAvailable != 0)
|
|
{
|
|
constAboveRow = aboveRef;
|
|
}
|
|
else
|
|
{
|
|
MemoryUtil.Copy(aboveRow, aboveRef, bs);
|
|
}
|
|
}
|
|
|
|
aboveRow[-1] = leftAvailable != 0 ? aboveRef[-1] : (ushort)(baseVal + 1);
|
|
}
|
|
else
|
|
{
|
|
MemoryUtil.Fill(aboveRow, (ushort)(baseVal - 1), bs);
|
|
aboveRow[-1] = (ushort)(baseVal - 1);
|
|
}
|
|
}
|
|
|
|
// NEED_ABOVERIGHT
|
|
if (needAboveRight != 0)
|
|
{
|
|
if (upAvailable != 0)
|
|
{
|
|
ushort* aboveRef = refr - refStride;
|
|
if (xd.MbToRightEdge < 0)
|
|
{
|
|
/* slower path if the block needs border extension */
|
|
if (x0 + (2 * bs) <= frameWidth)
|
|
{
|
|
if (rightAvailable != 0 && bs == 4)
|
|
{
|
|
MemoryUtil.Copy(aboveRow, aboveRef, 2 * bs);
|
|
}
|
|
else
|
|
{
|
|
MemoryUtil.Copy(aboveRow, aboveRef, bs);
|
|
MemoryUtil.Fill(aboveRow + bs, aboveRow[bs - 1], bs);
|
|
}
|
|
}
|
|
else if (x0 + bs <= frameWidth)
|
|
{
|
|
int r = frameWidth - x0;
|
|
if (rightAvailable != 0 && bs == 4)
|
|
{
|
|
MemoryUtil.Copy(aboveRow, aboveRef, r);
|
|
MemoryUtil.Fill(aboveRow + r, aboveRow[r - 1], x0 + (2 * bs) - frameWidth);
|
|
}
|
|
else
|
|
{
|
|
MemoryUtil.Copy(aboveRow, aboveRef, bs);
|
|
MemoryUtil.Fill(aboveRow + bs, aboveRow[bs - 1], bs);
|
|
}
|
|
}
|
|
else if (x0 <= frameWidth)
|
|
{
|
|
int r = frameWidth - x0;
|
|
MemoryUtil.Copy(aboveRow, aboveRef, r);
|
|
MemoryUtil.Fill(aboveRow + r, aboveRow[r - 1], x0 + (2 * bs) - frameWidth);
|
|
}
|
|
|
|
aboveRow[-1] = leftAvailable != 0 ? aboveRef[-1] : (ushort)(baseVal + 1);
|
|
}
|
|
else
|
|
{
|
|
/* faster path if the block does not need extension */
|
|
if (bs == 4 && rightAvailable != 0 && leftAvailable != 0)
|
|
{
|
|
constAboveRow = aboveRef;
|
|
}
|
|
else
|
|
{
|
|
MemoryUtil.Copy(aboveRow, aboveRef, bs);
|
|
if (bs == 4 && rightAvailable != 0)
|
|
{
|
|
MemoryUtil.Copy(aboveRow + bs, aboveRef + bs, bs);
|
|
}
|
|
else
|
|
{
|
|
MemoryUtil.Fill(aboveRow + bs, aboveRow[bs - 1], bs);
|
|
}
|
|
|
|
aboveRow[-1] = leftAvailable != 0 ? aboveRef[-1] : (ushort)(baseVal + 1);
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
MemoryUtil.Fill(aboveRow, (ushort)(baseVal - 1), bs * 2);
|
|
aboveRow[-1] = (ushort)(baseVal - 1);
|
|
}
|
|
}
|
|
|
|
// Predict
|
|
if (mode == PredictionMode.DcPred)
|
|
{
|
|
DcPredHigh[leftAvailable][upAvailable][(int)txSize](dst, dstStride, constAboveRow, leftCol, xd.Bd);
|
|
}
|
|
else
|
|
{
|
|
PredHigh[(int)mode][(int)txSize](dst, dstStride, constAboveRow, leftCol, xd.Bd);
|
|
}
|
|
}
|
|
|
|
public static unsafe void BuildIntraPredictors(
|
|
ref MacroBlockD xd,
|
|
byte* refr,
|
|
int refStride,
|
|
byte* dst,
|
|
int dstStride,
|
|
PredictionMode mode,
|
|
TxSize txSize,
|
|
int upAvailable,
|
|
int leftAvailable,
|
|
int rightAvailable,
|
|
int x,
|
|
int y,
|
|
int plane)
|
|
{
|
|
int i;
|
|
byte* leftCol = stackalloc byte[32];
|
|
byte* aboveData = stackalloc byte[64 + 16];
|
|
byte* aboveRow = aboveData + 16;
|
|
byte* constAboveRow = aboveRow;
|
|
int bs = 4 << (int)txSize;
|
|
int frameWidth, frameHeight;
|
|
int x0, y0;
|
|
ref MacroBlockDPlane pd = ref xd.Plane[plane];
|
|
|
|
// 127 127 127 .. 127 127 127 127 127 127
|
|
// 129 A B .. Y Z
|
|
// 129 C D .. W X
|
|
// 129 E F .. U V
|
|
// 129 G H .. S T T T T T
|
|
// ..
|
|
|
|
// Get current frame pointer, width and height.
|
|
if (plane == 0)
|
|
{
|
|
frameWidth = xd.CurBuf.Width;
|
|
frameHeight = xd.CurBuf.Height;
|
|
}
|
|
else
|
|
{
|
|
frameWidth = xd.CurBuf.UvWidth;
|
|
frameHeight = xd.CurBuf.UvHeight;
|
|
}
|
|
|
|
// Get block position in current frame.
|
|
x0 = (-xd.MbToLeftEdge >> (3 + pd.SubsamplingX)) + x;
|
|
y0 = (-xd.MbToTopEdge >> (3 + pd.SubsamplingY)) + y;
|
|
|
|
// NEED_LEFT
|
|
if ((ExtendModes[(int)mode] & NeedLeft) != 0)
|
|
{
|
|
if (leftAvailable != 0)
|
|
{
|
|
if (xd.MbToBottomEdge < 0)
|
|
{
|
|
/* Slower path if the block needs border extension */
|
|
if (y0 + bs <= frameHeight)
|
|
{
|
|
for (i = 0; i < bs; ++i)
|
|
{
|
|
leftCol[i] = refr[(i * refStride) - 1];
|
|
}
|
|
}
|
|
else
|
|
{
|
|
int extendBottom = frameHeight - y0;
|
|
for (i = 0; i < extendBottom; ++i)
|
|
{
|
|
leftCol[i] = refr[(i * refStride) - 1];
|
|
}
|
|
|
|
for (; i < bs; ++i)
|
|
{
|
|
leftCol[i] = refr[((extendBottom - 1) * refStride) - 1];
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
/* Faster path if the block does not need extension */
|
|
for (i = 0; i < bs; ++i)
|
|
{
|
|
leftCol[i] = refr[(i * refStride) - 1];
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
MemoryUtil.Fill(leftCol, (byte)129, bs);
|
|
}
|
|
}
|
|
|
|
// NEED_ABOVE
|
|
if ((ExtendModes[(int)mode] & NeedAbove) != 0)
|
|
{
|
|
if (upAvailable != 0)
|
|
{
|
|
byte* aboveRef = refr - refStride;
|
|
if (xd.MbToRightEdge < 0)
|
|
{
|
|
/* Slower path if the block needs border extension */
|
|
if (x0 + bs <= frameWidth)
|
|
{
|
|
MemoryUtil.Copy(aboveRow, aboveRef, bs);
|
|
}
|
|
else if (x0 <= frameWidth)
|
|
{
|
|
int r = frameWidth - x0;
|
|
MemoryUtil.Copy(aboveRow, aboveRef, r);
|
|
MemoryUtil.Fill(aboveRow + r, aboveRow[r - 1], x0 + bs - frameWidth);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
/* Faster path if the block does not need extension */
|
|
if (bs == 4 && rightAvailable != 0 && leftAvailable != 0)
|
|
{
|
|
constAboveRow = aboveRef;
|
|
}
|
|
else
|
|
{
|
|
MemoryUtil.Copy(aboveRow, aboveRef, bs);
|
|
}
|
|
}
|
|
|
|
aboveRow[-1] = leftAvailable != 0 ? aboveRef[-1] : (byte)129;
|
|
}
|
|
else
|
|
{
|
|
MemoryUtil.Fill(aboveRow, (byte)127, bs);
|
|
aboveRow[-1] = 127;
|
|
}
|
|
}
|
|
|
|
// NEED_ABOVERIGHT
|
|
if ((ExtendModes[(int)mode] & NeedAboveRight) != 0)
|
|
{
|
|
if (upAvailable != 0)
|
|
{
|
|
byte* aboveRef = refr - refStride;
|
|
if (xd.MbToRightEdge < 0)
|
|
{
|
|
/* Slower path if the block needs border extension */
|
|
if (x0 + (2 * bs) <= frameWidth)
|
|
{
|
|
if (rightAvailable != 0 && bs == 4)
|
|
{
|
|
MemoryUtil.Copy(aboveRow, aboveRef, 2 * bs);
|
|
}
|
|
else
|
|
{
|
|
MemoryUtil.Copy(aboveRow, aboveRef, bs);
|
|
MemoryUtil.Fill(aboveRow + bs, aboveRow[bs - 1], bs);
|
|
}
|
|
}
|
|
else if (x0 + bs <= frameWidth)
|
|
{
|
|
int r = frameWidth - x0;
|
|
if (rightAvailable != 0 && bs == 4)
|
|
{
|
|
MemoryUtil.Copy(aboveRow, aboveRef, r);
|
|
MemoryUtil.Fill(aboveRow + r, aboveRow[r - 1], x0 + (2 * bs) - frameWidth);
|
|
}
|
|
else
|
|
{
|
|
MemoryUtil.Copy(aboveRow, aboveRef, bs);
|
|
MemoryUtil.Fill(aboveRow + bs, aboveRow[bs - 1], bs);
|
|
}
|
|
}
|
|
else if (x0 <= frameWidth)
|
|
{
|
|
int r = frameWidth - x0;
|
|
MemoryUtil.Copy(aboveRow, aboveRef, r);
|
|
MemoryUtil.Fill(aboveRow + r, aboveRow[r - 1], x0 + (2 * bs) - frameWidth);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
/* Faster path if the block does not need extension */
|
|
if (bs == 4 && rightAvailable != 0 && leftAvailable != 0)
|
|
{
|
|
constAboveRow = aboveRef;
|
|
}
|
|
else
|
|
{
|
|
MemoryUtil.Copy(aboveRow, aboveRef, bs);
|
|
if (bs == 4 && rightAvailable != 0)
|
|
{
|
|
MemoryUtil.Copy(aboveRow + bs, aboveRef + bs, bs);
|
|
}
|
|
else
|
|
{
|
|
MemoryUtil.Fill(aboveRow + bs, aboveRow[bs - 1], bs);
|
|
}
|
|
}
|
|
}
|
|
|
|
aboveRow[-1] = leftAvailable != 0 ? aboveRef[-1] : (byte)129;
|
|
}
|
|
else
|
|
{
|
|
MemoryUtil.Fill(aboveRow, (byte)127, bs * 2);
|
|
aboveRow[-1] = 127;
|
|
}
|
|
}
|
|
|
|
// Predict
|
|
if (mode == PredictionMode.DcPred)
|
|
{
|
|
DcPred[leftAvailable][upAvailable][(int)txSize](dst, dstStride, constAboveRow, leftCol);
|
|
}
|
|
else
|
|
{
|
|
Pred[(int)mode][(int)txSize](dst, dstStride, constAboveRow, leftCol);
|
|
}
|
|
}
|
|
|
|
public static unsafe void PredictIntraBlock(
|
|
ref MacroBlockD xd,
|
|
int bwlIn,
|
|
TxSize txSize,
|
|
PredictionMode mode,
|
|
byte* refr,
|
|
int refStride,
|
|
byte* dst,
|
|
int dstStride,
|
|
int aoff,
|
|
int loff,
|
|
int plane)
|
|
{
|
|
int bw = 1 << bwlIn;
|
|
int txw = 1 << (int)txSize;
|
|
int haveTop = loff != 0 || !xd.AboveMi.IsNull ? 1 : 0;
|
|
int haveLeft = aoff != 0 || !xd.LeftMi.IsNull ? 1 : 0;
|
|
int haveRight = aoff + txw < bw ? 1 : 0;
|
|
int x = aoff * 4;
|
|
int y = loff * 4;
|
|
|
|
if (xd.CurBuf.HighBd)
|
|
{
|
|
BuildIntraPredictorsHigh(
|
|
ref xd,
|
|
refr,
|
|
refStride,
|
|
dst,
|
|
dstStride,
|
|
mode,
|
|
txSize,
|
|
haveTop,
|
|
haveLeft,
|
|
haveRight,
|
|
x,
|
|
y,
|
|
plane);
|
|
return;
|
|
}
|
|
|
|
BuildIntraPredictors(
|
|
ref xd,
|
|
refr,
|
|
refStride,
|
|
dst,
|
|
dstStride,
|
|
mode,
|
|
txSize,
|
|
haveTop,
|
|
haveLeft,
|
|
haveRight,
|
|
x,
|
|
y,
|
|
plane);
|
|
}
|
|
}
|
|
} |