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

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines