src/gallium/drivers/freedreno/ir3/ir3_cmdline.c

   1 /*
   2  * Copyright (C) 2014 Rob Clark <robclark@freedesktop.org>
   3  *
   4  * Permission is hereby granted, free of charge, to any person obtaining a
   5  * copy of this software and associated documentation files (the "Software"),
   6  * to deal in the Software without restriction, including without limitation
   7  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
   8  * and/or sell copies of the Software, and to permit persons to whom the
   9  * Software is furnished to do so, subject to the following conditions:
  10  *
  11  * The above copyright notice and this permission notice (including the next
  12  * paragraph) shall be included in all copies or substantial portions of the
  13  * Software.
  14  *
  15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  18  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  20  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  21  * SOFTWARE.
  22  *
  23  * Authors:
  24  *    Rob Clark <robclark@freedesktop.org>
  25  */
  26
  27 #include <sys/types.h>
  28 #include <sys/stat.h>
  29 #include <sys/mman.h>
  30 #include <fcntl.h>
  31 #include <stdint.h>
  32 #include <stdlib.h>
  33 #include <stdio.h>
  34 #include <err.h>
  35
  36 #include "tgsi/tgsi_parse.h"
  37 #include "tgsi/tgsi_text.h"
  38 #include "tgsi/tgsi_dump.h"
  39
  40 #include "freedreno_util.h"
  41
  42 #include "ir3_compiler.h"
  43 #include "instr-a3xx.h"
  44 #include "ir3.h"
  45
  46 static void dump_info(struct ir3_shader_variant *so, const char *str)
  47 {
  48         uint32_t *bin;
  49         const char *type = ir3_shader_stage(so->shader);
  50         // TODO make gpu_id configurable on cmdline
  51         bin = ir3_shader_assemble(so, 320);
  52         debug_printf("; %s: %s\n", type, str);
  53         ir3_shader_disasm(so, bin);
  54         free(bin);
  55 }
  56
  57
  58 static int
  59 read_file(const char *filename, void **ptr, size_t *size)
  60 {
  61         int fd, ret;
  62         struct stat st;
  63
  64         *ptr = MAP_FAILED;
  65
  66         fd = open(filename, O_RDONLY);
  67         if (fd == -1) {
  68                 warnx("couldn't open `%s'", filename);
  69                 return 1;
  70         }
  71
  72         ret = fstat(fd, &st);
  73         if (ret)
  74                 errx(1, "couldn't stat `%s'", filename);
  75
  76         *size = st.st_size;
  77         *ptr = mmap(NULL, st.st_size, PROT_READ, MAP_SHARED, fd, 0);
  78         if (*ptr == MAP_FAILED)
  79                 errx(1, "couldn't map `%s'", filename);
  80
  81         close(fd);
  82
  83         return 0;
  84 }
  85
  86 static void print_usage(void)
  87 {
  88         printf("Usage: ir3_compiler [OPTIONS]... FILE\n");
  89         printf("    --verbose         - verbose compiler/debug messages\n");
  90         printf("    --binning-pass    - generate binning pass shader (VERT)\n");
  91         printf("    --color-two-side  - emulate two-sided color (FRAG)\n");
  92         printf("    --half-precision  - use half-precision\n");
  93         printf("    --saturate-s MASK - bitmask of samplers to saturate S coord\n");
  94         printf("    --saturate-t MASK - bitmask of samplers to saturate T coord\n");
  95         printf("    --saturate-r MASK - bitmask of samplers to saturate R coord\n");
  96         printf("    --stream-out      - enable stream-out (aka transform feedback)\n");
  97         printf("    --ucp MASK        - bitmask of enabled user-clip-planes\n");
  98         printf("    --gpu GPU_ID      - specify gpu-id (default 320)\n");
  99         printf("    --help            - show this message\n");
 100 }
 101
 102 int main(int argc, char **argv)
 103 {
 104         int ret = 0, n = 1;
 105         const char *filename;
 106         struct tgsi_token toks[65536];
 107         struct tgsi_parse_context parse;
 108         struct ir3_compiler *compiler;
 109         struct ir3_shader_variant v;
 110         struct ir3_shader s;
 111         struct ir3_shader_key key = {};
 112         unsigned gpu_id = 320;
 113         const char *info;
 114         void *ptr;
 115         size_t size;
 116
 117         fd_mesa_debug |= FD_DBG_DISASM;
 118
 119         memset(&s, 0, sizeof(s));
 120         memset(&v, 0, sizeof(v));
 121
 122         /* cmdline args which impact shader variant get spit out in a
 123          * comment on the first line..  a quick/dirty way to preserve
 124          * that info so when ir3test recompiles the shader with a new
 125          * compiler version, we use the same shader-key settings:
 126          */
 127         debug_printf("; options:");
 128
 129         while (n < argc) {
 130                 if (!strcmp(argv[n], "--verbose")) {
 131                         fd_mesa_debug |= FD_DBG_MSGS | FD_DBG_OPTMSGS;
 132                         n++;
 133                         continue;
 134                 }
 135
 136                 if (!strcmp(argv[n], "--binning-pass")) {
 137                         debug_printf(" %s", argv[n]);
 138                         key.binning_pass = true;
 139                         n++;
 140                         continue;
 141                 }
 142
 143                 if (!strcmp(argv[n], "--color-two-side")) {
 144                         debug_printf(" %s", argv[n]);
 145                         key.color_two_side = true;
 146                         n++;
 147                         continue;
 148                 }
 149
 150                 if (!strcmp(argv[n], "--half-precision")) {
 151                         debug_printf(" %s", argv[n]);
 152                         key.half_precision = true;
 153                         n++;
 154                         continue;
 155                 }
 156
 157                 if (!strcmp(argv[n], "--saturate-s")) {
 158                         debug_printf(" %s %s", argv[n], argv[n+1]);
 159                         key.vsaturate_s = key.fsaturate_s = strtol(argv[n+1], NULL, 0);
 160                         n += 2;
 161                         continue;
 162                 }
 163
 164                 if (!strcmp(argv[n], "--saturate-t")) {
 165                         debug_printf(" %s %s", argv[n], argv[n+1]);
 166                         key.vsaturate_t = key.fsaturate_t = strtol(argv[n+1], NULL, 0);
 167                         n += 2;
 168                         continue;
 169                 }
 170
 171                 if (!strcmp(argv[n], "--saturate-r")) {
 172                         debug_printf(" %s %s", argv[n], argv[n+1]);
 173                         key.vsaturate_r = key.fsaturate_r = strtol(argv[n+1], NULL, 0);
 174                         n += 2;
 175                         continue;
 176                 }
 177
 178                 if (!strcmp(argv[n], "--stream-out")) {
 179                         struct pipe_stream_output_info *so = &s.stream_output;
 180                         debug_printf(" %s", argv[n]);
 181                         /* TODO more dynamic config based on number of outputs, etc
 182                          * rather than just hard-code for first output:
 183                          */
 184                         so->num_outputs = 1;
 185                         so->stride[0] = 4;
 186                         so->output[0].register_index = 0;
 187                         so->output[0].start_component = 0;
 188                         so->output[0].num_components = 4;
 189                         so->output[0].output_buffer = 0;
 190                         so->output[0].dst_offset = 2;
 191                         so->output[0].stream = 0;
 192                         n++;
 193                         continue;
 194                 }
 195
 196                 if (!strcmp(argv[n], "--ucp")) {
 197                         debug_printf(" %s %s", argv[n], argv[n+1]);
 198                         key.ucp_enables = strtol(argv[n+1], NULL, 0);
 199                         n += 2;
 200                         continue;
 201                 }
 202
 203                 if (!strcmp(argv[n], "--gpu")) {
 204                         debug_printf(" %s %s", argv[n], argv[n+1]);
 205                         gpu_id = strtol(argv[n+1], NULL, 0);
 206                         n += 2;
 207                         continue;
 208                 }
 209
 210                 if (!strcmp(argv[n], "--help")) {
 211                         print_usage();
 212                         return 0;
 213                 }
 214
 215                 break;
 216         }
 217         debug_printf("\n");
 218
 219         filename = argv[n];
 220
 221         ret = read_file(filename, &ptr, &size);
 222         if (ret) {
 223                 print_usage();
 224                 return ret;
 225         }
 226
 227         if (fd_mesa_debug & FD_DBG_OPTMSGS)
 228                 debug_printf("%s\n", (char *)ptr);
 229
 230         if (!tgsi_text_translate(ptr, toks, Elements(toks)))
 231                 errx(1, "could not parse `%s'", filename);
 232
 233         s.tokens = toks;
 234
 235         v.key = key;
 236         v.shader = &s;
 237
 238         tgsi_parse_init(&parse, toks);
 239         switch (parse.FullHeader.Processor.Processor) {
 240         case TGSI_PROCESSOR_FRAGMENT:
 241                 s.type = v.type = SHADER_FRAGMENT;
 242                 break;
 243         case TGSI_PROCESSOR_VERTEX:
 244                 s.type = v.type = SHADER_VERTEX;
 245                 break;
 246         case TGSI_PROCESSOR_COMPUTE:
 247                 s.type = v.type = SHADER_COMPUTE;
 248                 break;
 249         }
 250
 251         /* TODO cmdline option to target different gpus: */
 252         compiler = ir3_compiler_create(gpu_id);
 253
 254         info = "NIR compiler";
 255         ret = ir3_compile_shader_nir(compiler, &v);
 256         if (ret) {
 257                 fprintf(stderr, "compiler failed!\n");
 258                 return ret;
 259         }
 260         dump_info(&v, info);
 261 }