70ea449f9b8001017e07d4feb55bb24c4555ef1a
[dosdemo] / src / dos / main.c
1 #include <stdio.h>
2 #include <stdlib.h>
3 #include <math.h>
4 #include <string.h>
5 #include <limits.h>
6 #include <assert.h>
7 #include <conio.h>
8 #include "demo.h"
9 #include "keyb.h"
10 #include "mouse.h"
11 #include "timer.h"
12 #include "gfx.h"
13 #include "vmath.h"
14 #include "sball.h"
15 #include "cfgopt.h"
16 #include "logger.h"
17 #include "tinyfps.h"
18 #include "cdpmi.h"
19
20 #undef NOKEYB
21
22 static int handle_sball_event(sball_event *ev);
23 static void recalc_sball_matrix(float *xform);
24
25 static int quit;
26 static int use_mouse;
27 static long fbsize;
28
29 static int use_sball;
30 static vec3_t pos = {0, 0, 0};
31 static quat_t rot = {0, 0, 0, 1};
32
33 int main(int argc, char **argv)
34 {
35 #ifdef __DJGPP__
36         __djgpp_nearptr_enable();
37 #endif
38
39         fbsize = fb_width * fb_height * fb_bpp / 8;
40
41         init_logger("demo.log");
42
43         init_timer(100);
44 #ifndef NOKEYB
45         kb_init(32);
46 #endif
47
48         if((use_mouse = have_mouse())) {
49                 printf("initializing mouse input\n");
50                 set_mouse_limits(0, 0, fb_width, fb_height);
51                 set_mouse(fb_width / 2, fb_height / 2);
52         }
53
54         /* allocate a couple extra rows as a guard band, until we fucking fix the rasterizer */
55         if(!(fb_pixels = malloc(fbsize + (fb_width * fb_bpp / 8) * 2))) {
56                 fprintf(stderr, "failed to allocate backbuffer\n");
57                 return 1;
58         }
59         fb_pixels += fb_width;
60
61         if(!(vmem = set_video_mode(fb_width, fb_height, fb_bpp, 1))) {
62                 return 1;
63         }
64
65         if(demo_init(argc, argv) == -1) {
66                 set_text_mode();
67                 return 1;
68         }
69
70         if(opt.sball && sball_init() == 0) {
71                 use_sball = 1;
72         }
73
74         reset_timer();
75
76         while(!quit) {
77 #ifndef NOKEYB
78                 int key;
79                 while((key = kb_getkey()) != -1) {
80                         demo_keyboard(key, 1);
81                 }
82 #else
83                 if(kbhit()) {
84                         demo_keyboard(getch(), 1);
85                 }
86 #endif
87                 if(quit) goto break_evloop;
88
89                 if(use_mouse) {
90                         mouse_bmask = read_mouse(&mouse_x, &mouse_y);
91                 }
92                 if(use_sball && sball_pending()) {
93                         sball_event ev;
94                         while(sball_getevent(&ev)) {
95                                 handle_sball_event(&ev);
96                         }
97                         recalc_sball_matrix(sball_matrix);
98                 }
99
100                 time_msec = get_msec();
101                 demo_draw();
102         }
103
104 break_evloop:
105         set_text_mode();
106         demo_cleanup();
107 #ifndef NOKEYB
108         kb_shutdown();
109 #endif
110         if(use_sball) {
111                 sball_shutdown();
112         }
113         return 0;
114 }
115
116 void demo_quit(void)
117 {
118         quit = 1;
119 }
120
121 void swap_buffers(void *pixels)
122 {
123         if(!pixels) {
124                 pixels = fb_pixels;
125         }
126
127         /* just memcpy to the front buffer */
128         if(opt.vsync) {
129                 wait_vsync();
130         }
131         drawFps(pixels);
132         memcpy(vmem, pixels, fbsize);
133 }
134
135
136 #define TX(ev)  ((ev)->motion.motion[0])
137 #define TY(ev)  ((ev)->motion.motion[1])
138 #define TZ(ev)  ((ev)->motion.motion[2])
139 #define RX(ev)  ((ev)->motion.motion[3])
140 #define RY(ev)  ((ev)->motion.motion[4])
141 #define RZ(ev)  ((ev)->motion.motion[5])
142
143 static int handle_sball_event(sball_event *ev)
144 {
145         switch(ev->type) {
146         case SBALL_EV_MOTION:
147                 if(RX(ev) | RY(ev) | RZ(ev)) {
148                         float rx = (float)RX(ev);
149                         float ry = (float)RY(ev);
150                         float rz = (float)RZ(ev);
151                         float axis_len = sqrt(rx * rx + ry * ry + rz * rz);
152                         if(axis_len > 0.0) {
153                                 rot = quat_rotate(rot, axis_len * 0.001, -rx / axis_len,
154                                                 -ry / axis_len, -rz / axis_len);
155                         }
156                 }
157
158                 pos.x += TX(ev) * 0.001;
159                 pos.y += TY(ev) * 0.001;
160                 pos.z += TZ(ev) * 0.001;
161                 break;
162
163         case SBALL_EV_BUTTON:
164                 if(ev->button.pressed) {
165                         pos = v3_cons(0, 0, 0);
166                         rot = quat_cons(1, 0, 0, 0);
167                 }
168                 break;
169         }
170
171         return 0;
172 }
173
174 void recalc_sball_matrix(float *xform)
175 {
176         quat_to_mat(xform, rot);
177         xform[12] = pos.x;
178         xform[13] = pos.y;
179         xform[14] = pos.z;
180 }