2 * Copyright © 2008, 2009 Intel Corporation
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
20 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
21 * DEALINGS IN THE SOFTWARE.
27 #include <sys/types.h>
33 #include "glsl_parser_extras.h"
34 #include "glsl_parser.h"
35 #include "ir_optimization.h"
36 #include "ir_print_visitor.h"
39 /* Returned string will have 'ctx' as its talloc owner. */
41 load_text_file(void *ctx
, const char *file_name
)
45 ssize_t total_read
= 0;
46 int fd
= open(file_name
, O_RDONLY
);
52 if (fstat(fd
, & st
) == 0) {
53 text
= (char *) talloc_size(ctx
, st
.st_size
+ 1);
56 ssize_t bytes
= read(fd
, text
+ total_read
,
57 st
.st_size
- total_read
);
69 } while (total_read
< st
.st_size
);
71 text
[total_read
] = '\0';
82 usage_fail(const char *name
)
84 printf("%s <filename.frag|filename.vert>\n", name
);
94 const struct option compiler_opts
[] = {
95 { "dump-ast", 0, &dump_ast
, 1 },
96 { "dump-hir", 0, &dump_hir
, 1 },
97 { "dump-lir", 0, &dump_lir
, 1 },
98 { "link", 0, &do_link
, 1 },
103 steal_memory(ir_instruction
*ir
, void *new_ctx
)
105 talloc_steal(new_ctx
, ir
);
109 compile_shader(struct gl_shader
*shader
)
111 struct _mesa_glsl_parse_state
*state
;
113 state
= talloc_zero(talloc_parent(shader
), struct _mesa_glsl_parse_state
);
115 switch (shader
->Type
) {
116 case GL_VERTEX_SHADER
: state
->target
= vertex_shader
; break;
117 case GL_FRAGMENT_SHADER
: state
->target
= fragment_shader
; break;
118 case GL_GEOMETRY_SHADER
: state
->target
= geometry_shader
; break;
121 state
->scanner
= NULL
;
122 state
->translation_unit
.make_empty();
123 state
->symbols
= new(shader
) glsl_symbol_table
;
124 state
->info_log
= talloc_strdup(shader
, "");
125 state
->error
= false;
126 state
->temp_index
= 0;
127 state
->loop_or_switch_nesting
= NULL
;
128 state
->ARB_texture_rectangle_enable
= true;
130 state
->Const
.MaxDrawBuffers
= 2;
132 const char *source
= shader
->Source
;
133 state
->error
= preprocess(state
, &source
, &state
->info_log
);
136 _mesa_glsl_lexer_ctor(state
, source
);
137 _mesa_glsl_parse(state
);
138 _mesa_glsl_lexer_dtor(state
);
142 foreach_list_const(n
, &state
->translation_unit
) {
143 ast_node
*ast
= exec_node_data(ast_node
, n
, link
);
149 shader
->ir
= new(shader
) exec_list
;
150 if (!state
->error
&& !state
->translation_unit
.is_empty())
151 _mesa_ast_to_hir(shader
->ir
, state
);
153 validate_ir_tree(shader
->ir
);
155 /* Print out the unoptimized IR. */
156 if (!state
->error
&& dump_hir
) {
157 _mesa_print_ir(shader
->ir
, state
);
160 /* Optimization passes */
161 if (!state
->error
&& !shader
->ir
->is_empty()) {
166 progress
= do_function_inlining(shader
->ir
) || progress
;
167 progress
= do_if_simplification(shader
->ir
) || progress
;
168 progress
= do_copy_propagation(shader
->ir
) || progress
;
169 progress
= do_dead_code_local(shader
->ir
) || progress
;
170 progress
= do_dead_code_unlinked(state
, shader
->ir
) || progress
;
171 progress
= do_constant_variable_unlinked(shader
->ir
) || progress
;
172 progress
= do_constant_folding(shader
->ir
) || progress
;
173 progress
= do_vec_index_to_swizzle(shader
->ir
) || progress
;
174 progress
= do_swizzle_swizzle(shader
->ir
) || progress
;
178 validate_ir_tree(shader
->ir
);
180 /* Print out the resulting IR */
181 if (!state
->error
&& dump_lir
) {
182 _mesa_print_ir(shader
->ir
, state
);
185 shader
->symbols
= state
->symbols
;
186 shader
->CompileStatus
= !state
->error
;
189 talloc_free(shader
->InfoLog
);
191 shader
->InfoLog
= state
->info_log
;
193 /* Retain any live IR, but trash the rest. */
194 foreach_list(node
, shader
->ir
) {
195 visit_tree((ir_instruction
*) node
, steal_memory
, shader
);
204 main(int argc
, char **argv
)
206 int status
= EXIT_SUCCESS
;
210 while ((c
= getopt_long(argc
, argv
, "", compiler_opts
, &idx
)) != -1)
217 struct gl_shader_program
*whole_program
;
219 whole_program
= talloc_zero (NULL
, struct gl_shader_program
);
220 assert(whole_program
!= NULL
);
222 for (/* empty */; argc
> optind
; optind
++) {
223 whole_program
->Shaders
= (struct gl_shader
**)
224 talloc_realloc(whole_program
, whole_program
->Shaders
,
225 struct gl_shader
*, whole_program
->NumShaders
+ 1);
226 assert(whole_program
->Shaders
!= NULL
);
228 struct gl_shader
*shader
= talloc_zero(whole_program
, gl_shader
);
230 whole_program
->Shaders
[whole_program
->NumShaders
] = shader
;
231 whole_program
->NumShaders
++;
233 const unsigned len
= strlen(argv
[optind
]);
237 const char *const ext
= & argv
[optind
][len
- 5];
238 if (strncmp(".vert", ext
, 5) == 0)
239 shader
->Type
= GL_VERTEX_SHADER
;
240 else if (strncmp(".geom", ext
, 5) == 0)
241 shader
->Type
= GL_GEOMETRY_SHADER
;
242 else if (strncmp(".frag", ext
, 5) == 0)
243 shader
->Type
= GL_FRAGMENT_SHADER
;
247 shader
->Source
= load_text_file(whole_program
, argv
[optind
]);
248 if (shader
->Source
== NULL
) {
249 printf("File \"%s\" does not exist.\n", argv
[optind
]);
253 compile_shader(shader
);
255 if (!shader
->CompileStatus
) {
256 printf("Info log for %s:\n%s\n", argv
[optind
], shader
->InfoLog
);
257 status
= EXIT_FAILURE
;
262 if ((status
== EXIT_SUCCESS
) && do_link
) {
263 link_shaders(whole_program
);
264 status
= (whole_program
->LinkStatus
) ? EXIT_SUCCESS
: EXIT_FAILURE
;
267 talloc_free(whole_program
);
268 _mesa_glsl_release_types();