mirror of
https://github.com/RPCS3/rpcs3.git
synced 2025-03-12 13:13:43 +00:00
d3d12: Fragment program decompiler can decompile basic shader
This commit is contained in:
parent
bb643070bd
commit
9cb87552b8
@ -109,11 +109,11 @@ ID3D12PipelineState *PipelineStateObjectCache::getGraphicPipelineState(
|
||||
LOG_WARNING(RSX, "FP not found in buffer!");
|
||||
FragmentDecompiler FS(fragmentShader->addr, fragmentShader->size, fragmentShader->offset);
|
||||
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);
|
||||
|
||||
// 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)
|
||||
@ -125,7 +125,7 @@ ID3D12PipelineState *PipelineStateObjectCache::getGraphicPipelineState(
|
||||
AddVertexProgram(m_vertex_prog, *vertexShader);
|
||||
|
||||
// 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)
|
||||
@ -262,18 +262,7 @@ ID3D12PipelineState *PipelineStateObjectCache::getGraphicPipelineState(
|
||||
#define TO_STRING(x) #x
|
||||
|
||||
void Shader::Compile(const std::string &code, SHADER_TYPE st)
|
||||
{
|
||||
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;
|
||||
{ HRESULT hr;
|
||||
Microsoft::WRL::ComPtr<ID3DBlob> errorBlob;
|
||||
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());
|
||||
break;
|
||||
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)
|
||||
LOG_ERROR(RSX, "FS build failed:%s", errorBlob->GetBufferPointer());
|
||||
break;
|
||||
|
@ -6,6 +6,14 @@
|
||||
#include "Emu/Memory/Memory.h"
|
||||
#include "Emu/System.h"
|
||||
|
||||
static std::string typeName[] =
|
||||
{
|
||||
"float",
|
||||
"float2",
|
||||
"float3",
|
||||
"float4"
|
||||
};
|
||||
|
||||
FragmentDecompiler::FragmentDecompiler(u32 addr, u32& size, u32 ctrl) :
|
||||
m_addr(addr),
|
||||
m_size(size),
|
||||
@ -48,7 +56,7 @@ void FragmentDecompiler::SetDst(std::string code, bool append_mask)
|
||||
{
|
||||
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
|
||||
{
|
||||
@ -65,7 +73,7 @@ void FragmentDecompiler::SetDst(std::string code, bool append_mask)
|
||||
|
||||
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)
|
||||
{
|
||||
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)
|
||||
{
|
||||
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 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 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;
|
||||
}
|
||||
@ -122,8 +130,8 @@ std::string FragmentDecompiler::AddConst()
|
||||
u32 y = GetData(data[1]);
|
||||
u32 z = GetData(data[2]);
|
||||
u32 w = GetData(data[3]);
|
||||
return m_parr.AddParam(PARAM_UNIFORM, "vec4", name,
|
||||
std::string("vec4(") + std::to_string((float&)x) + ", " + std::to_string((float&)y)
|
||||
return m_parr.AddParam(PARAM_UNIFORM, typeName[3], name,
|
||||
std::string(typeName[3] + "(") + std::to_string((float&)x) + ", " + std::to_string((float&)y)
|
||||
+ ", " + std::to_string((float&)z) + ", " + std::to_string((float&)w) + ")");
|
||||
}
|
||||
|
||||
@ -201,7 +209,7 @@ std::string FragmentDecompiler::GetCond()
|
||||
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)
|
||||
@ -252,7 +260,7 @@ void FragmentDecompiler::AddCodeCond(const std::string& dst, const std::string&
|
||||
cond = "equal";
|
||||
}
|
||||
|
||||
cond = cond + "(" + AddCond() + swizzle + ", vec4(0.0))";
|
||||
cond = cond + "(" + AddCond() + swizzle + ", " + typeName[3] + "(0.0))";
|
||||
|
||||
ShaderVar dst_var(dst);
|
||||
dst_var.symplify();
|
||||
@ -261,7 +269,7 @@ void FragmentDecompiler::AddCodeCond(const std::string& dst, const std::string&
|
||||
|
||||
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
|
||||
{
|
||||
@ -299,12 +307,12 @@ template<typename T> std::string FragmentDecompiler::GetSRC(T src)
|
||||
default:
|
||||
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
|
||||
{
|
||||
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();
|
||||
}
|
||||
break;
|
||||
@ -351,22 +359,60 @@ std::string FragmentDecompiler::BuildCode()
|
||||
|
||||
for (int i = 0; i < sizeof(table) / sizeof(*table); ++i)
|
||||
{
|
||||
if (m_parr.HasParam(PARAM_NONE, "vec4", table[i].second))
|
||||
AddCode(m_parr.AddParam(PARAM_OUT, "vec4", table[i].first, i) + " = " + table[i].second + ";");
|
||||
if (m_parr.HasParam(PARAM_NONE, typeName[3], 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";
|
||||
|
||||
std::string p;
|
||||
std::stringstream OS;
|
||||
insertHeader(OS);
|
||||
insertIntputs(OS);
|
||||
insertMainStart(OS);
|
||||
OS << main << std::endl;
|
||||
insertMainEnd(OS);
|
||||
|
||||
for (auto& param : m_parr.params) {
|
||||
// p += param.Format();
|
||||
return OS.str();
|
||||
}
|
||||
|
||||
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"
|
||||
"\n"
|
||||
+ p + "\n"
|
||||
"void main()\n{\n" + main + "}\n");
|
||||
void FragmentDecompiler::insertMainStart(std::stringstream & OS)
|
||||
{
|
||||
OS << "float4 main(PSInput In) : SV_TARGET" << std::endl;
|
||||
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()
|
||||
|
@ -2,6 +2,7 @@
|
||||
#if defined(DX12_SUPPORT)
|
||||
#include "ShaderParam.h"
|
||||
#include "Emu/RSX/RSXFragmentProgram.h"
|
||||
#include <sstream>
|
||||
|
||||
class FragmentDecompiler
|
||||
{
|
||||
@ -35,6 +36,11 @@ class FragmentDecompiler
|
||||
std::string BuildCode();
|
||||
|
||||
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:
|
||||
FragmentDecompiler(u32 addr, u32& size, u32 ctrl);
|
||||
std::string Decompile();
|
||||
|
Loading…
x
Reference in New Issue
Block a user