ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/util/dctimestep.c
Revision: 2.35
Committed: Fri Sep 26 23:33:09 2014 UTC (9 years, 5 months ago) by greg
Content type: text/plain
Branch: MAIN
CVS Tags: rad4R2P2
Changes since 2.34: +4 -2 lines
Log Message:
Fixed bug in handling of sky matrices with default -n 0

File Contents

# Content
1 #ifndef lint
2 static const char RCSid[] = "$Id: dctimestep.c,v 2.34 2014/09/17 22:40:49 greg Exp $";
3 #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 #include "cmatrix.h"
13 #include "platform.h"
14 #include "resolu.h"
15
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 int myDT = DTfromHeader;
23 COLOR *scanline = NULL;
24 CMATRIX *pmat = NULL;
25 int myXR=0, myYR=0;
26 int i, y;
27
28 if (cv->ncols != 1)
29 error(INTERNAL, "expected vector in sum_images()");
30 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 char *err;
37 /* check for zero */
38 if ((scv[RED] == 0) & (scv[GRN] == 0) & (scv[BLU] == 0) &&
39 (myDT != DTfromHeader) | (i < cv->nrows-1))
40 continue;
41 /* open next picture */
42 sprintf(fname, fspec, i);
43 if ((fp = fopen(fname, "rb")) == NULL) {
44 sprintf(errmsg, "cannot open picture '%s'", fname);
45 error(SYSTEM, errmsg);
46 }
47 dt = DTfromHeader;
48 if ((err = cm_getheader(&dt, NULL, NULL, fp)) != NULL)
49 error(USER, err);
50 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 /* finish header */
64 fputformat((char *)cm_fmt_id[myDT], fout);
65 fputc('\n', fout);
66 fflush(fout);
67 } 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 i = cm_write(pmat, myDT, fout); /* write picture */
90 cm_free(pmat); /* free data */
91 return(i);
92 }
93
94 /* check to see if a string contains a %d or %o specification */
95 static int
96 hasNumberFormat(const char *s)
97 {
98 if (s == NULL)
99 return(0);
100
101 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 }
118
119 int
120 main(int argc, char *argv[])
121 {
122 int skyfmt = DTfromHeader;
123 int outfmt = DTascii;
124 int headout = 1;
125 int nsteps = 0;
126 char *ofspec = NULL;
127 FILE *ofp = stdout;
128 CMATRIX *cmtx; /* component vector/matrix result */
129 char fnbuf[256];
130 int a, i;
131
132 progname = argv[0];
133 /* get options */
134 for (a = 1; a < argc && argv[a][0] == '-'; a++)
135 switch (argv[a][1]) {
136 case 'n':
137 nsteps = atoi(argv[++a]);
138 if (nsteps < 0)
139 goto userr;
140 skyfmt = nsteps ? DTascii : DTfromHeader;
141 break;
142 case 'h':
143 headout = !headout;
144 break;
145 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 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 default:
179 goto userr;
180 }
181 if ((argc-a < 1) | (argc-a > 4))
182 goto userr;
183
184 if (argc-a > 2) { /* VTDs expression */
185 CMATRIX *smtx, *Dmat, *Tmat, *imtx;
186 /* get sky vector/matrix */
187 smtx = cm_load(argv[a+3], 0, nsteps, skyfmt);
188 nsteps = smtx->ncols;
189 /* load BSDF */
190 Tmat = cm_loadBTDF(argv[a+1]);
191 /* load Daylight matrix */
192 Dmat = cm_load(argv[a+2], Tmat==NULL ? 0 : Tmat->ncols,
193 smtx->nrows, DTfromHeader);
194 /* multiply vector through */
195 imtx = cm_multiply(Dmat, smtx);
196 cm_free(Dmat); cm_free(smtx);
197 cmtx = cm_multiply(Tmat, imtx);
198 cm_free(Tmat);
199 cm_free(imtx);
200 } else { /* sky vector/matrix only */
201 cmtx = cm_load(argv[a+1], 0, nsteps, skyfmt);
202 nsteps = cmtx->ncols;
203 }
204 /* prepare output stream */
205 if ((ofspec != NULL) & (nsteps == 1) && hasNumberFormat(ofspec)) {
206 sprintf(fnbuf, ofspec, 1);
207 ofspec = fnbuf;
208 }
209 if (ofspec != NULL && !hasNumberFormat(ofspec)) {
210 if ((ofp = fopen(ofspec, "w")) == NULL) {
211 fprintf(stderr, "%s: cannot open '%s' for output\n",
212 progname, ofspec);
213 return(1);
214 }
215 ofspec = NULL; /* only need to open once */
216 }
217 if (hasNumberFormat(argv[a])) { /* generating image(s) */
218 if (ofspec == NULL) {
219 SET_FILE_BINARY(ofp);
220 newheader("RADIANCE", ofp);
221 printargs(argc, argv, ofp);
222 fputnow(ofp);
223 }
224 if (nsteps > 1) /* multiple output frames? */
225 for (i = 0; i < nsteps; i++) {
226 CMATRIX *cvec = cm_column(cmtx, i);
227 if (ofspec != NULL) {
228 sprintf(fnbuf, ofspec, i+1);
229 if ((ofp = fopen(fnbuf, "wb")) == NULL) {
230 fprintf(stderr,
231 "%s: cannot open '%s' for output\n",
232 progname, fnbuf);
233 return(1);
234 }
235 newheader("RADIANCE", ofp);
236 printargs(argc, argv, ofp);
237 fputnow(ofp);
238 }
239 fprintf(ofp, "FRAME=%d\n", i+1);
240 if (!sum_images(argv[a], cvec, ofp))
241 return(1);
242 if (ofspec != NULL) {
243 if (fclose(ofp) == EOF) {
244 fprintf(stderr,
245 "%s: error writing to '%s'\n",
246 progname, fnbuf);
247 return(1);
248 }
249 ofp = stdout;
250 }
251 cm_free(cvec);
252 }
253 else if (!sum_images(argv[a], cmtx, ofp))
254 return(1);
255 } else { /* generating vector/matrix */
256 CMATRIX *Vmat = cm_load(argv[a], 0, cmtx->nrows, DTfromHeader);
257 CMATRIX *rmtx = cm_multiply(Vmat, cmtx);
258 cm_free(Vmat);
259 if (ofspec != NULL) { /* multiple vector files? */
260 const char *wtype = (outfmt==DTascii) ? "w" : "wb";
261 for (i = 0; i < nsteps; i++) {
262 CMATRIX *rvec = cm_column(rmtx, i);
263 sprintf(fnbuf, ofspec, i+1);
264 if ((ofp = fopen(fnbuf, wtype)) == NULL) {
265 fprintf(stderr,
266 "%s: cannot open '%s' for output\n",
267 progname, fnbuf);
268 return(1);
269 }
270 #ifdef getc_unlocked
271 flockfile(ofp);
272 #endif
273 if (headout) { /* header output */
274 newheader("RADIANCE", ofp);
275 printargs(argc, argv, ofp);
276 fputnow(ofp);
277 fprintf(ofp, "FRAME=%d\n", i+1);
278 fprintf(ofp, "NROWS=%d\n", rvec->nrows);
279 fputs("NCOLS=1\nNCOMP=3\n", ofp);
280 fputformat((char *)cm_fmt_id[outfmt], ofp);
281 fputc('\n', ofp);
282 }
283 cm_write(rvec, outfmt, ofp);
284 if (fclose(ofp) == EOF) {
285 fprintf(stderr,
286 "%s: error writing to '%s'\n",
287 progname, fnbuf);
288 return(1);
289 }
290 ofp = stdout;
291 cm_free(rvec);
292 }
293 } else {
294 #ifdef getc_unlocked
295 flockfile(ofp);
296 #endif
297 if (outfmt != DTascii)
298 SET_FILE_BINARY(ofp);
299 if (headout) { /* header output */
300 newheader("RADIANCE", ofp);
301 printargs(argc, argv, ofp);
302 fputnow(ofp);
303 fprintf(ofp, "NROWS=%d\n", rmtx->nrows);
304 fprintf(ofp, "NCOLS=%d\n", rmtx->ncols);
305 fputs("NCOMP=3\n", ofp);
306 fputformat((char *)cm_fmt_id[outfmt], ofp);
307 fputc('\n', ofp);
308 }
309 cm_write(rmtx, outfmt, ofp);
310 }
311 cm_free(rmtx);
312 }
313 if (fflush(ofp) == EOF) { /* final clean-up */
314 fprintf(stderr, "%s: write error on output\n", progname);
315 return(1);
316 }
317 cm_free(cmtx);
318 return(0);
319 userr:
320 fprintf(stderr, "Usage: %s [-n nsteps][-o ospec][-i{f|d|h}][-o{f|d}] DCspec [skyf]\n",
321 progname);
322 fprintf(stderr, " or: %s [-n nsteps][-o ospec][-i{f|d|h}][-o{f|d}] Vspec Tbsdf.xml Dmat.dat [skyf]\n",
323 progname);
324 return(1);
325 }