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

Comparing src/util/dctimestep.c (file contents):
Revision 2.17 by greg, Fri Apr 15 18:43:57 2011 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  
19 < /* 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 */
19 > /* Sum together a set of images and write result to fout */
20   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 – /* Multiply two matrices (or a matrix and a vector) and allocate the result*/
256 – static CMATRIX *
257 – cm_multiply(const CMATRIX *cm1, const CMATRIX *cm2)
258 – {
259 –        CMATRIX *cmr;
260 –        int     dr, dc, i;
261 –
262 –        if ((cm1->ncols <= 0) | (cm1->ncols != cm2->nrows))
263 –                error(INTERNAL, "matrix dimension mismatch in cm_multiply()");
264 –        cmr = cm_alloc(cm1->nrows, cm2->ncols);
265 –        if (cmr == NULL)
266 –                return(NULL);
267 –        for (dr = 0; dr < cmr->nrows; dr++)
268 –            for (dc = 0; dc < cmr->ncols; dc++) {
269 –                COLORV  *dp = cm_lval(cmr,dr,dc);
270 –                dp[0] = dp[1] = dp[2] = 0;
271 –                for (i = 0; i < cm1->ncols; i++) {
272 –                    const COLORV        *cp1 = cm_lval(cm1,dr,i);
273 –                    const COLORV        *cp2 = cm_lval(cm2,i,dc);
274 –                    dp[0] += cp1[0] * cp2[0];
275 –                    dp[1] += cp1[1] * cp2[1];
276 –                    dp[2] += cp1[2] * cp2[2];
277 –                }
278 –            }
279 –        return(cmr);
280 – }
281 –
282 – /* print out matrix as ASCII text -- no header */
283 – static void
284 – cm_print(const CMATRIX *cm, FILE *fp)
285 – {
286 –        int             r, c;
287 –        const COLORV    *mp = cm->cmem;
288 –        
289 –        for (r = 0; r < cm->nrows; r++) {
290 –                for (c = 0; c < cm->ncols; c++, mp += 3)
291 –                        fprintf(fp, "\t%.6e %.6e %.6e", mp[0], mp[1], mp[2]);
292 –                fputc('\n', fp);
293 –        }
294 – }
295 –
296 – /* Convert a BSDF to our matrix representation */
297 – static CMATRIX *
298 – cm_bsdf(const COLOR bsdfLamb, const COLOR specCol, const SDMat *bsdf)
299 – {
300 –        CMATRIX *cm = cm_alloc(bsdf->nout, bsdf->ninc);
301 –        int     nbadohm = 0;
302 –        int     nneg = 0;
303 –        int     r, c;
304 –                                        /* reciprocity is "transparent" */
305 –        for (c = 0; c < cm->ncols; c++) {
306 –                const double    dom = mBSDF_incohm(bsdf,c);
307 –                                        /* projected solid angle */
308 –                nbadohm += (dom <= 0);
309 –
310 –                for (r = 0; r < cm->nrows; r++) {
311 –                        float   f = mBSDF_value(bsdf,c,r);
312 –                        COLORV  *mp = cm_lval(cm,r,c);
313 –                                        /* check BSDF value */
314 –                        if ((f <= 0) | (dom <= 0)) {
315 –                                nneg += (f < -FTINY);
316 –                                f = .0f;
317 –                        }
318 –                        copycolor(mp, specCol);
319 –                        scalecolor(mp, f);
320 –                        addcolor(mp, bsdfLamb);
321 –                        scalecolor(mp, dom);
322 –                }
323 –        }
324 –        if (nneg | nbadohm) {
325 –                sprintf(errmsg,
326 –                    "BTDF has %d negatives and %d bad incoming solid angles",
327 –                                nneg, nbadohm);
328 –                error(WARNING, errmsg);
329 –        }
330 –        return(cm);
331 – }
332 –
333 – /* Load and convert a matrix BSDF from the given XML file */
334 – static CMATRIX *
335 – cm_loadBSDF(char *fname)
336 – {
337 –        CMATRIX *Tmat;
338 –        char    *fpath;
339 –        SDError ec;
340 –        SDData  myBSDF;
341 –        COLOR   bsdfLamb, specCol;
342 –                                        /* find path to BSDF file */
343 –        fpath = getpath(fname, getrlibpath(), R_OK);
344 –        if (fpath == NULL) {
345 –                sprintf(errmsg, "cannot find BSDF file '%s'", fname);
346 –                error(USER, errmsg);
347 –        }
348 –        SDclearBSDF(&myBSDF, fname);    /* load XML and check type */
349 –        ec = SDloadFile(&myBSDF, fpath);
350 –        if (ec)
351 –                error(USER, transSDError(ec));
352 –        if (myBSDF.tf == NULL || myBSDF.tf->ncomp != 1 ||
353 –                        myBSDF.tf->comp[0].func != &SDhandleMtx) {
354 –                sprintf(errmsg, "unsupported BSDF '%s'", fpath);
355 –                error(USER, errmsg);
356 –        }
357 –                                        /* convert BTDF to matrix */
358 –        ccy2rgb(&myBSDF.tLamb.spec, myBSDF.tLamb.cieY/PI, bsdfLamb);
359 –        ccy2rgb(&myBSDF.tf->comp[0].cspec[0], 1., specCol);
360 –        Tmat = cm_bsdf(bsdfLamb, specCol, (SDMat *)myBSDF.tf->comp[0].dist);
361 –                                        /* free BSDF and return */
362 –        SDfreeBSDF(&myBSDF);
363 –        return(Tmat);
364 – }
365 –
366 – /* Sum together a set of images and write result to stdout */
367 – static int
21   sum_images(const char *fspec, const CMATRIX *cv, FILE *fout)
22   {
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)
# Line 381 | 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))
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 400 | 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);
411 –                        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 417 | 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)
438 <                        return(0);
439 <        cm_free(pmat);                                  /* all done */
440 <        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 */
100   static int
101   hasNumberFormat(const char *s)
102   {
103 <        while (*s && *s != '%')
448 <                s++;
449 <        if (!*s)
103 >        if (s == NULL)
104                  return(0);
451 –        do
452 –                ++s;
453 –        while (isdigit(*s));
105  
106 <        return((*s == 'd') | (*s == 'i') | (*s == 'o') |
107 <                        (*s == 'x') | (*s == 'X'));
106 >        while (*s) {
107 >                while (*s != '%')
108 >                        if (!*s++)
109 >                                return(0);
110 >                if (*++s == '%') {              /* ignore "%%" */
111 >                        ++s;
112 >                        continue;
113 >                }
114 >                while (isdigit(*s))             /* field length */
115 >                        ++s;
116 >                                                /* field we'll use? */
117 >                if ((*s == 'd') | (*s == 'i') | (*s == 'o') |
118 >                                        (*s == 'x') | (*s == 'X'))
119 >                        return(1);
120 >        }
121 >        return(0);                              /* didn't find one */
122   }
123  
124   int
125   main(int argc, char *argv[])
126   {
127 <        CMATRIX                 *cvec;
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 */
134 >        char            fnbuf[256];
135 >        int             a, i;
136  
137          progname = argv[0];
138 +                                        /* get options */
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)
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 +                        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;
185 +                }
186 +        if ((argc-a < 1) | (argc-a > 4))
187 +                goto userr;
188  
189 <        if ((argc < 2) | (argc > 5)) {
190 <                fprintf(stderr, "Usage: %s DCspec [tregvec]\n", progname);
191 <                fprintf(stderr, "   or: %s Vspec Tbsdf.xml Dmat.dat [tregvec]\n",
192 <                                        progname);
193 <                return(1);
194 <        }
472 <
473 <        if (argc > 3) {                         /* VTDs expression */
474 <                CMATRIX *svec, *Dmat, *Tmat, *ivec;
475 <                                                /* get sky vector */
476 <                svec = cm_load(argv[4], 0, 1, DTascii);
189 >        if (argc-a > 2) {                       /* VTDs expression */
190 >                CMATRIX         *smtx, *Dmat, *Tmat, *imtx;
191 >                const char      *ccp;
192 >                                                /* get sky vector/matrix */
193 >                smtx = cm_load(argv[a+3], 0, nsteps, skyfmt);
194 >                nsteps = smtx->ncols;
195                                                  /* load BSDF */
196 <                Tmat = cm_loadBSDF(argv[2]);
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[3], Tmat->ncols, svec->nrows, DTfromHeader);
203 >                Dmat = cm_load(argv[a+2], Tmat->ncols,
204 >                                        smtx->nrows, DTfromHeader);
205                                                  /* multiply vector through */
206 <                ivec = cm_multiply(Dmat, svec);
207 <                cm_free(Dmat); cm_free(svec);
208 <                cvec = cm_multiply(Tmat, ivec); /* cvec = component vector */
209 <                cm_free(Tmat); cm_free(ivec);
210 <        } else {                                /* else just use sky vector */
211 <                cvec = cm_load(argv[2], 0, 1, DTascii);
206 >                imtx = cm_multiply(Dmat, smtx);
207 >                cm_free(Dmat); cm_free(smtx);
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, skyfmt);
213 >                nsteps = cmtx->ncols;
214          }
215 <
216 <        if (hasNumberFormat(argv[1])) {         /* generating image */
217 <                SET_FILE_BINARY(stdout);
218 <                newheader("RADIANCE", stdout);
219 <                printargs(argc, argv, stdout);
220 <                fputnow(stdout);
221 <                if (!sum_images(argv[1], cvec, stdout))
215 >                                                /* prepare output stream */
216 >        if ((ofspec != NULL) & (nsteps == 1) && hasNumberFormat(ofspec)) {
217 >                sprintf(fnbuf, ofspec, 1);
218 >                ofspec = fnbuf;
219 >        }
220 >        if (ofspec != NULL && !hasNumberFormat(ofspec)) {
221 >                if ((ofp = fopen(ofspec, "w")) == NULL) {
222 >                        fprintf(stderr, "%s: cannot open '%s' for output\n",
223 >                                        progname, ofspec);
224                          return(1);
225 <        } else {                                /* generating vector */
226 <                CMATRIX *Vmat = cm_load(argv[1], 0, cvec->nrows, DTfromHeader);
227 <                CMATRIX *rvec = cm_multiply(Vmat, cvec);
225 >                }
226 >                ofspec = NULL;                  /* only need to open once */
227 >        }
228 >        if (hasNumberFormat(argv[a])) {         /* generating image(s) */
229 >                if (ofspec == NULL) {
230 >                        SET_FILE_BINARY(ofp);
231 >                        newheader("RADIANCE", ofp);
232 >                        printargs(argc, argv, ofp);
233 >                        fputnow(ofp);
234 >                }
235 >                if (nsteps > 1)                 /* multiple output frames? */
236 >                        for (i = 0; i < nsteps; i++) {
237 >                                CMATRIX *cvec = cm_column(cmtx, i);
238 >                                if (ofspec != NULL) {
239 >                                        sprintf(fnbuf, ofspec, i+1);
240 >                                        if ((ofp = fopen(fnbuf, "wb")) == NULL) {
241 >                                                fprintf(stderr,
242 >                                        "%s: cannot open '%s' for output\n",
243 >                                                        progname, fnbuf);
244 >                                                return(1);
245 >                                        }
246 >                                        newheader("RADIANCE", ofp);
247 >                                        printargs(argc, argv, ofp);
248 >                                        fputnow(ofp);
249 >                                }
250 >                                fprintf(ofp, "FRAME=%d\n", i+1);
251 >                                if (!sum_images(argv[a], cvec, ofp))
252 >                                        return(1);
253 >                                if (ofspec != NULL) {
254 >                                        if (fclose(ofp) == EOF) {
255 >                                                fprintf(stderr,
256 >                                                "%s: error writing to '%s'\n",
257 >                                                        progname, fnbuf);
258 >                                                return(1);
259 >                                        }
260 >                                        ofp = stdout;
261 >                                }
262 >                                cm_free(cvec);
263 >                        }
264 >                else if (!sum_images(argv[a], cmtx, ofp))
265 >                        return(1);
266 >        } else {                                /* generating vector/matrix */
267 >                CMATRIX *Vmat = cm_load(argv[a], 0, cmtx->nrows, DTfromHeader);
268 >                CMATRIX *rmtx = cm_multiply(Vmat, cmtx);
269                  cm_free(Vmat);
270 <                cm_print(rvec, stdout);
271 <                cm_free(rvec);
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, wtype)) == NULL) {
276 >                                        fprintf(stderr,
277 >                                        "%s: cannot open '%s' for output\n",
278 >                                                        progname, fnbuf);
279 >                                        return(1);
280 >                                }
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",
298 >                                                        progname, fnbuf);
299 >                                        return(1);
300 >                                }
301 >                                ofp = stdout;
302 >                                cm_free(rvec);
303 >                        }
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 <        cm_free(cvec);                          /* final clean-up */
324 >        if (fflush(ofp) == EOF) {               /* final clean-up */
325 >                fprintf(stderr, "%s: write error on output\n", progname);
326 >                return(1);
327 >        }
328 >        cm_free(cmtx);
329          return(0);
330 + userr:
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][-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