nuclear@0: #include <stdio.h>
nuclear@0: #include <stdlib.h>
nuclear@0: #include <string.h>
nuclear@0: #include <GL/glut.h>
nuclear@0: #include "sdr.h"
nuclear@0: 
nuclear@0: #ifdef FREEGLUT
nuclear@0: #include <GL/freeglut_ext.h>
nuclear@0: #define MOUSEWHEEL
nuclear@0: #endif
nuclear@0: 
nuclear@0: void redraw(void);
nuclear@0: void set_material_color(float r, float g, float b, float a);
nuclear@0: void setup_lights(void);
nuclear@0: void key_handler(unsigned char key, int x, int y);
nuclear@0: void key_up_handler(unsigned char key, int x, int y);
nuclear@0: void skey_handler(int key, int x, int y);
nuclear@0: void button_handler(int bn, int state, int x, int y);
nuclear@0: void mouse_handler(int x, int y);
nuclear@0: void reshape(int x, int y);
nuclear@0: #ifdef MOUSEWHEEL
nuclear@0: void wheel_handler(int unk, int dir, int x, int y);
nuclear@0: #endif
nuclear@0: void menu_handler(int val);
nuclear@0: void init_sdr(void);
nuclear@0: 
nuclear@0: int xres, yres;
nuclear@0: 
nuclear@0: float cam_x, cam_y = 0.6, cam_z = 4;
nuclear@0: float cam_rot, cam_pitch = 35.0;
nuclear@0: 
nuclear@0: float pan_offs = 0.025;
nuclear@0: 
nuclear@0: unsigned int prog;
nuclear@0: 
nuclear@0: float roughness = 0.24;
nuclear@0: float specularity = 0.86;
nuclear@0: float ior = 3.2;
nuclear@0: float spec_pow = 60.0;
nuclear@0: 
nuclear@0: enum {DSDR_LAMBERT, DSDR_OREN_NAYAR};
nuclear@0: enum {SSDR_PHONG, SSDR_BLINN, SSDR_COOK_TORR};
nuclear@0: 
nuclear@0: int dif_sdr = DSDR_OREN_NAYAR;
nuclear@0: int spec_sdr = SSDR_COOK_TORR;
nuclear@0: 
nuclear@0: int main(int argc, char **argv)
nuclear@0: {
nuclear@0: 	unsigned int vs, ps, pslib;
nuclear@0: 	int dif_menu, spec_menu, i;
nuclear@0: 
nuclear@0: 	for(i=1; i<argc; i++) {
nuclear@0: 		if(argv[i][0] == '-' && argv[i][2] == 0) {
nuclear@0: 			switch(argv[i][1]) {
nuclear@0: 			case 'd':
nuclear@0: 				if(strcmp(argv[++i], "lambert") == 0) {
nuclear@0: 					dif_sdr = DSDR_LAMBERT;
nuclear@0: 				} else if(strcmp(argv[i], "oren-nayar") == 0) {
nuclear@0: 					dif_sdr = DSDR_OREN_NAYAR;
nuclear@0: 				} else {
nuclear@0: 					printf("-d (diffuse model) choices:\n");
nuclear@0: 					printf("  lambert        \"Photometria sive de mensura de gratibus luminis, colorum et umbrae\", 1760\n");
nuclear@0: 					printf("  oren-nayar     \"Generalization of Lambert's Reflectance Model\", SIGGRAPH 1994\n");
nuclear@0: 					return EXIT_FAILURE;
nuclear@0: 				}
nuclear@0: 				break;
nuclear@0: 
nuclear@0: 			case 's':
nuclear@0: 				if(strcmp(argv[++i], "phong") == 0) {
nuclear@0: 					spec_sdr = SSDR_PHONG;
nuclear@0: 				} else if(strcmp(argv[i], "cook-torrance") == 0) {
nuclear@0: 					spec_sdr = SSDR_COOK_TORR;
nuclear@0: 				} else {
nuclear@0: 					printf("-s (specular model) choices:\n");
nuclear@0: 					printf("  phong          \"Illumination for Computer Generated Pictures\", CACM 1975\n");
nuclear@0: 					printf("  blinn          \"Models of Light Reflection for Computer Synthesized Pictures\", SIGGRAPH 1977\n");
nuclear@0: 					printf("  cook-torrance  \"A Reflectance Model for Computer Graphics\", SIGGRAPH 1981\n");
nuclear@0: 					return EXIT_FAILURE;
nuclear@0: 				}
nuclear@0: 				break;
nuclear@0: 
nuclear@0: 			default:
nuclear@0: 				fprintf(stderr, "unrecognized argument: %s\n", argv[i]);
nuclear@0: 				return EXIT_FAILURE;
nuclear@0: 			}
nuclear@0: 		}
nuclear@0: 	}
nuclear@0: 
nuclear@0: 	glutInitWindowSize(800, 600);
nuclear@0: 	glutInit(&argc, argv);
nuclear@0: 	glutInitDisplayMode(GLUT_RGBA | GLUT_DEPTH | GLUT_DOUBLE);
nuclear@0: 	glutCreateWindow("1.roughness  2.specularity  3.ior  4.phong power");
nuclear@0: 	xres = glutGet(GLUT_WINDOW_WIDTH);
nuclear@0: 	yres = glutGet(GLUT_WINDOW_HEIGHT);
nuclear@0: 
nuclear@0: 	glutKeyboardFunc(key_handler);
nuclear@0: 	glutKeyboardUpFunc(key_up_handler);
nuclear@0: 	glutSpecialFunc(skey_handler);
nuclear@0: 	glutMotionFunc(mouse_handler);
nuclear@0: 	glutMouseFunc(button_handler);
nuclear@0: 	glutReshapeFunc(reshape);
nuclear@0: 	glutDisplayFunc(redraw);
nuclear@0: #ifdef MOUSEWHEEL
nuclear@0: 	glutMouseWheelFunc(wheel_handler);
nuclear@0: #endif
nuclear@0: 
nuclear@0: 	/* create the menus */
nuclear@0: 	dif_menu = glutCreateMenu(menu_handler);
nuclear@0: 	glutAddMenuEntry("lambert", DSDR_LAMBERT);
nuclear@0: 	glutAddMenuEntry("oren-nayar", DSDR_OREN_NAYAR);
nuclear@0: 
nuclear@0: 	spec_menu = glutCreateMenu(menu_handler);
nuclear@0: 	glutAddMenuEntry("phong", SSDR_PHONG + 10);
nuclear@0: 	glutAddMenuEntry("blinn", SSDR_BLINN + 10);
nuclear@0: 	glutAddMenuEntry("cook-torrance", SSDR_COOK_TORR + 10);
nuclear@0: 
nuclear@0: 	glutCreateMenu(menu_handler);
nuclear@0: 	glutAddSubMenu("diffuse models", dif_menu);
nuclear@0: 	glutAddSubMenu("specular models", spec_menu);
nuclear@0: 
nuclear@0: 	glutAttachMenu(GLUT_RIGHT_BUTTON);
nuclear@0: 
nuclear@0: 
nuclear@0: 	glEnable(GL_DEPTH_TEST);
nuclear@0: 	glEnable(GL_CULL_FACE);
nuclear@0: 	
nuclear@0: 	glEnable(GL_LIGHTING);
nuclear@0: 	glEnable(GL_LIGHT0);
nuclear@0: 
nuclear@0: 	glMatrixMode(GL_PROJECTION);
nuclear@0: 	gluPerspective(45.0, (float)xres / (float)yres, 1.0, 100.0);
nuclear@0: 	glMatrixMode(GL_MODELVIEW);
nuclear@0: 
nuclear@0: 	init_sdr();
nuclear@0: 
nuclear@0: 	if(!(vs = load_vertex_shader("sdr.vs.glsl"))) {
nuclear@0: 		fprintf(stderr, "failed to load vertex shader\n");
nuclear@0: 		return EXIT_FAILURE;
nuclear@0: 	}
nuclear@0: 
nuclear@0: 	if(!(pslib = load_pixel_shader("sdr_ref_models.glsl"))) {
nuclear@0: 		fprintf(stderr, "failed to load reflection models shader\n");
nuclear@0: 		return EXIT_FAILURE;
nuclear@0: 	}
nuclear@0: 
nuclear@0: 	if(!(ps = load_pixel_shader("sdr.ps.glsl"))) {
nuclear@0: 		fprintf(stderr, "failed to load pixel shader\n");
nuclear@0: 		return EXIT_FAILURE;
nuclear@0: 	}
nuclear@0: 
nuclear@0: 	if(!(prog = create_program())) {
nuclear@0: 		fprintf(stderr, "failed to create GPU program\n");
nuclear@0: 		return EXIT_FAILURE;
nuclear@0: 	}
nuclear@0: 	attach_shader(prog, vs);
nuclear@0: 	attach_shader(prog, pslib);
nuclear@0: 	attach_shader(prog, ps);
nuclear@0: 
nuclear@0: 	if(link_program(prog) == -1) {
nuclear@0: 		fprintf(stderr, "failed to link GPU program\n");
nuclear@0: 		return EXIT_FAILURE;
nuclear@0: 	}
nuclear@0: 
nuclear@0: 	glutMainLoop();
nuclear@0: 	return 0;
nuclear@0: }
nuclear@0: 
nuclear@0: void setup_lights(void)
nuclear@0: {
nuclear@0: 	float lpos[] = {-6, 8, 10, 1};
nuclear@0: 	glLightfv(GL_LIGHT0, GL_POSITION, lpos);
nuclear@0: }
nuclear@0: 
nuclear@0: void redraw(void)
nuclear@0: {
nuclear@0: 	glClearColor(0, 0, 0, 0);
nuclear@0: 	glClearDepth(1.0);
nuclear@0: 
nuclear@0: 	glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
nuclear@0: 
nuclear@0: 	glMatrixMode(GL_MODELVIEW);
nuclear@0: 	glLoadIdentity();
nuclear@0: 	glTranslatef(-cam_x, -cam_y, -cam_z);
nuclear@0: 	glRotatef(cam_pitch, 1, 0, 0);
nuclear@0: 	glRotatef(cam_rot, 0, 1, 0);
nuclear@0: 
nuclear@0: 	setup_lights();
nuclear@0: 
nuclear@0: 	glPushMatrix();
nuclear@0: 	glTranslatef(0, 0.8, 0);
nuclear@0: 
nuclear@0: 	bind_program(prog);
nuclear@0: 	set_uniform_float(prog, "rough", roughness);
nuclear@0: 	set_uniform_float(prog, "specularity", specularity);
nuclear@0: 	set_uniform_float(prog, "ior", ior);
nuclear@0: 	set_uniform_float(prog, "dif_sdr", dif_sdr);
nuclear@0: 	set_uniform_float(prog, "spec_sdr", spec_sdr);
nuclear@0: 	glFrontFace(GL_CW);
nuclear@0: 	set_material_color(0.55, 0.20, 0.10, 1.0);
nuclear@0: 	/*set_material_color(0.87, 0.72, 0.62, 1.0);*/
nuclear@0: 	glutSolidTeapot(1.0);
nuclear@0: 	glFrontFace(GL_CCW);
nuclear@0: 	bind_program(0);
nuclear@0: 
nuclear@0: 	glPopMatrix();
nuclear@0: 	
nuclear@0: 
nuclear@0: 	glutSwapBuffers();
nuclear@0: }
nuclear@0: 
nuclear@0: void set_material_color(float r, float g, float b, float a)
nuclear@0: {
nuclear@0: 	float col[4];
nuclear@0: 	col[0] = r;
nuclear@0: 	col[1] = g;
nuclear@0: 	col[2] = b;
nuclear@0: 	col[3] = a;
nuclear@0: 	glMaterialfv(GL_FRONT_AND_BACK, GL_AMBIENT_AND_DIFFUSE, col);
nuclear@0: 	glMaterialf(GL_FRONT_AND_BACK, GL_SHININESS, spec_pow);
nuclear@0: }
nuclear@0: 
nuclear@0: int mod_rough, mod_spec, mod_ior, mod_spow;
nuclear@0: 
nuclear@0: void key_handler(unsigned char key, int x, int y)
nuclear@0: {
nuclear@0: 	switch(key) {
nuclear@0: 	case 27:
nuclear@0: 		exit(0);
nuclear@0: 
nuclear@0: 	case '1':
nuclear@0: 		mod_rough = 1;
nuclear@0: 		break;
nuclear@0: 
nuclear@0: 	case '2':
nuclear@0: 		mod_spec = 1;
nuclear@0: 		break;
nuclear@0: 
nuclear@0: 	case '3':
nuclear@0: 		mod_ior = 1;
nuclear@0: 		break;
nuclear@0: 
nuclear@0: 	case '4':
nuclear@0: 		mod_spow = 1;
nuclear@0: 		break;
nuclear@0: 
nuclear@0: 	default:
nuclear@0: 		break;
nuclear@0: 	}
nuclear@0: }
nuclear@0: 
nuclear@0: void key_up_handler(unsigned char key, int x, int y)
nuclear@0: {
nuclear@0: 	switch(key) {
nuclear@0: 	case '1':
nuclear@0: 		mod_rough = 0;
nuclear@0: 		break;
nuclear@0: 
nuclear@0: 	case '2':
nuclear@0: 		mod_spec = 0;
nuclear@0: 		break;
nuclear@0: 
nuclear@0: 	case '3':
nuclear@0: 		mod_ior = 0;
nuclear@0: 		break;
nuclear@0: 
nuclear@0: 	case '4':
nuclear@0: 		mod_spow = 0;
nuclear@0: 		break;
nuclear@0: 
nuclear@0: 	default:
nuclear@0: 		break;
nuclear@0: 	}
nuclear@0: }
nuclear@0: 
nuclear@0: void skey_handler(int key, int x, int y)
nuclear@0: {
nuclear@0: 	switch(key) {
nuclear@0: 	default:
nuclear@0: 		break;
nuclear@0: 	}
nuclear@0: }
nuclear@0: 
nuclear@0: static int prev_x = -1;
nuclear@0: static int prev_y = -1;
nuclear@0: static int pbn;
nuclear@0: 
nuclear@0: void button_handler(int button, int state, int x, int y)
nuclear@0: {
nuclear@0: 	if(state == GLUT_DOWN) {
nuclear@0: 		prev_x = x;
nuclear@0: 		prev_y = y;
nuclear@0: 		pbn = button;
nuclear@0: 	} else {
nuclear@0: 		prev_x = -1;
nuclear@0: 		prev_y = -1;
nuclear@0: 	}
nuclear@0: }
nuclear@0: 
nuclear@0: 
nuclear@0: #define MIN(a, b)		((a) < (b) ? (a) : (b))
nuclear@0: #define MAX(a, b)		((a) > (b) ? (a) : (b))
nuclear@0: 
nuclear@0: void mouse_handler(int x, int y)
nuclear@0: {
nuclear@0: 	if(pbn == GLUT_LEFT_BUTTON) {
nuclear@0: 		if(mod_rough || mod_spec || mod_ior || mod_spow) {
nuclear@0: 			float dx = (float)(x - prev_x) / (float)xres;
nuclear@0: 
nuclear@0: 			if(mod_rough) {
nuclear@0: 				roughness = MAX(0.0, MIN(roughness + dx, 1.0));
nuclear@0: 				printf("roughness: %.3f\n", roughness);
nuclear@0: 			}
nuclear@0: 			if(mod_spec) {
nuclear@0: 				specularity = MAX(0.0, MIN(specularity + dx, 1.0));
nuclear@0: 				printf("specularity: %.3f\n", specularity);
nuclear@0: 			}
nuclear@0: 			if(mod_ior) {
nuclear@0: 				ior = MAX(ior + dx, 0.0);
nuclear@0: 				printf("ior: %.3f\n", ior);
nuclear@0: 			}
nuclear@0: 			if(mod_spow) {
nuclear@0: 				spec_pow = MAX(spec_pow + dx * 50.0, 0.01);
nuclear@0: 				printf("specular power: %.3f\n", spec_pow);
nuclear@0: 			}
nuclear@0: 		} else {
nuclear@0: 			/*
nuclear@0: 			cam_x += (float)(prev_x - x) * pan_offs;
nuclear@0: 			cam_y += (float)(y - prev_y) * pan_offs;
nuclear@0: 			*/
nuclear@0: 		}
nuclear@0: 
nuclear@0: 		prev_x = x;
nuclear@0: 		prev_y = y;
nuclear@0: 		glutPostRedisplay();
nuclear@0: 
nuclear@0: 	} else if(pbn == GLUT_MIDDLE_BUTTON) {
nuclear@0: 		cam_rot += (float)(x - prev_x) / 2.0f;
nuclear@0: 		cam_pitch += (float)(y - prev_y) / 2.0f;
nuclear@0: 		if(cam_pitch > 90.0) cam_pitch = 90.0;
nuclear@0: 		if(cam_pitch < -90.0) cam_pitch = -90.0;
nuclear@0: 		
nuclear@0: 		prev_x = x;
nuclear@0: 		prev_y = y;
nuclear@0: 		glutPostRedisplay();
nuclear@0: 
nuclear@0: 	} else if(pbn == GLUT_RIGHT_BUTTON) {
nuclear@0: 		/*
nuclear@0: 		cam_z -= (float)(prev_y - y) / 2.0f;
nuclear@0: 		prev_y = y;
nuclear@0: 		glutPostRedisplay();
nuclear@0: 		*/
nuclear@0: 	}
nuclear@0: }
nuclear@0: 
nuclear@0: void reshape(int x, int y)
nuclear@0: {
nuclear@0: 	glViewport(0, 0, x, y);
nuclear@0: 	xres = x;
nuclear@0: 	yres = y;
nuclear@0: 
nuclear@0: 	glMatrixMode(GL_PROJECTION);
nuclear@0: 	glLoadIdentity();
nuclear@0: 	gluPerspective(45.0, (float)xres / (float)yres, 1.0, 100.0);
nuclear@0: 	glMatrixMode(GL_MODELVIEW);
nuclear@0: }
nuclear@0: 
nuclear@0: #ifdef MOUSEWHEEL
nuclear@0: void wheel_handler(int unk, int dir, int x, int y)
nuclear@0: {
nuclear@0: 	if(dir > 0) {
nuclear@0: 		cam_z -= pan_offs * 5.0;
nuclear@0: 	} else {
nuclear@0: 		cam_z += pan_offs * 5.0;
nuclear@0: 	}
nuclear@0: 	glutPostRedisplay();
nuclear@0: }
nuclear@0: #endif
nuclear@0: 
nuclear@0: void menu_handler(int val)
nuclear@0: {
nuclear@0: 	if(val < 10) {
nuclear@0: 		dif_sdr = val;
nuclear@0: 	} else {
nuclear@0: 		spec_sdr = val - 10;
nuclear@0: 	}
nuclear@0: 	glutPostRedisplay();
nuclear@0: }