From 0cb83d4668f6d23b00c9cb7df545e505bed9e37d Mon Sep 17 00:00:00 2001
From: Samuliak <samuliak77@gmail.com>
Date: Mon, 29 Jul 2024 20:04:41 +0200
Subject: [PATCH] fix: support buffer data & fix: depth

---
 src/Cafe/HW/Latte/Core/LatteShader.cpp        |  6 ++++--
 .../LatteDecompilerEmitMSL.cpp                | 19 +++++++++++++------
 2 files changed, 17 insertions(+), 8 deletions(-)

diff --git a/src/Cafe/HW/Latte/Core/LatteShader.cpp b/src/Cafe/HW/Latte/Core/LatteShader.cpp
index 6561e642..486516ef 100644
--- a/src/Cafe/HW/Latte/Core/LatteShader.cpp
+++ b/src/Cafe/HW/Latte/Core/LatteShader.cpp
@@ -615,7 +615,8 @@ LatteDecompilerShader* LatteShader_CreateShaderFromDecompilerOutput(LatteDecompi
 	LatteDecompilerShader* shader = decompilerOutput.shader;
 	shader->baseHash = baseHash;
 	// copy resource mapping
-	if(g_renderer->GetType() == RendererAPI::Vulkan)
+	// HACK
+	if (g_renderer->GetType() != RendererAPI::OpenGL)
 		shader->resourceMapping = decompilerOutput.resourceMappingVK;
 	else
 		shader->resourceMapping = decompilerOutput.resourceMappingGL;
@@ -626,7 +627,8 @@ LatteDecompilerShader* LatteShader_CreateShaderFromDecompilerOutput(LatteDecompi
 	shader->hasStreamoutBufferWrite = decompilerOutput.streamoutBufferWriteMask.any();
 	// copy uniform offsets
 	// for OpenGL these are retrieved in _prepareSeparableUniforms()
-	if (g_renderer->GetType() == RendererAPI::Vulkan)
+	// HACK
+	if (g_renderer->GetType() != RendererAPI::OpenGL)
 	{
 		shader->uniform.loc_remapped = decompilerOutput.uniformOffsetsVK.offset_remapped;
 		shader->uniform.loc_uniformRegister = decompilerOutput.uniformOffsetsVK.offset_uniformRegister;
diff --git a/src/Cafe/HW/Latte/LegacyShaderDecompiler/LatteDecompilerEmitMSL.cpp b/src/Cafe/HW/Latte/LegacyShaderDecompiler/LatteDecompilerEmitMSL.cpp
index 320d8e24..21587437 100644
--- a/src/Cafe/HW/Latte/LegacyShaderDecompiler/LatteDecompilerEmitMSL.cpp
+++ b/src/Cafe/HW/Latte/LegacyShaderDecompiler/LatteDecompilerEmitMSL.cpp
@@ -1531,7 +1531,7 @@ static void _emitALUOP2InstructionCode(LatteDecompilerShaderContext* shaderConte
 			debugBreakpoint();
 		_emitOperandInputCode(shaderContext, aluInstruction, 1, LATTE_DECOMPILER_DTYPE_SIGNED_INT);
 		src->add(")");
-		src->add(") discard;");
+		src->add(") discard_fragment();");
 		src->add(_CRLF);
 	}
 	else if( aluInstruction->opcode == ALU_OP2_INST_KILLGT ||
@@ -1551,7 +1551,7 @@ static void _emitALUOP2InstructionCode(LatteDecompilerShaderContext* shaderConte
 			debugBreakpoint();
 		_emitOperandInputCode(shaderContext, aluInstruction, 1, LATTE_DECOMPILER_DTYPE_FLOAT);
 		src->add(")");
-		src->add(") discard;");
+		src->add(") discard_fragment();");
 		src->add(_CRLF);
 	}
 	else
@@ -3136,7 +3136,7 @@ static void _emitExportCode(LatteDecompilerShaderContext* shaderContext, LatteDe
 				if( pixelColorOutputIndex == 0 && alphaTestEnable && alphaTestFunc == Latte::E_COMPAREFUNC::NEVER )
 				{
 					// never pass alpha test
-					src->add("discard;" _CRLF);
+					src->add("discard_fragment();" _CRLF);
 				}
 				else if( pixelColorOutputIndex == 0 && alphaTestEnable && alphaTestFunc != Latte::E_COMPAREFUNC::ALWAYS)
 				{
@@ -3166,7 +3166,7 @@ static void _emitExportCode(LatteDecompilerShaderContext* shaderContext, LatteDe
 						break;
 					}
 					src->add(" supportBuffer.alphaTestRef");
-					src->add(") == false) discard;" _CRLF);
+					src->add(") == false) discard_fragment();" _CRLF);
 				}
 				// pixel color output
 				src->addFmt("out.passPixelColor{} = ", pixelColorOutputIndex);
@@ -4067,14 +4067,21 @@ void LatteDecompiler_emitMSLShader(LatteDecompilerShaderContext* shaderContext,
 	}
 	for(auto& cfInstruction : shaderContext->cfInstructions)
 		LatteDecompiler_emitClauseCodeMSL(shaderContext, &cfInstruction, false);
-	if( shader->shaderType == LatteConst::ShaderType::Geometry )
-		src->add("EndPrimitive();" _CRLF);
+	//if(shader->shaderType == LatteConst::ShaderType::Geometry)
+	//	src->add("EndPrimitive();" _CRLF);
 	// vertex shader should write renderstate point size at the end if required but not modified by shader
 	if (shaderContext->analyzer.outputPointSize && shaderContext->analyzer.writesPointSize == false)
 	{
 		if (shader->shaderType == LatteConst::ShaderType::Vertex && shaderContext->options->usesGeometryShader == false)
 			src->add("out.pointSize = supportBuffer.pointSize;" _CRLF);
 	}
+	// HACK: this should be handled outside of the shader, because clipping currently wouldn't work
+	if (shader->shaderType == LatteConst::ShaderType::Vertex)
+	{
+	    // Convert depth from the range of [-1, 1] to [0, 1]
+		src->add("out.position /= out.position.w;" _CRLF);
+		src->add("out.position.z = out.position.z * 0.5 + 0.5;" _CRLF);
+	}
 	// return
 	src->add("return out;" _CRLF);
 	// end of shader main