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

Comparing ray/src/util/dctimestep.c (file contents):
Revision 2.24 by greg, Fri Jan 11 05:07:47 2013 UTC vs.
Revision 2.38 by greg, Tue Apr 10 23:22:42 2018 UTC

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

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines