]> Pileus Git - grits/blob - src/roam.c
Miscellaneous updates
[grits] / src / roam.c
1 /*
2  * Copyright (C) 2009 Andy Spencer <spenceal@rose-hulman.edu>
3  *
4  * This program is free software: you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation, either version 3 of the License, or
7  * (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
16  */
17
18 #include <glib.h>
19 #include <math.h>
20 #include <string.h>
21 #include "gpqueue.h"
22 #include <GL/gl.h>
23 #include <GL/glu.h>
24
25 #include "gis-world.h"
26
27 #include "roam.h"
28
29
30 /**
31  * TODO:
32  *   - Optimize for memory consumption
33  *   - Profile for computation speed
34  *   - Target polygon count/detail
35  */
36
37 /* Misc */
38 RoamView *roam_view_new()
39 {
40         return g_new0(RoamView, 1);
41 }
42 void roam_view_update(RoamView *view)
43 {
44         glGetDoublev (GL_MODELVIEW_MATRIX,  view->model);
45         glGetDoublev (GL_PROJECTION_MATRIX, view->proj);
46         glGetIntegerv(GL_VIEWPORT,          view->view);
47         view->version++;
48 }
49
50 /* For GPQueue comparators */
51 static gint tri_cmp(RoamTriangle *a, RoamTriangle *b, gpointer data)
52 {
53         if      (a->error < b->error) return  1;
54         else if (a->error > b->error) return -1;
55         else                          return  0;
56 }
57 static gint dia_cmp(RoamDiamond *a, RoamDiamond *b, gpointer data)
58 {
59         if      (a->error < b->error) return -1;
60         else if (a->error > b->error) return  1;
61         else                          return  0;
62 }
63
64 /*************
65  * RoamPoint *
66  *************/
67 RoamPoint *roam_point_new(gdouble x, gdouble y, gdouble z)
68 {
69         RoamPoint *self = g_new0(RoamPoint, 1);
70         self->x = x;
71         self->y = y;
72         self->z = z;
73         return self;
74 }
75 RoamPoint *roam_point_dup(RoamPoint *self)
76 {
77         RoamPoint *new = g_memdup(self, sizeof(RoamPoint));
78         new->tris = 0;
79         return new;
80 }
81 void roam_point_add_triangle(RoamPoint *self, RoamTriangle *triangle)
82 {
83         for (int i = 0; i < 3; i++) {
84                 self->norm[i] *= self->tris;
85                 self->norm[i] += triangle->norm[i];
86         }
87         self->tris++;
88         for (int i = 0; i < 3; i++)
89                 self->norm[i] /= self->tris;
90 }
91 void roam_point_remove_triangle(RoamPoint *self, RoamTriangle *triangle)
92 {
93         for (int i = 0; i < 3; i++) {
94                 self->norm[i] *= self->tris;
95                 self->norm[i] -= triangle->norm[i];
96         }
97         self->tris--;
98         if (self->tris)
99                 for (int i = 0; i < 3; i++)
100                         self->norm[i] /= self->tris;
101 }
102 void roam_point_update_height(RoamPoint *self, RoamSphere *sphere)
103 {
104         if (sphere->height_func) {
105                 sphere->height_func(self, sphere->user_data);
106         } else {
107                 gdouble dist = sqrt(self->x * self->x +
108                                     self->y * self->y +
109                                     self->z * self->z);
110                 self->x = self->x/dist * EARTH_R;
111                 self->y = self->y/dist * EARTH_R;
112                 self->z = self->z/dist * EARTH_R;
113         }
114 }
115 void roam_point_update_projection(RoamPoint *self, RoamSphere *sphere)
116 {
117         static int count   = 0;
118         static int version = 0;
119         if (version != sphere->view->version) {
120                 g_debug("Projected %d points", count);
121                 count   = 0;
122                 version = sphere->view->version;
123         }
124
125         if (self->pversion != sphere->view->version) {
126                 /* Cache projection */
127                 gluProject(self->x, self->y, self->z,
128                         sphere->view->model, sphere->view->proj, sphere->view->view,
129                         &self->px, &self->py, &self->pz);
130                 self->pversion = sphere->view->version;
131                 count++;
132         }
133 }
134
135 /****************
136  * RoamTriangle *
137  ****************/
138 RoamTriangle *roam_triangle_new(RoamPoint *l, RoamPoint *m, RoamPoint *r)
139 {
140         RoamTriangle *self = g_new0(RoamTriangle, 1);
141
142         self->p.l = l;
143         self->p.m = m;
144         self->p.r = r;
145         self->split = roam_point_new(
146                 (l->x + r->x)/2,
147                 (l->y + r->y)/2,
148                 (l->z + r->z)/2);
149
150         self->error = 0;
151
152         /* Update normal */
153         double pa[3];
154         double pb[3];
155         pa[0] = self->p.l->x - self->p.m->x;
156         pa[1] = self->p.l->y - self->p.m->y;
157         pa[2] = self->p.l->z - self->p.m->z;
158
159         pb[0] = self->p.r->x - self->p.m->x;
160         pb[1] = self->p.r->y - self->p.m->y;
161         pb[2] = self->p.r->z - self->p.m->z;
162
163         self->norm[0] = pa[1] * pb[2] - pa[2] * pb[1];
164         self->norm[1] = pa[2] * pb[0] - pa[0] * pb[2];
165         self->norm[2] = pa[0] * pb[1] - pa[1] * pb[0];
166
167         double total = sqrt(self->norm[0] * self->norm[0] +
168                             self->norm[1] * self->norm[1] +
169                             self->norm[2] * self->norm[2]);
170
171         self->norm[0] /= total;
172         self->norm[1] /= total;
173         self->norm[2] /= total;
174
175         //g_message("roam_triangle_new: %p", self);
176         return self;
177 }
178
179 void roam_triangle_free(RoamTriangle *self)
180 {
181         g_free(self->split);
182         g_free(self);
183 }
184
185 void roam_triangle_add(RoamTriangle *self,
186                 RoamTriangle *left, RoamTriangle *base, RoamTriangle *right,
187                 RoamSphere *sphere)
188 {
189         self->t.l = left;
190         self->t.b = base;
191         self->t.r = right;
192
193         roam_point_add_triangle(self->p.l, self);
194         roam_point_add_triangle(self->p.m, self);
195         roam_point_add_triangle(self->p.r, self);
196
197         if (sphere->view)
198                 roam_triangle_update_errors(self, sphere);
199
200         self->handle = g_pqueue_push(sphere->triangles, self);
201 }
202
203 void roam_triangle_remove(RoamTriangle *self, RoamSphere *sphere)
204 {
205         /* Update vertex normals */
206         roam_point_remove_triangle(self->p.l, self);
207         roam_point_remove_triangle(self->p.m, self);
208         roam_point_remove_triangle(self->p.r, self);
209
210         g_pqueue_remove(sphere->triangles, self->handle);
211 }
212
213 void roam_triangle_sync_neighbors(RoamTriangle *new, RoamTriangle *old, RoamTriangle *neigh)
214 {
215         if      (neigh->t.l == old) neigh->t.l = new;
216         else if (neigh->t.b == old) neigh->t.b = new;
217         else if (neigh->t.r == old) neigh->t.r = new;
218         else g_assert_not_reached();
219 }
220
221 gboolean roam_point_visible(RoamPoint *self, RoamSphere *sphere)
222 {
223         gint *view = sphere->view->view;
224         return self->px > view[0] && self->px < view[2] &&
225                self->py > view[1] && self->py < view[3] &&
226                self->pz > 0       && self->pz < 1;
227 }
228 gboolean roam_triangle_visible(RoamTriangle *self, RoamSphere *sphere)
229 {
230         /* Do this with a bounding box */
231         return roam_point_visible(self->p.l, sphere) ||
232                roam_point_visible(self->p.m, sphere) ||
233                roam_point_visible(self->p.r, sphere);
234 }
235
236 void roam_triangle_update_errors(RoamTriangle *self, RoamSphere *sphere)
237 {
238         /* Update points */
239         roam_point_update_projection(self->p.l, sphere);
240         roam_point_update_projection(self->p.m, sphere);
241         roam_point_update_projection(self->p.r, sphere);
242         roam_point_update_projection(self->split, sphere);
243
244         /* Not exactly correct, could be out on both sides (middle in) */
245         if (!roam_triangle_visible(self, sphere)) {
246                 self->error = -1;
247         } else {
248                 RoamPoint *l = self->p.l;
249                 RoamPoint *m = self->p.m;
250                 RoamPoint *r = self->p.r;
251                 RoamPoint *split = self->split;
252
253                 /*               l-r midpoint        projected l-r midpoint */
254                 gdouble pxdist = (l->px + r->px)/2 - split->px;
255                 gdouble pydist = (l->py + r->py)/2 - split->py;
256
257                 self->error = sqrt(pxdist*pxdist + pydist*pydist);
258
259                 /* Multiply by size of triangle */
260                 double size = -( l->px * (m->py - r->py) +
261                                  m->px * (r->py - l->py) +
262                                  r->px * (l->py - m->py) ) / 2.0;
263
264                 /* Size < 0 == backface */
265                 self->error *= size;
266         }
267 }
268
269 void roam_triangle_split(RoamTriangle *self, RoamSphere *sphere)
270 {
271         //g_message("roam_triangle_split: %p, e=%f\n", self, self->error);
272
273         sphere->polys += 2;
274
275         if (self != self->t.b->t.b)
276                 roam_triangle_split(self->t.b, sphere);
277
278         RoamTriangle *base = self->t.b;
279
280         /* Add new triangles */
281         RoamPoint *mid = self->split;
282         RoamTriangle *sl = roam_triangle_new(self->p.m, mid, self->p.l); // Self Left
283         RoamTriangle *sr = roam_triangle_new(self->p.r, mid, self->p.m); // Self Right
284         RoamTriangle *bl = roam_triangle_new(base->p.m, mid, base->p.l); // Base Left
285         RoamTriangle *br = roam_triangle_new(base->p.r, mid, base->p.m); // Base Right
286
287         roam_point_update_height(sl->split, sphere);
288         roam_point_update_height(sr->split, sphere);
289         roam_point_update_height(bl->split, sphere);
290         roam_point_update_height(br->split, sphere);
291
292         /*                tri,l,  base,      r,  sphere */
293         roam_triangle_add(sl, sr, self->t.l, br, sphere);
294         roam_triangle_add(sr, bl, self->t.r, sl, sphere);
295         roam_triangle_add(bl, br, base->t.l, sr, sphere);
296         roam_triangle_add(br, sl, base->t.r, bl, sphere);
297
298         roam_triangle_sync_neighbors(sl, self, self->t.l);
299         roam_triangle_sync_neighbors(sr, self, self->t.r);
300         roam_triangle_sync_neighbors(bl, base, base->t.l);
301         roam_triangle_sync_neighbors(br, base, base->t.r);
302
303         /* Remove old triangles */
304         roam_triangle_remove(self, sphere);
305         roam_triangle_remove(base, sphere);
306
307         /* Add/Remove diamonds */
308         RoamDiamond *diamond = roam_diamond_new(self, base, sl, sr, bl, br);
309         roam_diamond_update_errors(diamond, sphere);
310         roam_diamond_add(diamond, sphere);
311         roam_diamond_remove(self->parent, sphere);
312         roam_diamond_remove(base->parent, sphere);
313 }
314
315 void roam_triangle_draw(RoamTriangle *self)
316 {
317         glBegin(GL_TRIANGLES);
318         glNormal3dv(self->p.r->norm); glVertex3dv((double*)self->p.r);
319         glNormal3dv(self->p.m->norm); glVertex3dv((double*)self->p.m);
320         glNormal3dv(self->p.l->norm); glVertex3dv((double*)self->p.l);
321         glEnd();
322         return;
323 }
324
325 void roam_triangle_draw_normal(RoamTriangle *self)
326 {
327         double center[] = {
328                 (self->p.l->x + self->p.m->x + self->p.r->x)/3.0,
329                 (self->p.l->y + self->p.m->y + self->p.r->y)/3.0,
330                 (self->p.l->z + self->p.m->z + self->p.r->z)/3.0,
331         };
332         double end[] = {
333                 center[0]+self->norm[0]*2000000,
334                 center[1]+self->norm[1]*2000000,
335                 center[2]+self->norm[2]*2000000,
336         };
337         glBegin(GL_LINES);
338         glVertex3dv(center);
339         glVertex3dv(end);
340         glEnd();
341 }
342
343 /***************
344  * RoamDiamond *
345  ***************/
346 RoamDiamond *roam_diamond_new(
347                 RoamTriangle *parent0, RoamTriangle *parent1,
348                 RoamTriangle *kid0, RoamTriangle *kid1,
349                 RoamTriangle *kid2, RoamTriangle *kid3)
350 {
351         RoamDiamond *self = g_new0(RoamDiamond, 1);
352
353         self->kids[0] = kid0;
354         self->kids[1] = kid1;
355         self->kids[2] = kid2;
356         self->kids[3] = kid3;
357
358         kid0->parent = self;
359         kid1->parent = self;
360         kid2->parent = self;
361         kid3->parent = self;
362
363         self->parents[0] = parent0;
364         self->parents[1] = parent1;
365
366         return self;
367 }
368 void roam_diamond_add(RoamDiamond *self, RoamSphere *sphere)
369 {
370         self->active = TRUE;
371         self->error  = MAX(self->parents[0]->error, self->parents[1]->error);
372         self->handle = g_pqueue_push(sphere->diamonds, self);
373 }
374 void roam_diamond_remove(RoamDiamond *self, RoamSphere *sphere)
375 {
376         if (self && self->active) {
377                 self->active = FALSE;
378                 g_pqueue_remove(sphere->diamonds, self->handle);
379         }
380 }
381 void roam_diamond_merge(RoamDiamond *self, RoamSphere *sphere)
382 {
383         //g_message("roam_diamond_merge: %p, e=%f\n", self, self->error);
384
385         sphere->polys -= 2;
386
387         /* TODO: pick the best split */
388         RoamTriangle **kids = self->kids;
389
390         /* Create triangles */
391         RoamTriangle *tri  = self->parents[0];
392         RoamTriangle *base = self->parents[1];
393
394         roam_triangle_add(tri,  kids[0]->t.b, base, kids[1]->t.b, sphere);
395         roam_triangle_add(base, kids[2]->t.b, tri,  kids[3]->t.b, sphere);
396
397         roam_triangle_sync_neighbors(tri,  kids[0], kids[0]->t.b);
398         roam_triangle_sync_neighbors(tri,  kids[1], kids[1]->t.b);
399         roam_triangle_sync_neighbors(base, kids[2], kids[2]->t.b);
400         roam_triangle_sync_neighbors(base, kids[3], kids[3]->t.b);
401
402         /* Remove triangles */
403         roam_triangle_remove(kids[0], sphere);
404         roam_triangle_remove(kids[1], sphere);
405         roam_triangle_remove(kids[2], sphere);
406         roam_triangle_remove(kids[3], sphere);
407
408         /* Add/Remove triangles */
409         if (tri->t.l->t.l == tri->t.r->t.r &&
410             tri->t.l->t.l != tri && tri->parent) {
411                 roam_diamond_update_errors(tri->parent, sphere);
412                 roam_diamond_add(tri->parent, sphere);
413         }
414
415         if (base->t.l->t.l == base->t.r->t.r &&
416             base->t.l->t.l != base && base->parent) {
417                 roam_diamond_update_errors(base->parent, sphere);
418                 roam_diamond_add(base->parent, sphere);
419         }
420
421         /* Remove and free diamond and child triangles */
422         roam_diamond_remove(self, sphere);
423         g_assert(self->kids[0]->p.m == self->kids[1]->p.m &&
424                  self->kids[1]->p.m == self->kids[2]->p.m &&
425                  self->kids[2]->p.m == self->kids[3]->p.m);
426         g_assert(self->kids[0]->p.m->tris == 0);
427         roam_triangle_free(self->kids[0]);
428         roam_triangle_free(self->kids[1]);
429         roam_triangle_free(self->kids[2]);
430         roam_triangle_free(self->kids[3]);
431         g_free(self);
432 }
433 void roam_diamond_update_errors(RoamDiamond *self, RoamSphere *sphere)
434 {
435         roam_triangle_update_errors(self->parents[0], sphere);
436         roam_triangle_update_errors(self->parents[1], sphere);
437         self->error = MAX(self->parents[0]->error, self->parents[1]->error);
438 }
439
440 /**************
441  * RoamSphere *
442  **************/
443 RoamSphere *roam_sphere_new(gpointer user_data)
444 {
445         RoamSphere *self = g_new0(RoamSphere, 1);
446         self->user_data   = user_data;
447         self->polys       = 12;
448         self->triangles   = g_pqueue_new((GCompareDataFunc)tri_cmp, NULL);
449         self->diamonds    = g_pqueue_new((GCompareDataFunc)dia_cmp, NULL);
450
451         RoamPoint *vertexes[] = {
452                 roam_point_new( 1, 1, 1), // 0
453                 roam_point_new( 1, 1,-1), // 1
454                 roam_point_new( 1,-1, 1), // 2
455                 roam_point_new( 1,-1,-1), // 3
456                 roam_point_new(-1, 1, 1), // 4
457                 roam_point_new(-1, 1,-1), // 5
458                 roam_point_new(-1,-1, 1), // 6
459                 roam_point_new(-1,-1,-1), // 7
460         };
461         int _triangles[][2][3] = {
462                 /*lv mv rv   ln, bn, rn */
463                 {{3,2,0}, {10, 1, 7}}, // 0
464                 {{0,1,3}, { 9, 0, 2}}, // 1
465                 {{7,3,1}, {11, 3, 1}}, // 2
466                 {{1,5,7}, { 8, 2, 4}}, // 3
467                 {{6,7,5}, {11, 5, 3}}, // 4
468                 {{5,4,6}, { 8, 4, 6}}, // 5
469                 {{2,6,4}, {10, 7, 5}}, // 6
470                 {{4,0,2}, { 9, 6, 0}}, // 7
471                 {{4,5,1}, { 5, 9, 3}}, // 8
472                 {{1,0,4}, { 1, 8, 7}}, // 9
473                 {{6,2,3}, { 6,11, 0}}, // 10
474                 {{3,7,6}, { 2,10, 4}}, // 11
475         };
476         RoamTriangle *triangles[12];
477
478         for (int i = 0; i < 8; i++)
479                 roam_point_update_height(vertexes[i], self);
480         for (int i = 0; i < 12; i++)
481                 triangles[i] = roam_triangle_new(
482                         vertexes[_triangles[i][0][0]],
483                         vertexes[_triangles[i][0][1]],
484                         vertexes[_triangles[i][0][2]]);
485         for (int i = 0; i < 12; i++)
486                 roam_point_update_height(triangles[i]->split, self);
487         for (int i = 0; i < 12; i++)
488                 roam_triangle_add(triangles[i],
489                         triangles[_triangles[i][1][0]],
490                         triangles[_triangles[i][1][1]],
491                         triangles[_triangles[i][1][2]],
492                         self);
493
494         return self;
495 }
496 void roam_sphere_update_errors(RoamSphere *self)
497 {
498         g_debug("RoamSphere: update_errors - polys=%d", self->polys);
499         if (!self->view)
500                 self->view = roam_view_new();
501         roam_view_update(self->view);
502
503         GPtrArray *tris = g_pqueue_get_array(self->triangles);
504         GPtrArray *dias = g_pqueue_get_array(self->diamonds);
505
506         for (int i = 0; i < tris->len; i++) {
507                 RoamTriangle *tri = tris->pdata[i];
508                 roam_triangle_update_errors(tri, self);
509                 g_pqueue_priority_changed(self->triangles, tri->handle);
510         }
511
512         for (int i = 0; i < dias->len; i++) {
513                 RoamDiamond *dia = dias->pdata[i];
514                 roam_diamond_update_errors(dia, self);
515                 g_pqueue_priority_changed(self->diamonds, dia->handle);
516         }
517
518         g_ptr_array_free(tris, TRUE);
519         g_ptr_array_free(dias, TRUE);
520 }
521
522 void roam_sphere_split_one(RoamSphere *self)
523 {
524         RoamTriangle *to_split = g_pqueue_peek(self->triangles);
525         if (!to_split) return;
526         roam_triangle_split(to_split, self);
527 }
528 void roam_sphere_merge_one(RoamSphere *self)
529 {
530         RoamDiamond *to_merge = g_pqueue_peek(self->diamonds);
531         if (!to_merge) return;
532         roam_diamond_merge(to_merge, self);
533 }
534 gint roam_sphere_split_merge(RoamSphere *self)
535 {
536         gint iters = 0, max_iters = 500;
537         //gint target = 4000;
538         gint target = 2000;
539         //gint target = 500;
540
541         if (!self->view)
542                 return 0;
543
544         if (target - self->polys > 100) {
545                 //g_debug("RoamSphere: split_merge - Splitting %d - %d > 100", target, self->polys);
546                 while (self->polys < target && iters++ < max_iters)
547                         roam_sphere_split_one(self);
548         }
549
550         if (self->polys - target > 100) {
551                 //g_debug("RoamSphere: split_merge - Merging %d - %d > 100", self->polys, target);
552                 while (self->polys > target && iters++ < max_iters)
553                         roam_sphere_merge_one(self);
554         }
555
556         while (((RoamTriangle*)g_pqueue_peek(self->triangles))->error >
557                ((RoamDiamond *)g_pqueue_peek(self->diamonds ))->error &&
558                iters++ < max_iters) {
559                 //g_debug("RoamSphere: split_merge - Fixing 1 %f > %f && %d < %d",
560                 //              ((RoamTriangle*)g_pqueue_peek(self->triangles))->error,
561                 //              ((RoamDiamond *)g_pqueue_peek(self->diamonds ))->error,
562                 //              iters-1, max_iters);
563                 roam_sphere_merge_one(self);
564                 roam_sphere_split_one(self);
565         }
566
567         return iters;
568 }
569 void roam_sphere_draw(RoamSphere *self)
570 {
571         g_debug("RoamSphere: draw");
572         g_pqueue_foreach(self->triangles, (GFunc)roam_triangle_draw, NULL);
573 }
574 void roam_sphere_draw_normals(RoamSphere *self)
575 {
576         g_debug("RoamSphere: draw_normal");
577         g_pqueue_foreach(self->triangles, (GFunc)roam_triangle_draw_normal, NULL);
578 }
579 void roam_sphere_free_tri(RoamTriangle *tri)
580 {
581         if (--tri->p.l->tris == 0) g_free(tri->p.l);
582         if (--tri->p.m->tris == 0) g_free(tri->p.m);
583         if (--tri->p.r->tris == 0) g_free(tri->p.r);
584         roam_triangle_free(tri);
585 }
586 void roam_sphere_free(RoamSphere *self)
587 {
588         /* Slow method, but it should work */
589         while (self->polys > 12)
590                 roam_sphere_merge_one(self);
591         /* TODO: free points */
592         g_pqueue_foreach(self->triangles, (GFunc)roam_sphere_free_tri, NULL);
593         g_pqueue_free(self->triangles);
594         g_pqueue_free(self->diamonds);
595         g_free(self->view);
596         g_free(self);
597 }