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.31 by greg, Fri May 30 00:00:54 2014 UTC vs.
Revision 2.52 by greg, Sat Mar 22 01:27:22 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"
# 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;
28  
29          if (cv->ncols != 1)
30 <                error(INTERNAL, "expected matrix in sum_images()");
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;
# Line 44 | Line 47 | sum_images(const char *fspec, const CMATRIX *cv, FILE
47                          sprintf(errmsg, "cannot open picture '%s'", fname);
48                          error(SYSTEM, errmsg);
49                  }
50 + #ifdef getc_unlocked
51 +                flockfile(fp);
52 + #endif
53                  dt = DTfromHeader;
54 <                if ((err = cm_getheader(&dt, NULL, NULL, fp)) != NULL)
54 >                if ((err = cm_getheader(&dt, NULL, NULL, NULL, NULL, fp)) != NULL)
55                          error(USER, err);
56                  if ((dt != DTrgbe) & (dt != DTxyze) ||
57                                  !fscnresolu(&xr, &yr, fp)) {
# Line 55 | Line 61 | sum_images(const char *fspec, const CMATRIX *cv, FILE
61                  if (myDT == DTfromHeader) {             /* on first one */
62                          myDT = dt;
63                          myXR = xr; myYR = yr;
64 <                        scanline = (COLOR *)malloc(sizeof(COLOR)*myXR);
64 >                        scanline = (COLR *)malloc(sizeof(COLR)*myXR);
65                          if (scanline == NULL)
66                                  error(SYSTEM, "out of memory in sum_images()");
67                          pmat = cm_alloc(myYR, myXR);
68                          memset(pmat->cmem, 0, sizeof(COLOR)*myXR*myYR);
69                                                          /* finish header */
70 <                        fputformat((char *)cm_fmt_id[myDT], fout);
70 >                        fputformat(cm_fmt_id[myDT], fout);
71                          fputc('\n', fout);
72                          fflush(fout);
73                  } else if ((dt != myDT) | (xr != myXR) | (yr != myYR)) {
# Line 69 | Line 75 | sum_images(const char *fspec, const CMATRIX *cv, FILE
75                                          fname);
76                          error(USER, errmsg);
77                  }
78 +                                                        /* flat file check */
79 +                if ((data_start = ftell(fp)) > 0 && fseek(fp, 0L, SEEK_END) == 0) {
80 +                        flat_file = (ftell(fp) >= data_start + sizeof(COLR)*xr*yr);
81 +                        if (fseek(fp, data_start, SEEK_SET) < 0) {
82 +                                sprintf(errmsg, "cannot seek on picture '%s'", fname);
83 +                                error(SYSTEM, errmsg);
84 +                        }
85 +                }
86                  psp = pmat->cmem;
87                  for (y = 0; y < yr; y++) {              /* read it in */
88 +                        COLOR   col;
89                          int     x;
90 <                        if (freadscan(scanline, xr, fp) < 0) {
90 >                        if (flat_file ? getbinary(scanline, sizeof(COLR), xr, fp) != xr :
91 >                                        freadcolrs(scanline, xr, fp) < 0) {
92                                  sprintf(errmsg, "error reading picture '%s'",
93                                                  fname);
94                                  error(SYSTEM, errmsg);
95                          }
96                                                          /* sum in scanline */
97                          for (x = 0; x < xr; x++, psp += 3) {
98 <                                multcolor(scanline[x], scv);
99 <                                addcolor(psp, scanline[x]);
98 >                                if (!scanline[x][EXP])
99 >                                        continue;       /* skip zeroes */
100 >                                colr_color(col, scanline[x]);
101 >                                multcolor(col, scv);
102 >                                addcolor(psp, col);
103                          }
104                  }
105                  fclose(fp);                             /* done this picture */
# Line 91 | Line 110 | sum_images(const char *fspec, const CMATRIX *cv, FILE
110          return(i);
111   }
112  
113 + /* adjust matrix dimensions according to user size(s) */
114 + static int
115 + alt_dim(CMATRIX *cm, int nr, int nc)
116 + {
117 +        if ((nr <= 0) & (nc <= 0))
118 +                return(0);
119 +        if ((nr == cm->nrows) & (nc == cm->ncols))
120 +                return(0);
121 +        if (nr > 0) {
122 +                if (nc <= 0)
123 +                        nc = cm->nrows*cm->ncols/nr;
124 +                if (nr*nc != cm->nrows*cm->ncols) {
125 +                        fprintf(stderr, "Bad dimensions: %dx%d != %dx%d\n",
126 +                                        nr, nc, cm->nrows, cm->ncols);
127 +                        return(-1);
128 +                }
129 +        } else /* nc > 0 */ {
130 +                nr = cm->nrows*cm->ncols/nc;
131 +                if (nc*nr != cm->nrows*cm->ncols) {
132 +                        fprintf(stderr, "Bad dimensions: %d does not divide %dx%d evenly\n",
133 +                                        nc, cm->nrows, cm->ncols);
134 +                        return(-1);
135 +                }
136 +        }
137 +        cm->nrows = nr;
138 +        cm->ncols = nc;
139 +        return(1);
140 + }
141 +
142   /* check to see if a string contains a %d or %o specification */
143   static int
144   hasNumberFormat(const char *s)
# Line 119 | Line 167 | hasNumberFormat(const char *s)
167   int
168   main(int argc, char *argv[])
169   {
170 <        int             skyfmt = DTascii;
170 >        int             skyfmt = DTfromHeader;
171          int             outfmt = DTascii;
172 <        int             nsteps = 1;
172 >        int             headout = 1;
173 >        int             nsteps = 0;
174          char            *ofspec = NULL;
175          FILE            *ofp = stdout;
176 +        int             xres=0, yres=0;
177          CMATRIX         *cmtx;          /* component vector/matrix result */
178          char            fnbuf[256];
179          int             a, i;
# Line 136 | Line 186 | main(int argc, char *argv[])
186                          nsteps = atoi(argv[++a]);
187                          if (nsteps < 0)
188                                  goto userr;
189 +                        skyfmt = nsteps ? DTascii : DTfromHeader;
190                          break;
191 +                case 'h':
192 +                        headout = !headout;
193 +                        break;
194                  case 'i':
195                          switch (argv[a][2]) {
196                          case 'f':
# Line 148 | Line 202 | main(int argc, char *argv[])
202                          case 'a':
203                                  skyfmt = DTascii;
204                                  break;
151 –                        case 'h':
152 –                                skyfmt = DTfromHeader;
153 –                                break;
205                          default:
206                                  goto userr;
207                          }
# Line 169 | Line 220 | main(int argc, char *argv[])
220                          case 'a':
221                                  outfmt = DTascii;
222                                  break;
223 +                        case 'c':
224 +                                outfmt = DTrgbe;
225 +                                break;
226                          default:
227                                  goto userr;
228                          }
229                          break;
230 +                case 'x':
231 +                        xres = atoi(argv[++a]);
232 +                        break;
233 +                case 'y':
234 +                        yres = atoi(argv[++a]);
235 +                        break;
236                  default:
237                          goto userr;
238                  }
# Line 180 | Line 240 | main(int argc, char *argv[])
240                  goto userr;
241  
242          if (argc-a > 2) {                       /* VTDs expression */
243 <                CMATRIX *smtx, *Dmat, *Tmat, *imtx;
243 >                CMATRIX         *smtx, *Dmat, *Tmat, *imtx;
244 >                const char      *ccp;
245                                                  /* get sky vector/matrix */
246                  smtx = cm_load(argv[a+3], 0, nsteps, skyfmt);
247 +                nsteps = smtx->ncols;
248                                                  /* load BSDF */
249 <                Tmat = cm_loadBTDF(argv[a+1]);
249 >                if (argv[a+1][0] != '!' &&
250 >                                (ccp = strrchr(argv[a+1], '.')) > argv[a+1] &&
251 >                                !strcasecmp(ccp+1, "XML"))
252 >                        Tmat = cm_loadBTDF(argv[a+1]);
253 >                else
254 >                        Tmat = cm_load(argv[a+1], 0, 0, DTfromHeader);
255                                                  /* load Daylight matrix */
256 <                Dmat = cm_load(argv[a+2], Tmat==NULL ? 0 : Tmat->ncols,
256 >                Dmat = cm_load(argv[a+2], Tmat->ncols,
257                                          smtx->nrows, DTfromHeader);
258                                                  /* multiply vector through */
259                  imtx = cm_multiply(Dmat, smtx);
# Line 196 | Line 263 | main(int argc, char *argv[])
263                  cm_free(imtx);
264          } else {                                /* sky vector/matrix only */
265                  cmtx = cm_load(argv[a+1], 0, nsteps, skyfmt);
266 +                nsteps = cmtx->ncols;
267          }
268                                                  /* prepare output stream */
269          if ((ofspec != NULL) & (nsteps == 1) && hasNumberFormat(ofspec)) {
270 <                sprintf(fnbuf, ofspec, 1);
270 >                sprintf(fnbuf, ofspec, 0);
271                  ofspec = fnbuf;
272          }
273          if (ofspec != NULL && !hasNumberFormat(ofspec)) {
# Line 210 | Line 278 | main(int argc, char *argv[])
278                  }
279                  ofspec = NULL;                  /* only need to open once */
280          }
281 <        if (hasNumberFormat(argv[a])) {         /* generating image(s) */
281 >        if (hasNumberFormat(argv[a])) {         /* loading image vector(s) */
282 >                if (outfmt != DTrgbe) {
283 >                        error(WARNING, "changing output type to -oc");
284 >                        outfmt = DTrgbe;
285 >                }
286                  if (ofspec == NULL) {
287                          SET_FILE_BINARY(ofp);
288                          newheader("RADIANCE", ofp);
# Line 221 | Line 293 | main(int argc, char *argv[])
293                          for (i = 0; i < nsteps; i++) {
294                                  CMATRIX *cvec = cm_column(cmtx, i);
295                                  if (ofspec != NULL) {
296 <                                        sprintf(fnbuf, ofspec, i+1);
296 >                                        sprintf(fnbuf, ofspec, i);
297                                          if ((ofp = fopen(fnbuf, "wb")) == NULL) {
298                                                  fprintf(stderr,
299                                          "%s: cannot open '%s' for output\n",
# Line 232 | Line 304 | main(int argc, char *argv[])
304                                          printargs(argc, argv, ofp);
305                                          fputnow(ofp);
306                                  }
307 <                                fprintf(ofp, "FRAME=%d\n", i+1);
307 >                                fprintf(ofp, "FRAME=%d\n", i);
308                                  if (!sum_images(argv[a], cvec, ofp))
309                                          return(1);
310                                  if (ofspec != NULL) {
# Line 248 | Line 320 | main(int argc, char *argv[])
320                          }
321                  else if (!sum_images(argv[a], cmtx, ofp))
322                          return(1);
323 <        } else {                                /* generating vector/matrix */
323 >        } else {                                /* loading view matrix */
324                  CMATRIX *Vmat = cm_load(argv[a], 0, cmtx->nrows, DTfromHeader);
325                  CMATRIX *rmtx = cm_multiply(Vmat, cmtx);
326                  cm_free(Vmat);
# Line 256 | Line 328 | main(int argc, char *argv[])
328                          const char      *wtype = (outfmt==DTascii) ? "w" : "wb";
329                          for (i = 0; i < nsteps; i++) {
330                                  CMATRIX *rvec = cm_column(rmtx, i);
331 <                                sprintf(fnbuf, ofspec, i+1);
331 >                                if (alt_dim(rvec, yres, xres) < 0)
332 >                                        return(1);
333 >                                sprintf(fnbuf, ofspec, i);
334                                  if ((ofp = fopen(fnbuf, wtype)) == NULL) {
335                                          fprintf(stderr,
336                                          "%s: cannot open '%s' for output\n",
# Line 266 | Line 340 | main(int argc, char *argv[])
340   #ifdef getc_unlocked
341                                  flockfile(ofp);
342   #endif
343 +                                if (headout) {  /* header output */
344 +                                        newheader("RADIANCE", ofp);
345 +                                        printargs(argc, argv, ofp);
346 +                                        fputnow(ofp);
347 +                                        fprintf(ofp, "FRAME=%d\n", i);
348 +                                        if ((outfmt != DTrgbe) & (outfmt != DTxyze)) {
349 +                                                fprintf(ofp, "NROWS=%d\n", rvec->nrows);
350 +                                                fprintf(ofp, "NCOLS=%d\n", rvec->ncols);
351 +                                                fputncomp(3, ofp);
352 +                                        }
353 +                                        if ((outfmt == DTfloat) | (outfmt == DTdouble))
354 +                                                fputendian(ofp);
355 +                                        fputformat(cm_fmt_id[outfmt], ofp);
356 +                                        fputc('\n', ofp);
357 +                                }
358                                  cm_write(rvec, outfmt, ofp);
359                                  if (fclose(ofp) == EOF) {
360                                          fprintf(stderr,
# Line 282 | Line 371 | main(int argc, char *argv[])
371   #endif
372                          if (outfmt != DTascii)
373                                  SET_FILE_BINARY(ofp);
374 <                        if (rmtx->ncols > 1) {  /* header if actual matrix */
374 >                        if (alt_dim(rmtx, yres, xres) < 0)
375 >                                return(1);
376 >                        if (headout) {          /* header output */
377                                  newheader("RADIANCE", ofp);
378                                  printargs(argc, argv, ofp);
379                                  fputnow(ofp);
380 <                                fprintf(ofp, "NCOLS=%d\n", rmtx->ncols);
381 <                                fputs("NCOMP=3\n", ofp);
382 <                                fputformat((char *)cm_fmt_id[outfmt], ofp);
380 >                                if ((outfmt != DTrgbe) & (outfmt != DTxyze)) {
381 >                                        fprintf(ofp, "NROWS=%d\n", rmtx->nrows);
382 >                                        fprintf(ofp, "NCOLS=%d\n", rmtx->ncols);
383 >                                        fputncomp(3, ofp);
384 >                                }
385 >                                if ((outfmt == DTfloat) | (outfmt == DTdouble))
386 >                                        fputendian(ofp);
387 >                                fputformat(cm_fmt_id[outfmt], ofp);
388                                  fputc('\n', ofp);
389                          }
390                          cm_write(rmtx, outfmt, ofp);
# Line 302 | Line 398 | main(int argc, char *argv[])
398          cm_free(cmtx);
399          return(0);
400   userr:
401 <        fprintf(stderr, "Usage: %s [-n nsteps][-o ospec][-i{f|d}][-o{f|d}] DCspec [skyf]\n",
401 >        fprintf(stderr, "Usage: %s [-n nsteps][-o ospec][-x xr][-y yr][-i{f|d|h}][-o{f|d|c}] DCspec [skyf]\n",
402                                  progname);
403 <        fprintf(stderr, "   or: %s [-n nsteps][-o ospec][-i{f|d}][-o{f|d}] Vspec Tbsdf.xml Dmat.dat [skyf]\n",
403 >        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",
404                                  progname);
405          return(1);
406   }

Diff Legend

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