volmetrics

annotate src/main.cc @ 21:c22866ecb7ae

find changesets by author, revision, files, or words in the commit message
frame :p
author Eleni Maria Stea <elene.mst@gmail.com>
date Tue, 25 Mar 2014 00:17:39 +0200
parents 21bc62bb3e14
children 4e120dcd55ec
rev   line source
eleni@0 1 #include <GL/glew.h>
eleni@0 2 #include <GL/glut.h>
elene@9 3 #include <GL/glui.h>
eleni@0 4
elene@17 5 #include <math.h>
eleni@0 6 #include <stdio.h>
eleni@0 7 #include <assert.h>
eleni@0 8
elene@9 9 #include <vector>
elene@9 10
elene@9 11 #include "mesh.h"
elene@6 12 #include "volume.h"
eleni@1 13
elene@10 14 static bool init(void);
elene@10 15 static GLUI *create_ui(void);
elene@10 16 static void display(void);
elene@10 17 static void reshape(int x, int y);
elene@10 18 static void keyboard(unsigned char key, int x, int y);
elene@10 19 static void keyboard_up(unsigned char key, int x, int y);
elene@10 20 static void mouse(int button, int state, int x, int y);
elene@10 21 static void motion(int x, int y);
elene@16 22 static bool init_xfer(void);
elene@16 23 static void display_xfer(void);
elene@16 24 static void reshape_xfer(int x, int y);
elene@17 25 static void mouse_xfer(int button, int state, int x, int y);
elene@16 26 static void motion_xfer(int x, int y);
elene@18 27 static void volume_preview ();
elene@16 28
elene@16 29 static int mainwin_id, xferwin_id;
elene@18 30 //todo keyb esc
elene@16 31
eleni@0 32 static int win_xsz, win_ysz;
elene@9 33 static float cam_phi, cam_theta, cam_dist = 6;
elene@9 34 static std::vector<bool> key_state(256);
eleni@0 35
elene@11 36 static const char *vol_fname = "data/test1.vol";
elene@6 37
elene@6 38 static Volume *vol;
elene@9 39 static Mesh *mesh;
elene@12 40 static float cur_z, thres = 0.5, thres2 = 1.0;
elene@20 41 static float zeta;
elene@9 42
elene@9 43 static int use_orig_vol_res = 1;
elene@9 44 static int vol_res[3]; // volume sampling resolution x/y/z
elene@9 45
elene@9 46 static GLUI *ui;
eleni@1 47
eleni@0 48 int main(int argc, char **argv)
eleni@0 49 {
eleni@0 50 glutInit(&argc, argv);
elene@16 51 if(argv[1])
elene@16 52 vol_fname = argv[1];
elene@16 53
elene@16 54 if(!init()) {
elene@16 55 fprintf(stderr, "Failed to initialize program.\n");
elene@16 56 return 1;
elene@16 57 }
elene@16 58
elene@16 59 init_xfer();
elene@16 60
elene@16 61 glutMainLoop();
elene@16 62 return 0;
elene@16 63 }
elene@16 64
elene@16 65 static bool init()
elene@16 66 {
elene@7 67 glutInitWindowSize(512, 512);
elene@11 68 glutInitDisplayMode(GLUT_RGB | GLUT_DEPTH | GLUT_DOUBLE);
eleni@0 69
elene@16 70 mainwin_id = glutCreateWindow("CT scan");
eleni@0 71 glutDisplayFunc(display);
eleni@0 72 glutReshapeFunc(reshape);
eleni@0 73 glutKeyboardFunc(keyboard);
elene@9 74 glutKeyboardUpFunc(keyboard_up);
eleni@0 75 glutMouseFunc(mouse);
eleni@0 76 glutMotionFunc(motion);
eleni@0 77
eleni@0 78 glewInit();
eleni@0 79
elene@11 80 glEnable(GL_DEPTH_TEST);
elene@11 81 glEnable(GL_NORMALIZE);
elene@11 82
elene@11 83 glEnable(GL_LIGHTING); //TODO: shaders
elene@11 84 glEnable(GL_LIGHT0);
elene@11 85 glLightModeli(GL_LIGHT_MODEL_TWO_SIDE, 1);
elene@11 86
elene@9 87 vol = new Volume;
elene@11 88 if(!vol->load(vol_fname)) {
elene@11 89 fprintf(stderr, "Failed to load %s", vol_fname);
elene@9 90 return false;
elene@9 91 }
elene@9 92 mesh = new Mesh;
elene@9 93 cur_z = 0.5;
elene@9 94
elene@9 95 vol_res[0] = vol->get_slice(0)->get_width();
elene@9 96 vol_res[1] = vol->get_slice(0)->get_height();
elene@9 97 vol_res[2] = vol->get_slice_count();
elene@9 98
elene@9 99 if(!(ui = create_ui())) {
elene@9 100 return false;
elene@9 101 }
elene@9 102
elene@9 103 return true;
elene@9 104 }
elene@9 105
elene@9 106 static GLUI_Spinner *res_spin[3];
elene@9 107 static void toggle_use_orig(int id)
elene@9 108 {
elene@9 109 for(int i=0; i<3; i++) {
elene@9 110 if(use_orig_vol_res) {
elene@9 111 res_spin[i]->disable();
elene@9 112 } else {
elene@9 113 res_spin[i]->enable();
elene@9 114 }
elene@9 115 }
elene@9 116 }
elene@9 117 static void thres_change(int id)
elene@9 118 {
elene@9 119 static float prev_thres = thres;
elene@12 120 static float prev_thres2 = thres2;
elene@9 121
elene@12 122 if(prev_thres != thres || prev_thres2 != thres2) {
elene@9 123 prev_thres = thres;
elene@12 124 prev_thres2 = thres2;
elene@12 125 mesh->clear();
elene@12 126 }
elene@16 127
elene@16 128 glutSetWindow(xferwin_id);
elene@16 129 glutPostRedisplay();
elene@16 130 glutSetWindow(mainwin_id);
elene@16 131 glutPostRedisplay();
elene@12 132 }
elene@12 133 static void res_change(int id)
elene@12 134 {
elene@12 135 static float prev_resx = vol_res[0];
elene@12 136 static float prev_resy = vol_res[1];
elene@12 137 static float prev_resz = vol_res[2];
elene@12 138
elene@12 139 if(prev_resx != vol_res[0] || prev_resy != vol_res[1] || prev_resz != vol_res[2]) {
elene@12 140 prev_resx = vol_res[0];
elene@12 141 prev_resy = vol_res[1];
elene@12 142 prev_resz = vol_res[2];
elene@9 143 mesh->clear();
elene@9 144 }
elene@9 145 }
elene@10 146 static GLUI *create_ui()
elene@9 147 {
elene@9 148 GLUI *ui = GLUI_Master.create_glui("ui");
elene@9 149 assert(ui);
elene@9 150
elene@9 151 ui->set_main_gfx_window(glutGetWindow());
elene@9 152
elene@12 153 GLUI_Panel *thres_panel = ui->add_panel("iso thresholds");
elene@12 154
elene@12 155 GLUI_Spinner *thres_spin = ui->add_spinner_to_panel(thres_panel, "T1", GLUI_SPINNER_FLOAT, &thres, 0, thres_change);
elene@9 156 thres_spin->set_float_limits(0, 1);
elene@9 157
elene@12 158 GLUI_Spinner *thres2_spin = ui->add_spinner_to_panel(thres_panel, "T2", GLUI_SPINNER_FLOAT, &thres2, 0, thres_change);
elene@12 159 thres2_spin->set_float_limits(0, 1);
elene@12 160
elene@9 161 GLUI_Panel *res_panel = ui->add_panel("volume resolution");
elene@9 162
elene@9 163 ui->add_checkbox_to_panel(res_panel, "original resolution", &use_orig_vol_res, 0, toggle_use_orig);
elene@9 164
elene@9 165 static const char *res_spin_name[] = {"x", "y", "z"};
elene@9 166 for(int i=0; i<3; i++) {
elene@12 167 res_spin[i] = ui->add_spinner_to_panel(res_panel, res_spin_name[i], GLUI_SPINNER_INT, vol_res + i, 0, res_change);
elene@9 168 res_spin[i]->set_int_limits(8, 256); // TODO limits as arguments or config
elene@9 169 res_spin[i]->disable();
elene@9 170 }
elene@9 171
elene@20 172 GLUI_Panel *preview_panel = ui->add_panel("volume preview");
elene@20 173
elene@20 174 GLUI_Spinner *preview_spin = ui->add_spinner_to_panel(preview_panel, "slice z", GLUI_SPINNER_FLOAT, &zeta, 0);
elene@20 175 preview_spin->set_float_limits(0, 1);
elene@20 176
elene@9 177 return ui;
elene@9 178 }
elene@9 179
elene@18 180 static void volume_preview ()
elene@18 181 {
elene@18 182 float aspect = win_xsz / win_ysz;
elene@18 183
elene@18 184 glDisable(GL_LIGHTING);
elene@21 185 glDisable(GL_DEPTH_TEST);
elene@18 186
elene@18 187 glMatrixMode(GL_MODELVIEW);
elene@18 188 glPushMatrix();
elene@18 189 glLoadIdentity();
elene@18 190
elene@18 191 glMatrixMode(GL_PROJECTION);
elene@18 192 glPushMatrix();
elene@18 193 glLoadIdentity();
elene@18 194 glOrtho(-aspect, aspect, -1.0, 1.0, -1.0, 1.0);
elene@18 195
elene@18 196 glBindTexture(GL_TEXTURE_3D, vol->get_texture());
elene@18 197 glEnable(GL_TEXTURE_3D);
elene@18 198 glBegin(GL_QUADS);
elene@21 199 glColor3f(1.0, 1.0, 1.0);
elene@20 200 glTexCoord3f(0, 0, zeta); glVertex3f(-1.0, 1.0, 0.0);
elene@20 201 glTexCoord3f(0, 1, zeta); glVertex3f(-1.0, 0.5, 0.0);
elene@20 202 glTexCoord3f(1, 1, zeta); glVertex3f(-0.5, 0.5, 0.0);
elene@20 203 glTexCoord3f(1, 0, zeta); glVertex3f(-0.5, 1.0, 0.0);
elene@18 204 glEnd();
elene@18 205 glDisable(GL_TEXTURE_3D);
elene@18 206
elene@21 207 glLineWidth(3);
elene@21 208 glBegin(GL_LINE_LOOP);
elene@21 209 glColor3f(0.4, 0.8, 0.4);
elene@21 210 glVertex3f(-1.0, 1.0, 0.0);
elene@21 211 glVertex3f(-1.0, 0.5, 0.0);
elene@21 212 glVertex3f(-0.5, 0.5, 0.0);
elene@21 213 glVertex3f(-0.5, 1.0, 0.0);
elene@21 214 glEnd();
elene@21 215
elene@18 216 glMatrixMode(GL_PROJECTION);
elene@18 217 glPopMatrix();
elene@18 218
elene@18 219 glMatrixMode(GL_MODELVIEW);
elene@18 220 glPopMatrix();
elene@18 221
elene@21 222 glEnable(GL_DEPTH_TEST);
elene@18 223 glEnable(GL_LIGHTING);
elene@18 224 }
elene@18 225
elene@10 226 static void display(void)
eleni@0 227 {
elene@9 228 glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
elene@9 229
elene@9 230 glMatrixMode(GL_MODELVIEW);
elene@9 231 glLoadIdentity();
elene@9 232 glTranslatef(0, 0, -cam_dist);
elene@9 233 glRotatef(cam_phi, 1, 0, 0);
elene@9 234 glRotatef(cam_theta, 0, 1, 0);
elene@9 235
elene@9 236 /*
elene@6 237 glBindTexture(GL_TEXTURE_3D, vol->get_texture());
elene@6 238 glEnable(GL_TEXTURE_3D);
eleni@1 239 glBegin(GL_QUADS);
elene@6 240 glTexCoord3f(0, 0, cur_z); glVertex3f(-1, -1, 0);
elene@6 241 glTexCoord3f(0, 1, cur_z); glVertex3f(-1, 1, 0);
elene@6 242 glTexCoord3f(1, 1, cur_z); glVertex3f(1, 1, 0);
elene@6 243 glTexCoord3f(1, 0, cur_z); glVertex3f(1, -1, 0);
eleni@1 244 glEnd();
elene@6 245 glDisable(GL_TEXTURE_3D);
elene@9 246 */
eleni@0 247
elene@9 248 if(mesh->is_empty()) {
elene@9 249 printf("recalculating isosurface ... ");
elene@9 250 fflush(stdout);
elene@12 251 vol->create_mesh(mesh, thres, thres2, vol_res[0], vol_res[1], vol_res[2]);
elene@9 252 printf("done.\n");
elene@9 253 }
elene@9 254 mesh->draw();
elene@9 255
elene@18 256 volume_preview();
eleni@0 257 glutSwapBuffers();
eleni@0 258 assert(glGetError() == GL_NO_ERROR);
eleni@0 259 }
eleni@0 260
elene@10 261 static void reshape(int x, int y)
eleni@0 262 {
eleni@0 263 glViewport(0, 0, x, y);
elene@9 264 glMatrixMode(GL_PROJECTION);
elene@9 265 glLoadIdentity();
elene@9 266 gluPerspective(45, (float)x / (float)y, 0.5, 500);
elene@9 267
eleni@0 268 if(x != win_xsz || y != win_ysz) {
eleni@0 269 win_xsz = x;
eleni@0 270 win_ysz = y;
eleni@0 271 }
eleni@0 272 }
eleni@0 273
elene@10 274 static void keyboard(unsigned char key, int x, int y)
eleni@0 275 {
elene@9 276 key_state[(int)key] = true;
elene@9 277
eleni@0 278 switch(key) {
eleni@0 279 case 27:
eleni@0 280 exit(0);
elene@9 281 case 'w':
elene@9 282 {
elene@9 283 static bool wire;
elene@9 284 wire = !wire;
elene@9 285 glPolygonMode(GL_FRONT_AND_BACK, wire ? GL_LINE : GL_FILL);
elene@9 286 }
elene@9 287 break;
eleni@0 288 default:
eleni@0 289 break;
eleni@0 290 }
eleni@0 291 }
eleni@0 292
elene@10 293 static void keyboard_up(unsigned char key, int x, int y)
elene@9 294 {
elene@9 295 key_state[(int) key] = false;
elene@9 296 }
elene@9 297
eleni@0 298 static int prev_x, prev_y;
elene@9 299 static bool bn_state[32];
elene@12 300 static float prev_thres, prev_thres2;
elene@9 301
elene@10 302 static void mouse(int bn, int state, int x, int y)
eleni@0 303 {
eleni@0 304 prev_x = x;
eleni@0 305 prev_y = y;
elene@9 306
elene@9 307 bn_state[bn - GLUT_LEFT_BUTTON] = (state == GLUT_DOWN);
elene@9 308
elene@9 309 if(state == GLUT_DOWN) {
elene@9 310 prev_thres = thres;
elene@12 311 prev_thres2 = thres2;
elene@9 312 }
elene@9 313 else {
elene@12 314 if(thres != prev_thres || thres2 != prev_thres2) {
elene@9 315 mesh->clear();
elene@9 316 }
elene@9 317 }
eleni@0 318 }
eleni@0 319
elene@10 320 static void motion(int x, int y)
eleni@0 321 {
eleni@0 322 int dx = x - prev_x;
eleni@0 323 int dy = y - prev_y;
elene@9 324
elene@9 325 if(!dx && !dy)
elene@9 326 return;
elene@9 327
eleni@0 328 prev_x = x;
eleni@0 329 prev_y = y;
elene@6 330
elene@9 331 if(key_state[(int)'t']) {
elene@9 332 thres = (float)x / (float)win_xsz;
elene@9 333 printf("threshold: %f\n", thres);
elene@9 334
elene@9 335 glutPostRedisplay();
elene@9 336 return;
elene@9 337 }
elene@9 338
elene@9 339 // camera
elene@9 340 if(bn_state[0]) {
elene@9 341 if(key_state[(int)'z']) {
elene@9 342 //zoom the camera
elene@9 343
elene@9 344 cam_dist += dy * 0.1;
elene@9 345
elene@9 346 if(cam_dist < 0)
elene@9 347 cam_dist = 0;
elene@9 348 }
elene@9 349 else {
elene@9 350 //rotate the camera
elene@9 351
elene@9 352 cam_phi += dy * 0.5;
elene@9 353 cam_theta += dx * 0.5;
elene@9 354
elene@9 355 if(cam_phi > 90)
elene@9 356 cam_phi = 90;
elene@9 357 if(cam_phi < -90)
elene@9 358 cam_phi = -90;
elene@9 359 }
elene@9 360
elene@6 361 glutPostRedisplay();
elene@6 362 }
eleni@0 363 }
elene@16 364
elene@16 365 static bool init_xfer(void)
elene@16 366 {
elene@16 367 glutSetWindow(mainwin_id);
elene@16 368 int x = glutGet(GLUT_WINDOW_X);
elene@16 369 int y = glutGet(GLUT_WINDOW_Y) + glutGet(GLUT_WINDOW_HEIGHT);
elene@16 370
elene@16 371 glutInitDisplayMode(GLUT_RGB | GLUT_DOUBLE);
elene@16 372 glutInitWindowSize(512, 100);
elene@16 373 glutInitWindowPosition(x, y);
elene@16 374
elene@16 375 xferwin_id = glutCreateWindow("Transfer Function");
elene@16 376 glutDisplayFunc(display_xfer);
elene@16 377 glutReshapeFunc(reshape_xfer);
elene@17 378 glutMouseFunc(mouse_xfer);
elene@16 379 glutMotionFunc(motion_xfer);
elene@16 380
elene@16 381 return true;
elene@16 382 }
elene@16 383
elene@16 384 static void display_xfer(void)
elene@16 385 {
elene@16 386 glClear(GL_COLOR_BUFFER_BIT);
elene@16 387
elene@16 388 int num_val = glutGet(GLUT_WINDOW_WIDTH) / 4.0;
elene@16 389 float x = 0;
elene@16 390 float dx = 1.0 / num_val;
elene@16 391 float low, high;
elene@16 392 if(thres < thres2) {
elene@16 393 low = thres;
elene@16 394 high = thres2;
elene@16 395 }
elene@16 396 else {
elene@16 397 low = thres2;
elene@16 398 high = thres;
elene@16 399 }
elene@16 400
elene@17 401 //drawing the transfer function curve
elene@17 402
elene@17 403 glLineWidth(3);
elene@17 404 glBegin(GL_LINE_STRIP);
elene@17 405 glColor3f(0.8, 0.3, 0.8);
elene@17 406 for(int i=0; i<num_val; i++) {
elene@17 407 float val = transfer_function(x, low, high);
elene@17 408 glVertex2f(x, val);
elene@17 409 x += dx;
elene@17 410 }
elene@17 411 glEnd();
elene@17 412
elene@17 413 //threshold bars
elene@17 414
elene@17 415 glLineWidth(2);
elene@17 416 glBegin(GL_LINES);
elene@17 417 glColor3f(0.4, 0.4, 0.8);
elene@17 418 glVertex2f(low, 0);
elene@17 419 glVertex2f(low, 1);
elene@17 420 glColor3f(0.4, 0.8, 0.4);
elene@17 421 glVertex2f(high, 0);
elene@17 422 glVertex2f(high, 1);
elene@17 423 glEnd();
elene@17 424
elene@17 425 /*
elene@17 426 * gradient
elene@17 427 */
elene@17 428 /* glBegin(GL_QUADS);
elene@16 429 for(int i=0; i<num_val; i++) {
elene@16 430 float val = transfer_function(x, low, high);
elene@16 431 glColor3f(val, val, val);
elene@16 432 glVertex3f(x, 1.0, 0.0);
elene@16 433 glVertex3f(x, 0.0, 0.0);
elene@16 434
elene@16 435 val = transfer_function(x + dx, low, high);
elene@16 436 glColor3f(val, val, val);
elene@16 437 glVertex3f(x + dx, 0.0, 0.0);
elene@16 438 glVertex3f(x + dx, 1.0, 0.0);
elene@16 439 x += dx;
elene@16 440 }
elene@17 441 glEnd(); */
elene@16 442
elene@16 443 glutSwapBuffers();
elene@16 444 assert(glGetError() == GL_NO_ERROR);
elene@16 445 }
elene@16 446
elene@16 447 static void reshape_xfer(int x, int y)
elene@16 448 {
elene@16 449 glViewport(0.0, 0.0, x, y);
elene@16 450 glMatrixMode(GL_PROJECTION);
elene@16 451 glLoadIdentity();
elene@16 452 glOrtho(0.0, 1.0, 0.0, 1.0, -1.0, 1.0);
elene@16 453 }
elene@16 454
elene@17 455 static float *select_thres;
elene@17 456 static float prev_select_thres;
elene@17 457 static void mouse_xfer(int button, int state, int x, int y)
elene@17 458 {
elene@17 459 if(button == GLUT_LEFT_BUTTON) {
elene@17 460 if(state == GLUT_DOWN) {
elene@17 461 int width = glutGet(GLUT_WINDOW_WIDTH);
elene@17 462 float xpos = (float)x / (float)width;
elene@17 463 if(fabs(xpos - thres) <= fabs(xpos - thres2)) {
elene@17 464 select_thres = &thres;
elene@17 465 }
elene@17 466 else {
elene@17 467 select_thres = &thres2;
elene@17 468 }
elene@17 469 prev_select_thres = *select_thres;
elene@17 470 }
elene@17 471 else {
elene@17 472 if(fabs(*select_thres - prev_select_thres) > 0.001) {
elene@17 473 mesh->clear();
elene@17 474 ui->sync_live();
elene@17 475 }
elene@17 476 select_thres = 0;
elene@17 477 }
elene@17 478 }
elene@17 479 }
elene@17 480
elene@16 481 static void motion_xfer(int x, int y)
elene@16 482 {
elene@17 483 if(!select_thres)
elene@17 484 return;
elene@17 485
elene@17 486 int width = glutGet(GLUT_WINDOW_WIDTH);
elene@17 487 float xpos = (float)x / (float)width;
elene@17 488
elene@17 489 *select_thres = xpos;
elene@17 490
elene@17 491 glutPostRedisplay();
elene@17 492 glutSetWindow(mainwin_id);
elene@17 493 glutPostRedisplay();
elene@16 494 }