gpuray_glsl
diff src/glsdr.c @ 0:f234630e38ff
initial commit
author | John Tsiombikas <nuclear@member.fsf.org> |
---|---|
date | Sun, 09 Nov 2014 13:03:36 +0200 |
parents | |
children |
line diff
1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000 1.2 +++ b/src/glsdr.c Sun Nov 09 13:03:36 2014 +0200 1.3 @@ -0,0 +1,411 @@ 1.4 +#include <stdio.h> 1.5 +#include <stdlib.h> 1.6 +#include <string.h> 1.7 +#include <errno.h> 1.8 +#include <stdarg.h> 1.9 +#include <assert.h> 1.10 +#include <GL/glew.h> 1.11 + 1.12 +#if defined(unix) || defined(__unix__) 1.13 +#include <unistd.h> 1.14 +#include <sys/stat.h> 1.15 +#endif /* unix */ 1.16 + 1.17 +#include "glsdr.h" 1.18 + 1.19 +static const char *sdrtypestr(unsigned int sdrtype); 1.20 + 1.21 +unsigned int create_vertex_shader(const char *src) 1.22 +{ 1.23 + return create_shader(src, GL_VERTEX_SHADER); 1.24 +} 1.25 + 1.26 +unsigned int create_pixel_shader(const char *src) 1.27 +{ 1.28 + return create_shader(src, GL_FRAGMENT_SHADER); 1.29 +} 1.30 + 1.31 +unsigned int create_tessctl_shader(const char *src) 1.32 +{ 1.33 + return create_shader(src, GL_TESS_CONTROL_SHADER); 1.34 +} 1.35 + 1.36 +unsigned int create_tesseval_shader(const char *src) 1.37 +{ 1.38 + return create_shader(src, GL_TESS_EVALUATION_SHADER); 1.39 +} 1.40 + 1.41 +unsigned int create_geometry_shader(const char *src) 1.42 +{ 1.43 + return create_shader(src, GL_GEOMETRY_SHADER); 1.44 +} 1.45 + 1.46 +unsigned int create_shader(const char *src, unsigned int sdr_type) 1.47 +{ 1.48 + unsigned int sdr; 1.49 + int success, info_len; 1.50 + char *info_str = 0; 1.51 + GLenum err; 1.52 + 1.53 + sdr = glCreateShader(sdr_type); 1.54 + assert(glGetError() == GL_NO_ERROR); 1.55 + glShaderSource(sdr, 1, &src, 0); 1.56 + err = glGetError(); 1.57 + assert(err == GL_NO_ERROR); 1.58 + glCompileShader(sdr); 1.59 + assert(glGetError() == GL_NO_ERROR); 1.60 + 1.61 + glGetShaderiv(sdr, GL_COMPILE_STATUS, &success); 1.62 + assert(glGetError() == GL_NO_ERROR); 1.63 + glGetShaderiv(sdr, GL_INFO_LOG_LENGTH, &info_len); 1.64 + assert(glGetError() == GL_NO_ERROR); 1.65 + 1.66 + if(info_len) { 1.67 + if((info_str = malloc(info_len + 1))) { 1.68 + glGetShaderInfoLog(sdr, info_len, 0, info_str); 1.69 + assert(glGetError() == GL_NO_ERROR); 1.70 + } 1.71 + } 1.72 + 1.73 + if(success) { 1.74 + fprintf(stderr, info_str ? "done: %s\n" : "done\n", info_str); 1.75 + } else { 1.76 + fprintf(stderr, info_str ? "failed: %s\n" : "failed\n", info_str); 1.77 + glDeleteShader(sdr); 1.78 + sdr = 0; 1.79 + } 1.80 + 1.81 + free(info_str); 1.82 + return sdr; 1.83 +} 1.84 + 1.85 +void free_shader(unsigned int sdr) 1.86 +{ 1.87 + glDeleteShader(sdr); 1.88 +} 1.89 + 1.90 +unsigned int load_vertex_shader(const char *fname) 1.91 +{ 1.92 + return load_shader(fname, GL_VERTEX_SHADER); 1.93 +} 1.94 + 1.95 +unsigned int load_pixel_shader(const char *fname) 1.96 +{ 1.97 + return load_shader(fname, GL_FRAGMENT_SHADER); 1.98 +} 1.99 + 1.100 +unsigned int load_tessctl_shader(const char *fname) 1.101 +{ 1.102 + return load_shader(fname, GL_TESS_CONTROL_SHADER); 1.103 +} 1.104 + 1.105 +unsigned int load_tesseval_shader(const char *fname) 1.106 +{ 1.107 + return load_shader(fname, GL_TESS_EVALUATION_SHADER); 1.108 +} 1.109 + 1.110 +unsigned int load_geometry_shader(const char *fname) 1.111 +{ 1.112 + return load_shader(fname, GL_GEOMETRY_SHADER); 1.113 +} 1.114 + 1.115 +unsigned int load_shader(const char *fname, unsigned int sdr_type) 1.116 +{ 1.117 +#if defined(unix) || defined(__unix__) 1.118 + struct stat st; 1.119 +#endif 1.120 + unsigned int sdr; 1.121 + size_t filesize; 1.122 + FILE *fp; 1.123 + char *src; 1.124 + 1.125 + if(!(fp = fopen(fname, "r"))) { 1.126 + fprintf(stderr, "failed to open shader %s: %s\n", fname, strerror(errno)); 1.127 + return 0; 1.128 + } 1.129 + 1.130 +#if defined(unix) || defined(__unix__) 1.131 + fstat(fileno(fp), &st); 1.132 + filesize = st.st_size; 1.133 +#else 1.134 + fseek(fp, 0, SEEK_END); 1.135 + filesize = ftell(fp); 1.136 + fseek(fp, 0, SEEK_SET); 1.137 +#endif /* unix */ 1.138 + 1.139 + if(!(src = malloc(filesize + 1))) { 1.140 + fclose(fp); 1.141 + return 0; 1.142 + } 1.143 + fread(src, 1, filesize, fp); 1.144 + src[filesize] = 0; 1.145 + fclose(fp); 1.146 + 1.147 + fprintf(stderr, "compiling %s shader: %s... ", sdrtypestr(sdr_type), fname); 1.148 + sdr = create_shader(src, sdr_type); 1.149 + 1.150 + free(src); 1.151 + return sdr; 1.152 +} 1.153 + 1.154 + 1.155 +unsigned int get_vertex_shader(const char *fname) 1.156 +{ 1.157 + return get_shader(fname, GL_VERTEX_SHADER); 1.158 +} 1.159 + 1.160 +unsigned int get_pixel_shader(const char *fname) 1.161 +{ 1.162 + return get_shader(fname, GL_FRAGMENT_SHADER); 1.163 +} 1.164 + 1.165 +unsigned int get_tessctl_shader(const char *fname) 1.166 +{ 1.167 + return get_shader(fname, GL_TESS_CONTROL_SHADER); 1.168 +} 1.169 + 1.170 +unsigned int get_tesseval_shader(const char *fname) 1.171 +{ 1.172 + return get_shader(fname, GL_TESS_EVALUATION_SHADER); 1.173 +} 1.174 + 1.175 +unsigned int get_geometry_shader(const char *fname) 1.176 +{ 1.177 + return get_shader(fname, GL_GEOMETRY_SHADER); 1.178 +} 1.179 + 1.180 +unsigned int get_shader(const char *fname, unsigned int sdr_type) 1.181 +{ 1.182 + unsigned int sdr; 1.183 + if(!(sdr = load_shader(fname, sdr_type))) { 1.184 + return 0; 1.185 + } 1.186 + return sdr; 1.187 +} 1.188 + 1.189 + 1.190 +/* ---- gpu programs ---- */ 1.191 + 1.192 +unsigned int create_program(void) 1.193 +{ 1.194 + unsigned int prog = glCreateProgram(); 1.195 + assert(glGetError() == GL_NO_ERROR); 1.196 + return prog; 1.197 +} 1.198 + 1.199 +unsigned int create_program_link(unsigned int sdr0, ...) 1.200 +{ 1.201 + unsigned int prog, sdr; 1.202 + va_list ap; 1.203 + 1.204 + if(!(prog = create_program())) { 1.205 + return 0; 1.206 + } 1.207 + 1.208 + attach_shader(prog, sdr0); 1.209 + if(glGetError()) { 1.210 + return 0; 1.211 + } 1.212 + 1.213 + va_start(ap, sdr0); 1.214 + while((sdr = va_arg(ap, unsigned int))) { 1.215 + attach_shader(prog, sdr); 1.216 + if(glGetError()) { 1.217 + return 0; 1.218 + } 1.219 + } 1.220 + va_end(ap); 1.221 + 1.222 + if(link_program(prog) == -1) { 1.223 + free_program(prog); 1.224 + return 0; 1.225 + } 1.226 + return prog; 1.227 +} 1.228 + 1.229 +unsigned int create_program_load(const char *vfile, const char *pfile) 1.230 +{ 1.231 + unsigned int sdr[2]; 1.232 + int i = 0; 1.233 + 1.234 + if(vfile && *vfile && !(sdr[i++] = get_vertex_shader(vfile))) { 1.235 + return 0; 1.236 + } 1.237 + if(pfile && *pfile && !(sdr[i++] = get_pixel_shader(pfile))) { 1.238 + return 0; 1.239 + } 1.240 + return create_program_link(sdr[0], sdr[1], 0); 1.241 +} 1.242 + 1.243 +void free_program(unsigned int sdr) 1.244 +{ 1.245 + glDeleteProgram(sdr); 1.246 +} 1.247 + 1.248 +void attach_shader(unsigned int prog, unsigned int sdr) 1.249 +{ 1.250 + glAttachShader(prog, sdr); 1.251 + assert(glGetError() == GL_NO_ERROR); 1.252 +} 1.253 + 1.254 +int link_program(unsigned int prog) 1.255 +{ 1.256 + int linked, info_len, retval = 0; 1.257 + char *info_str = 0; 1.258 + 1.259 + glLinkProgram(prog); 1.260 + assert(glGetError() == GL_NO_ERROR); 1.261 + glGetProgramiv(prog, GL_LINK_STATUS, &linked); 1.262 + assert(glGetError() == GL_NO_ERROR); 1.263 + glGetProgramiv(prog, GL_INFO_LOG_LENGTH, &info_len); 1.264 + assert(glGetError() == GL_NO_ERROR); 1.265 + 1.266 + if(info_len) { 1.267 + if((info_str = malloc(info_len + 1))) { 1.268 + glGetProgramInfoLog(prog, info_len, 0, info_str); 1.269 + assert(glGetError() == GL_NO_ERROR); 1.270 + } 1.271 + } 1.272 + 1.273 + if(linked) { 1.274 + fprintf(stderr, info_str ? "linking done: %s\n" : "linking done\n", info_str); 1.275 + } else { 1.276 + fprintf(stderr, info_str ? "linking failed: %s\n" : "linking failed\n", info_str); 1.277 + retval = -1; 1.278 + } 1.279 + 1.280 + free(info_str); 1.281 + return retval; 1.282 +} 1.283 + 1.284 +int bind_program(unsigned int prog) 1.285 +{ 1.286 + GLenum err; 1.287 + 1.288 + glUseProgram(prog); 1.289 + if(prog && (err = glGetError()) != GL_NO_ERROR) { 1.290 + /* maybe the program is not linked, try linking first */ 1.291 + if(err == GL_INVALID_OPERATION) { 1.292 + if(link_program(prog) == -1) { 1.293 + return -1; 1.294 + } 1.295 + glUseProgram(prog); 1.296 + return glGetError() == GL_NO_ERROR ? 0 : -1; 1.297 + } 1.298 + return -1; 1.299 + } 1.300 + return 0; 1.301 +} 1.302 + 1.303 +/* ugly but I'm not going to write the same bloody code over and over */ 1.304 +#define BEGIN_UNIFORM_CODE \ 1.305 + int loc, curr_prog; \ 1.306 + glGetIntegerv(GL_CURRENT_PROGRAM, &curr_prog); \ 1.307 + if((unsigned int)curr_prog != prog && bind_program(prog) == -1) { \ 1.308 + return -1; \ 1.309 + } \ 1.310 + if((loc = glGetUniformLocation(prog, name)) != -1) 1.311 + 1.312 +#define END_UNIFORM_CODE \ 1.313 + if((unsigned int)curr_prog != prog) { \ 1.314 + bind_program(curr_prog); \ 1.315 + } \ 1.316 + return loc == -1 ? -1 : 0 1.317 + 1.318 +int set_uniform_int(unsigned int prog, const char *name, int val) 1.319 +{ 1.320 + BEGIN_UNIFORM_CODE { 1.321 + glUniform1i(loc, val); 1.322 + } 1.323 + END_UNIFORM_CODE; 1.324 +} 1.325 + 1.326 +int set_uniform_float(unsigned int prog, const char *name, float val) 1.327 +{ 1.328 + BEGIN_UNIFORM_CODE { 1.329 + glUniform1f(loc, val); 1.330 + } 1.331 + END_UNIFORM_CODE; 1.332 +} 1.333 + 1.334 +int set_uniform_float2(unsigned int prog, const char *name, float x, float y) 1.335 +{ 1.336 + BEGIN_UNIFORM_CODE { 1.337 + glUniform2f(loc, x, y); 1.338 + } 1.339 + END_UNIFORM_CODE; 1.340 +} 1.341 + 1.342 +int set_uniform_float3(unsigned int prog, const char *name, float x, float y, float z) 1.343 +{ 1.344 + BEGIN_UNIFORM_CODE { 1.345 + glUniform3f(loc, x, y, z); 1.346 + } 1.347 + END_UNIFORM_CODE; 1.348 +} 1.349 + 1.350 +int set_uniform_float4(unsigned int prog, const char *name, float x, float y, float z, float w) 1.351 +{ 1.352 + BEGIN_UNIFORM_CODE { 1.353 + glUniform4f(loc, x, y, z, w); 1.354 + } 1.355 + END_UNIFORM_CODE; 1.356 +} 1.357 + 1.358 +int set_uniform_matrix4(unsigned int prog, const char *name, float *mat) 1.359 +{ 1.360 + BEGIN_UNIFORM_CODE { 1.361 + glUniformMatrix4fv(loc, 1, GL_FALSE, mat); 1.362 + } 1.363 + END_UNIFORM_CODE; 1.364 +} 1.365 + 1.366 +int set_uniform_matrix4_transposed(unsigned int prog, const char *name, float *mat) 1.367 +{ 1.368 + BEGIN_UNIFORM_CODE { 1.369 + glUniformMatrix4fv(loc, 1, GL_TRUE, mat); 1.370 + } 1.371 + END_UNIFORM_CODE; 1.372 +} 1.373 + 1.374 +int get_attrib_loc(unsigned int prog, const char *name) 1.375 +{ 1.376 + int loc, curr_prog; 1.377 + 1.378 + glGetIntegerv(GL_CURRENT_PROGRAM, &curr_prog); 1.379 + if((unsigned int)curr_prog != prog && bind_program(prog) == -1) { 1.380 + return -1; 1.381 + } 1.382 + 1.383 + loc = glGetAttribLocation(prog, (char*)name); 1.384 + 1.385 + if((unsigned int)curr_prog != prog) { 1.386 + bind_program(curr_prog); 1.387 + } 1.388 + return loc; 1.389 +} 1.390 + 1.391 +void set_attrib_float3(int attr_loc, float x, float y, float z) 1.392 +{ 1.393 + glVertexAttrib3f(attr_loc, x, y, z); 1.394 +} 1.395 + 1.396 +static const char *sdrtypestr(unsigned int sdrtype) 1.397 +{ 1.398 + switch(sdrtype) { 1.399 + case GL_VERTEX_SHADER: 1.400 + return "vertex"; 1.401 + case GL_FRAGMENT_SHADER: 1.402 + return "pixel"; 1.403 + case GL_TESS_CONTROL_SHADER: 1.404 + return "tessellation control"; 1.405 + case GL_TESS_EVALUATION_SHADER: 1.406 + return "tessellation evaluation"; 1.407 + case GL_GEOMETRY_SHADER: 1.408 + return "geometry"; 1.409 + 1.410 + default: 1.411 + break; 1.412 + } 1.413 + return "<unknown>"; 1.414 +}