2 RetroRay - integrated standalone vintage modeller/renderer
3 Copyright (C) 2023 John Tsiombikas <nuclear@mutantstargoat.com>
5 This program is free software: you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation, either version 3 of the License, or
8 (at your option) any later version.
10 This program is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 GNU General Public License for more details.
15 You should have received a copy of the GNU General Public License
16 along with this program. If not, see <https://www.gnu.org/licenses/>.
28 TBN_NEW, TBN_OPEN, TBN_SAVE, TBN_SEP1,
29 TBN_SEL, TBN_MOVE, TBN_ROT, TBN_SCALE, TBN_SEP2,
30 TBN_ADD, TBN_RM, TBN_SEP3,
31 TBN_UNION, TBN_ISECT, TBN_DIFF, TBN_SEP4,
32 TBN_MTL, TBN_REND, TBN_VIEWREND, TBN_SEP5, TBN_CFG,
36 static const char *tbn_icon_name[] = {
37 "new", "open", "save", 0,
38 "sel", "move", "rot", "scale", 0,
40 "union", "isect", "diff", 0,
41 "mtl", "rend", "viewrend", 0, "cfg"
43 static int tbn_icon_pos[][2] = {
44 {0,0}, {16,0}, {32,0}, {-1,-1},
45 {48,0}, {64,0}, {80,0}, {96,0}, {-1,-1},
46 {112,0}, {112,16}, {-1,-1},
47 {0,16}, {16,16}, {32,16}, {-1,-1},
48 {48,16}, {64,16}, {80,16}, {-1,-1}, {96,16}
50 static int tbn_istool[] = {
57 static rtk_icon *tbn_icons[NUM_TOOL_BUTTONS];
58 static rtk_widget *tbn_buttons[NUM_TOOL_BUTTONS];
60 #define TOOLBAR_HEIGHT 26
63 TOOL_SEL, TOOL_MOVE, TOOL_ROT, TOOL_SCALE,
64 TOOL_UNION, TOOL_ISECT, TOOL_DIFF,
67 static rtk_widget *tools[NUM_TOOLS];
70 static int mdl_init(void);
71 static void mdl_destroy(void);
72 static int mdl_start(void);
73 static void mdl_stop(void);
74 static void mdl_display(void);
75 static void mdl_reshape(int x, int y);
76 static void mdl_keyb(int key, int press);
77 static void mdl_mouse(int bn, int press, int x, int y);
78 static void mdl_motion(int x, int y);
80 static void draw_object(struct object *obj);
81 static void draw_grid(void);
82 static void tbn_callback(rtk_widget *w, void *cls);
84 static void draw_rband(void);
85 static void primray(cgm_ray *ray, int x, int y);
88 struct app_screen scr_model = {
90 mdl_init, mdl_destroy,
92 mdl_display, mdl_reshape,
93 mdl_keyb, mdl_mouse, mdl_motion
96 static rtk_widget *toolbar;
97 static rtk_iconsheet *icons;
99 static struct cmesh *mesh_sph;
101 static float cam_theta, cam_phi = 20, cam_dist = 8;
102 static float view_matrix[16], proj_matrix[16];
103 static float view_matrix_inv[16], proj_matrix_inv[16];
104 static int viewport[4];
105 static cgm_ray pickray;
108 static int selobj = -1;
110 static rtk_rect rband;
111 static int rband_valid;
113 static int mdl_init(void)
118 if(!(icons = rtk_load_iconsheet("data/icons.png"))) {
119 errormsg("failed to load iconsheet\n");
122 for(i=0; i<NUM_TOOL_BUTTONS; i++) {
123 if(tbn_icon_name[i]) {
124 tbn_icons[i] = rtk_define_icon(icons, tbn_icon_name[i],
125 tbn_icon_pos[i][0], tbn_icon_pos[i][1], 16, 16);
131 if(!(toolbar = rtk_create_window(0, "toolbar", 0, 0, win_width, TOOLBAR_HEIGHT))) {
134 rtk_win_layout(toolbar, RTK_HBOX);
137 for(i=0; i<NUM_TOOL_BUTTONS; i++) {
139 rtk_create_separator(toolbar);
141 if(!(w = rtk_create_iconbutton(toolbar, tbn_icons[i], 0))) {
145 rtk_set_callback(w, tbn_callback, (void*)(intptr_t)i);
147 rtk_bn_mode(w, RTK_TOGGLEBN);
148 tools[toolidx++] = w;
155 assert(toolidx == NUM_TOOLS);
157 if(!(mesh_sph = cmesh_alloc())) {
158 errormsg("failed to allocate sphere vis mesh\n");
161 gen_sphere(mesh_sph, 1.0f, 16, 8, 1.0f, 1.0f);
165 static void mdl_destroy(void)
167 cmesh_free(mesh_sph);
168 rtk_free_iconsheet(icons);
171 static int mdl_start(void)
173 gaw_clear_color(0.125, 0.125, 0.125, 1);
175 gaw_enable(GAW_DEPTH_TEST);
176 gaw_enable(GAW_CULL_FACE);
177 gaw_enable(GAW_LIGHTING);
178 gaw_enable(GAW_LIGHT0);
182 static void mdl_stop(void)
186 static void mdl_display(void)
190 gaw_clear(GAW_COLORBUF | GAW_DEPTHBUF);
192 rtk_draw_widget(toolbar);
194 gaw_matrix_mode(GAW_MODELVIEW);
196 gaw_translate(0, 0, -cam_dist);
197 gaw_rotate(cam_phi, 1, 0, 0);
198 gaw_rotate(cam_theta, 0, 1, 0);
199 gaw_get_modelview(view_matrix);
200 cgm_mcopy(view_matrix_inv, view_matrix);
201 cgm_minverse(view_matrix_inv);
205 gaw_mtl_diffuse(0.5, 0.5, 0.5, 1);
207 num = scn_num_objects(scn);
208 for(i=0; i<num; i++) {
211 gaw_enable(GAW_POLYGON_OFFSET);
212 draw_object(scn->objects[i]);
213 gaw_disable(GAW_POLYGON_OFFSET);
216 gaw_disable(GAW_LIGHTING);
218 gaw_color3f(0, 1, 0);
219 draw_object(scn->objects[i]);
223 draw_object(scn->objects[i]);
232 static void draw_object(struct object *obj)
236 calc_object_matrix(obj);
238 gaw_mult_matrix(obj->xform);
242 sph = (struct sphere*)obj;
243 gaw_scale(sph->rad, sph->rad, sph->rad);
244 cmesh_draw(mesh_sph);
254 static void draw_grid(void)
257 gaw_disable(GAW_LIGHTING);
259 gaw_begin(GAW_LINES);
260 gaw_color3f(0.5, 0, 0);
261 gaw_vertex3f(0, 0, 0);
262 gaw_vertex3f(-100, 0, 0);
263 gaw_vertex3f(0, 0, 0);
264 gaw_vertex3f(100, 0, 0);
265 gaw_color3f(0, 0.5, 0);
266 gaw_vertex3f(0, 0, 0);
267 gaw_vertex3f(0, 0, -100);
268 gaw_vertex3f(0, 0, 0);
269 gaw_vertex3f(0, 0, 100);
275 static void mdl_reshape(int x, int y)
277 float aspect = (float)x / (float)(y - TOOLBAR_HEIGHT);
280 viewport[1] = TOOLBAR_HEIGHT;
282 viewport[3] = y - TOOLBAR_HEIGHT;
283 gaw_viewport(viewport[0], viewport[1], viewport[2], viewport[3]);
285 gaw_matrix_mode(GAW_PROJECTION);
287 gaw_perspective(50, aspect, 0.5, 100.0);
288 gaw_get_projection(proj_matrix);
289 cgm_mcopy(proj_matrix_inv, proj_matrix);
290 cgm_minverse(proj_matrix_inv);
292 rtk_resize(toolbar, win_width, TOOLBAR_HEIGHT);
295 static void mdl_keyb(int key, int press)
297 if(rtk_input_key(toolbar, key, press)) {
305 static void mdl_mouse(int bn, int press, int x, int y)
308 if(!vpdrag && rtk_input_mbutton(toolbar, bn, press, x, y)) {
318 vpdrag &= ~(1 << bn);
321 printf("rubber band: %d,%d %dx%d\n", rband.x, rband.y, rband.width, rband.height);
324 } else if(bn == 0 && x == rband.x && y == rband.y) {
325 primray(&pickray, x, y);
326 if(scn_intersect(scn, &pickray, &hit)) {
327 selobj = scn_object_index(scn, hit.obj);
336 static void mdl_motion(int x, int y)
340 if(!vpdrag && rtk_input_mmotion(toolbar, x, y)) {
351 cam_theta += dx * 0.5f;
352 cam_phi += dy * 0.5f;
353 if(cam_phi < -90) cam_phi = -90;
354 if(cam_phi > 90) cam_phi = 90;
359 cam_dist += dy * 0.1f;
360 if(cam_dist < 0) cam_dist = 0;
365 if(rband.x != x || rband.y != y) {
366 rband.width = x - rband.x;
367 rband.height = y - rband.y;
375 static void add_sphere(void)
379 if(!(obj = create_object(OBJ_SPHERE))) {
382 scn_add_object(scn, obj);
385 static void tbn_callback(rtk_widget *w, void *cls)
387 int i, id = (intptr_t)cls;
400 idx = id - TBN_UNION + TOOL_UNION;
403 for(i=0; i<NUM_TOOLS; i++) {
405 rtk_set_value(tools[i], 0);
411 idx = scn_num_objects(scn);
418 scn_rm_object(scn, selobj);
429 static void draw_rband(void)
432 uint32_t *fbptr, *bptr;
437 if(rband.width < 0) {
443 if(rband.height < 0) {
450 fbptr = framebuf + y * win_width + x;
451 bptr = fbptr + win_width * (h - 1);
454 fbptr[i] ^= 0xffffff;
458 for(i=0; i<h-2; i++) {
459 fbptr[0] ^= 0xffffff;
460 fbptr[w - 1] ^= 0xffffff;
465 static void primray(cgm_ray *ray, int x, int y)
468 cgm_vec3 npos, farpt;
472 nx = (float)(x - viewport[0]) / (float)viewport[2];
473 ny = (float)(y - viewport[1]) / (float)viewport[3];
475 cgm_mcopy(inv_pv, proj_matrix_inv);
476 cgm_mmul(inv_pv, view_matrix_inv);
478 cgm_vcons(&npos, nx, ny, 0.0f);
479 cgm_unproject(&ray->origin, &npos, inv_pv);
481 cgm_unproject(&farpt, &npos, inv_pv);
483 ray->dir.x = farpt.x - ray->origin.x;
484 ray->dir.y = farpt.y - ray->origin.y;
485 ray->dir.z = farpt.z - ray->origin.z;