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