metasurf

annotate examples/metaballs/src/metaballs.c @ 2:9ab057fba0c5

find changesets by author, revision, files, or words in the commit message
added copyright statements, added comments, yada yada
author John Tsiombikas <nuclear@member.fsf.org>
date Tue, 25 Oct 2011 08:59:07 +0300
parents dc0e882ec3f9
children 52664d3451ad
rev   line source
nuclear@0 1 #include <stdio.h>
nuclear@0 2 #include <stdlib.h>
nuclear@0 3 #include <math.h>
nuclear@0 4 #include <assert.h>
nuclear@0 5
nuclear@2 6 #ifndef NO_SHADERS
nuclear@0 7 #include <GL/glew.h>
nuclear@2 8 #include "sdr.h"
nuclear@2 9 #endif
nuclear@2 10
nuclear@0 11 #ifndef __APPLE__
nuclear@0 12 #include <GL/glut.h>
nuclear@0 13 #else
nuclear@0 14 #include <GLUT/glut.h>
nuclear@0 15 #endif
nuclear@0 16
nuclear@0 17 #include "cam.h"
nuclear@0 18 #include "metasurf.h"
nuclear@0 19
nuclear@0 20 #define RES 38
nuclear@0 21
nuclear@0 22 struct metaball {
nuclear@0 23 float energy;
nuclear@0 24 float x, y, z;
nuclear@0 25 } mball[] = {
nuclear@0 26 {1.0, 0, 0, 0},
nuclear@0 27 {0.25, 0.45, 0, 0.25},
nuclear@0 28 {0.15, -0.3, 0.2, 0.1}
nuclear@0 29 };
nuclear@0 30
nuclear@0 31 int num_mballs = sizeof mball / sizeof *mball;
nuclear@0 32
nuclear@0 33 float eval(float x, float y, float z);
nuclear@0 34 float eval_cached(float x, float y, float z);
nuclear@0 35 void vertex(float x, float y, float z);
nuclear@0 36 void render(void);
nuclear@0 37 void disp(void);
nuclear@0 38 void reshape(int x, int y);
nuclear@0 39 void keyb(unsigned char key, int x, int y);
nuclear@0 40 void mouse(int bn, int state, int x, int y);
nuclear@0 41 void motion(int x, int y);
nuclear@0 42 void sball_button(int bn, int state);
nuclear@0 43 void sball_motion(int x, int y, int z);
nuclear@0 44 int parse_args(int argc, char **argv);
nuclear@0 45
nuclear@2 46 int stereo, fullscreen;
nuclear@2 47 int orig_xsz, orig_ysz;
nuclear@2 48
nuclear@0 49 struct metasurface *msurf;
nuclear@0 50 float threshold = 12;
nuclear@2 51 #ifndef NO_SHADERS
nuclear@0 52 unsigned int sdr;
nuclear@2 53 #endif
nuclear@0 54 int bidx = 1;
nuclear@0 55
nuclear@0 56 int main(int argc, char **argv)
nuclear@0 57 {
nuclear@0 58 float amb[] = {0, 0, 0, 0};
nuclear@0 59 float lpos[] = {-0.2, 0.2, 1, 0};
nuclear@0 60
nuclear@0 61 glutInitWindowSize(1280, 720);
nuclear@0 62 glutInit(&argc, argv);
nuclear@0 63
nuclear@0 64 if(parse_args(argc, argv) == -1) {
nuclear@0 65 return 1;
nuclear@0 66 }
nuclear@0 67 glutInitDisplayMode(GLUT_RGB | GLUT_DEPTH | GLUT_DOUBLE | (stereo ? GLUT_STEREO : 0));
nuclear@0 68 glutCreateWindow("metasurf");
nuclear@0 69
nuclear@2 70 orig_xsz = glutGet(GLUT_WINDOW_WIDTH);
nuclear@2 71 orig_ysz = glutGet(GLUT_WINDOW_HEIGHT);
nuclear@2 72
nuclear@2 73 if(fullscreen) {
nuclear@2 74 glutFullScreen();
nuclear@2 75 }
nuclear@2 76
nuclear@0 77 glutDisplayFunc(disp);
nuclear@0 78 glutReshapeFunc(reshape);
nuclear@0 79 glutKeyboardFunc(keyb);
nuclear@0 80 glutMouseFunc(mouse);
nuclear@0 81 glutMotionFunc(motion);
nuclear@0 82 glutSpaceballButtonFunc(sball_button);
nuclear@0 83 glutSpaceballMotionFunc(sball_motion);
nuclear@0 84
nuclear@2 85 #ifndef NO_SHADERS
nuclear@0 86 glewInit();
nuclear@2 87 if(!(sdr = create_program_load("sdr/vert.glsl", "sdr/frag.glsl"))) {
nuclear@2 88 return 1;
nuclear@2 89 }
nuclear@2 90 #endif
nuclear@0 91
nuclear@0 92 glEnable(GL_CULL_FACE);
nuclear@0 93 glEnable(GL_DEPTH_TEST);
nuclear@0 94
nuclear@0 95 glLightModelfv(GL_LIGHT_MODEL_AMBIENT, amb);
nuclear@0 96
nuclear@0 97 glEnable(GL_LIGHTING);
nuclear@0 98 glEnable(GL_LIGHT0);
nuclear@0 99 glLightfv(GL_LIGHT0, GL_POSITION, lpos);
nuclear@0 100
nuclear@0 101 glEnable(GL_NORMALIZE);
nuclear@0 102
nuclear@0 103 cam_focus_dist(2.0);
nuclear@0 104 cam_clip(0.1, 200.0);
nuclear@0 105 cam_rotate(0, 0);
nuclear@0 106 cam_dolly(2);
nuclear@0 107
nuclear@0 108 msurf = msurf_create();
nuclear@0 109 msurf_eval_func(msurf, eval);
nuclear@0 110 msurf_vertex_func(msurf, vertex);
nuclear@0 111 msurf_threshold(msurf, threshold);
nuclear@0 112 msurf_resolution(msurf, RES, RES, RES);
nuclear@0 113 msurf_bounds(msurf, -1, -1, -1, 1, 1, 1);
nuclear@0 114
nuclear@0 115 glClearColor(0.8, 0.8, 0.8, 1.0);
nuclear@0 116
nuclear@0 117 glutMainLoop();
nuclear@0 118 return 0;
nuclear@0 119 }
nuclear@0 120
nuclear@0 121 float eval(float x, float y, float z)
nuclear@0 122 {
nuclear@0 123 int i;
nuclear@0 124 float val = 0.0f;
nuclear@0 125
nuclear@0 126 for(i=0; i<num_mballs; i++) {
nuclear@0 127 float dx = mball[i].x - x;
nuclear@0 128 float dy = mball[i].y - y;
nuclear@0 129 float dz = mball[i].z - z;
nuclear@0 130 float dist_sq = dx * dx + dy * dy + dz * dz;
nuclear@0 131
nuclear@0 132 if(dist_sq < 1e-6) {
nuclear@0 133 val += 100.0;
nuclear@0 134 } else {
nuclear@0 135 val += mball[i].energy / dist_sq;
nuclear@0 136 }
nuclear@0 137 }
nuclear@0 138 return val;
nuclear@0 139 }
nuclear@0 140
nuclear@0 141 void vertex(float x, float y, float z)
nuclear@0 142 {
nuclear@0 143 const float dt = 0.001;
nuclear@0 144 float dfdx = eval(x - dt, y, z) - eval(x + dt, y, z);
nuclear@0 145 float dfdy = eval(x, y - dt, z) - eval(x, y + dt, z);
nuclear@0 146 float dfdz = eval(x, y, z - dt) - eval(x, y, z + dt);
nuclear@0 147
nuclear@0 148 glNormal3f(dfdx, dfdy, dfdz);
nuclear@0 149 glVertex3f(x, y, z);
nuclear@0 150 }
nuclear@0 151
nuclear@0 152 void render(void)
nuclear@0 153 {
nuclear@0 154 float kd[] = {0.7, 0.28, 0.2, 1.0};
nuclear@0 155 float ks[] = {0.9, 0.9, 0.9, 1.0};
nuclear@0 156
nuclear@0 157 glMaterialfv(GL_FRONT_AND_BACK, GL_AMBIENT_AND_DIFFUSE, kd);
nuclear@0 158 glMaterialfv(GL_FRONT_AND_BACK, GL_SPECULAR, ks);
nuclear@0 159 glMaterialf(GL_FRONT_AND_BACK, GL_SHININESS, 60.0);
nuclear@0 160
nuclear@2 161 #ifndef NO_SHADERS
nuclear@0 162 bind_program(sdr);
nuclear@2 163 #endif
nuclear@0 164
nuclear@0 165 glBegin(GL_TRIANGLES);
nuclear@0 166 msurf_polygonize(msurf);
nuclear@0 167 glEnd();
nuclear@0 168
nuclear@0 169 assert(glGetError() == GL_NO_ERROR);
nuclear@0 170 }
nuclear@0 171
nuclear@0 172 void disp(void)
nuclear@0 173 {
nuclear@0 174 if(stereo) {
nuclear@0 175 glDrawBuffer(GL_BACK_LEFT);
nuclear@0 176 }
nuclear@0 177
nuclear@0 178 glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
nuclear@0 179
nuclear@0 180 glMatrixMode(GL_PROJECTION);
nuclear@0 181 glLoadIdentity();
nuclear@0 182 cam_stereo_proj_matrix(stereo ? CAM_LEFT : CAM_CENTER);
nuclear@0 183
nuclear@0 184 glMatrixMode(GL_MODELVIEW);
nuclear@0 185 glLoadIdentity();
nuclear@0 186 cam_stereo_view_matrix(stereo ? CAM_LEFT : CAM_CENTER);
nuclear@0 187
nuclear@0 188 render();
nuclear@0 189
nuclear@0 190 if(stereo) {
nuclear@0 191 glDrawBuffer(GL_BACK_RIGHT);
nuclear@0 192 glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
nuclear@0 193
nuclear@0 194 glMatrixMode(GL_PROJECTION);
nuclear@0 195 glLoadIdentity();
nuclear@0 196 cam_stereo_proj_matrix(CAM_RIGHT);
nuclear@0 197
nuclear@0 198 glMatrixMode(GL_MODELVIEW);
nuclear@0 199 glLoadIdentity();
nuclear@0 200 cam_stereo_view_matrix(CAM_RIGHT);
nuclear@0 201
nuclear@0 202 render();
nuclear@0 203 }
nuclear@0 204 glutSwapBuffers();
nuclear@0 205 }
nuclear@0 206
nuclear@0 207 void reshape(int x, int y)
nuclear@0 208 {
nuclear@0 209 glViewport(0, 0, x, y);
nuclear@0 210 cam_aspect((float)x / (float)y);
nuclear@0 211 }
nuclear@0 212
nuclear@0 213 void keyb(unsigned char key, int x, int y)
nuclear@0 214 {
nuclear@0 215 static int wire;
nuclear@0 216
nuclear@0 217 switch(key) {
nuclear@0 218 case 27:
nuclear@0 219 exit(0);
nuclear@0 220
nuclear@2 221 case 'f':
nuclear@2 222 fullscreen = !fullscreen;
nuclear@2 223 if(fullscreen) {
nuclear@2 224 glutFullScreen();
nuclear@2 225 } else {
nuclear@2 226 glutReshapeWindow(orig_xsz, orig_ysz);
nuclear@2 227 }
nuclear@2 228 break;
nuclear@2 229
nuclear@0 230 case 's':
nuclear@0 231 stereo = !stereo;
nuclear@0 232 glutPostRedisplay();
nuclear@0 233 break;
nuclear@0 234
nuclear@0 235 case 'w':
nuclear@0 236 wire = !wire;
nuclear@0 237 glPolygonMode(GL_FRONT_AND_BACK, wire ? GL_LINE : GL_FILL);
nuclear@0 238 glutPostRedisplay();
nuclear@0 239 break;
nuclear@0 240
nuclear@0 241 case '=':
nuclear@0 242 threshold += 0.05;
nuclear@0 243 msurf_threshold(msurf, threshold);
nuclear@0 244 printf("threshold: %f\n", threshold);
nuclear@0 245 glutPostRedisplay();
nuclear@0 246 break;
nuclear@0 247
nuclear@0 248 case '-':
nuclear@0 249 threshold -= 0.05;
nuclear@0 250 msurf_threshold(msurf, threshold);
nuclear@0 251 printf("threshold: %f\n", threshold);
nuclear@0 252 glutPostRedisplay();
nuclear@0 253 break;
nuclear@0 254
nuclear@0 255 case ' ':
nuclear@0 256 bidx = (bidx + 1) % num_mballs;
nuclear@0 257 break;
nuclear@0 258
nuclear@0 259 default:
nuclear@0 260 break;
nuclear@0 261 }
nuclear@0 262 }
nuclear@0 263
nuclear@0 264 int bnstate[32];
nuclear@0 265 int prev_x, prev_y;
nuclear@0 266
nuclear@0 267 void mouse(int bn, int state, int x, int y)
nuclear@0 268 {
nuclear@0 269 bnstate[bn] = state == GLUT_DOWN;
nuclear@0 270 prev_x = x;
nuclear@0 271 prev_y = y;
nuclear@0 272 }
nuclear@0 273
nuclear@0 274 void motion(int x, int y)
nuclear@0 275 {
nuclear@0 276 int dx, dy;
nuclear@0 277
nuclear@0 278 dx = x - prev_x;
nuclear@0 279 dy = y - prev_y;
nuclear@0 280 prev_x = x;
nuclear@0 281 prev_y = y;
nuclear@0 282
nuclear@2 283 if(glutGetModifiers()) {
nuclear@2 284 if(bnstate[GLUT_LEFT_BUTTON]) {
nuclear@2 285 cam_inp_rotate(dx, dy);
nuclear@2 286 }
nuclear@2 287 if(bnstate[GLUT_RIGHT_BUTTON]) {
nuclear@2 288 cam_inp_zoom(dy);
nuclear@2 289 }
nuclear@2 290 } else {
nuclear@2 291 mball[bidx].x += (float)dx * 0.005;
nuclear@2 292 mball[bidx].y -= (float)dy * 0.005;
nuclear@0 293 }
nuclear@2 294 glutPostRedisplay();
nuclear@0 295 }
nuclear@0 296
nuclear@0 297 void sball_button(int bn, int state)
nuclear@0 298 {
nuclear@0 299 if(state) return;
nuclear@0 300
nuclear@0 301 if(bn < num_mballs) {
nuclear@0 302 bidx = bn;
nuclear@0 303 } else {
nuclear@0 304 bidx = (bidx + 1) % num_mballs;
nuclear@0 305 }
nuclear@0 306 }
nuclear@0 307
nuclear@0 308 void sball_motion(int x, int y, int z)
nuclear@0 309 {
nuclear@2 310 mball[bidx].x += (float)x / 16384.0;
nuclear@2 311 mball[bidx].y += (float)y / 16384.0;
nuclear@2 312 mball[bidx].z -= (float)z / 16384.0;
nuclear@0 313 glutPostRedisplay();
nuclear@0 314 }
nuclear@0 315
nuclear@0 316 int parse_args(int argc, char **argv)
nuclear@0 317 {
nuclear@0 318 int i;
nuclear@0 319
nuclear@0 320 for(i=1; i<argc; i++) {
nuclear@0 321 if(argv[i][0] == '-' && argv[i][2] == 0) {
nuclear@0 322 switch(argv[i][1]) {
nuclear@2 323 case 'f':
nuclear@2 324 fullscreen = !fullscreen;
nuclear@2 325 break;
nuclear@2 326
nuclear@0 327 case 's':
nuclear@0 328 stereo = !stereo;
nuclear@0 329 break;
nuclear@0 330
nuclear@2 331 case 'h':
nuclear@2 332 printf("usage: %s [opt]\n", argv[0]);
nuclear@2 333 printf("options:\n");
nuclear@2 334 printf(" -f start in fullscreen\n");
nuclear@2 335 printf(" -s enable stereoscopic rendering\n");
nuclear@2 336 printf(" -h print usage and exit\n");
nuclear@2 337 exit(0);
nuclear@2 338
nuclear@0 339 default:
nuclear@0 340 fprintf(stderr, "unrecognized option: %s\n", argv[i]);
nuclear@0 341 return -1;
nuclear@0 342 }
nuclear@0 343 } else {
nuclear@0 344 fprintf(stderr, "unexpected argument: %s\n", argv[i]);
nuclear@0 345 return -1;
nuclear@0 346 }
nuclear@0 347 }
nuclear@0 348 return 0;
nuclear@0 349 }