nuclear@0: #include <stdio.h>
nuclear@0: #include <stdlib.h>
nuclear@0: #include <math.h>
nuclear@0: #include <assert.h>
nuclear@0: 
nuclear@0: #include <GL/glew.h>
nuclear@0: #ifndef __APPLE__
nuclear@0: #include <GL/glut.h>
nuclear@0: #else
nuclear@0: #include <GLUT/glut.h>
nuclear@0: #endif
nuclear@0: 
nuclear@0: #include "cam.h"
nuclear@0: #include "sdr.h"
nuclear@0: #include "metasurf.h"
nuclear@0: 
nuclear@0: #define RES		38
nuclear@0: 
nuclear@0: struct metaball {
nuclear@0: 	float energy;
nuclear@0: 	float x, y, z;
nuclear@0: } mball[] = {
nuclear@0: 	{1.0, 0, 0, 0},
nuclear@0: 	{0.25, 0.45, 0, 0.25},
nuclear@0: 	{0.15, -0.3, 0.2, 0.1}
nuclear@0: };
nuclear@0: 
nuclear@0: int num_mballs = sizeof mball / sizeof *mball;
nuclear@0: 
nuclear@0: float eval(float x, float y, float z);
nuclear@0: float eval_cached(float x, float y, float z);
nuclear@0: void vertex(float x, float y, float z);
nuclear@0: void render(void);
nuclear@0: void disp(void);
nuclear@0: void reshape(int x, int y);
nuclear@0: void keyb(unsigned char key, int x, int y);
nuclear@0: void mouse(int bn, int state, int x, int y);
nuclear@0: void motion(int x, int y);
nuclear@0: void sball_button(int bn, int state);
nuclear@0: void sball_motion(int x, int y, int z);
nuclear@0: int parse_args(int argc, char **argv);
nuclear@0: 
nuclear@0: int stereo;
nuclear@0: struct metasurface *msurf;
nuclear@0: float threshold = 12;
nuclear@0: unsigned int sdr;
nuclear@0: int bidx = 1;
nuclear@0: 
nuclear@0: int main(int argc, char **argv)
nuclear@0: {
nuclear@0: 	float amb[] = {0, 0, 0, 0};
nuclear@0: 	float lpos[] = {-0.2, 0.2, 1, 0};
nuclear@0: 
nuclear@0: 	glutInitWindowSize(1280, 720);
nuclear@0: 	glutInit(&argc, argv);
nuclear@0: 
nuclear@0: 	if(parse_args(argc, argv) == -1) {
nuclear@0: 		return 1;
nuclear@0: 	}
nuclear@0: 	glutInitDisplayMode(GLUT_RGB | GLUT_DEPTH | GLUT_DOUBLE | (stereo ? GLUT_STEREO : 0));
nuclear@0: 	glutCreateWindow("metasurf");
nuclear@0: 
nuclear@0: 	glutDisplayFunc(disp);
nuclear@0: 	glutReshapeFunc(reshape);
nuclear@0: 	glutKeyboardFunc(keyb);
nuclear@0: 	glutMouseFunc(mouse);
nuclear@0: 	glutMotionFunc(motion);
nuclear@0: 	glutSpaceballButtonFunc(sball_button);
nuclear@0: 	glutSpaceballMotionFunc(sball_motion);
nuclear@0: 
nuclear@0: 	glewInit();
nuclear@0: 
nuclear@0: 	glEnable(GL_CULL_FACE);
nuclear@0: 	glEnable(GL_DEPTH_TEST);
nuclear@0: 
nuclear@0: 	glLightModelfv(GL_LIGHT_MODEL_AMBIENT, amb);
nuclear@0: 
nuclear@0: 	glEnable(GL_LIGHTING);
nuclear@0: 	glEnable(GL_LIGHT0);
nuclear@0: 	glLightfv(GL_LIGHT0, GL_POSITION, lpos);
nuclear@0: 
nuclear@0: 	glEnable(GL_NORMALIZE);
nuclear@0: 
nuclear@0: 	cam_focus_dist(2.0);
nuclear@0: 	cam_clip(0.1, 200.0);
nuclear@0: 	cam_rotate(0, 0);
nuclear@0: 	cam_dolly(2);
nuclear@0: 
nuclear@0: 	msurf = msurf_create();
nuclear@0: 	msurf_eval_func(msurf, eval);
nuclear@0: 	msurf_vertex_func(msurf, vertex);
nuclear@0: 	msurf_threshold(msurf, threshold);
nuclear@0: 	msurf_resolution(msurf, RES, RES, RES);
nuclear@0: 	msurf_bounds(msurf, -1, -1, -1, 1, 1, 1);
nuclear@0: 
nuclear@0: 	glClearColor(0.8, 0.8, 0.8, 1.0);
nuclear@0: 
nuclear@0: 	if(!(sdr = create_program_load("sdr/vert.glsl", "sdr/frag.glsl"))) {
nuclear@0: 		return 1;
nuclear@0: 	}
nuclear@0: 
nuclear@0: 	glutMainLoop();
nuclear@0: 	return 0;
nuclear@0: }
nuclear@0: 
nuclear@0: float eval(float x, float y, float z)
nuclear@0: {
nuclear@0: 	int i;
nuclear@0: 	float val = 0.0f;
nuclear@0: 
nuclear@0: 	for(i=0; i<num_mballs; i++) {
nuclear@0: 		float dx = mball[i].x - x;
nuclear@0: 		float dy = mball[i].y - y;
nuclear@0: 		float dz = mball[i].z - z;
nuclear@0: 		float dist_sq = dx * dx + dy * dy + dz * dz;
nuclear@0: 
nuclear@0: 		if(dist_sq < 1e-6) {
nuclear@0: 			val += 100.0;
nuclear@0: 		} else {
nuclear@0: 			val += mball[i].energy / dist_sq;
nuclear@0: 		}
nuclear@0: 	}
nuclear@0: 	return val;
nuclear@0: }
nuclear@0: 
nuclear@0: void vertex(float x, float y, float z)
nuclear@0: {
nuclear@0: 	const float dt = 0.001;
nuclear@0: 	float dfdx = eval(x - dt, y, z) - eval(x + dt, y, z);
nuclear@0: 	float dfdy = eval(x, y - dt, z) - eval(x, y + dt, z);
nuclear@0: 	float dfdz = eval(x, y, z - dt) - eval(x, y, z + dt);
nuclear@0: 
nuclear@0: 	glNormal3f(dfdx, dfdy, dfdz);
nuclear@0: 	glVertex3f(x, y, z);
nuclear@0: }
nuclear@0: 
nuclear@0: void render(void)
nuclear@0: {
nuclear@0: 	float kd[] = {0.7, 0.28, 0.2, 1.0};
nuclear@0: 	float ks[] = {0.9, 0.9, 0.9, 1.0};
nuclear@0: 
nuclear@0: 	glMaterialfv(GL_FRONT_AND_BACK, GL_AMBIENT_AND_DIFFUSE, kd);
nuclear@0: 	glMaterialfv(GL_FRONT_AND_BACK, GL_SPECULAR, ks);
nuclear@0: 	glMaterialf(GL_FRONT_AND_BACK, GL_SHININESS, 60.0);
nuclear@0: 
nuclear@0: 	bind_program(sdr);
nuclear@0: 
nuclear@0: 	glBegin(GL_TRIANGLES);
nuclear@0: 	msurf_polygonize(msurf);
nuclear@0: 	glEnd();
nuclear@0: 
nuclear@0: 	assert(glGetError() == GL_NO_ERROR);
nuclear@0: }
nuclear@0: 
nuclear@0: void disp(void)
nuclear@0: {
nuclear@0: 	if(stereo) {
nuclear@0: 		glDrawBuffer(GL_BACK_LEFT);
nuclear@0: 	}
nuclear@0: 
nuclear@0: 	glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
nuclear@0: 
nuclear@0: 	glMatrixMode(GL_PROJECTION);
nuclear@0: 	glLoadIdentity();
nuclear@0: 	cam_stereo_proj_matrix(stereo ? CAM_LEFT : CAM_CENTER);
nuclear@0: 
nuclear@0: 	glMatrixMode(GL_MODELVIEW);
nuclear@0: 	glLoadIdentity();
nuclear@0: 	cam_stereo_view_matrix(stereo ? CAM_LEFT : CAM_CENTER);
nuclear@0: 
nuclear@0: 	render();
nuclear@0: 
nuclear@0: 	if(stereo) {
nuclear@0: 		glDrawBuffer(GL_BACK_RIGHT);
nuclear@0: 		glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
nuclear@0: 
nuclear@0: 		glMatrixMode(GL_PROJECTION);
nuclear@0: 		glLoadIdentity();
nuclear@0: 		cam_stereo_proj_matrix(CAM_RIGHT);
nuclear@0: 
nuclear@0: 		glMatrixMode(GL_MODELVIEW);
nuclear@0: 		glLoadIdentity();
nuclear@0: 		cam_stereo_view_matrix(CAM_RIGHT);
nuclear@0: 
nuclear@0: 		render();
nuclear@0: 	}
nuclear@0: 	glutSwapBuffers();
nuclear@0: }
nuclear@0: 
nuclear@0: void reshape(int x, int y)
nuclear@0: {
nuclear@0: 	glViewport(0, 0, x, y);
nuclear@0: 	cam_aspect((float)x / (float)y);
nuclear@0: }
nuclear@0: 
nuclear@0: void keyb(unsigned char key, int x, int y)
nuclear@0: {
nuclear@0: 	static int wire;
nuclear@0: 
nuclear@0: 	switch(key) {
nuclear@0: 	case 27:
nuclear@0: 		exit(0);
nuclear@0: 
nuclear@0: 	case 's':
nuclear@0: 		stereo = !stereo;
nuclear@0: 		glutPostRedisplay();
nuclear@0: 		break;
nuclear@0: 
nuclear@0: 	case 'w':
nuclear@0: 		wire = !wire;
nuclear@0: 		glPolygonMode(GL_FRONT_AND_BACK, wire ? GL_LINE : GL_FILL);
nuclear@0: 		glutPostRedisplay();
nuclear@0: 		break;
nuclear@0: 
nuclear@0: 	case '=':
nuclear@0: 		threshold += 0.05;
nuclear@0: 		msurf_threshold(msurf, threshold);
nuclear@0: 		printf("threshold: %f\n", threshold);
nuclear@0: 		glutPostRedisplay();
nuclear@0: 		break;
nuclear@0: 
nuclear@0: 	case '-':
nuclear@0: 		threshold -= 0.05;
nuclear@0: 		msurf_threshold(msurf, threshold);
nuclear@0: 		printf("threshold: %f\n", threshold);
nuclear@0: 		glutPostRedisplay();
nuclear@0: 		break;
nuclear@0: 
nuclear@0: 	case ' ':
nuclear@0: 		bidx = (bidx + 1) % num_mballs;
nuclear@0: 		break;
nuclear@0: 
nuclear@0: 	default:
nuclear@0: 		break;
nuclear@0: 	}
nuclear@0: }
nuclear@0: 
nuclear@0: int bnstate[32];
nuclear@0: int prev_x, prev_y;
nuclear@0: 
nuclear@0: void mouse(int bn, int state, int x, int y)
nuclear@0: {
nuclear@0: 	bnstate[bn] = state == GLUT_DOWN;
nuclear@0: 	prev_x = x;
nuclear@0: 	prev_y = y;
nuclear@0: }
nuclear@0: 
nuclear@0: void motion(int x, int y)
nuclear@0: {
nuclear@0: 	int dx, dy;
nuclear@0: 
nuclear@0: 	dx = x - prev_x;
nuclear@0: 	dy = y - prev_y;
nuclear@0: 	prev_x = x;
nuclear@0: 	prev_y = y;
nuclear@0: 
nuclear@0: 	if(bnstate[GLUT_LEFT_BUTTON]) {
nuclear@0: 		cam_inp_rotate(dx, dy);
nuclear@0: 		glutPostRedisplay();
nuclear@0: 	}
nuclear@0: 	if(bnstate[GLUT_RIGHT_BUTTON]) {
nuclear@0: 		cam_inp_zoom(dy);
nuclear@0: 		glutPostRedisplay();
nuclear@0: 	}
nuclear@0: }
nuclear@0: 
nuclear@0: void sball_button(int bn, int state)
nuclear@0: {
nuclear@0: 	if(state) return;
nuclear@0: 
nuclear@0: 	if(bn < num_mballs) {
nuclear@0: 		bidx = bn;
nuclear@0: 	} else {
nuclear@0: 		bidx = (bidx + 1) % num_mballs;
nuclear@0: 	}
nuclear@0: }
nuclear@0: 
nuclear@0: void sball_motion(int x, int y, int z)
nuclear@0: {
nuclear@0: 	mball[bidx].x += (float)x / 32768.0;
nuclear@0: 	mball[bidx].y += (float)y / 32768.0;
nuclear@0: 	mball[bidx].z -= (float)z / 32768.0;
nuclear@0: 	glutPostRedisplay();
nuclear@0: }
nuclear@0: 
nuclear@0: int parse_args(int argc, char **argv)
nuclear@0: {
nuclear@0: 	int 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 's':
nuclear@0: 				stereo = !stereo;
nuclear@0: 				break;
nuclear@0: 
nuclear@0: 			default:
nuclear@0: 				fprintf(stderr, "unrecognized option: %s\n", argv[i]);
nuclear@0: 				return -1;
nuclear@0: 			}
nuclear@0: 		} else {
nuclear@0: 			fprintf(stderr, "unexpected argument: %s\n", argv[i]);
nuclear@0: 			return -1;
nuclear@0: 		}
nuclear@0: 	}
nuclear@0: 	return 0;
nuclear@0: }