mirror of
https://github.com/Ryujinx/Ryujinx.git
synced 2024-08-19 23:28:42 +02:00
* first try * second try * working update * Final impl * Fixes nits * Fix everything * remove leftover * Update FFmpegContext.cs * Update Surface.cs * Addresses gdkchan feedback * bool not byte * Addresses gdkchan feedback
84 lines
2.5 KiB
C#
84 lines
2.5 KiB
C#
using Ryujinx.Common.Logging;
|
|
using Ryujinx.Graphics.Device;
|
|
using Ryujinx.Graphics.Gpu.Memory;
|
|
using Ryujinx.Graphics.Nvdec.Image;
|
|
using System.Collections.Concurrent;
|
|
using System.Collections.Generic;
|
|
using System.Threading;
|
|
|
|
namespace Ryujinx.Graphics.Nvdec
|
|
{
|
|
public class NvdecDevice : IDeviceStateWithContext
|
|
{
|
|
private readonly ResourceManager _rm;
|
|
private readonly DeviceState<NvdecRegisters> _state;
|
|
|
|
private long _currentId;
|
|
private ConcurrentDictionary<long, NvdecDecoderContext> _contexts;
|
|
private NvdecDecoderContext _currentContext;
|
|
|
|
public NvdecDevice(MemoryManager gmm)
|
|
{
|
|
_rm = new ResourceManager(gmm, new SurfaceCache(gmm));
|
|
_state = new DeviceState<NvdecRegisters>(new Dictionary<string, RwCallback>
|
|
{
|
|
{ nameof(NvdecRegisters.Execute), new RwCallback(Execute, null) }
|
|
});
|
|
_contexts = new ConcurrentDictionary<long, NvdecDecoderContext>();
|
|
}
|
|
|
|
public long CreateContext()
|
|
{
|
|
long id = Interlocked.Increment(ref _currentId);
|
|
_contexts.TryAdd(id, new NvdecDecoderContext());
|
|
|
|
return id;
|
|
}
|
|
|
|
public void DestroyContext(long id)
|
|
{
|
|
if (_contexts.TryRemove(id, out var context))
|
|
{
|
|
context.Dispose();
|
|
}
|
|
|
|
_rm.Cache.Trim();
|
|
}
|
|
|
|
public void BindContext(long id)
|
|
{
|
|
if (_contexts.TryGetValue(id, out var context))
|
|
{
|
|
_currentContext = context;
|
|
}
|
|
}
|
|
|
|
public int Read(int offset) => _state.Read(offset);
|
|
public void Write(int offset, int data) => _state.Write(offset, data);
|
|
|
|
private void Execute(int data)
|
|
{
|
|
Decode((CodecId)_state.State.SetCodecID);
|
|
}
|
|
|
|
private void Decode(CodecId codecId)
|
|
{
|
|
switch (codecId)
|
|
{
|
|
case CodecId.H264:
|
|
H264Decoder.Decode(_currentContext, _rm, ref _state.State);
|
|
break;
|
|
case CodecId.Vp8:
|
|
Vp8Decoder.Decode(_currentContext, _rm, ref _state.State);
|
|
break;
|
|
case CodecId.Vp9:
|
|
Vp9Decoder.Decode(_rm, ref _state.State);
|
|
break;
|
|
default:
|
|
Logger.Error?.Print(LogClass.Nvdec, $"Unsupported codec \"{codecId}\".");
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|