select and move exhibits
[laserbrain_demo] / src / app.cc
1 #include <stdio.h>
2 #include <assert.h>
3 #include <goatvr.h>
4 #include "app.h"
5 #include "opengl.h"
6 #include "sdr.h"
7 #include "texture.h"
8 #include "mesh.h"
9 #include "meshgen.h"
10 #include "scene.h"
11 #include "metascene.h"
12 #include "datamap.h"
13 #include "ui.h"
14 #include "opt.h"
15 #include "post.h"
16 #include "renderer.h"
17 #include "avatar.h"
18 #include "vrinput.h"
19 #include "exman.h"
20 #include "blob_exhibit.h"
21 #include "dbg_gui.h"
22
23 #define NEAR_CLIP       5.0
24 #define FAR_CLIP        10000.0
25
26 static void draw_scene();
27 static void toggle_flight();
28 static void calc_framerate();
29 static Ray calc_pick_ray(int x, int y);
30
31 long time_msec;
32 int win_width, win_height;
33 float win_aspect;
34 bool fb_srgb;
35 bool opt_gear_wireframe;
36
37 TextureSet texman;
38 SceneSet sceneman;
39
40 unsigned int sdr_ltmap, sdr_ltmap_notex;
41
42 int fpexcept_enabled;
43
44 static Avatar avatar;
45
46 static float cam_dist = 0.0;
47 static float floor_y;   // last floor height
48 static float user_eye_height = 165;
49
50 static float walk_speed = 300.0f;
51 static float mouse_speed = 0.5f;
52 static bool show_walk_mesh, noclip = false;
53
54 static bool have_headtracking, have_handtracking, should_swap;
55
56 static int prev_mx, prev_my;
57 static bool bnstate[8];
58 static bool keystate[256];
59 static bool gpad_bnstate[64];
60 static Vec2 joy_move, joy_look;
61 static float joy_deadzone = 0.01;
62
63 static float framerate;
64
65 static Mat4 view_matrix, mouse_view_matrix, proj_matrix;
66 static MetaScene *mscn;
67 static unsigned int sdr_post_gamma;
68
69 static long prev_msec;
70
71 static ExhibitManager *exman;
72 static BlobExhibit *blobs;
73 static bool show_blobs;
74
75 ExSelection exsel_grab, exsel_hover;
76
77 static Renderer *rend;
78
79 static Ray last_pick_ray;
80
81
82 bool app_init(int argc, char **argv)
83 {
84         set_log_file("demo.log");
85
86         char *env = getenv("FPEXCEPT");
87         if(env && atoi(env)) {
88                 info_log("enabling floating point exceptions\n");
89                 fpexcept_enabled = 1;
90                 enable_fpexcept();
91         }
92
93         if(init_opengl() == -1) {
94                 return false;
95         }
96
97         if(!init_options(argc, argv, "demo.conf")) {
98                 return false;
99         }
100         app_resize(opt.width, opt.height);
101         app_fullscreen(opt.fullscreen);
102
103         if(opt.vr) {
104                 if(goatvr_init() == -1) {
105                         return false;
106                 }
107                 goatvr_set_origin_mode(GOATVR_HEAD);
108                 goatvr_set_units_scale(100.0f);
109
110                 goatvr_startvr();
111                 should_swap = goatvr_should_swap() != 0;
112                 user_eye_height = goatvr_get_eye_height();
113                 have_headtracking = goatvr_have_headtracking();
114                 have_handtracking = goatvr_have_handtracking();
115
116                 goatvr_recenter();
117         }
118
119         if(fb_srgb) {
120                 int srgb_capable;
121                 glGetIntegerv(GL_FRAMEBUFFER_SRGB_CAPABLE_EXT, &srgb_capable);
122                 printf("Framebuffer %s sRGB-capable\n", srgb_capable ? "is" : "is not");
123                 if(srgb_capable) {
124                         glEnable(GL_FRAMEBUFFER_SRGB);
125                 } else {
126                         fb_srgb = 0;
127                 }
128         }
129
130         glEnable(GL_MULTISAMPLE);
131         glEnable(GL_DEPTH_TEST);
132         glEnable(GL_CULL_FACE);
133         glEnable(GL_NORMALIZE);
134
135         if(!init_debug_gui()) {
136                 return false;
137         }
138
139         Mesh::use_custom_sdr_attr = false;
140
141         float ambient[] = {0.0, 0.0, 0.0, 0.0};
142         glLightModelfv(GL_LIGHT_MODEL_AMBIENT, ambient);
143
144         glClearColor(1, 1, 1, 1);
145
146         init_audio();
147
148         if(!init_vrhands()) {
149                 return false;
150         }
151
152         mscn = new MetaScene;
153         if(!mscn->load(opt.scenefile ? opt.scenefile : "data/museum.scene")) {
154                 return false;
155         }
156
157         avatar.pos = mscn->start_pos;
158         Vec3 dir = rotate(Vec3(0, 0, 1), mscn->start_rot);
159         dir.y = 0;
160         avatar.body_rot = rad_to_deg(acos(dot(dir, Vec3(0, 0, 1))));
161
162         exman = new ExhibitManager;
163         /*
164         if(!exman->load(mscn, "data/exhibits")) {
165                 //return false;
166         }
167         */
168
169         blobs = new BlobExhibit;
170         blobs->node = new SceneNode;
171         blobs->init();
172         blobs->node->set_position(Vec3(-680, 160, -100));
173         blobs->node->set_scaling(Vec3(28, 28, 28));
174         blobs->node->update(0);
175
176         exman->add(blobs);
177
178         if(!(sdr_ltmap_notex = create_program_load("sdr/lightmap.v.glsl", "sdr/lightmap-notex.p.glsl"))) {
179                 return false;
180         }
181         set_uniform_int(sdr_ltmap_notex, "texmap", MTL_TEX_DIFFUSE);
182         set_uniform_int(sdr_ltmap_notex, "lightmap", MTL_TEX_LIGHTMAP);
183
184         if(!(sdr_ltmap = create_program_load("sdr/lightmap.v.glsl", "sdr/lightmap-tex.p.glsl"))) {
185                 return false;
186         }
187         set_uniform_int(sdr_ltmap, "texmap", MTL_TEX_DIFFUSE);
188         set_uniform_int(sdr_ltmap, "lightmap", MTL_TEX_LIGHTMAP);
189
190         if(!fb_srgb) {
191                 sdr_post_gamma = create_program_load("sdr/post_gamma.v.glsl", "sdr/post_gamma.p.glsl");
192         }
193
194         rend = new Renderer;
195         rend->set_scene(mscn);
196
197         glUseProgram(0);
198
199         if(opt.vr || opt.fullscreen) {
200                 app_grab_mouse(true);
201         }
202
203         if(mscn->music && opt.music) {
204                 mscn->music->play(AUDIO_PLAYMODE_LOOP);
205         }
206         return true;
207 }
208
209 void app_cleanup()
210 {
211         if(mscn->music) {
212                 mscn->music->stop();
213         }
214         destroy_audio();
215
216         app_grab_mouse(false);
217         if(opt.vr) {
218                 goatvr_shutdown();
219         }
220         destroy_vrhands();
221
222         delete rend;
223
224         delete exman;
225
226         texman.clear();
227         sceneman.clear();
228
229         cleanup_debug_gui();
230 }
231
232 static bool constrain_walk_mesh(const Vec3 &v, Vec3 *newv)
233 {
234         Mesh *wm = mscn->walk_mesh;
235         if(!wm) {
236                 *newv = v;
237                 return true;
238         }
239
240         Ray downray = Ray(v, Vec3(0, -1, 0));
241         HitPoint hit;
242         if(mscn->walk_mesh->intersect(downray, &hit)) {
243                 *newv = hit.pos;
244                 newv->y += user_eye_height;
245                 return true;
246         }
247         return false;
248 }
249
250 static void update(float dt)
251 {
252         texman.update();
253         sceneman.update();
254
255         mscn->update(dt);
256         exman->update(dt);
257
258         float speed = walk_speed * dt;
259         Vec3 dir;
260
261         // joystick
262         float jdeadsq = joy_deadzone * joy_deadzone;
263         float jmove_lensq = length_sq(joy_move);
264         float jlook_lensq = length_sq(joy_look);
265
266         if(jmove_lensq > jdeadsq) {
267                 float len = sqrt(jmove_lensq);
268                 jmove_lensq -= jdeadsq;
269
270                 float mag = len * len;
271                 dir.x += mag * joy_move.x / len * 2.0 * speed;
272                 dir.z += mag * joy_move.y / len * 2.0 * speed;
273         }
274         if(jlook_lensq > jdeadsq) {
275                 float len = sqrt(jlook_lensq);
276                 jlook_lensq -= jdeadsq;
277
278                 float mag = len * len;
279                 avatar.body_rot += mag * joy_look.x / len * 200.0 * dt;
280                 avatar.head_alt += mag * joy_look.y / len * 100.0 * dt;
281                 if(avatar.head_alt < -90.0f) avatar.head_alt = -90.0f;
282                 if(avatar.head_alt > 90.0f) avatar.head_alt = 90.0f;
283         }
284
285         // keyboard move
286         if(keystate[(int)'w']) {
287                 dir.z -= speed;
288         }
289         if(keystate[(int)'s']) {
290                 dir.z += speed;
291         }
292         if(keystate[(int)'d']) {
293                 dir.x += speed;
294         }
295         if(keystate[(int)'a']) {
296                 dir.x -= speed;
297         }
298         if(keystate[(int)'q'] || gpad_bnstate[GPAD_UP]) {
299                 avatar.pos.y += speed;
300         }
301         if(keystate[(int)'z'] || gpad_bnstate[GPAD_DOWN]) {
302                 avatar.pos.y -= speed;
303         }
304
305         float theta = M_PI * avatar.body_rot / 180.0f;
306         Vec3 newpos;
307         newpos.x = avatar.pos.x + cos(theta) * dir.x - sin(theta) * dir.z;
308         newpos.y = avatar.pos.y;
309         newpos.z = avatar.pos.z + sin(theta) * dir.x + cos(theta) * dir.z;
310
311         if(noclip) {
312                 avatar.pos = newpos;
313         } else {
314                 if(!constrain_walk_mesh(newpos, &avatar.pos)) {
315                         float dtheta = M_PI / 32.0;
316                         float theta = dtheta;
317                         Vec2 dir2d = newpos.xz() - avatar.pos.xz();
318
319                         for(int i=0; i<16; i++) {
320                                 Vec2 dvec = rotate(dir2d, theta);
321                                 Vec3 pos = avatar.pos + Vec3(dvec.x, 0, dvec.y);
322                                 if(constrain_walk_mesh(pos, &avatar.pos)) {
323                                         break;
324                                 }
325                                 dvec = rotate(dir2d, -theta);
326                                 pos = avatar.pos + Vec3(dvec.x, 0, dvec.y);
327                                 if(constrain_walk_mesh(pos, &avatar.pos)) {
328                                         break;
329                                 }
330                                 theta += dtheta;
331                         }
332                 }
333                 floor_y = avatar.pos.y - user_eye_height;
334         }
335
336         // TODO move to avatar
337         // calculate mouselook view matrix
338         mouse_view_matrix = Mat4::identity;
339         mouse_view_matrix.pre_translate(0, 0, -cam_dist);
340         if(!have_headtracking) {
341                 mouse_view_matrix.pre_rotate_x(deg_to_rad(avatar.head_alt));
342         }
343         mouse_view_matrix.pre_rotate_y(deg_to_rad(avatar.body_rot));
344         mouse_view_matrix.pre_translate(-avatar.pos.x, -avatar.pos.y, -avatar.pos.z);
345
346         // check if an exhibit is hovered-over by mouse or 6dof
347         // XXX note: using previous view/proj matrix lattency shouldn't be an issue but
348         //           make sure state-creep doesn't get us
349         // XXX also this mouse-picking probably should only be active in non-VR mode
350         Ray ray = calc_pick_ray(prev_mx, prev_my);
351         exsel_hover = exman->select(ray);
352
353         // update hand-tracking
354         if(have_handtracking) {
355                 update_vrhands(&avatar);
356         }
357 }
358
359 void app_display()
360 {
361         float dt = (float)(time_msec - prev_msec) / 1000.0f;
362         prev_msec = time_msec;
363
364         if(debug_gui) {
365                 ImGui::GetIOPtr()->DeltaTime = dt;
366                 ImGui::NewFrame();
367
368                 ImGui::ShowTestWindow();
369         }
370
371         if(opt.vr) {
372                 // VR mode
373                 goatvr_draw_start();
374                 glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
375
376                 update(dt);
377
378                 for(int i=0; i<2; i++) {
379                         // for each eye
380                         goatvr_draw_eye(i);
381
382                         proj_matrix = goatvr_projection_matrix(i, NEAR_CLIP, FAR_CLIP);
383                         glMatrixMode(GL_PROJECTION);
384                         glLoadMatrixf(proj_matrix[0]);
385
386                         view_matrix = mouse_view_matrix * Mat4(goatvr_view_matrix(i));
387                         glMatrixMode(GL_MODELVIEW);
388                         glLoadMatrixf(view_matrix[0]);
389
390                         draw_scene();
391                         if(have_handtracking) {
392                                 draw_vrhands();
393                         }
394
395                         if(debug_gui) {
396                                 ImGui::Render();
397                         }
398                 }
399                 goatvr_draw_done();
400
401                 if(should_swap) {
402                         app_swap_buffers();
403                 }
404
405         } else {
406                 glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
407
408                 update(dt);
409
410                 proj_matrix.perspective(deg_to_rad(50.0), win_aspect, NEAR_CLIP, FAR_CLIP);
411                 glMatrixMode(GL_PROJECTION);
412                 glLoadMatrixf(proj_matrix[0]);
413
414                 view_matrix = mouse_view_matrix;
415                 glMatrixMode(GL_MODELVIEW);
416                 glLoadMatrixf(view_matrix[0]);
417
418                 draw_scene();
419
420                 if(!fb_srgb && sdr_post_gamma) {
421                         slow_post(sdr_post_gamma);
422                         glUseProgram(0);
423                 }
424
425                 if(debug_gui) {
426                         ImGui::Render();
427                 }
428                 app_swap_buffers();
429         }
430         assert(glGetError() == GL_NO_ERROR);
431
432         calc_framerate();
433 }
434
435
436 static void draw_scene()
437 {
438         rend->draw();
439         exman->draw();
440
441         /*
442         if(have_handtracking) {
443                 Mat4 head_xform = inverse(mouse_view_matrix);//goatvr_head_matrix();
444                 Mat4 head_dir_xform = head_xform.upper3x3();
445
446                 glUseProgram(0);
447                 glPushAttrib(GL_ENABLE_BIT);
448                 glDisable(GL_LIGHTING);
449                 glBegin(GL_LINES);
450                 for(int i=0; i<2; i++) {
451                         if(hand[i].valid) {
452                                 glColor3f(i, 1 - i, i);
453                         } else {
454                                 glColor3f(0.5, 0.5, 0.5);
455                         }
456                         Vec3 v = head_xform * hand[i].pos;
457                         Vec3 dir = head_dir_xform * rotate(Vec3(0, 0, -1), hand[i].rot) * 20.0f;
458                         Vec3 up = head_dir_xform * rotate(Vec3(0, 1, 0), hand[i].rot) * 10.0f;
459                         Vec3 right = head_dir_xform * rotate(Vec3(1, 0, 0), hand[i].rot) * 10.0f;
460
461                         glVertex3f(v.x, v.y, v.z);
462                         glVertex3f(v.x + dir.x, v.y + dir.y, v.z + dir.z);
463                         glVertex3f(v.x - right.x, v.y - right.y, v.z - right.z);
464                         glVertex3f(v.x + right.x, v.y + right.y, v.z + right.z);
465                         glVertex3f(v.x - up.x, v.y - up.y, v.z - up.z);
466                         glVertex3f(v.x + up.x, v.y + up.y, v.z + up.z);
467                 }
468                 glEnd();
469                 glPopAttrib();
470         }
471         */
472
473         if(show_walk_mesh && mscn->walk_mesh) {
474                 glPushAttrib(GL_ENABLE_BIT);
475                 glEnable(GL_BLEND);
476                 glBlendFunc(GL_ONE, GL_ONE);
477                 glEnable(GL_POLYGON_OFFSET_FILL);
478
479                 glUseProgram(0);
480
481                 glPolygonOffset(-1, 1);
482                 glDepthMask(0);
483
484                 glColor3f(0.3, 0.08, 0.01);
485                 mscn->walk_mesh->draw();
486
487                 glDepthMask(1);
488
489                 glPopAttrib();
490         }
491
492         print_text(Vec2(9 * win_width / 10, 20), Vec3(1, 1, 0), "fps: %.1f", framerate);
493         draw_ui();
494 }
495
496
497 void app_reshape(int x, int y)
498 {
499         glViewport(0, 0, x, y);
500         goatvr_set_fb_size(x, y, 1.0f);
501         debug_gui_reshape(x, y);
502 }
503
504 void app_keyboard(int key, bool pressed)
505 {
506         unsigned int mod = app_get_modifiers();
507
508         if(debug_gui && !(pressed && (key == '`' || key == 27))) {
509                 debug_gui_key(key, pressed, mod);
510                 return; // ignore all keystrokes when GUI is visible
511         }
512
513         if(pressed) {
514                 switch(key) {
515                 case 27:
516                         app_quit();
517                         break;
518
519                 case '\n':
520                 case '\r':
521                         if(mod & MOD_ALT) {
522                                 app_toggle_fullscreen();
523                         }
524                         break;
525
526                 case '`':
527                         debug_gui = !debug_gui;
528                         show_message("debug gui %s", debug_gui ? "enabled" : "disabled");
529                         break;
530
531                 case 'm':
532                         app_toggle_grab_mouse();
533                         show_message("mouse %s", app_is_mouse_grabbed() ? "grabbed" : "released");
534                         break;
535
536                 case 'w':
537                         if(mod & MOD_CTRL) {
538                                 show_walk_mesh = !show_walk_mesh;
539                                 show_message("walk mesh: %s", show_walk_mesh ? "on" : "off");
540                         }
541                         break;
542
543                 case 'c':
544                         if(mod & MOD_CTRL) {
545                                 noclip = !noclip;
546                                 show_message(noclip ? "no clip" : "clip");
547                         }
548                         break;
549
550                 case 'f':
551                         toggle_flight();
552                         break;
553
554                 case 'p':
555                         if(mod & MOD_CTRL) {
556                                 fb_srgb = !fb_srgb;
557                                 show_message("gamma correction for non-sRGB framebuffers: %s\n", fb_srgb ? "off" : "on");
558                         }
559                         break;
560
561                 case '=':
562                         walk_speed *= 1.25;
563                         show_message("walk speed: %g", walk_speed);
564                         break;
565
566                 case '-':
567                         walk_speed *= 0.75;
568                         show_message("walk speed: %g", walk_speed);
569                         break;
570
571                 case ']':
572                         mouse_speed *= 1.2;
573                         show_message("mouse speed: %g", mouse_speed);
574                         break;
575
576                 case '[':
577                         mouse_speed *= 0.8;
578                         show_message("mouse speed: %g", mouse_speed);
579                         break;
580
581                 case 'b':
582                         show_blobs = !show_blobs;
583                         show_message("blobs: %s\n", show_blobs ? "on" : "off");
584                         break;
585
586                 case ' ':
587                         goatvr_recenter();
588                         show_message("VR recenter\n");
589                         break;
590
591                 case 'x':
592                         exman->load(mscn, "data/exhibits");
593                         break;
594                 }
595         }
596
597         if(key < 256 && !(mod & (MOD_CTRL | MOD_ALT))) {
598                 keystate[key] = pressed;
599         }
600 }
601
602 void app_mouse_button(int bn, bool pressed, int x, int y)
603 {
604         if(debug_gui) {
605                 debug_gui_mbutton(bn, pressed, x, y);
606                 return; // ignore mouse events while GUI is visible
607         }
608
609         prev_mx = x;
610         prev_my = y;
611         bnstate[bn] = pressed;
612
613         if(bn == 0) {
614                 if(!pressed) {
615                         if(exsel_grab.ex) {
616                                 exsel_grab.ex = 0;
617                         } else {
618                                 Ray ray = calc_pick_ray(x, y);
619                                 exsel_grab = exman->select(ray);
620                         }
621                 }
622         }
623 }
624
625 static inline void mouse_look(float dx, float dy)
626 {
627         float scrsz = (float)win_height;
628         avatar.body_rot += dx * 512.0 / scrsz;
629         avatar.head_alt += dy * 512.0 / scrsz;
630
631         if(avatar.head_alt < -90) avatar.head_alt = -90;
632         if(avatar.head_alt > 90) avatar.head_alt = 90;
633 }
634
635 static void mouse_zoom(float dx, float dy)
636 {
637         cam_dist += dy * 0.1;
638         if(cam_dist < 0.0) cam_dist = 0.0;
639 }
640
641 void app_mouse_motion(int x, int y)
642 {
643         if(debug_gui) {
644                 debug_gui_mmotion(x, y);
645                 return; // ignore mouse events while GUI is visible
646         }
647
648         int dx = x - prev_mx;
649         int dy = y - prev_my;
650         prev_mx = x;
651         prev_my = y;
652
653         if(!dx && !dy) return;
654
655         if(exsel_grab.ex) {
656                 Vec3 pos = exsel_grab.ex->node->get_node_position();
657                 Vec3 dir = transpose(view_matrix.upper3x3()) * Vec3(dx * 1.0, dy * -1.0, 0);
658
659                 exsel_grab.ex->node->set_position(pos + dir);
660         }
661
662         if(bnstate[2]) {
663                 mouse_look(dx, dy);
664         }
665 }
666
667 void app_mouse_delta(int dx, int dy)
668 {
669         if(bnstate[2]) {
670                 mouse_zoom(dx * mouse_speed, dy * mouse_speed);
671         } else {
672                 mouse_look(dx * mouse_speed, dy * mouse_speed);
673         }
674 }
675
676 void app_mouse_wheel(int dir)
677 {
678         if(debug_gui) {
679                 debug_gui_wheel(dir);
680         }
681 }
682
683 void app_gamepad_axis(int axis, float val)
684 {
685         switch(axis) {
686         case 0:
687                 joy_move.x = val;
688                 break;
689         case 1:
690                 joy_move.y = val;
691                 break;
692
693         case 2:
694                 joy_look.x = val;
695                 break;
696         case 3:
697                 joy_look.y = val;
698                 break;
699         }
700 }
701
702 void app_gamepad_button(int bn, bool pressed)
703 {
704         gpad_bnstate[bn] = pressed;
705
706         if(pressed) {
707                 switch(bn) {
708                 case GPAD_LSTICK:
709                         toggle_flight();
710                         break;
711
712                 case GPAD_X:
713                         show_blobs = !show_blobs;
714                         show_message("blobs: %s\n", show_blobs ? "on" : "off");
715                         break;
716
717                 case GPAD_START:
718                         goatvr_recenter();
719                         show_message("VR recenter\n");
720                         break;
721
722                 default:
723                         break;
724                 }
725         }
726 }
727
728 static void toggle_flight()
729 {
730         static float prev_walk_speed = -1.0;
731         if(prev_walk_speed < 0.0) {
732                 noclip = true;
733                 prev_walk_speed = walk_speed;
734                 walk_speed = 1000.0;
735                 show_message("fly mode\n");
736         } else {
737                 noclip = false;
738                 walk_speed = prev_walk_speed;
739                 prev_walk_speed = -1.0;
740                 show_message("walk mode\n");
741         }
742 }
743
744 static void calc_framerate()
745 {
746         //static int ncalc;
747         static int nframes;
748         static long prev_upd;
749
750         long elapsed = time_msec - prev_upd;
751         if(elapsed >= 1000) {
752                 framerate = (float)nframes / (float)(elapsed * 0.001);
753                 nframes = 1;
754                 prev_upd = time_msec;
755
756                 /*if(++ncalc >= 5) {
757                         printf("fps: %f\n", framerate);
758                         ncalc = 0;
759                 }*/
760         } else {
761                 ++nframes;
762         }
763 }
764
765 static Ray calc_pick_ray(int x, int y)
766 {
767         float nx = (float)x / (float)win_width;
768         float ny = (float)(win_height - y) / (float)win_height;
769
770         last_pick_ray = mouse_pick_ray(nx, ny, view_matrix, proj_matrix);
771         return last_pick_ray;
772 }