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
84 lines
2.6 KiB
C#
84 lines
2.6 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((ApplicationId)_state.State.SetApplicationId);
|
|
}
|
|
|
|
private void Decode(ApplicationId applicationId)
|
|
{
|
|
switch (applicationId)
|
|
{
|
|
case ApplicationId.H264:
|
|
H264Decoder.Decode(_currentContext, _rm, ref _state.State);
|
|
break;
|
|
case ApplicationId.Vp8:
|
|
Vp8Decoder.Decode(_currentContext, _rm, ref _state.State);
|
|
break;
|
|
case ApplicationId.Vp9:
|
|
Vp9Decoder.Decode(_rm, ref _state.State);
|
|
break;
|
|
default:
|
|
Logger.Error?.Print(LogClass.Nvdec, $"Unsupported codec \"{applicationId}\".");
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|