From 03e088a4f44af1212da0c7c23f95293a6e129a35 Mon Sep 17 00:00:00 2001
From: ReinUsesLisp <reinuseslisp@airmail.cc>
Date: Sat, 22 Dec 2018 01:20:57 -0300
Subject: shader_ir: Fixup TEX and TEXS and partially fix TLD4 decompiling

---
 src/video_core/shader/glsl_decompiler.cpp | 29 ++++++++++++++++++++---------
 1 file changed, 20 insertions(+), 9 deletions(-)

(limited to 'src/video_core/shader/glsl_decompiler.cpp')

diff --git a/src/video_core/shader/glsl_decompiler.cpp b/src/video_core/shader/glsl_decompiler.cpp
index 3ca3fae6d..a513c0c4b 100644
--- a/src/video_core/shader/glsl_decompiler.cpp
+++ b/src/video_core/shader/glsl_decompiler.cpp
@@ -635,8 +635,10 @@ private:
                                                          result_type));
     }
 
+#pragma optimize("", off)
+
     std::string GenerateTexture(Operation operation, const std::string& func,
-                                const std::string& extra_cast = "") {
+                                std::string extra_cast(std::string) = nullptr) {
         constexpr std::array<const char*, 4> coord_constructors = {"float", "vec2", "vec3", "vec4"};
 
         const auto& meta = std::get<MetaTexture>(operation.GetMeta());
@@ -651,15 +653,17 @@ private:
         expr += '(';
         for (u32 i = 0; i < count; ++i) {
             const bool is_extra = i >= meta.coords_count;
-            const bool do_cast = is_extra && !extra_cast.empty();
-            if (do_cast) {
-                expr += extra_cast;
-                expr += '(';
+            const bool is_array = i == meta.array_index;
+
+            std::string operand = Visit(operation[i]);
+            if (is_extra && extra_cast != nullptr) {
+                operand = extra_cast(operand);
             }
-            expr += Visit(operation[i]);
-            if (do_cast) {
-                expr += ')';
+            if (is_array) {
+                ASSERT(!is_extra);
+                operand = "float(ftoi(" + operand + "))";
             }
+            expr += operand;
             if (i + 1 == meta.coords_count) {
                 expr += ')';
             }
@@ -1065,7 +1069,14 @@ private:
     }
 
     std::string F4TextureGather(Operation operation) {
-        return GenerateTexture(operation, "textureGather", "int");
+        const bool is_shadow = std::get<MetaTexture>(operation.GetMeta()).sampler.IsShadow();
+        if (is_shadow) {
+            return GenerateTexture(operation, "textureGather",
+                                   [](std::string ref_z) { return ref_z; });
+        } else {
+            return GenerateTexture(operation, "textureGather",
+                                   [](std::string comp) { return "ftoi(" + comp + ')'; });
+        }
     }
 
     std::string F4TextureQueryDimensions(Operation operation) {
-- 
cgit v1.2.3