From 20bfe9ee05cf341b7e4f880ef690bd155ecedda6 Mon Sep 17 00:00:00 2001
From: Alyssa Rosenzweig <alyssa.rosenzweig@collabora.com>
Date: Tue, 8 Dec 2020 08:39:20 -0500
Subject: [PATCH] pan/bi: Implement load_vary via the builder

Signed-off-by: Alyssa Rosenzweig <alyssa.rosenzweig@collabora.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/8135>
---
 src/panfrost/bifrost/bifrost_compile.c | 54 ++++++++++++++++++++++++++
 1 file changed, 54 insertions(+)

diff --git a/src/panfrost/bifrost/bifrost_compile.c b/src/panfrost/bifrost/bifrost_compile.c
index 80e77461c44..cd9f9c748f7 100644
--- a/src/panfrost/bifrost/bifrost_compile.c
+++ b/src/panfrost/bifrost/bifrost_compile.c
@@ -228,6 +228,60 @@ bi_is_intr_immediate(nir_intrinsic_instr *instr, unsigned *immediate)
         return (*immediate) < 20;
 }
 
+static void
+bi_emit_load_vary(bi_builder *b, nir_intrinsic_instr *instr)
+{
+        enum bi_sample sample = BI_SAMPLE_CENTER;
+        enum bi_update update = BI_UPDATE_STORE;
+        enum bi_register_format regfmt = BI_REGISTER_FORMAT_AUTO;
+        enum bi_vecsize vecsize = instr->num_components - 1;
+        bool smooth = instr->intrinsic == nir_intrinsic_load_interpolated_input;
+
+        if (smooth) {
+                nir_intrinsic_instr *parent = nir_src_as_intrinsic(instr->src[0]);
+                assert(parent);
+
+                sample = bi_interp_for_intrinsic(parent->intrinsic);
+        } else {
+                regfmt = bi_reg_fmt_for_nir(nir_intrinsic_dest_type(instr));
+        }
+
+        /* Ignored for non-conditional center and retrieve modes (use an
+         * efficient encoding), otherwise R61 for sample mask XXX RA */
+
+        bi_index src0 = (sample == BI_SAMPLE_CENTER) ? bi_passthrough(BIFROST_SRC_FAU_HI) :
+                bi_register(61);
+
+        nir_src *offset = nir_get_io_offset_src(instr);
+        unsigned imm_index = 0;
+        bool immediate = bi_is_intr_immediate(instr, &imm_index);
+
+        if (immediate && smooth) {
+                bi_ld_var_imm_to(b, bi_dest_index(&instr->dest),
+                                src0, regfmt, sample, update, vecsize,
+                                imm_index);
+        } else if (immediate && !smooth) {
+                bi_ld_var_flat_imm_to(b, bi_dest_index(&instr->dest),
+                                BI_FUNCTION_NONE, regfmt, vecsize, imm_index);
+        } else {
+                bi_index idx = bi_src_index(offset);
+                unsigned base = nir_intrinsic_base(instr);
+
+                if (base != 0)
+                        idx = bi_iadd_u32(b, idx, bi_imm_u32(base), false);
+
+                if (smooth) {
+                        bi_ld_var_to(b, bi_dest_index(&instr->dest),
+                                        src0, idx, regfmt, sample, update,
+                                        vecsize);
+                } else {
+                        bi_ld_var_flat_to(b, bi_dest_index(&instr->dest),
+                                        idx, BI_FUNCTION_NONE, regfmt,
+                                        vecsize);
+                }
+        }
+}
+
 static void
 bi_emit_ld_vary(bi_context *ctx, nir_intrinsic_instr *instr)
 {