ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/util/ranimove1.c
Revision: 3.19
Committed: Sun Sep 30 19:18:41 2012 UTC (11 years, 7 months ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 3.18: +32 -40 lines
Log Message:
Minor code clean-ups and efficiency improvements

File Contents

# User Rev Content
1 greg 3.1 #ifndef lint
2 greg 3.19 static const char RCSid[] = "$Id: ranimove1.c,v 3.18 2012/09/28 22:20:49 greg Exp $";
3 greg 3.1 #endif
4     /*
5     * ranimove1.c
6     *
7 greg 3.11 * Basic frame rendering routines for ranimove(1).
8 greg 3.1 *
9     * Created by Gregory Ward on Wed Jan 08 2003.
10     */
11    
12 greg 3.2 #include "copyright.h"
13 greg 3.1
14 schorsch 3.4 #include <string.h>
15    
16 schorsch 3.3 #include "platform.h"
17 greg 3.1 #include "ranimove.h"
18     #include "otypes.h"
19 schorsch 3.8 #include "source.h"
20 greg 3.1 #include "random.h"
21    
22     double acctab[256]; /* accuracy value table */
23    
24     int hres, vres; /* frame resolution (fcur) */
25     double pixaspect; /* pixel aspect ratio */
26    
27     VIEW vw; /* view for this frame */
28     COLOR *cbuffer; /* color at each pixel */
29     float *zbuffer; /* depth at each pixel */
30     OBJECT *obuffer; /* object id at each pixel */
31     short *xmbuffer; /* x motion at each pixel */
32     short *ymbuffer; /* y motion at each pixel */
33 greg 3.17 uby8 *abuffer; /* accuracy at each pixel */
34     uby8 *sbuffer; /* sample count per pixel */
35 greg 3.1
36     VIEW vwprev; /* last frame's view */
37     COLOR *cprev; /* last frame colors */
38     float *zprev; /* last frame depth */
39     OBJECT *oprev; /* last frame objects */
40 greg 3.17 uby8 *aprev; /* last frame accuracy */
41 greg 3.1
42     float *cerrmap; /* conspicuous error map */
43     COLOR *val2map; /* value-squared map for variance */
44    
45     double frm_stop; /* when to stop rendering this frame */
46    
47     double hlsmax; /* maximum high-level saliency this frame */
48    
49    
50 schorsch 3.8 static void next_frame(void);
51     static int sample_here(int x, int y);
52     static int offset_cmp(const void *p1, const void *p2);
53     static void setmotion(int n, FVECT wpos);
54     static void init_frame_sample(void);
55    
56    
57 greg 3.14 #if 0
58 greg 3.19 void
59 schorsch 3.8 write_map( /* write out float map (debugging) */
60     float *mp,
61     char *fn
62     )
63 greg 3.1 {
64     FILE *fp = fopen(fn, "w");
65     COLOR scanbuf[2048];
66     int x, y;
67    
68     if (fp == NULL)
69     return;
70     newheader("RADIANCE", fp);
71     fputformat(COLRFMT, fp);
72     fputc('\n', fp); /* end header */
73     fprtresolu(hres, vres, fp);
74     for (y = vres; y--; ) { /* write scanlines */
75     float *bp = mp + (y+1)*hres - 1;
76     for (x = hres; x--; bp--)
77     setcolor(scanbuf[x], *bp, *bp, *bp);
78     if (fwritescan(scanbuf, hres, fp) < 0)
79     break;
80     }
81     fclose(fp);
82     }
83 greg 3.14 #endif
84 greg 3.1
85    
86     static void
87 schorsch 3.8 next_frame(void) /* prepare next frame buffer */
88 greg 3.1 {
89     VIEW *fv;
90     char *err;
91     /* get previous view */
92     if (vw.type != 0)
93 schorsch 3.5 vwprev = vw;
94 greg 3.1 else if (fcur > 1 && (fv = getview(fcur-1)) != NULL) {
95 schorsch 3.5 vwprev = *fv;
96 greg 3.1 if (setview(&vwprev) != NULL)
97     vwprev.type = 0;
98     }
99     /* get current view */
100     if ((fv = getview(fcur)) == NULL) {
101     sprintf(errmsg, "cannot get view for frame %d", fcur);
102     error(USER, errmsg);
103     }
104 schorsch 3.5 vw = *fv;
105 greg 3.1 if ((err = setview(&vw)) != NULL) {
106     sprintf(errmsg, "view error at frame %d: %s", fcur, err);
107     error(USER, errmsg);
108     }
109     if (cbuffer == NULL) {
110 greg 3.16 int n;
111 greg 3.1 /* compute resolution and allocate */
112     switch (sscanf(vval(RESOLUTION), "%d %d %lf",
113     &hres, &vres, &pixaspect)) {
114     case 1:
115     vres = hres;
116     /* fall through */
117     case 2:
118     pixaspect = 1.;
119     /* fall through */
120     case 3:
121 schorsch 3.6 if ((hres > 0) & (vres > 0))
122 greg 3.1 break;
123     /* fall through */
124     default:
125     sprintf(errmsg, "bad %s value", vnam(RESOLUTION));
126     error(USER, errmsg);
127     }
128     normaspect(viewaspect(&vw), &pixaspect, &hres, &vres);
129     cbuffer = (COLOR *)malloc(sizeof(COLOR)*hres*vres);
130     zbuffer = (float *)malloc(sizeof(float)*hres*vres);
131     obuffer = (OBJECT *)malloc(sizeof(OBJECT)*hres*vres);
132     xmbuffer = (short *)malloc(sizeof(short)*hres*vres);
133     ymbuffer = (short *)malloc(sizeof(short)*hres*vres);
134 greg 3.17 abuffer = (uby8 *)calloc(hres*vres, sizeof(uby8));
135     sbuffer = (uby8 *)calloc(hres*vres, sizeof(uby8));
136 greg 3.1 cprev = (COLOR *)malloc(sizeof(COLOR)*hres*vres);
137     zprev = (float *)malloc(sizeof(float)*hres*vres);
138     oprev = (OBJECT *)malloc(sizeof(OBJECT)*hres*vres);
139 greg 3.17 aprev = (uby8 *)malloc(sizeof(uby8)*hres*vres);
140 schorsch 3.6 if ((cbuffer==NULL) | (zbuffer==NULL) | (obuffer==NULL) |
141     (xmbuffer==NULL) | (ymbuffer==NULL) |
142     (abuffer==NULL) | (sbuffer==NULL) |
143     (cprev==NULL) | (zprev == NULL) |
144     (oprev==NULL) | (aprev==NULL))
145 greg 3.1 error(SYSTEM, "out of memory in init_frame");
146 greg 3.16 for (n = hres*vres; n--; ) {
147     zprev[n] = -1.f;
148 greg 3.18 xmbuffer[n] = ymbuffer[n] = MO_UNK;
149 greg 3.16 oprev[n] = OVOID;
150     }
151 greg 3.1 frm_stop = getTime() + rtperfrm;
152     } else {
153     COLOR *cp; /* else just swap buffers */
154     float *fp;
155     OBJECT *op;
156 greg 3.17 uby8 *bp;
157 greg 3.1 cp = cprev; cprev = cbuffer; cbuffer = cp;
158     fp = zprev; zprev = zbuffer; zbuffer = fp;
159     op = oprev; oprev = obuffer; obuffer = op;
160     bp = aprev; aprev = abuffer; abuffer = bp;
161 greg 3.17 memset(abuffer, '\0', sizeof(uby8)*hres*vres);
162     memset(sbuffer, '\0', sizeof(uby8)*hres*vres);
163 greg 3.1 frm_stop += rtperfrm;
164     }
165     cerrmap = NULL;
166     val2map = NULL;
167     }
168    
169    
170     #define SAMPDIST 3 /* Maximum distance to neighbor sample */
171     #define SAMPDIST2 (SAMPDIST*SAMPDIST)
172    
173    
174     static int
175 schorsch 3.8 sample_here( /* 4x4 quincunx sample at this pixel? */
176 greg 3.18 int x,
177     int y
178 schorsch 3.8 )
179 greg 3.1 {
180     if (y & 0x1) /* every other row has samples */
181     return(0);
182     if (y & 0x3) /* every fourth row is offset */
183     x += 2;
184     return((x & 0x3) == 0); /* every fourth column is sampled */
185     }
186    
187    
188 greg 3.19 void
189 schorsch 3.8 sample_pos( /* compute jittered sample position */
190     double hv[2],
191     int x,
192     int y,
193     int sn
194     )
195 greg 3.1 {
196     int hl[2];
197    
198     hl[0] = x; hl[1] = y;
199     multisamp(hv, 2, urand(ilhash(hl,2) + sn));
200     hv[0] = ((double)x + hv[0]) / (double)hres;
201     hv[1] = ((double)y + hv[1]) / (double)vres;
202     }
203    
204    
205 greg 3.19 double
206 schorsch 3.8 sample_wt( /* compute interpolant sample weight */
207     int xo,
208     int yo
209     )
210 greg 3.1 {
211     static double etab[400];
212 schorsch 3.7 /* we can't use the name rad2 here, for some reason Visual C
213     thinks that is a constant (compiler bug?) */
214     int rad_2 = xo*xo + yo*yo;
215 greg 3.1 int i;
216    
217     if (etab[0] <= FTINY) /* initialize exponent table */
218     for (i = 400; i--; )
219     etab[i] = exp(-0.1*i);
220    
221     /* look up Gaussian */
222 schorsch 3.7 i = (int)((10.*3./(double)SAMPDIST2)*rad_2 + .5);
223 greg 3.1 if (i >= 400)
224     return(0.0);
225     return(etab[i]);
226     }
227    
228    
229     static int
230 schorsch 3.8 offset_cmp( /* compare offset distances */
231     const void *p1,
232     const void *p2
233     )
234 greg 3.1 {
235     return(*(const int *)p1 - *(const int *)p2);
236     }
237    
238    
239 greg 3.19 int
240 schorsch 3.8 getclosest( /* get nc closest neighbors on same object */
241     int *iarr,
242     int nc,
243     int x,
244     int y
245     )
246 greg 3.1 {
247     #define NSCHECK ((2*SAMPDIST+1)*(2*SAMPDIST+1))
248     static int hro, vro;
249     static int ioffs[NSCHECK];
250     OBJECT myobj;
251     int i0, nf;
252 greg 3.18 int i, j;
253 greg 3.1 /* get our object number */
254     myobj = obuffer[fndx(x, y)];
255     /* special case for borders */
256 schorsch 3.6 if ((x < SAMPDIST) | (x >= hres-SAMPDIST) |
257     (y < SAMPDIST) | (y >= vres-SAMPDIST)) {
258 greg 3.1 int tndx[NSCHECK][2];
259     nf = 0;
260     for (j = y - SAMPDIST; j <= y + SAMPDIST; j++) {
261     if (j >= vres) break;
262     if (j < 0) j = 0;
263     for (i = x - SAMPDIST; i <= x + SAMPDIST; i++) {
264     if (i >= hres) break;
265     if (i < 0) i = 0;
266     i0 = fndx(i, j);
267     if (!sbuffer[i0])
268     continue;
269 schorsch 3.6 if ((myobj != OVOID) & (obuffer[i0] != myobj))
270 greg 3.1 continue;
271     tndx[nf][0] = (i-x)*(i-x) + (j-y)*(j-y);
272     tndx[nf][1] = i0;
273     nf++;
274     }
275     }
276     qsort((void *)tndx, nf, 2*sizeof(int), offset_cmp);
277     if (nf > nc)
278     nf = nc;
279     for (i = nf; i--; )
280     iarr[i] = tndx[i][1];
281     return(nf);
282     }
283     /* initialize offset array */
284 schorsch 3.6 if ((hres != hro) | (vres != vro)) {
285 greg 3.1 int toffs[NSCHECK][2];
286     i0 = fndx(SAMPDIST, SAMPDIST);
287     nf = 0;
288     for (i = 0; i <= 2*SAMPDIST; i++)
289     for (j = 0; j <= 2*SAMPDIST; j++) {
290     toffs[nf][0] = (i-SAMPDIST)*(i-SAMPDIST) +
291     (j-SAMPDIST)*(j-SAMPDIST);
292     toffs[nf][1] = fndx(i, j) - i0;
293     nf++;
294     }
295     qsort((void *)toffs, nf, 2*sizeof(int), offset_cmp);
296     for (i = NSCHECK; i--; )
297     ioffs[i] = toffs[i][1];
298     hro = hres;
299     vro = vres;
300     }
301     /* find up to nc neighbors */
302     i0 = fndx(x, y);
303 schorsch 3.6 for (j = 0, nf = 0; (j < NSCHECK) & (nf < nc); j++) {
304 greg 3.1 i = i0 + ioffs[j];
305 schorsch 3.6 if (sbuffer[i] && (myobj == OVOID) | (obuffer[i] == myobj))
306 greg 3.1 iarr[nf++] = i;
307     }
308     /* return number found */
309     return(nf);
310     #undef NSCHECK
311     }
312    
313    
314     static void
315 schorsch 3.8 setmotion( /* compute motion vector for this pixel */
316 greg 3.18 int n,
317 schorsch 3.8 FVECT wpos
318     )
319 greg 3.1 {
320     FVECT ovp;
321     int moi;
322     int xp, yp;
323     /* ID object and update maximum HLS */
324     moi = getmove(obuffer[n]);
325     if (moi >= 0 && obj_move[moi].cprio > hlsmax)
326     hlsmax = obj_move[moi].cprio;
327     if (vwprev.type == 0) /* return leaves MO_UNK */
328     return;
329     if (moi >= 0) { /* move object point back */
330     multp3(ovp, wpos, obj_move[moi].bxfm);
331     wpos = ovp;
332     }
333     viewloc(ovp, &vwprev, wpos);
334     if (ovp[2] <= FTINY)
335     return;
336     xp = (int)(ovp[0]*hres);
337     yp = (int)(ovp[1]*vres);
338     xmbuffer[n] = xp - (n % hres);
339     ymbuffer[n] = yp - (n / hres);
340 schorsch 3.6 if ((xp < 0) | (xp >= hres))
341 greg 3.1 return;
342 schorsch 3.6 if ((yp < 0) | (yp >= vres))
343 greg 3.1 return;
344     n = fndx(xp, yp);
345 schorsch 3.6 if ((zprev[n] < 0.97*ovp[2]) | (zprev[n] > 1.03*ovp[2]))
346 greg 3.1 oprev[n] = OVOID; /* assume it's a bad match */
347     }
348    
349    
350     static void
351 schorsch 3.8 init_frame_sample(void) /* sample our initial frame */
352 greg 3.1 {
353     RAY ir;
354     int x, y;
355 greg 3.18 int n;
356 greg 3.1
357     if (!silent) {
358     printf("\tComputing initial samples...");
359     fflush(stdout);
360     }
361     hlsmax = CSF_SMN;
362     for (y = vres; y--; )
363     for (x = hres; x--; ) {
364     double hv[2];
365     n = fndx(x, y);
366     xmbuffer[n] = MO_UNK;
367     ymbuffer[n] = MO_UNK;
368     sample_pos(hv, x, y, 0);
369     ir.rmax = viewray(ir.rorg, ir.rdir, &vw, hv[0], hv[1]);
370     if (ir.rmax < -FTINY) {
371     setcolor(cbuffer[n], 0., 0., 0.);
372     zbuffer[n] = FHUGE;
373     obuffer[n] = OVOID;
374     abuffer[n] = ADISTANT;
375     continue;
376     }
377     if (!sample_here(x, y)) { /* just cast */
378 greg 3.10 rayorigin(&ir, PRIMARY, NULL, NULL);
379 greg 3.1 if (!localhit(&ir, &thescene)) {
380 greg 3.16 if (ir.ro != &Aftplane && sourcehit(&ir)) {
381     rayshade(&ir, ir.ro->omod);
382     rayparticipate(&ir);
383     }
384 greg 3.1 copycolor(cbuffer[n], ir.rcol);
385     zbuffer[n] = ir.rot;
386     obuffer[n] = ir.robj;
387     abuffer[n] = ADISTANT;
388     sbuffer[n] = 1;
389     } else {
390     zbuffer[n] = ir.rot;
391     obuffer[n] = ir.robj;
392     setmotion(n, ir.rop);
393     }
394     continue;
395     }
396     if (nprocs > 1) { /* get sample */
397     int rval;
398 greg 3.10 rayorigin(&ir, PRIMARY, NULL, NULL);
399 greg 3.1 ir.rno = n;
400     rval = ray_pqueue(&ir);
401     if (!rval)
402     continue;
403     if (rval < 0)
404     quit(1);
405     n = ir.rno;
406     } else
407     ray_trace(&ir);
408     copycolor(cbuffer[n], ir.rcol);
409     zbuffer[n] = ir.rot;
410     obuffer[n] = ir.robj;
411     sbuffer[n] = 1;
412 greg 3.16 if (ir.rot >= 0.99*FHUGE)
413 greg 3.1 abuffer[n] = ADISTANT;
414     else {
415     abuffer[n] = ALOWQ;
416     setmotion(n, ir.rop);
417     }
418     }
419     if (nprocs > 1) /* get stragglers */
420     while (ray_presult(&ir, 0)) {
421     n = ir.rno;
422     copycolor(cbuffer[n], ir.rcol);
423     zbuffer[n] = ir.rot;
424     obuffer[n] = ir.robj;
425     sbuffer[n] = 1;
426     if (ir.rot >= FHUGE)
427     abuffer[n] = ADISTANT;
428     else {
429     abuffer[n] = ALOWQ;
430     setmotion(n, ir.rop);
431     }
432     }
433     /* ambiguate object boundaries */
434     for (y = vres-1; y--; )
435     for (x = hres-1; x--; ) {
436     OBJECT obj;
437     n = fndx(x, y);
438     if ((obj = obuffer[n]) == OVOID)
439     continue;
440 schorsch 3.6 if ((obuffer[n+1] != OVOID) & (obuffer[n+1] != obj)) {
441 greg 3.1 obuffer[n] = OVOID;
442     obuffer[n+1] = OVOID;
443     }
444 schorsch 3.6 if ((obuffer[n+hres] != OVOID) & (obuffer[n+hres] != obj)) {
445 greg 3.1 obuffer[n] = OVOID;
446     obuffer[n+hres] = OVOID;
447     }
448     }
449    
450     if (!silent)
451     printf("done\n");
452     }
453    
454    
455 greg 3.19 int
456 schorsch 3.8 getambcolor( /* get ambient color for object if we can */
457     COLOR clr,
458     int obj
459     )
460 greg 3.1 {
461 greg 3.18 OBJREC *op;
462 greg 3.1
463     if (obj == OVOID)
464     return(0);
465 greg 3.16 op = objptr(obj); /* search for material */
466     if (op->omod == OVOID)
467     return(0);
468     op = findmaterial(objptr(op->omod));
469     if (op == NULL)
470 greg 3.1 return(0);
471     /*
472     * Since this routine is called to compute the difference
473     * from rendering with and without interreflections,
474     * we don't want to return colors for materials that are
475     * explicitly excluded from the HQ ambient calculation.
476     */
477     if (hirendparams.ambincl >= 0) {
478     int i;
479     char *lv;
480     for (i = 0; (lv = rpambmod(&hirendparams,i)) != NULL; i++)
481     if (lv[0] == op->oname[0] &&
482     !strcmp(lv+1, op->oname+1))
483     break;
484 greg 3.16 if ((lv != NULL) ^ hirendparams.ambincl)
485 greg 3.1 return(0);
486     }
487     switch (op->otype) {
488     case MAT_PLASTIC:
489     case MAT_METAL:
490     case MAT_PLASTIC2:
491     case MAT_METAL2:
492     case MAT_PFUNC:
493     case MAT_MFUNC:
494     case MAT_PDATA:
495     case MAT_MDATA:
496     case MAT_TRANS:
497     case MAT_TRANS2:
498     case MAT_TFUNC:
499     case MAT_TDATA:
500     if (op->oargs.nfargs < 3)
501     return(0);
502     setcolor(clr, op->oargs.farg[0], op->oargs.farg[1],
503     op->oargs.farg[2]);
504     return(1);
505     case MAT_BRTDF:
506     if (op->oargs.nfargs < 6)
507     return(0);
508     setcolor(clr, op->oargs.farg[0]+op->oargs.farg[3],
509     op->oargs.farg[1]+op->oargs.farg[4],
510     op->oargs.farg[2]+op->oargs.farg[5]);
511     scalecolor(clr, 0.5);
512     return(1);
513     case MAT_LIGHT:
514     case MAT_GLOW:
515     case MAT_ILLUM:
516     setcolor(clr, 0., 0., 0.);
517     return(1);
518     }
519     return(0);
520     }
521    
522    
523 greg 3.19 double
524 schorsch 3.8 estimaterr( /* estimate relative error from samples */
525     COLOR cs,
526     COLOR cs2,
527     int ns,
528     int ns0
529     )
530 greg 3.1 {
531     double d, d2, brt;
532    
533     if (ns <= 1 || (brt = bright(cs)/ns) < 1e-14)
534     return(1.0);
535     /* use largest of RGB std. dev. */
536     d2 = colval(cs2,RED) - colval(cs,RED)*colval(cs,RED)/ns;
537     d = colval(cs2,GRN) - colval(cs,GRN)*colval(cs,GRN)/ns;
538     if (d > d2) d2 = d;
539     d = colval(cs2,BLU) - colval(cs,BLU)*colval(cs,BLU)/ns;
540     if (d > d2) d2 = d;
541     /* use s.d. if <= 1 central sample */
542     if (ns0 <= 1)
543     return(sqrt(d2/(ns-1))/brt);
544     /* use s.d./sqrt(ns0) otherwise */
545     return(sqrt(d2/((ns-1)*ns0))/brt);
546     }
547    
548    
549 greg 3.19 double
550 schorsch 3.8 comperr( /* estimate relative error in neighborhood */
551     int *neigh,
552     int nc,
553     int ns0
554     )
555 greg 3.1 {
556     COLOR csum, csum2;
557     COLOR ctmp;
558     int i;
559     int ns;
560 greg 3.18 int n;
561 greg 3.1 /* add together samples */
562     setcolor(csum, 0., 0., 0.);
563     setcolor(csum2, 0., 0., 0.);
564 schorsch 3.6 for (i = 0, ns = 0; (i < nc) & (ns < NSAMPOK); i++) {
565 greg 3.1 n = neigh[i];
566     addcolor(csum, cbuffer[n]);
567     if (val2map != NULL) {
568     addcolor(csum2, val2map[n]);
569     ns += sbuffer[n];
570     continue;
571     }
572     if (sbuffer[n] != 1)
573     error(CONSISTENCY, "bad count in comperr");
574     setcolor(ctmp,
575     colval(cbuffer[n],RED)*colval(cbuffer[n],RED),
576     colval(cbuffer[n],GRN)*colval(cbuffer[n],GRN),
577     colval(cbuffer[n],BLU)*colval(cbuffer[n],BLU));
578     addcolor(csum2, ctmp);
579     ns++;
580     }
581     return(estimaterr(csum, csum2, ns, ns0));
582     }
583    
584    
585 greg 3.19 void
586 schorsch 3.8 comp_frame_error(void) /* initialize frame error values */
587 greg 3.1 {
588 greg 3.17 uby8 *edone = NULL;
589 greg 3.1 COLOR objamb;
590     double eest;
591     int neigh[NSAMPOK];
592     int nc;
593 schorsch 3.8 int x, y, i;
594 greg 3.18 int n;
595 greg 3.1
596     if (!silent) {
597     printf("\tComputing error map\n");
598     fflush(stdout);
599     }
600     if (acctab[0] <= FTINY) /* initialize accuracy table */
601     for (i = 256; i--; )
602     acctab[i] = errorf(i);
603     /* estimate sample error */
604     if (!curparams->ambounce && hirendparams.ambounce) {
605     /*
606     * Our error estimate for the initial value is based
607     * on the assumption that most of it comes from the
608     * lack of an interreflection calculation. The relative
609     * error should be less than the ambient value divided
610     * by the returned ray value -- we take half of this.
611     */
612 greg 3.17 edone = (uby8 *)calloc(hres*vres, sizeof(uby8));
613 greg 3.19 for (n = hres*vres; n--; ) {
614 schorsch 3.6 if ((abuffer[n] != ALOWQ) | (obuffer[n] == OVOID))
615 greg 3.1 continue;
616     if (!getambcolor(objamb, obuffer[n]))
617     continue;
618     multcolor(objamb, ambval);
619     if ((eest = bright(cbuffer[n])) <= FTINY)
620     continue;
621     eest = bright(objamb) / eest;
622     if (eest > 1.) /* should we report this? */
623     continue;
624     eest *= 0.50; /* use 50% ambient error */
625     i = errori(eest);
626     if (i < AMIN) i = AMIN;
627     else if (i >= ADISTANT/2) i = ADISTANT/2-1;
628     abuffer[n] = i;
629     edone[n] = 1;
630 greg 3.19 }
631 greg 3.1 }
632     /* final statistical estimate */
633     for (y = vres; y--; )
634     for (x = hres; x--; ) {
635     n = fndx(x, y);
636     if (abuffer[n] == ADISTANT)
637     continue; /* don't update these */
638     if (edone != NULL && edone[n])
639     continue; /* already done this */
640     if (sbuffer[n] >= 255) {
641     abuffer[n] = ADISTANT;
642     continue; /* can't take any more */
643     }
644     nc = getclosest(neigh, NSAMPOK, x, y);
645     if (nc <= 0) {
646     abuffer[n] = ANOVAL;
647     continue; /* no clue what to do for him */
648     }
649     i = errori(comperr(neigh, nc, sbuffer[n]));
650     if (i < AMIN) i = AMIN;
651     else if (i >= ADISTANT) i = ADISTANT-1;
652     abuffer[n] = i;
653     /* can't be better than closest */
654     if (i < abuffer[neigh[0]] && abuffer[neigh[0]] >= AMIN)
655     abuffer[n] = abuffer[neigh[0]];
656     }
657     if (edone != NULL)
658     free((void *)edone);
659     }
660    
661    
662 greg 3.19 void
663 schorsch 3.8 init_frame(void) /* render base (low quality) frame */
664 greg 3.1 {
665     int restart;
666     /* allocate/swap buffers */
667     next_frame();
668     /* check rendering status */
669     restart = (!nobjects || vdef(MOVE));
670     if (!restart && curparams != &lorendparams && nprocs > 1)
671     restart = -1;
672     /* post low quality parameters */
673     if (curparams != &lorendparams)
674     ray_restore(curparams = &lorendparams);
675     if (restart > 0) { /* load new octree */
676     char *oct = getoctspec(fcur);
677     if (oct == NULL) {
678     sprintf(errmsg, "cannot get scene for frame %d", fcur);
679     error(USER, errmsg);
680     }
681     if (!silent) {
682     printf("\tLoading octree...");
683     fflush(stdout);
684     }
685     if (nprocs > 1)
686     ray_pinit(oct, nprocs);
687     else
688     ray_init(oct);
689     } else if (restart < 0) { /* update children */
690     if (!silent) {
691     printf("\tRestarting %d processes...", nprocs);
692     fflush(stdout);
693     }
694     ray_pclose(0);
695     ray_popen(nprocs);
696     }
697     if (restart && !silent)
698     printf("done\n");
699     /* sample frame buffer */
700     init_frame_sample();
701     /* initialize frame error */
702     comp_frame_error();
703 greg 3.14 #if 0
704 greg 3.1 {
705     float *ebuf = (float *)malloc(sizeof(float)*hres*vres);
706     char fnm[256];
707 greg 3.18 int n;
708 greg 3.1 for (n = hres*vres; n--; )
709     ebuf[n] = acctab[abuffer[n]];
710     sprintf(fnm, vval(BASENAME), fcur);
711     strcat(fnm, "_inerr.pic");
712     write_map(ebuf, fnm);
713     free((void *)ebuf);
714     }
715 greg 3.14 #endif
716 greg 3.1 }
717    
718    
719 greg 3.19 void
720 schorsch 3.8 filter_frame(void) /* interpolation, motion-blur, and exposure */
721 greg 3.1 {
722 greg 3.19 const double expval = expspec_val(getexp(fcur));
723     int x, y;
724     int neigh[NPINTERP];
725     int nc;
726     COLOR cval;
727     double w, wsum;
728     int n;
729 greg 3.1
730     #if 0
731 schorsch 3.8 /* XXX TEMPORARY!! */
732     conspicuity();
733     write_map(cerrmap, "outcmap.pic");
734     {
735     float *ebuf = (float *)malloc(sizeof(float)*hres*vres);
736     for (n = hres*vres; n--; )
737     ebuf[n] = acctab[abuffer[n]];
738     write_map(ebuf, "outerr.pic");
739     free((void *)ebuf);
740     }
741 greg 3.1 #endif
742    
743     if (!silent) {
744     printf("\tFiltering frame\n");
745     fflush(stdout);
746     }
747 greg 3.9 /* normalize samples */
748 greg 3.19 for (n = hres*vres; n--; ) {
749 greg 3.9 if (sbuffer[n] <= 1)
750     continue;
751     w = 1.0/(double)sbuffer[n];
752     scalecolor(cbuffer[n], w);
753 greg 3.19 }
754 greg 3.9 /* interpolate samples */
755     for (y = vres; y--; )
756     for (x = hres; x--; ) {
757     n = fndx(x, y);
758     if (sbuffer[n])
759     continue;
760     nc = getclosest(neigh, NPINTERP, x, y);
761     setcolor(cbuffer[n], 0., 0., 0.);
762 greg 3.12 if (nc <= 0) { /* no acceptable neighbors */
763     if (y < vres-1)
764     nc = fndx(x, y+1);
765     else if (x < hres-1)
766     nc = fndx(x+1, y);
767     else
768     continue;
769     copycolor(cbuffer[n], cbuffer[nc]);
770     continue;
771     }
772 greg 3.9 wsum = 0.;
773     while (nc-- > 0) {
774     copycolor(cval, cbuffer[neigh[nc]]);
775     w = sample_wt((neigh[nc]%hres) - x,
776     (neigh[nc]/hres) - y);
777     scalecolor(cval, w);
778     addcolor(cbuffer[n], cval);
779     wsum += w;
780     }
781 greg 3.12 w = 1.0/wsum;
782     scalecolor(cbuffer[n], w);
783 greg 3.9 }
784     /* motion blur if requested */
785 greg 3.1 if (mblur > .02) {
786 greg 3.9 int xs, ys, xl, yl;
787     int rise, run;
788     long rise2, run2;
789     int n2;
790     int cnt;
791     /* sum in motion streaks */
792     memset(outbuffer, '\0', sizeof(COLOR)*hres*vres);
793     memset(wbuffer, '\0', sizeof(float)*hres*vres);
794     for (y = vres; y--; )
795     for (x = hres; x--; ) {
796 greg 3.1 n = fndx(x, y);
797     if (xmbuffer[n] == MO_UNK) {
798     run = rise = 0;
799     } else {
800     run = (int)(mblur*xmbuffer[n]);
801     rise = (int)(mblur*ymbuffer[n]);
802     }
803     if (!(run | rise)) {
804     addcolor(outbuffer[n], cbuffer[n]);
805     wbuffer[n] += 1.;
806     continue;
807     }
808     xl = x - run/4;
809     yl = y - rise/4;
810     if (run < 0) { xs = -1; run = -run; }
811     else xs = 1;
812     if (rise < 0) { ys = -1; rise = -rise; }
813     else ys = 1;
814     rise2 = run2 = 0L;
815     if (rise > run) {
816 greg 3.9 cnt = rise + 1;
817     w = 1./cnt;
818     copycolor(cval, cbuffer[n]);
819     scalecolor(cval, w);
820     while (cnt)
821     if (rise2 >= run2) {
822     if ((xl >= 0) & (xl < hres) &
823 schorsch 3.8 (yl >= 0) & (yl < vres)) {
824 greg 3.9 n2 = fndx(xl, yl);
825     addcolor(outbuffer[n2],
826     cval);
827     wbuffer[n2] += w;
828     }
829     yl += ys;
830     run2 += run;
831     cnt--;
832     } else {
833     xl += xs;
834     rise2 += rise;
835     }
836 greg 3.1 } else {
837     cnt = run + 1;
838     w = 1./cnt;
839     copycolor(cval, cbuffer[n]);
840     scalecolor(cval, w);
841 greg 3.9 while (cnt)
842     if (run2 >= rise2) {
843     if ((xl >= 0) & (xl < hres) &
844 schorsch 3.8 (yl >= 0) & (yl < vres)) {
845 greg 3.9 n2 = fndx(xl, yl);
846     addcolor(outbuffer[n2],
847     cval);
848     wbuffer[n2] += w;
849     }
850     xl += xs;
851     rise2 += rise;
852     cnt--;
853     } else {
854     yl += ys;
855     run2 += run;
856 greg 3.1 }
857 schorsch 3.8 }
858 greg 3.9 }
859     /* compute final results */
860 greg 3.19 for (n = hres*vres; n--; ) {
861 greg 3.9 if (wbuffer[n] <= FTINY)
862     continue;
863 greg 3.19 w = expval/wbuffer[n];
864 greg 3.9 scalecolor(outbuffer[n], w);
865 greg 3.19 }
866     } else { /* no blur -- just exposure */
867     memcpy(outbuffer, cbuffer, sizeof(COLOR)*hres*vres);
868     for (n = ((expval < 0.99) | (expval > 1.01))*hres*vres; n--; )
869     scalecolor(outbuffer[n], expval);
870     }
871 schorsch 3.8 /*
872     for (n = hres*vres; n--; )
873 greg 3.9 if (!sbuffer[n])
874     setcolor(outbuffer[n], 0., 0., 0.);
875 schorsch 3.8 */
876 greg 3.9 #if 0
877 schorsch 3.8 {
878     float *sbuf = (float *)malloc(sizeof(float)*hres*vres);
879     char fnm[256];
880     sprintf(fnm, vval(BASENAME), fcur);
881     strcat(fnm, "_outsamp.pic");
882     for (n = hres*vres; n--; )
883     sbuf[n] = (float)sbuffer[n];
884     write_map(sbuf, fnm);
885     free((void *)sbuf);
886     }
887 greg 3.9 #endif
888 greg 3.1 }
889    
890    
891 greg 3.19 void
892 schorsch 3.8 send_frame(void) /* send frame to destination */
893 greg 3.1 {
894 greg 3.18 char fname[1024];
895 greg 3.1 double d;
896     FILE *fp;
897     int y;
898     /* open output picture */
899 greg 3.18 sprintf(fname, vval(BASENAME), fcur);
900     strcat(fname, ".hdr");
901     fp = fopen(fname, "w");
902 greg 3.1 if (fp == NULL) {
903 greg 3.18 sprintf(errmsg, "cannot open output frame \"%s\"", fname);
904 greg 3.1 error(SYSTEM, errmsg);
905     }
906 schorsch 3.3 SET_FILE_BINARY(fp);
907 greg 3.1 if (!silent) {
908 greg 3.18 printf("\tWriting to \"%s\"\n", fname);
909 greg 3.1 fflush(stdout);
910     }
911     /* write header */
912     newheader("RADIANCE", fp);
913     printargs(gargc, gargv, fp);
914     fprintf(fp, "SOFTWARE= %s\n", VersionID);
915     fprintf(fp, "FRAME=%d\n", fcur);
916     fputnow(fp);
917     fputs(VIEWSTR, fp); fprintview(&vw, fp); fputc('\n', fp);
918     d = expspec_val(getexp(fcur));
919 schorsch 3.6 if ((d < 0.99) | (d > 1.01))
920 greg 3.1 fputexpos(d, fp);
921     d = viewaspect(&vw) * hres / vres;
922 schorsch 3.6 if ((d < 0.99) | (d > 1.01))
923 greg 3.1 fputaspect(d, fp);
924     fputformat(COLRFMT, fp);
925     fputc('\n', fp); /* end header */
926     fprtresolu(hres, vres, fp);
927     if (fflush(fp) == EOF)
928     goto writerr;
929     #if (PIXSTANDARD != (YMAJOR|YDECR))
930     error(CONSISTENCY, "bad code in send_frame");
931     #endif
932     for (y = vres; y--; ) /* write scanlines */
933     if (fwritescan(outbuffer+y*hres, hres, fp) < 0)
934     goto writerr;
935     if (fclose(fp) == EOF)
936     goto writerr;
937 greg 3.18 if (vdef(ZNAME)) { /* output z-buffer */
938     sprintf(fname, vval(ZNAME), fcur);
939     strcat(fname, ".zbf");
940     fp = fopen(fname, "w");
941     if (fp == NULL) {
942     sprintf(errmsg, "cannot open z-file \"%s\"\n", fname);
943     error(SYSTEM, errmsg);
944     }
945     SET_FILE_BINARY(fp);
946     if (!silent) {
947     printf("\tWriting depths to \"%s\"\n", fname);
948     fflush(stdout);
949     }
950     for (y = vres; y--; )
951     if (fwrite(zbuffer+y*hres, sizeof(float),
952     hres, fp) != hres)
953     goto writerr;
954     if (fclose(fp) == EOF)
955     goto writerr;
956     }
957     if (vdef(MNAME)) { /* output motion buffer */
958     unsigned short *mbuffer = (unsigned short *)malloc(
959     sizeof(unsigned short)*3*hres);
960     int x;
961     if (mbuffer == NULL)
962     error(SYSTEM, "out of memory in send_frame()");
963     sprintf(fname, vval(MNAME), fcur);
964     strcat(fname, ".mvo");
965     fp = fopen(fname, "w");
966     if (fp == NULL) {
967     sprintf(errmsg, "cannot open motion file \"%s\"\n",
968     fname);
969     error(SYSTEM, errmsg);
970     }
971     SET_FILE_BINARY(fp);
972     if (!silent) {
973     printf("\tWriting motion vectors to \"%s\"\n", fname);
974     fflush(stdout);
975     }
976     for (y = vres; y--; ) {
977     for (x = hres; x--; ) {
978     mbuffer[3*x] = xmbuffer[fndx(x,y)] + 0x8000;
979     mbuffer[3*x+1] = ymbuffer[fndx(x,y)] + 0x8000;
980     mbuffer[3*x+2] = (oprev[fndx(x,y)]!=OVOID)*0x8000;
981     }
982     if (fwrite(mbuffer, sizeof(*mbuffer),
983     3*hres, fp) != 3*hres)
984     goto writerr;
985     }
986     free((void *)mbuffer);
987     if (fclose(fp) == EOF)
988     goto writerr;
989     }
990 greg 3.1 return; /* all is well */
991     writerr:
992 greg 3.18 sprintf(errmsg, "error writing file \"%s\"", fname);
993 greg 3.1 error(SYSTEM, errmsg);
994     }
995    
996    
997 greg 3.19 void
998 schorsch 3.8 free_frame(void) /* free frame allocation */
999 greg 3.1 {
1000     if (cbuffer == NULL)
1001     return;
1002     free((void *)cbuffer); cbuffer = NULL;
1003     free((void *)zbuffer); zbuffer = NULL;
1004     free((void *)obuffer); obuffer = NULL;
1005     free((void *)xmbuffer); xmbuffer = NULL;
1006     free((void *)ymbuffer); ymbuffer = NULL;
1007     free((void *)cprev); cprev = NULL;
1008     free((void *)zprev); zprev = NULL;
1009     free((void *)oprev); oprev = NULL;
1010     cerrmap = NULL;
1011     val2map = NULL;
1012     hres = vres = 0;
1013     vw.type = vwprev.type = 0;
1014     frm_stop = 0;
1015     }