ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/util/dctimestep.c
Revision: 2.36
Committed: Tue Jul 14 16:08:28 2015 UTC (8 years, 8 months ago) by greg
Content type: text/plain
Branch: MAIN
CVS Tags: rad5R0
Changes since 2.35: +11 -5 lines
Log Message:
Added ability for dctimestep to accept matrix files in place of XML for T

File Contents

# User Rev Content
1 greg 2.1 #ifndef lint
2 greg 2.36 static const char RCSid[] = "$Id: dctimestep.c,v 2.35 2014/09/26 23:33:09 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     #include "standard.h"
12 greg 2.28 #include "cmatrix.h"
13 greg 2.1 #include "platform.h"
14     #include "resolu.h"
15    
16     char *progname; /* global argv[0] */
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     int myDT = DTfromHeader;
23 greg 2.24 COLOR *scanline = NULL;
24     CMATRIX *pmat = NULL;
25     int myXR=0, myYR=0;
26 greg 2.1 int i, y;
27    
28     if (cv->ncols != 1)
29 greg 2.35 error(INTERNAL, "expected vector in sum_images()");
30 greg 2.1 for (i = 0; i < cv->nrows; i++) {
31     const COLORV *scv = cv_lval(cv,i);
32     char fname[1024];
33     FILE *fp;
34     int dt, xr, yr;
35     COLORV *psp;
36 greg 2.31 char *err;
37 greg 2.14 /* check for zero */
38 greg 2.19 if ((scv[RED] == 0) & (scv[GRN] == 0) & (scv[BLU] == 0) &&
39     (myDT != DTfromHeader) | (i < cv->nrows-1))
40 greg 2.14 continue;
41 greg 2.1 /* open next picture */
42     sprintf(fname, fspec, i);
43 greg 2.31 if ((fp = fopen(fname, "rb")) == NULL) {
44 greg 2.1 sprintf(errmsg, "cannot open picture '%s'", fname);
45     error(SYSTEM, errmsg);
46     }
47 greg 2.31 dt = DTfromHeader;
48     if ((err = cm_getheader(&dt, NULL, NULL, fp)) != NULL)
49     error(USER, err);
50 greg 2.1 if ((dt != DTrgbe) & (dt != DTxyze) ||
51     !fscnresolu(&xr, &yr, fp)) {
52     sprintf(errmsg, "file '%s' not a picture", fname);
53     error(USER, errmsg);
54     }
55     if (myDT == DTfromHeader) { /* on first one */
56     myDT = dt;
57     myXR = xr; myYR = yr;
58     scanline = (COLOR *)malloc(sizeof(COLOR)*myXR);
59     if (scanline == NULL)
60     error(SYSTEM, "out of memory in sum_images()");
61     pmat = cm_alloc(myYR, myXR);
62     memset(pmat->cmem, 0, sizeof(COLOR)*myXR*myYR);
63 greg 2.2 /* finish header */
64 greg 2.30 fputformat((char *)cm_fmt_id[myDT], fout);
65 greg 2.13 fputc('\n', fout);
66     fflush(fout);
67 greg 2.1 } else if ((dt != myDT) | (xr != myXR) | (yr != myYR)) {
68     sprintf(errmsg, "picture '%s' format/size mismatch",
69     fname);
70     error(USER, errmsg);
71     }
72     psp = pmat->cmem;
73     for (y = 0; y < yr; y++) { /* read it in */
74     int x;
75     if (freadscan(scanline, xr, fp) < 0) {
76     sprintf(errmsg, "error reading picture '%s'",
77     fname);
78     error(SYSTEM, errmsg);
79     }
80     /* sum in scanline */
81     for (x = 0; x < xr; x++, psp += 3) {
82     multcolor(scanline[x], scv);
83     addcolor(psp, scanline[x]);
84     }
85     }
86     fclose(fp); /* done this picture */
87     }
88     free(scanline);
89 greg 2.30 i = cm_write(pmat, myDT, fout); /* write picture */
90     cm_free(pmat); /* free data */
91     return(i);
92 greg 2.1 }
93    
94 greg 2.11 /* check to see if a string contains a %d or %o specification */
95     static int
96     hasNumberFormat(const char *s)
97 greg 2.1 {
98 greg 2.24 if (s == NULL)
99     return(0);
100    
101 greg 2.18 while (*s) {
102     while (*s != '%')
103     if (!*s++)
104     return(0);
105     if (*++s == '%') { /* ignore "%%" */
106     ++s;
107     continue;
108     }
109     while (isdigit(*s)) /* field length */
110     ++s;
111     /* field we'll use? */
112     if ((*s == 'd') | (*s == 'i') | (*s == 'o') |
113     (*s == 'x') | (*s == 'X'))
114     return(1);
115     }
116     return(0); /* didn't find one */
117 greg 2.1 }
118    
119     int
120     main(int argc, char *argv[])
121     {
122 greg 2.34 int skyfmt = DTfromHeader;
123 greg 2.30 int outfmt = DTascii;
124 greg 2.34 int headout = 1;
125     int nsteps = 0;
126 greg 2.24 char *ofspec = NULL;
127     FILE *ofp = stdout;
128     CMATRIX *cmtx; /* component vector/matrix result */
129     char fnbuf[256];
130     int a, i;
131 greg 2.1
132     progname = argv[0];
133 greg 2.24 /* get options */
134 greg 2.25 for (a = 1; a < argc && argv[a][0] == '-'; a++)
135     switch (argv[a][1]) {
136 greg 2.24 case 'n':
137     nsteps = atoi(argv[++a]);
138 greg 2.31 if (nsteps < 0)
139 greg 2.24 goto userr;
140 greg 2.34 skyfmt = nsteps ? DTascii : DTfromHeader;
141     break;
142     case 'h':
143     headout = !headout;
144 greg 2.24 break;
145 greg 2.25 case 'i':
146     switch (argv[a][2]) {
147     case 'f':
148     skyfmt = DTfloat;
149     break;
150     case 'd':
151     skyfmt = DTdouble;
152     break;
153     case 'a':
154     skyfmt = DTascii;
155     break;
156     default:
157     goto userr;
158     }
159     break;
160 greg 2.30 case 'o':
161     switch (argv[a][2]) {
162     case '\0': /* output specification (not format) */
163     ofspec = argv[++a];
164     break;
165     case 'f':
166     outfmt = DTfloat;
167     break;
168     case 'd':
169     outfmt = DTdouble;
170     break;
171     case 'a':
172     outfmt = DTascii;
173     break;
174     default:
175     goto userr;
176     }
177     break;
178 greg 2.24 default:
179     goto userr;
180     }
181     if ((argc-a < 1) | (argc-a > 4))
182     goto userr;
183 greg 2.1
184 greg 2.24 if (argc-a > 2) { /* VTDs expression */
185 greg 2.36 CMATRIX *smtx, *Dmat, *Tmat, *imtx;
186     const char *ccp;
187 greg 2.24 /* get sky vector/matrix */
188 greg 2.25 smtx = cm_load(argv[a+3], 0, nsteps, skyfmt);
189 greg 2.35 nsteps = smtx->ncols;
190 greg 2.12 /* load BSDF */
191 greg 2.36 if (argv[a+1][0] != '!' &&
192     (ccp = strrchr(argv[a+1], '.')) != NULL &&
193     !strcasecmp(ccp+1, "XML"))
194     Tmat = cm_loadBTDF(argv[a+1]);
195     else
196     Tmat = cm_load(argv[a+1], 0, 0, DTfromHeader);
197 greg 2.5 /* load Daylight matrix */
198 greg 2.36 Dmat = cm_load(argv[a+2], Tmat->ncols,
199 greg 2.24 smtx->nrows, DTfromHeader);
200 greg 2.1 /* multiply vector through */
201 greg 2.24 imtx = cm_multiply(Dmat, smtx);
202     cm_free(Dmat); cm_free(smtx);
203 greg 2.29 cmtx = cm_multiply(Tmat, imtx);
204     cm_free(Tmat);
205 greg 2.24 cm_free(imtx);
206     } else { /* sky vector/matrix only */
207 greg 2.25 cmtx = cm_load(argv[a+1], 0, nsteps, skyfmt);
208 greg 2.35 nsteps = cmtx->ncols;
209 greg 2.24 }
210     /* prepare output stream */
211     if ((ofspec != NULL) & (nsteps == 1) && hasNumberFormat(ofspec)) {
212     sprintf(fnbuf, ofspec, 1);
213     ofspec = fnbuf;
214     }
215     if (ofspec != NULL && !hasNumberFormat(ofspec)) {
216     if ((ofp = fopen(ofspec, "w")) == NULL) {
217     fprintf(stderr, "%s: cannot open '%s' for output\n",
218     progname, ofspec);
219     return(1);
220     }
221     ofspec = NULL; /* only need to open once */
222 greg 2.12 }
223 greg 2.24 if (hasNumberFormat(argv[a])) { /* generating image(s) */
224     if (ofspec == NULL) {
225     SET_FILE_BINARY(ofp);
226     newheader("RADIANCE", ofp);
227     printargs(argc, argv, ofp);
228     fputnow(ofp);
229     }
230     if (nsteps > 1) /* multiple output frames? */
231     for (i = 0; i < nsteps; i++) {
232     CMATRIX *cvec = cm_column(cmtx, i);
233     if (ofspec != NULL) {
234     sprintf(fnbuf, ofspec, i+1);
235     if ((ofp = fopen(fnbuf, "wb")) == NULL) {
236     fprintf(stderr,
237     "%s: cannot open '%s' for output\n",
238     progname, fnbuf);
239     return(1);
240     }
241     newheader("RADIANCE", ofp);
242     printargs(argc, argv, ofp);
243     fputnow(ofp);
244     }
245     fprintf(ofp, "FRAME=%d\n", i+1);
246     if (!sum_images(argv[a], cvec, ofp))
247     return(1);
248     if (ofspec != NULL) {
249     if (fclose(ofp) == EOF) {
250     fprintf(stderr,
251     "%s: error writing to '%s'\n",
252     progname, fnbuf);
253     return(1);
254     }
255     ofp = stdout;
256     }
257     cm_free(cvec);
258     }
259     else if (!sum_images(argv[a], cmtx, ofp))
260 greg 2.1 return(1);
261 greg 2.24 } else { /* generating vector/matrix */
262     CMATRIX *Vmat = cm_load(argv[a], 0, cmtx->nrows, DTfromHeader);
263     CMATRIX *rmtx = cm_multiply(Vmat, cmtx);
264 greg 2.1 cm_free(Vmat);
265 greg 2.30 if (ofspec != NULL) { /* multiple vector files? */
266     const char *wtype = (outfmt==DTascii) ? "w" : "wb";
267 greg 2.24 for (i = 0; i < nsteps; i++) {
268     CMATRIX *rvec = cm_column(rmtx, i);
269     sprintf(fnbuf, ofspec, i+1);
270 greg 2.30 if ((ofp = fopen(fnbuf, wtype)) == NULL) {
271 greg 2.24 fprintf(stderr,
272     "%s: cannot open '%s' for output\n",
273     progname, fnbuf);
274     return(1);
275     }
276 greg 2.31 #ifdef getc_unlocked
277     flockfile(ofp);
278     #endif
279 greg 2.34 if (headout) { /* header output */
280     newheader("RADIANCE", ofp);
281     printargs(argc, argv, ofp);
282     fputnow(ofp);
283     fprintf(ofp, "FRAME=%d\n", i+1);
284     fprintf(ofp, "NROWS=%d\n", rvec->nrows);
285     fputs("NCOLS=1\nNCOMP=3\n", ofp);
286     fputformat((char *)cm_fmt_id[outfmt], ofp);
287     fputc('\n', ofp);
288     }
289 greg 2.30 cm_write(rvec, outfmt, ofp);
290 greg 2.24 if (fclose(ofp) == EOF) {
291     fprintf(stderr,
292     "%s: error writing to '%s'\n",
293     progname, fnbuf);
294     return(1);
295     }
296     ofp = stdout;
297     cm_free(rvec);
298     }
299 greg 2.30 } else {
300 greg 2.31 #ifdef getc_unlocked
301     flockfile(ofp);
302     #endif
303 greg 2.30 if (outfmt != DTascii)
304     SET_FILE_BINARY(ofp);
305 greg 2.34 if (headout) { /* header output */
306 greg 2.30 newheader("RADIANCE", ofp);
307     printargs(argc, argv, ofp);
308     fputnow(ofp);
309 greg 2.33 fprintf(ofp, "NROWS=%d\n", rmtx->nrows);
310 greg 2.31 fprintf(ofp, "NCOLS=%d\n", rmtx->ncols);
311     fputs("NCOMP=3\n", ofp);
312 greg 2.30 fputformat((char *)cm_fmt_id[outfmt], ofp);
313     fputc('\n', ofp);
314     }
315     cm_write(rmtx, outfmt, ofp);
316     }
317 greg 2.24 cm_free(rmtx);
318     }
319     if (fflush(ofp) == EOF) { /* final clean-up */
320     fprintf(stderr, "%s: write error on output\n", progname);
321     return(1);
322 greg 2.1 }
323 greg 2.24 cm_free(cmtx);
324 greg 2.1 return(0);
325 greg 2.24 userr:
326 greg 2.32 fprintf(stderr, "Usage: %s [-n nsteps][-o ospec][-i{f|d|h}][-o{f|d}] DCspec [skyf]\n",
327 greg 2.24 progname);
328 greg 2.36 fprintf(stderr, " or: %s [-n nsteps][-o ospec][-i{f|d|h}][-o{f|d}] Vspec Tbsdf Dmat.dat [skyf]\n",
329 greg 2.24 progname);
330     return(1);
331 greg 2.1 }