using Ryujinx.Common.Logging; using Ryujinx.Graphics.GAL; using Ryujinx.Graphics.Shader; namespace Ryujinx.Graphics.Gpu.Shader { /// <summary> /// Represents a GPU state and memory accessor. /// </summary> class GpuAccessor : TextureDescriptorCapableGpuAccessor, IGpuAccessor { private readonly GpuChannel _channel; private readonly GpuAccessorState _state; private readonly int _stageIndex; private readonly bool _compute; private readonly int _localSizeX; private readonly int _localSizeY; private readonly int _localSizeZ; private readonly int _localMemorySize; private readonly int _sharedMemorySize; public int Cb1DataSize { get; private set; } /// <summary> /// Creates a new instance of the GPU state accessor for graphics shader translation. /// </summary> /// <param name="context">GPU context</param> /// <param name="channel">GPU channel</param> /// <param name="state">Current GPU state</param> /// <param name="stageIndex">Graphics shader stage index (0 = Vertex, 4 = Fragment)</param> public GpuAccessor(GpuContext context, GpuChannel channel, GpuAccessorState state, int stageIndex) : base(context) { _channel = channel; _state = state; _stageIndex = stageIndex; } /// <summary> /// Creates a new instance of the GPU state accessor for compute shader translation. /// </summary> /// <param name="context">GPU context</param> /// <param name="channel">GPU channel</param> /// <param name="state">Current GPU state</param> /// <param name="localSizeX">Local group size X of the compute shader</param> /// <param name="localSizeY">Local group size Y of the compute shader</param> /// <param name="localSizeZ">Local group size Z of the compute shader</param> /// <param name="localMemorySize">Local memory size of the compute shader</param> /// <param name="sharedMemorySize">Shared memory size of the compute shader</param> public GpuAccessor( GpuContext context, GpuChannel channel, GpuAccessorState state, int localSizeX, int localSizeY, int localSizeZ, int localMemorySize, int sharedMemorySize) : base(context) { _channel = channel; _state = state; _compute = true; _localSizeX = localSizeX; _localSizeY = localSizeY; _localSizeZ = localSizeZ; _localMemorySize = localMemorySize; _sharedMemorySize = sharedMemorySize; } /// <summary> /// Reads data from the constant buffer 1. /// </summary> /// <param name="offset">Offset in bytes to read from</param> /// <returns>Value at the given offset</returns> public uint ConstantBuffer1Read(int offset) { if (Cb1DataSize < offset + 4) { Cb1DataSize = offset + 4; } ulong baseAddress = _compute ? _channel.BufferManager.GetComputeUniformBufferAddress(1) : _channel.BufferManager.GetGraphicsUniformBufferAddress(_stageIndex, 1); return _channel.MemoryManager.Physical.Read<uint>(baseAddress + (ulong)offset); } /// <summary> /// Prints a log message. /// </summary> /// <param name="message">Message to print</param> public void Log(string message) { Logger.Warning?.Print(LogClass.Gpu, $"Shader translator: {message}"); } /// <summary> /// Reads data from GPU memory. /// </summary> /// <typeparam name="T">Type of the data to be read</typeparam> /// <param name="address">GPU virtual address of the data</param> /// <returns>Data at the memory location</returns> public override T MemoryRead<T>(ulong address) { return _channel.MemoryManager.Read<T>(address); } /// <summary> /// Checks if a given memory address is mapped. /// </summary> /// <param name="address">GPU virtual address to be checked</param> /// <returns>True if the address is mapped, false otherwise</returns> public bool MemoryMapped(ulong address) { return _channel.MemoryManager.IsMapped(address); } /// <summary> /// Queries Local Size X for compute shaders. /// </summary> /// <returns>Local Size X</returns> public int QueryComputeLocalSizeX() => _localSizeX; /// <summary> /// Queries Local Size Y for compute shaders. /// </summary> /// <returns>Local Size Y</returns> public int QueryComputeLocalSizeY() => _localSizeY; /// <summary> /// Queries Local Size Z for compute shaders. /// </summary> /// <returns>Local Size Z</returns> public int QueryComputeLocalSizeZ() => _localSizeZ; /// <summary> /// Queries Local Memory size in bytes for compute shaders. /// </summary> /// <returns>Local Memory size in bytes</returns> public int QueryComputeLocalMemorySize() => _localMemorySize; /// <summary> /// Queries Shared Memory size in bytes for compute shaders. /// </summary> /// <returns>Shared Memory size in bytes</returns> public int QueryComputeSharedMemorySize() => _sharedMemorySize; /// <summary> /// Queries Constant Buffer usage information. /// </summary> /// <returns>A mask where each bit set indicates a bound constant buffer</returns> public uint QueryConstantBufferUse() { return _compute ? _channel.BufferManager.GetComputeUniformBufferUseMask() : _channel.BufferManager.GetGraphicsUniformBufferUseMask(_stageIndex); } /// <summary> /// Queries current primitive topology for geometry shaders. /// </summary> /// <returns>Current primitive topology</returns> public InputTopology QueryPrimitiveTopology() { return _state.Topology switch { PrimitiveTopology.Points => InputTopology.Points, PrimitiveTopology.Lines or PrimitiveTopology.LineLoop or PrimitiveTopology.LineStrip => InputTopology.Lines, PrimitiveTopology.LinesAdjacency or PrimitiveTopology.LineStripAdjacency => InputTopology.LinesAdjacency, PrimitiveTopology.Triangles or PrimitiveTopology.TriangleStrip or PrimitiveTopology.TriangleFan => InputTopology.Triangles, PrimitiveTopology.TrianglesAdjacency or PrimitiveTopology.TriangleStripAdjacency => InputTopology.TrianglesAdjacency, _ => InputTopology.Points, }; } /// <summary> /// Gets the texture descriptor for a given texture on the pool. /// </summary> /// <param name="handle">Index of the texture (this is the word offset of the handle in the constant buffer)</param> /// <param name="cbufSlot">Constant buffer slot for the texture handle</param> /// <returns>Texture descriptor</returns> public override Image.ITextureDescriptor GetTextureDescriptor(int handle, int cbufSlot) { if (_compute) { return _channel.TextureManager.GetComputeTextureDescriptor( _state.TexturePoolGpuVa, _state.TextureBufferIndex, _state.TexturePoolMaximumId, handle, cbufSlot); } else { return _channel.TextureManager.GetGraphicsTextureDescriptor( _state.TexturePoolGpuVa, _state.TextureBufferIndex, _state.TexturePoolMaximumId, _stageIndex, handle, cbufSlot); } } /// <summary> /// Queries if host state forces early depth testing. /// </summary> /// <returns>True if early depth testing is forced</returns> public bool QueryEarlyZForce() { return _state.EarlyZForce; } } }