metasurf

annotate examples/volume/src/volume.c @ 11:430d8dde62aa

find changesets by author, revision, files, or words in the commit message
random changes
author John Tsiombikas <nuclear@member.fsf.org>
date Sun, 23 Aug 2015 07:17:56 +0300
parents c1a60ab45bf7
children
rev   line source
nuclear@3 1 #include <stdio.h>
nuclear@3 2 #include <stdlib.h>
nuclear@3 3 #include <math.h>
nuclear@3 4 #include <assert.h>
nuclear@3 5
nuclear@3 6 #ifndef NO_SHADERS
nuclear@3 7 #include <GL/glew.h>
nuclear@3 8 #include "sdr.h"
nuclear@3 9 #endif
nuclear@3 10
nuclear@3 11 #ifndef __APPLE__
nuclear@3 12 #include <GL/glut.h>
nuclear@3 13 #else
nuclear@3 14 #include <GLUT/glut.h>
nuclear@3 15 #endif
nuclear@3 16
nuclear@3 17 #include <imago2.h>
nuclear@3 18
nuclear@3 19 #include "cam.h"
nuclear@3 20 #include "metasurf.h"
nuclear@3 21
nuclear@3 22 float eval(float x, float y, float z);
nuclear@3 23 void vertex(float x, float y, float z);
nuclear@3 24 void render(void);
nuclear@3 25 void disp(void);
nuclear@3 26 void reshape(int x, int y);
nuclear@3 27 void keyb(unsigned char key, int x, int y);
nuclear@11 28 void keyb_up(unsigned char key, int x, int y);
nuclear@3 29 void mouse(int bn, int state, int x, int y);
nuclear@3 30 void motion(int x, int y);
nuclear@3 31 int parse_args(int argc, char **argv);
nuclear@3 32
nuclear@3 33 int stereo, fullscreen;
nuclear@3 34 int orig_xsz, orig_ysz;
nuclear@3 35
nuclear@3 36 struct metasurface *msurf;
nuclear@3 37 float threshold = 0.5;
nuclear@3 38 #ifndef NO_SHADERS
nuclear@3 39 unsigned int sdr;
nuclear@3 40 #endif
nuclear@3 41
nuclear@11 42 float yscale = 1.0;
nuclear@11 43
nuclear@3 44 struct img_pixmap *volume;
nuclear@3 45 int xres, yres, num_slices;
nuclear@3 46
nuclear@3 47 int dlist, need_update = 1;
nuclear@3 48
nuclear@3 49 int main(int argc, char **argv)
nuclear@3 50 {
nuclear@3 51 float amb[] = {0, 0, 0, 0};
nuclear@3 52 float lpos[] = {-0.2, 0.2, 1, 0};
nuclear@3 53
nuclear@3 54 glutInitWindowSize(1280, 720);
nuclear@3 55 glutInit(&argc, argv);
nuclear@3 56
nuclear@3 57 if(parse_args(argc, argv) == -1) {
nuclear@3 58 return 1;
nuclear@3 59 }
nuclear@3 60 glutInitDisplayMode(GLUT_RGB | GLUT_DEPTH | GLUT_DOUBLE | (stereo ? GLUT_STEREO : 0));
nuclear@3 61 glutCreateWindow("metasurf - volume rendering");
nuclear@3 62
nuclear@3 63 orig_xsz = glutGet(GLUT_WINDOW_WIDTH);
nuclear@3 64 orig_ysz = glutGet(GLUT_WINDOW_HEIGHT);
nuclear@3 65
nuclear@3 66 if(fullscreen) {
nuclear@3 67 glutFullScreen();
nuclear@3 68 }
nuclear@3 69
nuclear@3 70 glutDisplayFunc(disp);
nuclear@3 71 glutReshapeFunc(reshape);
nuclear@3 72 glutKeyboardFunc(keyb);
nuclear@11 73 glutKeyboardUpFunc(keyb_up);
nuclear@3 74 glutMouseFunc(mouse);
nuclear@3 75 glutMotionFunc(motion);
nuclear@3 76
nuclear@3 77 #ifndef NO_SHADERS
nuclear@3 78 glewInit();
nuclear@3 79 if(!(sdr = create_program_load("sdr/vert.glsl", "sdr/frag.glsl"))) {
nuclear@3 80 return 1;
nuclear@3 81 }
nuclear@3 82 #endif
nuclear@3 83
nuclear@3 84 glEnable(GL_CULL_FACE);
nuclear@3 85 glEnable(GL_DEPTH_TEST);
nuclear@3 86
nuclear@3 87 glLightModelfv(GL_LIGHT_MODEL_AMBIENT, amb);
nuclear@3 88
nuclear@3 89 glEnable(GL_LIGHTING);
nuclear@3 90 glEnable(GL_LIGHT0);
nuclear@3 91 glLightfv(GL_LIGHT0, GL_POSITION, lpos);
nuclear@3 92
nuclear@3 93 glEnable(GL_NORMALIZE);
nuclear@3 94
nuclear@3 95 cam_focus_dist(3.0);
nuclear@3 96 cam_clip(0.1, 200.0);
nuclear@3 97 cam_rotate(0, 0);
nuclear@3 98 cam_dolly(2);
nuclear@3 99
nuclear@3 100 msurf = msurf_create();
nuclear@3 101 msurf_eval_func(msurf, eval);
nuclear@3 102 msurf_vertex_func(msurf, vertex);
nuclear@3 103 msurf_threshold(msurf, threshold);
nuclear@3 104 msurf_resolution(msurf, xres, yres, num_slices);
nuclear@3 105 msurf_bounds(msurf, -1, -1, -1, 1, 1, 1);
nuclear@3 106
nuclear@3 107 glClearColor(0.6, 0.6, 0.6, 1.0);
nuclear@3 108
nuclear@3 109 dlist = glGenLists(1);
nuclear@3 110
nuclear@3 111 glutMainLoop();
nuclear@3 112 return 0;
nuclear@3 113 }
nuclear@3 114
nuclear@3 115 float eval(float x, float y, float z)
nuclear@3 116 {
nuclear@3 117 int px, py, slice;
nuclear@3 118 struct img_pixmap *img;
nuclear@3 119
nuclear@3 120 px = round((x * 0.5 + 0.5) * xres);
nuclear@3 121 py = round((y * 0.5 + 0.5) * yres);
nuclear@3 122 slice = round((z * 0.5 + 0.5) * num_slices);
nuclear@3 123
nuclear@3 124 if(px < 0) px = 0;
nuclear@3 125 if(px >= xres) px = xres - 1;
nuclear@3 126
nuclear@3 127 if(py < 0) py = 0;
nuclear@3 128 if(py >= yres) py = yres - 1;
nuclear@3 129
nuclear@3 130 if(slice < 0) slice = 0;
nuclear@3 131 if(slice >= num_slices) slice = num_slices - 1;
nuclear@3 132
nuclear@3 133 img = volume + slice;
nuclear@3 134 return *((unsigned char*)img->pixels + py * img->width + px) / 255.0;
nuclear@3 135 }
nuclear@3 136
nuclear@3 137 void vertex(float x, float y, float z)
nuclear@3 138 {
nuclear@3 139 float dx = 1.0 / xres;
nuclear@3 140 float dy = 1.0 / yres;
nuclear@3 141 float dz = 1.0 / num_slices;
nuclear@3 142 float dfdx = eval(x - dx, y, z) - eval(x + dx, y, z);
nuclear@3 143 float dfdy = eval(x, y - dy, z) - eval(x, y + dy, z);
nuclear@3 144 float dfdz = eval(x, y, z - dz) - eval(x, y, z + dz);
nuclear@3 145
nuclear@3 146 glNormal3f(dfdx, dfdy, dfdz);
nuclear@3 147 glVertex3f(x, y, z);
nuclear@3 148 }
nuclear@3 149
nuclear@3 150 void render(void)
nuclear@3 151 {
nuclear@3 152 float kd[] = {0.87, 0.82, 0.74, 1.0};
nuclear@3 153 float ks[] = {0.9, 0.9, 0.9, 1.0};
nuclear@3 154
nuclear@3 155 glMaterialfv(GL_FRONT_AND_BACK, GL_AMBIENT_AND_DIFFUSE, kd);
nuclear@3 156 glMaterialfv(GL_FRONT_AND_BACK, GL_SPECULAR, ks);
nuclear@3 157 glMaterialf(GL_FRONT_AND_BACK, GL_SHININESS, 60.0);
nuclear@3 158
nuclear@3 159 #ifndef NO_SHADERS
nuclear@3 160 bind_program(sdr);
nuclear@3 161 #endif
nuclear@3 162
nuclear@3 163 glMatrixMode(GL_MODELVIEW);
nuclear@3 164 glPushMatrix();
nuclear@11 165 glScalef(1.0, yscale, 1.0);
nuclear@3 166 glRotatef(90, 1, 0, 0);
nuclear@3 167
nuclear@3 168 if(need_update) {
nuclear@3 169 glNewList(dlist, GL_COMPILE);
nuclear@3 170 glBegin(GL_TRIANGLES);
nuclear@3 171 printf("generating mesh... ");
nuclear@3 172 fflush(stdout);
nuclear@3 173 msurf_polygonize(msurf);
nuclear@3 174 glEnd();
nuclear@3 175 glEndList();
nuclear@3 176 need_update = 0;
nuclear@3 177 printf("done\n");
nuclear@3 178 }
nuclear@3 179 glCallList(dlist);
nuclear@3 180
nuclear@3 181 glPopMatrix();
nuclear@3 182
nuclear@3 183 assert(glGetError() == GL_NO_ERROR);
nuclear@3 184 }
nuclear@3 185
nuclear@3 186 void disp(void)
nuclear@3 187 {
nuclear@3 188 if(stereo) {
nuclear@3 189 glDrawBuffer(GL_BACK_LEFT);
nuclear@3 190 }
nuclear@3 191
nuclear@3 192 glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
nuclear@3 193
nuclear@3 194 glMatrixMode(GL_PROJECTION);
nuclear@3 195 glLoadIdentity();
nuclear@3 196 cam_stereo_proj_matrix(stereo ? CAM_LEFT : CAM_CENTER);
nuclear@3 197
nuclear@3 198 glMatrixMode(GL_MODELVIEW);
nuclear@3 199 glLoadIdentity();
nuclear@3 200 cam_stereo_view_matrix(stereo ? CAM_LEFT : CAM_CENTER);
nuclear@3 201
nuclear@3 202 render();
nuclear@3 203
nuclear@3 204 if(stereo) {
nuclear@3 205 glDrawBuffer(GL_BACK_RIGHT);
nuclear@3 206 glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
nuclear@3 207
nuclear@3 208 glMatrixMode(GL_PROJECTION);
nuclear@3 209 glLoadIdentity();
nuclear@3 210 cam_stereo_proj_matrix(CAM_RIGHT);
nuclear@3 211
nuclear@3 212 glMatrixMode(GL_MODELVIEW);
nuclear@3 213 glLoadIdentity();
nuclear@3 214 cam_stereo_view_matrix(CAM_RIGHT);
nuclear@3 215
nuclear@3 216 render();
nuclear@3 217 }
nuclear@3 218 glutSwapBuffers();
nuclear@3 219 }
nuclear@3 220
nuclear@3 221 void reshape(int x, int y)
nuclear@3 222 {
nuclear@3 223 glViewport(0, 0, x, y);
nuclear@3 224 cam_aspect((float)x / (float)y);
nuclear@3 225 }
nuclear@3 226
nuclear@11 227 int mode_scale;
nuclear@11 228
nuclear@3 229 void keyb(unsigned char key, int x, int y)
nuclear@3 230 {
nuclear@3 231 static int wire;
nuclear@3 232
nuclear@3 233 switch(key) {
nuclear@3 234 case 27:
nuclear@3 235 exit(0);
nuclear@3 236
nuclear@3 237 case 'f':
nuclear@3 238 fullscreen = !fullscreen;
nuclear@3 239 if(fullscreen) {
nuclear@3 240 glutFullScreen();
nuclear@3 241 } else {
nuclear@3 242 glutReshapeWindow(orig_xsz, orig_ysz);
nuclear@3 243 }
nuclear@3 244 break;
nuclear@3 245
nuclear@3 246 case 's':
nuclear@3 247 stereo = !stereo;
nuclear@3 248 glutPostRedisplay();
nuclear@3 249 break;
nuclear@3 250
nuclear@3 251 case 'w':
nuclear@3 252 wire = !wire;
nuclear@3 253 glPolygonMode(GL_FRONT_AND_BACK, wire ? GL_LINE : GL_FILL);
nuclear@3 254 glutPostRedisplay();
nuclear@3 255 break;
nuclear@3 256
nuclear@3 257 case '=':
nuclear@3 258 threshold += 0.05;
nuclear@3 259 msurf_threshold(msurf, threshold);
nuclear@3 260 printf("threshold: %f\n", threshold);
nuclear@3 261 glutPostRedisplay();
nuclear@3 262 need_update = 1;
nuclear@3 263 break;
nuclear@3 264
nuclear@3 265 case '-':
nuclear@3 266 threshold -= 0.05;
nuclear@3 267 msurf_threshold(msurf, threshold);
nuclear@3 268 printf("threshold: %f\n", threshold);
nuclear@3 269 glutPostRedisplay();
nuclear@3 270 need_update = 1;
nuclear@3 271 break;
nuclear@3 272
nuclear@11 273 case 'y':
nuclear@11 274 mode_scale = 1;
nuclear@11 275 break;
nuclear@11 276
nuclear@3 277 default:
nuclear@3 278 break;
nuclear@3 279 }
nuclear@3 280 }
nuclear@3 281
nuclear@11 282 void keyb_up(unsigned char key, int x, int y)
nuclear@11 283 {
nuclear@11 284 switch(key) {
nuclear@11 285 case 'y':
nuclear@11 286 mode_scale = 0;
nuclear@11 287 break;
nuclear@11 288 }
nuclear@11 289 }
nuclear@11 290
nuclear@3 291 int bnstate[32];
nuclear@3 292 int prev_x, prev_y;
nuclear@3 293
nuclear@3 294 void mouse(int bn, int state, int x, int y)
nuclear@3 295 {
nuclear@3 296 bnstate[bn] = state == GLUT_DOWN;
nuclear@3 297 prev_x = x;
nuclear@3 298 prev_y = y;
nuclear@3 299 }
nuclear@3 300
nuclear@3 301 void motion(int x, int y)
nuclear@3 302 {
nuclear@3 303 int dx, dy;
nuclear@3 304
nuclear@3 305 dx = x - prev_x;
nuclear@3 306 dy = y - prev_y;
nuclear@3 307 prev_x = x;
nuclear@3 308 prev_y = y;
nuclear@3 309
nuclear@11 310 if(mode_scale) {
nuclear@11 311 yscale += dy * 0.001;
nuclear@11 312 } else {
nuclear@11 313 if(bnstate[GLUT_LEFT_BUTTON]) {
nuclear@11 314 cam_inp_rotate(dx, dy);
nuclear@11 315 }
nuclear@11 316 if(bnstate[GLUT_RIGHT_BUTTON]) {
nuclear@11 317 cam_inp_zoom(dy);
nuclear@11 318 }
nuclear@3 319 }
nuclear@3 320 glutPostRedisplay();
nuclear@3 321 }
nuclear@3 322
nuclear@3 323 struct list_node {
nuclear@3 324 struct img_pixmap img;
nuclear@3 325 struct list_node *next;
nuclear@3 326 };
nuclear@3 327
nuclear@3 328 int parse_args(int argc, char **argv)
nuclear@3 329 {
nuclear@3 330 int i;
nuclear@6 331 char *endp;
nuclear@3 332 struct list_node *head = 0, *tail = 0;
nuclear@3 333
nuclear@3 334 for(i=1; i<argc; i++) {
nuclear@3 335 if(argv[i][0] == '-' && argv[i][2] == 0) {
nuclear@3 336 switch(argv[i][1]) {
nuclear@3 337 case 'f':
nuclear@3 338 fullscreen = !fullscreen;
nuclear@3 339 break;
nuclear@3 340
nuclear@3 341 case 's':
nuclear@3 342 stereo = !stereo;
nuclear@3 343 break;
nuclear@3 344
nuclear@6 345 case 't':
nuclear@6 346 threshold = strtod(argv[++i], &endp);
nuclear@6 347 if(endp == argv[i]) {
nuclear@6 348 fprintf(stderr, "-t must be followed by a number\n");
nuclear@6 349 return -1;
nuclear@6 350 }
nuclear@6 351 break;
nuclear@6 352
nuclear@3 353 case 'h':
nuclear@3 354 printf("usage: %s [opt]\n", argv[0]);
nuclear@3 355 printf("options:\n");
nuclear@3 356 printf(" -f start in fullscreen\n");
nuclear@3 357 printf(" -s enable stereoscopic rendering\n");
nuclear@3 358 printf(" -h print usage and exit\n");
nuclear@3 359 exit(0);
nuclear@3 360
nuclear@3 361 default:
nuclear@3 362 fprintf(stderr, "unrecognized option: %s\n", argv[i]);
nuclear@3 363 return -1;
nuclear@3 364 }
nuclear@3 365 } else {
nuclear@3 366 struct list_node *slice;
nuclear@3 367
nuclear@3 368 if(!(slice = malloc(sizeof *slice))) {
nuclear@3 369 fprintf(stderr, "failed to allocate volume slice: %d\n", num_slices);
nuclear@3 370 return -1;
nuclear@3 371 }
nuclear@3 372 slice->next = 0;
nuclear@3 373
nuclear@3 374 img_init(&slice->img);
nuclear@3 375 if(img_load(&slice->img, argv[i]) == -1) {
nuclear@3 376 fprintf(stderr, "failed to load volume slice %d: %s\n", num_slices, argv[i]);
nuclear@3 377 free(slice);
nuclear@3 378 return -1;
nuclear@3 379 }
nuclear@3 380 img_convert(&slice->img, IMG_FMT_GREY8);
nuclear@3 381
nuclear@3 382 if(num_slices > 0 && (xres != slice->img.width || yres != slice->img.height)) {
nuclear@3 383 fprintf(stderr, "error: slice %d (%s) is %dx%d, up to now we had %dx%d images\n", num_slices, argv[i],
nuclear@3 384 slice->img.width, slice->img.height, xres, yres);
nuclear@3 385 img_destroy(&slice->img);
nuclear@3 386 free(slice);
nuclear@3 387 return -1;
nuclear@3 388 }
nuclear@3 389 xres = slice->img.width;
nuclear@3 390 yres = slice->img.height;
nuclear@3 391
nuclear@3 392 if(head) {
nuclear@3 393 tail->next = slice;
nuclear@3 394 tail = slice;
nuclear@3 395 } else {
nuclear@3 396 head = tail = slice;
nuclear@3 397 }
nuclear@3 398 printf("loaded volume slice %d: %s\n", num_slices++, argv[i]);
nuclear@3 399 }
nuclear@3 400 }
nuclear@3 401
nuclear@3 402 if(!head) {
nuclear@3 403 fprintf(stderr, "you must specify a list of images for the volume data slices\n");
nuclear@3 404 return -1;
nuclear@3 405 }
nuclear@3 406
nuclear@3 407 if(!(volume = malloc(num_slices * sizeof *volume))) {
nuclear@3 408 fprintf(stderr, "failed to allocate volume data (%d slices)\n", num_slices);
nuclear@3 409 return -1;
nuclear@3 410 }
nuclear@3 411
nuclear@3 412 for(i=0; i<num_slices; i++) {
nuclear@3 413 void *tmp;
nuclear@3 414
nuclear@3 415 assert(head);
nuclear@3 416 volume[i] = head->img;
nuclear@3 417
nuclear@3 418 tmp = head;
nuclear@3 419 head = head->next;
nuclear@3 420 free(tmp);
nuclear@3 421 }
nuclear@3 422
nuclear@3 423 return 0;
nuclear@3 424 }