curvedraw

annotate src/app.cc @ 6:6e980fddbf3b

find changesets by author, revision, files, or words in the commit message
curve export
author John Tsiombikas <nuclear@member.fsf.org>
date Fri, 18 Dec 2015 07:07:19 +0200
parents 2b7ae76c173f
children 95fada20c638
rev   line source
nuclear@0 1 #include <stdlib.h>
nuclear@0 2 #include <float.h>
nuclear@2 3 #include <assert.h>
nuclear@0 4 #include <vector>
nuclear@0 5 #include <algorithm>
nuclear@0 6 #include "opengl.h"
nuclear@0 7 #include "app.h"
nuclear@0 8 #include "curve.h"
nuclear@0 9 #include "widgets.h"
nuclear@6 10 #include "curvefile.h"
nuclear@0 11
nuclear@3 12 enum SnapMode {
nuclear@3 13 SNAP_NONE,
nuclear@3 14 SNAP_GRID,
nuclear@3 15 SNAP_POINT
nuclear@3 16 };
nuclear@3 17
nuclear@1 18 int win_width, win_height;
nuclear@1 19 float win_aspect;
nuclear@1 20
nuclear@0 21 static void draw_grid(float sz, float sep, float alpha = 1.0f);
nuclear@0 22 static void draw_curve(const Curve *curve);
nuclear@0 23 static void on_click(int bn, float u, float v);
nuclear@0 24
nuclear@2 25 // viewport control
nuclear@2 26 static Vector2 view_pan;
nuclear@3 27 static float view_scale = 0.2f;
nuclear@3 28 static Matrix4x4 view_matrix;
nuclear@3 29
nuclear@3 30 static float grid_size = 1.0;
nuclear@3 31 static SnapMode snap_mode;
nuclear@0 32
nuclear@0 33 static std::vector<Curve*> curves;
nuclear@2 34 static Curve *sel_curve; // selected curve being edited
nuclear@2 35 static Curve *new_curve; // new curve being entered
nuclear@2 36 static Curve *hover_curve; // curve the mouse is hovering over (click to select)
nuclear@2 37 static int sel_pidx = -1; // selected point of the selected or hovered-over curve
nuclear@0 38
nuclear@2 39 static Label *weight_label; // floating label for the cp weight
nuclear@0 40
nuclear@3 41 #ifdef DRAW_MOUSE_POINTER
nuclear@3 42 static Vector2 mouse_pointer;
nuclear@3 43 #endif
nuclear@3 44
nuclear@0 45
nuclear@0 46 bool app_init(int argc, char **argv)
nuclear@0 47 {
nuclear@5 48 glewInit();
nuclear@5 49
nuclear@0 50 glEnable(GL_MULTISAMPLE);
nuclear@0 51 glEnable(GL_CULL_FACE);
nuclear@2 52
nuclear@2 53 glEnable(GL_BLEND);
nuclear@2 54 glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
nuclear@0 55 return true;
nuclear@0 56 }
nuclear@0 57
nuclear@0 58 void app_cleanup()
nuclear@0 59 {
nuclear@0 60 for(size_t i=0; i<curves.size(); i++) {
nuclear@0 61 delete curves[i];
nuclear@0 62 }
nuclear@0 63 curves.clear();
nuclear@0 64 }
nuclear@0 65
nuclear@0 66 void app_draw()
nuclear@0 67 {
nuclear@1 68 glClearColor(0.1, 0.1, 0.1, 1);
nuclear@0 69 glClear(GL_COLOR_BUFFER_BIT);
nuclear@0 70
nuclear@0 71 glMatrixMode(GL_MODELVIEW);
nuclear@0 72 glLoadIdentity();
nuclear@3 73 glTranslatef(view_pan.x * view_scale, view_pan.y * view_scale, 0);
nuclear@0 74 glScalef(view_scale, view_scale, view_scale);
nuclear@0 75
nuclear@3 76 float max_aspect = std::max(win_aspect, 1.0f / win_aspect);
nuclear@3 77 draw_grid(max_aspect, grid_size);
nuclear@0 78
nuclear@0 79 for(size_t i=0; i<curves.size(); i++) {
nuclear@0 80 draw_curve(curves[i]);
nuclear@0 81 }
nuclear@0 82 if(new_curve) {
nuclear@0 83 draw_curve(new_curve);
nuclear@0 84 }
nuclear@3 85
nuclear@3 86 #ifdef DRAW_MOUSE_POINTER
nuclear@3 87 glPointSize(6.0);
nuclear@3 88 glBegin(GL_POINTS);
nuclear@3 89 glColor3f(0, 0, 1);
nuclear@3 90 glVertex2f(mouse_pointer.x, mouse_pointer.y);
nuclear@3 91 glEnd();
nuclear@3 92 #endif
nuclear@4 93
nuclear@4 94 glMatrixMode(GL_MODELVIEW);
nuclear@4 95 glLoadIdentity();
nuclear@4 96
nuclear@4 97 if(weight_label) {
nuclear@4 98 weight_label->draw();
nuclear@4 99 }
nuclear@0 100 }
nuclear@0 101
nuclear@0 102 static void draw_grid(float sz, float sep, float alpha)
nuclear@0 103 {
nuclear@0 104 float x = 0.0f;
nuclear@3 105 float s = 1.0 / view_scale;
nuclear@3 106
nuclear@3 107 sz *= s;
nuclear@3 108 sz += sep; // one more step for when we have non-zero fractional pan
nuclear@3 109 float end = std::min(sz, 100.0f * sep);
nuclear@3 110
nuclear@3 111 // fractional pan
nuclear@3 112 Vector2 pan = view_pan;
nuclear@3 113 Vector2 fpan = Vector2(fmod(pan.x, sep), fmod(pan.y, sep));
nuclear@3 114 Vector2 offset = fpan - pan;
nuclear@3 115
nuclear@3 116 glMatrixMode(GL_MODELVIEW);
nuclear@3 117 glPushMatrix();
nuclear@3 118 glTranslatef(offset.x, offset.y, 0);
nuclear@3 119
nuclear@3 120 glBegin(GL_LINES);
nuclear@3 121 glColor4f(0.35, 0.35, 0.35, alpha);
nuclear@3 122 while(x <= end) {
nuclear@3 123 glVertex2f(-end, x);
nuclear@3 124 glVertex2f(end, x);
nuclear@3 125 glVertex2f(-end, -x);
nuclear@3 126 glVertex2f(end, -x);
nuclear@3 127 glVertex2f(x, -end);
nuclear@3 128 glVertex2f(x, end);
nuclear@3 129 glVertex2f(-x, -end);
nuclear@3 130 glVertex2f(-x, end);
nuclear@3 131 x += sep;
nuclear@3 132 }
nuclear@3 133 glEnd();
nuclear@3 134 glPopMatrix();
nuclear@3 135
nuclear@0 136
nuclear@0 137 glLineWidth(1.0);
nuclear@0 138 glBegin(GL_LINES);
nuclear@0 139 glColor4f(0.6, 0.3, 0.2, alpha);
nuclear@3 140 glVertex2f(-sz + offset.x, 0);
nuclear@3 141 glVertex2f(sz + offset.x, 0);
nuclear@0 142 glColor4f(0.2, 0.3, 0.6, alpha);
nuclear@3 143 glVertex2f(0, -sz + offset.y);
nuclear@3 144 glVertex2f(0, sz + offset.y);
nuclear@0 145 glEnd();
nuclear@3 146
nuclear@0 147 }
nuclear@0 148
nuclear@0 149 static void draw_curve(const Curve *curve)
nuclear@0 150 {
nuclear@2 151 int numpt = curve->size();
nuclear@2 152 int segm = numpt * 16;
nuclear@0 153
nuclear@2 154 glLineWidth(curve == hover_curve ? 4.0 : 2.0);
nuclear@1 155 if(curve == sel_curve) {
nuclear@1 156 glColor3f(0.3, 0.4, 1.0);
nuclear@1 157 } else if(curve == new_curve) {
nuclear@1 158 glColor3f(1.0, 0.75, 0.3);
nuclear@1 159 } else {
nuclear@2 160 glColor3f(0.6, 0.6, 0.6);
nuclear@1 161 }
nuclear@0 162 glBegin(GL_LINE_STRIP);
nuclear@0 163 for(int i=0; i<segm; i++) {
nuclear@0 164 float t = (float)i / (float)(segm - 1);
nuclear@0 165 Vector2 v = curve->interpolate(t);
nuclear@0 166 glVertex2f(v.x, v.y);
nuclear@0 167 }
nuclear@0 168 glEnd();
nuclear@0 169 glLineWidth(1.0);
nuclear@0 170
nuclear@2 171 glPointSize(curve == hover_curve ? 10.0 : 7.0);
nuclear@0 172 glBegin(GL_POINTS);
nuclear@1 173 if(curve == new_curve) {
nuclear@1 174 glColor3f(1.0, 0.0, 0.0);
nuclear@1 175 } else {
nuclear@1 176 glColor3f(0.6, 0.3, 0.2);
nuclear@1 177 }
nuclear@0 178 for(int i=0; i<numpt; i++) {
nuclear@1 179 if(curve == sel_curve) {
nuclear@1 180 if(i == sel_pidx) {
nuclear@1 181 glColor3f(1.0, 0.2, 0.1);
nuclear@1 182 } else {
nuclear@1 183 glColor3f(0.2, 1.0, 0.2);
nuclear@1 184 }
nuclear@1 185 }
nuclear@0 186 Vector2 pt = curve->get_point(i);
nuclear@0 187 glVertex2f(pt.x, pt.y);
nuclear@0 188 }
nuclear@0 189 glEnd();
nuclear@0 190 glPointSize(1.0);
nuclear@0 191 }
nuclear@0 192
nuclear@0 193 void app_reshape(int x, int y)
nuclear@0 194 {
nuclear@0 195 win_width = x;
nuclear@0 196 win_height = y;
nuclear@0 197 win_aspect = (float)x / (float)y;
nuclear@0 198
nuclear@0 199 glViewport(0, 0, x, y);
nuclear@0 200 glMatrixMode(GL_PROJECTION);
nuclear@0 201 glLoadIdentity();
nuclear@0 202 glOrtho(-win_aspect, win_aspect, -1, 1, -1, 1);
nuclear@0 203 }
nuclear@0 204
nuclear@0 205 void app_keyboard(int key, bool pressed)
nuclear@0 206 {
nuclear@0 207 if(pressed) {
nuclear@0 208 switch(key) {
nuclear@1 209 case 'q':
nuclear@1 210 case 'Q':
nuclear@1 211 exit(0);
nuclear@1 212
nuclear@0 213 case 27:
nuclear@1 214 if(new_curve) {
nuclear@1 215 delete new_curve;
nuclear@1 216 new_curve = 0;
nuclear@1 217 post_redisplay();
nuclear@1 218 }
nuclear@1 219 break;
nuclear@0 220
nuclear@0 221 case 'l':
nuclear@0 222 case 'L':
nuclear@0 223 if(sel_curve) {
nuclear@0 224 sel_curve->set_type(CURVE_LINEAR);
nuclear@0 225 post_redisplay();
nuclear@0 226 }
nuclear@1 227 if(new_curve) {
nuclear@1 228 new_curve->set_type(CURVE_LINEAR);
nuclear@1 229 post_redisplay();
nuclear@1 230 }
nuclear@0 231 break;
nuclear@0 232
nuclear@0 233 case 'b':
nuclear@0 234 case 'B':
nuclear@0 235 if(sel_curve) {
nuclear@0 236 sel_curve->set_type(CURVE_BSPLINE);
nuclear@0 237 post_redisplay();
nuclear@0 238 }
nuclear@1 239 if(new_curve) {
nuclear@1 240 new_curve->set_type(CURVE_BSPLINE);
nuclear@1 241 post_redisplay();
nuclear@1 242 }
nuclear@0 243 break;
nuclear@0 244
nuclear@0 245 case 'h':
nuclear@0 246 case 'H':
nuclear@0 247 if(sel_curve) {
nuclear@0 248 sel_curve->set_type(CURVE_HERMITE);
nuclear@0 249 post_redisplay();
nuclear@0 250 }
nuclear@1 251 if(new_curve) {
nuclear@1 252 new_curve->set_type(CURVE_HERMITE);
nuclear@1 253 post_redisplay();
nuclear@1 254 }
nuclear@0 255 break;
nuclear@6 256
nuclear@6 257 case 'e':
nuclear@6 258 case 'E':
nuclear@6 259 // TODO: GUI for filename at least
nuclear@6 260 if(!save_curves(stdout, &curves[0], (int)curves.size())) {
nuclear@6 261 fprintf(stderr, "failed to export curves\n");
nuclear@6 262 }
nuclear@6 263 printf("exported %d curves\n", (int)curves.size());
nuclear@6 264 break;
nuclear@0 265 }
nuclear@0 266 }
nuclear@3 267
nuclear@3 268
nuclear@3 269 switch(key) {
nuclear@3 270 case 's':
nuclear@3 271 snap_mode = pressed ? SNAP_GRID : SNAP_NONE;
nuclear@3 272 break;
nuclear@3 273
nuclear@3 274 case 'S':
nuclear@3 275 snap_mode = pressed ? SNAP_POINT : SNAP_NONE;
nuclear@3 276 break;
nuclear@3 277
nuclear@3 278 default:
nuclear@3 279 break;
nuclear@3 280 }
nuclear@3 281 }
nuclear@3 282
nuclear@3 283 static void calc_view_matrix()
nuclear@3 284 {
nuclear@3 285 view_matrix.reset_identity();
nuclear@3 286 view_matrix.scale(Vector3(view_scale, view_scale, view_scale));
nuclear@3 287 view_matrix.translate(Vector3(view_pan.x, view_pan.y, 0.0));
nuclear@0 288 }
nuclear@0 289
nuclear@0 290 static Vector2 pixel_to_uv(int x, int y)
nuclear@0 291 {
nuclear@0 292 float u = win_aspect * (2.0 * (float)x / (float)win_width - 1.0);
nuclear@0 293 float v = 1.0 - 2.0 * (float)y / (float)win_height;
nuclear@3 294
nuclear@3 295 u = u / view_scale - view_pan.x;
nuclear@3 296 v = v / view_scale - view_pan.y;
nuclear@0 297 return Vector2(u, v);
nuclear@3 298 /*
nuclear@3 299 Matrix4x4 inv_view_matrix = view_matrix.inverse();
nuclear@3 300 Vector4 res = Vector4(u, v, 0.0, 1.0).transformed(inv_view_matrix);
nuclear@3 301
nuclear@3 302 return Vector2(res.x, res.y);
nuclear@3 303 */
nuclear@0 304 }
nuclear@0 305
nuclear@0 306 static int prev_x, prev_y;
nuclear@0 307 static int click_pos[8][2];
nuclear@0 308 static unsigned int bnstate;
nuclear@0 309
nuclear@0 310 #define BNBIT(x) (1 << (x))
nuclear@0 311
nuclear@0 312 void app_mouse_button(int bn, bool pressed, int x, int y)
nuclear@0 313 {
nuclear@0 314 prev_x = x;
nuclear@0 315 prev_y = y;
nuclear@0 316 if(pressed) {
nuclear@0 317 bnstate |= BNBIT(bn);
nuclear@0 318 } else {
nuclear@0 319 bnstate &= ~BNBIT(bn);
nuclear@0 320 }
nuclear@0 321
nuclear@0 322 if(pressed) {
nuclear@0 323 click_pos[bn][0] = x;
nuclear@0 324 click_pos[bn][1] = y;
nuclear@0 325 } else {
nuclear@0 326 int dx = x - click_pos[bn][0];
nuclear@0 327 int dy = y - click_pos[bn][1];
nuclear@0 328
nuclear@0 329 if(abs(dx) + abs(dy) < 3) {
nuclear@0 330 Vector2 uv = pixel_to_uv(x, y);
nuclear@0 331 on_click(bn, uv.x, uv.y);
nuclear@0 332 }
nuclear@0 333
nuclear@0 334 if(!(bnstate & BNBIT(2))) {
nuclear@0 335 delete weight_label;
nuclear@0 336 weight_label = 0;
nuclear@0 337 post_redisplay();
nuclear@0 338 }
nuclear@0 339 }
nuclear@0 340 }
nuclear@0 341
nuclear@2 342 static bool point_hit_test(const Vector2 &pos, Curve **curveret, int *pidxret)
nuclear@0 343 {
nuclear@3 344 float thres = 0.02 / view_scale;
nuclear@0 345
nuclear@0 346 for(size_t i=0; i<curves.size(); i++) {
nuclear@2 347 int pidx = curves[i]->nearest_point(pos);
nuclear@0 348 if(pidx == -1) continue;
nuclear@0 349
nuclear@0 350 Vector2 cp = curves[i]->get_point(pidx);
nuclear@2 351 if((cp - pos).length_sq() < thres * thres) {
nuclear@2 352 *curveret = curves[i];
nuclear@2 353 *pidxret = pidx;
nuclear@2 354 return true;
nuclear@0 355 }
nuclear@0 356 }
nuclear@2 357 *curveret = 0;
nuclear@2 358 *pidxret = -1;
nuclear@2 359 return false;
nuclear@0 360 }
nuclear@0 361
nuclear@3 362 static Vector2 snap(const Vector2 &p)
nuclear@3 363 {
nuclear@3 364 switch(snap_mode) {
nuclear@3 365 case SNAP_GRID:
nuclear@3 366 return Vector2(round(p.x / grid_size) * grid_size, round(p.y / grid_size) * grid_size);
nuclear@3 367 case SNAP_POINT:
nuclear@3 368 // TODO
nuclear@3 369 default:
nuclear@3 370 break;
nuclear@3 371 }
nuclear@3 372 return p;
nuclear@3 373 }
nuclear@3 374
nuclear@0 375 void app_mouse_motion(int x, int y)
nuclear@0 376 {
nuclear@3 377 Vector2 prev_uv = pixel_to_uv(prev_x, prev_y);
nuclear@3 378
nuclear@0 379 int dx = x - prev_x;
nuclear@0 380 int dy = y - prev_y;
nuclear@0 381 prev_x = x;
nuclear@0 382 prev_y = y;
nuclear@0 383
nuclear@0 384 if(!dx && !dy) return;
nuclear@0 385
nuclear@2 386 Vector2 uv = pixel_to_uv(x, y);
nuclear@3 387 #ifdef DRAW_MOUSE_POINTER
nuclear@3 388 mouse_pointer = uv;
nuclear@3 389 post_redisplay();
nuclear@3 390 #endif
nuclear@2 391
nuclear@2 392 /* when entering a new curve, have the last (extra) point following
nuclear@2 393 * the mouse until it's entered by a click (see on_click).
nuclear@2 394 */
nuclear@3 395 if(new_curve) {
nuclear@3 396 new_curve->move_point(new_curve->size() - 1, snap(uv));
nuclear@2 397 post_redisplay();
nuclear@2 398 }
nuclear@2 399
nuclear@4 400 if(!new_curve && !bnstate) {
nuclear@3 401 // not dragging, highlight curve under mouse
nuclear@2 402 point_hit_test(uv, &hover_curve, &sel_pidx);
nuclear@0 403 post_redisplay();
nuclear@0 404
nuclear@3 405 } else {
nuclear@3 406 // we're dragging with one or more buttons held down
nuclear@0 407
nuclear@3 408 if(sel_curve && sel_pidx != -1) {
nuclear@3 409 // we have a curve and a point of the curve selected
nuclear@0 410
nuclear@3 411 if(bnstate & BNBIT(0)) {
nuclear@3 412 // dragging point with left button: move it
nuclear@3 413 sel_curve->move_point(sel_pidx, snap(uv));
nuclear@3 414 post_redisplay();
nuclear@0 415 }
nuclear@3 416
nuclear@3 417 if(bnstate & BNBIT(2)) {
nuclear@3 418 // dragging point with right button: change weight
nuclear@3 419 float w = sel_curve->get_weight(sel_pidx);
nuclear@3 420 w -= dy * 0.01;
nuclear@3 421 if(w < FLT_MIN) w = FLT_MIN;
nuclear@3 422 sel_curve->set_weight(sel_pidx, w);
nuclear@3 423
nuclear@3 424 // popup floating weight label if not already there
nuclear@3 425 if(!weight_label) {
nuclear@3 426 weight_label = new Label;
nuclear@3 427 }
nuclear@3 428 weight_label->set_position(uv);
nuclear@3 429 weight_label->set_textf("w=%g", w);
nuclear@3 430 post_redisplay();
nuclear@3 431 }
nuclear@3 432 } else {
nuclear@3 433 // no selection, we're dragging in empty space: manipulate viewport
nuclear@3 434 Vector2 dir = uv - prev_uv;
nuclear@3 435
nuclear@3 436 if(bnstate & (BNBIT(0) | BNBIT(1))) {
nuclear@3 437 // panning
nuclear@3 438 view_pan += dir;
nuclear@3 439 calc_view_matrix();
nuclear@3 440 post_redisplay();
nuclear@3 441 }
nuclear@3 442 if(bnstate & BNBIT(2)) {
nuclear@3 443 // zooming
nuclear@3 444 view_scale -= ((float)dy / (float)win_height) * view_scale * 5.0;
nuclear@3 445 if(view_scale < 1e-4) view_scale = 1e-4;
nuclear@3 446 calc_view_matrix();
nuclear@3 447 post_redisplay();
nuclear@3 448 }
nuclear@0 449 }
nuclear@0 450 }
nuclear@0 451 }
nuclear@0 452
nuclear@0 453 static void on_click(int bn, float u, float v)
nuclear@0 454 {
nuclear@2 455 Vector2 uv = Vector2(u, v);
nuclear@2 456
nuclear@0 457 switch(bn) {
nuclear@2 458 case 0: // ------- LEFT CLICK ------
nuclear@2 459 if(hover_curve) {
nuclear@2 460 // if we're hovering: click selects
nuclear@2 461 sel_curve = hover_curve;
nuclear@2 462 hover_curve = 0;
nuclear@2 463 } else if(sel_curve) {
nuclear@2 464 // if we have a selected curve: click adds point (enter new_curve mode)
nuclear@2 465 std::vector<Curve*>::iterator it = std::find(curves.begin(), curves.end(), sel_curve);
nuclear@2 466 assert(it != curves.end());
nuclear@2 467 curves.erase(it, it + 1);
nuclear@2 468
nuclear@2 469 new_curve = sel_curve;
nuclear@2 470 sel_curve = 0;
nuclear@2 471 sel_pidx = -1;
nuclear@2 472
nuclear@2 473 new_curve->add_point(uv);
nuclear@2 474 } else {
nuclear@2 475 // otherwise, click starts a new curve
nuclear@2 476 if(!new_curve) {
nuclear@2 477 new_curve = new Curve;
nuclear@2 478 new_curve->add_point(uv);
nuclear@2 479 }
nuclear@2 480 new_curve->add_point(uv);
nuclear@0 481 }
nuclear@0 482 post_redisplay();
nuclear@0 483 break;
nuclear@0 484
nuclear@2 485 case 2: // ------- RIGHT CLICK ------
nuclear@2 486 if(new_curve) {
nuclear@2 487 // in new-curve mode: finish curve (cancels last floating segment)
nuclear@2 488 new_curve->remove_point(new_curve->size() - 1);
nuclear@2 489 if(new_curve->empty()) {
nuclear@2 490 delete new_curve;
nuclear@2 491 } else {
nuclear@2 492 curves.push_back(new_curve);
nuclear@2 493 }
nuclear@2 494 new_curve = 0;
nuclear@2 495
nuclear@2 496 } else if(sel_curve) {
nuclear@2 497 // in selected curve mode: delete control point or unselect
nuclear@2 498 Curve *hit_curve;
nuclear@2 499 int hit_pidx;
nuclear@2 500 if(point_hit_test(uv, &hit_curve, &hit_pidx) && hit_curve == sel_curve) {
nuclear@2 501 hit_curve->remove_point(hit_pidx);
nuclear@2 502 sel_pidx = -1;
nuclear@2 503 } else {
nuclear@2 504 sel_curve = 0;
nuclear@2 505 sel_pidx = -1;
nuclear@2 506 }
nuclear@2 507 }
nuclear@0 508 post_redisplay();
nuclear@0 509 break;
nuclear@0 510
nuclear@0 511 default:
nuclear@0 512 break;
nuclear@0 513 }
nuclear@0 514 }