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