4 * GLSL wrapper code to hide the differences between the different gl/sl APIs.
6 * Copyright (c) 2007-2010 Nathan Keynes.
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 2 of the License, or
11 * (at your option) any later version.
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
23 #include "pvr2/glutil.h"
24 #include "pvr2/shaders.h"
26 #define MAX_ERROR_BUF 4096
27 #define INVALID_SHADER 0
28 #define INVALID_PROGRAM 0
30 #ifdef HAVE_OPENGL_SHADER_ARB
31 typedef GLhandleARB gl_program_t;
32 typedef GLhandleARB gl_shader_t;
34 typedef GLuint gl_program_t;
35 typedef GLuint gl_shader_t;
38 gboolean glsl_is_supported();
39 gl_shader_t glsl_create_vertex_shader( const char *source );
40 gl_shader_t glsl_create_fragment_shader( const char *source );
41 gl_program_t glsl_create_program( gl_shader_t *shaderv );
42 void glsl_use_program(gl_program_t program);
43 void glsl_destroy_shader(gl_shader_t shader);
44 void glsl_destroy_program(gl_program_t program);
46 typedef void (*program_cleanup_fn_t)();
47 static void glsl_set_cleanup_fn( program_cleanup_fn_t );
49 #ifdef HAVE_OPENGL_SHADER_ARB
51 gboolean glsl_is_supported()
53 return isGLExtensionSupported("GL_ARB_fragment_shader") &&
54 isGLExtensionSupported("GL_ARB_vertex_shader") &&
55 isGLExtensionSupported("GL_ARB_shading_language_100");
58 const char *glsl_get_version()
60 return glGetString(GL_SHADING_LANGUAGE_VERSION_ARB);
63 void glsl_print_error( char *msg, GLhandleARB obj )
65 char buf[MAX_ERROR_BUF];
67 glGetInfoLogARB( obj, sizeof(buf), &length, buf );
68 ERROR( "%s: %s", msg, buf );
71 gboolean glsl_check_shader_error( char *msg, GLhandleARB obj )
75 glGetObjectParameterivARB(obj, GL_OBJECT_COMPILE_STATUS_ARB, &value);
77 glsl_print_error(msg, obj);
83 gboolean glsl_check_program_error( char *msg, GLhandleARB obj )
85 if( glGetError() != GL_NO_ERROR ) {
86 glsl_print_error(msg, obj);
91 gl_shader_t glsl_create_vertex_shader( const char *source )
94 gl_shader_t shader = glCreateShaderObjectARB(GL_VERTEX_SHADER_ARB);
96 glShaderSourceARB( shader, 1, &source, NULL );
97 glCompileShaderARB(shader);
98 ok = glsl_check_shader_error("Failed to compile vertex shader", shader);
100 glDeleteObjectARB(shader);
101 return INVALID_SHADER;
107 gl_shader_t glsl_create_fragment_shader( const char *source )
110 gl_shader_t shader = glCreateShaderObjectARB(GL_FRAGMENT_SHADER_ARB);
112 glShaderSourceARB( shader, 1, &source, NULL );
113 glCompileShaderARB(shader);
114 ok = glsl_check_shader_error("Failed to compile fragment shader", shader);
116 glDeleteObjectARB(shader);
117 return INVALID_SHADER;
123 gl_program_t glsl_create_program( gl_shader_t *shaderv )
127 gl_program_t program = glCreateProgramObjectARB();
129 for( i=0; shaderv[i] != INVALID_SHADER; i++ ) {
130 glAttachObjectARB(program, shaderv[i]);
133 glLinkProgramARB(program);
134 ok = glsl_check_program_error( "Failed to link shader program", program );
136 glDeleteObjectARB(program);
137 return INVALID_PROGRAM;
143 void glsl_use_program(gl_program_t program)
145 glUseProgramObjectARB(program);
148 void glsl_destroy_shader(gl_shader_t shader)
150 glDeleteObjectARB(shader);
153 void glsl_destroy_program(gl_program_t program)
155 glDeleteObjectARB(program);
158 static inline GLint glsl_get_uniform_location(gl_program_t program, const char *name)
160 return glGetUniformLocationARB(program, name);
163 static inline GLint glsl_get_attrib_location(gl_program_t program, const char *name)
165 return glGetAttribLocationARB(program, name);
168 #define glsl_set_uniform_sampler1D(id,v) glUniform1iARB(id,v)
169 #define glsl_set_uniform_sampler2D(id,v) glUniform1iARB(id,v)
170 #define glsl_set_uniform_float(id,v) glUniform1fARB(id,v)
171 #define glsl_set_uniform_vec2(id,v) glUniform2fvARB(id,1,v)
172 #define glsl_set_uniform_vec3(id,v) glUniform3fvARB(id,1,v)
173 #define glsl_set_uniform_vec4(id,v) glUniform4fvARB(id,1,v)
174 #define glsl_set_uniform_mat4(id,v) glUniformMatrix4fvARB(id,1,GL_FALSE,v)
175 #define glsl_set_attrib_vec2(id,stride,v) glVertexAttribPointerARB(id, 2, GL_FLOAT, GL_FALSE, stride, v)
176 #define glsl_set_attrib_vec3(id,stride,v) glVertexAttribPointerARB(id, 3, GL_FLOAT, GL_FALSE, stride, v)
177 #define glsl_set_attrib_vec4(id,stride,v) glVertexAttribPointerARB(id, 4, GL_FLOAT, GL_FALSE, stride, v)
178 #define glsl_enable_attrib(id) glEnableVertexAttribArrayARB(id)
179 #define glsl_disable_attrib(id) glDisableVertexAttribArrayARB(id)
181 #elif HAVE_OPENGL_SHADER
183 gboolean glsl_is_supported()
185 return isGLExtensionSupported("GL_ARB_fragment_shader") &&
186 isGLExtensionSupported("GL_ARB_vertex_shader") &&
187 isGLExtensionSupported("GL_ARB_shading_language_100");
190 const char *glsl_get_version()
192 return glGetString(GL_SHADING_LANGUAGE_VERSION);
195 gboolean glsl_check_shader_error( char *msg, GLuint shader )
199 glGetShaderiv( shader, GL_COMPILE_STATUS, &value );
201 char buf[MAX_ERROR_BUF];
203 glGetShaderInfoLog( shader, sizeof(buf), &length, buf );
204 ERROR( "%s: %s", msg, buf );
210 gboolean glsl_check_program_error( char *msg, GLuint program )
212 if( glGetError() != GL_NO_ERROR ) {
213 char buf[MAX_ERROR_BUF];
215 glGetProgramInfoLog( program, sizeof(buf), &length, buf );
216 ERROR( "%s: %s", msg, buf );
222 gl_shader_t glsl_create_vertex_shader( const char *source )
225 gl_shader_t shader = glCreateShader(GL_VERTEX_SHADER);
227 glShaderSource( shader, 1, &source, NULL );
228 glCompileShader(shader);
229 ok = glsl_check_shader_error( "Failed to compile vertex shader", shader );
231 glDeleteShader(shader);
232 return INVALID_SHADER;
239 gl_shader_t glsl_create_fragment_shader( const char *source )
242 gl_shader_t shader = glCreateShader(GL_FRAGMENT_SHADER);
244 glShaderSource( shader, 1, &source, NULL );
245 glCompileShader(shader);
246 ok = glsl_check_shader_error( "Failed to compile fragment shader", shader );
248 glDeleteShader(shader);
249 return INVALID_SHADER;
255 gl_program_t glsl_create_program( gl_shader_t *shaderv )
259 gl_program_t program = glCreateProgram();
261 for( i=0; shaderv[i] != INVALID_SHADER; i++ ) {
262 glAttachShader(program, shaderv[i]);
264 glLinkProgram(program);
265 ok = glsl_check_program_error( "Failed to link shader program", program );
267 glDeleteProgram(program);
268 return INVALID_PROGRAM;
274 void glsl_use_program(gl_program_t program)
276 glUseProgram(program);
279 void glsl_destroy_shader(gl_shader_t shader)
281 glDeleteShader(shader);
284 void glsl_destroy_program(gl_program_t program)
286 glDeleteProgram(program);
289 static inline GLint glsl_get_uniform_location(gl_program_t program, const char *name)
291 return glGetUniformLocation(program, name);
293 static inline GLint glsl_get_attrib_location(gl_program_t program, const char *name)
295 return glGetAttribLocation(program, name);
298 #define glsl_set_uniform_sampler1D(id,v) glUniform1i(id,v)
299 #define glsl_set_uniform_sampler2D(id,v) glUniform1i(id,v)
300 #define glsl_set_uniform_float(id,v) glUniform1f(id,v)
301 #define glsl_set_uniform_vec2(id,v) glUniform2fv(id,1,v)
302 #define glsl_set_uniform_vec3(id,v) glUniform3fv(id,1,v)
303 #define glsl_set_uniform_vec4(id,v) glUniform4fv(id,1,v)
304 #define glsl_set_uniform_mat4(id,v) glUniformMatrix4fv(id,1,GL_FALSE,v)
305 #define glsl_set_attrib_vec2(id,stride,v) glVertexAttribPointer(id, 2, GL_FLOAT, GL_FALSE, stride, v)
306 #define glsl_set_attrib_vec3(id,stride,v) glVertexAttribPointer(id, 3, GL_FLOAT, GL_FALSE, stride, v)
307 #define glsl_set_attrib_vec4(id,stride,v) glVertexAttribPointer(id, 4, GL_FLOAT, GL_FALSE, stride, v)
308 #define glsl_enable_attrib(id) glEnableVertexAttribArray(id)
309 #define glsl_disable_attrib(id) glDisableVertexAttribArray(id)
313 gboolean glsl_is_supported()
318 const char *glsl_get_version()
323 gl_shader_t glsl_create_vertex_shader( const char *source )
328 gl_shader_t glsl_create_fragment_shader( const char *source )
333 gl_program_t glsl_create_program( gl_shader_t *shaderv )
338 void glsl_use_program(gl_program_t program)
342 void glsl_destroy_shader(gl_shader_t shader)
346 void glsl_destroy_program(gl_program_t program)
350 static inline GLint glsl_get_uniform_location(gl_program_t program, const char *name)
355 static inline GLint glsl_get_attrib_location(gl_program_t program, const char *name)
360 #define glsl_set_uniform_sampler1D(id,v)
361 #define glsl_set_uniform_sampler2D(id,v)
362 #define glsl_set_uniform_float(id,v)
363 #define glsl_set_uniform_vec2(id,v)
364 #define glsl_set_uniform_vec3(id,v)
365 #define glsl_set_uniform_vec4(id,v)
366 #define glsl_set_uniform_mat4(id,v)
367 #define glsl_set_attrib_vec2(id,stride,v)
368 #define glsl_set_attrib_vec3(id,stride,v)
369 #define glsl_set_attrib_vec4(id,stride,v)
370 #define glsl_enable_attrib(id)
371 #define glsl_disable_attrib(id)
376 /****************************************************************************/
378 program_cleanup_fn_t current_cleanup_fn = NULL;
380 /* Pull in the auto-generated shader definitions */
382 #include "pvr2/shaders.def"
384 static gl_program_t program_array[GLSL_NUM_PROGRAMS];
387 gboolean glsl_load_shaders()
389 gl_shader_t shader_array[GLSL_NUM_SHADERS];
392 for( i=0; i<GLSL_NUM_SHADERS; i++ )
393 shader_array[i] = INVALID_SHADER;
394 for( i=0; i<GLSL_NUM_PROGRAMS; i++ )
395 program_array[i] = INVALID_PROGRAM;
397 /* Compile the shader fragments */
398 for( i=0; shader_source[i].type != GLSL_NO_SHADER; i++ ) {
399 gl_shader_t shader = INVALID_SHADER;
400 switch(shader_source[i].type) {
401 case GLSL_VERTEX_SHADER:
402 shader = glsl_create_vertex_shader(shader_source[i].source);
404 case GLSL_FRAGMENT_SHADER:
405 shader = glsl_create_fragment_shader(shader_source[i].source);
408 if( shader == INVALID_SHADER ) {
412 shader_array[i] = shader;
416 /* Link the programs */
417 if(ok) for( i=0; program_list[i][0] != GLSL_NO_SHADER; i++ ) {
418 gl_shader_t shaderv[GLSL_NUM_SHADERS+1];
419 for( j=0; program_list[i][j] != GLSL_NO_SHADER; j++ ) {
420 shaderv[j] = shader_array[program_list[i][j]];
422 shaderv[j] = INVALID_SHADER;
423 gl_program_t program = glsl_create_program(shaderv);
424 if( program == INVALID_PROGRAM ) {
428 /* Check that we can actually use the program (can this really fail?) */
429 glsl_use_program(program);
430 if( !glsl_check_program_error( "Failed to activate shader program", program ) ) {
433 program_array[i] = program;
438 * Destroy the compiled fragments (the linked programs don't need them
441 for( i=0; i<GLSL_NUM_SHADERS; i++ ) {
442 if( shader_array[i] != INVALID_SHADER )
443 glsl_destroy_shader(shader_array[i]);
447 * If we errored, delete the programs. It's all or nothing.
450 glsl_unload_shaders();
454 glsl_init_programs(program_array);
459 static void glsl_set_cleanup_fn( program_cleanup_fn_t fn )
461 if( fn != current_cleanup_fn ) {
462 if( current_cleanup_fn != NULL ) {
463 current_cleanup_fn();
465 current_cleanup_fn = fn;
469 static void glsl_run_cleanup_fn()
471 if( current_cleanup_fn ) {
472 current_cleanup_fn();
474 current_cleanup_fn = NULL;
477 void glsl_unload_shaders()
480 glsl_run_cleanup_fn();
481 for( i=0; i<GLSL_NUM_PROGRAMS; i++ ) {
482 if( program_array[i] != INVALID_PROGRAM ) {
483 glsl_destroy_program(program_array[i]);
484 program_array[i] = INVALID_PROGRAM;
489 void glsl_clear_shader()
491 glsl_run_cleanup_fn();
.