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.23 by greg, Fri Nov 2 00:26:55 2012 UTC vs.
Revision 2.54 by greg, Fri Jun 6 19:11:21 2025 UTC

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

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines