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"
34 compile_shader(char **argv
)
36 struct gl_shader_program
*prog
;
38 unsigned shader_types
[2] = {
43 struct standalone_options options
= {
48 static struct gl_context local_ctx
;
50 prog
= standalone_compile_shader(&options
, 2, argv
, &local_ctx
);
51 prog
->_LinkedShaders
[MESA_SHADER_FRAGMENT
]->Program
->info
.stage
= MESA_SHADER_FRAGMENT
;
54 struct 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
);
61 NIR_PASS_V(nir
[i
], nir_lower_alu_to_scalar
, NULL
);
63 /* before buffers and vars_to_ssa */
64 NIR_PASS_V(nir
[i
], gl_nir_lower_bindless_images
);
66 NIR_PASS_V(nir
[i
], gl_nir_lower_buffers
, prog
);
67 NIR_PASS_V(nir
[i
], nir_opt_constant_folding
);
68 bifrost_compile_shader_nir(nir
[i
], &compiled
);
74 disassemble(const char *filename
)
76 FILE *fp
= fopen(filename
, "rb");
79 fseek(fp
, 0, SEEK_END
);
80 int filesize
= ftell(fp
);
83 unsigned char *code
= malloc(filesize
);
84 int res
= fread(code
, 1, filesize
, fp
);
85 if (res
!= filesize
) {
86 printf("Couldn't read full file\n");
90 disassemble_bifrost(code
, filesize
, false);
95 main(int argc
, char **argv
)
98 printf("Pass a command\n");
101 if (strcmp(argv
[1], "compile") == 0) {
102 compile_shader(&argv
[2]);
104 else if (strcmp(argv
[1], "disasm") == 0) {
105 disassemble(argv
[2]);