ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/px/pinterp.c
Revision: 2.45
Committed: Mon May 20 21:52:06 2013 UTC (10 years, 11 months ago) by greg
Content type: text/plain
Branch: MAIN
CVS Tags: rad4R2P2, rad5R0, rad5R1, rad4R2, rad4R2P1
Changes since 2.44: +2 -3 lines
Log Message:
Fixed never-noticed bug in pinterp with -n option

File Contents

# Content
1 #ifndef lint
2 static const char RCSid[] = "$Id: pinterp.c,v 2.44 2013/03/24 19:00:33 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] /= DOT(theirview.vdir, tdir);
757 pt[0] += tdir[0]*pos[2];
758 pt[1] += tdir[1]*pos[2];
759 pt[2] += tdir[2]*pos[2];
760 viewloc(pos, &ourview, pt);
761 if (pos[2] <= 0)
762 return(0);
763 }
764 if ((pos[0] < 0) | (pos[0] >= 1-FTINY) | (pos[1] < 0) | (pos[1] >= 1-FTINY))
765 return(0);
766 if (!averaging)
767 return(1);
768 /* compute pixel weight */
769 if (ourview.type == VT_PAR) {
770 d = DOT(ourview.vdir,tdir);
771 d = 1. - d*d;
772 } else {
773 VSUB(odir, pt, ourview.vp);
774 d = DOT(odir,tdir);
775 d = 1. - d*d/DOT(odir,odir);
776 }
777 if (d <= 1./MAXWT/MAXWT)
778 return(MAXWT); /* clip to maximum weight */
779 return(1./sqrt(d));
780 }
781
782
783 static int
784 getperim( /* compute overlapping image area */
785 struct bound *xl,
786 struct bound *yl,
787 float *zline,
788 int zfd
789 )
790 {
791 int step;
792 FVECT pos;
793 int x, y;
794 /* set up step size */
795 if (scanlen(&tresolu) < numscans(&tresolu))
796 step = scanlen(&tresolu)/NSTEPS;
797 else
798 step = numscans(&tresolu)/NSTEPS;
799 if (step < MINSTEP) { /* not worth cropping? */
800 yl->min = 0;
801 yl->max = numscans(&tresolu) - 1;
802 x = scanlen(&tresolu) - 1;
803 for (y = numscans(&tresolu); y--; ) {
804 xl[y].min = 0;
805 xl[y].max = x;
806 }
807 return(1);
808 }
809 yl->min = 32000; yl->max = 0; /* search for points on image */
810 for (y = step - 1; y < numscans(&tresolu); y += step) {
811 if (zfd != -1) {
812 if (lseek(zfd, (off_t)y*scanlen(&tresolu)*sizeof(float),
813 SEEK_SET) < 0)
814 syserror("lseek");
815 if (read(zfd, (char *)zline,
816 scanlen(&tresolu)*sizeof(float))
817 < scanlen(&tresolu)*sizeof(float))
818 syserror("read");
819 }
820 xl[y].min = 32000; xl[y].max = 0; /* x max */
821 for (x = scanlen(&tresolu); (x -= step) > 0; ) {
822 pix2loc(pos, &tresolu, x, y);
823 pos[2] = zline[x];
824 if (movepixel(pos) > FTINY) {
825 xl[y].max = x + step - 1;
826 xl[y].min = x - step + 1; /* x min */
827 if (xl[y].min < 0)
828 xl[y].min = 0;
829 for (x = step - 1; x < xl[y].max; x += step) {
830 pix2loc(pos, &tresolu, x, y);
831 pos[2] = zline[x];
832 if (movepixel(pos) > FTINY) {
833 xl[y].min = x - step + 1;
834 break;
835 }
836 }
837 if (y < yl->min) /* y limits */
838 yl->min = y - step + 1;
839 yl->max = y + step - 1;
840 break;
841 }
842 }
843 /* fill in between */
844 if (y < step) {
845 xl[y-1].min = xl[y].min;
846 xl[y-1].max = xl[y].max;
847 } else {
848 if (xl[y].min < xl[y-step].min)
849 xl[y-1].min = xl[y].min;
850 else
851 xl[y-1].min = xl[y-step].min;
852 if (xl[y].max > xl[y-step].max)
853 xl[y-1].max = xl[y].max;
854 else
855 xl[y-1].max = xl[y-step].max;
856 }
857 for (x = 2; x < step; x++)
858 *(xl+y-x) = *(xl+y-1);
859 }
860 if (yl->max >= numscans(&tresolu))
861 yl->max = numscans(&tresolu) - 1;
862 y -= step;
863 for (x = numscans(&tresolu) - 1; x > y; x--) /* fill bottom rows */
864 *(xl+x) = *(xl+y);
865 return(yl->max >= yl->min);
866 }
867
868
869 static void
870 backpicture( /* background fill algorithm */
871 fillfunc_t *fill,
872 int samp
873 )
874 {
875 int *yback, xback;
876 int y;
877 int x, i;
878 /* get back buffer */
879 yback = (int *)malloc(hresolu*sizeof(int));
880 if (yback == NULL)
881 syserror(progname);
882 for (x = 0; x < hresolu; x++)
883 yback[x] = -2;
884 /*
885 * Xback and yback are the pixel locations of suitable
886 * background values in each direction.
887 * A value of -2 means unassigned, and -1 means
888 * that there is no suitable background in this direction.
889 */
890 /* fill image */
891 for (y = 0; y < vresolu; y++) {
892 xback = -2;
893 for (x = 0; x < hresolu; x++)
894 if (zscan(y)[x] <= 0) { /* empty pixel */
895 /*
896 * First, find background from above or below.
897 * (farthest assigned pixel)
898 */
899 if (yback[x] == -2) {
900 for (i = y+1; i < vresolu; i++)
901 if (zscan(i)[x] > 0)
902 break;
903 if (i < vresolu
904 && (y <= 0 || zscan(y-1)[x] < zscan(i)[x]))
905 yback[x] = i;
906 else
907 yback[x] = y-1;
908 }
909 /*
910 * Next, find background from left or right.
911 */
912 if (xback == -2) {
913 for (i = x+1; i < hresolu; i++)
914 if (zscan(y)[i] > 0)
915 break;
916 if (i < hresolu
917 && (x <= 0 || zscan(y)[x-1] < zscan(y)[i]))
918 xback = i;
919 else
920 xback = x-1;
921 }
922 /*
923 * If we have no background for this pixel,
924 * use the given fill function.
925 */
926 if (xback < 0 && yback[x] < 0)
927 goto fillit;
928 /*
929 * Compare, and use the background that is
930 * farther, unless one of them is next to us.
931 * If the background is too distant, call
932 * the fill function.
933 */
934 if ( yback[x] < 0
935 || (xback >= 0 && ABS(x-xback) <= 1)
936 || ( ABS(y-yback[x]) > 1
937 && zscan(yback[x])[x]
938 < zscan(y)[xback] ) ) {
939 if (samp > 0 && ABS(x-xback) >= samp)
940 goto fillit;
941 if (averaging) {
942 copycolor(sscan(y)[x],
943 sscan(y)[xback]);
944 wscan(y)[x] = wscan(y)[xback];
945 } else
946 copycolr(pscan(y)[x],
947 pscan(y)[xback]);
948 zscan(y)[x] = zscan(y)[xback];
949 } else {
950 if (samp > 0 && ABS(y-yback[x]) > samp)
951 goto fillit;
952 if (averaging) {
953 copycolor(sscan(y)[x],
954 sscan(yback[x])[x]);
955 wscan(y)[x] =
956 wscan(yback[x])[x];
957 } else
958 copycolr(pscan(y)[x],
959 pscan(yback[x])[x]);
960 zscan(y)[x] = zscan(yback[x])[x];
961 }
962 continue;
963 fillit:
964 (*fill)(x,y);
965 if (fill == rcalfill) { /* use it */
966 clearqueue();
967 xback = x;
968 yback[x] = y;
969 }
970 } else { /* full pixel */
971 yback[x] = -2;
972 xback = -2;
973 }
974 }
975 free((void *)yback);
976 }
977
978
979 static void
980 fillpicture( /* paint in empty pixels using fill */
981 fillfunc_t *fill
982 )
983 {
984 int x, y;
985
986 for (y = 0; y < vresolu; y++)
987 for (x = 0; x < hresolu; x++)
988 if (zscan(y)[x] <= 0)
989 (*fill)(x,y);
990 if (fill == rcalfill)
991 clearqueue();
992 }
993
994
995 static int
996 clipaft(void) /* perform aft clipping as indicated */
997 {
998 int x, y;
999 int adjtest = (ourview.type == VT_PER) & zisnorm;
1000 double tstdist;
1001 double yzn2, vx;
1002
1003 if (ourview.vaft <= FTINY)
1004 return(0);
1005 tstdist = ourview.vaft - ourview.vfore;
1006 for (y = 0; y < vresolu; y++) {
1007 if (adjtest) { /* adjust test */
1008 yzn2 = (y+.5)/vresolu + ourview.voff - .5;
1009 yzn2 = 1. + yzn2*yzn2*ourview.vn2;
1010 tstdist = (ourview.vaft - ourview.vfore)*sqrt(yzn2);
1011 }
1012 for (x = 0; x < hresolu; x++)
1013 if (zscan(y)[x] > tstdist) {
1014 if (adjtest) {
1015 vx = (x+.5)/hresolu + ourview.hoff - .5;
1016 if (zscan(y)[x] <= (ourview.vaft -
1017 ourview.vfore) *
1018 sqrt(vx*vx*ourview.hn2 + yzn2))
1019 continue;
1020 }
1021 if (averaging)
1022 memset(sscan(y)[x], '\0', sizeof(COLOR));
1023 else
1024 memset(pscan(y)[x], '\0', sizeof(COLR));
1025 zscan(y)[x] = 0.0;
1026 }
1027 }
1028 return(1);
1029 }
1030
1031
1032 static int
1033 addblur(void) /* add to blurred picture */
1034 {
1035 COLOR cval;
1036 double d;
1037 int i;
1038
1039 if (!blurring)
1040 return(0);
1041 i = hresolu*vresolu;
1042 if (nvavg < 2)
1043 if (averaging)
1044 while (i--) {
1045 copycolor(ourbpict[i], ourspict[i]);
1046 d = 1.0/ourweigh[i];
1047 scalecolor(ourbpict[i], d);
1048 }
1049 else
1050 while (i--)
1051 colr_color(ourbpict[i], ourpict[i]);
1052 else
1053 if (averaging)
1054 while (i--) {
1055 copycolor(cval, ourspict[i]);
1056 d = 1.0/ourweigh[i];
1057 scalecolor(cval, d);
1058 addcolor(ourbpict[i], cval);
1059 }
1060 else
1061 while (i--) {
1062 colr_color(cval, ourpict[i]);
1063 addcolor(ourbpict[i], cval);
1064 }
1065 /* print view */
1066 printf("VIEW%d:", nvavg);
1067 fprintview(&ourview, stdout);
1068 putchar('\n');
1069 return(1);
1070 }
1071
1072
1073 static void
1074 writepicture(void) /* write out picture (alters buffer) */
1075 {
1076 int y;
1077 int x;
1078 double d;
1079
1080 fprtresolu(hresolu, vresolu, stdout);
1081 for (y = vresolu-1; y >= 0; y--)
1082 if (blurring) {
1083 for (x = 0; x < hresolu; x++) { /* compute avg. */
1084 d = rexpadj/nvavg;
1085 scalecolor(bscan(y)[x], d);
1086 }
1087 if (fwritescan(bscan(y), hresolu, stdout) < 0)
1088 syserror(progname);
1089 } else if (averaging) {
1090 for (x = 0; x < hresolu; x++) { /* average pixels */
1091 d = rexpadj/wscan(y)[x];
1092 scalecolor(sscan(y)[x], d);
1093 }
1094 if (fwritescan(sscan(y), hresolu, stdout) < 0)
1095 syserror(progname);
1096 } else {
1097 if (expadj)
1098 shiftcolrs(pscan(y), hresolu, expadj);
1099 if (fwritecolrs(pscan(y), hresolu, stdout) < 0)
1100 syserror(progname);
1101 }
1102 }
1103
1104
1105 static void
1106 writedistance( /* write out z file (alters buffer) */
1107 char *fname
1108 )
1109 {
1110 int donorm = normdist & !zisnorm ? 1 :
1111 (ourview.type == VT_PER) & !normdist & zisnorm ? -1 : 0;
1112 int fd;
1113 int y;
1114
1115 if ((fd = open(fname, O_WRONLY|O_CREAT|O_TRUNC, 0666)) == -1)
1116 syserror(fname);
1117 for (y = vresolu-1; y >= 0; y--) {
1118 if (donorm) {
1119 double vx, yzn2, d;
1120 int x;
1121 yzn2 = (y+.5)/vresolu + ourview.voff - .5;
1122 yzn2 = 1. + yzn2*yzn2*ourview.vn2;
1123 for (x = 0; x < hresolu; x++) {
1124 vx = (x+.5)/hresolu + ourview.hoff - .5;
1125 d = sqrt(vx*vx*ourview.hn2 + yzn2);
1126 if (donorm > 0)
1127 zscan(y)[x] *= d;
1128 else
1129 zscan(y)[x] /= d;
1130 }
1131 }
1132 if (write(fd, (char *)zscan(y), hresolu*sizeof(float))
1133 < hresolu*sizeof(float))
1134 syserror(fname);
1135 }
1136 close(fd);
1137 }
1138
1139
1140 static void
1141 backfill( /* fill pixel with background */
1142 int x,
1143 int y
1144 )
1145 {
1146 if (averaging) {
1147 copycolor(sscan(y)[x], backcolor);
1148 wscan(y)[x] = 1;
1149 } else
1150 copycolr(pscan(y)[x], backcolr);
1151 zscan(y)[x] = backz;
1152 }
1153
1154
1155 static void
1156 calstart( /* start fill calculation */
1157 char *prog,
1158 char *args
1159 )
1160 {
1161 char combuf[512];
1162 char *argv[64];
1163 int rval;
1164 char **wp, *cp;
1165
1166 if (PDesc.running) {
1167 fprintf(stderr, "%s: too many calculations\n", progname);
1168 exit(1);
1169 }
1170 strcpy(combuf, prog);
1171 strcat(combuf, args);
1172 cp = combuf;
1173 wp = argv;
1174 for ( ; ; ) {
1175 while (isspace(*cp)) /* nullify spaces */
1176 *cp++ = '\0';
1177 if (!*cp) /* all done? */
1178 break;
1179 *wp++ = cp; /* add argument to list */
1180 while (*++cp && !isspace(*cp))
1181 ;
1182 }
1183 *wp = NULL;
1184 /* start process */
1185 if ((rval = open_process(&PDesc, argv)) < 0)
1186 syserror(progname);
1187 if (rval == 0) {
1188 fprintf(stderr, "%s: command not found\n", argv[0]);
1189 exit(1);
1190 }
1191 packsiz = rval/(6*sizeof(float)) - 1;
1192 if (packsiz > PACKSIZ)
1193 packsiz = PACKSIZ;
1194 queuesiz = 0;
1195 }
1196
1197
1198 static void
1199 caldone(void) /* done with calculation */
1200 {
1201 if (!PDesc.running)
1202 return;
1203 clearqueue();
1204 close_process(&PDesc);
1205 }
1206
1207
1208 static void
1209 rcalfill( /* fill with ray-calculated pixel */
1210 int x,
1211 int y
1212 )
1213 {
1214 if (queuesiz >= packsiz) /* flush queue if needed */
1215 clearqueue();
1216 /* add position to queue */
1217 queue[queuesiz][0] = x;
1218 queue[queuesiz][1] = y;
1219 queuesiz++;
1220 }
1221
1222
1223 static void
1224 clearqueue(void) /* process queue */
1225 {
1226 FVECT orig, dir;
1227 float fbuf[6*(PACKSIZ+1)];
1228 float *fbp;
1229 int i;
1230 double vx, vy;
1231
1232 if (queuesiz == 0)
1233 return;
1234 fbp = fbuf;
1235 for (i = 0; i < queuesiz; i++) {
1236 viewray(orig, dir, &ourview,
1237 (queue[i][0]+.5)/hresolu,
1238 (queue[i][1]+.5)/vresolu);
1239 *fbp++ = orig[0]; *fbp++ = orig[1]; *fbp++ = orig[2];
1240 *fbp++ = dir[0]; *fbp++ = dir[1]; *fbp++ = dir[2];
1241 }
1242 /* mark end and get results */
1243 memset((char *)fbp, '\0', 6*sizeof(float));
1244 if (process(&PDesc, (char *)fbuf, (char *)fbuf,
1245 4*sizeof(float)*(queuesiz+1),
1246 6*sizeof(float)*(queuesiz+1)) !=
1247 4*sizeof(float)*(queuesiz+1)) {
1248 fprintf(stderr, "%s: error reading from rtrace process\n",
1249 progname);
1250 exit(1);
1251 }
1252 fbp = fbuf;
1253 for (i = 0; i < queuesiz; i++) {
1254 if (ourexp > 0 && ourexp != 1.0) {
1255 fbp[0] *= ourexp;
1256 fbp[1] *= ourexp;
1257 fbp[2] *= ourexp;
1258 }
1259 if (averaging) {
1260 setcolor(sscan(queue[i][1])[queue[i][0]],
1261 fbp[0], fbp[1], fbp[2]);
1262 wscan(queue[i][1])[queue[i][0]] = 1;
1263 } else
1264 setcolr(pscan(queue[i][1])[queue[i][0]],
1265 fbp[0], fbp[1], fbp[2]);
1266 if (zisnorm)
1267 zscan(queue[i][1])[queue[i][0]] = fbp[3];
1268 else {
1269 vx = (queue[i][0]+.5)/hresolu + ourview.hoff - .5;
1270 vy = (queue[i][1]+.5)/vresolu + ourview.voff - .5;
1271 zscan(queue[i][1])[queue[i][0]] = fbp[3] / sqrt(1. +
1272 vx*vx*ourview.hn2 + vy*vy*ourview.vn2);
1273 }
1274 fbp += 4;
1275 }
1276 queuesiz = 0;
1277 }
1278
1279
1280 static void
1281 syserror( /* report error and exit */
1282 char *s
1283 )
1284 {
1285 perror(s);
1286 exit(1);
1287 }