ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/util/dctimestep.c
Revision: 2.54
Committed: Fri Jun 6 19:11:21 2025 UTC (16 hours, 45 minutes ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.53: +4 -5 lines
Log Message:
refactor: Making use of printargs() more consistent with fixargv0()

File Contents

# User Rev Content
1 greg 2.1 #ifndef lint
2 greg 2.54 static const char RCSid[] = "$Id: dctimestep.c,v 2.53 2025/03/27 16:34:23 greg Exp $";
3 greg 2.1 #endif
4     /*
5     * Compute time-step result using Daylight Coefficient method.
6     *
7     * G. Ward
8     */
9    
10     #include <ctype.h>
11 greg 2.54 #include "paths.h"
12 schorsch 2.37 #include "platform.h"
13 greg 2.1 #include "standard.h"
14 greg 2.28 #include "cmatrix.h"
15 greg 2.1 #include "platform.h"
16     #include "resolu.h"
17    
18 greg 2.24 /* Sum together a set of images and write result to fout */
19 greg 2.1 static int
20 greg 2.13 sum_images(const char *fspec, const CMATRIX *cv, FILE *fout)
21 greg 2.1 {
22 greg 2.53 static int runcnt = 0;
23 greg 2.1 int myDT = DTfromHeader;
24 greg 2.38 COLR *scanline = NULL;
25 greg 2.24 CMATRIX *pmat = NULL;
26     int myXR=0, myYR=0;
27 greg 2.1 int i, y;
28    
29     if (cv->ncols != 1)
30 greg 2.35 error(INTERNAL, "expected vector in sum_images()");
31 greg 2.53 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 greg 2.39 int flat_file = 0;
35 greg 2.1 char fname[1024];
36     FILE *fp;
37 greg 2.39 long data_start;
38 greg 2.1 int dt, xr, yr;
39     COLORV *psp;
40 greg 2.31 char *err;
41 greg 2.14 /* check for zero */
42 greg 2.19 if ((scv[RED] == 0) & (scv[GRN] == 0) & (scv[BLU] == 0) &&
43 greg 2.53 (myDT != DTfromHeader) | (i > 0))
44 greg 2.14 continue;
45 greg 2.1 /* open next picture */
46 greg 2.53 sprintf(fname, fspec, r);
47 greg 2.31 if ((fp = fopen(fname, "rb")) == NULL) {
48 greg 2.1 sprintf(errmsg, "cannot open picture '%s'", fname);
49     error(SYSTEM, errmsg);
50     }
51 greg 2.50 #ifdef getc_unlocked
52     flockfile(fp);
53     #endif
54 greg 2.31 dt = DTfromHeader;
55 greg 2.45 if ((err = cm_getheader(&dt, NULL, NULL, NULL, NULL, fp)) != NULL)
56 greg 2.31 error(USER, err);
57 greg 2.1 if ((dt != DTrgbe) & (dt != DTxyze) ||
58     !fscnresolu(&xr, &yr, fp)) {
59     sprintf(errmsg, "file '%s' not a picture", fname);
60     error(USER, errmsg);
61     }
62     if (myDT == DTfromHeader) { /* on first one */
63     myDT = dt;
64     myXR = xr; myYR = yr;
65 greg 2.38 scanline = (COLR *)malloc(sizeof(COLR)*myXR);
66 greg 2.1 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 greg 2.2 /* finish header */
71 greg 2.46 fputformat(cm_fmt_id[myDT], fout);
72 greg 2.13 fputc('\n', fout);
73     fflush(fout);
74 greg 2.1 } else if ((dt != myDT) | (xr != myXR) | (yr != myYR)) {
75     sprintf(errmsg, "picture '%s' format/size mismatch",
76     fname);
77     error(USER, errmsg);
78     }
79 greg 2.39 /* flat file check */
80     if ((data_start = ftell(fp)) > 0 && fseek(fp, 0L, SEEK_END) == 0) {
81 greg 2.51 flat_file = (ftell(fp) >= data_start + sizeof(COLR)*xr*yr);
82 greg 2.39 if (fseek(fp, data_start, SEEK_SET) < 0) {
83     sprintf(errmsg, "cannot seek on picture '%s'", fname);
84     error(SYSTEM, errmsg);
85     }
86     }
87 greg 2.1 psp = pmat->cmem;
88     for (y = 0; y < yr; y++) { /* read it in */
89 greg 2.38 COLOR col;
90 greg 2.1 int x;
91 greg 2.39 if (flat_file ? getbinary(scanline, sizeof(COLR), xr, fp) != xr :
92     freadcolrs(scanline, xr, fp) < 0) {
93 greg 2.1 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 greg 2.38 if (!scanline[x][EXP])
100     continue; /* skip zeroes */
101     colr_color(col, scanline[x]);
102     multcolor(col, scv);
103     addcolor(psp, col);
104 greg 2.1 }
105     }
106     fclose(fp); /* done this picture */
107     }
108     free(scanline);
109 greg 2.30 i = cm_write(pmat, myDT, fout); /* write picture */
110     cm_free(pmat); /* free data */
111 greg 2.53 ++runcnt; /* for zig-zagging */
112 greg 2.30 return(i);
113 greg 2.1 }
114    
115 greg 2.49 /* 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 greg 2.11 /* check to see if a string contains a %d or %o specification */
145     static int
146     hasNumberFormat(const char *s)
147 greg 2.1 {
148 greg 2.24 if (s == NULL)
149     return(0);
150    
151 greg 2.18 while (*s) {
152     while (*s != '%')
153     if (!*s++)
154     return(0);
155     if (*++s == '%') { /* ignore "%%" */
156     ++s;
157     continue;
158     }
159     while (isdigit(*s)) /* field length */
160     ++s;
161     /* field we'll use? */
162     if ((*s == 'd') | (*s == 'i') | (*s == 'o') |
163     (*s == 'x') | (*s == 'X'))
164     return(1);
165     }
166     return(0); /* didn't find one */
167 greg 2.1 }
168    
169     int
170     main(int argc, char *argv[])
171     {
172 greg 2.34 int skyfmt = DTfromHeader;
173 greg 2.30 int outfmt = DTascii;
174 greg 2.34 int headout = 1;
175     int nsteps = 0;
176 greg 2.24 char *ofspec = NULL;
177     FILE *ofp = stdout;
178 greg 2.47 int xres=0, yres=0;
179 greg 2.24 CMATRIX *cmtx; /* component vector/matrix result */
180     char fnbuf[256];
181     int a, i;
182 greg 2.54 /* set global progname */
183     fixargv0(argv[0]);
184 greg 2.24 /* get options */
185 greg 2.25 for (a = 1; a < argc && argv[a][0] == '-'; a++)
186     switch (argv[a][1]) {
187 greg 2.24 case 'n':
188     nsteps = atoi(argv[++a]);
189 greg 2.31 if (nsteps < 0)
190 greg 2.24 goto userr;
191 greg 2.34 skyfmt = nsteps ? DTascii : DTfromHeader;
192     break;
193     case 'h':
194     headout = !headout;
195 greg 2.24 break;
196 greg 2.25 case 'i':
197     switch (argv[a][2]) {
198     case 'f':
199     skyfmt = DTfloat;
200     break;
201     case 'd':
202     skyfmt = DTdouble;
203     break;
204     case 'a':
205     skyfmt = DTascii;
206     break;
207     default:
208     goto userr;
209     }
210     break;
211 greg 2.30 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 greg 2.43 case 'c':
226     outfmt = DTrgbe;
227     break;
228 greg 2.30 default:
229     goto userr;
230     }
231     break;
232 greg 2.47 case 'x':
233     xres = atoi(argv[++a]);
234     break;
235     case 'y':
236     yres = atoi(argv[++a]);
237     break;
238 greg 2.24 default:
239     goto userr;
240     }
241     if ((argc-a < 1) | (argc-a > 4))
242     goto userr;
243 greg 2.1
244 greg 2.24 if (argc-a > 2) { /* VTDs expression */
245 greg 2.36 CMATRIX *smtx, *Dmat, *Tmat, *imtx;
246     const char *ccp;
247 greg 2.24 /* get sky vector/matrix */
248 greg 2.25 smtx = cm_load(argv[a+3], 0, nsteps, skyfmt);
249 greg 2.35 nsteps = smtx->ncols;
250 greg 2.12 /* load BSDF */
251 greg 2.36 if (argv[a+1][0] != '!' &&
252 greg 2.52 (ccp = strrchr(argv[a+1], '.')) > argv[a+1] &&
253 greg 2.36 !strcasecmp(ccp+1, "XML"))
254     Tmat = cm_loadBTDF(argv[a+1]);
255     else
256     Tmat = cm_load(argv[a+1], 0, 0, DTfromHeader);
257 greg 2.5 /* load Daylight matrix */
258 greg 2.36 Dmat = cm_load(argv[a+2], Tmat->ncols,
259 greg 2.24 smtx->nrows, DTfromHeader);
260 greg 2.1 /* multiply vector through */
261 greg 2.24 imtx = cm_multiply(Dmat, smtx);
262     cm_free(Dmat); cm_free(smtx);
263 greg 2.29 cmtx = cm_multiply(Tmat, imtx);
264     cm_free(Tmat);
265 greg 2.24 cm_free(imtx);
266     } else { /* sky vector/matrix only */
267 greg 2.25 cmtx = cm_load(argv[a+1], 0, nsteps, skyfmt);
268 greg 2.35 nsteps = cmtx->ncols;
269 greg 2.24 }
270     /* prepare output stream */
271     if ((ofspec != NULL) & (nsteps == 1) && hasNumberFormat(ofspec)) {
272 greg 2.52 sprintf(fnbuf, ofspec, 0);
273 greg 2.24 ofspec = fnbuf;
274     }
275     if (ofspec != NULL && !hasNumberFormat(ofspec)) {
276     if ((ofp = fopen(ofspec, "w")) == NULL) {
277     fprintf(stderr, "%s: cannot open '%s' for output\n",
278     progname, ofspec);
279     return(1);
280     }
281     ofspec = NULL; /* only need to open once */
282 greg 2.12 }
283 greg 2.52 if (hasNumberFormat(argv[a])) { /* loading image vector(s) */
284 greg 2.44 if (outfmt != DTrgbe) {
285     error(WARNING, "changing output type to -oc");
286     outfmt = DTrgbe;
287     }
288 greg 2.24 if (ofspec == NULL) {
289     SET_FILE_BINARY(ofp);
290     newheader("RADIANCE", ofp);
291     printargs(argc, argv, ofp);
292     fputnow(ofp);
293     }
294     if (nsteps > 1) /* multiple output frames? */
295     for (i = 0; i < nsteps; i++) {
296     CMATRIX *cvec = cm_column(cmtx, i);
297     if (ofspec != NULL) {
298 greg 2.40 sprintf(fnbuf, ofspec, i);
299 greg 2.24 if ((ofp = fopen(fnbuf, "wb")) == NULL) {
300     fprintf(stderr,
301     "%s: cannot open '%s' for output\n",
302     progname, fnbuf);
303     return(1);
304     }
305     newheader("RADIANCE", ofp);
306     printargs(argc, argv, ofp);
307     fputnow(ofp);
308     }
309 greg 2.40 fprintf(ofp, "FRAME=%d\n", i);
310 greg 2.24 if (!sum_images(argv[a], cvec, ofp))
311     return(1);
312     if (ofspec != NULL) {
313     if (fclose(ofp) == EOF) {
314     fprintf(stderr,
315     "%s: error writing to '%s'\n",
316     progname, fnbuf);
317     return(1);
318     }
319     ofp = stdout;
320     }
321     cm_free(cvec);
322     }
323     else if (!sum_images(argv[a], cmtx, ofp))
324 greg 2.1 return(1);
325 greg 2.52 } else { /* loading view matrix */
326 greg 2.24 CMATRIX *Vmat = cm_load(argv[a], 0, cmtx->nrows, DTfromHeader);
327     CMATRIX *rmtx = cm_multiply(Vmat, cmtx);
328 greg 2.1 cm_free(Vmat);
329 greg 2.30 if (ofspec != NULL) { /* multiple vector files? */
330     const char *wtype = (outfmt==DTascii) ? "w" : "wb";
331 greg 2.24 for (i = 0; i < nsteps; i++) {
332     CMATRIX *rvec = cm_column(rmtx, i);
333 greg 2.49 if (alt_dim(rvec, yres, xres) < 0)
334     return(1);
335 greg 2.40 sprintf(fnbuf, ofspec, i);
336 greg 2.30 if ((ofp = fopen(fnbuf, wtype)) == NULL) {
337 greg 2.24 fprintf(stderr,
338     "%s: cannot open '%s' for output\n",
339     progname, fnbuf);
340     return(1);
341     }
342 greg 2.31 #ifdef getc_unlocked
343     flockfile(ofp);
344     #endif
345 greg 2.34 if (headout) { /* header output */
346     newheader("RADIANCE", ofp);
347     printargs(argc, argv, ofp);
348     fputnow(ofp);
349 greg 2.40 fprintf(ofp, "FRAME=%d\n", i);
350 greg 2.47 if ((outfmt != DTrgbe) & (outfmt != DTxyze)) {
351     fprintf(ofp, "NROWS=%d\n", rvec->nrows);
352     fprintf(ofp, "NCOLS=%d\n", rvec->ncols);
353 greg 2.52 fputncomp(3, ofp);
354 greg 2.47 }
355     if ((outfmt == DTfloat) | (outfmt == DTdouble))
356 greg 2.42 fputendian(ofp);
357 greg 2.46 fputformat(cm_fmt_id[outfmt], ofp);
358 greg 2.34 fputc('\n', ofp);
359     }
360 greg 2.30 cm_write(rvec, outfmt, ofp);
361 greg 2.24 if (fclose(ofp) == EOF) {
362     fprintf(stderr,
363     "%s: error writing to '%s'\n",
364     progname, fnbuf);
365     return(1);
366     }
367     ofp = stdout;
368     cm_free(rvec);
369     }
370 greg 2.30 } else {
371 greg 2.31 #ifdef getc_unlocked
372     flockfile(ofp);
373     #endif
374 greg 2.30 if (outfmt != DTascii)
375     SET_FILE_BINARY(ofp);
376 greg 2.49 if (alt_dim(rmtx, yres, xres) < 0)
377     return(1);
378 greg 2.34 if (headout) { /* header output */
379 greg 2.30 newheader("RADIANCE", ofp);
380     printargs(argc, argv, ofp);
381     fputnow(ofp);
382 greg 2.47 if ((outfmt != DTrgbe) & (outfmt != DTxyze)) {
383     fprintf(ofp, "NROWS=%d\n", rmtx->nrows);
384     fprintf(ofp, "NCOLS=%d\n", rmtx->ncols);
385 greg 2.52 fputncomp(3, ofp);
386 greg 2.47 }
387     if ((outfmt == DTfloat) | (outfmt == DTdouble))
388 greg 2.42 fputendian(ofp);
389 greg 2.46 fputformat(cm_fmt_id[outfmt], ofp);
390 greg 2.30 fputc('\n', ofp);
391     }
392     cm_write(rmtx, outfmt, ofp);
393     }
394 greg 2.24 cm_free(rmtx);
395     }
396     if (fflush(ofp) == EOF) { /* final clean-up */
397     fprintf(stderr, "%s: write error on output\n", progname);
398     return(1);
399 greg 2.1 }
400 greg 2.24 cm_free(cmtx);
401 greg 2.1 return(0);
402 greg 2.24 userr:
403 greg 2.48 fprintf(stderr, "Usage: %s [-n nsteps][-o ospec][-x xr][-y yr][-i{f|d|h}][-o{f|d|c}] DCspec [skyf]\n",
404 greg 2.24 progname);
405 greg 2.48 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 greg 2.24 progname);
407     return(1);
408 greg 2.1 }