Ryujinx-git/Ryujinx.Graphics.Gpu/Engine/Compute.cs

141 lines
5.2 KiB
C#
Raw Normal View History

2019-12-29 17:41:50 +00:00
using Ryujinx.Graphics.GAL;
using Ryujinx.Graphics.Gpu.Image;
using Ryujinx.Graphics.Gpu.Shader;
2019-10-13 06:02:07 +00:00
using Ryujinx.Graphics.Gpu.State;
using Ryujinx.Graphics.Shader;
using System;
using System.Runtime.InteropServices;
namespace Ryujinx.Graphics.Gpu.Engine
{
partial class Methods
{
/// <summary>
/// Dispatches compute work.
/// </summary>
/// <param name="state">Current GPU state</param>
/// <param name="argument">Method call argument</param>
2019-11-22 02:46:14 +00:00
public void Dispatch(GpuState state, int argument)
2019-10-13 06:02:07 +00:00
{
2019-11-22 02:46:14 +00:00
uint dispatchParamsAddress = (uint)state.Get<int>(MethodOffset.DispatchParamsAddress);
2019-10-13 06:02:07 +00:00
var dispatchParams = _context.MemoryAccessor.Read<ComputeParams>((ulong)dispatchParamsAddress << 8);
2019-11-22 02:46:14 +00:00
GpuVa shaderBaseAddress = state.Get<GpuVa>(MethodOffset.ShaderBaseAddress);
2019-10-13 06:02:07 +00:00
ulong shaderGpuVa = shaderBaseAddress.Pack() + (uint)dispatchParams.ShaderOffset;
// Note: A size of 0 is also invalid, the size must be at least 1.
int sharedMemorySize = Math.Clamp(dispatchParams.SharedMemorySize & 0xffff, 1, _context.Capabilities.MaximumComputeSharedMemorySize);
2019-12-31 22:09:49 +00:00
ComputeShader cs = ShaderCache.GetComputeShader(
2019-10-13 06:02:07 +00:00
shaderGpuVa,
sharedMemorySize,
2019-10-13 06:02:07 +00:00
dispatchParams.UnpackBlockSizeX(),
dispatchParams.UnpackBlockSizeY(),
dispatchParams.UnpackBlockSizeZ());
2019-12-29 17:41:50 +00:00
_context.Renderer.Pipeline.SetProgram(cs.HostProgram);
2019-11-22 02:46:14 +00:00
var samplerPool = state.Get<PoolState>(MethodOffset.SamplerPoolState);
2019-12-29 17:41:50 +00:00
TextureManager.SetComputeSamplerPool(samplerPool.Address.Pack(), samplerPool.MaximumId, dispatchParams.SamplerIndex);
2019-11-22 02:46:14 +00:00
var texturePool = state.Get<PoolState>(MethodOffset.TexturePoolState);
2019-12-29 17:41:50 +00:00
TextureManager.SetComputeTexturePool(texturePool.Address.Pack(), texturePool.MaximumId);
2019-12-29 17:41:50 +00:00
TextureManager.SetComputeTextureBufferIndex(state.Get<int>(MethodOffset.TextureBufferIndex));
2019-10-13 06:02:07 +00:00
ShaderProgramInfo info = cs.Shader.Program.Info;
2019-10-13 06:02:07 +00:00
uint sbEnableMask = 0;
uint ubEnableMask = dispatchParams.UnpackUniformBuffersEnableMask();
for (int index = 0; index < dispatchParams.UniformBuffers.Length; index++)
{
if ((ubEnableMask & (1 << index)) == 0)
{
continue;
}
ulong gpuVa = dispatchParams.UniformBuffers[index].PackAddress();
ulong size = dispatchParams.UniformBuffers[index].UnpackSize();
2019-12-29 17:41:50 +00:00
BufferManager.SetComputeUniformBuffer(index, gpuVa, size);
2019-10-13 06:02:07 +00:00
}
for (int index = 0; index < info.SBuffers.Count; index++)
{
BufferDescriptor sb = info.SBuffers[index];
sbEnableMask |= 1u << sb.Slot;
2019-12-29 17:41:50 +00:00
ulong sbDescAddress = BufferManager.GetComputeUniformBufferAddress(0);
2019-10-13 06:02:07 +00:00
int sbDescOffset = 0x310 + sb.Slot * 0x10;
sbDescAddress += (ulong)sbDescOffset;
ReadOnlySpan<byte> sbDescriptorData = _context.PhysicalMemory.GetSpan(sbDescAddress, 0x10);
2019-10-13 06:02:07 +00:00
SbDescriptor sbDescriptor = MemoryMarshal.Cast<byte, SbDescriptor>(sbDescriptorData)[0];
2019-12-29 17:41:50 +00:00
BufferManager.SetComputeStorageBuffer(sb.Slot, sbDescriptor.PackAddress(), (uint)sbDescriptor.Size);
2019-10-13 06:02:07 +00:00
}
ubEnableMask = 0;
for (int index = 0; index < info.CBuffers.Count; index++)
{
ubEnableMask |= 1u << info.CBuffers[index].Slot;
}
2019-12-29 17:41:50 +00:00
BufferManager.SetComputeStorageBufferEnableMask(sbEnableMask);
BufferManager.SetComputeUniformBufferEnableMask(ubEnableMask);
2019-10-13 06:02:07 +00:00
var textureBindings = new TextureBindingInfo[info.Textures.Count];
for (int index = 0; index < info.Textures.Count; index++)
{
var descriptor = info.Textures[index];
Target target = GetTarget(descriptor.Type);
if (descriptor.IsBindless)
{
textureBindings[index] = new TextureBindingInfo(target, descriptor.CbufOffset, descriptor.CbufSlot);
}
else
{
textureBindings[index] = new TextureBindingInfo(target, descriptor.HandleIndex);
}
}
2019-12-29 17:41:50 +00:00
TextureManager.SetComputeTextures(textureBindings);
var imageBindings = new TextureBindingInfo[info.Images.Count];
for (int index = 0; index < info.Images.Count; index++)
{
var descriptor = info.Images[index];
Target target = GetTarget(descriptor.Type);
imageBindings[index] = new TextureBindingInfo(target, descriptor.HandleIndex);
}
2019-12-29 17:41:50 +00:00
TextureManager.SetComputeImages(imageBindings);
2019-12-29 17:41:50 +00:00
BufferManager.CommitComputeBindings();
TextureManager.CommitComputeBindings();
2019-10-13 06:02:07 +00:00
2019-12-29 17:41:50 +00:00
_context.Renderer.Pipeline.DispatchCompute(
2019-10-13 06:02:07 +00:00
dispatchParams.UnpackGridSizeX(),
dispatchParams.UnpackGridSizeY(),
dispatchParams.UnpackGridSizeZ());
2019-11-22 02:46:14 +00:00
UpdateShaderState(state);
2019-10-13 06:02:07 +00:00
}
}
}