ShuriZma
/
rpcs3
mirror of https://github.com/RPCS3/rpcs3.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

d3d12: Fragment program decompiler can decompile basic shader

This commit is contained in:
vlj 2015-05-14 20:57:33 +02:00 committed by Vincent Lejeune
parent bb643070bd
commit 9cb87552b8
3 changed files with 79 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
rpcs3/Emu/RSX/D3D12/D3D12PipelineState.cpp
View File

@ -109,11 +109,11 @@ ID3D12PipelineState *PipelineStateObjectCache::getGraphicPipelineState(
LOG_WARNING(RSX, "FP not found in buffer!"); LOG_WARNING(RSX, "FP not found in buffer!");
FragmentDecompiler FS(fragmentShader->addr, fragmentShader->size, fragmentShader->offset); FragmentDecompiler FS(fragmentShader->addr, fragmentShader->size, fragmentShader->offset);
const std::string &shader = FS.Decompile(); const std::string &shader = FS.Decompile();
m_fragment_prog.Compile("", SHADER_TYPE::SHADER_TYPE_FRAGMENT); m_fragment_prog.Compile(shader, SHADER_TYPE::SHADER_TYPE_FRAGMENT);
AddFragmentProgram(m_fragment_prog, *fragmentShader); AddFragmentProgram(m_fragment_prog, *fragmentShader);
// TODO: This shouldn't use current dir // TODO: This shouldn't use current dir
fs::file("./FragmentProgram.txt", o_write | o_create | o_trunc).write(shader.c_str(), shader.size()); fs::file("./FragmentProgram.hlsl", o_write | o_create | o_trunc).write(shader.c_str(), shader.size());
} }
if (!m_vp_buf_num) if (!m_vp_buf_num)
@ -125,7 +125,7 @@ ID3D12PipelineState *PipelineStateObjectCache::getGraphicPipelineState(
AddVertexProgram(m_vertex_prog, *vertexShader); AddVertexProgram(m_vertex_prog, *vertexShader);
// TODO: This shouldn't use current dir // TODO: This shouldn't use current dir
fs::file("./VertexProgram.txt", o_write | o_create | o_trunc).write(shaderCode.c_str(), shaderCode.size()); fs::file("./VertexProgram.hlsl", o_write | o_create | o_trunc).write(shaderCode.c_str(), shaderCode.size());
} }
if (m_fp_buf_num && m_vp_buf_num) if (m_fp_buf_num && m_vp_buf_num)
@ -262,18 +262,7 @@ ID3D12PipelineState *PipelineStateObjectCache::getGraphicPipelineState(
#define TO_STRING(x) #x #define TO_STRING(x) #x
void Shader::Compile(const std::string &code, SHADER_TYPE st) void Shader::Compile(const std::string &code, SHADER_TYPE st)
{ { HRESULT hr;
static const char FSstring[] = TO_STRING(
struct pixel {
float4 dst_reg0 : SV_POSITION;
float4 dst_reg1 : TEXCOORD0;
};
float4 main(pixel In) : SV_TARGET
{
return In.dst_reg1;
});
HRESULT hr;
Microsoft::WRL::ComPtr<ID3DBlob> errorBlob; Microsoft::WRL::ComPtr<ID3DBlob> errorBlob;
switch (st) switch (st)
{ {
@ -283,7 +272,7 @@ void Shader::Compile(const std::string &code, SHADER_TYPE st)
LOG_ERROR(RSX, "VS build failed:%s", errorBlob->GetBufferPointer()); LOG_ERROR(RSX, "VS build failed:%s", errorBlob->GetBufferPointer());
break; break;
case SHADER_TYPE::SHADER_TYPE_FRAGMENT: case SHADER_TYPE::SHADER_TYPE_FRAGMENT:
hr = D3DCompile(FSstring, sizeof(FSstring), "test", nullptr, nullptr, "main", "ps_5_0", 0, 0, &bytecode, errorBlob.GetAddressOf()); hr = D3DCompile(code.c_str(), code.size(), "test", nullptr, nullptr, "main", "ps_5_0", 0, 0, &bytecode, errorBlob.GetAddressOf());
if (hr != S_OK) if (hr != S_OK)
LOG_ERROR(RSX, "FS build failed:%s", errorBlob->GetBufferPointer()); LOG_ERROR(RSX, "FS build failed:%s", errorBlob->GetBufferPointer());
break; break;

90
rpcs3/Emu/RSX/D3D12/FragmentProgramDecompiler.cpp
View File

@ -6,6 +6,14 @@
#include "Emu/Memory/Memory.h" #include "Emu/Memory/Memory.h"
#include "Emu/System.h" #include "Emu/System.h"
static std::string typeName[] =
{
"float",
"float2",
"float3",
"float4"
};
FragmentDecompiler::FragmentDecompiler(u32 addr, u32& size, u32 ctrl) : FragmentDecompiler::FragmentDecompiler(u32 addr, u32& size, u32 ctrl) :
m_addr(addr), m_addr(addr),
m_size(size), m_size(size),
@ -48,7 +56,7 @@ void FragmentDecompiler::SetDst(std::string code, bool append_mask)
{ {
if (dst.set_cond) if (dst.set_cond)
{ {
AddCode("$ifcond " + m_parr.AddParam(PARAM_NONE, "vec4", "cc" + std::to_string(src0.cond_mod_reg_index)) + "$m = " + code + ";"); AddCode("$ifcond " + m_parr.AddParam(PARAM_NONE, typeName[3], "cc" + std::to_string(src0.cond_mod_reg_index)) + "$m = " + code + ";");
} }
else else
{ {
@ -65,7 +73,7 @@ void FragmentDecompiler::SetDst(std::string code, bool append_mask)
if (dst.set_cond) if (dst.set_cond)
{ {
AddCode(m_parr.AddParam(PARAM_NONE, "vec4", "cc" + std::to_string(src0.cond_mod_reg_index)) + "$m = " + dest + ";"); AddCode(m_parr.AddParam(PARAM_NONE, typeName[3], "cc" + std::to_string(src0.cond_mod_reg_index)) + "$m = " + dest + ";");
} }
} }
@ -93,24 +101,24 @@ std::string FragmentDecompiler::GetMask()
std::string FragmentDecompiler::AddReg(u32 index, int fp16) std::string FragmentDecompiler::AddReg(u32 index, int fp16)
{ {
return m_parr.AddParam(PARAM_NONE, "vec4", std::string(fp16 ? "h" : "r") + std::to_string(index), "vec4(0.0)"); return m_parr.AddParam(PARAM_NONE, typeName[3], std::string(fp16 ? "h" : "r") + std::to_string(index), typeName[3] + "(0.0)");
} }
bool FragmentDecompiler::HasReg(u32 index, int fp16) bool FragmentDecompiler::HasReg(u32 index, int fp16)
{ {
return m_parr.HasParam(PARAM_NONE, "vec4", return m_parr.HasParam(PARAM_NONE, typeName[3],
std::string(fp16 ? "h" : "r") + std::to_string(index)); std::string(fp16 ? "h" : "r") + std::to_string(index));
} }
std::string FragmentDecompiler::AddCond() std::string FragmentDecompiler::AddCond()
{ {
return m_parr.AddParam(PARAM_NONE, "vec4", "cc" + std::to_string(src0.cond_reg_index)); return m_parr.AddParam(PARAM_NONE, typeName[3], "cc" + std::to_string(src0.cond_reg_index));
} }
std::string FragmentDecompiler::AddConst() std::string FragmentDecompiler::AddConst()
{ {
std::string name = std::string("fc") + std::to_string(m_size + 4 * 4); std::string name = std::string("fc") + std::to_string(m_size + 4 * 4);
if (m_parr.HasParam(PARAM_UNIFORM, "vec4", name)) if (m_parr.HasParam(PARAM_UNIFORM, typeName[3], name))
{ {
return name; return name;
} }
@ -122,8 +130,8 @@ std::string FragmentDecompiler::AddConst()
u32 y = GetData(data[1]); u32 y = GetData(data[1]);
u32 z = GetData(data[2]); u32 z = GetData(data[2]);
u32 w = GetData(data[3]); u32 w = GetData(data[3]);
return m_parr.AddParam(PARAM_UNIFORM, "vec4", name, return m_parr.AddParam(PARAM_UNIFORM, typeName[3], name,
std::string("vec4(") + std::to_string((float&)x) + ", " + std::to_string((float&)y) std::string(typeName[3] + "(") + std::to_string((float&)x) + ", " + std::to_string((float&)y)
+ ", " + std::to_string((float&)z) + ", " + std::to_string((float&)w) + ")"); + ", " + std::to_string((float&)z) + ", " + std::to_string((float&)w) + ")");
} }
@ -201,7 +209,7 @@ std::string FragmentDecompiler::GetCond()
cond = "equal"; cond = "equal";
} }
return "any(" + cond + "(" + AddCond() + swizzle + ", vec4(0.0)))"; return "any(" + cond + "(" + AddCond() + swizzle + ", " + typeName[3] +"(0.0)))";
} }
void FragmentDecompiler::AddCodeCond(const std::string& dst, const std::string& src) void FragmentDecompiler::AddCodeCond(const std::string& dst, const std::string& src)
@ -252,7 +260,7 @@ void FragmentDecompiler::AddCodeCond(const std::string& dst, const std::string&
cond = "equal"; cond = "equal";
} }
cond = cond + "(" + AddCond() + swizzle + ", vec4(0.0))"; cond = cond + "(" + AddCond() + swizzle + ", " + typeName[3] + "(0.0))";
ShaderVar dst_var(dst); ShaderVar dst_var(dst);
dst_var.symplify(); dst_var.symplify();
@ -261,7 +269,7 @@ void FragmentDecompiler::AddCodeCond(const std::string& dst, const std::string&
if (dst_var.swizzles[0].length() == 1) if (dst_var.swizzles[0].length() == 1)
{ {
AddCode("if (" + cond + ".x) " + dst + " = vec4(" + src + ").x;"); AddCode("if (" + cond + ".x) " + dst + " = " + typeName[3] + "(" + src + ").x;");
} }
else else
{ {
@ -299,12 +307,12 @@ template<typename T> std::string FragmentDecompiler::GetSRC(T src)
default: default:
if (dst.src_attr_reg_num < sizeof(reg_table) / sizeof(reg_table[0])) if (dst.src_attr_reg_num < sizeof(reg_table) / sizeof(reg_table[0]))
{ {
ret += m_parr.AddParam(PARAM_IN, "vec4", reg_table[dst.src_attr_reg_num]); ret += m_parr.AddParam(PARAM_IN, typeName[3], reg_table[dst.src_attr_reg_num]);
} }
else else
{ {
LOG_ERROR(RSX, "Bad src reg num: %d", fmt::by_value(dst.src_attr_reg_num)); LOG_ERROR(RSX, "Bad src reg num: %d", fmt::by_value(dst.src_attr_reg_num));
ret += m_parr.AddParam(PARAM_IN, "vec4", "unk"); ret += m_parr.AddParam(PARAM_IN, typeName[3], "unk");
Emu.Pause(); Emu.Pause();
} }
break; break;
@ -351,22 +359,60 @@ std::string FragmentDecompiler::BuildCode()
for (int i = 0; i < sizeof(table) / sizeof(*table); ++i) for (int i = 0; i < sizeof(table) / sizeof(*table); ++i)
{ {
if (m_parr.HasParam(PARAM_NONE, "vec4", table[i].second)) if (m_parr.HasParam(PARAM_NONE, typeName[3], table[i].second))
AddCode(m_parr.AddParam(PARAM_OUT, "vec4", table[i].first, i) + " = " + table[i].second + ";"); AddCode(m_parr.AddParam(PARAM_OUT, typeName[3], table[i].first, i) + " = " + table[i].second + ";");
} }
if (m_ctrl & 0xe) main += m_ctrl & 0x40 ? "\tgl_FragDepth = r1.z;\n" : "\tgl_FragDepth = h2.z;\n"; if (m_ctrl & 0xe) main += m_ctrl & 0x40 ? "\tgl_FragDepth = r1.z;\n" : "\tgl_FragDepth = h2.z;\n";
std::string p; std::stringstream OS;
insertHeader(OS);
insertIntputs(OS);
insertMainStart(OS);
OS << main << std::endl;
insertMainEnd(OS);
for (auto& param : m_parr.params) { return OS.str();
// p += param.Format(); }
void FragmentDecompiler::insertHeader(std::stringstream & OS)
{
OS << "// Nothing" << std::endl;
}
void FragmentDecompiler::insertIntputs(std::stringstream & OS)
{
OS << "struct PSInput" << std::endl;
OS << "{" << std::endl;
OS << " float4 dst_reg0 : SV_POSITION;" << std::endl;
size_t index = 0;
for (ParamType PT : m_parr.params[PARAM_IN])
{
for (ParamItem PI : PT.items)
{
OS << " " << PT.type << " " << PI.name << " : TEXCOORD" << index << ";" << std::endl;
index++;
}
} }
OS << "};" << std::endl;
}
return std::string("#version 420\n" void FragmentDecompiler::insertMainStart(std::stringstream & OS)
"\n" {
+ p + "\n" OS << "float4 main(PSInput In) : SV_TARGET" << std::endl;
"void main()\n{\n" + main + "}\n"); OS << "{" << std::endl;
OS << " float4 r0;" << std::endl;
for (ParamType PT : m_parr.params[PARAM_IN])
{
for (ParamItem PI : PT.items)
OS << " " << PT.type << " " << PI.name << " = In." << PI.name << ";" << std::endl;
}
}
void FragmentDecompiler::insertMainEnd(std::stringstream & OS)
{
OS << " return r0;" << std::endl;
OS << "}" << std::endl;
} }
std::string FragmentDecompiler::Decompile() std::string FragmentDecompiler::Decompile()

6
rpcs3/Emu/RSX/D3D12/FragmentProgramDecompiler.h
View File

@ -2,6 +2,7 @@
#if defined(DX12_SUPPORT) #if defined(DX12_SUPPORT)
#include "ShaderParam.h" #include "ShaderParam.h"
#include "Emu/RSX/RSXFragmentProgram.h" #include "Emu/RSX/RSXFragmentProgram.h"
#include <sstream>
class FragmentDecompiler class FragmentDecompiler
{ {
@ -35,6 +36,11 @@ class FragmentDecompiler
std::string BuildCode(); std::string BuildCode();
u32 GetData(const u32 d) const { return d << 16 | d >> 16; } u32 GetData(const u32 d) const { return d << 16 | d >> 16; }
protected:
virtual void insertHeader(std::stringstream &OS);
virtual void insertIntputs(std::stringstream &OS);
virtual void insertMainStart(std::stringstream &OS);
virtual void insertMainEnd(std::stringstream &OS);
public: public:
FragmentDecompiler(u32 addr, u32& size, u32 ctrl); FragmentDecompiler(u32 addr, u32& size, u32 ctrl);
std::string Decompile(); std::string Decompile();