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.29 by greg, Mon Jan 20 22:18:29 2014 UTC vs.
Revision 2.54 by greg, Fri Jun 6 19:11:21 2025 UTC

# Line 8 | Line 8 | static const char RCSid[] = "$Id$";
8   */
9  
10   #include <ctype.h>
11 + #include "paths.h"
12 + #include "platform.h"
13   #include "standard.h"
14   #include "cmatrix.h"
15   #include "platform.h"
16   #include "resolu.h"
17  
16 – char    *progname;                      /* global argv[0] */
17 –
18   /* Sum together a set of images and write result to fout */
19   static int
20   sum_images(const char *fspec, const CMATRIX *cv, FILE *fout)
21   {
22 +        static int      runcnt = 0;
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;
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);
31 >        for (i = cv->nrows; i-- > 0; ) {
32 >                const int       r = runcnt&1 ? i : cv->nrows-1 - i;
33 >                const COLORV    *scv = cv_lval(cv,r);
34 >                int             flat_file = 0;
35                  char            fname[1024];
36                  FILE            *fp;
37 +                long            data_start;
38                  int             dt, xr, yr;
39                  COLORV          *psp;
40 +                char            *err;
41                                                          /* check for zero */
42                  if ((scv[RED] == 0) & (scv[GRN] == 0) & (scv[BLU] == 0) &&
43 <                                (myDT != DTfromHeader) | (i < cv->nrows-1))
43 >                                (myDT != DTfromHeader) | (i > 0))
44                          continue;
45                                                          /* open next picture */
46 <                sprintf(fname, fspec, i);
47 <                if ((fp = fopen(fname, "r")) == NULL) {
46 >                sprintf(fname, fspec, r);
47 >                if ((fp = fopen(fname, "rb")) == NULL) {
48                          sprintf(errmsg, "cannot open picture '%s'", fname);
49                          error(SYSTEM, errmsg);
50                  }
51 <                SET_FILE_BINARY(fp);
52 <                dt = getDTfromHeader(fp);
51 > #ifdef getc_unlocked
52 >                flockfile(fp);
53 > #endif
54 >                dt = DTfromHeader;
55 >                if ((err = cm_getheader(&dt, NULL, NULL, NULL, NULL, fp)) != NULL)
56 >                        error(USER, err);
57                  if ((dt != DTrgbe) & (dt != DTxyze) ||
58                                  !fscnresolu(&xr, &yr, fp)) {
59                          sprintf(errmsg, "file '%s' not a picture", fname);
# Line 53 | Line 62 | sum_images(const char *fspec, const CMATRIX *cv, FILE
62                  if (myDT == DTfromHeader) {             /* on first one */
63                          myDT = dt;
64                          myXR = xr; myYR = yr;
65 <                        scanline = (COLOR *)malloc(sizeof(COLOR)*myXR);
65 >                        scanline = (COLR *)malloc(sizeof(COLR)*myXR);
66                          if (scanline == NULL)
67                                  error(SYSTEM, "out of memory in sum_images()");
68                          pmat = cm_alloc(myYR, myXR);
69                          memset(pmat->cmem, 0, sizeof(COLOR)*myXR*myYR);
70                                                          /* finish header */
71 <                        fputformat(myDT==DTrgbe ? COLRFMT : CIEFMT, fout);
71 >                        fputformat(cm_fmt_id[myDT], fout);
72                          fputc('\n', fout);
64 –                        fprtresolu(myXR, myYR, fout);
73                          fflush(fout);
74                  } else if ((dt != myDT) | (xr != myXR) | (yr != myYR)) {
75                          sprintf(errmsg, "picture '%s' format/size mismatch",
76                                          fname);
77                          error(USER, errmsg);
78                  }
79 +                                                        /* flat file check */
80 +                if ((data_start = ftell(fp)) > 0 && fseek(fp, 0L, SEEK_END) == 0) {
81 +                        flat_file = (ftell(fp) >= data_start + sizeof(COLR)*xr*yr);
82 +                        if (fseek(fp, data_start, SEEK_SET) < 0) {
83 +                                sprintf(errmsg, "cannot seek on picture '%s'", fname);
84 +                                error(SYSTEM, errmsg);
85 +                        }
86 +                }
87                  psp = pmat->cmem;
88                  for (y = 0; y < yr; y++) {              /* read it in */
89 +                        COLOR   col;
90                          int     x;
91 <                        if (freadscan(scanline, xr, fp) < 0) {
91 >                        if (flat_file ? getbinary(scanline, sizeof(COLR), xr, fp) != xr :
92 >                                        freadcolrs(scanline, xr, fp) < 0) {
93                                  sprintf(errmsg, "error reading picture '%s'",
94                                                  fname);
95                                  error(SYSTEM, errmsg);
96                          }
97                                                          /* sum in scanline */
98                          for (x = 0; x < xr; x++, psp += 3) {
99 <                                multcolor(scanline[x], scv);
100 <                                addcolor(psp, scanline[x]);
99 >                                if (!scanline[x][EXP])
100 >                                        continue;       /* skip zeroes */
101 >                                colr_color(col, scanline[x]);
102 >                                multcolor(col, scv);
103 >                                addcolor(psp, col);
104                          }
105                  }
106                  fclose(fp);                             /* done this picture */
107          }
108          free(scanline);
109 <                                                        /* write scanlines */
110 <        for (y = 0; y < myYR; y++)
111 <                if (fwritescan((COLOR *)cm_lval(pmat, y, 0), myXR, fout) < 0)
112 <                        return(0);
92 <        cm_free(pmat);                                  /* all done */
93 <        return(fflush(fout) == 0);
109 >        i = cm_write(pmat, myDT, fout);                 /* write picture */
110 >        cm_free(pmat);                                  /* free data */
111 >        ++runcnt;                                       /* for zig-zagging */
112 >        return(i);
113   }
114  
115 + /* adjust matrix dimensions according to user size(s) */
116 + static int
117 + alt_dim(CMATRIX *cm, int nr, int nc)
118 + {
119 +        if ((nr <= 0) & (nc <= 0))
120 +                return(0);
121 +        if ((nr == cm->nrows) & (nc == cm->ncols))
122 +                return(0);
123 +        if (nr > 0) {
124 +                if (nc <= 0)
125 +                        nc = cm->nrows*cm->ncols/nr;
126 +                if (nr*nc != cm->nrows*cm->ncols) {
127 +                        fprintf(stderr, "Bad dimensions: %dx%d != %dx%d\n",
128 +                                        nr, nc, cm->nrows, cm->ncols);
129 +                        return(-1);
130 +                }
131 +        } else /* nc > 0 */ {
132 +                nr = cm->nrows*cm->ncols/nc;
133 +                if (nc*nr != cm->nrows*cm->ncols) {
134 +                        fprintf(stderr, "Bad dimensions: %d does not divide %dx%d evenly\n",
135 +                                        nc, cm->nrows, cm->ncols);
136 +                        return(-1);
137 +                }
138 +        }
139 +        cm->nrows = nr;
140 +        cm->ncols = nc;
141 +        return(1);
142 + }
143 +
144   /* check to see if a string contains a %d or %o specification */
145   static int
146   hasNumberFormat(const char *s)
# Line 121 | Line 169 | hasNumberFormat(const char *s)
169   int
170   main(int argc, char *argv[])
171   {
172 <        int             skyfmt = DTascii;
173 <        int             nsteps = 1;
172 >        int             skyfmt = DTfromHeader;
173 >        int             outfmt = DTascii;
174 >        int             headout = 1;
175 >        int             nsteps = 0;
176          char            *ofspec = NULL;
177          FILE            *ofp = stdout;
178 +        int             xres=0, yres=0;
179          CMATRIX         *cmtx;          /* component vector/matrix result */
180          char            fnbuf[256];
181          int             a, i;
182 <
183 <        progname = argv[0];
182 >                                        /* set global progname */
183 >        fixargv0(argv[0]);
184                                          /* get options */
185          for (a = 1; a < argc && argv[a][0] == '-'; a++)
186                  switch (argv[a][1]) {
187                  case 'n':
188                          nsteps = atoi(argv[++a]);
189 <                        if (nsteps <= 0)
189 >                        if (nsteps < 0)
190                                  goto userr;
191 +                        skyfmt = nsteps ? DTascii : DTfromHeader;
192                          break;
193 <                case 'o':
194 <                        ofspec = argv[++a];
193 >                case 'h':
194 >                        headout = !headout;
195                          break;
196                  case 'i':
197                          switch (argv[a][2]) {
# Line 156 | Line 208 | main(int argc, char *argv[])
208                                  goto userr;
209                          }
210                          break;
211 +                case 'o':
212 +                        switch (argv[a][2]) {
213 +                        case '\0':      /* output specification (not format) */
214 +                                ofspec = argv[++a];
215 +                                break;
216 +                        case 'f':
217 +                                outfmt = DTfloat;
218 +                                break;
219 +                        case 'd':
220 +                                outfmt = DTdouble;
221 +                                break;
222 +                        case 'a':
223 +                                outfmt = DTascii;
224 +                                break;
225 +                        case 'c':
226 +                                outfmt = DTrgbe;
227 +                                break;
228 +                        default:
229 +                                goto userr;
230 +                        }
231 +                        break;
232 +                case 'x':
233 +                        xres = atoi(argv[++a]);
234 +                        break;
235 +                case 'y':
236 +                        yres = atoi(argv[++a]);
237 +                        break;
238                  default:
239                          goto userr;
240                  }
# Line 163 | Line 242 | main(int argc, char *argv[])
242                  goto userr;
243  
244          if (argc-a > 2) {                       /* VTDs expression */
245 <                CMATRIX *smtx, *Dmat, *Tmat, *imtx;
245 >                CMATRIX         *smtx, *Dmat, *Tmat, *imtx;
246 >                const char      *ccp;
247                                                  /* get sky vector/matrix */
248                  smtx = cm_load(argv[a+3], 0, nsteps, skyfmt);
249 +                nsteps = smtx->ncols;
250                                                  /* load BSDF */
251 <                Tmat = cm_loadBTDF(argv[a+1]);
251 >                if (argv[a+1][0] != '!' &&
252 >                                (ccp = strrchr(argv[a+1], '.')) > argv[a+1] &&
253 >                                !strcasecmp(ccp+1, "XML"))
254 >                        Tmat = cm_loadBTDF(argv[a+1]);
255 >                else
256 >                        Tmat = cm_load(argv[a+1], 0, 0, DTfromHeader);
257                                                  /* load Daylight matrix */
258 <                Dmat = cm_load(argv[a+2], Tmat==NULL ? 0 : Tmat->ncols,
258 >                Dmat = cm_load(argv[a+2], Tmat->ncols,
259                                          smtx->nrows, DTfromHeader);
260                                                  /* multiply vector through */
261                  imtx = cm_multiply(Dmat, smtx);
# Line 179 | Line 265 | main(int argc, char *argv[])
265                  cm_free(imtx);
266          } else {                                /* sky vector/matrix only */
267                  cmtx = cm_load(argv[a+1], 0, nsteps, skyfmt);
268 +                nsteps = cmtx->ncols;
269          }
270                                                  /* prepare output stream */
271          if ((ofspec != NULL) & (nsteps == 1) && hasNumberFormat(ofspec)) {
272 <                sprintf(fnbuf, ofspec, 1);
272 >                sprintf(fnbuf, ofspec, 0);
273                  ofspec = fnbuf;
274          }
275          if (ofspec != NULL && !hasNumberFormat(ofspec)) {
# Line 193 | Line 280 | main(int argc, char *argv[])
280                  }
281                  ofspec = NULL;                  /* only need to open once */
282          }
283 <        if (hasNumberFormat(argv[a])) {         /* generating image(s) */
283 >        if (hasNumberFormat(argv[a])) {         /* loading image vector(s) */
284 >                if (outfmt != DTrgbe) {
285 >                        error(WARNING, "changing output type to -oc");
286 >                        outfmt = DTrgbe;
287 >                }
288                  if (ofspec == NULL) {
289                          SET_FILE_BINARY(ofp);
290                          newheader("RADIANCE", ofp);
# Line 204 | Line 295 | main(int argc, char *argv[])
295                          for (i = 0; i < nsteps; i++) {
296                                  CMATRIX *cvec = cm_column(cmtx, i);
297                                  if (ofspec != NULL) {
298 <                                        sprintf(fnbuf, ofspec, i+1);
298 >                                        sprintf(fnbuf, ofspec, i);
299                                          if ((ofp = fopen(fnbuf, "wb")) == NULL) {
300                                                  fprintf(stderr,
301                                          "%s: cannot open '%s' for output\n",
# Line 215 | Line 306 | main(int argc, char *argv[])
306                                          printargs(argc, argv, ofp);
307                                          fputnow(ofp);
308                                  }
309 <                                fprintf(ofp, "FRAME=%d\n", i+1);
309 >                                fprintf(ofp, "FRAME=%d\n", i);
310                                  if (!sum_images(argv[a], cvec, ofp))
311                                          return(1);
312                                  if (ofspec != NULL) {
# Line 231 | Line 322 | main(int argc, char *argv[])
322                          }
323                  else if (!sum_images(argv[a], cmtx, ofp))
324                          return(1);
325 <        } else {                                /* generating vector/matrix */
325 >        } else {                                /* loading view matrix */
326                  CMATRIX *Vmat = cm_load(argv[a], 0, cmtx->nrows, DTfromHeader);
327                  CMATRIX *rmtx = cm_multiply(Vmat, cmtx);
328                  cm_free(Vmat);
329 <                if (ofspec != NULL)             /* multiple vector files? */
329 >                if (ofspec != NULL) {           /* multiple vector files? */
330 >                        const char      *wtype = (outfmt==DTascii) ? "w" : "wb";
331                          for (i = 0; i < nsteps; i++) {
332                                  CMATRIX *rvec = cm_column(rmtx, i);
333 <                                sprintf(fnbuf, ofspec, i+1);
334 <                                if ((ofp = fopen(fnbuf, "w")) == NULL) {
333 >                                if (alt_dim(rvec, yres, xres) < 0)
334 >                                        return(1);
335 >                                sprintf(fnbuf, ofspec, i);
336 >                                if ((ofp = fopen(fnbuf, wtype)) == NULL) {
337                                          fprintf(stderr,
338                                          "%s: cannot open '%s' for output\n",
339                                                          progname, fnbuf);
340                                          return(1);
341                                  }
342 <                                cm_print(rvec, ofp);
342 > #ifdef getc_unlocked
343 >                                flockfile(ofp);
344 > #endif
345 >                                if (headout) {  /* header output */
346 >                                        newheader("RADIANCE", ofp);
347 >                                        printargs(argc, argv, ofp);
348 >                                        fputnow(ofp);
349 >                                        fprintf(ofp, "FRAME=%d\n", i);
350 >                                        if ((outfmt != DTrgbe) & (outfmt != DTxyze)) {
351 >                                                fprintf(ofp, "NROWS=%d\n", rvec->nrows);
352 >                                                fprintf(ofp, "NCOLS=%d\n", rvec->ncols);
353 >                                                fputncomp(3, ofp);
354 >                                        }
355 >                                        if ((outfmt == DTfloat) | (outfmt == DTdouble))
356 >                                                fputendian(ofp);
357 >                                        fputformat(cm_fmt_id[outfmt], ofp);
358 >                                        fputc('\n', ofp);
359 >                                }
360 >                                cm_write(rvec, outfmt, ofp);
361                                  if (fclose(ofp) == EOF) {
362                                          fprintf(stderr,
363                                                  "%s: error writing to '%s'\n",
# Line 255 | Line 367 | main(int argc, char *argv[])
367                                  ofp = stdout;
368                                  cm_free(rvec);
369                          }
370 <                else
371 <                        cm_print(rmtx, ofp);
370 >                } else {
371 > #ifdef getc_unlocked
372 >                        flockfile(ofp);
373 > #endif
374 >                        if (outfmt != DTascii)
375 >                                SET_FILE_BINARY(ofp);
376 >                        if (alt_dim(rmtx, yres, xres) < 0)
377 >                                return(1);
378 >                        if (headout) {          /* header output */
379 >                                newheader("RADIANCE", ofp);
380 >                                printargs(argc, argv, ofp);
381 >                                fputnow(ofp);
382 >                                if ((outfmt != DTrgbe) & (outfmt != DTxyze)) {
383 >                                        fprintf(ofp, "NROWS=%d\n", rmtx->nrows);
384 >                                        fprintf(ofp, "NCOLS=%d\n", rmtx->ncols);
385 >                                        fputncomp(3, ofp);
386 >                                }
387 >                                if ((outfmt == DTfloat) | (outfmt == DTdouble))
388 >                                        fputendian(ofp);
389 >                                fputformat(cm_fmt_id[outfmt], ofp);
390 >                                fputc('\n', ofp);
391 >                        }
392 >                        cm_write(rmtx, outfmt, ofp);
393 >                }
394                  cm_free(rmtx);
395          }
396          if (fflush(ofp) == EOF) {               /* final clean-up */
# Line 266 | Line 400 | main(int argc, char *argv[])
400          cm_free(cmtx);
401          return(0);
402   userr:
403 <        fprintf(stderr, "Usage: %s [-n nsteps][-o ospec][-i{f|d}] DCspec [skyf]\n",
403 >        fprintf(stderr, "Usage: %s [-n nsteps][-o ospec][-x xr][-y yr][-i{f|d|h}][-o{f|d|c}] DCspec [skyf]\n",
404                                  progname);
405 <        fprintf(stderr, "   or: %s [-n nsteps][-o ospec][-i{f|d}] Vspec Tbsdf.xml Dmat.dat [skyf]\n",
405 >        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",
406                                  progname);
407          return(1);
408   }

Diff Legend

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