d3d12: Use the compiler vertex program

It works with the primitive sample.
This commit is contained in:
vlj 2015-05-14 19:48:49 +02:00 committed by Vincent Lejeune
parent 96c4e87bf5
commit 3960555e45
3 changed files with 46 additions and 62 deletions

View File

@ -108,7 +108,7 @@ ID3D12PipelineState *PipelineStateObjectCache::getGraphicPipelineState(
{ {
LOG_WARNING(RSX, "FP not found in buffer!"); LOG_WARNING(RSX, "FP not found in buffer!");
// Decompile(*fragmentShader); // Decompile(*fragmentShader);
m_fragment_prog.Compile(SHADER_TYPE::SHADER_TYPE_FRAGMENT); m_fragment_prog.Compile("", 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
@ -120,7 +120,7 @@ ID3D12PipelineState *PipelineStateObjectCache::getGraphicPipelineState(
LOG_WARNING(RSX, "VP not found in buffer!"); LOG_WARNING(RSX, "VP not found in buffer!");
VertexDecompiler VS(vertexShader->data); VertexDecompiler VS(vertexShader->data);
std::string shaderCode = VS.Decompile(); std::string shaderCode = VS.Decompile();
m_vertex_prog.Compile(SHADER_TYPE::SHADER_TYPE_VERTEX); m_vertex_prog.Compile(shaderCode, SHADER_TYPE::SHADER_TYPE_VERTEX);
AddVertexProgram(m_vertex_prog, *vertexShader); AddVertexProgram(m_vertex_prog, *vertexShader);
// TODO: This shouldn't use current dir // TODO: This shouldn't use current dir
@ -260,51 +260,16 @@ ID3D12PipelineState *PipelineStateObjectCache::getGraphicPipelineState(
#define TO_STRING(x) #x #define TO_STRING(x) #x
void Shader::Compile(SHADER_TYPE st) void Shader::Compile(const std::string &code, SHADER_TYPE st)
{ {
static const char VSstring[] = TO_STRING(
cbuffer SCALE_OFFSET : register(b0)
{
float4x4 scaleOffsetMat;
};
cbuffer CONSTANT : register(b1)
{
float4 vc[468];
};
struct vertex {
float4 pos : TEXCOORD0;
float4 color : TEXCOORD3;
};
struct pixel {
float4 pos : SV_POSITION;
float4 color : TEXCOORD0;
};
pixel main(vertex In)
{
pixel Out;
float4 pos = In.pos;
pos.w = dot(pos, vc[259]);
pos.z = dot(pos, vc[258]);
pos.y = dot(pos, vc[257]);
pos.x = dot(pos, vc[256]);
pos.z = -pos.z;
Out.pos = mul(pos, scaleOffsetMat);
Out.color = In.color;
return Out;
});
static const char FSstring[] = TO_STRING( static const char FSstring[] = TO_STRING(
struct pixel { struct pixel {
float4 pos : SV_POSITION; float4 dst_reg0 : SV_POSITION;
float4 color : TEXCOORD0; float4 dst_reg1 : TEXCOORD0;
}; };
float4 main(pixel In) : SV_TARGET float4 main(pixel In) : SV_TARGET
{ {
return In.color; return In.dst_reg1;
}); });
HRESULT hr; HRESULT hr;
@ -312,7 +277,7 @@ void Shader::Compile(SHADER_TYPE st)
switch (st) switch (st)
{ {
case SHADER_TYPE::SHADER_TYPE_VERTEX: case SHADER_TYPE::SHADER_TYPE_VERTEX:
hr = D3DCompile(VSstring, sizeof(VSstring), "test", nullptr, nullptr, "main", "vs_5_0", 0, 0, &bytecode, errorBlob.GetAddressOf()); hr = D3DCompile(code.c_str(), code.size(), "test", nullptr, nullptr, "main", "vs_5_0", 0, 0, &bytecode, errorBlob.GetAddressOf());
if (hr != S_OK) if (hr != S_OK)
LOG_ERROR(RSX, "VS build failed:%s", errorBlob->GetBufferPointer()); LOG_ERROR(RSX, "VS build failed:%s", errorBlob->GetBufferPointer());
break; break;

View File

@ -37,7 +37,7 @@ public:
// void Decompile(RSXFragmentProgram& prog) // void Decompile(RSXFragmentProgram& prog)
/** Compile the decompiled fragment shader into a format we can use with OpenGL. */ /** Compile the decompiled fragment shader into a format we can use with OpenGL. */
void Compile(SHADER_TYPE st); void Compile(const std::string &code, SHADER_TYPE st);
}; };
// Based on // Based on

View File

@ -5,6 +5,13 @@
#include "Utilities/Log.h" #include "Utilities/Log.h"
#include "Emu/System.h" #include "Emu/System.h"
static std::string typeName[] =
{
"float",
"float2",
"float3",
"float4"
};
std::string VertexDecompiler::GetMask(bool is_sca) std::string VertexDecompiler::GetMask(bool is_sca)
{ {
@ -45,13 +52,13 @@ std::string VertexDecompiler::GetDST(bool isSca)
switch (isSca ? 0x1f : d3.dst) switch (isSca ? 0x1f : d3.dst)
{ {
case 0x1f: case 0x1f:
ret += m_parr.AddParam(PARAM_NONE, "vec4", std::string("tmp") + std::to_string(isSca ? d3.sca_dst_tmp : d0.dst_tmp)); ret += m_parr.AddParam(PARAM_NONE, typeName[3], std::string("tmp") + std::to_string(isSca ? d3.sca_dst_tmp : d0.dst_tmp));
break; break;
default: default:
if (d3.dst > 15) if (d3.dst > 15)
LOG_ERROR(RSX, fmt::Format("dst index out of range: %u", d3.dst)); LOG_ERROR(RSX, fmt::Format("dst index out of range: %u", d3.dst));
ret += m_parr.AddParam(PARAM_NONE, "vec4", std::string("dst_reg") + std::to_string(d3.dst), d3.dst == 0 ? "vec4(0.0f, 0.0f, 0.0f, 1.0f)" : "vec4(0.0)"); ret += m_parr.AddParam(PARAM_NONE, typeName[3], std::string("dst_reg") + std::to_string(d3.dst), d3.dst == 0 ? typeName[3] + "(0.0f, 0.0f, 0.0f, 1.0f)" : typeName[3] + "(0.0, 0.0, 0.0, 0.0)");
break; break;
} }
@ -75,21 +82,21 @@ std::string VertexDecompiler::GetSRC(const u32 n)
switch (src[n].reg_type) switch (src[n].reg_type)
{ {
case 1: //temp case 1: //temp
ret += m_parr.AddParam(PARAM_NONE, "vec4", "tmp" + std::to_string(src[n].tmp_src)); ret += m_parr.AddParam(PARAM_NONE, typeName[3], "tmp" + std::to_string(src[n].tmp_src));
break; break;
case 2: //input case 2: //input
if (d1.input_src < (sizeof(reg_table) / sizeof(reg_table[0]))) if (d1.input_src < (sizeof(reg_table) / sizeof(reg_table[0])))
{ {
ret += m_parr.AddParam(PARAM_IN, "vec4", reg_table[d1.input_src], d1.input_src); ret += m_parr.AddParam(PARAM_IN, typeName[3], reg_table[d1.input_src], d1.input_src);
} }
else else
{ {
LOG_ERROR(RSX, "Bad input src num: %d", fmt::by_value(d1.input_src)); LOG_ERROR(RSX, "Bad input src num: %d", fmt::by_value(d1.input_src));
ret += m_parr.AddParam(PARAM_IN, "vec4", "in_unk", d1.input_src); ret += m_parr.AddParam(PARAM_IN, typeName[3], "in_unk", d1.input_src);
} }
break; break;
case 3: //const case 3: //const
m_parr.AddParam(PARAM_UNIFORM, "vec4", std::string("vc[468]")); m_parr.AddParam(PARAM_UNIFORM, typeName[3], std::string("vc[468]"));
ret += std::string("vc[") + std::to_string(d1.const_src) + (d3.index_const ? " + " + AddAddrReg() : "") + "]"; ret += std::string("vc[") + std::to_string(d1.const_src) + (d3.index_const ? " + " + AddAddrReg() : "") + "]";
break; break;
@ -154,7 +161,7 @@ void VertexDecompiler::SetDST(bool is_sca, std::string value)
if (d0.cond_update_enable_0 && d0.cond_update_enable_1) if (d0.cond_update_enable_0 && d0.cond_update_enable_1)
{ {
dest = m_parr.AddParam(PARAM_NONE, "vec4", "cc" + std::to_string(d0.cond_reg_sel_1), "vec4(0.0)") + mask; dest = m_parr.AddParam(PARAM_NONE, typeName[3], "cc" + std::to_string(d0.cond_reg_sel_1), typeName[3] + "(0.0)") + mask;
} }
else if (d3.dst != 0x1f || (is_sca ? d3.sca_dst_tmp != 0x3f : d0.dst_tmp != 0x3f)) else if (d3.dst != 0x1f || (is_sca ? d3.sca_dst_tmp != 0x3f : d0.dst_tmp != 0x3f))
{ {
@ -312,7 +319,7 @@ void VertexDecompiler::AddCodeCond(const std::string& dst, const std::string& sr
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
{ {
@ -437,7 +444,7 @@ std::string VertexDecompiler::BuildCode()
insertInputs(OS, m_parr.params[PARAM_IN]); insertInputs(OS, m_parr.params[PARAM_IN]);
OS << std::endl; OS << std::endl;
insertOutputs(OS, m_parr.params[PARAM_OUT]); insertOutputs(OS, m_parr.params[PARAM_NONE]);
OS << std::endl; OS << std::endl;
insertConstants(OS, m_parr.params[PARAM_UNIFORM]); insertConstants(OS, m_parr.params[PARAM_UNIFORM]);
OS << std::endl; OS << std::endl;
@ -471,7 +478,7 @@ void VertexDecompiler::insertInputs(std::stringstream & OS, const std::vector<Pa
void VertexDecompiler::insertConstants(std::stringstream & OS, const std::vector<ParamType> & constants) void VertexDecompiler::insertConstants(std::stringstream & OS, const std::vector<ParamType> & constants)
{ {
OS << "cbuffer CONSTANT_BUFFER" << std::endl; OS << "cbuffer CONSTANT_BUFFER : register(b1)" << std::endl;
OS << "{" << std::endl; OS << "{" << std::endl;
for (const ParamType PT : constants) for (const ParamType PT : constants)
{ {
@ -485,11 +492,15 @@ void VertexDecompiler::insertOutputs(std::stringstream & OS, const std::vector<P
{ {
OS << "struct PixelInput" << std::endl; OS << "struct PixelInput" << std::endl;
OS << "{" << std::endl; OS << "{" << std::endl;
OS << " float4 position : SV_POSITION;" << std::endl; OS << " float4 dst_reg0 : SV_POSITION;" << std::endl;
size_t outputIndex = 0;
for (const ParamType PT : outputs) for (const ParamType PT : outputs)
{ {
for (const ParamItem &PI : PT.items) for (const ParamItem &PI : PT.items)
OS << " " << PT.type << " " << PI.name << ": TEXCOORD" << PI.location << ";" << std::endl; {
if (PI.name == "dst_reg0") continue;
OS << " " << PT.type << " " << PI.name << ": TEXCOORD" << outputIndex++ << ";" << std::endl;
}
} }
OS << "};" << std::endl; OS << "};" << std::endl;
} }
@ -532,14 +543,20 @@ static const reg_info reg_table[] =
void VertexDecompiler::insertMainStart(std::stringstream & OS) void VertexDecompiler::insertMainStart(std::stringstream & OS)
{ {
OS << "PixelInput main(VertexInput)" << std::endl; OS << "PixelInput main(VertexInput In)" << std::endl;
OS << "{" << std::endl; OS << "{" << std::endl;
// Declare inside main function // Declare inside main function
for (auto &i : reg_table) for (auto &i : reg_table)
{ {
if (m_parr.HasParam(PARAM_NONE, "vec4", i.src_reg)) if (m_parr.HasParam(PARAM_NONE, typeName[3], i.src_reg))
OS << " vec4 " << i.src_reg << ";" << std::endl; OS << " float4 " << i.src_reg << ";" << std::endl;
}
for (const ParamType PT : m_parr.params[PARAM_IN])
{
for (const ParamItem &PI : PT.items)
OS << " " << PT.type << " " << PI.name << " = In." << PI.name << ";" << std::endl;
} }
} }
@ -550,10 +567,12 @@ void VertexDecompiler::insertMainEnd(std::stringstream & OS)
// Declare inside main function // Declare inside main function
for (auto &i : reg_table) for (auto &i : reg_table)
{ {
if (m_parr.HasParam(PARAM_NONE, "vec4", i.src_reg)) if (m_parr.HasParam(PARAM_NONE, typeName[3], i.src_reg))
OS << " Out." << i.src_reg << " = " << i.src_reg << ";" << std::endl; OS << " Out." << i.src_reg << " = " << i.src_reg << ";" << std::endl;
} }
OS << " Out.position = mul(dst_reg0, scaleOffsetMat);" << std::endl; // TODO: Find why I need to do this
OS << " Out.dst_reg0.z *= -1.;" << std::endl;
OS << " Out.dst_reg0 = mul(Out.dst_reg0, scaleOffsetMat);" << std::endl;
OS << " return Out;" << std::endl; OS << " return Out;" << std::endl;
OS << "}" << std::endl; OS << "}" << std::endl;
} }
@ -688,7 +707,7 @@ std::string VertexDecompiler::Decompile()
case RSX_SCA_OPCODE_RSQ: SetDSTSca("inversesqrt(abs($s))"); break; case RSX_SCA_OPCODE_RSQ: SetDSTSca("inversesqrt(abs($s))"); break;
case RSX_SCA_OPCODE_EXP: SetDSTSca("exp($s)"); break; case RSX_SCA_OPCODE_EXP: SetDSTSca("exp($s)"); break;
case RSX_SCA_OPCODE_LOG: SetDSTSca("log($s)"); break; case RSX_SCA_OPCODE_LOG: SetDSTSca("log($s)"); break;
case RSX_SCA_OPCODE_LIT: SetDSTSca("vec4(1.0, $s.x, ($s.x > 0.0 ? exp($s.w * log2($s.y)) : 0.0), 1.0)"); break; case RSX_SCA_OPCODE_LIT: SetDSTSca(typeName[3] + "(1.0, $s.x, ($s.x > 0.0 ? exp($s.w * log2($s.y)) : 0.0), 1.0)"); break;
case RSX_SCA_OPCODE_BRA: case RSX_SCA_OPCODE_BRA:
{ {
AddCode("$if ($cond)"); AddCode("$if ($cond)");
@ -782,7 +801,7 @@ std::string VertexDecompiler::Decompile()
case RSX_VEC_OPCODE_MAD: SetDSTVec("($0 * $1 + $2)"); break; case RSX_VEC_OPCODE_MAD: SetDSTVec("($0 * $1 + $2)"); break;
case RSX_VEC_OPCODE_DP3: SetDSTVec("vec4(dot($0.xyz, $1.xyz))"); break; case RSX_VEC_OPCODE_DP3: SetDSTVec("vec4(dot($0.xyz, $1.xyz))"); break;
case RSX_VEC_OPCODE_DPH: SetDSTVec("vec4(dot(vec4($0.xyz, 1.0), $1))"); break; case RSX_VEC_OPCODE_DPH: SetDSTVec("vec4(dot(vec4($0.xyz, 1.0), $1))"); break;
case RSX_VEC_OPCODE_DP4: SetDSTVec("vec4(dot($0, $1))"); break; case RSX_VEC_OPCODE_DP4: SetDSTVec(typeName[3] + "(dot($0, $1), dot($0, $1), dot($0, $1), dot($0, $1))"); break;
case RSX_VEC_OPCODE_DST: SetDSTVec("vec4(distance($0, $1))"); break; case RSX_VEC_OPCODE_DST: SetDSTVec("vec4(distance($0, $1))"); break;
case RSX_VEC_OPCODE_MIN: SetDSTVec("min($0, $1)"); break; case RSX_VEC_OPCODE_MIN: SetDSTVec("min($0, $1)"); break;
case RSX_VEC_OPCODE_MAX: SetDSTVec("max($0, $1)"); break; case RSX_VEC_OPCODE_MAX: SetDSTVec("max($0, $1)"); break;