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.28 by greg, Mon Jan 20 21:29:04 2014 UTC vs.
Revision 2.49 by greg, Fri Mar 11 02:44:33 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"
# Line 20 | Line 21 | static int
21   sum_images(const char *fspec, const CMATRIX *cv, FILE *fout)
22   {
23          int     myDT = DTfromHeader;
24 <        COLOR   *scanline = NULL;
24 >        COLR    *scanline = NULL;
25          CMATRIX *pmat = NULL;
26          int     myXR=0, myYR=0;
27          int     i, y;
# Line 29 | Line 30 | sum_images(const char *fspec, const CMATRIX *cv, FILE
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 53 | Line 58 | sum_images(const char *fspec, const CMATRIX *cv, FILE
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, fout);
67 >                        fputformat(cm_fmt_id[myDT], fout);
68                          fputc('\n', fout);
64 –                        fprtresolu(myXR, myYR, 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, fout) < 0)
91 <                        return(0);
92 <        cm_free(pmat);                                  /* all done */
93 <        return(fflush(fout) == 0);
105 >        i = cm_write(pmat, myDT, fout);                 /* write picture */
106 >        cm_free(pmat);                                  /* free data */
107 >        return(i);
108   }
109  
110 + /* adjust matrix dimensions according to user size(s) */
111 + static int
112 + alt_dim(CMATRIX *cm, int nr, int nc)
113 + {
114 +        if ((nr <= 0) & (nc <= 0))
115 +                return(0);
116 +        if ((nr == cm->nrows) & (nc == cm->ncols))
117 +                return(0);
118 +        if (nr > 0) {
119 +                if (nc <= 0)
120 +                        nc = cm->nrows*cm->ncols/nr;
121 +                if (nr*nc != cm->nrows*cm->ncols) {
122 +                        fprintf(stderr, "Bad dimensions: %dx%d != %dx%d\n",
123 +                                        nr, nc, cm->nrows, cm->ncols);
124 +                        return(-1);
125 +                }
126 +        } else /* nc > 0 */ {
127 +                nr = cm->nrows*cm->ncols/nc;
128 +                if (nc*nr != cm->nrows*cm->ncols) {
129 +                        fprintf(stderr, "Bad dimensions: %d does not divide %dx%d evenly\n",
130 +                                        nc, cm->nrows, cm->ncols);
131 +                        return(-1);
132 +                }
133 +        }
134 +        cm->nrows = nr;
135 +        cm->ncols = nc;
136 +        return(1);
137 + }
138 +
139   /* check to see if a string contains a %d or %o specification */
140   static int
141   hasNumberFormat(const char *s)
# Line 121 | Line 164 | hasNumberFormat(const char *s)
164   int
165   main(int argc, char *argv[])
166   {
167 <        int             skyfmt = DTascii;
168 <        int             nsteps = 1;
167 >        int             skyfmt = DTfromHeader;
168 >        int             outfmt = DTascii;
169 >        int             headout = 1;
170 >        int             nsteps = 0;
171          char            *ofspec = NULL;
172          FILE            *ofp = stdout;
173 +        int             xres=0, yres=0;
174          CMATRIX         *cmtx;          /* component vector/matrix result */
175          char            fnbuf[256];
176          int             a, i;
# Line 135 | Line 181 | main(int argc, char *argv[])
181                  switch (argv[a][1]) {
182                  case 'n':
183                          nsteps = atoi(argv[++a]);
184 <                        if (nsteps <= 0)
184 >                        if (nsteps < 0)
185                                  goto userr;
186 +                        skyfmt = nsteps ? DTascii : DTfromHeader;
187                          break;
188 <                case 'o':
189 <                        ofspec = argv[++a];
188 >                case 'h':
189 >                        headout = !headout;
190                          break;
191                  case 'i':
192                          switch (argv[a][2]) {
# Line 156 | Line 203 | main(int argc, char *argv[])
203                                  goto userr;
204                          }
205                          break;
206 +                case 'o':
207 +                        switch (argv[a][2]) {
208 +                        case '\0':      /* output specification (not format) */
209 +                                ofspec = argv[++a];
210 +                                break;
211 +                        case 'f':
212 +                                outfmt = DTfloat;
213 +                                break;
214 +                        case 'd':
215 +                                outfmt = DTdouble;
216 +                                break;
217 +                        case 'a':
218 +                                outfmt = DTascii;
219 +                                break;
220 +                        case 'c':
221 +                                outfmt = DTrgbe;
222 +                                break;
223 +                        default:
224 +                                goto userr;
225 +                        }
226 +                        break;
227 +                case 'x':
228 +                        xres = atoi(argv[++a]);
229 +                        break;
230 +                case 'y':
231 +                        yres = atoi(argv[++a]);
232 +                        break;
233                  default:
234                          goto userr;
235                  }
# Line 163 | Line 237 | main(int argc, char *argv[])
237                  goto userr;
238  
239          if (argc-a > 2) {                       /* VTDs expression */
240 <                CMATRIX *smtx, *Dmat, *Tmat, *imtx;
241 <                COLOR   tLamb;
240 >                CMATRIX         *smtx, *Dmat, *Tmat, *imtx;
241 >                const char      *ccp;
242                                                  /* get sky vector/matrix */
243                  smtx = cm_load(argv[a+3], 0, nsteps, skyfmt);
244 +                nsteps = smtx->ncols;
245                                                  /* load BSDF */
246 <                Tmat = cm_loadBSDF(argv[a+1], tLamb);
246 >                if (argv[a+1][0] != '!' &&
247 >                                (ccp = strrchr(argv[a+1], '.')) != NULL &&
248 >                                !strcasecmp(ccp+1, "XML"))
249 >                        Tmat = cm_loadBTDF(argv[a+1]);
250 >                else
251 >                        Tmat = cm_load(argv[a+1], 0, 0, DTfromHeader);
252                                                  /* load Daylight matrix */
253 <                Dmat = cm_load(argv[a+2], Tmat==NULL ? 0 : Tmat->ncols,
253 >                Dmat = cm_load(argv[a+2], Tmat->ncols,
254                                          smtx->nrows, DTfromHeader);
255                                                  /* multiply vector through */
256                  imtx = cm_multiply(Dmat, smtx);
257                  cm_free(Dmat); cm_free(smtx);
258 <                if (Tmat == NULL) {             /* diffuse only */
259 <                        cmtx = cm_scale(imtx, tLamb);
180 <                } else {                        /* else apply BTDF matrix */
181 <                        cmtx = cm_multiply(Tmat, imtx);
182 <                        cm_free(Tmat);
183 <                }
258 >                cmtx = cm_multiply(Tmat, imtx);
259 >                cm_free(Tmat);
260                  cm_free(imtx);
261          } else {                                /* sky vector/matrix only */
262                  cmtx = cm_load(argv[a+1], 0, nsteps, skyfmt);
263 +                nsteps = cmtx->ncols;
264          }
265                                                  /* prepare output stream */
266          if ((ofspec != NULL) & (nsteps == 1) && hasNumberFormat(ofspec)) {
# Line 199 | Line 276 | main(int argc, char *argv[])
276                  ofspec = NULL;                  /* only need to open once */
277          }
278          if (hasNumberFormat(argv[a])) {         /* generating image(s) */
279 +                if (outfmt != DTrgbe) {
280 +                        error(WARNING, "changing output type to -oc");
281 +                        outfmt = DTrgbe;
282 +                }
283                  if (ofspec == NULL) {
284                          SET_FILE_BINARY(ofp);
285                          newheader("RADIANCE", ofp);
# Line 209 | Line 290 | main(int argc, char *argv[])
290                          for (i = 0; i < nsteps; i++) {
291                                  CMATRIX *cvec = cm_column(cmtx, i);
292                                  if (ofspec != NULL) {
293 <                                        sprintf(fnbuf, ofspec, i+1);
293 >                                        sprintf(fnbuf, ofspec, i);
294                                          if ((ofp = fopen(fnbuf, "wb")) == NULL) {
295                                                  fprintf(stderr,
296                                          "%s: cannot open '%s' for output\n",
# Line 220 | Line 301 | main(int argc, char *argv[])
301                                          printargs(argc, argv, ofp);
302                                          fputnow(ofp);
303                                  }
304 <                                fprintf(ofp, "FRAME=%d\n", i+1);
304 >                                fprintf(ofp, "FRAME=%d\n", i);
305                                  if (!sum_images(argv[a], cvec, ofp))
306                                          return(1);
307                                  if (ofspec != NULL) {
# Line 240 | Line 321 | main(int argc, char *argv[])
321                  CMATRIX *Vmat = cm_load(argv[a], 0, cmtx->nrows, DTfromHeader);
322                  CMATRIX *rmtx = cm_multiply(Vmat, cmtx);
323                  cm_free(Vmat);
324 <                if (ofspec != NULL)             /* multiple vector files? */
324 >                if (ofspec != NULL) {           /* multiple vector files? */
325 >                        const char      *wtype = (outfmt==DTascii) ? "w" : "wb";
326                          for (i = 0; i < nsteps; i++) {
327                                  CMATRIX *rvec = cm_column(rmtx, i);
328 <                                sprintf(fnbuf, ofspec, i+1);
329 <                                if ((ofp = fopen(fnbuf, "w")) == NULL) {
328 >                                if (alt_dim(rvec, yres, xres) < 0)
329 >                                        return(1);
330 >                                sprintf(fnbuf, ofspec, i);
331 >                                if ((ofp = fopen(fnbuf, wtype)) == NULL) {
332                                          fprintf(stderr,
333                                          "%s: cannot open '%s' for output\n",
334                                                          progname, fnbuf);
335                                          return(1);
336                                  }
337 <                                cm_print(rvec, ofp);
337 > #ifdef getc_unlocked
338 >                                flockfile(ofp);
339 > #endif
340 >                                if (headout) {  /* header output */
341 >                                        newheader("RADIANCE", ofp);
342 >                                        printargs(argc, argv, ofp);
343 >                                        fputnow(ofp);
344 >                                        fprintf(ofp, "FRAME=%d\n", i);
345 >                                        if ((outfmt != DTrgbe) & (outfmt != DTxyze)) {
346 >                                                fprintf(ofp, "NROWS=%d\n", rvec->nrows);
347 >                                                fprintf(ofp, "NCOLS=%d\n", rvec->ncols);
348 >                                                fputs("NCOMP=3\n", ofp);
349 >                                        }
350 >                                        if ((outfmt == DTfloat) | (outfmt == DTdouble))
351 >                                                fputendian(ofp);
352 >                                        fputformat(cm_fmt_id[outfmt], ofp);
353 >                                        fputc('\n', ofp);
354 >                                }
355 >                                cm_write(rvec, outfmt, ofp);
356                                  if (fclose(ofp) == EOF) {
357                                          fprintf(stderr,
358                                                  "%s: error writing to '%s'\n",
# Line 260 | Line 362 | main(int argc, char *argv[])
362                                  ofp = stdout;
363                                  cm_free(rvec);
364                          }
365 <                else
366 <                        cm_print(rmtx, ofp);
365 >                } else {
366 > #ifdef getc_unlocked
367 >                        flockfile(ofp);
368 > #endif
369 >                        if (outfmt != DTascii)
370 >                                SET_FILE_BINARY(ofp);
371 >                        if (alt_dim(rmtx, yres, xres) < 0)
372 >                                return(1);
373 >                        if (headout) {          /* header output */
374 >                                newheader("RADIANCE", ofp);
375 >                                printargs(argc, argv, ofp);
376 >                                fputnow(ofp);
377 >                                if ((outfmt != DTrgbe) & (outfmt != DTxyze)) {
378 >                                        fprintf(ofp, "NROWS=%d\n", rmtx->nrows);
379 >                                        fprintf(ofp, "NCOLS=%d\n", rmtx->ncols);
380 >                                        fputs("NCOMP=3\n", ofp);
381 >                                }
382 >                                if ((outfmt == DTfloat) | (outfmt == DTdouble))
383 >                                        fputendian(ofp);
384 >                                fputformat(cm_fmt_id[outfmt], ofp);
385 >                                fputc('\n', ofp);
386 >                        }
387 >                        cm_write(rmtx, outfmt, ofp);
388 >                }
389                  cm_free(rmtx);
390          }
391          if (fflush(ofp) == EOF) {               /* final clean-up */
# Line 271 | Line 395 | main(int argc, char *argv[])
395          cm_free(cmtx);
396          return(0);
397   userr:
398 <        fprintf(stderr, "Usage: %s [-n nsteps][-o ospec][-i{f|d}] DCspec [skyf]\n",
398 >        fprintf(stderr, "Usage: %s [-n nsteps][-o ospec][-x xr][-y yr][-i{f|d|h}][-o{f|d|c}] DCspec [skyf]\n",
399                                  progname);
400 <        fprintf(stderr, "   or: %s [-n nsteps][-o ospec][-i{f|d}] Vspec Tbsdf.xml Dmat.dat [skyf]\n",
400 >        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",
401                                  progname);
402          return(1);
403   }

Diff Legend

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