mirror of
https://github.com/Ryujinx/Ryujinx.git
synced 2024-08-17 06:08:43 +02:00
131 lines
4 KiB
C#
131 lines
4 KiB
C#
using Ryujinx.Graphics.Shader.Translation;
|
|
using System;
|
|
using System.IO;
|
|
|
|
namespace Ryujinx.Graphics.Gpu.Shader
|
|
{
|
|
/// <summary>
|
|
/// Shader dumper, writes binary shader code to disk.
|
|
/// </summary>
|
|
class ShaderDumper
|
|
{
|
|
private string _runtimeDir;
|
|
private string _dumpPath;
|
|
private int _dumpIndex;
|
|
|
|
public int CurrentDumpIndex => _dumpIndex;
|
|
|
|
public ShaderDumper()
|
|
{
|
|
_dumpIndex = 1;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Dumps shader code to disk.
|
|
/// </summary>
|
|
/// <param name="code">Code to be dumped</param>
|
|
/// <param name="compute">True for compute shader code, false for graphics shader code</param>
|
|
/// <param name="fullPath">Output path for the shader code with header included</param>
|
|
/// <param name="codePath">Output path for the shader code without header</param>
|
|
public void Dump(ReadOnlySpan<byte> code, bool compute, out string fullPath, out string codePath)
|
|
{
|
|
_dumpPath = GraphicsConfig.ShadersDumpPath;
|
|
|
|
if (string.IsNullOrWhiteSpace(_dumpPath))
|
|
{
|
|
fullPath = null;
|
|
codePath = null;
|
|
|
|
return;
|
|
}
|
|
|
|
string fileName = "Shader" + _dumpIndex.ToString("d4") + ".bin";
|
|
|
|
fullPath = Path.Combine(FullDir(), fileName);
|
|
codePath = Path.Combine(CodeDir(), fileName);
|
|
|
|
_dumpIndex++;
|
|
|
|
code = Translator.ExtractCode(code, compute, out int headerSize);
|
|
|
|
using (MemoryStream stream = new MemoryStream(code.ToArray()))
|
|
{
|
|
BinaryReader codeReader = new BinaryReader(stream);
|
|
|
|
using (FileStream fullFile = File.Create(fullPath))
|
|
using (FileStream codeFile = File.Create(codePath))
|
|
{
|
|
BinaryWriter fullWriter = new BinaryWriter(fullFile);
|
|
BinaryWriter codeWriter = new BinaryWriter(codeFile);
|
|
|
|
fullWriter.Write(codeReader.ReadBytes(headerSize));
|
|
|
|
byte[] temp = codeReader.ReadBytes(code.Length - headerSize);
|
|
|
|
fullWriter.Write(temp);
|
|
codeWriter.Write(temp);
|
|
|
|
// Align to meet nvdisasm requirements.
|
|
while (codeFile.Length % 0x20 != 0)
|
|
{
|
|
codeWriter.Write(0);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns the output directory for shader code with header.
|
|
/// </summary>
|
|
/// <returns>Directory path</returns>
|
|
private string FullDir()
|
|
{
|
|
return CreateAndReturn(Path.Combine(DumpDir(), "Full"));
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns the output directory for shader code without header.
|
|
/// </summary>
|
|
/// <returns>Directory path</returns>
|
|
private string CodeDir()
|
|
{
|
|
return CreateAndReturn(Path.Combine(DumpDir(), "Code"));
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns the full output directory for the current shader dump.
|
|
/// </summary>
|
|
/// <returns>Directory path</returns>
|
|
private string DumpDir()
|
|
{
|
|
if (string.IsNullOrEmpty(_runtimeDir))
|
|
{
|
|
int index = 1;
|
|
|
|
do
|
|
{
|
|
_runtimeDir = Path.Combine(_dumpPath, "Dumps" + index.ToString("d2"));
|
|
|
|
index++;
|
|
}
|
|
while (Directory.Exists(_runtimeDir));
|
|
|
|
Directory.CreateDirectory(_runtimeDir);
|
|
}
|
|
|
|
return _runtimeDir;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Creates a new specified directory if needed.
|
|
/// </summary>
|
|
/// <param name="dir">The directory to create</param>
|
|
/// <returns>The same directory passed to the method</returns>
|
|
private static string CreateAndReturn(string dir)
|
|
{
|
|
Directory.CreateDirectory(dir);
|
|
|
|
return dir;
|
|
}
|
|
}
|
|
} |