refmod_test

annotate refmod_test.c @ 0:b469e6a72636

find changesets by author, revision, files, or words in the commit message
initial commit
author John Tsiombikas <nuclear@member.fsf.org>
date Thu, 18 Feb 2016 23:15:43 +0200
parents
children 7e911c994ef2
rev   line source
nuclear@0 1 #include <stdio.h>
nuclear@0 2 #include <stdlib.h>
nuclear@0 3 #include <string.h>
nuclear@0 4 #include <GL/glut.h>
nuclear@0 5 #include "sdr.h"
nuclear@0 6
nuclear@0 7 #ifdef FREEGLUT
nuclear@0 8 #include <GL/freeglut_ext.h>
nuclear@0 9 #define MOUSEWHEEL
nuclear@0 10 #endif
nuclear@0 11
nuclear@0 12 void redraw(void);
nuclear@0 13 void set_material_color(float r, float g, float b, float a);
nuclear@0 14 void setup_lights(void);
nuclear@0 15 void key_handler(unsigned char key, int x, int y);
nuclear@0 16 void key_up_handler(unsigned char key, int x, int y);
nuclear@0 17 void skey_handler(int key, int x, int y);
nuclear@0 18 void button_handler(int bn, int state, int x, int y);
nuclear@0 19 void mouse_handler(int x, int y);
nuclear@0 20 void reshape(int x, int y);
nuclear@0 21 #ifdef MOUSEWHEEL
nuclear@0 22 void wheel_handler(int unk, int dir, int x, int y);
nuclear@0 23 #endif
nuclear@0 24 void menu_handler(int val);
nuclear@0 25 void init_sdr(void);
nuclear@0 26
nuclear@0 27 int xres, yres;
nuclear@0 28
nuclear@0 29 float cam_x, cam_y = 0.6, cam_z = 4;
nuclear@0 30 float cam_rot, cam_pitch = 35.0;
nuclear@0 31
nuclear@0 32 float pan_offs = 0.025;
nuclear@0 33
nuclear@0 34 unsigned int prog;
nuclear@0 35
nuclear@0 36 float roughness = 0.24;
nuclear@0 37 float specularity = 0.86;
nuclear@0 38 float ior = 3.2;
nuclear@0 39 float spec_pow = 60.0;
nuclear@0 40
nuclear@0 41 enum {DSDR_LAMBERT, DSDR_OREN_NAYAR};
nuclear@0 42 enum {SSDR_PHONG, SSDR_BLINN, SSDR_COOK_TORR};
nuclear@0 43
nuclear@0 44 int dif_sdr = DSDR_OREN_NAYAR;
nuclear@0 45 int spec_sdr = SSDR_COOK_TORR;
nuclear@0 46
nuclear@0 47 int main(int argc, char **argv)
nuclear@0 48 {
nuclear@0 49 unsigned int vs, ps, pslib;
nuclear@0 50 int dif_menu, spec_menu, i;
nuclear@0 51
nuclear@0 52 for(i=1; i<argc; i++) {
nuclear@0 53 if(argv[i][0] == '-' && argv[i][2] == 0) {
nuclear@0 54 switch(argv[i][1]) {
nuclear@0 55 case 'd':
nuclear@0 56 if(strcmp(argv[++i], "lambert") == 0) {
nuclear@0 57 dif_sdr = DSDR_LAMBERT;
nuclear@0 58 } else if(strcmp(argv[i], "oren-nayar") == 0) {
nuclear@0 59 dif_sdr = DSDR_OREN_NAYAR;
nuclear@0 60 } else {
nuclear@0 61 printf("-d (diffuse model) choices:\n");
nuclear@0 62 printf(" lambert \"Photometria sive de mensura de gratibus luminis, colorum et umbrae\", 1760\n");
nuclear@0 63 printf(" oren-nayar \"Generalization of Lambert's Reflectance Model\", SIGGRAPH 1994\n");
nuclear@0 64 return EXIT_FAILURE;
nuclear@0 65 }
nuclear@0 66 break;
nuclear@0 67
nuclear@0 68 case 's':
nuclear@0 69 if(strcmp(argv[++i], "phong") == 0) {
nuclear@0 70 spec_sdr = SSDR_PHONG;
nuclear@0 71 } else if(strcmp(argv[i], "cook-torrance") == 0) {
nuclear@0 72 spec_sdr = SSDR_COOK_TORR;
nuclear@0 73 } else {
nuclear@0 74 printf("-s (specular model) choices:\n");
nuclear@0 75 printf(" phong \"Illumination for Computer Generated Pictures\", CACM 1975\n");
nuclear@0 76 printf(" blinn \"Models of Light Reflection for Computer Synthesized Pictures\", SIGGRAPH 1977\n");
nuclear@0 77 printf(" cook-torrance \"A Reflectance Model for Computer Graphics\", SIGGRAPH 1981\n");
nuclear@0 78 return EXIT_FAILURE;
nuclear@0 79 }
nuclear@0 80 break;
nuclear@0 81
nuclear@0 82 default:
nuclear@0 83 fprintf(stderr, "unrecognized argument: %s\n", argv[i]);
nuclear@0 84 return EXIT_FAILURE;
nuclear@0 85 }
nuclear@0 86 }
nuclear@0 87 }
nuclear@0 88
nuclear@0 89 glutInitWindowSize(800, 600);
nuclear@0 90 glutInit(&argc, argv);
nuclear@0 91 glutInitDisplayMode(GLUT_RGBA | GLUT_DEPTH | GLUT_DOUBLE);
nuclear@0 92 glutCreateWindow("1.roughness 2.specularity 3.ior 4.phong power");
nuclear@0 93 xres = glutGet(GLUT_WINDOW_WIDTH);
nuclear@0 94 yres = glutGet(GLUT_WINDOW_HEIGHT);
nuclear@0 95
nuclear@0 96 glutKeyboardFunc(key_handler);
nuclear@0 97 glutKeyboardUpFunc(key_up_handler);
nuclear@0 98 glutSpecialFunc(skey_handler);
nuclear@0 99 glutMotionFunc(mouse_handler);
nuclear@0 100 glutMouseFunc(button_handler);
nuclear@0 101 glutReshapeFunc(reshape);
nuclear@0 102 glutDisplayFunc(redraw);
nuclear@0 103 #ifdef MOUSEWHEEL
nuclear@0 104 glutMouseWheelFunc(wheel_handler);
nuclear@0 105 #endif
nuclear@0 106
nuclear@0 107 /* create the menus */
nuclear@0 108 dif_menu = glutCreateMenu(menu_handler);
nuclear@0 109 glutAddMenuEntry("lambert", DSDR_LAMBERT);
nuclear@0 110 glutAddMenuEntry("oren-nayar", DSDR_OREN_NAYAR);
nuclear@0 111
nuclear@0 112 spec_menu = glutCreateMenu(menu_handler);
nuclear@0 113 glutAddMenuEntry("phong", SSDR_PHONG + 10);
nuclear@0 114 glutAddMenuEntry("blinn", SSDR_BLINN + 10);
nuclear@0 115 glutAddMenuEntry("cook-torrance", SSDR_COOK_TORR + 10);
nuclear@0 116
nuclear@0 117 glutCreateMenu(menu_handler);
nuclear@0 118 glutAddSubMenu("diffuse models", dif_menu);
nuclear@0 119 glutAddSubMenu("specular models", spec_menu);
nuclear@0 120
nuclear@0 121 glutAttachMenu(GLUT_RIGHT_BUTTON);
nuclear@0 122
nuclear@0 123
nuclear@0 124 glEnable(GL_DEPTH_TEST);
nuclear@0 125 glEnable(GL_CULL_FACE);
nuclear@0 126
nuclear@0 127 glEnable(GL_LIGHTING);
nuclear@0 128 glEnable(GL_LIGHT0);
nuclear@0 129
nuclear@0 130 glMatrixMode(GL_PROJECTION);
nuclear@0 131 gluPerspective(45.0, (float)xres / (float)yres, 1.0, 100.0);
nuclear@0 132 glMatrixMode(GL_MODELVIEW);
nuclear@0 133
nuclear@0 134 init_sdr();
nuclear@0 135
nuclear@0 136 if(!(vs = load_vertex_shader("sdr.vs.glsl"))) {
nuclear@0 137 fprintf(stderr, "failed to load vertex shader\n");
nuclear@0 138 return EXIT_FAILURE;
nuclear@0 139 }
nuclear@0 140
nuclear@0 141 if(!(pslib = load_pixel_shader("sdr_ref_models.glsl"))) {
nuclear@0 142 fprintf(stderr, "failed to load reflection models shader\n");
nuclear@0 143 return EXIT_FAILURE;
nuclear@0 144 }
nuclear@0 145
nuclear@0 146 if(!(ps = load_pixel_shader("sdr.ps.glsl"))) {
nuclear@0 147 fprintf(stderr, "failed to load pixel shader\n");
nuclear@0 148 return EXIT_FAILURE;
nuclear@0 149 }
nuclear@0 150
nuclear@0 151 if(!(prog = create_program())) {
nuclear@0 152 fprintf(stderr, "failed to create GPU program\n");
nuclear@0 153 return EXIT_FAILURE;
nuclear@0 154 }
nuclear@0 155 attach_shader(prog, vs);
nuclear@0 156 attach_shader(prog, pslib);
nuclear@0 157 attach_shader(prog, ps);
nuclear@0 158
nuclear@0 159 if(link_program(prog) == -1) {
nuclear@0 160 fprintf(stderr, "failed to link GPU program\n");
nuclear@0 161 return EXIT_FAILURE;
nuclear@0 162 }
nuclear@0 163
nuclear@0 164 glutMainLoop();
nuclear@0 165 return 0;
nuclear@0 166 }
nuclear@0 167
nuclear@0 168 void setup_lights(void)
nuclear@0 169 {
nuclear@0 170 float lpos[] = {-6, 8, 10, 1};
nuclear@0 171 glLightfv(GL_LIGHT0, GL_POSITION, lpos);
nuclear@0 172 }
nuclear@0 173
nuclear@0 174 void redraw(void)
nuclear@0 175 {
nuclear@0 176 glClearColor(0, 0, 0, 0);
nuclear@0 177 glClearDepth(1.0);
nuclear@0 178
nuclear@0 179 glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
nuclear@0 180
nuclear@0 181 glMatrixMode(GL_MODELVIEW);
nuclear@0 182 glLoadIdentity();
nuclear@0 183 glTranslatef(-cam_x, -cam_y, -cam_z);
nuclear@0 184 glRotatef(cam_pitch, 1, 0, 0);
nuclear@0 185 glRotatef(cam_rot, 0, 1, 0);
nuclear@0 186
nuclear@0 187 setup_lights();
nuclear@0 188
nuclear@0 189 glPushMatrix();
nuclear@0 190 glTranslatef(0, 0.8, 0);
nuclear@0 191
nuclear@0 192 bind_program(prog);
nuclear@0 193 set_uniform_float(prog, "rough", roughness);
nuclear@0 194 set_uniform_float(prog, "specularity", specularity);
nuclear@0 195 set_uniform_float(prog, "ior", ior);
nuclear@0 196 set_uniform_float(prog, "dif_sdr", dif_sdr);
nuclear@0 197 set_uniform_float(prog, "spec_sdr", spec_sdr);
nuclear@0 198 glFrontFace(GL_CW);
nuclear@0 199 set_material_color(0.55, 0.20, 0.10, 1.0);
nuclear@0 200 /*set_material_color(0.87, 0.72, 0.62, 1.0);*/
nuclear@0 201 glutSolidTeapot(1.0);
nuclear@0 202 glFrontFace(GL_CCW);
nuclear@0 203 bind_program(0);
nuclear@0 204
nuclear@0 205 glPopMatrix();
nuclear@0 206
nuclear@0 207
nuclear@0 208 glutSwapBuffers();
nuclear@0 209 }
nuclear@0 210
nuclear@0 211 void set_material_color(float r, float g, float b, float a)
nuclear@0 212 {
nuclear@0 213 float col[4];
nuclear@0 214 col[0] = r;
nuclear@0 215 col[1] = g;
nuclear@0 216 col[2] = b;
nuclear@0 217 col[3] = a;
nuclear@0 218 glMaterialfv(GL_FRONT_AND_BACK, GL_AMBIENT_AND_DIFFUSE, col);
nuclear@0 219 glMaterialf(GL_FRONT_AND_BACK, GL_SHININESS, spec_pow);
nuclear@0 220 }
nuclear@0 221
nuclear@0 222 int mod_rough, mod_spec, mod_ior, mod_spow;
nuclear@0 223
nuclear@0 224 void key_handler(unsigned char key, int x, int y)
nuclear@0 225 {
nuclear@0 226 switch(key) {
nuclear@0 227 case 27:
nuclear@0 228 exit(0);
nuclear@0 229
nuclear@0 230 case '1':
nuclear@0 231 mod_rough = 1;
nuclear@0 232 break;
nuclear@0 233
nuclear@0 234 case '2':
nuclear@0 235 mod_spec = 1;
nuclear@0 236 break;
nuclear@0 237
nuclear@0 238 case '3':
nuclear@0 239 mod_ior = 1;
nuclear@0 240 break;
nuclear@0 241
nuclear@0 242 case '4':
nuclear@0 243 mod_spow = 1;
nuclear@0 244 break;
nuclear@0 245
nuclear@0 246 default:
nuclear@0 247 break;
nuclear@0 248 }
nuclear@0 249 }
nuclear@0 250
nuclear@0 251 void key_up_handler(unsigned char key, int x, int y)
nuclear@0 252 {
nuclear@0 253 switch(key) {
nuclear@0 254 case '1':
nuclear@0 255 mod_rough = 0;
nuclear@0 256 break;
nuclear@0 257
nuclear@0 258 case '2':
nuclear@0 259 mod_spec = 0;
nuclear@0 260 break;
nuclear@0 261
nuclear@0 262 case '3':
nuclear@0 263 mod_ior = 0;
nuclear@0 264 break;
nuclear@0 265
nuclear@0 266 case '4':
nuclear@0 267 mod_spow = 0;
nuclear@0 268 break;
nuclear@0 269
nuclear@0 270 default:
nuclear@0 271 break;
nuclear@0 272 }
nuclear@0 273 }
nuclear@0 274
nuclear@0 275 void skey_handler(int key, int x, int y)
nuclear@0 276 {
nuclear@0 277 switch(key) {
nuclear@0 278 default:
nuclear@0 279 break;
nuclear@0 280 }
nuclear@0 281 }
nuclear@0 282
nuclear@0 283 static int prev_x = -1;
nuclear@0 284 static int prev_y = -1;
nuclear@0 285 static int pbn;
nuclear@0 286
nuclear@0 287 void button_handler(int button, int state, int x, int y)
nuclear@0 288 {
nuclear@0 289 if(state == GLUT_DOWN) {
nuclear@0 290 prev_x = x;
nuclear@0 291 prev_y = y;
nuclear@0 292 pbn = button;
nuclear@0 293 } else {
nuclear@0 294 prev_x = -1;
nuclear@0 295 prev_y = -1;
nuclear@0 296 }
nuclear@0 297 }
nuclear@0 298
nuclear@0 299
nuclear@0 300 #define MIN(a, b) ((a) < (b) ? (a) : (b))
nuclear@0 301 #define MAX(a, b) ((a) > (b) ? (a) : (b))
nuclear@0 302
nuclear@0 303 void mouse_handler(int x, int y)
nuclear@0 304 {
nuclear@0 305 if(pbn == GLUT_LEFT_BUTTON) {
nuclear@0 306 if(mod_rough || mod_spec || mod_ior || mod_spow) {
nuclear@0 307 float dx = (float)(x - prev_x) / (float)xres;
nuclear@0 308
nuclear@0 309 if(mod_rough) {
nuclear@0 310 roughness = MAX(0.0, MIN(roughness + dx, 1.0));
nuclear@0 311 printf("roughness: %.3f\n", roughness);
nuclear@0 312 }
nuclear@0 313 if(mod_spec) {
nuclear@0 314 specularity = MAX(0.0, MIN(specularity + dx, 1.0));
nuclear@0 315 printf("specularity: %.3f\n", specularity);
nuclear@0 316 }
nuclear@0 317 if(mod_ior) {
nuclear@0 318 ior = MAX(ior + dx, 0.0);
nuclear@0 319 printf("ior: %.3f\n", ior);
nuclear@0 320 }
nuclear@0 321 if(mod_spow) {
nuclear@0 322 spec_pow = MAX(spec_pow + dx * 50.0, 0.01);
nuclear@0 323 printf("specular power: %.3f\n", spec_pow);
nuclear@0 324 }
nuclear@0 325 } else {
nuclear@0 326 /*
nuclear@0 327 cam_x += (float)(prev_x - x) * pan_offs;
nuclear@0 328 cam_y += (float)(y - prev_y) * pan_offs;
nuclear@0 329 */
nuclear@0 330 }
nuclear@0 331
nuclear@0 332 prev_x = x;
nuclear@0 333 prev_y = y;
nuclear@0 334 glutPostRedisplay();
nuclear@0 335
nuclear@0 336 } else if(pbn == GLUT_MIDDLE_BUTTON) {
nuclear@0 337 cam_rot += (float)(x - prev_x) / 2.0f;
nuclear@0 338 cam_pitch += (float)(y - prev_y) / 2.0f;
nuclear@0 339 if(cam_pitch > 90.0) cam_pitch = 90.0;
nuclear@0 340 if(cam_pitch < -90.0) cam_pitch = -90.0;
nuclear@0 341
nuclear@0 342 prev_x = x;
nuclear@0 343 prev_y = y;
nuclear@0 344 glutPostRedisplay();
nuclear@0 345
nuclear@0 346 } else if(pbn == GLUT_RIGHT_BUTTON) {
nuclear@0 347 /*
nuclear@0 348 cam_z -= (float)(prev_y - y) / 2.0f;
nuclear@0 349 prev_y = y;
nuclear@0 350 glutPostRedisplay();
nuclear@0 351 */
nuclear@0 352 }
nuclear@0 353 }
nuclear@0 354
nuclear@0 355 void reshape(int x, int y)
nuclear@0 356 {
nuclear@0 357 glViewport(0, 0, x, y);
nuclear@0 358 xres = x;
nuclear@0 359 yres = y;
nuclear@0 360
nuclear@0 361 glMatrixMode(GL_PROJECTION);
nuclear@0 362 glLoadIdentity();
nuclear@0 363 gluPerspective(45.0, (float)xres / (float)yres, 1.0, 100.0);
nuclear@0 364 glMatrixMode(GL_MODELVIEW);
nuclear@0 365 }
nuclear@0 366
nuclear@0 367 #ifdef MOUSEWHEEL
nuclear@0 368 void wheel_handler(int unk, int dir, int x, int y)
nuclear@0 369 {
nuclear@0 370 if(dir > 0) {
nuclear@0 371 cam_z -= pan_offs * 5.0;
nuclear@0 372 } else {
nuclear@0 373 cam_z += pan_offs * 5.0;
nuclear@0 374 }
nuclear@0 375 glutPostRedisplay();
nuclear@0 376 }
nuclear@0 377 #endif
nuclear@0 378
nuclear@0 379 void menu_handler(int val)
nuclear@0 380 {
nuclear@0 381 if(val < 10) {
nuclear@0 382 dif_sdr = val;
nuclear@0 383 } else {
nuclear@0 384 spec_sdr = val - 10;
nuclear@0 385 }
nuclear@0 386 glutPostRedisplay();
nuclear@0 387 }