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.51 by greg, Wed Mar 5 02:54:58 2025 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 495 | Line 30 | sum_images(const char *fspec, const CMATRIX *cv, FILE
30                  error(INTERNAL, "expected vector in sum_images()");
31          for (i = 0; i < cv->nrows; i++) {
32                  const COLORV    *scv = cv_lval(cv,i);
33 +                int             flat_file = 0;
34                  char            fname[1024];
35                  FILE            *fp;
36 +                long            data_start;
37                  int             dt, xr, yr;
38                  COLORV          *psp;
39 +                char            *err;
40                                                          /* check for zero */
41                  if ((scv[RED] == 0) & (scv[GRN] == 0) & (scv[BLU] == 0) &&
42                                  (myDT != DTfromHeader) | (i < cv->nrows-1))
43                          continue;
44                                                          /* open next picture */
45                  sprintf(fname, fspec, i);
46 <                if ((fp = fopen(fname, "r")) == NULL) {
46 >                if ((fp = fopen(fname, "rb")) == NULL) {
47                          sprintf(errmsg, "cannot open picture '%s'", fname);
48                          error(SYSTEM, errmsg);
49                  }
50 <                SET_FILE_BINARY(fp);
51 <                dt = getDTfromHeader(fp);
50 > #ifdef getc_unlocked
51 >                flockfile(fp);
52 > #endif
53 >                dt = DTfromHeader;
54 >                if ((err = cm_getheader(&dt, NULL, NULL, NULL, NULL, fp)) != NULL)
55 >                        error(USER, err);
56                  if ((dt != DTrgbe) & (dt != DTxyze) ||
57                                  !fscnresolu(&xr, &yr, fp)) {
58                          sprintf(errmsg, "file '%s' not a picture", fname);
# Line 519 | Line 61 | sum_images(const char *fspec, const CMATRIX *cv, FILE
61                  if (myDT == DTfromHeader) {             /* on first one */
62                          myDT = dt;
63                          myXR = xr; myYR = yr;
64 <                        scanline = (COLOR *)malloc(sizeof(COLOR)*myXR);
64 >                        scanline = (COLR *)malloc(sizeof(COLR)*myXR);
65                          if (scanline == NULL)
66                                  error(SYSTEM, "out of memory in sum_images()");
67                          pmat = cm_alloc(myYR, myXR);
68                          memset(pmat->cmem, 0, sizeof(COLOR)*myXR*myYR);
69                                                          /* finish header */
70 <                        fputformat(myDT==DTrgbe ? COLRFMT : CIEFMT, fout);
70 >                        fputformat(cm_fmt_id[myDT], fout);
71                          fputc('\n', fout);
530 –                        fprtresolu(myXR, myYR, fout);
72                          fflush(fout);
73                  } else if ((dt != myDT) | (xr != myXR) | (yr != myYR)) {
74                          sprintf(errmsg, "picture '%s' format/size mismatch",
75                                          fname);
76                          error(USER, errmsg);
77                  }
78 +                                                        /* flat file check */
79 +                if ((data_start = ftell(fp)) > 0 && fseek(fp, 0L, SEEK_END) == 0) {
80 +                        flat_file = (ftell(fp) >= data_start + sizeof(COLR)*xr*yr);
81 +                        if (fseek(fp, data_start, SEEK_SET) < 0) {
82 +                                sprintf(errmsg, "cannot seek on picture '%s'", fname);
83 +                                error(SYSTEM, errmsg);
84 +                        }
85 +                }
86                  psp = pmat->cmem;
87                  for (y = 0; y < yr; y++) {              /* read it in */
88 +                        COLOR   col;
89                          int     x;
90 <                        if (freadscan(scanline, xr, fp) < 0) {
90 >                        if (flat_file ? getbinary(scanline, sizeof(COLR), xr, fp) != xr :
91 >                                        freadcolrs(scanline, xr, fp) < 0) {
92                                  sprintf(errmsg, "error reading picture '%s'",
93                                                  fname);
94                                  error(SYSTEM, errmsg);
95                          }
96                                                          /* sum in scanline */
97                          for (x = 0; x < xr; x++, psp += 3) {
98 <                                multcolor(scanline[x], scv);
99 <                                addcolor(psp, scanline[x]);
98 >                                if (!scanline[x][EXP])
99 >                                        continue;       /* skip zeroes */
100 >                                colr_color(col, scanline[x]);
101 >                                multcolor(col, scv);
102 >                                addcolor(psp, col);
103                          }
104                  }
105                  fclose(fp);                             /* done this picture */
106          }
107          free(scanline);
108 <                                                        /* write scanlines */
109 <        for (y = 0; y < myYR; y++)
110 <                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);
108 >        i = cm_write(pmat, myDT, fout);                 /* write picture */
109 >        cm_free(pmat);                                  /* free data */
110 >        return(i);
111   }
112  
113 + /* adjust matrix dimensions according to user size(s) */
114 + static int
115 + alt_dim(CMATRIX *cm, int nr, int nc)
116 + {
117 +        if ((nr <= 0) & (nc <= 0))
118 +                return(0);
119 +        if ((nr == cm->nrows) & (nc == cm->ncols))
120 +                return(0);
121 +        if (nr > 0) {
122 +                if (nc <= 0)
123 +                        nc = cm->nrows*cm->ncols/nr;
124 +                if (nr*nc != cm->nrows*cm->ncols) {
125 +                        fprintf(stderr, "Bad dimensions: %dx%d != %dx%d\n",
126 +                                        nr, nc, cm->nrows, cm->ncols);
127 +                        return(-1);
128 +                }
129 +        } else /* nc > 0 */ {
130 +                nr = cm->nrows*cm->ncols/nc;
131 +                if (nc*nr != cm->nrows*cm->ncols) {
132 +                        fprintf(stderr, "Bad dimensions: %d does not divide %dx%d evenly\n",
133 +                                        nc, cm->nrows, cm->ncols);
134 +                        return(-1);
135 +                }
136 +        }
137 +        cm->nrows = nr;
138 +        cm->ncols = nc;
139 +        return(1);
140 + }
141 +
142   /* check to see if a string contains a %d or %o specification */
143   static int
144   hasNumberFormat(const char *s)
# Line 587 | Line 167 | hasNumberFormat(const char *s)
167   int
168   main(int argc, char *argv[])
169   {
170 <        int             nsteps = 1;
170 >        int             skyfmt = DTfromHeader;
171 >        int             outfmt = DTascii;
172 >        int             headout = 1;
173 >        int             nsteps = 0;
174          char            *ofspec = NULL;
175          FILE            *ofp = stdout;
176 +        int             xres=0, yres=0;
177          CMATRIX         *cmtx;          /* component vector/matrix result */
178          char            fnbuf[256];
179          int             a, i;
180  
181          progname = argv[0];
182                                          /* get options */
183 <        for (a = 1; a < argc-1 && argv[a][0] == '-'; a++)
184 <                switch (argv[0][1]) {
183 >        for (a = 1; a < argc && argv[a][0] == '-'; a++)
184 >                switch (argv[a][1]) {
185                  case 'n':
186                          nsteps = atoi(argv[++a]);
187 <                        if (nsteps <= 0)
187 >                        if (nsteps < 0)
188                                  goto userr;
189 +                        skyfmt = nsteps ? DTascii : DTfromHeader;
190                          break;
191 +                case 'h':
192 +                        headout = !headout;
193 +                        break;
194 +                case 'i':
195 +                        switch (argv[a][2]) {
196 +                        case 'f':
197 +                                skyfmt = DTfloat;
198 +                                break;
199 +                        case 'd':
200 +                                skyfmt = DTdouble;
201 +                                break;
202 +                        case 'a':
203 +                                skyfmt = DTascii;
204 +                                break;
205 +                        default:
206 +                                goto userr;
207 +                        }
208 +                        break;
209                  case 'o':
210 <                        ofspec = argv[++a];
210 >                        switch (argv[a][2]) {
211 >                        case '\0':      /* output specification (not format) */
212 >                                ofspec = argv[++a];
213 >                                break;
214 >                        case 'f':
215 >                                outfmt = DTfloat;
216 >                                break;
217 >                        case 'd':
218 >                                outfmt = DTdouble;
219 >                                break;
220 >                        case 'a':
221 >                                outfmt = DTascii;
222 >                                break;
223 >                        case 'c':
224 >                                outfmt = DTrgbe;
225 >                                break;
226 >                        default:
227 >                                goto userr;
228 >                        }
229                          break;
230 +                case 'x':
231 +                        xres = atoi(argv[++a]);
232 +                        break;
233 +                case 'y':
234 +                        yres = atoi(argv[++a]);
235 +                        break;
236                  default:
237                          goto userr;
238                  }
# Line 613 | Line 240 | main(int argc, char *argv[])
240                  goto userr;
241  
242          if (argc-a > 2) {                       /* VTDs expression */
243 <                CMATRIX *smtx, *Dmat, *Tmat, *imtx;
244 <                COLOR   tLamb;
243 >                CMATRIX         *smtx, *Dmat, *Tmat, *imtx;
244 >                const char      *ccp;
245                                                  /* get sky vector/matrix */
246 <                smtx = cm_load(argv[a+3], 0, nsteps, DTascii);
246 >                smtx = cm_load(argv[a+3], 0, nsteps, skyfmt);
247 >                nsteps = smtx->ncols;
248                                                  /* load BSDF */
249 <                Tmat = cm_loadBSDF(argv[a+1], tLamb);
249 >                if (argv[a+1][0] != '!' &&
250 >                                (ccp = strrchr(argv[a+1], '.')) != NULL &&
251 >                                !strcasecmp(ccp+1, "XML"))
252 >                        Tmat = cm_loadBTDF(argv[a+1]);
253 >                else
254 >                        Tmat = cm_load(argv[a+1], 0, 0, DTfromHeader);
255                                                  /* load Daylight matrix */
256 <                Dmat = cm_load(argv[a+2], Tmat==NULL ? 0 : Tmat->ncols,
256 >                Dmat = cm_load(argv[a+2], Tmat->ncols,
257                                          smtx->nrows, DTfromHeader);
258                                                  /* multiply vector through */
259                  imtx = cm_multiply(Dmat, smtx);
260                  cm_free(Dmat); cm_free(smtx);
261 <                if (Tmat == NULL) {             /* diffuse only */
262 <                        cmtx = cm_scale(imtx, tLamb);
630 <                } else {                        /* else apply BTDF matrix */
631 <                        cmtx = cm_multiply(Tmat, imtx);
632 <                        cm_free(Tmat);
633 <                }
261 >                cmtx = cm_multiply(Tmat, imtx);
262 >                cm_free(Tmat);
263                  cm_free(imtx);
264          } else {                                /* sky vector/matrix only */
265 <                cmtx = cm_load(argv[a+1], 0, nsteps, DTascii);
265 >                cmtx = cm_load(argv[a+1], 0, nsteps, skyfmt);
266 >                nsteps = cmtx->ncols;
267          }
268                                                  /* prepare output stream */
269          if ((ofspec != NULL) & (nsteps == 1) && hasNumberFormat(ofspec)) {
# Line 649 | Line 279 | main(int argc, char *argv[])
279                  ofspec = NULL;                  /* only need to open once */
280          }
281          if (hasNumberFormat(argv[a])) {         /* generating image(s) */
282 +                if (outfmt != DTrgbe) {
283 +                        error(WARNING, "changing output type to -oc");
284 +                        outfmt = DTrgbe;
285 +                }
286                  if (ofspec == NULL) {
287                          SET_FILE_BINARY(ofp);
288                          newheader("RADIANCE", ofp);
# Line 659 | Line 293 | main(int argc, char *argv[])
293                          for (i = 0; i < nsteps; i++) {
294                                  CMATRIX *cvec = cm_column(cmtx, i);
295                                  if (ofspec != NULL) {
296 <                                        sprintf(fnbuf, ofspec, i+1);
296 >                                        sprintf(fnbuf, ofspec, i);
297                                          if ((ofp = fopen(fnbuf, "wb")) == NULL) {
298                                                  fprintf(stderr,
299                                          "%s: cannot open '%s' for output\n",
# Line 670 | Line 304 | main(int argc, char *argv[])
304                                          printargs(argc, argv, ofp);
305                                          fputnow(ofp);
306                                  }
307 <                                fprintf(ofp, "FRAME=%d\n", i+1);
307 >                                fprintf(ofp, "FRAME=%d\n", i);
308                                  if (!sum_images(argv[a], cvec, ofp))
309                                          return(1);
310                                  if (ofspec != NULL) {
# Line 690 | Line 324 | main(int argc, char *argv[])
324                  CMATRIX *Vmat = cm_load(argv[a], 0, cmtx->nrows, DTfromHeader);
325                  CMATRIX *rmtx = cm_multiply(Vmat, cmtx);
326                  cm_free(Vmat);
327 <                if (ofspec != NULL)             /* multiple vector files? */
327 >                if (ofspec != NULL) {           /* multiple vector files? */
328 >                        const char      *wtype = (outfmt==DTascii) ? "w" : "wb";
329                          for (i = 0; i < nsteps; i++) {
330                                  CMATRIX *rvec = cm_column(rmtx, i);
331 <                                sprintf(fnbuf, ofspec, i+1);
332 <                                if ((ofp = fopen(fnbuf, "w")) == NULL) {
331 >                                if (alt_dim(rvec, yres, xres) < 0)
332 >                                        return(1);
333 >                                sprintf(fnbuf, ofspec, i);
334 >                                if ((ofp = fopen(fnbuf, wtype)) == NULL) {
335                                          fprintf(stderr,
336                                          "%s: cannot open '%s' for output\n",
337                                                          progname, fnbuf);
338                                          return(1);
339                                  }
340 <                                cm_print(rvec, ofp);
340 > #ifdef getc_unlocked
341 >                                flockfile(ofp);
342 > #endif
343 >                                if (headout) {  /* header output */
344 >                                        newheader("RADIANCE", ofp);
345 >                                        printargs(argc, argv, ofp);
346 >                                        fputnow(ofp);
347 >                                        fprintf(ofp, "FRAME=%d\n", i);
348 >                                        if ((outfmt != DTrgbe) & (outfmt != DTxyze)) {
349 >                                                fprintf(ofp, "NROWS=%d\n", rvec->nrows);
350 >                                                fprintf(ofp, "NCOLS=%d\n", rvec->ncols);
351 >                                                fputs("NCOMP=3\n", ofp);
352 >                                        }
353 >                                        if ((outfmt == DTfloat) | (outfmt == DTdouble))
354 >                                                fputendian(ofp);
355 >                                        fputformat(cm_fmt_id[outfmt], ofp);
356 >                                        fputc('\n', ofp);
357 >                                }
358 >                                cm_write(rvec, outfmt, ofp);
359                                  if (fclose(ofp) == EOF) {
360                                          fprintf(stderr,
361                                                  "%s: error writing to '%s'\n",
# Line 710 | Line 365 | main(int argc, char *argv[])
365                                  ofp = stdout;
366                                  cm_free(rvec);
367                          }
368 <                else
369 <                        cm_print(rmtx, ofp);
368 >                } else {
369 > #ifdef getc_unlocked
370 >                        flockfile(ofp);
371 > #endif
372 >                        if (outfmt != DTascii)
373 >                                SET_FILE_BINARY(ofp);
374 >                        if (alt_dim(rmtx, yres, xres) < 0)
375 >                                return(1);
376 >                        if (headout) {          /* header output */
377 >                                newheader("RADIANCE", ofp);
378 >                                printargs(argc, argv, ofp);
379 >                                fputnow(ofp);
380 >                                if ((outfmt != DTrgbe) & (outfmt != DTxyze)) {
381 >                                        fprintf(ofp, "NROWS=%d\n", rmtx->nrows);
382 >                                        fprintf(ofp, "NCOLS=%d\n", rmtx->ncols);
383 >                                        fputs("NCOMP=3\n", ofp);
384 >                                }
385 >                                if ((outfmt == DTfloat) | (outfmt == DTdouble))
386 >                                        fputendian(ofp);
387 >                                fputformat(cm_fmt_id[outfmt], ofp);
388 >                                fputc('\n', ofp);
389 >                        }
390 >                        cm_write(rmtx, outfmt, ofp);
391 >                }
392                  cm_free(rmtx);
393          }
394          if (fflush(ofp) == EOF) {               /* final clean-up */
# Line 721 | Line 398 | main(int argc, char *argv[])
398          cm_free(cmtx);
399          return(0);
400   userr:
401 <        fprintf(stderr, "Usage: %s [-n nsteps][-o ospec] DCspec [skyf]\n",
401 >        fprintf(stderr, "Usage: %s [-n nsteps][-o ospec][-x xr][-y yr][-i{f|d|h}][-o{f|d|c}] DCspec [skyf]\n",
402                                  progname);
403 <        fprintf(stderr, "   or: %s [-n nsteps][-o ospec] Vspec Tbsdf.xml Dmat.dat [skyf]\n",
403 >        fprintf(stderr, "   or: %s [-n nsteps][-o ospec][-x xr][-y yr][-i{f|d|h}][-o{f|d|c}] Vspec Tbsdf Dmat.dat [skyf]\n",
404                                  progname);
405          return(1);
406   }

Diff Legend

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