src/panfrost/bifrost/cmdline.c

   1 /*
   2  * Copyright (C) 2019 Ryan Houdek <Sonicadvance1@gmail.com>
   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
  24 #include "disassemble.h"
  25
  26 #include "main/mtypes.h"
  27 #include "compiler/glsl/standalone.h"
  28 #include "compiler/glsl/glsl_to_nir.h"
  29 #include "compiler/glsl/gl_nir.h"
  30 #include "compiler/nir_types.h"
  31 #include "util/u_dynarray.h"
  32 #include "bifrost_compile.h"
  33
  34 static void
  35 compile_shader(char **argv)
  36 {
  37         struct gl_shader_program *prog;
  38         nir_shader *nir[2];
  39         unsigned shader_types[2] = {
  40                 MESA_SHADER_VERTEX,
  41                 MESA_SHADER_FRAGMENT,
  42         };
  43
  44         struct standalone_options options = {
  45                 .glsl_version = 430,
  46                 .do_link = true,
  47         };
  48
  49         static struct gl_context local_ctx;
  50
  51         prog = standalone_compile_shader(&options, 2, argv, &local_ctx);
  52         prog->_LinkedShaders[MESA_SHADER_FRAGMENT]->Program->info.stage = MESA_SHADER_FRAGMENT;
  53
  54         bifrost_program compiled;
  55         for (unsigned i = 0; i < 2; ++i) {
  56                 nir[i] = glsl_to_nir(&local_ctx, prog, shader_types[i], &bifrost_nir_options);
  57                 NIR_PASS_V(nir[i], nir_lower_global_vars_to_local);
  58                 NIR_PASS_V(nir[i], nir_split_var_copies);
  59                 NIR_PASS_V(nir[i], nir_lower_var_copies);
  60
  61                 NIR_PASS_V(nir[i], nir_lower_alu_to_scalar, NULL, NULL);
  62
  63                 /* before buffers and vars_to_ssa */
  64                 NIR_PASS_V(nir[i], gl_nir_lower_images, true);
  65
  66                 NIR_PASS_V(nir[i], gl_nir_lower_buffers, prog);
  67                 NIR_PASS_V(nir[i], nir_opt_constant_folding);
  68
  69                 unsigned product_id = 0x7212; /* Mali G52 */
  70                 bifrost_compile_shader_nir(nir[i], &compiled, product_id);
  71         }
  72 }
  73
  74 static void
  75 disassemble(const char *filename)
  76 {
  77         FILE *fp = fopen(filename, "rb");
  78         assert(fp);
  79
  80         fseek(fp, 0, SEEK_END);
  81         unsigned filesize = ftell(fp);
  82         rewind(fp);
  83
  84         unsigned char *code = malloc(filesize);
  85         unsigned res = fread(code, 1, filesize, fp);
  86         if (res != filesize) {
  87                 printf("Couldn't read full file\n");
  88         }
  89         fclose(fp);
  90
  91         disassemble_bifrost(stdout, code, filesize, false);
  92         free(code);
  93 }
  94
  95 int
  96 main(int argc, char **argv)
  97 {
  98         if (argc < 2) {
  99                 printf("Pass a command\n");
 100                 exit(1);
 101         }
 102
 103         if (strcmp(argv[1], "compile") == 0)
 104                 compile_shader(&argv[2]);
 105         else if (strcmp(argv[1], "disasm") == 0)
 106                 disassemble(argv[2]);
 107         else
 108                 unreachable("Unknown command. Valid: compile/disasm");
 109
 110         return 0;
 111 }