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.4 by greg, Fri Jun 19 14:24:06 2009 UTC vs.
Revision 2.35 by greg, Fri Sep 26 23:33:09 2014 UTC

# Line 9 | Line 9 | static const char RCSid[] = "$Id$";
9  
10   #include <ctype.h>
11   #include "standard.h"
12 + #include "cmatrix.h"
13   #include "platform.h"
13 – #include "color.h"
14   #include "resolu.h"
15 – #include "bsdf.h"
15  
16   char    *progname;                      /* global argv[0] */
17  
18 < /* Data types for file loading */
20 < enum {DTfromHeader, DTascii, DTfloat, DTdouble, DTrgbe, DTxyze};
21 <
22 < /* A color coefficient matrix -- vectors have ncols==1 */
23 < typedef struct {
24 <        int     nrows, ncols;
25 <        COLORV  cmem[3];                /* extends struct */
26 < } CMATRIX;
27 <
28 < #define COLSPEC (sizeof(COLORV)==sizeof(float) ? "%f %f %f" : "%lf %lf %lf")
29 <
30 < #define cm_lval(cm,r,c) ((cm)->cmem + 3*((r)*(cm)->ncols + (c)))
31 <
32 < #define cv_lval(cm,i)   ((cm)->cmem + 3*(i))
33 <
34 < /* Allocate a color coefficient matrix */
35 < static CMATRIX *
36 < cm_alloc(int nrows, int ncols)
37 < {
38 <        CMATRIX *cm;
39 <
40 <        if ((nrows <= 0) | (ncols <= 0))
41 <                return(NULL);
42 <        cm = (CMATRIX *)malloc(sizeof(CMATRIX) +
43 <                                3*sizeof(COLORV)*(nrows*ncols - 1));
44 <        if (cm == NULL)
45 <                error(SYSTEM, "out of memory in cm_alloc()");
46 <        cm->nrows = nrows;
47 <        cm->ncols = ncols;
48 <        return(cm);
49 < }
50 <
51 < #define cm_free(cm)     free(cm)
52 <
53 < /* Resize color coefficient matrix */
54 < static CMATRIX *
55 < cm_resize(CMATRIX *cm, int nrows)
56 < {
57 <        if (nrows == cm->nrows)
58 <                return(cm);
59 <        if (nrows <= 0) {
60 <                cm_free(cm);
61 <                return(NULL);
62 <        }
63 <        cm = (CMATRIX *)realloc(cm, sizeof(CMATRIX) +
64 <                        3*sizeof(COLORV)*(nrows*cm->ncols - 1));
65 <        if (cm == NULL)
66 <                error(SYSTEM, "out of memory in cm_resize()");
67 <        cm->nrows = nrows;
68 <        return(cm);
69 < }
70 <
71 < /* Load header to obtain data type */
18 > /* Sum together a set of images and write result to fout */
19   static int
20 < getDT(char *s, void *p)
20 > sum_images(const char *fspec, const CMATRIX *cv, FILE *fout)
21   {
75 –        char    fmt[32];
76 –        
77 –        if (formatval(fmt, s)) {
78 –                if (!strcmp(fmt, "ascii"))
79 –                        *((int *)p) = DTascii;
80 –                else if (!strcmp(fmt, "float"))
81 –                        *((int *)p) = DTfloat;
82 –                else if (!strcmp(fmt, "double"))
83 –                        *((int *)p) = DTdouble;
84 –                else if (!strcmp(fmt, COLRFMT))
85 –                        *((int *)p) = DTrgbe;
86 –                else if (!strcmp(fmt, CIEFMT))
87 –                        *((int *)p) = DTxyze;
88 –        }
89 –        return(0);
90 – }
91 –
92 – static int
93 – getDTfromHeader(FILE *fp)
94 – {
95 –        int     dt = DTfromHeader;
96 –        
97 –        if (getheader(fp, getDT, &dt) < 0)
98 –                error(SYSTEM, "header read error");
99 –        if (dt == DTfromHeader)
100 –                error(USER, "missing data format in header");
101 –        return(dt);
102 – }
103 –
104 – /* Allocate and load a matrix from the given file (or stdin if NULL) */
105 – static CMATRIX *
106 – cm_load(const char *fname, int nrows, int ncols, int dtype)
107 – {
108 –        CMATRIX *cm;
109 –        FILE    *fp = stdin;
110 –
111 –        if (fname == NULL)
112 –                fname = "<stdin>";
113 –        else if ((fp = fopen(fname, "r")) == NULL) {
114 –                sprintf(errmsg, "cannot open file '%s'", fname);
115 –                error(SYSTEM, errmsg);
116 –        }
117 –        if (dtype != DTascii)
118 –                SET_FILE_BINARY(fp);
119 –        if (dtype == DTfromHeader)
120 –                dtype = getDTfromHeader(fp);
121 –        switch (dtype) {
122 –        case DTascii:
123 –        case DTfloat:
124 –        case DTdouble:
125 –                break;
126 –        default:
127 –                error(USER, "unexpected data type in cm_load()");
128 –        }
129 –        if (nrows <= 0) {                       /* don't know length? */
130 –                int     guessrows = 147;        /* usually big enough */
131 –                if ((dtype != DTascii) & (fp != stdin)) {
132 –                        long    startpos = ftell(fp);
133 –                        if (fseek(fp, 0L, SEEK_END) == 0) {
134 –                                long    endpos = ftell(fp);
135 –                                long    elemsiz = 3*(dtype==DTfloat ?
136 –                                            sizeof(float) : sizeof(double));
137 –
138 –                                if ((endpos - startpos) % (ncols*elemsiz)) {
139 –                                        sprintf(errmsg,
140 –                                        "improper length for binary file '%s'",
141 –                                                        fname);
142 –                                        error(USER, errmsg);
143 –                                }
144 –                                guessrows = (endpos - startpos)/(ncols*elemsiz);
145 –                                if (fseek(fp, startpos, SEEK_SET) < 0) {
146 –                                        sprintf(errmsg,
147 –                                                "fseek() error on file '%s'",
148 –                                                        fname);
149 –                                        error(SYSTEM, errmsg);
150 –                                }
151 –                                nrows = guessrows;      /* we're confident */
152 –                        }
153 –                }
154 –                cm = cm_alloc(guessrows, ncols);
155 –        } else
156 –                cm = cm_alloc(nrows, ncols);
157 –        if (cm == NULL)
158 –                return(NULL);
159 –        if (dtype == DTascii) {                         /* read text file */
160 –                int     maxrow = (nrows > 0 ? nrows : 32000);
161 –                int     r, c;
162 –                for (r = 0; r < maxrow; r++) {
163 –                    if (r >= cm->nrows)         /* need more space? */
164 –                        cm = cm_resize(cm, 2*cm->nrows);
165 –                    for (c = 0; c < ncols; c++) {
166 –                        COLORV  *cv = cm_lval(cm,r,c);
167 –                        if (fscanf(fp, COLSPEC, cv, cv+1, cv+2) != 3)
168 –                                if ((nrows <= 0) & (r > 0) & (c == 0)) {
169 –                                        cm = cm_resize(cm, maxrow=r);
170 –                                        break;
171 –                                } else
172 –                                        goto EOFerror;
173 –                    }
174 –                }
175 –                while ((c = getc(fp)) != EOF)
176 –                        if (!isspace(c)) {
177 –                                sprintf(errmsg,
178 –                                "unexpected data at end of ascii file %s",
179 –                                                fname);
180 –                                error(WARNING, errmsg);
181 –                                break;
182 –                        }
183 –        } else {                                        /* read binary file */
184 –                if (sizeof(COLORV) == (dtype==DTfloat ? sizeof(float) :
185 –                                                        sizeof(double))) {
186 –                        int     nread = 0;
187 –                        do {                            /* read all we can */
188 –                                nread += fread(cm->cmem + 3*nread,
189 –                                                3*sizeof(COLORV),
190 –                                                cm->nrows*cm->ncols - nread,
191 –                                                fp);
192 –                                if (nrows <= 0) {       /* unknown length */
193 –                                        if (nread == cm->nrows*cm->ncols)
194 –                                                        /* need more space? */
195 –                                                cm = cm_resize(cm, 2*cm->nrows);
196 –                                        else if (nread % cm->ncols == 0)
197 –                                                        /* seem to be  done */
198 –                                                cm = cm_resize(cm, nread/cm->ncols);
199 –                                        else            /* ended mid-row */
200 –                                                goto EOFerror;
201 –                                } else if (nread < cm->nrows*cm->ncols)
202 –                                        goto EOFerror;
203 –                        } while (nread < cm->nrows*cm->ncols);
204 –
205 –                } else if (dtype == DTdouble) {
206 –                        double  dc[3];                  /* load from double */
207 –                        COLORV  *cvp = cm->cmem;
208 –                        int     n = nrows*ncols;
209 –
210 –                        if (n <= 0)
211 –                                goto not_handled;
212 –                        while (n--) {
213 –                                if (fread(dc, sizeof(double), 3, fp) != 3)
214 –                                        goto EOFerror;
215 –                                copycolor(cvp, dc);
216 –                                cvp += 3;
217 –                        }
218 –                } else /* dtype == DTfloat */ {
219 –                        float   fc[3];                  /* load from float */
220 –                        COLORV  *cvp = cm->cmem;
221 –                        int     n = nrows*ncols;
222 –
223 –                        if (n <= 0)
224 –                                goto not_handled;
225 –                        while (n--) {
226 –                                if (fread(fc, sizeof(float), 3, fp) != 3)
227 –                                        goto EOFerror;
228 –                                copycolor(cvp, fc);
229 –                                cvp += 3;
230 –                        }
231 –                }
232 –                if (getc(fp) != EOF) {
233 –                                sprintf(errmsg,
234 –                                "unexpected data at end of binary file %s",
235 –                                                fname);
236 –                                error(WARNING, errmsg);
237 –                }
238 –        }
239 –        if (fp != stdin)
240 –                fclose(fp);
241 –        return(cm);
242 – EOFerror:
243 –        sprintf(errmsg, "unexpected EOF reading %s",
244 –                        fname);
245 –        error(USER, errmsg);
246 – not_handled:
247 –        error(INTERNAL, "unhandled data size or length in cm_load()");
248 –        return(NULL);   /* gratis return */
249 – }
250 –
251 – /* Multiply two matrices (or a matrix and a vector) and allocate the result*/
252 – static CMATRIX *
253 – cm_multiply(const CMATRIX *cm1, const CMATRIX *cm2)
254 – {
255 –        CMATRIX *cmr;
256 –        int     dr, dc, i;
257 –
258 –        if ((cm1->ncols <= 0) | (cm1->ncols != cm2->nrows))
259 –                error(INTERNAL, "matrix dimension mismatch in cm_multiply()");
260 –        cmr = cm_alloc(cm1->nrows, cm2->ncols);
261 –        if (cmr == NULL)
262 –                return(NULL);
263 –        for (dr = 0; dr < cmr->nrows; dr++)
264 –            for (dc = 0; dc < cmr->ncols; dc++) {
265 –                COLORV  *dp = cm_lval(cmr,dr,dc);
266 –                dp[0] = dp[1] = dp[2] = 0;
267 –                for (i = 0; i < cm1->ncols; i++) {
268 –                    const COLORV        *cp1 = cm_lval(cm1,dr,i);
269 –                    const COLORV        *cp2 = cm_lval(cm2,i,dc);
270 –                    dp[0] += cp1[0] * cp2[0];
271 –                    dp[1] += cp1[1] * cp2[1];
272 –                    dp[2] += cp1[2] * cp2[2];
273 –                }
274 –            }
275 –        return(cmr);
276 – }
277 –
278 – /* print out matrix as ASCII text -- no header */
279 – static void
280 – cm_print(const CMATRIX *cm, FILE *fp)
281 – {
282 –        int             r, c;
283 –        const COLORV    *mp = cm->cmem;
284 –        
285 –        for (r = 0; r < cm->nrows; r++) {
286 –                for (c = 0; c < cm->ncols; c++, mp += 3)
287 –                        fprintf(fp, "\t%.6e %.6e %.6e", mp[0], mp[1], mp[2]);
288 –                fputc('\n', fp);
289 –        }
290 – }
291 –
292 – /* convert a BSDF to our matrix representation */
293 – static CMATRIX *
294 – cm_bsdf(const struct BSDF_data *bsdf)
295 – {
296 –        CMATRIX *cm = cm_alloc(bsdf->nout, bsdf->ninc);
297 –        COLORV  *mp = cm->cmem;
298 –        int     nbadohm = 0;
299 –        int     nneg = 0;
300 –        int     r, c;
301 –        
302 –        for (r = 0; r < cm->nrows; r++)
303 –                for (c = 0; c < cm->ncols; c++, mp += 3) {
304 –                        float   f = BSDF_value(bsdf,c,r);
305 –                        float   dom = getBSDF_incohm(bsdf,c);
306 –                        FVECT   v;
307 –                        
308 –                        if (f <= .0) {
309 –                                nneg += (f < -FTINY);
310 –                                continue;
311 –                        }
312 –                        if (dom <= .0) {
313 –                                nbadohm++;
314 –                                continue;
315 –                        }
316 –                        if (!getBSDF_incvec(v,bsdf,c) || v[2] > FTINY)
317 –                                error(USER, "illegal incoming BTDF direction");
318 –                                
319 –                        mp[0] = mp[1] = mp[2] = f * dom * -v[2];
320 –                }
321 –        if (nneg || nbadohm) {
322 –                sprintf(errmsg,
323 –                    "BTDF has %d negatives and %d bad incoming solid angles",
324 –                                nneg, nbadohm);
325 –                error(WARNING, errmsg);
326 –        }
327 –        return(cm);
328 – }
329 –
330 – /* Sum together a set of images and write result to stdout */
331 – static int
332 – sum_images(const char *fspec, const CMATRIX *cv)
333 – {
22          int     myDT = DTfromHeader;
23 <        CMATRIX *pmat;
24 <        COLOR   *scanline;
25 <        int     myXR, myYR;
23 >        COLOR   *scanline = NULL;
24 >        CMATRIX *pmat = NULL;
25 >        int     myXR=0, myYR=0;
26          int     i, y;
27  
28          if (cv->ncols != 1)
# Line 345 | Line 33 | sum_images(const char *fspec, const CMATRIX *cv)
33                  FILE            *fp;
34                  int             dt, xr, yr;
35                  COLORV          *psp;
36 +                char            *err;
37 +                                                        /* check for zero */
38 +                if ((scv[RED] == 0) & (scv[GRN] == 0) & (scv[BLU] == 0) &&
39 +                                (myDT != DTfromHeader) | (i < cv->nrows-1))
40 +                        continue;
41                                                          /* open next picture */
42                  sprintf(fname, fspec, i);
43 <                if ((fp = fopen(fname, "r")) == NULL) {
43 >                if ((fp = fopen(fname, "rb")) == NULL) {
44                          sprintf(errmsg, "cannot open picture '%s'", fname);
45                          error(SYSTEM, errmsg);
46                  }
47 <                SET_FILE_BINARY(fp);
48 <                dt = getDTfromHeader(fp);
47 >                dt = DTfromHeader;
48 >                if ((err = cm_getheader(&dt, NULL, NULL, fp)) != NULL)
49 >                        error(USER, err);
50                  if ((dt != DTrgbe) & (dt != DTxyze) ||
51                                  !fscnresolu(&xr, &yr, fp)) {
52                          sprintf(errmsg, "file '%s' not a picture", fname);
# Line 367 | Line 61 | sum_images(const char *fspec, const CMATRIX *cv)
61                          pmat = cm_alloc(myYR, myXR);
62                          memset(pmat->cmem, 0, sizeof(COLOR)*myXR*myYR);
63                                                          /* finish header */
64 <                        fputformat(myDT==DTrgbe ? COLRFMT : CIEFMT, stdout);
65 <                        fputc('\n', stdout);
66 <                        fprtresolu(myXR, myYR, stdout);
373 <                        fflush(stdout);
64 >                        fputformat((char *)cm_fmt_id[myDT], fout);
65 >                        fputc('\n', fout);
66 >                        fflush(fout);
67                  } else if ((dt != myDT) | (xr != myXR) | (yr != myYR)) {
68                          sprintf(errmsg, "picture '%s' format/size mismatch",
69                                          fname);
# Line 393 | Line 86 | sum_images(const char *fspec, const CMATRIX *cv)
86                  fclose(fp);                             /* done this picture */
87          }
88          free(scanline);
89 <                                                        /* write scanlines */
90 <        for (y = 0; y < myYR; y++)
91 <                if (fwritescan((COLOR *)cm_lval(pmat, y, 0), myXR, stdout) < 0)
399 <                        return(0);
400 <        cm_free(pmat);                                  /* all done */
401 <        return(fflush(stdout) == 0);
89 >        i = cm_write(pmat, myDT, fout);                 /* write picture */
90 >        cm_free(pmat);                                  /* free data */
91 >        return(i);
92   }
93  
94 < /* check to see if a string contains a %d specification */
95 < int
96 < hasDecimalSpec(const char *s)
94 > /* check to see if a string contains a %d or %o specification */
95 > static int
96 > hasNumberFormat(const char *s)
97   {
98 <        while (*s && *s != '%')
409 <                s++;
410 <        if (!*s)
98 >        if (s == NULL)
99                  return(0);
412 –        do
413 –                ++s;
414 –        while (isdigit(*s));
100  
101 <        return(*s == 'd');
101 >        while (*s) {
102 >                while (*s != '%')
103 >                        if (!*s++)
104 >                                return(0);
105 >                if (*++s == '%') {              /* ignore "%%" */
106 >                        ++s;
107 >                        continue;
108 >                }
109 >                while (isdigit(*s))             /* field length */
110 >                        ++s;
111 >                                                /* field we'll use? */
112 >                if ((*s == 'd') | (*s == 'i') | (*s == 'o') |
113 >                                        (*s == 'x') | (*s == 'X'))
114 >                        return(1);
115 >        }
116 >        return(0);                              /* didn't find one */
117   }
118  
119   int
120   main(int argc, char *argv[])
121   {
122 <        CMATRIX                 *tvec, *Dmat, *Tmat, *ivec, *cvec;
123 <        struct BSDF_data        *btdf;
122 >        int             skyfmt = DTfromHeader;
123 >        int             outfmt = DTascii;
124 >        int             headout = 1;
125 >        int             nsteps = 0;
126 >        char            *ofspec = NULL;
127 >        FILE            *ofp = stdout;
128 >        CMATRIX         *cmtx;          /* component vector/matrix result */
129 >        char            fnbuf[256];
130 >        int             a, i;
131  
132          progname = argv[0];
133 +                                        /* get options */
134 +        for (a = 1; a < argc && argv[a][0] == '-'; a++)
135 +                switch (argv[a][1]) {
136 +                case 'n':
137 +                        nsteps = atoi(argv[++a]);
138 +                        if (nsteps < 0)
139 +                                goto userr;
140 +                        skyfmt = nsteps ? DTascii : DTfromHeader;
141 +                        break;
142 +                case 'h':
143 +                        headout = !headout;
144 +                        break;
145 +                case 'i':
146 +                        switch (argv[a][2]) {
147 +                        case 'f':
148 +                                skyfmt = DTfloat;
149 +                                break;
150 +                        case 'd':
151 +                                skyfmt = DTdouble;
152 +                                break;
153 +                        case 'a':
154 +                                skyfmt = DTascii;
155 +                                break;
156 +                        default:
157 +                                goto userr;
158 +                        }
159 +                        break;
160 +                case 'o':
161 +                        switch (argv[a][2]) {
162 +                        case '\0':      /* output specification (not format) */
163 +                                ofspec = argv[++a];
164 +                                break;
165 +                        case 'f':
166 +                                outfmt = DTfloat;
167 +                                break;
168 +                        case 'd':
169 +                                outfmt = DTdouble;
170 +                                break;
171 +                        case 'a':
172 +                                outfmt = DTascii;
173 +                                break;
174 +                        default:
175 +                                goto userr;
176 +                        }
177 +                        break;
178 +                default:
179 +                        goto userr;
180 +                }
181 +        if ((argc-a < 1) | (argc-a > 4))
182 +                goto userr;
183  
184 <        if ((argc < 4) | (argc > 5)) {
185 <                fprintf(stderr, "Usage: %s Vspec Tbsdf.xml Dmat.dat [tregvec]\n",
186 <                                progname);
187 <                return(1);
184 >        if (argc-a > 2) {                       /* VTDs expression */
185 >                CMATRIX *smtx, *Dmat, *Tmat, *imtx;
186 >                                                /* get sky vector/matrix */
187 >                smtx = cm_load(argv[a+3], 0, nsteps, skyfmt);
188 >                nsteps = smtx->ncols;
189 >                                                /* load BSDF */
190 >                Tmat = cm_loadBTDF(argv[a+1]);
191 >                                                /* load Daylight matrix */
192 >                Dmat = cm_load(argv[a+2], Tmat==NULL ? 0 : Tmat->ncols,
193 >                                        smtx->nrows, DTfromHeader);
194 >                                                /* multiply vector through */
195 >                imtx = cm_multiply(Dmat, smtx);
196 >                cm_free(Dmat); cm_free(smtx);
197 >                cmtx = cm_multiply(Tmat, imtx);
198 >                cm_free(Tmat);
199 >                cm_free(imtx);
200 >        } else {                                /* sky vector/matrix only */
201 >                cmtx = cm_load(argv[a+1], 0, nsteps, skyfmt);
202 >                nsteps = cmtx->ncols;
203          }
204 <        tvec = cm_load(argv[4], 0, 1, DTascii); /* argv[4]==NULL iff argc==4 */
205 <        Dmat = cm_load(argv[3], 0, tvec->nrows, DTfromHeader);
206 <        btdf = load_BSDF(argv[2]);
207 <        if (btdf == NULL)
436 <                return(1);
437 <        if (btdf->ninc != Dmat->nrows) {
438 <                sprintf(errmsg, "Incoming BTDF dir (%d) mismatch to D (%d)",
439 <                                btdf->ninc, Dmat->nrows);
440 <                error(USER, errmsg);
204 >                                                /* prepare output stream */
205 >        if ((ofspec != NULL) & (nsteps == 1) && hasNumberFormat(ofspec)) {
206 >                sprintf(fnbuf, ofspec, 1);
207 >                ofspec = fnbuf;
208          }
209 <                                                /* multiply vector through */
210 <        ivec = cm_multiply(Dmat, tvec);
211 <        cm_free(Dmat); cm_free(tvec);
212 <        Tmat = cm_bsdf(btdf);                   /* convert BTDF to matrix */
446 <        free_BSDF(btdf);
447 <        cvec = cm_multiply(Tmat, ivec);         /* cvec = component vector */
448 <        cm_free(Tmat); cm_free(ivec);
449 <        if (hasDecimalSpec(argv[1])) {          /* generating image */
450 <                SET_FILE_BINARY(stdout);
451 <                newheader("RADIANCE", stdout);
452 <                printargs(argc, argv, stdout);
453 <                fputnow(stdout);
454 <                if (!sum_images(argv[1], cvec))
209 >        if (ofspec != NULL && !hasNumberFormat(ofspec)) {
210 >                if ((ofp = fopen(ofspec, "w")) == NULL) {
211 >                        fprintf(stderr, "%s: cannot open '%s' for output\n",
212 >                                        progname, ofspec);
213                          return(1);
214 <        } else {                                /* generating vector */
215 <                CMATRIX *Vmat = cm_load(argv[1], 0, cvec->nrows, DTfromHeader);
216 <                CMATRIX *rvec = cm_multiply(Vmat, cvec);
214 >                }
215 >                ofspec = NULL;                  /* only need to open once */
216 >        }
217 >        if (hasNumberFormat(argv[a])) {         /* generating image(s) */
218 >                if (ofspec == NULL) {
219 >                        SET_FILE_BINARY(ofp);
220 >                        newheader("RADIANCE", ofp);
221 >                        printargs(argc, argv, ofp);
222 >                        fputnow(ofp);
223 >                }
224 >                if (nsteps > 1)                 /* multiple output frames? */
225 >                        for (i = 0; i < nsteps; i++) {
226 >                                CMATRIX *cvec = cm_column(cmtx, i);
227 >                                if (ofspec != NULL) {
228 >                                        sprintf(fnbuf, ofspec, i+1);
229 >                                        if ((ofp = fopen(fnbuf, "wb")) == NULL) {
230 >                                                fprintf(stderr,
231 >                                        "%s: cannot open '%s' for output\n",
232 >                                                        progname, fnbuf);
233 >                                                return(1);
234 >                                        }
235 >                                        newheader("RADIANCE", ofp);
236 >                                        printargs(argc, argv, ofp);
237 >                                        fputnow(ofp);
238 >                                }
239 >                                fprintf(ofp, "FRAME=%d\n", i+1);
240 >                                if (!sum_images(argv[a], cvec, ofp))
241 >                                        return(1);
242 >                                if (ofspec != NULL) {
243 >                                        if (fclose(ofp) == EOF) {
244 >                                                fprintf(stderr,
245 >                                                "%s: error writing to '%s'\n",
246 >                                                        progname, fnbuf);
247 >                                                return(1);
248 >                                        }
249 >                                        ofp = stdout;
250 >                                }
251 >                                cm_free(cvec);
252 >                        }
253 >                else if (!sum_images(argv[a], cmtx, ofp))
254 >                        return(1);
255 >        } else {                                /* generating vector/matrix */
256 >                CMATRIX *Vmat = cm_load(argv[a], 0, cmtx->nrows, DTfromHeader);
257 >                CMATRIX *rmtx = cm_multiply(Vmat, cmtx);
258                  cm_free(Vmat);
259 <                cm_print(rvec, stdout);
260 <                cm_free(rvec);
259 >                if (ofspec != NULL) {           /* multiple vector files? */
260 >                        const char      *wtype = (outfmt==DTascii) ? "w" : "wb";
261 >                        for (i = 0; i < nsteps; i++) {
262 >                                CMATRIX *rvec = cm_column(rmtx, i);
263 >                                sprintf(fnbuf, ofspec, i+1);
264 >                                if ((ofp = fopen(fnbuf, wtype)) == NULL) {
265 >                                        fprintf(stderr,
266 >                                        "%s: cannot open '%s' for output\n",
267 >                                                        progname, fnbuf);
268 >                                        return(1);
269 >                                }
270 > #ifdef getc_unlocked
271 >                                flockfile(ofp);
272 > #endif
273 >                                if (headout) {  /* header output */
274 >                                        newheader("RADIANCE", ofp);
275 >                                        printargs(argc, argv, ofp);
276 >                                        fputnow(ofp);
277 >                                        fprintf(ofp, "FRAME=%d\n", i+1);
278 >                                        fprintf(ofp, "NROWS=%d\n", rvec->nrows);
279 >                                        fputs("NCOLS=1\nNCOMP=3\n", ofp);
280 >                                        fputformat((char *)cm_fmt_id[outfmt], ofp);
281 >                                        fputc('\n', ofp);
282 >                                }
283 >                                cm_write(rvec, outfmt, ofp);
284 >                                if (fclose(ofp) == EOF) {
285 >                                        fprintf(stderr,
286 >                                                "%s: error writing to '%s'\n",
287 >                                                        progname, fnbuf);
288 >                                        return(1);
289 >                                }
290 >                                ofp = stdout;
291 >                                cm_free(rvec);
292 >                        }
293 >                } else {
294 > #ifdef getc_unlocked
295 >                        flockfile(ofp);
296 > #endif
297 >                        if (outfmt != DTascii)
298 >                                SET_FILE_BINARY(ofp);
299 >                        if (headout) {          /* header output */
300 >                                newheader("RADIANCE", ofp);
301 >                                printargs(argc, argv, ofp);
302 >                                fputnow(ofp);
303 >                                fprintf(ofp, "NROWS=%d\n", rmtx->nrows);
304 >                                fprintf(ofp, "NCOLS=%d\n", rmtx->ncols);
305 >                                fputs("NCOMP=3\n", ofp);
306 >                                fputformat((char *)cm_fmt_id[outfmt], ofp);
307 >                                fputc('\n', ofp);
308 >                        }
309 >                        cm_write(rmtx, outfmt, ofp);
310 >                }
311 >                cm_free(rmtx);
312          }
313 <        cm_free(cvec);                          /* final clean-up */
313 >        if (fflush(ofp) == EOF) {               /* final clean-up */
314 >                fprintf(stderr, "%s: write error on output\n", progname);
315 >                return(1);
316 >        }
317 >        cm_free(cmtx);
318          return(0);
319 + userr:
320 +        fprintf(stderr, "Usage: %s [-n nsteps][-o ospec][-i{f|d|h}][-o{f|d}] DCspec [skyf]\n",
321 +                                progname);
322 +        fprintf(stderr, "   or: %s [-n nsteps][-o ospec][-i{f|d|h}][-o{f|d}] Vspec Tbsdf.xml Dmat.dat [skyf]\n",
323 +                                progname);
324 +        return(1);
325   }

Diff Legend

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