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.13 by greg, Thu Jul 1 22:44:25 2010 UTC vs.
Revision 2.53 by greg, Thu Mar 27 16:34:23 2025 UTC

# Line 8 | Line 8 | static const char RCSid[] = "$Id$";
8   */
9  
10   #include <ctype.h>
11 + #include "platform.h"
12   #include "standard.h"
13 + #include "cmatrix.h"
14   #include "platform.h"
13 – #include "color.h"
15   #include "resolu.h"
15 – #include "bsdf.h"
16  
17   char    *progname;                      /* global argv[0] */
18  
19 < /* 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 <                error(USER, "attempt to create empty matrix");
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 */
19 > /* Sum together a set of images and write result to fout */
20   static int
73 – getDT(char *s, void *p)
74 – {
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 (ncols <= 0)
112 –                error(USER, "Non-positive number of columns");
113 –        if (fname == NULL)
114 –                fname = "<stdin>";
115 –        else if ((fp = fopen(fname, "r")) == NULL) {
116 –                sprintf(errmsg, "cannot open file '%s'", fname);
117 –                error(SYSTEM, errmsg);
118 –        }
119 –        if (dtype != DTascii)
120 –                SET_FILE_BINARY(fp);
121 –        if (dtype == DTfromHeader)
122 –                dtype = getDTfromHeader(fp);
123 –        switch (dtype) {
124 –        case DTascii:
125 –        case DTfloat:
126 –        case DTdouble:
127 –                break;
128 –        default:
129 –                error(USER, "unexpected data type in cm_load()");
130 –        }
131 –        if (nrows <= 0) {                       /* don't know length? */
132 –                int     guessrows = 147;        /* usually big enough */
133 –                if ((dtype != DTascii) & (fp != stdin)) {
134 –                        long    startpos = ftell(fp);
135 –                        if (fseek(fp, 0L, SEEK_END) == 0) {
136 –                                long    endpos = ftell(fp);
137 –                                long    elemsiz = 3*(dtype==DTfloat ?
138 –                                            sizeof(float) : sizeof(double));
139 –
140 –                                if ((endpos - startpos) % (ncols*elemsiz)) {
141 –                                        sprintf(errmsg,
142 –                                        "improper length for binary file '%s'",
143 –                                                        fname);
144 –                                        error(USER, errmsg);
145 –                                }
146 –                                guessrows = (endpos - startpos)/(ncols*elemsiz);
147 –                                if (fseek(fp, startpos, SEEK_SET) < 0) {
148 –                                        sprintf(errmsg,
149 –                                                "fseek() error on file '%s'",
150 –                                                        fname);
151 –                                        error(SYSTEM, errmsg);
152 –                                }
153 –                                nrows = guessrows;      /* we're confident */
154 –                        }
155 –                }
156 –                cm = cm_alloc(guessrows, ncols);
157 –        } else
158 –                cm = cm_alloc(nrows, ncols);
159 –        if (cm == NULL)
160 –                return(NULL);
161 –        if (dtype == DTascii) {                         /* read text file */
162 –                int     maxrow = (nrows > 0 ? nrows : 32000);
163 –                int     r, c;
164 –                for (r = 0; r < maxrow; r++) {
165 –                    if (r >= cm->nrows)                 /* need more space? */
166 –                        cm = cm_resize(cm, 2*cm->nrows);
167 –                    for (c = 0; c < ncols; c++) {
168 –                        COLORV  *cv = cm_lval(cm,r,c);
169 –                        if (fscanf(fp, COLSPEC, cv, cv+1, cv+2) != 3)
170 –                                if ((nrows <= 0) & (r > 0) & !c) {
171 –                                        cm = cm_resize(cm, maxrow=r);
172 –                                        break;
173 –                                } else
174 –                                        goto EOFerror;
175 –                    }
176 –                }
177 –                while ((c = getc(fp)) != EOF)
178 –                        if (!isspace(c)) {
179 –                                sprintf(errmsg,
180 –                                "unexpected data at end of ascii file %s",
181 –                                                fname);
182 –                                error(WARNING, errmsg);
183 –                                break;
184 –                        }
185 –        } else {                                        /* read binary file */
186 –                if (sizeof(COLORV) == (dtype==DTfloat ? sizeof(float) :
187 –                                                        sizeof(double))) {
188 –                        int     nread = 0;
189 –                        do {                            /* read all we can */
190 –                                nread += fread(cm->cmem + 3*nread,
191 –                                                3*sizeof(COLORV),
192 –                                                cm->nrows*cm->ncols - nread,
193 –                                                fp);
194 –                                if (nrows <= 0) {       /* unknown length */
195 –                                        if (nread == cm->nrows*cm->ncols)
196 –                                                        /* need more space? */
197 –                                                cm = cm_resize(cm, 2*cm->nrows);
198 –                                        else if (nread && !(nread % cm->ncols))
199 –                                                        /* seem to be  done */
200 –                                                cm = cm_resize(cm, nread/cm->ncols);
201 –                                        else            /* ended mid-row */
202 –                                                goto EOFerror;
203 –                                } else if (nread < cm->nrows*cm->ncols)
204 –                                        goto EOFerror;
205 –                        } while (nread < cm->nrows*cm->ncols);
206 –
207 –                } else if (dtype == DTdouble) {
208 –                        double  dc[3];                  /* load from double */
209 –                        COLORV  *cvp = cm->cmem;
210 –                        int     n = nrows*ncols;
211 –
212 –                        if (n <= 0)
213 –                                goto not_handled;
214 –                        while (n--) {
215 –                                if (fread(dc, sizeof(double), 3, fp) != 3)
216 –                                        goto EOFerror;
217 –                                copycolor(cvp, dc);
218 –                                cvp += 3;
219 –                        }
220 –                } else /* dtype == DTfloat */ {
221 –                        float   fc[3];                  /* load from float */
222 –                        COLORV  *cvp = cm->cmem;
223 –                        int     n = nrows*ncols;
224 –
225 –                        if (n <= 0)
226 –                                goto not_handled;
227 –                        while (n--) {
228 –                                if (fread(fc, sizeof(float), 3, fp) != 3)
229 –                                        goto EOFerror;
230 –                                copycolor(cvp, fc);
231 –                                cvp += 3;
232 –                        }
233 –                }
234 –                if (getc(fp) != EOF) {
235 –                                sprintf(errmsg,
236 –                                "unexpected data at end of binary file %s",
237 –                                                fname);
238 –                                error(WARNING, errmsg);
239 –                }
240 –        }
241 –        if (fp != stdin)
242 –                fclose(fp);
243 –        return(cm);
244 – EOFerror:
245 –        sprintf(errmsg, "unexpected EOF reading %s",
246 –                        fname);
247 –        error(USER, errmsg);
248 – not_handled:
249 –        error(INTERNAL, "unhandled data size or length in cm_load()");
250 –        return(NULL);   /* gratis return */
251 – }
252 –
253 – /* Multiply two matrices (or a matrix and a vector) and allocate the result*/
254 – static CMATRIX *
255 – cm_multiply(const CMATRIX *cm1, const CMATRIX *cm2)
256 – {
257 –        CMATRIX *cmr;
258 –        int     dr, dc, i;
259 –
260 –        if ((cm1->ncols <= 0) | (cm1->ncols != cm2->nrows))
261 –                error(INTERNAL, "matrix dimension mismatch in cm_multiply()");
262 –        cmr = cm_alloc(cm1->nrows, cm2->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 –                COLORV  *dp = cm_lval(cmr,dr,dc);
268 –                dp[0] = dp[1] = dp[2] = 0;
269 –                for (i = 0; i < cm1->ncols; i++) {
270 –                    const COLORV        *cp1 = cm_lval(cm1,dr,i);
271 –                    const COLORV        *cp2 = cm_lval(cm2,i,dc);
272 –                    dp[0] += cp1[0] * cp2[0];
273 –                    dp[1] += cp1[1] * cp2[1];
274 –                    dp[2] += cp1[2] * cp2[2];
275 –                }
276 –            }
277 –        return(cmr);
278 – }
279 –
280 – /* print out matrix as ASCII text -- no header */
281 – static void
282 – cm_print(const CMATRIX *cm, FILE *fp)
283 – {
284 –        int             r, c;
285 –        const COLORV    *mp = cm->cmem;
286 –        
287 –        for (r = 0; r < cm->nrows; r++) {
288 –                for (c = 0; c < cm->ncols; c++, mp += 3)
289 –                        fprintf(fp, "\t%.6e %.6e %.6e", mp[0], mp[1], mp[2]);
290 –                fputc('\n', fp);
291 –        }
292 – }
293 –
294 – /* convert a BSDF to our matrix representation */
295 – static CMATRIX *
296 – cm_bsdf(const struct BSDF_data *bsdf)
297 – {
298 –        CMATRIX *cm = cm_alloc(bsdf->nout, bsdf->ninc);
299 –        int     nbadohm = 0;
300 –        int     nneg = 0;
301 –        int     r, c;
302 –        
303 –        for (c = 0; c < cm->ncols; c++) {
304 –                float   dom = getBSDF_incohm(bsdf,c);
305 –                FVECT   v;
306 –                
307 –                if (dom <= .0) {
308 –                        nbadohm++;
309 –                        continue;
310 –                }
311 –                if (!getBSDF_incvec(v,bsdf,c) || v[2] > FTINY)
312 –                        error(USER, "illegal incoming BTDF direction");
313 –                dom *= -v[2];
314 –
315 –                for (r = 0; r < cm->nrows; r++) {
316 –                        float   f = BSDF_value(bsdf,c,r);
317 –                        COLORV  *mp = cm_lval(cm,r,c);
318 –
319 –                        if (f <= .0) {
320 –                                nneg += (f < -FTINY);
321 –                                f = .0f;
322 –                        }
323 –                        mp[0] = mp[1] = mp[2] = f * dom;
324 –                }
325 –        }
326 –        if (nneg || nbadohm) {
327 –                sprintf(errmsg,
328 –                    "BTDF has %d negatives and %d bad incoming solid angles",
329 –                                nneg, nbadohm);
330 –                error(WARNING, errmsg);
331 –        }
332 –        return(cm);
333 – }
334 –
335 – /* Sum together a set of images and write result to stdout */
336 – static int
21   sum_images(const char *fspec, const CMATRIX *cv, FILE *fout)
22   {
23 +        static int      runcnt = 0;
24          int     myDT = DTfromHeader;
25 <        CMATRIX *pmat;
26 <        COLOR   *scanline;
27 <        int     myXR, myYR;
25 >        COLR    *scanline = NULL;
26 >        CMATRIX *pmat = NULL;
27 >        int     myXR=0, myYR=0;
28          int     i, y;
29  
30          if (cv->ncols != 1)
31                  error(INTERNAL, "expected vector in sum_images()");
32 <        for (i = 0; i < cv->nrows; i++) {
33 <                const COLORV    *scv = cv_lval(cv,i);
32 >        for (i = cv->nrows; i-- > 0; ) {
33 >                const int       r = runcnt&1 ? i : cv->nrows-1 - i;
34 >                const COLORV    *scv = cv_lval(cv,r);
35 >                int             flat_file = 0;
36                  char            fname[1024];
37                  FILE            *fp;
38 +                long            data_start;
39                  int             dt, xr, yr;
40                  COLORV          *psp;
41 +                char            *err;
42 +                                                        /* check for zero */
43 +                if ((scv[RED] == 0) & (scv[GRN] == 0) & (scv[BLU] == 0) &&
44 +                                (myDT != DTfromHeader) | (i > 0))
45 +                        continue;
46                                                          /* open next picture */
47 <                sprintf(fname, fspec, i);
48 <                if ((fp = fopen(fname, "r")) == NULL) {
47 >                sprintf(fname, fspec, r);
48 >                if ((fp = fopen(fname, "rb")) == NULL) {
49                          sprintf(errmsg, "cannot open picture '%s'", fname);
50                          error(SYSTEM, errmsg);
51                  }
52 <                SET_FILE_BINARY(fp);
53 <                dt = getDTfromHeader(fp);
52 > #ifdef getc_unlocked
53 >                flockfile(fp);
54 > #endif
55 >                dt = DTfromHeader;
56 >                if ((err = cm_getheader(&dt, NULL, NULL, NULL, NULL, fp)) != NULL)
57 >                        error(USER, err);
58                  if ((dt != DTrgbe) & (dt != DTxyze) ||
59                                  !fscnresolu(&xr, &yr, fp)) {
60                          sprintf(errmsg, "file '%s' not a picture", fname);
# Line 366 | Line 63 | sum_images(const char *fspec, const CMATRIX *cv, FILE
63                  if (myDT == DTfromHeader) {             /* on first one */
64                          myDT = dt;
65                          myXR = xr; myYR = yr;
66 <                        scanline = (COLOR *)malloc(sizeof(COLOR)*myXR);
66 >                        scanline = (COLR *)malloc(sizeof(COLR)*myXR);
67                          if (scanline == NULL)
68                                  error(SYSTEM, "out of memory in sum_images()");
69                          pmat = cm_alloc(myYR, myXR);
70                          memset(pmat->cmem, 0, sizeof(COLOR)*myXR*myYR);
71                                                          /* finish header */
72 <                        fputformat(myDT==DTrgbe ? COLRFMT : CIEFMT, fout);
72 >                        fputformat(cm_fmt_id[myDT], fout);
73                          fputc('\n', fout);
377 –                        fprtresolu(myXR, myYR, fout);
74                          fflush(fout);
75                  } else if ((dt != myDT) | (xr != myXR) | (yr != myYR)) {
76                          sprintf(errmsg, "picture '%s' format/size mismatch",
77                                          fname);
78                          error(USER, errmsg);
79                  }
80 +                                                        /* flat file check */
81 +                if ((data_start = ftell(fp)) > 0 && fseek(fp, 0L, SEEK_END) == 0) {
82 +                        flat_file = (ftell(fp) >= data_start + sizeof(COLR)*xr*yr);
83 +                        if (fseek(fp, data_start, SEEK_SET) < 0) {
84 +                                sprintf(errmsg, "cannot seek on picture '%s'", fname);
85 +                                error(SYSTEM, errmsg);
86 +                        }
87 +                }
88                  psp = pmat->cmem;
89                  for (y = 0; y < yr; y++) {              /* read it in */
90 +                        COLOR   col;
91                          int     x;
92 <                        if (freadscan(scanline, xr, fp) < 0) {
92 >                        if (flat_file ? getbinary(scanline, sizeof(COLR), xr, fp) != xr :
93 >                                        freadcolrs(scanline, xr, fp) < 0) {
94                                  sprintf(errmsg, "error reading picture '%s'",
95                                                  fname);
96                                  error(SYSTEM, errmsg);
97                          }
98                                                          /* sum in scanline */
99                          for (x = 0; x < xr; x++, psp += 3) {
100 <                                multcolor(scanline[x], scv);
101 <                                addcolor(psp, scanline[x]);
100 >                                if (!scanline[x][EXP])
101 >                                        continue;       /* skip zeroes */
102 >                                colr_color(col, scanline[x]);
103 >                                multcolor(col, scv);
104 >                                addcolor(psp, col);
105                          }
106                  }
107                  fclose(fp);                             /* done this picture */
108          }
109          free(scanline);
110 <                                                        /* write scanlines */
111 <        for (y = 0; y < myYR; y++)
112 <                if (fwritescan((COLOR *)cm_lval(pmat, y, 0), myXR, fout) < 0)
113 <                        return(0);
405 <        cm_free(pmat);                                  /* all done */
406 <        return(fflush(fout) == 0);
110 >        i = cm_write(pmat, myDT, fout);                 /* write picture */
111 >        cm_free(pmat);                                  /* free data */
112 >        ++runcnt;                                       /* for zig-zagging */
113 >        return(i);
114   }
115  
116 + /* adjust matrix dimensions according to user size(s) */
117 + static int
118 + alt_dim(CMATRIX *cm, int nr, int nc)
119 + {
120 +        if ((nr <= 0) & (nc <= 0))
121 +                return(0);
122 +        if ((nr == cm->nrows) & (nc == cm->ncols))
123 +                return(0);
124 +        if (nr > 0) {
125 +                if (nc <= 0)
126 +                        nc = cm->nrows*cm->ncols/nr;
127 +                if (nr*nc != cm->nrows*cm->ncols) {
128 +                        fprintf(stderr, "Bad dimensions: %dx%d != %dx%d\n",
129 +                                        nr, nc, cm->nrows, cm->ncols);
130 +                        return(-1);
131 +                }
132 +        } else /* nc > 0 */ {
133 +                nr = cm->nrows*cm->ncols/nc;
134 +                if (nc*nr != cm->nrows*cm->ncols) {
135 +                        fprintf(stderr, "Bad dimensions: %d does not divide %dx%d evenly\n",
136 +                                        nc, cm->nrows, cm->ncols);
137 +                        return(-1);
138 +                }
139 +        }
140 +        cm->nrows = nr;
141 +        cm->ncols = nc;
142 +        return(1);
143 + }
144 +
145   /* check to see if a string contains a %d or %o specification */
146   static int
147   hasNumberFormat(const char *s)
148   {
149 <        while (*s && *s != '%')
414 <                s++;
415 <        if (!*s)
149 >        if (s == NULL)
150                  return(0);
417 –        do
418 –                ++s;
419 –        while (isdigit(*s));
151  
152 <        return(*s == 'd' | *s == 'i' | *s == 'o' | *s == 'x' | *s == 'X');
152 >        while (*s) {
153 >                while (*s != '%')
154 >                        if (!*s++)
155 >                                return(0);
156 >                if (*++s == '%') {              /* ignore "%%" */
157 >                        ++s;
158 >                        continue;
159 >                }
160 >                while (isdigit(*s))             /* field length */
161 >                        ++s;
162 >                                                /* field we'll use? */
163 >                if ((*s == 'd') | (*s == 'i') | (*s == 'o') |
164 >                                        (*s == 'x') | (*s == 'X'))
165 >                        return(1);
166 >        }
167 >        return(0);                              /* didn't find one */
168   }
169  
170   int
171   main(int argc, char *argv[])
172   {
173 <        CMATRIX                 *cvec;
173 >        int             skyfmt = DTfromHeader;
174 >        int             outfmt = DTascii;
175 >        int             headout = 1;
176 >        int             nsteps = 0;
177 >        char            *ofspec = NULL;
178 >        FILE            *ofp = stdout;
179 >        int             xres=0, yres=0;
180 >        CMATRIX         *cmtx;          /* component vector/matrix result */
181 >        char            fnbuf[256];
182 >        int             a, i;
183  
184          progname = argv[0];
185 +                                        /* get options */
186 +        for (a = 1; a < argc && argv[a][0] == '-'; a++)
187 +                switch (argv[a][1]) {
188 +                case 'n':
189 +                        nsteps = atoi(argv[++a]);
190 +                        if (nsteps < 0)
191 +                                goto userr;
192 +                        skyfmt = nsteps ? DTascii : DTfromHeader;
193 +                        break;
194 +                case 'h':
195 +                        headout = !headout;
196 +                        break;
197 +                case 'i':
198 +                        switch (argv[a][2]) {
199 +                        case 'f':
200 +                                skyfmt = DTfloat;
201 +                                break;
202 +                        case 'd':
203 +                                skyfmt = DTdouble;
204 +                                break;
205 +                        case 'a':
206 +                                skyfmt = DTascii;
207 +                                break;
208 +                        default:
209 +                                goto userr;
210 +                        }
211 +                        break;
212 +                case 'o':
213 +                        switch (argv[a][2]) {
214 +                        case '\0':      /* output specification (not format) */
215 +                                ofspec = argv[++a];
216 +                                break;
217 +                        case 'f':
218 +                                outfmt = DTfloat;
219 +                                break;
220 +                        case 'd':
221 +                                outfmt = DTdouble;
222 +                                break;
223 +                        case 'a':
224 +                                outfmt = DTascii;
225 +                                break;
226 +                        case 'c':
227 +                                outfmt = DTrgbe;
228 +                                break;
229 +                        default:
230 +                                goto userr;
231 +                        }
232 +                        break;
233 +                case 'x':
234 +                        xres = atoi(argv[++a]);
235 +                        break;
236 +                case 'y':
237 +                        yres = atoi(argv[++a]);
238 +                        break;
239 +                default:
240 +                        goto userr;
241 +                }
242 +        if ((argc-a < 1) | (argc-a > 4))
243 +                goto userr;
244  
245 <        if ((argc < 2) | (argc > 5)) {
246 <                fprintf(stderr, "Usage: %s DCspec [tregvec]\n", progname);
247 <                fprintf(stderr, "   or: %s Vspec Tbsdf.xml Dmat.dat [tregvec]\n",
248 <                                        progname);
249 <                return(1);
250 <        }
437 <
438 <        if (argc > 3) {                         /* VTDs expression */
439 <                CMATRIX *svec, *Dmat, *Tmat, *ivec;
440 <                struct BSDF_data        *btdf;
441 <                                                /* get sky vector */
442 <                svec = cm_load(argv[4], 0, 1, DTascii);
245 >        if (argc-a > 2) {                       /* VTDs expression */
246 >                CMATRIX         *smtx, *Dmat, *Tmat, *imtx;
247 >                const char      *ccp;
248 >                                                /* get sky vector/matrix */
249 >                smtx = cm_load(argv[a+3], 0, nsteps, skyfmt);
250 >                nsteps = smtx->ncols;
251                                                  /* load BSDF */
252 <                btdf = load_BSDF(argv[2]);
253 <                if (btdf == NULL)
254 <                        return(1);
252 >                if (argv[a+1][0] != '!' &&
253 >                                (ccp = strrchr(argv[a+1], '.')) > argv[a+1] &&
254 >                                !strcasecmp(ccp+1, "XML"))
255 >                        Tmat = cm_loadBTDF(argv[a+1]);
256 >                else
257 >                        Tmat = cm_load(argv[a+1], 0, 0, DTfromHeader);
258                                                  /* load Daylight matrix */
259 <                Dmat = cm_load(argv[3], btdf->ninc, svec->nrows, DTfromHeader);
259 >                Dmat = cm_load(argv[a+2], Tmat->ncols,
260 >                                        smtx->nrows, DTfromHeader);
261                                                  /* multiply vector through */
262 <                ivec = cm_multiply(Dmat, svec);
263 <                cm_free(Dmat); cm_free(svec);
264 <                Tmat = cm_bsdf(btdf);           /* convert BTDF to matrix */
265 <                free_BSDF(btdf);
266 <                cvec = cm_multiply(Tmat, ivec); /* cvec = component vector */
267 <                cm_free(Tmat); cm_free(ivec);
268 <        } else {                                /* else just use sky vector */
269 <                cvec = cm_load(argv[2], 0, 1, DTascii);
262 >                imtx = cm_multiply(Dmat, smtx);
263 >                cm_free(Dmat); cm_free(smtx);
264 >                cmtx = cm_multiply(Tmat, imtx);
265 >                cm_free(Tmat);
266 >                cm_free(imtx);
267 >        } else {                                /* sky vector/matrix only */
268 >                cmtx = cm_load(argv[a+1], 0, nsteps, skyfmt);
269 >                nsteps = cmtx->ncols;
270          }
271 <
272 <        if (hasNumberFormat(argv[1])) {         /* generating image */
273 <                SET_FILE_BINARY(stdout);
274 <                newheader("RADIANCE", stdout);
275 <                printargs(argc, argv, stdout);
276 <                fputnow(stdout);
277 <                if (!sum_images(argv[1], cvec, stdout))
271 >                                                /* prepare output stream */
272 >        if ((ofspec != NULL) & (nsteps == 1) && hasNumberFormat(ofspec)) {
273 >                sprintf(fnbuf, ofspec, 0);
274 >                ofspec = fnbuf;
275 >        }
276 >        if (ofspec != NULL && !hasNumberFormat(ofspec)) {
277 >                if ((ofp = fopen(ofspec, "w")) == NULL) {
278 >                        fprintf(stderr, "%s: cannot open '%s' for output\n",
279 >                                        progname, ofspec);
280                          return(1);
281 <        } else {                                /* generating vector */
282 <                CMATRIX *Vmat = cm_load(argv[1], 0, cvec->nrows, DTfromHeader);
283 <                CMATRIX *rvec = cm_multiply(Vmat, cvec);
281 >                }
282 >                ofspec = NULL;                  /* only need to open once */
283 >        }
284 >        if (hasNumberFormat(argv[a])) {         /* loading image vector(s) */
285 >                if (outfmt != DTrgbe) {
286 >                        error(WARNING, "changing output type to -oc");
287 >                        outfmt = DTrgbe;
288 >                }
289 >                if (ofspec == NULL) {
290 >                        SET_FILE_BINARY(ofp);
291 >                        newheader("RADIANCE", ofp);
292 >                        printargs(argc, argv, ofp);
293 >                        fputnow(ofp);
294 >                }
295 >                if (nsteps > 1)                 /* multiple output frames? */
296 >                        for (i = 0; i < nsteps; i++) {
297 >                                CMATRIX *cvec = cm_column(cmtx, i);
298 >                                if (ofspec != NULL) {
299 >                                        sprintf(fnbuf, ofspec, i);
300 >                                        if ((ofp = fopen(fnbuf, "wb")) == NULL) {
301 >                                                fprintf(stderr,
302 >                                        "%s: cannot open '%s' for output\n",
303 >                                                        progname, fnbuf);
304 >                                                return(1);
305 >                                        }
306 >                                        newheader("RADIANCE", ofp);
307 >                                        printargs(argc, argv, ofp);
308 >                                        fputnow(ofp);
309 >                                }
310 >                                fprintf(ofp, "FRAME=%d\n", i);
311 >                                if (!sum_images(argv[a], cvec, ofp))
312 >                                        return(1);
313 >                                if (ofspec != NULL) {
314 >                                        if (fclose(ofp) == EOF) {
315 >                                                fprintf(stderr,
316 >                                                "%s: error writing to '%s'\n",
317 >                                                        progname, fnbuf);
318 >                                                return(1);
319 >                                        }
320 >                                        ofp = stdout;
321 >                                }
322 >                                cm_free(cvec);
323 >                        }
324 >                else if (!sum_images(argv[a], cmtx, ofp))
325 >                        return(1);
326 >        } else {                                /* loading view matrix */
327 >                CMATRIX *Vmat = cm_load(argv[a], 0, cmtx->nrows, DTfromHeader);
328 >                CMATRIX *rmtx = cm_multiply(Vmat, cmtx);
329                  cm_free(Vmat);
330 <                cm_print(rvec, stdout);
331 <                cm_free(rvec);
330 >                if (ofspec != NULL) {           /* multiple vector files? */
331 >                        const char      *wtype = (outfmt==DTascii) ? "w" : "wb";
332 >                        for (i = 0; i < nsteps; i++) {
333 >                                CMATRIX *rvec = cm_column(rmtx, i);
334 >                                if (alt_dim(rvec, yres, xres) < 0)
335 >                                        return(1);
336 >                                sprintf(fnbuf, ofspec, i);
337 >                                if ((ofp = fopen(fnbuf, wtype)) == NULL) {
338 >                                        fprintf(stderr,
339 >                                        "%s: cannot open '%s' for output\n",
340 >                                                        progname, fnbuf);
341 >                                        return(1);
342 >                                }
343 > #ifdef getc_unlocked
344 >                                flockfile(ofp);
345 > #endif
346 >                                if (headout) {  /* header output */
347 >                                        newheader("RADIANCE", ofp);
348 >                                        printargs(argc, argv, ofp);
349 >                                        fputnow(ofp);
350 >                                        fprintf(ofp, "FRAME=%d\n", i);
351 >                                        if ((outfmt != DTrgbe) & (outfmt != DTxyze)) {
352 >                                                fprintf(ofp, "NROWS=%d\n", rvec->nrows);
353 >                                                fprintf(ofp, "NCOLS=%d\n", rvec->ncols);
354 >                                                fputncomp(3, ofp);
355 >                                        }
356 >                                        if ((outfmt == DTfloat) | (outfmt == DTdouble))
357 >                                                fputendian(ofp);
358 >                                        fputformat(cm_fmt_id[outfmt], ofp);
359 >                                        fputc('\n', ofp);
360 >                                }
361 >                                cm_write(rvec, outfmt, ofp);
362 >                                if (fclose(ofp) == EOF) {
363 >                                        fprintf(stderr,
364 >                                                "%s: error writing to '%s'\n",
365 >                                                        progname, fnbuf);
366 >                                        return(1);
367 >                                }
368 >                                ofp = stdout;
369 >                                cm_free(rvec);
370 >                        }
371 >                } else {
372 > #ifdef getc_unlocked
373 >                        flockfile(ofp);
374 > #endif
375 >                        if (outfmt != DTascii)
376 >                                SET_FILE_BINARY(ofp);
377 >                        if (alt_dim(rmtx, yres, xres) < 0)
378 >                                return(1);
379 >                        if (headout) {          /* header output */
380 >                                newheader("RADIANCE", ofp);
381 >                                printargs(argc, argv, ofp);
382 >                                fputnow(ofp);
383 >                                if ((outfmt != DTrgbe) & (outfmt != DTxyze)) {
384 >                                        fprintf(ofp, "NROWS=%d\n", rmtx->nrows);
385 >                                        fprintf(ofp, "NCOLS=%d\n", rmtx->ncols);
386 >                                        fputncomp(3, ofp);
387 >                                }
388 >                                if ((outfmt == DTfloat) | (outfmt == DTdouble))
389 >                                        fputendian(ofp);
390 >                                fputformat(cm_fmt_id[outfmt], ofp);
391 >                                fputc('\n', ofp);
392 >                        }
393 >                        cm_write(rmtx, outfmt, ofp);
394 >                }
395 >                cm_free(rmtx);
396          }
397 <        cm_free(cvec);                          /* final clean-up */
397 >        if (fflush(ofp) == EOF) {               /* final clean-up */
398 >                fprintf(stderr, "%s: write error on output\n", progname);
399 >                return(1);
400 >        }
401 >        cm_free(cmtx);
402          return(0);
403 + userr:
404 +        fprintf(stderr, "Usage: %s [-n nsteps][-o ospec][-x xr][-y yr][-i{f|d|h}][-o{f|d|c}] DCspec [skyf]\n",
405 +                                progname);
406 +        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",
407 +                                progname);
408 +        return(1);
409   }

Diff Legend

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