| 14 |
|
#ifndef LFREEPCT |
| 15 |
|
#define LFREEPCT 25 |
| 16 |
|
#endif |
| 17 |
+ |
/* maximum allowed angle difference (deg.) */ |
| 18 |
+ |
#ifndef MAXANG |
| 19 |
+ |
#define MAXANG 20. |
| 20 |
+ |
#endif |
| 21 |
|
|
| 22 |
+ |
#define MAXDIFF2 (long)( MAXANG*MAXANG /90./90.*(1L<<15)*(1L<<15)) |
| 23 |
+ |
|
| 24 |
+ |
#define abs(i) ((i) < 0 ? -(i) : (i)) |
| 25 |
+ |
|
| 26 |
|
RTREE qtrunk; /* our quadtree trunk */ |
| 27 |
< |
double qtDepthEps = .02; /* epsilon to compare depths (z fraction) */ |
| 27 |
> |
double qtDepthEps = .05; /* epsilon to compare depths (z fraction) */ |
| 28 |
|
int qtMinNodesiz = 2; /* minimum node dimension (pixels) */ |
| 29 |
|
struct rleaves qtL; /* our pile of leaves */ |
| 30 |
|
|
| 33 |
|
static RTREE **twigbundle; /* free twig blocks (NULL term.) */ |
| 34 |
|
static int nexttwig; /* next free twig */ |
| 35 |
|
|
| 36 |
< |
#define is_stump(t) (!((t)->flgs & (BR_ANY|LF_ANY))) |
| 36 |
> |
#define ungetleaf(li) (qtL.tl=(li)) /* dangerous if used improperly */ |
| 37 |
|
|
| 38 |
|
|
| 39 |
|
static RTREE * |
| 70 |
|
{ |
| 71 |
|
register int i; |
| 72 |
|
|
| 73 |
< |
qtrunk.flgs = CH_ANY; |
| 66 |
< |
nexttwig = 0; |
| 73 |
> |
qtrunk.flgs = CH_ANY; /* chop down tree */ |
| 74 |
|
if (twigbundle == NULL) |
| 75 |
|
return; |
| 76 |
+ |
i = (TBUNDLESIZ-1+nexttwig)/TBUNDLESIZ; |
| 77 |
+ |
nexttwig = 0; |
| 78 |
|
if (!really) { /* just clear allocated blocks */ |
| 79 |
< |
for (i = 0; twigbundle[i] != NULL; i++) |
| 79 |
> |
while (i--) |
| 80 |
|
bzero((char *)twigbundle[i], TBUNDLESIZ*sizeof(RTREE)); |
| 81 |
|
return; |
| 82 |
|
} |
| 102 |
|
} |
| 103 |
|
|
| 104 |
|
|
| 105 |
< |
#define LEAFSIZ (3*sizeof(float)+sizeof(TMbright)+6*sizeof(BYTE)) |
| 105 |
> |
#define LEAFSIZ (3*sizeof(float)+2*sizeof(short)+\ |
| 106 |
> |
sizeof(TMbright)+6*sizeof(BYTE)) |
| 107 |
|
|
| 108 |
|
int |
| 109 |
|
qtAllocLeaves(n) /* allocate space for n leaves */ |
| 129 |
|
return(0); |
| 130 |
|
/* assign larger alignment types earlier */ |
| 131 |
|
qtL.wp = (float (*)[3])qtL.base; |
| 132 |
< |
qtL.brt = (TMbright *)(qtL.wp + n); |
| 132 |
> |
qtL.wd = (short (*)[2])(qtL.wp + n); |
| 133 |
> |
qtL.brt = (TMbright *)(qtL.wd + n); |
| 134 |
|
qtL.chr = (BYTE (*)[3])(qtL.brt + n); |
| 135 |
|
qtL.rgb = (BYTE (*)[3])(qtL.chr + n); |
| 136 |
|
qtL.nl = n; |
| 202 |
|
} |
| 203 |
|
|
| 204 |
|
|
| 205 |
+ |
static |
| 206 |
+ |
encodedir(pa, dv) /* encode a normalized direction vector */ |
| 207 |
+ |
short pa[2]; |
| 208 |
+ |
FVECT dv; |
| 209 |
+ |
{ |
| 210 |
+ |
pa[1] = 0; |
| 211 |
+ |
if (dv[2] >= 1.) |
| 212 |
+ |
pa[0] = (1L<<15)-1; |
| 213 |
+ |
else if (dv[2] <= -1.) |
| 214 |
+ |
pa[0] = -((1L<<15)-1); |
| 215 |
+ |
else { |
| 216 |
+ |
pa[0] = ((1L<<15)-1)/(PI/2.) * asin(dv[2]); |
| 217 |
+ |
pa[1] = ((1L<<15)-1)/PI * atan2(dv[1], dv[0]); |
| 218 |
+ |
} |
| 219 |
+ |
} |
| 220 |
+ |
|
| 221 |
+ |
|
| 222 |
+ |
#define ALTSHFT 5 |
| 223 |
+ |
#define NALT (1<<ALTSHFT) |
| 224 |
+ |
#define azisft(alt) azisftab[abs(alt)>>(15-ALTSHFT)] |
| 225 |
+ |
|
| 226 |
+ |
static unsigned short azisftab[NALT]; |
| 227 |
+ |
|
| 228 |
+ |
static |
| 229 |
+ |
azisftinit(alt) /* initialize azimuth scale factor table */ |
| 230 |
+ |
int alt; |
| 231 |
+ |
{ |
| 232 |
+ |
register int i; |
| 233 |
+ |
|
| 234 |
+ |
for (i = NALT; i--; ) |
| 235 |
+ |
azisftab[i] = 2.*(1L<<15) * cos(PI/2.*(i+.5)/NALT); |
| 236 |
+ |
return(azisft(alt)); |
| 237 |
+ |
} |
| 238 |
+ |
|
| 239 |
+ |
#define azisf(alt) (azisftab[0] ? azisft(alt) : azisftinit(alt)) >> 15 |
| 240 |
+ |
|
| 241 |
+ |
static long |
| 242 |
+ |
dir2diff(pa1, pa2) /* relative distance^2 between directions */ |
| 243 |
+ |
short pa1[2], pa2[2]; |
| 244 |
+ |
{ |
| 245 |
+ |
long altd2, azid2; |
| 246 |
+ |
int alt; |
| 247 |
+ |
|
| 248 |
+ |
altd2 = pa1[0] - pa2[0]; /* get altitude difference^2 */ |
| 249 |
+ |
altd2 *= altd2; |
| 250 |
+ |
if (altd2 > MAXDIFF2) |
| 251 |
+ |
return(altd2); /* too large already */ |
| 252 |
+ |
azid2 = pa1[1] - pa2[1]; /* get adjusted azimuth difference^2 */ |
| 253 |
+ |
if (azid2 < 0) azid2 = -azid2; |
| 254 |
+ |
if (azid2 >= 1L<<15) { /* wrap sphere */ |
| 255 |
+ |
azid2 -= 1L<<16; |
| 256 |
+ |
if (azid2 < 0) azid2 = -azid2; |
| 257 |
+ |
} |
| 258 |
+ |
alt = (pa1[0]+pa2[0])/2; |
| 259 |
+ |
azid2 = azid2*azisf(alt); /* evaluation order is important */ |
| 260 |
+ |
azid2 *= azid2; |
| 261 |
+ |
return(altd2 + azid2); |
| 262 |
+ |
} |
| 263 |
+ |
|
| 264 |
+ |
|
| 265 |
|
int |
| 266 |
|
qtFindLeaf(x, y) /* find closest leaf to (x,y) */ |
| 267 |
|
int x, y; |
| 306 |
|
register RTREE *tp = &qtrunk; |
| 307 |
|
int x0=0, y0=0, x1=odev.hres, y1=odev.vres; |
| 308 |
|
int lo = -1; |
| 309 |
+ |
long d2; |
| 310 |
+ |
short dc[2]; |
| 311 |
|
int x, y, mx, my; |
| 312 |
|
double z; |
| 313 |
|
FVECT ip, wp; |
| 314 |
|
register int q; |
| 315 |
< |
/* compute leaf location */ |
| 315 |
> |
/* compute leaf location in view */ |
| 316 |
|
VCOPY(wp, qtL.wp[li]); |
| 317 |
|
viewloc(ip, &odev.v, wp); |
| 318 |
|
if (ip[2] <= 0. || ip[0] < 0. || ip[0] >= 1. |
| 319 |
|
|| ip[1] < 0. || ip[1] >= 1.) |
| 320 |
< |
return; |
| 320 |
> |
return(0); /* behind or outside view */ |
| 321 |
> |
#ifdef DEBUG |
| 322 |
> |
if (odev.v.type == VT_PAR | odev.v.vfore > FTINY) |
| 323 |
> |
error(INTERNAL, "bad view assumption in addleaf"); |
| 324 |
> |
#endif |
| 325 |
> |
for (q = 0; q < 3; q++) |
| 326 |
> |
wp[q] = (wp[q] - odev.v.vp[q])/ip[2]; |
| 327 |
> |
encodedir(dc, wp); /* compute pixel direction */ |
| 328 |
> |
d2 = dir2diff(dc, qtL.wd[li]); |
| 329 |
> |
if (d2 > MAXDIFF2) |
| 330 |
> |
return(0); /* leaf dir. too far off */ |
| 331 |
|
x = ip[0] * odev.hres; |
| 332 |
|
y = ip[1] * odev.vres; |
| 333 |
|
z = ip[2]; |
| 350 |
|
tp->flgs |= CHLFF(q); |
| 351 |
|
break; |
| 352 |
|
} |
| 353 |
< |
/* check existing leaf */ |
| 271 |
< |
if (lo != tp->k[q].li) { |
| 353 |
> |
if (lo != tp->k[q].li) { /* check old leaf */ |
| 354 |
|
lo = tp->k[q].li; |
| 355 |
|
VCOPY(wp, qtL.wp[lo]); |
| 356 |
|
viewloc(ip, &odev.v, wp); |
| 357 |
|
} |
| 358 |
|
/* is node minimum size? */ |
| 359 |
< |
if (x1-x0 <= qtMinNodesiz || y1-y0 <= qtMinNodesiz) { |
| 360 |
< |
if (z > (1.-qtDepthEps)*ip[2]) /* who is closer? */ |
| 361 |
< |
return; /* old one is */ |
| 362 |
< |
tp->k[q].li = li; /* new one is */ |
| 359 |
> |
if (y1-y0 <= qtMinNodesiz || x1-x0 <= qtMinNodesiz) { |
| 360 |
> |
if (z > (1.+qtDepthEps)*ip[2]) |
| 361 |
> |
return(0); /* old one closer */ |
| 362 |
> |
if (z >= (1.-qtDepthEps)*ip[2] && |
| 363 |
> |
dir2diff(dc, qtL.wd[lo]) < d2) |
| 364 |
> |
return(0); /* old one better */ |
| 365 |
> |
tp->k[q].li = li; /* else new one is */ |
| 366 |
|
tp->flgs |= CHF(q); |
| 367 |
|
break; |
| 368 |
|
} |
| 377 |
|
tp->k[q].li = lo; |
| 378 |
|
tp->flgs |= LFF(q)|CH_ANY; /* all new */ |
| 379 |
|
} |
| 380 |
+ |
return(1); /* done */ |
| 381 |
|
} |
| 382 |
|
|
| 383 |
|
|
| 384 |
< |
dev_value(c, p) /* add a pixel value to our output queue */ |
| 384 |
> |
dev_value(c, p, v) /* add a pixel value to our quadtree */ |
| 385 |
|
COLR c; |
| 386 |
< |
FVECT p; |
| 386 |
> |
FVECT p, v; |
| 387 |
|
{ |
| 388 |
|
register int li; |
| 389 |
|
|
| 390 |
|
li = newleaf(); |
| 391 |
|
VCOPY(qtL.wp[li], p); |
| 392 |
+ |
encodedir(qtL.wd[li], v); |
| 393 |
|
tmCvColrs(&qtL.brt[li], qtL.chr[li], c, 1); |
| 394 |
< |
addleaf(li); |
| 394 |
> |
if (!addleaf(li)) |
| 395 |
> |
ungetleaf(li); |
| 396 |
|
} |
| 397 |
|
|
| 398 |
|
|
| 448 |
|
qtL.chr+borg, blen); |
| 449 |
|
qtL.tml = qtL.tl; |
| 450 |
|
return(1); |
| 363 |
– |
} |
| 364 |
– |
|
| 365 |
– |
|
| 366 |
– |
static |
| 367 |
– |
redraw(ca, tp, x0, y0, x1, y1, l) /* redraw portion of a tree */ |
| 368 |
– |
BYTE ca[3]; /* returned average color */ |
| 369 |
– |
register RTREE *tp; |
| 370 |
– |
int x0, y0, x1, y1; |
| 371 |
– |
int l[2][2]; |
| 372 |
– |
{ |
| 373 |
– |
int csm[3], nc; |
| 374 |
– |
register BYTE *cp; |
| 375 |
– |
BYTE rgb[3]; |
| 376 |
– |
int quads = CH_ANY; |
| 377 |
– |
int mx, my; |
| 378 |
– |
register int i; |
| 379 |
– |
/* compute midpoint */ |
| 380 |
– |
mx = (x0 + x1) >> 1; |
| 381 |
– |
my = (y0 + y1) >> 1; |
| 382 |
– |
/* see what to do */ |
| 383 |
– |
if (l[0][0] >= mx) |
| 384 |
– |
quads &= ~(CHF(2)|CHF(0)); |
| 385 |
– |
else if (l[0][1] <= mx) |
| 386 |
– |
quads &= ~(CHF(3)|CHF(1)); |
| 387 |
– |
if (l[1][0] >= my) |
| 388 |
– |
quads &= ~(CHF(1)|CHF(0)); |
| 389 |
– |
else if (l[1][1] <= my) |
| 390 |
– |
quads &= ~(CHF(3)|CHF(2)); |
| 391 |
– |
tp->flgs &= ~quads; /* mark them done */ |
| 392 |
– |
csm[0] = csm[1] = csm[2] = nc = 0; |
| 393 |
– |
/* do leaves first */ |
| 394 |
– |
for (i = 0; i < 4; i++) |
| 395 |
– |
if (quads & CHF(i) && tp->flgs & LFF(i)) { |
| 396 |
– |
dev_paintr(cp=qtL.rgb[tp->k[i].li], |
| 397 |
– |
i&01 ? mx : x0, i&02 ? my : y0, |
| 398 |
– |
i&01 ? x1 : mx, i&02 ? y1 : my); |
| 399 |
– |
csm[0] += cp[0]; csm[1] += cp[1]; csm[2] += cp[2]; |
| 400 |
– |
nc++; |
| 401 |
– |
quads &= ~CHF(i); |
| 402 |
– |
} |
| 403 |
– |
/* now do branches */ |
| 404 |
– |
for (i = 0; i < 4; i++) |
| 405 |
– |
if (quads & CHF(i) && tp->flgs & BRF(i)) { |
| 406 |
– |
redraw(rgb, tp->k[i].b, i&01 ? mx : x0, i&02 ? my : y0, |
| 407 |
– |
i&01 ? x1 : mx, i&02 ? y1 : my, l); |
| 408 |
– |
csm[0] += rgb[0]; csm[1] += rgb[1]; csm[2] += rgb[2]; |
| 409 |
– |
nc++; |
| 410 |
– |
quads &= ~CHF(i); |
| 411 |
– |
} |
| 412 |
– |
if (nc > 1) { |
| 413 |
– |
ca[0] = csm[0]/nc; ca[1] = csm[1]/nc; ca[2] = csm[2]/nc; |
| 414 |
– |
} else { |
| 415 |
– |
ca[0] = csm[0]; ca[1] = csm[1]; ca[2] = csm[2]; |
| 416 |
– |
} |
| 417 |
– |
if (!quads) return; |
| 418 |
– |
/* fill in gaps with average */ |
| 419 |
– |
for (i = 0; i < 4; i++) |
| 420 |
– |
if (quads & CHF(i)) |
| 421 |
– |
dev_paintr(ca, i&01 ? mx : x0, i&02 ? my : y0, |
| 422 |
– |
i&01 ? x1 : mx, i&02 ? y1 : my); |
| 423 |
– |
} |
| 424 |
– |
|
| 425 |
– |
|
| 426 |
– |
static |
| 427 |
– |
update(ca, tp, x0, y0, x1, y1) /* update tree display as needed */ |
| 428 |
– |
BYTE ca[3]; /* returned average color */ |
| 429 |
– |
register RTREE *tp; |
| 430 |
– |
int x0, y0, x1, y1; |
| 431 |
– |
{ |
| 432 |
– |
int csm[3], nc; |
| 433 |
– |
register BYTE *cp; |
| 434 |
– |
BYTE rgb[3]; |
| 435 |
– |
int gaps = 0; |
| 436 |
– |
int mx, my; |
| 437 |
– |
register int i; |
| 438 |
– |
/* compute midpoint */ |
| 439 |
– |
mx = (x0 + x1) >> 1; |
| 440 |
– |
my = (y0 + y1) >> 1; |
| 441 |
– |
csm[0] = csm[1] = csm[2] = nc = 0; |
| 442 |
– |
/* do leaves first */ |
| 443 |
– |
for (i = 0; i < 4; i++) { |
| 444 |
– |
if (!(tp->flgs & CHF(i))) |
| 445 |
– |
continue; |
| 446 |
– |
if (tp->flgs & LFF(i)) { |
| 447 |
– |
dev_paintr(cp=qtL.rgb[tp->k[i].li], |
| 448 |
– |
i&01 ? mx : x0, i&02 ? my : y0, |
| 449 |
– |
i&01 ? x1 : mx, i&02 ? y1 : my); |
| 450 |
– |
csm[0] += cp[0]; csm[1] += cp[1]; csm[2] += cp[2]; |
| 451 |
– |
nc++; |
| 452 |
– |
} else if (!(tp->flgs & BRF(i))) |
| 453 |
– |
gaps |= 1<<i; /* empty stem */ |
| 454 |
– |
} |
| 455 |
– |
/* now do branches */ |
| 456 |
– |
for (i = 0; i < 4; i++) |
| 457 |
– |
if ((tp->flgs & CHBRF(i)) == CHBRF(i)) { |
| 458 |
– |
update(rgb, tp->k[i].b, i&01 ? mx : x0, i&02 ? my : y0, |
| 459 |
– |
i&01 ? x1 : mx, i&02 ? y1 : my); |
| 460 |
– |
csm[0] += rgb[0]; csm[1] += rgb[1]; csm[2] += rgb[2]; |
| 461 |
– |
nc++; |
| 462 |
– |
} |
| 463 |
– |
if (nc > 1) { |
| 464 |
– |
ca[0] = csm[0]/nc; ca[1] = csm[1]/nc; ca[2] = csm[2]/nc; |
| 465 |
– |
} else { |
| 466 |
– |
ca[0] = csm[0]; ca[1] = csm[1]; ca[2] = csm[2]; |
| 467 |
– |
} |
| 468 |
– |
/* fill in gaps with average */ |
| 469 |
– |
for (i = 0; gaps && i < 4; gaps >>= 1, i++) |
| 470 |
– |
if (gaps & 01) |
| 471 |
– |
dev_paintr(ca, i&01 ? mx : x0, i&02 ? my : y0, |
| 472 |
– |
i&01 ? x1 : mx, i&02 ? y1 : my); |
| 473 |
– |
tp->flgs &= ~CH_ANY; /* all done */ |
| 474 |
– |
} |
| 475 |
– |
|
| 476 |
– |
|
| 477 |
– |
qtRedraw(x0, y0, x1, y1) /* redraw part or all of our screen */ |
| 478 |
– |
int x0, y0, x1, y1; |
| 479 |
– |
{ |
| 480 |
– |
int lim[2][2]; |
| 481 |
– |
BYTE ca[3]; |
| 482 |
– |
|
| 483 |
– |
if (is_stump(&qtrunk)) |
| 484 |
– |
return; |
| 485 |
– |
if (!qtMapLeaves((lim[0][0]=x0) <= 0 & (lim[1][0]=y0) <= 0 & |
| 486 |
– |
(lim[0][1]=x1) >= odev.hres-1 & (lim[1][1]=y1) >= odev.vres-1)) |
| 487 |
– |
return; |
| 488 |
– |
redraw(ca, &qtrunk, 0, 0, odev.hres, odev.vres, lim); |
| 489 |
– |
} |
| 490 |
– |
|
| 491 |
– |
|
| 492 |
– |
qtUpdate() /* update our tree display */ |
| 493 |
– |
{ |
| 494 |
– |
BYTE ca[3]; |
| 495 |
– |
|
| 496 |
– |
if (is_stump(&qtrunk)) |
| 497 |
– |
return; |
| 498 |
– |
if (!qtMapLeaves(0)) |
| 499 |
– |
return; |
| 500 |
– |
update(ca, &qtrunk, 0, 0, odev.hres, odev.vres); |
| 451 |
|
} |