2 * Copyright (C) 2019 Ryan Houdek <Sonicadvance1@gmail.com>
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:
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
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
24 #include "disassemble.h"
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"
33 #include "bifrost_compile.h"
36 compile_shader(char **argv
)
38 struct gl_shader_program
*prog
;
40 unsigned shader_types
[2] = {
45 struct standalone_options options
= {
50 static struct gl_context local_ctx
;
52 prog
= standalone_compile_shader(&options
, 2, argv
, &local_ctx
);
53 prog
->_LinkedShaders
[MESA_SHADER_FRAGMENT
]->Program
->info
.stage
= MESA_SHADER_FRAGMENT
;
56 struct bifrost_program compiled
;
57 for (unsigned i
= 0; i
< 2; ++i
) {
58 nir
[i
] = glsl_to_nir(&local_ctx
, prog
, shader_types
[i
], &bifrost_nir_options
);
59 NIR_PASS_V(nir
[i
], nir_lower_global_vars_to_local
);
60 NIR_PASS_V(nir
[i
], nir_split_var_copies
);
61 NIR_PASS_V(nir
[i
], nir_lower_var_copies
);
63 NIR_PASS_V(nir
[i
], nir_lower_alu_to_scalar
, NULL
);
65 /* before buffers and vars_to_ssa */
66 NIR_PASS_V(nir
[i
], gl_nir_lower_bindless_images
);
68 NIR_PASS_V(nir
[i
], gl_nir_lower_buffers
, prog
);
69 NIR_PASS_V(nir
[i
], nir_opt_constant_folding
);
70 bifrost_compile_shader_nir(nir
[i
], &compiled
);
76 disassemble(const char *filename
)
78 FILE *fp
= fopen(filename
, "rb");
81 fseek(fp
, 0, SEEK_END
);
82 int filesize
= ftell(fp
);
85 unsigned char *code
= malloc(filesize
);
86 int res
= fread(code
, 1, filesize
, fp
);
87 if (res
!= filesize
) {
88 printf("Couldn't read full file\n");
92 disassemble_bifrost(code
, filesize
, false);
97 main(int argc
, char **argv
)
100 printf("Pass a command\n");
103 if (strcmp(argv
[1], "compile") == 0) {
104 compile_shader(&argv
[2]);
105 } else if (strcmp(argv
[1], "disasm") == 0) {
106 disassemble(argv
[2]);