ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/px/pinterp.c
Revision: 2.44
Committed: Sun Mar 24 19:00:33 2013 UTC (11 years, 1 month ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.43: +22 -22 lines
Log Message:
Corrected definition of matrix

File Contents

# Content
1 #ifndef lint
2 static const char RCSid[] = "$Id: pinterp.c,v 2.43 2012/02/08 13:43:21 greg Exp $";
3 #endif
4 /*
5 * Interpolate and extrapolate pictures with different view parameters.
6 *
7 * Greg Ward 09Dec89
8 */
9
10 #include "copyright.h"
11
12 #include <ctype.h>
13 #include <string.h>
14
15 #include "platform.h"
16 #include "standard.h"
17 #include "rtprocess.h" /* Windows: must come before color.h */
18 #include "view.h"
19 #include "color.h"
20
21 #define LOG2 0.69314718055994530942
22
23 #define pscan(y) (ourpict+(y)*hresolu)
24 #define sscan(y) (ourspict+(y)*hresolu)
25 #define wscan(y) (ourweigh+(y)*hresolu)
26 #define zscan(y) (ourzbuf+(y)*hresolu)
27 #define bscan(y) (ourbpict+(y)*hresolu)
28 #define averaging (ourweigh != NULL)
29 #define blurring (ourbpict != NULL)
30 #define usematrix (hasmatrix & !averaging)
31 #define zisnorm ((!usematrix) | (ourview.type != VT_PER))
32
33 #define MAXWT 1000. /* maximum pixel weight (averaging) */
34
35 #define F_FORE 1 /* fill foreground */
36 #define F_BACK 2 /* fill background */
37
38 #define PACKSIZ 256 /* max. calculation packet size */
39
40 #define RTCOM "rtrace -h- -ovl -fff -ld- -i- -I- "
41
42 #define ABS(x) ((x)>0?(x):-(x))
43
44 struct position {int x,y; float z;};
45
46 #define NSTEPS 64 /* number steps in overlap prescan */
47 #define MINSTEP 4 /* minimum worthwhile preview step */
48
49 struct bound {int min,max;};
50
51 VIEW ourview = STDVIEW; /* desired view */
52 int hresolu = 512; /* horizontal resolution */
53 int vresolu = 512; /* vertical resolution */
54 double pixaspect = 1.0; /* pixel aspect ratio */
55
56 double zeps = .02; /* allowed z epsilon */
57
58 COLR *ourpict; /* output picture (COLR's) */
59 COLOR *ourspict; /* output pixel sums (averaging) */
60 float *ourweigh = NULL; /* output pixel weights (averaging) */
61 float *ourzbuf; /* corresponding z-buffer */
62 COLOR *ourbpict = NULL; /* blurred picture (view averaging) */
63
64 VIEW avgview; /* average view for -B option */
65 int nvavg; /* number of views averaged */
66
67 char *progname;
68
69 int fillo = F_FORE|F_BACK; /* selected fill options */
70 int fillsamp = 0; /* sample separation (0 == inf) */
71 COLR backcolr = BLKCOLR; /* background color */
72 COLOR backcolor = BLKCOLOR; /* background color (float) */
73 double backz = 0.0; /* background z value */
74 int normdist = 1; /* i/o normalized distance? */
75 char ourfmt[LPICFMT+1] = PICFMT; /* original picture format */
76 double ourexp = -1; /* original picture exposure */
77 int expadj = 0; /* exposure adjustment (f-stops) */
78 double rexpadj = 1; /* real exposure adjustment */
79
80 VIEW theirview; /* input view */
81 int gotview; /* got input view? */
82 int wrongformat = 0; /* input in another format? */
83 RESOLU tresolu; /* input resolution */
84 double theirexp; /* input picture exposure */
85 MAT4 theirs2ours; /* transformation matrix */
86 int hasmatrix = 0; /* has transformation matrix */
87
88 static SUBPROC PDesc = SP_INACTIVE; /* rtrace process descriptor */
89 unsigned short queue[PACKSIZ][2]; /* pending pixels */
90 int packsiz; /* actual packet size */
91 int queuesiz = 0; /* number of pixels pending */
92
93 typedef void fillfunc_t(int x, int y);
94
95 static gethfunc headline;
96 static int nextview(FILE *fp);
97 static void compavgview(void);
98 static void addpicture(char *pfile, char *zspec);
99 static int pixform(MAT4 xfmat, VIEW *vw1, VIEW *vw2);
100 static void addscanline(struct bound *xl, int y,
101 COLR *pline, float *zline, struct position *lasty);
102 static void addpixel(struct position *p0, struct position *p1,
103 struct position *p2, COLR pix, double w, double z);
104 static double movepixel(FVECT pos);
105 static int getperim(struct bound *xl, struct bound *yl, float *zline, int zfd);
106 static void backpicture(fillfunc_t *fill, int samp);
107 static void fillpicture(fillfunc_t *fill);
108 static int clipaft(void);
109 static int addblur(void);
110 static void writepicture(void);
111 static void writedistance(char *fname);
112 static fillfunc_t backfill;
113 static fillfunc_t rcalfill;
114 static void calstart(char *prog, char *args);
115 static void caldone(void);
116 static void clearqueue(void);
117 static void syserror(char *s);
118
119 fillfunc_t *fillfunc = backfill; /* selected fill function */
120
121 int
122 main( /* interpolate pictures */
123 int argc,
124 char *argv[]
125 )
126 {
127 #define check(ol,al) if (argv[an][ol] || \
128 badarg(argc-an-1,argv+an+1,al)) \
129 goto badopt
130 int gotvfile = 0;
131 int doavg = -1;
132 int doblur = 0;
133 char *zfile = NULL;
134 char *expcomp = NULL;
135 int i, an, rval;
136
137 SET_DEFAULT_BINARY();
138 SET_FILE_BINARY(stdout);
139
140 progname = argv[0];
141
142 for (an = 1; an < argc && argv[an][0] == '-'; an++) {
143 rval = getviewopt(&ourview, argc-an, argv+an);
144 if (rval >= 0) {
145 an += rval;
146 continue;
147 }
148 switch (argv[an][1]) {
149 case 'e': /* exposure */
150 check(2,"f");
151 expcomp = argv[++an];
152 break;
153 case 't': /* threshold */
154 check(2,"f");
155 zeps = atof(argv[++an]);
156 break;
157 case 'a': /* average */
158 check(2,NULL);
159 doavg = 1;
160 break;
161 case 'B': /* blur views */
162 check(2,NULL);
163 doblur = 1;
164 break;
165 case 'q': /* quick (no avg.) */
166 check(2,NULL);
167 doavg = 0;
168 break;
169 case 'n': /* dist. normalized? */
170 check(2,NULL);
171 normdist = !normdist;
172 break;
173 case 'f': /* fill type */
174 switch (argv[an][2]) {
175 case '0': /* none */
176 check(3,NULL);
177 fillo = 0;
178 break;
179 case 'f': /* foreground */
180 check(3,NULL);
181 fillo = F_FORE;
182 break;
183 case 'b': /* background */
184 check(3,NULL);
185 fillo = F_BACK;
186 break;
187 case 'a': /* all */
188 check(3,NULL);
189 fillo = F_FORE|F_BACK;
190 break;
191 case 's': /* sample */
192 check(3,"i");
193 fillsamp = atoi(argv[++an]);
194 break;
195 case 'c': /* color */
196 check(3,"fff");
197 fillfunc = backfill;
198 setcolor(backcolor, atof(argv[an+1]),
199 atof(argv[an+2]), atof(argv[an+3]));
200 setcolr(backcolr, colval(backcolor,RED),
201 colval(backcolor,GRN),
202 colval(backcolor,BLU));
203 an += 3;
204 break;
205 case 'z': /* z value */
206 check(3,"f");
207 fillfunc = backfill;
208 backz = atof(argv[++an]);
209 break;
210 case 'r': /* rtrace */
211 check(3,"s");
212 fillfunc = rcalfill;
213 calstart(RTCOM, argv[++an]);
214 break;
215 default:
216 goto badopt;
217 }
218 break;
219 case 'z': /* z file */
220 check(2,"s");
221 zfile = argv[++an];
222 break;
223 case 'x': /* x resolution */
224 check(2,"i");
225 hresolu = atoi(argv[++an]);
226 break;
227 case 'y': /* y resolution */
228 check(2,"i");
229 vresolu = atoi(argv[++an]);
230 break;
231 case 'p': /* pixel aspect */
232 if (argv[an][2] != 'a')
233 goto badopt;
234 check(3,"f");
235 pixaspect = atof(argv[++an]);
236 break;
237 case 'v': /* view file */
238 if (argv[an][2] != 'f')
239 goto badopt;
240 check(3,"s");
241 gotvfile = viewfile(argv[++an], &ourview, NULL);
242 if (gotvfile < 0)
243 syserror(argv[an]);
244 else if (gotvfile == 0) {
245 fprintf(stderr, "%s: bad view file\n",
246 argv[an]);
247 exit(1);
248 }
249 break;
250 default:
251 badopt:
252 fprintf(stderr, "%s: command line error at '%s'\n",
253 progname, argv[an]);
254 goto userr;
255 }
256 }
257 /* check arguments */
258 if ((argc-an)%2)
259 goto userr;
260 if (fillsamp == 1)
261 fillo &= ~F_BACK;
262 if (doavg < 0)
263 doavg = (argc-an) > 2;
264 if (expcomp != NULL) {
265 if ((expcomp[0] == '+') | (expcomp[0] == '-')) {
266 expadj = atof(expcomp) + (expcomp[0]=='+' ? .5 : -.5);
267 if (doavg | doblur)
268 rexpadj = pow(2.0, atof(expcomp));
269 else
270 rexpadj = pow(2.0, (double)expadj);
271 } else {
272 if (!isflt(expcomp))
273 goto userr;
274 rexpadj = atof(expcomp);
275 expadj = log(rexpadj)/LOG2 + (rexpadj>1 ? .5 : -.5);
276 if (!(doavg | doblur))
277 rexpadj = pow(2.0, (double)expadj);
278 }
279 }
280 /* set view */
281 if (nextview(doblur ? stdin : (FILE *)NULL) == EOF) {
282 fprintf(stderr, "%s: no view on standard input!\n",
283 progname);
284 exit(1);
285 }
286 normaspect(viewaspect(&ourview), &pixaspect, &hresolu, &vresolu);
287 /* allocate frame */
288 if (doavg) {
289 ourspict = (COLOR *)bmalloc(hresolu*vresolu*sizeof(COLOR));
290 ourweigh = (float *)bmalloc(hresolu*vresolu*sizeof(float));
291 if ((ourspict == NULL) | (ourweigh == NULL))
292 syserror(progname);
293 } else {
294 ourpict = (COLR *)bmalloc(hresolu*vresolu*sizeof(COLR));
295 if (ourpict == NULL)
296 syserror(progname);
297 }
298 if (doblur) {
299 ourbpict = (COLOR *)bmalloc(hresolu*vresolu*sizeof(COLOR));
300 if (ourbpict == NULL)
301 syserror(progname);
302 }
303 ourzbuf = (float *)bmalloc(hresolu*vresolu*sizeof(float));
304 if (ourzbuf == NULL)
305 syserror(progname);
306 /* new header */
307 newheader("RADIANCE", stdout);
308 fputnow(stdout);
309 /* run pictures */
310 do {
311 memset((char *)ourzbuf, '\0', hresolu*vresolu*sizeof(float));
312 for (i = an; i < argc; i += 2)
313 addpicture(argv[i], argv[i+1]);
314 if (fillo&F_BACK) /* fill in spaces */
315 backpicture(fillfunc, fillsamp);
316 else
317 fillpicture(fillfunc);
318 /* aft clipping */
319 clipaft();
320 } while (addblur() && nextview(stdin) != EOF);
321 /* close calculation */
322 caldone();
323 /* add to header */
324 printargs(argc, argv, stdout);
325 compavgview();
326 if (doblur | gotvfile) {
327 fputs(VIEWSTR, stdout);
328 fprintview(&avgview, stdout);
329 putc('\n', stdout);
330 }
331 if ((pixaspect < .99) | (pixaspect > 1.01))
332 fputaspect(pixaspect, stdout);
333 if (ourexp > 0)
334 ourexp *= rexpadj;
335 else
336 ourexp = rexpadj;
337 if ((ourexp < .995) | (ourexp > 1.005))
338 fputexpos(ourexp, stdout);
339 if (strcmp(ourfmt, PICFMT)) /* print format if known */
340 fputformat(ourfmt, stdout);
341 putc('\n', stdout);
342 /* write picture */
343 writepicture();
344 /* write z file */
345 if (zfile != NULL)
346 writedistance(zfile);
347
348 exit(0);
349 userr:
350 fprintf(stderr,
351 "Usage: %s [view opts][-t eps][-z zout][-e spec][-B][-a|-q][-fT][-n] pfile zspec ..\n",
352 progname);
353 exit(1);
354 #undef check
355 }
356
357
358 static int
359 headline( /* process header string */
360 char *s,
361 void *p
362 )
363 {
364 char fmt[32];
365
366 if (isheadid(s))
367 return(0);
368 if (formatval(fmt, s)) {
369 if (globmatch(ourfmt, fmt)) {
370 wrongformat = 0;
371 strcpy(ourfmt, fmt);
372 } else
373 wrongformat = 1;
374 return(0);
375 }
376 if (nvavg < 2) {
377 putc('\t', stdout);
378 fputs(s, stdout);
379 }
380 if (isexpos(s)) {
381 theirexp *= exposval(s);
382 return(0);
383 }
384 if (isview(s) && sscanview(&theirview, s) > 0)
385 gotview++;
386 return(0);
387 }
388
389
390 static int
391 nextview( /* get and set next view */
392 FILE *fp
393 )
394 {
395 char linebuf[256];
396 char *err;
397 int i;
398
399 if (fp != NULL) {
400 do /* get new view */
401 if (fgets(linebuf, sizeof(linebuf), fp) == NULL)
402 return(EOF);
403 while (!isview(linebuf) || !sscanview(&ourview, linebuf));
404 }
405 /* set new view */
406 if ((err = setview(&ourview)) != NULL) {
407 fprintf(stderr, "%s: %s\n", progname, err);
408 exit(1);
409 }
410 if (!nvavg) { /* first view */
411 avgview = ourview;
412 return(nvavg++);
413 }
414 /* add to average view */
415 for (i = 0; i < 3; i++) {
416 avgview.vp[i] += ourview.vp[i];
417 avgview.vdir[i] += ourview.vdir[i];
418 avgview.vup[i] += ourview.vup[i];
419 }
420 avgview.vdist += ourview.vdist;
421 avgview.horiz += ourview.horiz;
422 avgview.vert += ourview.vert;
423 avgview.hoff += ourview.hoff;
424 avgview.voff += ourview.voff;
425 avgview.vfore += ourview.vfore;
426 avgview.vaft += ourview.vaft;
427 return(nvavg++);
428 }
429
430
431 static void
432 compavgview(void) /* compute average view */
433 {
434 int i;
435 double f;
436
437 if (nvavg < 2)
438 return;
439 f = 1.0/nvavg;
440 for (i = 0; i < 3; i++) {
441 avgview.vp[i] *= f;
442 avgview.vdir[i] *= f;
443 avgview.vup[i] *= f;
444 }
445 avgview.vdist *= f;
446 avgview.horiz *= f;
447 avgview.vert *= f;
448 avgview.hoff *= f;
449 avgview.voff *= f;
450 avgview.vfore *= f;
451 avgview.vaft *= f;
452 if (setview(&avgview) != NULL) /* in case of emergency... */
453 avgview = ourview;
454 pixaspect = viewaspect(&avgview) * hresolu / vresolu;
455 }
456
457
458 static void
459 addpicture( /* add picture to output */
460 char *pfile,
461 char *zspec
462 )
463 {
464 FILE *pfp;
465 int zfd;
466 char *err;
467 COLR *scanin;
468 float *zin;
469 struct position *plast;
470 struct bound *xlim, ylim;
471 int y;
472 /* open picture file */
473 if ((pfp = fopen(pfile, "r")) == NULL)
474 syserror(pfile);
475 /* get header with exposure and view */
476 theirexp = 1.0;
477 theirview = stdview;
478 gotview = 0;
479 if (nvavg < 2)
480 printf("%s:\n", pfile);
481 getheader(pfp, headline, NULL);
482 if (wrongformat || !gotview || !fgetsresolu(&tresolu, pfp)) {
483 fprintf(stderr, "%s: picture format error\n", pfile);
484 exit(1);
485 }
486 if (ourexp <= 0)
487 ourexp = theirexp;
488 else if (ABS(theirexp-ourexp) > .01*ourexp)
489 fprintf(stderr, "%s: different exposure (warning)\n", pfile);
490 if ( (err = setview(&theirview)) ) {
491 fprintf(stderr, "%s: %s\n", pfile, err);
492 exit(1);
493 }
494 /* compute transformation */
495 hasmatrix = pixform(theirs2ours, &theirview, &ourview);
496 /* get z specification or file */
497 zin = (float *)malloc(scanlen(&tresolu)*sizeof(float));
498 if (zin == NULL)
499 syserror(progname);
500 if ((zfd = open(zspec, O_RDONLY)) == -1) {
501 double zvalue;
502 int x;
503 if (!isflt(zspec) || (zvalue = atof(zspec)) <= 0.0)
504 syserror(zspec);
505 for (x = scanlen(&tresolu); x-- > 0; )
506 zin[x] = zvalue;
507 }
508 /* compute transferrable perimeter */
509 xlim = (struct bound *)malloc(numscans(&tresolu)*sizeof(struct bound));
510 if (xlim == NULL)
511 syserror(progname);
512 if (!getperim(xlim, &ylim, zin, zfd)) { /* overlapping area? */
513 free((void *)zin);
514 free((void *)xlim);
515 if (zfd != -1)
516 close(zfd);
517 fclose(pfp);
518 return;
519 }
520 /* allocate scanlines */
521 scanin = (COLR *)malloc(scanlen(&tresolu)*sizeof(COLR));
522 plast = (struct position *)calloc(scanlen(&tresolu),
523 sizeof(struct position));
524 if ((scanin == NULL) | (plast == NULL))
525 syserror(progname);
526 /* skip to starting point */
527 for (y = 0; y < ylim.min; y++)
528 if (freadcolrs(scanin, scanlen(&tresolu), pfp) < 0) {
529 fprintf(stderr, "%s: read error\n", pfile);
530 exit(1);
531 }
532 if (zfd != -1 && lseek(zfd,
533 (off_t)ylim.min*scanlen(&tresolu)*sizeof(float),
534 SEEK_SET) < 0)
535 syserror(zspec);
536 /* load image */
537 for (y = ylim.min; y <= ylim.max; y++) {
538 if (freadcolrs(scanin, scanlen(&tresolu), pfp) < 0) {
539 fprintf(stderr, "%s: read error\n", pfile);
540 exit(1);
541 }
542 if (zfd != -1 && read(zfd, (char *)zin,
543 scanlen(&tresolu)*sizeof(float))
544 < scanlen(&tresolu)*sizeof(float))
545 syserror(zspec);
546 addscanline(xlim+y, y, scanin, zin, plast);
547 }
548 /* clean up */
549 free((void *)xlim);
550 free((void *)scanin);
551 free((void *)zin);
552 free((void *)plast);
553 fclose(pfp);
554 if (zfd != -1)
555 close(zfd);
556 }
557
558
559 static int
560 pixform( /* compute view1 to view2 matrix */
561 MAT4 xfmat,
562 VIEW *vw1,
563 VIEW *vw2
564 )
565 {
566 MAT4 m4t;
567
568 if ((vw1->type != VT_PER) & (vw1->type != VT_PAR))
569 return(0);
570 if ((vw2->type != VT_PER) & (vw2->type != VT_PAR))
571 return(0);
572 setident4(xfmat);
573 xfmat[0][0] = vw1->hvec[0];
574 xfmat[0][1] = vw1->hvec[1];
575 xfmat[0][2] = vw1->hvec[2];
576 xfmat[1][0] = vw1->vvec[0];
577 xfmat[1][1] = vw1->vvec[1];
578 xfmat[1][2] = vw1->vvec[2];
579 xfmat[2][0] = vw1->vdir[0];
580 xfmat[2][1] = vw1->vdir[1];
581 xfmat[2][2] = vw1->vdir[2];
582 xfmat[3][0] = vw1->vp[0];
583 xfmat[3][1] = vw1->vp[1];
584 xfmat[3][2] = vw1->vp[2];
585 setident4(m4t);
586 m4t[0][0] = vw2->hvec[0]/vw2->hn2;
587 m4t[1][0] = vw2->hvec[1]/vw2->hn2;
588 m4t[2][0] = vw2->hvec[2]/vw2->hn2;
589 m4t[3][0] = -DOT(vw2->vp,vw2->hvec)/vw2->hn2;
590 m4t[0][1] = vw2->vvec[0]/vw2->vn2;
591 m4t[1][1] = vw2->vvec[1]/vw2->vn2;
592 m4t[2][1] = vw2->vvec[2]/vw2->vn2;
593 m4t[3][1] = -DOT(vw2->vp,vw2->vvec)/vw2->vn2;
594 m4t[0][2] = vw2->vdir[0];
595 m4t[1][2] = vw2->vdir[1];
596 m4t[2][2] = vw2->vdir[2];
597 m4t[3][2] = -DOT(vw2->vp,vw2->vdir);
598 multmat4(xfmat, xfmat, m4t);
599 return(1);
600 }
601
602
603 static void
604 addscanline( /* add scanline to output */
605 struct bound *xl,
606 int y,
607 COLR *pline,
608 float *zline,
609 struct position *lasty /* input/output */
610 )
611 {
612 FVECT pos;
613 struct position lastx, newpos;
614 double wt;
615 int x;
616
617 lastx.z = 0;
618 for (x = xl->max; x >= xl->min; x--) {
619 pix2loc(pos, &tresolu, x, y);
620 pos[2] = zline[x];
621 if ((wt = movepixel(pos)) <= FTINY) {
622 lasty[x].z = lastx.z = 0; /* mark invalid */
623 continue;
624 }
625 /* add pixel to our image */
626 newpos.x = pos[0] * hresolu;
627 newpos.y = pos[1] * vresolu;
628 newpos.z = zline[x];
629 addpixel(&newpos, &lastx, &lasty[x], pline[x], wt, pos[2]);
630 lasty[x].x = lastx.x = newpos.x;
631 lasty[x].y = lastx.y = newpos.y;
632 lasty[x].z = lastx.z = newpos.z;
633 }
634 }
635
636
637 static void
638 addpixel( /* fill in pixel parallelogram */
639 struct position *p0,
640 struct position *p1,
641 struct position *p2,
642 COLR pix,
643 double w,
644 double z
645 )
646 {
647 double zt = 2.*zeps*p0->z; /* threshold */
648 COLOR pval; /* converted+weighted pixel */
649 int s1x, s1y, s2x, s2y; /* step sizes */
650 int l1, l2, c1, c2; /* side lengths and counters */
651 int p1isy; /* p0p1 along y? */
652 int x1, y1; /* p1 position */
653 int x, y; /* final position */
654
655 /* compute vector p0p1 */
656 if (fillo&F_FORE && ABS(p1->z-p0->z) <= zt) {
657 s1x = p1->x - p0->x;
658 s1y = p1->y - p0->y;
659 l1 = ABS(s1x);
660 if ( (p1isy = (ABS(s1y) > l1)) )
661 l1 = ABS(s1y);
662 else if (l1 < 1)
663 l1 = 1;
664 } else {
665 l1 = s1x = s1y = 1;
666 p1isy = -1;
667 }
668 /* compute vector p0p2 */
669 if (fillo&F_FORE && ABS(p2->z-p0->z) <= zt) {
670 s2x = p2->x - p0->x;
671 s2y = p2->y - p0->y;
672 if (p1isy == 1)
673 l2 = ABS(s2x);
674 else {
675 l2 = ABS(s2y);
676 if (p1isy != 0 && ABS(s2x) > l2)
677 l2 = ABS(s2x);
678 }
679 if (l2 < 1)
680 l2 = 1;
681 } else
682 l2 = s2x = s2y = 1;
683 /* fill the parallelogram */
684 if (averaging) {
685 colr_color(pval, pix);
686 scalecolor(pval, w);
687 }
688 for (c1 = l1; c1-- > 0; ) {
689 x1 = p0->x + c1*s1x/l1;
690 y1 = p0->y + c1*s1y/l1;
691 for (c2 = l2; c2-- > 0; ) {
692 x = x1 + c2*s2x/l2;
693 if ((x < 0) | (x >= hresolu))
694 continue;
695 y = y1 + c2*s2y/l2;
696 if ((y < 0) | (y >= vresolu))
697 continue;
698 if (averaging) {
699 if (zscan(y)[x] <= 0 || zscan(y)[x]-z
700 > zeps*zscan(y)[x]) {
701 copycolor(sscan(y)[x], pval);
702 wscan(y)[x] = w;
703 zscan(y)[x] = z;
704 } else if (z-zscan(y)[x] <= zeps*zscan(y)[x]) {
705 addcolor(sscan(y)[x], pval);
706 wscan(y)[x] += w;
707 }
708 } else if (zscan(y)[x] <= 0 || zscan(y)[x]-z
709 > zeps*zscan(y)[x]) {
710 copycolr(pscan(y)[x], pix);
711 zscan(y)[x] = z;
712 }
713 }
714 }
715 }
716
717
718 static double
719 movepixel( /* reposition image point */
720 FVECT pos
721 )
722 {
723 FVECT pt, tdir, odir;
724 double d;
725
726 if (pos[2] <= 0) /* empty pixel */
727 return(0);
728 if (usematrix) {
729 pos[0] += theirview.hoff - .5;
730 pos[1] += theirview.voff - .5;
731 if (normdist & (theirview.type == VT_PER))
732 d = sqrt(1. + pos[0]*pos[0]*theirview.hn2
733 + pos[1]*pos[1]*theirview.vn2);
734 else
735 d = 1.;
736 pos[2] += d*theirview.vfore;
737 if (theirview.type == VT_PER) {
738 pos[2] /= d;
739 pos[0] *= pos[2];
740 pos[1] *= pos[2];
741 }
742 multp3(pos, pos, theirs2ours);
743 if (pos[2] <= ourview.vfore)
744 return(0);
745 if (ourview.type == VT_PER) {
746 pos[0] /= pos[2];
747 pos[1] /= pos[2];
748 }
749 pos[0] += .5 - ourview.hoff;
750 pos[1] += .5 - ourview.voff;
751 pos[2] -= ourview.vfore;
752 } else {
753 if (viewray(pt, tdir, &theirview, pos[0], pos[1]) < -FTINY)
754 return(0);
755 if ((!normdist) & (theirview.type == VT_PER)) /* adjust */
756 pos[2] *= sqrt(1. + pos[0]*pos[0]*theirview.hn2
757 + pos[1]*pos[1]*theirview.vn2);
758 pt[0] += tdir[0]*pos[2];
759 pt[1] += tdir[1]*pos[2];
760 pt[2] += tdir[2]*pos[2];
761 viewloc(pos, &ourview, pt);
762 if (pos[2] <= 0)
763 return(0);
764 }
765 if ((pos[0] < 0) | (pos[0] >= 1-FTINY) | (pos[1] < 0) | (pos[1] >= 1-FTINY))
766 return(0);
767 if (!averaging)
768 return(1);
769 /* compute pixel weight */
770 if (ourview.type == VT_PAR) {
771 d = DOT(ourview.vdir,tdir);
772 d = 1. - d*d;
773 } else {
774 VSUB(odir, pt, ourview.vp);
775 d = DOT(odir,tdir);
776 d = 1. - d*d/DOT(odir,odir);
777 }
778 if (d <= 1./MAXWT/MAXWT)
779 return(MAXWT); /* clip to maximum weight */
780 return(1./sqrt(d));
781 }
782
783
784 static int
785 getperim( /* compute overlapping image area */
786 struct bound *xl,
787 struct bound *yl,
788 float *zline,
789 int zfd
790 )
791 {
792 int step;
793 FVECT pos;
794 int x, y;
795 /* set up step size */
796 if (scanlen(&tresolu) < numscans(&tresolu))
797 step = scanlen(&tresolu)/NSTEPS;
798 else
799 step = numscans(&tresolu)/NSTEPS;
800 if (step < MINSTEP) { /* not worth cropping? */
801 yl->min = 0;
802 yl->max = numscans(&tresolu) - 1;
803 x = scanlen(&tresolu) - 1;
804 for (y = numscans(&tresolu); y--; ) {
805 xl[y].min = 0;
806 xl[y].max = x;
807 }
808 return(1);
809 }
810 yl->min = 32000; yl->max = 0; /* search for points on image */
811 for (y = step - 1; y < numscans(&tresolu); y += step) {
812 if (zfd != -1) {
813 if (lseek(zfd, (off_t)y*scanlen(&tresolu)*sizeof(float),
814 SEEK_SET) < 0)
815 syserror("lseek");
816 if (read(zfd, (char *)zline,
817 scanlen(&tresolu)*sizeof(float))
818 < scanlen(&tresolu)*sizeof(float))
819 syserror("read");
820 }
821 xl[y].min = 32000; xl[y].max = 0; /* x max */
822 for (x = scanlen(&tresolu); (x -= step) > 0; ) {
823 pix2loc(pos, &tresolu, x, y);
824 pos[2] = zline[x];
825 if (movepixel(pos) > FTINY) {
826 xl[y].max = x + step - 1;
827 xl[y].min = x - step + 1; /* x min */
828 if (xl[y].min < 0)
829 xl[y].min = 0;
830 for (x = step - 1; x < xl[y].max; x += step) {
831 pix2loc(pos, &tresolu, x, y);
832 pos[2] = zline[x];
833 if (movepixel(pos) > FTINY) {
834 xl[y].min = x - step + 1;
835 break;
836 }
837 }
838 if (y < yl->min) /* y limits */
839 yl->min = y - step + 1;
840 yl->max = y + step - 1;
841 break;
842 }
843 }
844 /* fill in between */
845 if (y < step) {
846 xl[y-1].min = xl[y].min;
847 xl[y-1].max = xl[y].max;
848 } else {
849 if (xl[y].min < xl[y-step].min)
850 xl[y-1].min = xl[y].min;
851 else
852 xl[y-1].min = xl[y-step].min;
853 if (xl[y].max > xl[y-step].max)
854 xl[y-1].max = xl[y].max;
855 else
856 xl[y-1].max = xl[y-step].max;
857 }
858 for (x = 2; x < step; x++)
859 *(xl+y-x) = *(xl+y-1);
860 }
861 if (yl->max >= numscans(&tresolu))
862 yl->max = numscans(&tresolu) - 1;
863 y -= step;
864 for (x = numscans(&tresolu) - 1; x > y; x--) /* fill bottom rows */
865 *(xl+x) = *(xl+y);
866 return(yl->max >= yl->min);
867 }
868
869
870 static void
871 backpicture( /* background fill algorithm */
872 fillfunc_t *fill,
873 int samp
874 )
875 {
876 int *yback, xback;
877 int y;
878 int x, i;
879 /* get back buffer */
880 yback = (int *)malloc(hresolu*sizeof(int));
881 if (yback == NULL)
882 syserror(progname);
883 for (x = 0; x < hresolu; x++)
884 yback[x] = -2;
885 /*
886 * Xback and yback are the pixel locations of suitable
887 * background values in each direction.
888 * A value of -2 means unassigned, and -1 means
889 * that there is no suitable background in this direction.
890 */
891 /* fill image */
892 for (y = 0; y < vresolu; y++) {
893 xback = -2;
894 for (x = 0; x < hresolu; x++)
895 if (zscan(y)[x] <= 0) { /* empty pixel */
896 /*
897 * First, find background from above or below.
898 * (farthest assigned pixel)
899 */
900 if (yback[x] == -2) {
901 for (i = y+1; i < vresolu; i++)
902 if (zscan(i)[x] > 0)
903 break;
904 if (i < vresolu
905 && (y <= 0 || zscan(y-1)[x] < zscan(i)[x]))
906 yback[x] = i;
907 else
908 yback[x] = y-1;
909 }
910 /*
911 * Next, find background from left or right.
912 */
913 if (xback == -2) {
914 for (i = x+1; i < hresolu; i++)
915 if (zscan(y)[i] > 0)
916 break;
917 if (i < hresolu
918 && (x <= 0 || zscan(y)[x-1] < zscan(y)[i]))
919 xback = i;
920 else
921 xback = x-1;
922 }
923 /*
924 * If we have no background for this pixel,
925 * use the given fill function.
926 */
927 if (xback < 0 && yback[x] < 0)
928 goto fillit;
929 /*
930 * Compare, and use the background that is
931 * farther, unless one of them is next to us.
932 * If the background is too distant, call
933 * the fill function.
934 */
935 if ( yback[x] < 0
936 || (xback >= 0 && ABS(x-xback) <= 1)
937 || ( ABS(y-yback[x]) > 1
938 && zscan(yback[x])[x]
939 < zscan(y)[xback] ) ) {
940 if (samp > 0 && ABS(x-xback) >= samp)
941 goto fillit;
942 if (averaging) {
943 copycolor(sscan(y)[x],
944 sscan(y)[xback]);
945 wscan(y)[x] = wscan(y)[xback];
946 } else
947 copycolr(pscan(y)[x],
948 pscan(y)[xback]);
949 zscan(y)[x] = zscan(y)[xback];
950 } else {
951 if (samp > 0 && ABS(y-yback[x]) > samp)
952 goto fillit;
953 if (averaging) {
954 copycolor(sscan(y)[x],
955 sscan(yback[x])[x]);
956 wscan(y)[x] =
957 wscan(yback[x])[x];
958 } else
959 copycolr(pscan(y)[x],
960 pscan(yback[x])[x]);
961 zscan(y)[x] = zscan(yback[x])[x];
962 }
963 continue;
964 fillit:
965 (*fill)(x,y);
966 if (fill == rcalfill) { /* use it */
967 clearqueue();
968 xback = x;
969 yback[x] = y;
970 }
971 } else { /* full pixel */
972 yback[x] = -2;
973 xback = -2;
974 }
975 }
976 free((void *)yback);
977 }
978
979
980 static void
981 fillpicture( /* paint in empty pixels using fill */
982 fillfunc_t *fill
983 )
984 {
985 int x, y;
986
987 for (y = 0; y < vresolu; y++)
988 for (x = 0; x < hresolu; x++)
989 if (zscan(y)[x] <= 0)
990 (*fill)(x,y);
991 if (fill == rcalfill)
992 clearqueue();
993 }
994
995
996 static int
997 clipaft(void) /* perform aft clipping as indicated */
998 {
999 int x, y;
1000 int adjtest = (ourview.type == VT_PER) & zisnorm;
1001 double tstdist;
1002 double yzn2, vx;
1003
1004 if (ourview.vaft <= FTINY)
1005 return(0);
1006 tstdist = ourview.vaft - ourview.vfore;
1007 for (y = 0; y < vresolu; y++) {
1008 if (adjtest) { /* adjust test */
1009 yzn2 = (y+.5)/vresolu + ourview.voff - .5;
1010 yzn2 = 1. + yzn2*yzn2*ourview.vn2;
1011 tstdist = (ourview.vaft - ourview.vfore)*sqrt(yzn2);
1012 }
1013 for (x = 0; x < hresolu; x++)
1014 if (zscan(y)[x] > tstdist) {
1015 if (adjtest) {
1016 vx = (x+.5)/hresolu + ourview.hoff - .5;
1017 if (zscan(y)[x] <= (ourview.vaft -
1018 ourview.vfore) *
1019 sqrt(vx*vx*ourview.hn2 + yzn2))
1020 continue;
1021 }
1022 if (averaging)
1023 memset(sscan(y)[x], '\0', sizeof(COLOR));
1024 else
1025 memset(pscan(y)[x], '\0', sizeof(COLR));
1026 zscan(y)[x] = 0.0;
1027 }
1028 }
1029 return(1);
1030 }
1031
1032
1033 static int
1034 addblur(void) /* add to blurred picture */
1035 {
1036 COLOR cval;
1037 double d;
1038 int i;
1039
1040 if (!blurring)
1041 return(0);
1042 i = hresolu*vresolu;
1043 if (nvavg < 2)
1044 if (averaging)
1045 while (i--) {
1046 copycolor(ourbpict[i], ourspict[i]);
1047 d = 1.0/ourweigh[i];
1048 scalecolor(ourbpict[i], d);
1049 }
1050 else
1051 while (i--)
1052 colr_color(ourbpict[i], ourpict[i]);
1053 else
1054 if (averaging)
1055 while (i--) {
1056 copycolor(cval, ourspict[i]);
1057 d = 1.0/ourweigh[i];
1058 scalecolor(cval, d);
1059 addcolor(ourbpict[i], cval);
1060 }
1061 else
1062 while (i--) {
1063 colr_color(cval, ourpict[i]);
1064 addcolor(ourbpict[i], cval);
1065 }
1066 /* print view */
1067 printf("VIEW%d:", nvavg);
1068 fprintview(&ourview, stdout);
1069 putchar('\n');
1070 return(1);
1071 }
1072
1073
1074 static void
1075 writepicture(void) /* write out picture (alters buffer) */
1076 {
1077 int y;
1078 int x;
1079 double d;
1080
1081 fprtresolu(hresolu, vresolu, stdout);
1082 for (y = vresolu-1; y >= 0; y--)
1083 if (blurring) {
1084 for (x = 0; x < hresolu; x++) { /* compute avg. */
1085 d = rexpadj/nvavg;
1086 scalecolor(bscan(y)[x], d);
1087 }
1088 if (fwritescan(bscan(y), hresolu, stdout) < 0)
1089 syserror(progname);
1090 } else if (averaging) {
1091 for (x = 0; x < hresolu; x++) { /* average pixels */
1092 d = rexpadj/wscan(y)[x];
1093 scalecolor(sscan(y)[x], d);
1094 }
1095 if (fwritescan(sscan(y), hresolu, stdout) < 0)
1096 syserror(progname);
1097 } else {
1098 if (expadj)
1099 shiftcolrs(pscan(y), hresolu, expadj);
1100 if (fwritecolrs(pscan(y), hresolu, stdout) < 0)
1101 syserror(progname);
1102 }
1103 }
1104
1105
1106 static void
1107 writedistance( /* write out z file (alters buffer) */
1108 char *fname
1109 )
1110 {
1111 int donorm = normdist & !zisnorm ? 1 :
1112 (ourview.type == VT_PER) & !normdist & zisnorm ? -1 : 0;
1113 int fd;
1114 int y;
1115
1116 if ((fd = open(fname, O_WRONLY|O_CREAT|O_TRUNC, 0666)) == -1)
1117 syserror(fname);
1118 for (y = vresolu-1; y >= 0; y--) {
1119 if (donorm) {
1120 double vx, yzn2, d;
1121 int x;
1122 yzn2 = (y+.5)/vresolu + ourview.voff - .5;
1123 yzn2 = 1. + yzn2*yzn2*ourview.vn2;
1124 for (x = 0; x < hresolu; x++) {
1125 vx = (x+.5)/hresolu + ourview.hoff - .5;
1126 d = sqrt(vx*vx*ourview.hn2 + yzn2);
1127 if (donorm > 0)
1128 zscan(y)[x] *= d;
1129 else
1130 zscan(y)[x] /= d;
1131 }
1132 }
1133 if (write(fd, (char *)zscan(y), hresolu*sizeof(float))
1134 < hresolu*sizeof(float))
1135 syserror(fname);
1136 }
1137 close(fd);
1138 }
1139
1140
1141 static void
1142 backfill( /* fill pixel with background */
1143 int x,
1144 int y
1145 )
1146 {
1147 if (averaging) {
1148 copycolor(sscan(y)[x], backcolor);
1149 wscan(y)[x] = 1;
1150 } else
1151 copycolr(pscan(y)[x], backcolr);
1152 zscan(y)[x] = backz;
1153 }
1154
1155
1156 static void
1157 calstart( /* start fill calculation */
1158 char *prog,
1159 char *args
1160 )
1161 {
1162 char combuf[512];
1163 char *argv[64];
1164 int rval;
1165 char **wp, *cp;
1166
1167 if (PDesc.running) {
1168 fprintf(stderr, "%s: too many calculations\n", progname);
1169 exit(1);
1170 }
1171 strcpy(combuf, prog);
1172 strcat(combuf, args);
1173 cp = combuf;
1174 wp = argv;
1175 for ( ; ; ) {
1176 while (isspace(*cp)) /* nullify spaces */
1177 *cp++ = '\0';
1178 if (!*cp) /* all done? */
1179 break;
1180 *wp++ = cp; /* add argument to list */
1181 while (*++cp && !isspace(*cp))
1182 ;
1183 }
1184 *wp = NULL;
1185 /* start process */
1186 if ((rval = open_process(&PDesc, argv)) < 0)
1187 syserror(progname);
1188 if (rval == 0) {
1189 fprintf(stderr, "%s: command not found\n", argv[0]);
1190 exit(1);
1191 }
1192 packsiz = rval/(6*sizeof(float)) - 1;
1193 if (packsiz > PACKSIZ)
1194 packsiz = PACKSIZ;
1195 queuesiz = 0;
1196 }
1197
1198
1199 static void
1200 caldone(void) /* done with calculation */
1201 {
1202 if (!PDesc.running)
1203 return;
1204 clearqueue();
1205 close_process(&PDesc);
1206 }
1207
1208
1209 static void
1210 rcalfill( /* fill with ray-calculated pixel */
1211 int x,
1212 int y
1213 )
1214 {
1215 if (queuesiz >= packsiz) /* flush queue if needed */
1216 clearqueue();
1217 /* add position to queue */
1218 queue[queuesiz][0] = x;
1219 queue[queuesiz][1] = y;
1220 queuesiz++;
1221 }
1222
1223
1224 static void
1225 clearqueue(void) /* process queue */
1226 {
1227 FVECT orig, dir;
1228 float fbuf[6*(PACKSIZ+1)];
1229 float *fbp;
1230 int i;
1231 double vx, vy;
1232
1233 if (queuesiz == 0)
1234 return;
1235 fbp = fbuf;
1236 for (i = 0; i < queuesiz; i++) {
1237 viewray(orig, dir, &ourview,
1238 (queue[i][0]+.5)/hresolu,
1239 (queue[i][1]+.5)/vresolu);
1240 *fbp++ = orig[0]; *fbp++ = orig[1]; *fbp++ = orig[2];
1241 *fbp++ = dir[0]; *fbp++ = dir[1]; *fbp++ = dir[2];
1242 }
1243 /* mark end and get results */
1244 memset((char *)fbp, '\0', 6*sizeof(float));
1245 if (process(&PDesc, (char *)fbuf, (char *)fbuf,
1246 4*sizeof(float)*(queuesiz+1),
1247 6*sizeof(float)*(queuesiz+1)) !=
1248 4*sizeof(float)*(queuesiz+1)) {
1249 fprintf(stderr, "%s: error reading from rtrace process\n",
1250 progname);
1251 exit(1);
1252 }
1253 fbp = fbuf;
1254 for (i = 0; i < queuesiz; i++) {
1255 if (ourexp > 0 && ourexp != 1.0) {
1256 fbp[0] *= ourexp;
1257 fbp[1] *= ourexp;
1258 fbp[2] *= ourexp;
1259 }
1260 if (averaging) {
1261 setcolor(sscan(queue[i][1])[queue[i][0]],
1262 fbp[0], fbp[1], fbp[2]);
1263 wscan(queue[i][1])[queue[i][0]] = 1;
1264 } else
1265 setcolr(pscan(queue[i][1])[queue[i][0]],
1266 fbp[0], fbp[1], fbp[2]);
1267 if (zisnorm)
1268 zscan(queue[i][1])[queue[i][0]] = fbp[3];
1269 else {
1270 vx = (queue[i][0]+.5)/hresolu + ourview.hoff - .5;
1271 vy = (queue[i][1]+.5)/vresolu + ourview.voff - .5;
1272 zscan(queue[i][1])[queue[i][0]] = fbp[3] / sqrt(1. +
1273 vx*vx*ourview.hn2 + vy*vy*ourview.vn2);
1274 }
1275 fbp += 4;
1276 }
1277 queuesiz = 0;
1278 }
1279
1280
1281 static void
1282 syserror( /* report error and exit */
1283 char *s
1284 )
1285 {
1286 perror(s);
1287 exit(1);
1288 }