ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/src/util/dctimestep.c
(Generate patch)

Comparing src/util/dctimestep.c (file contents):
Revision 2.27 by greg, Sat Jan 19 22:30:55 2013 UTC vs.
Revision 2.34 by greg, Wed Sep 17 22:40:49 2014 UTC

# Line 9 | Line 9 | static const char RCSid[] = "$Id$";
9  
10   #include <ctype.h>
11   #include "standard.h"
12 + #include "cmatrix.h"
13   #include "platform.h"
13 – #include "paths.h"
14 – #include "color.h"
14   #include "resolu.h"
16 – #include "bsdf.h"
17 – #include "bsdf_m.h"
15  
16   char    *progname;                      /* global argv[0] */
17  
21 – /* Data types for file loading */
22 – enum {DTfromHeader, DTascii, DTfloat, DTdouble, DTrgbe, DTxyze};
23 –
24 – /* A color coefficient matrix -- vectors have ncols==1 */
25 – typedef struct {
26 –        int     nrows, ncols;
27 –        COLORV  cmem[3];                /* extends struct */
28 – } CMATRIX;
29 –
30 – #define COLSPEC (sizeof(COLORV)==sizeof(float) ? "%f %f %f" : "%lf %lf %lf")
31 –
32 – #define cm_lval(cm,r,c) ((cm)->cmem + 3*((r)*(cm)->ncols + (c)))
33 –
34 – #define cv_lval(cm,i)   ((cm)->cmem + 3*(i))
35 –
36 – /* Allocate a color coefficient matrix */
37 – static CMATRIX *
38 – cm_alloc(int nrows, int ncols)
39 – {
40 –        CMATRIX *cm;
41 –
42 –        if ((nrows <= 0) | (ncols <= 0))
43 –                error(USER, "attempt to create empty matrix");
44 –        cm = (CMATRIX *)malloc(sizeof(CMATRIX) +
45 –                                3*sizeof(COLORV)*(nrows*ncols - 1));
46 –        if (cm == NULL)
47 –                error(SYSTEM, "out of memory in cm_alloc()");
48 –        cm->nrows = nrows;
49 –        cm->ncols = ncols;
50 –        return(cm);
51 – }
52 –
53 – #define cm_free(cm)     free(cm)
54 –
55 – /* Resize color coefficient matrix */
56 – static CMATRIX *
57 – cm_resize(CMATRIX *cm, int nrows)
58 – {
59 –        if (nrows == cm->nrows)
60 –                return(cm);
61 –        if (nrows <= 0) {
62 –                cm_free(cm);
63 –                return(NULL);
64 –        }
65 –        cm = (CMATRIX *)realloc(cm, sizeof(CMATRIX) +
66 –                        3*sizeof(COLORV)*(nrows*cm->ncols - 1));
67 –        if (cm == NULL)
68 –                error(SYSTEM, "out of memory in cm_resize()");
69 –        cm->nrows = nrows;
70 –        return(cm);
71 – }
72 –
73 – /* Load header to obtain data type */
74 – static int
75 – getDT(char *s, void *p)
76 – {
77 –        char    fmt[32];
78 –        
79 –        if (formatval(fmt, s)) {
80 –                if (!strcmp(fmt, "ascii"))
81 –                        *((int *)p) = DTascii;
82 –                else if (!strcmp(fmt, "float"))
83 –                        *((int *)p) = DTfloat;
84 –                else if (!strcmp(fmt, "double"))
85 –                        *((int *)p) = DTdouble;
86 –                else if (!strcmp(fmt, COLRFMT))
87 –                        *((int *)p) = DTrgbe;
88 –                else if (!strcmp(fmt, CIEFMT))
89 –                        *((int *)p) = DTxyze;
90 –        }
91 –        return(0);
92 – }
93 –
94 – static int
95 – getDTfromHeader(FILE *fp)
96 – {
97 –        int     dt = DTfromHeader;
98 –        
99 –        if (getheader(fp, getDT, &dt) < 0)
100 –                error(SYSTEM, "header read error");
101 –        if (dt == DTfromHeader)
102 –                error(USER, "missing data format in header");
103 –        return(dt);
104 – }
105 –
106 – /* Allocate and load a matrix from the given file (or stdin if NULL) */
107 – static CMATRIX *
108 – cm_load(const char *fname, int nrows, int ncols, int dtype)
109 – {
110 –        FILE    *fp = stdin;
111 –        CMATRIX *cm;
112 –
113 –        if (ncols <= 0)
114 –                error(USER, "Non-positive number of columns");
115 –        if (fname == NULL)
116 –                fname = "<stdin>";
117 –        else if ((fp = fopen(fname, "r")) == NULL) {
118 –                sprintf(errmsg, "cannot open file '%s'", fname);
119 –                error(SYSTEM, errmsg);
120 –        }
121 – #ifdef getc_unlocked
122 –        flockfile(fp);
123 – #endif
124 –        if (dtype != DTascii)
125 –                SET_FILE_BINARY(fp);            /* doesn't really work */
126 –        if (dtype == DTfromHeader)
127 –                dtype = getDTfromHeader(fp);
128 –        switch (dtype) {
129 –        case DTascii:
130 –        case DTfloat:
131 –        case DTdouble:
132 –                break;
133 –        default:
134 –                error(USER, "unexpected data type in cm_load()");
135 –        }
136 –        if (nrows <= 0) {                       /* don't know length? */
137 –                int     guessrows = 147;        /* usually big enough */
138 –                if ((dtype != DTascii) & (fp != stdin)) {
139 –                        long    startpos = ftell(fp);
140 –                        if (fseek(fp, 0L, SEEK_END) == 0) {
141 –                                long    endpos = ftell(fp);
142 –                                long    elemsiz = 3*(dtype==DTfloat ?
143 –                                            sizeof(float) : sizeof(double));
144 –
145 –                                if ((endpos - startpos) % (ncols*elemsiz)) {
146 –                                        sprintf(errmsg,
147 –                                        "improper length for binary file '%s'",
148 –                                                        fname);
149 –                                        error(USER, errmsg);
150 –                                }
151 –                                guessrows = (endpos - startpos)/(ncols*elemsiz);
152 –                                if (fseek(fp, startpos, SEEK_SET) < 0) {
153 –                                        sprintf(errmsg,
154 –                                                "fseek() error on file '%s'",
155 –                                                        fname);
156 –                                        error(SYSTEM, errmsg);
157 –                                }
158 –                                nrows = guessrows;      /* we're confident */
159 –                        }
160 –                }
161 –                cm = cm_alloc(guessrows, ncols);
162 –        } else
163 –                cm = cm_alloc(nrows, ncols);
164 –        if (cm == NULL)                                 /* XXX never happens */
165 –                return(NULL);
166 –        if (dtype == DTascii) {                         /* read text file */
167 –                int     maxrow = (nrows > 0 ? nrows : 32000);
168 –                int     r, c;
169 –                for (r = 0; r < maxrow; r++) {
170 –                    if (r >= cm->nrows)                 /* need more space? */
171 –                        cm = cm_resize(cm, 2*cm->nrows);
172 –                    for (c = 0; c < ncols; c++) {
173 –                        COLORV  *cv = cm_lval(cm,r,c);
174 –                        if (fscanf(fp, COLSPEC, cv, cv+1, cv+2) != 3)
175 –                                if ((nrows <= 0) & (r > 0) & !c) {
176 –                                        cm = cm_resize(cm, maxrow=r);
177 –                                        break;
178 –                                } else
179 –                                        goto EOFerror;
180 –                    }
181 –                }
182 –                while ((c = getc(fp)) != EOF)
183 –                        if (!isspace(c)) {
184 –                                sprintf(errmsg,
185 –                                "unexpected data at end of ascii file %s",
186 –                                                fname);
187 –                                error(WARNING, errmsg);
188 –                                break;
189 –                        }
190 –        } else {                                        /* read binary file */
191 –                if (sizeof(COLORV) == (dtype==DTfloat ? sizeof(float) :
192 –                                                        sizeof(double))) {
193 –                        int     nread = 0;
194 –                        do {                            /* read all we can */
195 –                                nread += fread(cm->cmem + 3*nread,
196 –                                                3*sizeof(COLORV),
197 –                                                cm->nrows*cm->ncols - nread,
198 –                                                fp);
199 –                                if (nrows <= 0) {       /* unknown length */
200 –                                        if (nread == cm->nrows*cm->ncols)
201 –                                                        /* need more space? */
202 –                                                cm = cm_resize(cm, 2*cm->nrows);
203 –                                        else if (nread && !(nread % cm->ncols))
204 –                                                        /* seem to be  done */
205 –                                                cm = cm_resize(cm, nread/cm->ncols);
206 –                                        else            /* ended mid-row */
207 –                                                goto EOFerror;
208 –                                } else if (nread < cm->nrows*cm->ncols)
209 –                                        goto EOFerror;
210 –                        } while (nread < cm->nrows*cm->ncols);
211 –
212 –                } else if (dtype == DTdouble) {
213 –                        double  dc[3];                  /* load from double */
214 –                        COLORV  *cvp = cm->cmem;
215 –                        int     n = nrows*ncols;
216 –
217 –                        if (n <= 0)
218 –                                goto not_handled;
219 –                        while (n--) {
220 –                                if (fread(dc, sizeof(double), 3, fp) != 3)
221 –                                        goto EOFerror;
222 –                                copycolor(cvp, dc);
223 –                                cvp += 3;
224 –                        }
225 –                } else /* dtype == DTfloat */ {
226 –                        float   fc[3];                  /* load from float */
227 –                        COLORV  *cvp = cm->cmem;
228 –                        int     n = nrows*ncols;
229 –
230 –                        if (n <= 0)
231 –                                goto not_handled;
232 –                        while (n--) {
233 –                                if (fread(fc, sizeof(float), 3, fp) != 3)
234 –                                        goto EOFerror;
235 –                                copycolor(cvp, fc);
236 –                                cvp += 3;
237 –                        }
238 –                }
239 –                if (fgetc(fp) != EOF) {
240 –                                sprintf(errmsg,
241 –                                "unexpected data at end of binary file %s",
242 –                                                fname);
243 –                                error(WARNING, errmsg);
244 –                }
245 –        }
246 –        if (fp != stdin)
247 –                fclose(fp);
248 – #ifdef getc_unlocked
249 –        else
250 –                funlockfile(fp);
251 – #endif
252 –        return(cm);
253 – EOFerror:
254 –        sprintf(errmsg, "unexpected EOF reading %s", fname);
255 –        error(USER, errmsg);
256 – not_handled:
257 –        error(INTERNAL, "unhandled data size or length in cm_load()");
258 –        return(NULL);   /* gratis return */
259 – }
260 –
261 – /* Extract a column vector from a matrix */
262 – static CMATRIX *
263 – cm_column(const CMATRIX *cm, int c)
264 – {
265 –        CMATRIX *cvr;
266 –        int     dr;
267 –
268 –        if ((c < 0) | (c >= cm->ncols))
269 –                error(INTERNAL, "column requested outside matrix");
270 –        cvr = cm_alloc(cm->nrows, 1);
271 –        if (cvr == NULL)
272 –                return(NULL);
273 –        for (dr = 0; dr < cm->nrows; dr++) {
274 –                const COLORV    *sp = cm_lval(cm,dr,c);
275 –                COLORV          *dp = cv_lval(cvr,dr);
276 –                dp[0] = sp[0];
277 –                dp[1] = sp[1];
278 –                dp[2] = sp[2];
279 –        }
280 –        return(cvr);
281 – }
282 –
283 – /* Scale a matrix by a single value */
284 – static CMATRIX *
285 – cm_scale(const CMATRIX *cm1, const COLOR sca)
286 – {
287 –        CMATRIX *cmr;
288 –        int     dr, dc;
289 –
290 –        cmr = cm_alloc(cm1->nrows, cm1->ncols);
291 –        if (cmr == NULL)
292 –                return(NULL);
293 –        for (dr = 0; dr < cmr->nrows; dr++)
294 –            for (dc = 0; dc < cmr->ncols; dc++) {
295 –                const COLORV    *sp = cm_lval(cm1,dr,dc);
296 –                COLORV          *dp = cm_lval(cmr,dr,dc);
297 –                dp[0] = sp[0] * sca[0];
298 –                dp[1] = sp[1] * sca[1];
299 –                dp[2] = sp[2] * sca[2];
300 –            }
301 –        return(cmr);
302 – }
303 –
304 – /* Multiply two matrices (or a matrix and a vector) and allocate the result */
305 – static CMATRIX *
306 – cm_multiply(const CMATRIX *cm1, const CMATRIX *cm2)
307 – {
308 –        char    *rowcheck=NULL, *colcheck=NULL;
309 –        CMATRIX *cmr;
310 –        int     dr, dc, i;
311 –
312 –        if ((cm1->ncols <= 0) | (cm1->ncols != cm2->nrows))
313 –                error(INTERNAL, "matrix dimension mismatch in cm_multiply()");
314 –        cmr = cm_alloc(cm1->nrows, cm2->ncols);
315 –        if (cmr == NULL)
316 –                return(NULL);
317 –                                /* optimization: check for zero rows & cols */
318 –        if (((cm1->nrows > 5) | (cm2->ncols > 5)) & (cm1->ncols > 5)) {
319 –                static const COLOR      czero;
320 –                rowcheck = (char *)calloc(cmr->nrows, 1);
321 –                for (dr = cm1->nrows*(rowcheck != NULL); dr--; )
322 –                    for (dc = cm1->ncols; dc--; )
323 –                        if (memcmp(cm_lval(cm1,dr,dc), czero, sizeof(COLOR))) {
324 –                                rowcheck[dr] = 1;
325 –                                break;
326 –                        }
327 –                colcheck = (char *)calloc(cmr->ncols, 1);
328 –                for (dc = cm2->ncols*(colcheck != NULL); dc--; )
329 –                    for (dr = cm2->nrows; dr--; )
330 –                        if (memcmp(cm_lval(cm2,dr,dc), czero, sizeof(COLOR))) {
331 –                                colcheck[dc] = 1;
332 –                                break;
333 –                        }
334 –        }
335 –        for (dr = 0; dr < cmr->nrows; dr++)
336 –            for (dc = 0; dc < cmr->ncols; dc++) {
337 –                COLORV  *dp = cm_lval(cmr,dr,dc);
338 –                dp[0] = dp[1] = dp[2] = 0;
339 –                if (rowcheck != NULL && !rowcheck[dr])
340 –                        continue;
341 –                if (colcheck != NULL && !colcheck[dc])
342 –                        continue;
343 –                for (i = 0; i < cm1->ncols; i++) {
344 –                    const COLORV        *cp1 = cm_lval(cm1,dr,i);
345 –                    const COLORV        *cp2 = cm_lval(cm2,i,dc);
346 –                    dp[0] += cp1[0] * cp2[0];
347 –                    dp[1] += cp1[1] * cp2[1];
348 –                    dp[2] += cp1[2] * cp2[2];
349 –                }
350 –            }
351 –        if (rowcheck != NULL) free(rowcheck);
352 –        if (colcheck != NULL) free(colcheck);
353 –        return(cmr);
354 – }
355 –
356 – /* print out matrix as ASCII text -- no header */
357 – static void
358 – cm_print(const CMATRIX *cm, FILE *fp)
359 – {
360 –        int             r, c;
361 –        const COLORV    *mp = cm->cmem;
362 –        
363 –        for (r = 0; r < cm->nrows; r++) {
364 –                for (c = 0; c < cm->ncols; c++, mp += 3)
365 –                        fprintf(fp, "\t%.6e %.6e %.6e", mp[0], mp[1], mp[2]);
366 –                fputc('\n', fp);
367 –        }
368 – }
369 –
370 – /* Convert a BSDF to our matrix representation */
371 – static CMATRIX *
372 – cm_bsdf(const COLOR bsdfLamb, const COLOR specCol, const SDMat *bsdf)
373 – {
374 –        CMATRIX *cm = cm_alloc(bsdf->nout, bsdf->ninc);
375 –        int     nbadohm = 0;
376 –        int     nneg = 0;
377 –        int     r, c;
378 –                                        /* loop over incident angles */
379 –        for (c = 0; c < cm->ncols; c++) {
380 –                const double    dom = mBSDF_incohm(bsdf,c);
381 –                                        /* projected solid angle */
382 –                nbadohm += (dom <= 0);
383 –
384 –                for (r = 0; r < cm->nrows; r++) {
385 –                        float   f = mBSDF_value(bsdf,c,r);
386 –                        COLORV  *mp = cm_lval(cm,r,c);
387 –                                        /* check BSDF value */
388 –                        if ((f <= 0) | (dom <= 0)) {
389 –                                nneg += (f < -FTINY);
390 –                                f = .0f;
391 –                        }
392 –                        copycolor(mp, specCol);
393 –                        scalecolor(mp, f);
394 –                        addcolor(mp, bsdfLamb);
395 –                        scalecolor(mp, dom);
396 –                }
397 –        }
398 –        if (nneg | nbadohm) {
399 –                sprintf(errmsg,
400 –                    "BTDF has %d negatives and %d bad incoming solid angles",
401 –                                nneg, nbadohm);
402 –                error(WARNING, errmsg);
403 –        }
404 –        return(cm);
405 – }
406 –
407 – /* Convert between input and output indices for reciprocity */
408 – static int
409 – recip_out_from_in(const SDMat *bsdf, int in_recip)
410 – {
411 –        FVECT   v;
412 –
413 –        if (!mBSDF_incvec(v, bsdf, in_recip+.5))
414 –                return(in_recip);               /* XXX should be error! */
415 –        v[2] = -v[2];
416 –        return(mBSDF_outndx(bsdf, v));
417 – }
418 –
419 – /* Convert between output and input indices for reciprocity */
420 – static int
421 – recip_in_from_out(const SDMat *bsdf, int out_recip)
422 – {
423 –        FVECT   v;
424 –
425 –        if (!mBSDF_outvec(v, bsdf, out_recip+.5))
426 –                return(out_recip);              /* XXX should be error! */
427 –        v[2] = -v[2];
428 –        return(mBSDF_incndx(bsdf, v));
429 – }
430 –
431 – /* Convert a BSDF to our matrix representation, applying reciprocity */
432 – static CMATRIX *
433 – cm_bsdf_recip(const COLOR bsdfLamb, const COLOR specCol, const SDMat *bsdf)
434 – {
435 –        CMATRIX *cm = cm_alloc(bsdf->ninc, bsdf->nout);
436 –        int     nbadohm = 0;
437 –        int     nneg = 0;
438 –        int     r, c;
439 –                                        /* loop over incident angles */
440 –        for (c = 0; c < cm->ncols; c++) {
441 –                const int       ro = recip_out_from_in(bsdf,c);
442 –                const double    dom = mBSDF_outohm(bsdf,ro);
443 –                                        /* projected solid angle */
444 –                nbadohm += (dom <= 0);
445 –
446 –                for (r = 0; r < cm->nrows; r++) {
447 –                        const int       ri = recip_in_from_out(bsdf,r);
448 –                        float           f = mBSDF_value(bsdf,ri,ro);
449 –                        COLORV          *mp = cm_lval(cm,r,c);
450 –                                        /* check BSDF value */
451 –                        if ((f <= 0) | (dom <= 0)) {
452 –                                nneg += (f < -FTINY);
453 –                                f = .0f;
454 –                        }
455 –                        copycolor(mp, specCol);
456 –                        scalecolor(mp, f);
457 –                        addcolor(mp, bsdfLamb);
458 –                        scalecolor(mp, dom);
459 –                }
460 –        }
461 –        if (nneg | nbadohm) {
462 –                sprintf(errmsg,
463 –                    "BTDF has %d negatives and %d bad incoming solid angles",
464 –                                nneg, nbadohm);
465 –                error(WARNING, errmsg);
466 –        }
467 –        return(cm);
468 – }
469 –
470 – /* Load and convert a matrix BSDF from the given XML file */
471 – static CMATRIX *
472 – cm_loadBSDF(char *fname, COLOR cLamb)
473 – {
474 –        CMATRIX         *Tmat;
475 –        char            *fpath;
476 –        int             recip;
477 –        SDError         ec;
478 –        SDData          myBSDF;
479 –        SDSpectralDF    *tdf;
480 –        COLOR           bsdfLamb, specCol;
481 –                                        /* find path to BSDF file */
482 –        fpath = getpath(fname, getrlibpath(), R_OK);
483 –        if (fpath == NULL) {
484 –                sprintf(errmsg, "cannot find BSDF file '%s'", fname);
485 –                error(USER, errmsg);
486 –        }
487 –        SDclearBSDF(&myBSDF, fname);    /* load XML and check type */
488 –        ec = SDloadFile(&myBSDF, fpath);
489 –        if (ec)
490 –                error(USER, transSDError(ec));
491 –        ccy2rgb(&myBSDF.tLamb.spec, myBSDF.tLamb.cieY/PI, bsdfLamb);
492 –        recip = (myBSDF.tb == NULL);
493 –        tdf = recip ? myBSDF.tf : myBSDF.tb;
494 –        if (tdf == NULL) {              /* no non-Lambertian transmission? */
495 –                if (cLamb != NULL)
496 –                        copycolor(cLamb, bsdfLamb);
497 –                SDfreeBSDF(&myBSDF);
498 –                return(NULL);
499 –        }
500 –        if (tdf->ncomp != 1 || tdf->comp[0].func != &SDhandleMtx) {
501 –                sprintf(errmsg, "unsupported BSDF '%s'", fpath);
502 –                error(USER, errmsg);
503 –        }
504 –                                        /* convert BTDF to matrix */
505 –        ccy2rgb(&tdf->comp[0].cspec[0], 1., specCol);
506 –        Tmat = recip ? cm_bsdf_recip(bsdfLamb, specCol, (SDMat *)tdf->comp[0].dist)
507 –                        : cm_bsdf(bsdfLamb, specCol, (SDMat *)tdf->comp[0].dist);
508 –        if (cLamb != NULL)              /* Lambertian is included */
509 –                setcolor(cLamb, .0, .0, .0);
510 –                                        /* free BSDF and return */
511 –        SDfreeBSDF(&myBSDF);
512 –        return(Tmat);
513 – }
514 –
18   /* Sum together a set of images and write result to fout */
19   static int
20   sum_images(const char *fspec, const CMATRIX *cv, FILE *fout)
# Line 523 | Line 26 | sum_images(const char *fspec, const CMATRIX *cv, FILE
26          int     i, y;
27  
28          if (cv->ncols != 1)
29 <                error(INTERNAL, "expected vector in sum_images()");
29 >                error(INTERNAL, "expected matrix in sum_images()");
30          for (i = 0; i < cv->nrows; i++) {
31                  const COLORV    *scv = cv_lval(cv,i);
32                  char            fname[1024];
33                  FILE            *fp;
34                  int             dt, xr, yr;
35                  COLORV          *psp;
36 +                char            *err;
37                                                          /* check for zero */
38                  if ((scv[RED] == 0) & (scv[GRN] == 0) & (scv[BLU] == 0) &&
39                                  (myDT != DTfromHeader) | (i < cv->nrows-1))
40                          continue;
41                                                          /* open next picture */
42                  sprintf(fname, fspec, i);
43 <                if ((fp = fopen(fname, "r")) == NULL) {
43 >                if ((fp = fopen(fname, "rb")) == NULL) {
44                          sprintf(errmsg, "cannot open picture '%s'", fname);
45                          error(SYSTEM, errmsg);
46                  }
47 <                SET_FILE_BINARY(fp);
48 <                dt = getDTfromHeader(fp);
47 >                dt = DTfromHeader;
48 >                if ((err = cm_getheader(&dt, NULL, NULL, fp)) != NULL)
49 >                        error(USER, err);
50                  if ((dt != DTrgbe) & (dt != DTxyze) ||
51                                  !fscnresolu(&xr, &yr, fp)) {
52                          sprintf(errmsg, "file '%s' not a picture", fname);
# Line 556 | Line 61 | sum_images(const char *fspec, const CMATRIX *cv, FILE
61                          pmat = cm_alloc(myYR, myXR);
62                          memset(pmat->cmem, 0, sizeof(COLOR)*myXR*myYR);
63                                                          /* finish header */
64 <                        fputformat(myDT==DTrgbe ? COLRFMT : CIEFMT, fout);
64 >                        fputformat((char *)cm_fmt_id[myDT], fout);
65                          fputc('\n', fout);
561 –                        fprtresolu(myXR, myYR, fout);
66                          fflush(fout);
67                  } else if ((dt != myDT) | (xr != myXR) | (yr != myYR)) {
68                          sprintf(errmsg, "picture '%s' format/size mismatch",
# Line 582 | Line 86 | sum_images(const char *fspec, const CMATRIX *cv, FILE
86                  fclose(fp);                             /* done this picture */
87          }
88          free(scanline);
89 <                                                        /* write scanlines */
90 <        for (y = 0; y < myYR; y++)
91 <                if (fwritescan((COLOR *)cm_lval(pmat, y, 0), myXR, fout) < 0)
588 <                        return(0);
589 <        cm_free(pmat);                                  /* all done */
590 <        return(fflush(fout) == 0);
89 >        i = cm_write(pmat, myDT, fout);                 /* write picture */
90 >        cm_free(pmat);                                  /* free data */
91 >        return(i);
92   }
93  
94   /* check to see if a string contains a %d or %o specification */
# Line 618 | Line 119 | hasNumberFormat(const char *s)
119   int
120   main(int argc, char *argv[])
121   {
122 <        int             skyfmt = DTascii;
123 <        int             nsteps = 1;
122 >        int             skyfmt = DTfromHeader;
123 >        int             outfmt = DTascii;
124 >        int             headout = 1;
125 >        int             nsteps = 0;
126          char            *ofspec = NULL;
127          FILE            *ofp = stdout;
128          CMATRIX         *cmtx;          /* component vector/matrix result */
# Line 632 | Line 135 | main(int argc, char *argv[])
135                  switch (argv[a][1]) {
136                  case 'n':
137                          nsteps = atoi(argv[++a]);
138 <                        if (nsteps <= 0)
138 >                        if (nsteps < 0)
139                                  goto userr;
140 +                        skyfmt = nsteps ? DTascii : DTfromHeader;
141                          break;
142 <                case 'o':
143 <                        ofspec = argv[++a];
142 >                case 'h':
143 >                        headout = !headout;
144                          break;
145                  case 'i':
146                          switch (argv[a][2]) {
# Line 653 | Line 157 | main(int argc, char *argv[])
157                                  goto userr;
158                          }
159                          break;
160 +                case 'o':
161 +                        switch (argv[a][2]) {
162 +                        case '\0':      /* output specification (not format) */
163 +                                ofspec = argv[++a];
164 +                                break;
165 +                        case 'f':
166 +                                outfmt = DTfloat;
167 +                                break;
168 +                        case 'd':
169 +                                outfmt = DTdouble;
170 +                                break;
171 +                        case 'a':
172 +                                outfmt = DTascii;
173 +                                break;
174 +                        default:
175 +                                goto userr;
176 +                        }
177 +                        break;
178                  default:
179                          goto userr;
180                  }
# Line 661 | Line 183 | main(int argc, char *argv[])
183  
184          if (argc-a > 2) {                       /* VTDs expression */
185                  CMATRIX *smtx, *Dmat, *Tmat, *imtx;
664 –                COLOR   tLamb;
186                                                  /* get sky vector/matrix */
187                  smtx = cm_load(argv[a+3], 0, nsteps, skyfmt);
188                                                  /* load BSDF */
189 <                Tmat = cm_loadBSDF(argv[a+1], tLamb);
189 >                Tmat = cm_loadBTDF(argv[a+1]);
190                                                  /* load Daylight matrix */
191                  Dmat = cm_load(argv[a+2], Tmat==NULL ? 0 : Tmat->ncols,
192                                          smtx->nrows, DTfromHeader);
193                                                  /* multiply vector through */
194                  imtx = cm_multiply(Dmat, smtx);
195                  cm_free(Dmat); cm_free(smtx);
196 <                if (Tmat == NULL) {             /* diffuse only */
197 <                        cmtx = cm_scale(imtx, tLamb);
677 <                } else {                        /* else apply BTDF matrix */
678 <                        cmtx = cm_multiply(Tmat, imtx);
679 <                        cm_free(Tmat);
680 <                }
196 >                cmtx = cm_multiply(Tmat, imtx);
197 >                cm_free(Tmat);
198                  cm_free(imtx);
199          } else {                                /* sky vector/matrix only */
200                  cmtx = cm_load(argv[a+1], 0, nsteps, skyfmt);
# Line 737 | Line 254 | main(int argc, char *argv[])
254                  CMATRIX *Vmat = cm_load(argv[a], 0, cmtx->nrows, DTfromHeader);
255                  CMATRIX *rmtx = cm_multiply(Vmat, cmtx);
256                  cm_free(Vmat);
257 <                if (ofspec != NULL)             /* multiple vector files? */
257 >                if (ofspec != NULL) {           /* multiple vector files? */
258 >                        const char      *wtype = (outfmt==DTascii) ? "w" : "wb";
259                          for (i = 0; i < nsteps; i++) {
260                                  CMATRIX *rvec = cm_column(rmtx, i);
261                                  sprintf(fnbuf, ofspec, i+1);
262 <                                if ((ofp = fopen(fnbuf, "w")) == NULL) {
262 >                                if ((ofp = fopen(fnbuf, wtype)) == NULL) {
263                                          fprintf(stderr,
264                                          "%s: cannot open '%s' for output\n",
265                                                          progname, fnbuf);
266                                          return(1);
267                                  }
268 <                                cm_print(rvec, ofp);
268 > #ifdef getc_unlocked
269 >                                flockfile(ofp);
270 > #endif
271 >                                if (headout) {  /* header output */
272 >                                        newheader("RADIANCE", ofp);
273 >                                        printargs(argc, argv, ofp);
274 >                                        fputnow(ofp);
275 >                                        fprintf(ofp, "FRAME=%d\n", i+1);
276 >                                        fprintf(ofp, "NROWS=%d\n", rvec->nrows);
277 >                                        fputs("NCOLS=1\nNCOMP=3\n", ofp);
278 >                                        fputformat((char *)cm_fmt_id[outfmt], ofp);
279 >                                        fputc('\n', ofp);
280 >                                }
281 >                                cm_write(rvec, outfmt, ofp);
282                                  if (fclose(ofp) == EOF) {
283                                          fprintf(stderr,
284                                                  "%s: error writing to '%s'\n",
# Line 757 | Line 288 | main(int argc, char *argv[])
288                                  ofp = stdout;
289                                  cm_free(rvec);
290                          }
291 <                else
292 <                        cm_print(rmtx, ofp);
291 >                } else {
292 > #ifdef getc_unlocked
293 >                        flockfile(ofp);
294 > #endif
295 >                        if (outfmt != DTascii)
296 >                                SET_FILE_BINARY(ofp);
297 >                        if (headout) {          /* header output */
298 >                                newheader("RADIANCE", ofp);
299 >                                printargs(argc, argv, ofp);
300 >                                fputnow(ofp);
301 >                                fprintf(ofp, "NROWS=%d\n", rmtx->nrows);
302 >                                fprintf(ofp, "NCOLS=%d\n", rmtx->ncols);
303 >                                fputs("NCOMP=3\n", ofp);
304 >                                fputformat((char *)cm_fmt_id[outfmt], ofp);
305 >                                fputc('\n', ofp);
306 >                        }
307 >                        cm_write(rmtx, outfmt, ofp);
308 >                }
309                  cm_free(rmtx);
310          }
311          if (fflush(ofp) == EOF) {               /* final clean-up */
# Line 768 | Line 315 | main(int argc, char *argv[])
315          cm_free(cmtx);
316          return(0);
317   userr:
318 <        fprintf(stderr, "Usage: %s [-n nsteps][-o ospec][-i{f|d}] DCspec [skyf]\n",
318 >        fprintf(stderr, "Usage: %s [-n nsteps][-o ospec][-i{f|d|h}][-o{f|d}] DCspec [skyf]\n",
319                                  progname);
320 <        fprintf(stderr, "   or: %s [-n nsteps][-o ospec][-i{f|d}] Vspec Tbsdf.xml Dmat.dat [skyf]\n",
320 >        fprintf(stderr, "   or: %s [-n nsteps][-o ospec][-i{f|d|h}][-o{f|d}] Vspec Tbsdf.xml Dmat.dat [skyf]\n",
321                                  progname);
322          return(1);
323   }

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines