1 /*
2  * Copyright © 2014 Intel Corporation
3  * Copyright © 2016 Advanced Micro Devices, Inc.
4  * All Rights Reserved.
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining a
7  * copy of this software and associated documentation files (the "Software"),
8  * to deal in the Software without restriction, including without limitation
9  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
10  * and/or sell copies of the Software, and to permit persons to whom the
11  * Software is furnished to do so, subject to the following conditions:
12  *
13  * The above copyright notice and this permission notice (including the next
14  * paragraph) shall be included in all copies or substantial portions of the
15  * Software.
16  *
17  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
18  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
19  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
20  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
21  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
22  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
23  * DEALINGS IN THE SOFTWARE.
24  */
25 
26 /**
27  * This program reads and compiles multiple GLSL shaders from one source file.
28  * Each shader must begin with the #shader directive. Syntax:
29  *     #shader [vs|tcs|tes|gs|ps|cs] [name]
30  *
31  * The shader name is printed, followed by the stderr output from
32  * glCreateShaderProgramv. (radeonsi prints the shader disassembly there)
33  *
34  * The optional parameter -mcpu=[processor] forces radeonsi to compile for
35  * the specified GPU processor. (e.g. tahiti, bonaire, tonga)
36  *
37  * The program doesn't check if the underlying driver is really radeonsi.
38  * OpenGL 4.3 Core profile is required.
39  */
40 
41 /* for asprintf() */
42 #define _GNU_SOURCE
43 
44 #include <stdio.h>
45 #include <fcntl.h>
46 #include <string.h>
47 #include <stdlib.h>
48 
49 #include <epoxy/gl.h>
50 #include <epoxy/egl.h>
51 #include <gbm.h>
52 
53 #define unlikely(x) __builtin_expect(!!(x), 0)
54 #define ARRAY_SIZE(x) (sizeof(x) / sizeof((x)[0]))
55 
56 static int fd;
57 static EGLDisplay egl_dpy;
58 static EGLContext ctx;
59 
60 static void
create_gl_core_context()61 create_gl_core_context()
62 {
63     const char *client_extensions = eglQueryString(EGL_NO_DISPLAY,
64                                                    EGL_EXTENSIONS);
65     if (!client_extensions) {
66         fprintf(stderr, "ERROR: Missing EGL_EXT_client_extensions\n");
67         exit(1);
68     }
69 
70     if (!strstr(client_extensions, "EGL_MESA_platform_gbm")) {
71         fprintf(stderr, "ERROR: Missing EGL_MESA_platform_gbm\n");
72         exit(1);
73     }
74 
75     fd = open("/dev/dri/renderD128", O_RDWR);
76     if (unlikely(fd < 0)) {
77         fprintf(stderr, "ERROR: Couldn't open /dev/dri/renderD128\n");
78         exit(1);
79     }
80 
81     struct gbm_device *gbm = gbm_create_device(fd);
82     if (unlikely(gbm == NULL)) {
83         fprintf(stderr, "ERROR: Couldn't create gbm device\n");
84         exit(1);
85     }
86 
87     egl_dpy = eglGetPlatformDisplayEXT(EGL_PLATFORM_GBM_MESA,
88                                        gbm, NULL);
89     if (unlikely(egl_dpy == EGL_NO_DISPLAY)) {
90         fprintf(stderr, "ERROR: eglGetDisplay() failed\n");
91         exit(1);
92     }
93 
94     if (unlikely(!eglInitialize(egl_dpy, NULL, NULL))) {
95         fprintf(stderr, "ERROR: eglInitialize() failed\n");
96         exit(1);
97     }
98 
99     static const char *egl_extension[] = {
100             "EGL_KHR_create_context",
101             "EGL_KHR_surfaceless_context"
102     };
103     const char *extension_string = eglQueryString(egl_dpy, EGL_EXTENSIONS);
104     for (int i = 0; i < ARRAY_SIZE(egl_extension); i++) {
105         if (strstr(extension_string, egl_extension[i]) == NULL) {
106             fprintf(stderr, "ERROR: Missing %s\n", egl_extension[i]);
107             exit(1);
108         }
109     }
110 
111     static const EGLint config_attribs[] = {
112         EGL_RENDERABLE_TYPE, EGL_OPENGL_BIT,
113         EGL_NONE
114     };
115     EGLConfig cfg;
116     EGLint count;
117 
118     if (!eglChooseConfig(egl_dpy, config_attribs, &cfg, 1, &count) ||
119         count == 0) {
120         fprintf(stderr, "ERROR: eglChooseConfig() failed\n");
121         exit(1);
122     }
123     eglBindAPI(EGL_OPENGL_API);
124 
125     static const EGLint attribs[] = {
126         EGL_CONTEXT_OPENGL_PROFILE_MASK_KHR,
127         EGL_CONTEXT_OPENGL_CORE_PROFILE_BIT_KHR,
128         EGL_CONTEXT_MAJOR_VERSION_KHR, 4,
129         EGL_CONTEXT_MINOR_VERSION_KHR, 3,
130         EGL_NONE
131     };
132     ctx = eglCreateContext(egl_dpy, cfg, EGL_NO_CONTEXT, attribs);
133     if (ctx == EGL_NO_CONTEXT) {
134         fprintf(stderr, "eglCreateContext(GL 3.2) failed.\n");
135         exit(1);
136     }
137 
138     if (!eglMakeCurrent(egl_dpy, EGL_NO_SURFACE, EGL_NO_SURFACE, ctx)) {
139         fprintf(stderr, "eglMakeCurrent failed.\n");
140         exit(1);
141     }
142 }
143 
144 static char *
read_file(const char * filename)145 read_file(const char *filename)
146 {
147     FILE *f = fopen(filename, "r");
148     if (!f) {
149         fprintf(stderr, "Can't open file: %s\n", filename);
150         exit(1);
151     }
152 
153     fseek(f, 0, SEEK_END);
154     int filesize = ftell(f);
155     fseek(f, 0, SEEK_SET);
156 
157     char *input = (char*)malloc(filesize + 1);
158     if (!input) {
159         fprintf(stderr, "malloc failed\n");
160         exit(1);
161     }
162 
163     if (fread(input, filesize, 1, f) != 1) {
164         fprintf(stderr, "fread failed\n");
165         exit(1);
166     }
167     fclose(f);
168 
169     input[filesize] = 0;
170     return input;
171 }
172 
addenv(const char * name,const char * value)173 static void addenv(const char *name, const char *value)
174 {
175     const char *orig = getenv(name);
176     if (orig) {
177         char *newval;
178         (void)!asprintf(&newval, "%s,%s", orig, value);
179         setenv(name, newval, 1);
180         free(newval);
181     } else {
182         setenv(name, value, 1);
183     }
184 }
185 
186 int
main(int argc,char ** argv)187 main(int argc, char **argv)
188 {
189     const char *filename = NULL;
190 
191     for (int i = 1; i < argc; i++) {
192         if (strstr(argv[i], "-mcpu=") == argv[i]) {
193             setenv("SI_FORCE_FAMILY", argv[i] + 6, 1);
194         } else if (filename == NULL) {
195             filename = argv[i];
196         } else {
197             if (strcmp(argv[i], "--help") != 0 && strcmp(argv[i], "-h") != 0)
198                 fprintf(stderr, "Unknown option: %s\n\n", argv[i]);
199 
200             fprintf(stderr, "Usage: amdgcn_glslc -mcpu=[chip name] [glsl filename]\n");
201             return 1;
202         }
203     }
204 
205     if (filename == NULL) {
206         fprintf(stderr, "No filename specified.\n");
207         return 1;
208     }
209 
210     addenv("R600_DEBUG", "precompile,vs,tcs,tes,gs,ps,cs,noir,notgsi");
211 
212     create_gl_core_context();
213 
214     /* Read the source. */
215     char *input = read_file(filename);
216 
217     /* Comment out lines beginning with ; (FileCheck prefix). */
218     if (input[0] == ';')
219         memcpy(input, "//", 2);
220 
221     char *s = input;
222     while (s = strstr(s, "\n;"))
223         memcpy(s + 1, "//", 2);
224 
225     s = input;
226     while (s && (s = strstr(s, "#shader "))) {
227         char type_str[16], name[128];
228         GLenum type;
229 
230         /* If #shader is not at the beginning of the line. */
231         if (s != input && s[-1] != '\n' && s[-1] != 0) {
232             s = strstr(s, "\n");
233             continue;
234         }
235 
236         /* Parse the #shader directive. */
237         if (sscanf(s + 8, "%s %s", type_str, name) != 2) {
238             fprintf(stderr, "Cannot parse #shader directive.\n");
239             continue;
240         }
241 
242         if (!strcmp(type_str, "vs"))
243             type = GL_VERTEX_SHADER;
244         else if (!strcmp(type_str, "tcs"))
245             type = GL_TESS_CONTROL_SHADER;
246         else if (!strcmp(type_str, "tes"))
247             type = GL_TESS_EVALUATION_SHADER;
248         else if (!strcmp(type_str, "gs"))
249             type = GL_GEOMETRY_SHADER;
250         else if (!strcmp(type_str, "fs"))
251             type = GL_FRAGMENT_SHADER;
252         else if (!strcmp(type_str, "cs"))
253             type = GL_COMPUTE_SHADER;
254 
255         /* Go the next line. */
256         s = strstr(s, "\n");
257         if (!s)
258             break;
259         s++;
260 
261         const char *source = s;
262 
263         /* Cut the shader source at the end. */
264         s = strstr(s, "#shader");
265         if (s && s[-1] == '\n')
266             s[-1] = 0;
267 
268         /* Compile the shader. */
269         printf("@%s:\n", name);
270 
271         /* Redirect stderr to stdout for the compiler. */
272         FILE *stderr_original = stderr;
273         stderr = stdout;
274         GLuint prog = glCreateShaderProgramv(type, 1, &source);
275         stderr = stderr_original;
276 
277         GLint linked;
278         glGetProgramiv(prog, GL_LINK_STATUS, &linked);
279 
280         if (!linked) {
281             char log[4096];
282             GLsizei length;
283 
284             glGetProgramInfoLog(prog, sizeof(log), &length, log);
285             fprintf(stderr, "ERROR: Compile failure:\n\n%s\n%s\n",
286                     source, log);
287             return 1;
288         }
289 
290         glDeleteProgram(prog);
291     }
292     return 0;
293 }
294