nuclear@0: #include <stdio.h>
nuclear@0: #include <stdlib.h>
nuclear@2: #include <string.h>
nuclear@1: #include <math.h>
nuclear@2: #include <errno.h>
nuclear@2: #include <dirent.h>
nuclear@0: #include <GL/glew.h>
nuclear@0: #include <GL/glut.h>
nuclear@0: #include "sdr.h"
nuclear@0: 
nuclear@3: enum {
nuclear@3: 	SDR_LINE_DIST_BIT = 1,
nuclear@3: 	SDR_PLOT_ORBIT_BIT = 2
nuclear@3: };
nuclear@3: const char *sdrdef[] = {
nuclear@3: 	"#define LINE_DIST\n",
nuclear@3: 	"#define PLOT_ORBIT\n"
nuclear@3: };
nuclear@3: #define NUM_SDR_BITS	(sizeof sdrdef / sizeof *sdrdef)
nuclear@3: #define NUM_SHADERS		(1 << NUM_SDR_BITS)
nuclear@3: 
nuclear@0: int init(void);
nuclear@0: void cleanup(void);
nuclear@0: void disp(void);
nuclear@0: void reshape(int x, int y);
nuclear@1: void keydown(unsigned char key, int x, int y);
nuclear@1: void keyup(unsigned char key, int x, int y);
nuclear@0: void mouse(int bn, int st, int x, int y);
nuclear@0: void motion(int x, int y);
nuclear@2: void screenshot(void);
nuclear@0: 
nuclear@1: static int win_width = 1280, win_height = 800;
nuclear@1: static float win_aspect;
nuclear@1: static int mouse_x = 640, mouse_y = 400;
nuclear@3: static unsigned int prog_mbrot[NUM_SHADERS];
nuclear@0: 
nuclear@1: static float view_center[2] = {0.7, 0.0};
nuclear@1: static float view_scale = 1.2;
nuclear@1: 
nuclear@3: static unsigned int sdrflags;
nuclear@1: 
nuclear@0: int main(int argc, char **argv)
nuclear@0: {
nuclear@0: 	glutInit(&argc, argv);
nuclear@0: 	glutInitWindowSize(1280, 800);
nuclear@0: 	glutInitDisplayMode(GLUT_RGB | GLUT_DOUBLE);
nuclear@0: 	glutCreateWindow("fractorb");
nuclear@0: 
nuclear@0: 	glutDisplayFunc(disp);
nuclear@0: 	glutReshapeFunc(reshape);
nuclear@1: 	glutKeyboardFunc(keydown);
nuclear@1: 	glutKeyboardUpFunc(keyup);
nuclear@0: 	glutMouseFunc(mouse);
nuclear@0: 	glutMotionFunc(motion);
nuclear@0: 
nuclear@0: 	if(init() == -1) {
nuclear@0: 		return 1;
nuclear@0: 	}
nuclear@0: 	atexit(cleanup);
nuclear@0: 
nuclear@0: 	glutMainLoop();
nuclear@0: 	return 0;
nuclear@0: }
nuclear@0: 
nuclear@0: 
nuclear@0: int init(void)
nuclear@0: {
nuclear@3: 	int i, j;
nuclear@3: 	unsigned int vs = 0, ps[NUM_SHADERS] = {0};
nuclear@3: 
nuclear@0: 	glewInit();
nuclear@0: 
nuclear@3: 	if(!(vs = load_vertex_shader("vertex.glsl"))) {
nuclear@0: 		return -1;
nuclear@0: 	}
nuclear@3: 
nuclear@3: 	for(i=0; i<NUM_SHADERS; i++) {
nuclear@3: 		clear_shader_header(GL_FRAGMENT_SHADER);
nuclear@3: 		for(j=0; j<NUM_SDR_BITS; j++) {
nuclear@3: 			if(i & (1 << j)) {
nuclear@3: 				add_shader_header(GL_FRAGMENT_SHADER, sdrdef[j]);
nuclear@3: 			}
nuclear@3: 		}
nuclear@3: 
nuclear@3: 		if(!(ps[i] = load_pixel_shader("mbrot.glsl"))) {
nuclear@3: 			goto err;
nuclear@3: 		}
nuclear@3: 
nuclear@3: 		if(!(prog_mbrot[i] = create_program_link(vs, ps[i], 0))) {
nuclear@3: 			goto err;
nuclear@3: 		}
nuclear@3: 	}
nuclear@0: 	return 0;
nuclear@3: 
nuclear@3: err:
nuclear@3: 	for(i=0; i<NUM_SHADERS; i++) {
nuclear@3: 		if(prog_mbrot[i]) {
nuclear@3: 			free_program(prog_mbrot[i]);
nuclear@3: 			prog_mbrot[i] = 0;
nuclear@3: 		}
nuclear@3: 		if(ps[i]) {
nuclear@3: 			free_shader(ps[i]);
nuclear@3: 		}
nuclear@3: 	}
nuclear@3: 	if(vs) {
nuclear@3: 		free_shader(vs);
nuclear@3: 	}
nuclear@3: 	return -1;
nuclear@0: }
nuclear@0: 
nuclear@0: void cleanup(void)
nuclear@0: {
nuclear@3: 	int i;
nuclear@3: 
nuclear@3: 	for(i=0; i<NUM_SHADERS; i++) {
nuclear@3: 		free_program(prog_mbrot[i]);
nuclear@3: 		prog_mbrot[i] = 0;
nuclear@3: 	}
nuclear@0: }
nuclear@0: 
nuclear@1: void pixel_to_complex(float *res, float px, float py)
nuclear@1: {
nuclear@1: 	float u = (2.0 * px / win_width - 1.0) * win_aspect;
nuclear@1: 	float v = 2.0 * py / win_height - 1.0;
nuclear@1: 	res[0] = u * view_scale - view_center[0];
nuclear@1: 	res[1] = v * view_scale - view_center[1];
nuclear@1: }
nuclear@1: 
nuclear@0: void disp(void)
nuclear@0: {
nuclear@1: 	int i;
nuclear@1: 	float seed[2];
nuclear@1: 	static const float verts[][2] = {{-1, -1}, {1, -1}, {1, 1}, {-1, 1}};
nuclear@1: 	static const float corners[][2] = {{0, 1}, {1, 1}, {1, 0}, {0, 0}};
nuclear@1: 
nuclear@1: 	pixel_to_complex(seed, mouse_x, mouse_y);
nuclear@1: 
nuclear@3: 	set_uniform_float2(prog_mbrot[sdrflags], "seed", seed[0], seed[1]);
nuclear@3: 	set_uniform_float(prog_mbrot[sdrflags], "view_scale", view_scale);
nuclear@3: 	set_uniform_float2(prog_mbrot[sdrflags], "view_center", view_center[0], view_center[1]);
nuclear@3: 	glUseProgram(prog_mbrot[sdrflags]);
nuclear@0: 
nuclear@0: 	glBegin(GL_QUADS);
nuclear@1: 	for(i=0; i<4; i++) {
nuclear@1: 		float bv[2];
nuclear@1: 		pixel_to_complex(bv, (win_width - 1) * corners[i][0], (win_height - 1) * corners[i][1]);
nuclear@1: 		glTexCoord2f(bv[0], bv[1]);
nuclear@1: 		glVertex2f(verts[i][0], verts[i][1]);
nuclear@1: 	}
nuclear@0: 	glEnd();
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: 
nuclear@1: 	win_aspect = (float)x / (float)y;
nuclear@1: 	win_width = x;
nuclear@1: 	win_height = y;
nuclear@0: }
nuclear@0: 
nuclear@1: static int keystate[256];
nuclear@1: 
nuclear@1: void keydown(unsigned char key, int x, int y)
nuclear@0: {
nuclear@1: 	static int fullscreen;
nuclear@1: 	static int prev_width, prev_height;
nuclear@1: 
nuclear@1: 	switch(key) {
nuclear@1: 	case 27:
nuclear@0: 		exit(0);
nuclear@1: 
nuclear@1: 	case 'f':
nuclear@1: 		fullscreen = !fullscreen;
nuclear@1: 		if(fullscreen) {
nuclear@1: 			prev_width = win_width;
nuclear@1: 			prev_height = win_height;
nuclear@1: 			glutFullScreen();
nuclear@1: 		} else {
nuclear@1: 			glutReshapeWindow(prev_width, prev_height);
nuclear@1: 		}
nuclear@1: 		break;
nuclear@1: 
nuclear@3: 	case ' ':
nuclear@3: 		sdrflags ^= SDR_LINE_DIST_BIT;
nuclear@3: 		glutPostRedisplay();
nuclear@3: 		break;
nuclear@3: 
nuclear@3: 	case 'o':
nuclear@3: 		sdrflags ^= SDR_PLOT_ORBIT_BIT;
nuclear@3: 		glutPostRedisplay();
nuclear@3: 		break;
nuclear@3: 
nuclear@2: 	case '`':
nuclear@2: 		screenshot();
nuclear@2: 		break;
nuclear@2: 
nuclear@1: 	default:
nuclear@1: 		break;
nuclear@0: 	}
nuclear@1: 	keystate[key] = 1;
nuclear@0: }
nuclear@0: 
nuclear@1: void keyup(unsigned char key, int x, int y)
nuclear@1: {
nuclear@1: 	keystate[key] = 0;
nuclear@1: }
nuclear@1: 
nuclear@1: static int prev_x, prev_y;
nuclear@1: static int bnstate[8];
nuclear@1: 
nuclear@0: void mouse(int bn, int st, int x, int y)
nuclear@0: {
nuclear@1: 	prev_x = x;
nuclear@1: 	prev_y = y;
nuclear@1: 	bnstate[bn - GLUT_LEFT_BUTTON] = st == GLUT_DOWN ? 1 : 0;
nuclear@0: }
nuclear@0: 
nuclear@0: void motion(int x, int y)
nuclear@0: {
nuclear@1: 	int dx = x - prev_x;
nuclear@1: 	int dy = y - prev_y;
nuclear@1: 	prev_x = x;
nuclear@1: 	prev_y = y;
nuclear@1: 
nuclear@1: 	if(!(dx | dy)) return;
nuclear@1: 
nuclear@1: 	if(bnstate[0]) {
nuclear@1: 		if(keystate['s']) {
nuclear@1: 			mouse_x = x;
nuclear@1: 			mouse_y = y;
nuclear@1: 		} else if(keystate['z']) {
nuclear@1: 			float s = sqrt(view_scale);
nuclear@1: 			view_scale += dy * 0.01 * s;
nuclear@1: 			if(view_scale < 1e-8) view_scale = 1e-8;
nuclear@1: 		} else {
nuclear@1: 			float s = sqrt(view_scale);
nuclear@1: 			view_center[0] += 0.01 * dx * s;
nuclear@1: 			view_center[1] += 0.01 * dy * s;
nuclear@1: 		}
nuclear@1: 		glutPostRedisplay();
nuclear@1: 	}
nuclear@0: }
nuclear@2: 
nuclear@2: void screenshot(void)
nuclear@2: {
nuclear@2: 	FILE *fp;
nuclear@2: 	DIR *dir;
nuclear@2: 	struct dirent *dent;
nuclear@2: 	unsigned char *img;
nuclear@2: 	static int shotnum = -1;
nuclear@2: 	char fname[128];
nuclear@2: 
nuclear@2: 	if(shotnum == -1) {
nuclear@2: 		shotnum = 1;
nuclear@2: 		if((dir = opendir("."))) {
nuclear@2: 			while((dent = readdir(dir))) {
nuclear@2: 				int num;
nuclear@2: 				if(sscanf(dent->d_name, "img%d.ppm", &num) == 1) {
nuclear@2: 					if(num >= shotnum) shotnum = num + 1;
nuclear@2: 				}
nuclear@2: 			}
nuclear@2: 			closedir(dir);
nuclear@2: 		}
nuclear@2: 	} else {
nuclear@2: 		++shotnum;
nuclear@2: 	}
nuclear@2: 
nuclear@2: 	sprintf(fname, "img%04d.ppm", shotnum > 0 ? shotnum : 0);
nuclear@2: 	if(!(fp = fopen(fname, "wb"))) {
nuclear@2: 		fprintf(stderr, "failed to open %s for writing: %s\n", fname, strerror(errno));
nuclear@2: 		return;
nuclear@2: 	}
nuclear@2: 
nuclear@2: 	if(!(img = malloc(win_width * win_height * 3))) {
nuclear@2: 		fprintf(stderr, "failed to allocate screenshot buffer\n");
nuclear@2: 		fclose(fp);
nuclear@2: 		return;
nuclear@2: 	}
nuclear@2: 
nuclear@2: 	glReadPixels(0, 0, win_width, win_height, GL_RGB, GL_UNSIGNED_BYTE, img);
nuclear@2: 
nuclear@2: 	fprintf(fp, "P6\n%d %d\n255\n", win_width, win_height);
nuclear@2: 	fwrite(img, 1, win_width * win_height * 3, fp);
nuclear@2: 	fclose(fp);
nuclear@2: 	free(img);
nuclear@2: 	return;
nuclear@2: }