nouveau: add a nouveau_compiler binary to compile TGSI into shader ISA

This makes it easy to compare output between different cards, especially
for ones that you don't have (and/or not in the current machine).

Signed-off-by: Ilia Mirkin <[email protected]>

3 files changed, 235 insertions, 0 deletions

diff --git a/src/gallium/drivers/nouveau/.gitignore b/src/gallium/drivers/nouveau/.gitignore
new file mode 100644
index 00000000000..829f951a050
--- /dev/null
+++ b/src/gallium/drivers/nouveau/.gitignore
@@ -0,0 +1 @@
+nouveau_compiler
diff --git a/src/gallium/drivers/nouveau/Makefile.am b/src/gallium/drivers/nouveau/Makefile.am
index 7c052231bfa..ac4f9bbcbf3 100644
--- a/src/gallium/drivers/nouveau/Makefile.am
+++ b/src/gallium/drivers/nouveau/Makefile.am
@@ -39,3 +39,15 @@ libnouveau_la_SOURCES = \
 	$(NV50_C_SOURCES) \
 	$(NVC0_CODEGEN_SOURCES) \
 	$(NVC0_C_SOURCES)
+
+noinst_PROGRAMS = nouveau_compiler
+
+nouveau_compiler_SOURCES = \
+	nouveau_compiler.c
+
+nouveau_compiler_LDADD = \
+	libnouveau.la \
+	../../auxiliary/libgallium.la \
+	-lstdc++ \
+	-lm \
+	-ldl
diff --git a/src/gallium/drivers/nouveau/nouveau_compiler.c b/src/gallium/drivers/nouveau/nouveau_compiler.c
new file mode 100644
index 00000000000..5f1e35a0d24
--- /dev/null
+++ b/src/gallium/drivers/nouveau/nouveau_compiler.c
@@ -0,0 +1,222 @@
+/*
+ * Copyright 2014 Ilia Mirkin
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a
+ * copy of this software and associated documentation files (the "Software"),
+ * to deal in the Software without restriction, including without limitation
+ * the rights to use, copy, modify, merge, publish, distribute, sublicense,
+ * and/or sell copies of the Software, and to permit persons to whom the
+ * Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in
+ * all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
+ * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR
+ * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
+ * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
+ * OTHER DEALINGS IN THE SOFTWARE.
+ */
+
+#include <errno.h>
+
+#include "tgsi/tgsi_text.h"
+#include "util/u_debug.h"
+
+#include "codegen/nv50_ir_driver.h"
+#include "nv50/nv50_context.h"
+
+/* these headers weren't really meant to be included together */
+#undef SB_DATA
+
+#include "nv30/nv30_state.h"
+#include "nv30/nvfx_shader.h"
+
+static int
+nv30_fp(int chipset, struct tgsi_token tokens[],
+        unsigned *size, unsigned **code) {
+   struct nv30_fragprog fp;
+   memset(&fp, 0, sizeof(fp));
+   fp.pipe.tokens = tokens;
+   tgsi_scan_shader(fp.pipe.tokens, &fp.info);
+   _nvfx_fragprog_translate(chipset >= 0x40 ? 0x4097 : 0x3097, &fp);
+   *size = fp.insn_len * 4;
+   *code = fp.insn;
+   return !fp.translated;
+}
+
+static int
+nv30_vp(int chipset, struct tgsi_token tokens[],
+        unsigned *size, unsigned **code) {
+   struct nv30_vertprog vp;
+   memset(&vp, 0, sizeof(vp));
+
+   vp.pipe.tokens = tokens;
+   tgsi_scan_shader(vp.pipe.tokens, &vp.info);
+   _nvfx_vertprog_translate(chipset >= 0x40 ? 0x4097 : 0x3097, &vp);
+   *size = vp.nr_insns * 16;
+   *code = (unsigned *)vp.insns;
+   return !vp.translated;
+}
+
+static int
+nv30_codegen(int chipset, int type, struct tgsi_token tokens[],
+             unsigned *size, unsigned **code) {
+   switch (type) {
+      case PIPE_SHADER_FRAGMENT:
+         return nv30_fp(chipset, tokens, size, code);
+      case PIPE_SHADER_VERTEX:
+         return nv30_vp(chipset, tokens, size, code);
+   }
+   _debug_printf("Unexpected shader type: %d\n", type);
+   return 1;
+}
+
+static int
+dummy_assign_slots(struct nv50_ir_prog_info *info)
+{
+   unsigned i, n, c;
+
+   n = 0;
+   for (i = 0; i < info->numInputs; ++i) {
+      for (c = 0; c < 4; ++c)
+         if (info->in[i].mask & (1 << c))
+            info->in[i].slot[c] = n++;
+   }
+
+   /* VertexID before InstanceID */
+   if (info->io.vertexId < info->numSysVals)
+      info->sv[info->io.vertexId].slot[0] = n++;
+   if (info->io.instanceId < info->numSysVals)
+      info->sv[info->io.instanceId].slot[0] = n++;
+
+   n = 0;
+   for (i = 0; i < info->numOutputs; ++i) {
+      for (c = 0; c < 4; ++c)
+         if (info->out[i].mask & (1 << c))
+            info->out[i].slot[c] = n++;
+   }
+   return 0;
+}
+
+static int
+nouveau_codegen(int chipset, int type, struct tgsi_token tokens[],
+                unsigned *size, unsigned **code) {
+   struct nv50_ir_prog_info info = {0};
+   int ret;
+
+   info.type = type;
+   info.target = chipset;
+   info.bin.sourceRep = NV50_PROGRAM_IR_TGSI;
+   info.bin.source = tokens;
+
+   info.io.ucpCBSlot = 15;
+   info.io.ucpBase = NV50_CB_AUX_UCP_OFFSET;
+
+   info.io.resInfoCBSlot = 15;
+   info.io.suInfoBase = NV50_CB_AUX_TEX_MS_OFFSET;
+   info.io.msInfoCBSlot = 15;
+   info.io.msInfoBase = NV50_CB_AUX_MS_OFFSET;
+
+   info.assignSlots = dummy_assign_slots;
+
+   info.optLevel = debug_get_num_option("NV50_PROG_OPTIMIZE", 3);
+   info.dbgFlags = debug_get_num_option("NV50_PROG_DEBUG", 0);
+
+   ret = nv50_ir_generate_code(&info);
+   if (ret) {
+      _debug_printf("Error compiling program: %d\n", ret);
+      return ret;
+   }
+
+   *size = info.bin.codeSize;
+   *code = info.bin.code;
+   return 0;
+}
+
+int
+main(int argc, char *argv[])
+{
+   struct tgsi_token tokens[1024];
+   int i, chipset = 0, type = -1;
+   const char *filename = NULL;
+   FILE *f;
+   char text[65536] = {0};
+   unsigned size, *code;
+
+   for (i = 1; i < argc; i++) {
+      if (!strcmp(argv[i], "-a"))
+         chipset = strtol(argv[++i], NULL, 16);
+      else
+         filename = argv[i];
+   }
+
+   if (!chipset) {
+      _debug_printf("Must specify a chipset (-a)\n");
+      return 1;
+   }
+
+   if (!filename) {
+      _debug_printf("Must specify a filename\n");
+      return 1;
+   }
+
+   if (!strcmp(filename, "-"))
+      f = stdin;
+   else
+      f = fopen(filename, "r");
+
+   if (f == NULL) {
+      _debug_printf("Error opening file '%s': %s\n", filename, strerror(errno));
+      return 1;
+   }
+
+   fread(text, 1, sizeof(text), f);
+   if (ferror(f)) {
+      _debug_printf("Error reading file '%s'\n", filename);
+      return 1;
+   }
+   fclose(f);
+
+   _debug_printf("Compiling for NV%X\n", chipset);
+
+   if (!strncmp(text, "FRAG", 4))
+      type = PIPE_SHADER_FRAGMENT;
+   else if (!strncmp(text, "VERT", 4))
+      type = PIPE_SHADER_VERTEX;
+   else if (!strncmp(text, "GEOM", 4))
+      type = PIPE_SHADER_GEOMETRY;
+   else if (!strncmp(text, "COMP", 4))
+      type = PIPE_SHADER_COMPUTE;
+   else {
+      _debug_printf("Unrecognized TGSI header\n");
+      return 1;
+   }
+
+   if (!tgsi_text_translate(text, tokens, Elements(tokens)))
+      return 1;
+
+   if (chipset >= 0x50) {
+      i = nouveau_codegen(chipset, type, tokens, &size, &code);
+   } else if (chipset >= 0x30) {
+      i = nv30_codegen(chipset, type, tokens, &size, &code);
+   } else {
+      _debug_printf("chipset NV%02X not supported\n", chipset);
+      i = 1;
+   }
+   if (i)
+      return i;
+
+   _debug_printf("program binary (%d bytes)\n", size);
+   for (i = 0; i < size; i += 4) {
+      printf("%08x ", code[i / 4]);
+      if (i % (8 * 4) == (7 * 4))
+         printf("\n");
+   }
+   if (i % (8 * 4) != 0)
+      printf("\n");
+
+   return 0;
+}