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
, size_t *size
)
45 ssize_t total_read
= 0;
46 int fd
= open(file_name
, O_RDONLY
);
53 if (fstat(fd
, & st
) == 0) {
54 text
= (char *) talloc_size(ctx
, st
.st_size
+ 1);
57 ssize_t bytes
= read(fd
, text
+ total_read
,
58 st
.st_size
- total_read
);
70 } while (total_read
< st
.st_size
);
72 text
[total_read
] = '\0';
84 usage_fail(const char *name
)
86 printf("%s <filename.frag|filename.vert>\n", name
);
96 const struct option compiler_opts
[] = {
97 { "dump-ast", 0, &dump_ast
, 1 },
98 { "dump-hir", 0, &dump_hir
, 1 },
99 { "dump-lir", 0, &dump_lir
, 1 },
100 { "link", 0, &do_link
, 1 },
105 compile_shader(struct glsl_shader
*shader
)
107 struct _mesa_glsl_parse_state
*state
;
109 state
= talloc_zero(talloc_parent(shader
), struct _mesa_glsl_parse_state
);
111 switch (shader
->Type
) {
112 case GL_VERTEX_SHADER
: state
->target
= vertex_shader
; break;
113 case GL_FRAGMENT_SHADER
: state
->target
= fragment_shader
; break;
114 case GL_GEOMETRY_SHADER
: state
->target
= geometry_shader
; break;
117 state
->scanner
= NULL
;
118 state
->translation_unit
.make_empty();
119 state
->symbols
= new(shader
) glsl_symbol_table
;
120 state
->info_log
= talloc_strdup(shader
, "");
121 state
->error
= false;
122 state
->temp_index
= 0;
123 state
->loop_or_switch_nesting
= NULL
;
124 state
->ARB_texture_rectangle_enable
= true;
126 /* Create a new context for the preprocessor output. Ultimately, this
127 * should probably be the parser context, but there isn't one yet.
129 const char *source
= shader
->Source
;
130 state
->error
= preprocess(shader
, &source
, &state
->info_log
);
133 _mesa_glsl_lexer_ctor(state
, source
);
134 _mesa_glsl_parse(state
);
135 _mesa_glsl_lexer_dtor(state
);
139 foreach_list_const(n
, &state
->translation_unit
) {
140 ast_node
*ast
= exec_node_data(ast_node
, n
, link
);
146 shader
->ir
.make_empty();
147 if (!state
->error
&& !state
->translation_unit
.is_empty())
148 _mesa_ast_to_hir(&shader
->ir
, state
);
150 validate_ir_tree(&shader
->ir
);
152 /* Print out the unoptimized IR. */
153 if (!state
->error
&& dump_hir
) {
154 _mesa_print_ir(&shader
->ir
, state
);
157 /* Optimization passes */
158 if (!state
->error
&& !shader
->ir
.is_empty()) {
163 progress
= do_function_inlining(&shader
->ir
) || progress
;
164 progress
= do_if_simplification(&shader
->ir
) || progress
;
165 progress
= do_copy_propagation(&shader
->ir
) || progress
;
166 progress
= do_dead_code_local(&shader
->ir
) || progress
;
167 progress
= do_dead_code_unlinked(state
, &shader
->ir
) || progress
;
168 progress
= do_constant_variable_unlinked(&shader
->ir
) || progress
;
169 progress
= do_constant_folding(&shader
->ir
) || progress
;
170 progress
= do_vec_index_to_swizzle(&shader
->ir
) || progress
;
171 progress
= do_swizzle_swizzle(&shader
->ir
) || progress
;
175 validate_ir_tree(&shader
->ir
);
177 /* Print out the resulting IR */
178 if (!state
->error
&& dump_lir
) {
179 _mesa_print_ir(&shader
->ir
, state
);
182 shader
->symbols
= state
->symbols
;
183 shader
->CompileStatus
= !state
->error
;
186 talloc_free(shader
->InfoLog
);
188 shader
->InfoLog
= state
->info_log
;
196 main(int argc
, char **argv
)
198 int status
= EXIT_SUCCESS
;
202 while ((c
= getopt_long(argc
, argv
, "", compiler_opts
, &idx
)) != -1)
209 struct glsl_program
*whole_program
;
211 whole_program
= talloc_zero (NULL
, struct glsl_program
);
212 assert(whole_program
!= NULL
);
214 for (/* empty */; argc
> optind
; optind
++) {
215 whole_program
->Shaders
= (struct glsl_shader
**)
216 talloc_realloc(whole_program
, whole_program
->Shaders
,
217 struct glsl_shader
*, whole_program
->NumShaders
+ 1);
218 assert(whole_program
->Shaders
!= NULL
);
220 struct glsl_shader
*shader
= talloc_zero(whole_program
, glsl_shader
);
222 whole_program
->Shaders
[whole_program
->NumShaders
] = shader
;
223 whole_program
->NumShaders
++;
225 const unsigned len
= strlen(argv
[optind
]);
229 const char *const ext
= & argv
[optind
][len
- 5];
230 if (strncmp(".vert", ext
, 5) == 0)
231 shader
->Type
= GL_VERTEX_SHADER
;
232 else if (strncmp(".geom", ext
, 5) == 0)
233 shader
->Type
= GL_GEOMETRY_SHADER
;
234 else if (strncmp(".frag", ext
, 5) == 0)
235 shader
->Type
= GL_FRAGMENT_SHADER
;
239 shader
->Source
= load_text_file(whole_program
,
240 argv
[optind
], &shader
->SourceLen
);
241 if (shader
->Source
== NULL
) {
242 printf("File \"%s\" does not exist.\n", argv
[optind
]);
246 compile_shader(shader
);
248 if (!shader
->CompileStatus
) {
249 printf("Info log for %s:\n%s\n", argv
[optind
], shader
->InfoLog
);
250 status
= EXIT_FAILURE
;
255 if ((status
== EXIT_SUCCESS
) && do_link
) {
256 link_shaders(whole_program
);
257 status
= (whole_program
->LinkStatus
) ? EXIT_SUCCESS
: EXIT_FAILURE
;
260 talloc_free(whole_program
);