mirror of
https://github.com/Ryujinx/Ryujinx.git
synced 2024-08-19 21:52:22 +00:00
* misc: Uses official names for NVDEC registers * Address gdkchan's comment * Address comments
91 lines
3.9 KiB
C#
91 lines
3.9 KiB
C#
using Ryujinx.Common;
|
|
using Ryujinx.Graphics.Gpu.Memory;
|
|
using Ryujinx.Graphics.Nvdec.Image;
|
|
using Ryujinx.Graphics.Nvdec.Types.Vp9;
|
|
using Ryujinx.Graphics.Nvdec.Vp9;
|
|
using Ryujinx.Graphics.Video;
|
|
using System;
|
|
using System.Runtime.CompilerServices;
|
|
using System.Runtime.InteropServices;
|
|
using static Ryujinx.Graphics.Nvdec.MemoryExtensions;
|
|
|
|
namespace Ryujinx.Graphics.Nvdec
|
|
{
|
|
static class Vp9Decoder
|
|
{
|
|
private static Decoder _decoder = new Decoder();
|
|
|
|
public unsafe static void Decode(ResourceManager rm, ref NvdecRegisters state)
|
|
{
|
|
PictureInfo pictureInfo = rm.Gmm.DeviceRead<PictureInfo>(state.SetDrvPicSetupOffset);
|
|
EntropyProbs entropy = rm.Gmm.DeviceRead<EntropyProbs>(state.Vp9SetProbTabBufOffset);
|
|
|
|
ISurface Rent(uint lumaOffset, uint chromaOffset, FrameSize size)
|
|
{
|
|
return rm.Cache.Get(_decoder, lumaOffset, chromaOffset, size.Width, size.Height);
|
|
}
|
|
|
|
ISurface lastSurface = Rent(state.SetPictureLumaOffset[0], state.SetPictureChromaOffset[0], pictureInfo.LastFrameSize);
|
|
ISurface goldenSurface = Rent(state.SetPictureLumaOffset[1], state.SetPictureChromaOffset[1], pictureInfo.GoldenFrameSize);
|
|
ISurface altSurface = Rent(state.SetPictureLumaOffset[2], state.SetPictureChromaOffset[2], pictureInfo.AltFrameSize);
|
|
ISurface currentSurface = Rent(state.SetPictureLumaOffset[3], state.SetPictureChromaOffset[3], pictureInfo.CurrentFrameSize);
|
|
|
|
Vp9PictureInfo info = pictureInfo.Convert();
|
|
|
|
info.LastReference = lastSurface;
|
|
info.GoldenReference = goldenSurface;
|
|
info.AltReference = altSurface;
|
|
|
|
entropy.Convert(ref info.Entropy);
|
|
|
|
ReadOnlySpan<byte> bitstream = rm.Gmm.DeviceGetSpan(state.SetInBufBaseOffset, (int)pictureInfo.BitstreamSize);
|
|
|
|
ReadOnlySpan<Vp9MvRef> mvsIn = ReadOnlySpan<Vp9MvRef>.Empty;
|
|
|
|
if (info.UsePrevInFindMvRefs)
|
|
{
|
|
mvsIn = GetMvsInput(rm.Gmm, pictureInfo.CurrentFrameSize, state.Vp9SetColMvReadBufOffset);
|
|
}
|
|
|
|
int miCols = BitUtils.DivRoundUp(pictureInfo.CurrentFrameSize.Width, 8);
|
|
int miRows = BitUtils.DivRoundUp(pictureInfo.CurrentFrameSize.Height, 8);
|
|
|
|
using var mvsRegion = rm.Gmm.GetWritableRegion(ExtendOffset(state.Vp9SetColMvWriteBufOffset), miRows * miCols * 16);
|
|
|
|
Span<Vp9MvRef> mvsOut = MemoryMarshal.Cast<byte, Vp9MvRef>(mvsRegion.Memory.Span);
|
|
|
|
uint lumaOffset = state.SetPictureLumaOffset[3];
|
|
uint chromaOffset = state.SetPictureChromaOffset[3];
|
|
|
|
if (_decoder.Decode(ref info, currentSurface, bitstream, mvsIn, mvsOut))
|
|
{
|
|
SurfaceWriter.Write(rm.Gmm, currentSurface, lumaOffset, chromaOffset);
|
|
}
|
|
|
|
WriteBackwardUpdates(rm.Gmm, state.Vp9SetCtxCounterBufOffset, ref info.BackwardUpdateCounts);
|
|
|
|
rm.Cache.Put(lastSurface);
|
|
rm.Cache.Put(goldenSurface);
|
|
rm.Cache.Put(altSurface);
|
|
rm.Cache.Put(currentSurface);
|
|
}
|
|
|
|
private static ReadOnlySpan<Vp9MvRef> GetMvsInput(MemoryManager gmm, FrameSize size, uint offset)
|
|
{
|
|
int miCols = BitUtils.DivRoundUp(size.Width, 8);
|
|
int miRows = BitUtils.DivRoundUp(size.Height, 8);
|
|
|
|
return MemoryMarshal.Cast<byte, Vp9MvRef>(gmm.DeviceGetSpan(offset, miRows * miCols * 16));
|
|
}
|
|
|
|
private static void WriteBackwardUpdates(MemoryManager gmm, uint offset, ref Vp9BackwardUpdates counts)
|
|
{
|
|
using var backwardUpdatesRegion = gmm.GetWritableRegion(ExtendOffset(offset), Unsafe.SizeOf<BackwardUpdates>());
|
|
|
|
ref var backwardUpdates = ref MemoryMarshal.Cast<byte, BackwardUpdates>(backwardUpdatesRegion.Memory.Span)[0];
|
|
|
|
backwardUpdates = new BackwardUpdates(ref counts);
|
|
}
|
|
}
|
|
}
|