10 |
|
#include <ctype.h> |
11 |
|
#include "standard.h" |
12 |
|
#include "platform.h" |
13 |
+ |
#include "paths.h" |
14 |
|
#include "color.h" |
15 |
|
#include "resolu.h" |
16 |
|
#include "bsdf.h" |
17 |
+ |
#include "bsdf_m.h" |
18 |
|
|
19 |
|
char *progname; /* global argv[0] */ |
20 |
|
|
40 |
|
CMATRIX *cm; |
41 |
|
|
42 |
|
if ((nrows <= 0) | (ncols <= 0)) |
43 |
< |
return(NULL); |
43 |
> |
error(USER, "attempt to create empty matrix"); |
44 |
|
cm = (CMATRIX *)malloc(sizeof(CMATRIX) + |
45 |
|
3*sizeof(COLORV)*(nrows*ncols - 1)); |
46 |
|
if (cm == NULL) |
110 |
|
CMATRIX *cm; |
111 |
|
FILE *fp = stdin; |
112 |
|
|
113 |
+ |
if (ncols <= 0) |
114 |
+ |
error(USER, "Non-positive number of columns"); |
115 |
|
if (fname == NULL) |
116 |
|
fname = "<stdin>"; |
117 |
|
else if ((fp = fopen(fname, "r")) == NULL) { |
164 |
|
int maxrow = (nrows > 0 ? nrows : 32000); |
165 |
|
int r, c; |
166 |
|
for (r = 0; r < maxrow; r++) { |
167 |
< |
if (r >= cm->nrows) /* need more space? */ |
167 |
> |
if (r >= cm->nrows) /* need more space? */ |
168 |
|
cm = cm_resize(cm, 2*cm->nrows); |
169 |
|
for (c = 0; c < ncols; c++) { |
170 |
|
COLORV *cv = cm_lval(cm,r,c); |
171 |
|
if (fscanf(fp, COLSPEC, cv, cv+1, cv+2) != 3) |
172 |
< |
if ((nrows <= 0) & (r > 0) & (c == 0)) { |
172 |
> |
if ((nrows <= 0) & (r > 0) & !c) { |
173 |
|
cm = cm_resize(cm, maxrow=r); |
174 |
|
break; |
175 |
|
} else |
197 |
|
if (nread == cm->nrows*cm->ncols) |
198 |
|
/* need more space? */ |
199 |
|
cm = cm_resize(cm, 2*cm->nrows); |
200 |
< |
else if (nread % cm->ncols == 0) |
200 |
> |
else if (nread && !(nread % cm->ncols)) |
201 |
|
/* seem to be done */ |
202 |
|
cm = cm_resize(cm, nread/cm->ncols); |
203 |
|
else /* ended mid-row */ |
293 |
|
} |
294 |
|
} |
295 |
|
|
296 |
< |
/* convert a BSDF to our matrix representation */ |
296 |
> |
/* Convert a BSDF to our matrix representation */ |
297 |
|
static CMATRIX * |
298 |
< |
cm_bsdf(const struct BSDF_data *bsdf) |
298 |
> |
cm_bsdf(const COLOR bsdfLamb, const COLOR specCol, const SDMat *bsdf) |
299 |
|
{ |
300 |
< |
CMATRIX *cm = cm_alloc(bsdf->nout, bsdf->ninc); |
297 |
< |
COLORV *mp = cm->cmem; |
300 |
> |
CMATRIX *cm; |
301 |
|
int nbadohm = 0; |
302 |
|
int nneg = 0; |
303 |
+ |
FVECT v; |
304 |
+ |
float dom; |
305 |
+ |
int doforward; |
306 |
|
int r, c; |
307 |
< |
|
308 |
< |
for (r = 0; r < cm->nrows; r++) |
309 |
< |
for (c = 0; c < cm->ncols; c++, mp += 3) { |
310 |
< |
float f = BSDF_value(bsdf,c,r); |
311 |
< |
float dom = getBSDF_incohm(bsdf,c); |
312 |
< |
FVECT v; |
307 |
> |
|
308 |
> |
mBSDF_incvec(v, bsdf, .5); /* check BTDF orientation */ |
309 |
> |
|
310 |
> |
if (v[2] < 0) { /* incident from outside */ |
311 |
> |
cm = cm_alloc(bsdf->nout, bsdf->ninc); |
312 |
> |
for (c = 0; c < cm->ncols; c++) { |
313 |
> |
dom = mBSDF_incohm(bsdf,c); |
314 |
|
|
315 |
< |
if (f <= .0) { |
316 |
< |
nneg += (f < -FTINY); |
317 |
< |
continue; |
315 |
> |
nbadohm += (dom <= 0); |
316 |
> |
|
317 |
> |
if (!mBSDF_incvec(v,bsdf,c+.5) || v[2] > 0) |
318 |
> |
error(USER, "illegal incoming BTDF direction"); |
319 |
> |
dom *= -v[2]; |
320 |
> |
|
321 |
> |
for (r = 0; r < cm->nrows; r++) { |
322 |
> |
float f = mBSDF_value(bsdf,c,r); |
323 |
> |
COLORV *mp = cm_lval(cm,r,c); |
324 |
> |
|
325 |
> |
if ((f <= 0) | (dom <= 0)) { |
326 |
> |
nneg += (f < -FTINY); |
327 |
> |
f = .0f; |
328 |
> |
} |
329 |
> |
copycolor(mp, specCol); |
330 |
> |
f *= dom; |
331 |
> |
scalecolor(mp, f); |
332 |
> |
addcolor(mp, bsdfLamb); |
333 |
|
} |
334 |
< |
if (dom <= .0) { |
335 |
< |
nbadohm++; |
336 |
< |
continue; |
334 |
> |
} |
335 |
> |
} else { /* rely on reciprocity */ |
336 |
> |
cm = cm_alloc(bsdf->ninc, bsdf->nout); |
337 |
> |
for (c = 0; c < cm->ncols; c++) { |
338 |
> |
dom = mBSDF_outohm(bsdf,c); |
339 |
> |
|
340 |
> |
nbadohm += (dom <= 0); |
341 |
> |
|
342 |
> |
if (!mBSDF_outvec(v,bsdf,c+.5) || v[2] > 0) |
343 |
> |
error(USER, "illegal outgoing BTDF direction"); |
344 |
> |
dom *= -v[2]; |
345 |
> |
|
346 |
> |
for (r = 0; r < cm->nrows; r++) { |
347 |
> |
float f = mBSDF_value(bsdf,r,c); |
348 |
> |
COLORV *mp = cm_lval(cm,r,c); |
349 |
> |
|
350 |
> |
if ((f <= 0) | (dom <= 0)) { |
351 |
> |
nneg += (f < -FTINY); |
352 |
> |
f = .0f; |
353 |
> |
} |
354 |
> |
copycolor(mp, specCol); |
355 |
> |
f *= dom; |
356 |
> |
scalecolor(mp, f); |
357 |
> |
addcolor(mp, bsdfLamb); |
358 |
|
} |
316 |
– |
if (!getBSDF_incvec(v,bsdf,c) || v[2] > FTINY) |
317 |
– |
error(USER, "illegal incoming BTDF direction"); |
318 |
– |
|
319 |
– |
mp[0] = mp[1] = mp[2] = f * dom * -v[2]; |
359 |
|
} |
360 |
< |
if (nbadohm | nneg) { |
360 |
> |
} |
361 |
> |
if (nneg | nbadohm) { |
362 |
|
sprintf(errmsg, |
363 |
< |
"BTDF has %d negatives and %d bad incoming solid angles"); |
363 |
> |
"BTDF has %d negatives and %d bad incoming solid angles", |
364 |
> |
nneg, nbadohm); |
365 |
|
error(WARNING, errmsg); |
366 |
|
} |
367 |
|
return(cm); |
368 |
|
} |
369 |
|
|
370 |
+ |
/* Load and convert a matrix BSDF from the given XML file */ |
371 |
+ |
static CMATRIX * |
372 |
+ |
cm_loadBSDF(char *fname) |
373 |
+ |
{ |
374 |
+ |
CMATRIX *Tmat; |
375 |
+ |
char *fpath; |
376 |
+ |
SDError ec; |
377 |
+ |
SDData myBSDF; |
378 |
+ |
COLOR bsdfLamb, specCol; |
379 |
+ |
/* find path to BSDF file */ |
380 |
+ |
fpath = getpath(fname, getrlibpath(), R_OK); |
381 |
+ |
if (fpath == NULL) { |
382 |
+ |
sprintf(errmsg, "cannot find BSDF file '%s'", fname); |
383 |
+ |
error(USER, errmsg); |
384 |
+ |
} |
385 |
+ |
SDclipName(myBSDF.name, fname); |
386 |
+ |
/* load XML and check type */ |
387 |
+ |
ec = SDloadFile(&myBSDF, fpath); |
388 |
+ |
if (ec) |
389 |
+ |
error(USER, transSDError(ec)); |
390 |
+ |
if (myBSDF.tf == NULL || myBSDF.tf->ncomp != 1 || |
391 |
+ |
myBSDF.tf->comp[0].func != &SDhandleMtx) { |
392 |
+ |
sprintf(errmsg, "unsupported BSDF '%s'", fpath); |
393 |
+ |
error(USER, errmsg); |
394 |
+ |
} |
395 |
+ |
/* convert BTDF to matrix */ |
396 |
+ |
ccy2rgb(&myBSDF.tLamb.spec, myBSDF.tLamb.cieY/PI, bsdfLamb); |
397 |
+ |
ccy2rgb(&myBSDF.tf->comp[0].cspec[0], 1., specCol); |
398 |
+ |
Tmat = cm_bsdf(bsdfLamb, specCol, (SDMat *)myBSDF.tf->comp[0].dist); |
399 |
+ |
/* free BSDF and return */ |
400 |
+ |
SDfreeBSDF(&myBSDF); |
401 |
+ |
return(Tmat); |
402 |
+ |
} |
403 |
+ |
|
404 |
|
/* Sum together a set of images and write result to stdout */ |
405 |
|
static int |
406 |
< |
sum_images(const char *fspec, const CMATRIX *cv) |
406 |
> |
sum_images(const char *fspec, const CMATRIX *cv, FILE *fout) |
407 |
|
{ |
408 |
|
int myDT = DTfromHeader; |
409 |
|
CMATRIX *pmat; |
419 |
|
FILE *fp; |
420 |
|
int dt, xr, yr; |
421 |
|
COLORV *psp; |
422 |
+ |
/* check for zero */ |
423 |
+ |
if ((scv[RED] == 0) & (scv[GRN] == 0) & (scv[BLU] == 0)) |
424 |
+ |
continue; |
425 |
|
/* open next picture */ |
426 |
|
sprintf(fname, fspec, i); |
427 |
|
if ((fp = fopen(fname, "r")) == NULL) { |
444 |
|
pmat = cm_alloc(myYR, myXR); |
445 |
|
memset(pmat->cmem, 0, sizeof(COLOR)*myXR*myYR); |
446 |
|
/* finish header */ |
447 |
< |
fputformat(myDT==DTrgbe ? COLRFMT : CIEFMT, stdout); |
448 |
< |
fputc('\n', stdout); |
449 |
< |
fprtresolu(myXR, myYR, stdout); |
450 |
< |
fflush(stdout); |
447 |
> |
fputformat(myDT==DTrgbe ? COLRFMT : CIEFMT, fout); |
448 |
> |
fputc('\n', fout); |
449 |
> |
fprtresolu(myXR, myYR, fout); |
450 |
> |
fflush(fout); |
451 |
|
} else if ((dt != myDT) | (xr != myXR) | (yr != myYR)) { |
452 |
|
sprintf(errmsg, "picture '%s' format/size mismatch", |
453 |
|
fname); |
472 |
|
free(scanline); |
473 |
|
/* write scanlines */ |
474 |
|
for (y = 0; y < myYR; y++) |
475 |
< |
if (fwritescan((COLOR *)cm_lval(pmat, y, 0), myXR, stdout) < 0) |
475 |
> |
if (fwritescan((COLOR *)cm_lval(pmat, y, 0), myXR, fout) < 0) |
476 |
|
return(0); |
477 |
|
cm_free(pmat); /* all done */ |
478 |
< |
return(fflush(stdout) == 0); |
478 |
> |
return(fflush(fout) == 0); |
479 |
|
} |
480 |
|
|
481 |
< |
/* check to see if a string contains a %d specification */ |
482 |
< |
int |
483 |
< |
hasDecimalSpec(const char *s) |
481 |
> |
/* check to see if a string contains a %d or %o specification */ |
482 |
> |
static int |
483 |
> |
hasNumberFormat(const char *s) |
484 |
|
{ |
485 |
|
while (*s && *s != '%') |
486 |
|
s++; |
490 |
|
++s; |
491 |
|
while (isdigit(*s)); |
492 |
|
|
493 |
< |
return(*s == 'd'); |
493 |
> |
return((*s == 'd') | (*s == 'i') | (*s == 'o') | |
494 |
> |
(*s == 'x') | (*s == 'X')); |
495 |
|
} |
496 |
|
|
497 |
|
int |
498 |
|
main(int argc, char *argv[]) |
499 |
|
{ |
500 |
< |
CMATRIX *tvec, *Dmat, *Tmat, *ivec, *cvec; |
422 |
< |
struct BSDF_data *btdf; |
500 |
> |
CMATRIX *cvec; |
501 |
|
|
502 |
|
progname = argv[0]; |
503 |
|
|
504 |
< |
if ((argc < 4) | (argc > 5)) { |
505 |
< |
fprintf(stderr, "Usage: %s Vspec Tbsdf.xml Dmat.dat [tregvec]\n", |
506 |
< |
progname); |
504 |
> |
if ((argc < 2) | (argc > 5)) { |
505 |
> |
fprintf(stderr, "Usage: %s DCspec [tregvec]\n", progname); |
506 |
> |
fprintf(stderr, " or: %s Vspec Tbsdf.xml Dmat.dat [tregvec]\n", |
507 |
> |
progname); |
508 |
|
return(1); |
509 |
|
} |
510 |
< |
tvec = cm_load(argv[4], 0, 1, DTascii); /* argv[4]==NULL iff argc==4 */ |
511 |
< |
Dmat = cm_load(argv[3], 0, tvec->nrows, DTfromHeader); |
512 |
< |
btdf = load_BSDF(argv[2]); |
513 |
< |
if (btdf == NULL) |
514 |
< |
return(1); |
515 |
< |
if (btdf->ninc != Dmat->nrows) { |
516 |
< |
sprintf(errmsg, "Incoming BTDF dir (%d) mismatch to D (%d)", |
517 |
< |
btdf->ninc, Dmat->nrows); |
518 |
< |
error(USER, errmsg); |
440 |
< |
} |
510 |
> |
|
511 |
> |
if (argc > 3) { /* VTDs expression */ |
512 |
> |
CMATRIX *svec, *Dmat, *Tmat, *ivec; |
513 |
> |
/* get sky vector */ |
514 |
> |
svec = cm_load(argv[4], 0, 1, DTascii); |
515 |
> |
/* load BSDF */ |
516 |
> |
Tmat = cm_loadBSDF(argv[2]); |
517 |
> |
/* load Daylight matrix */ |
518 |
> |
Dmat = cm_load(argv[3], Tmat->ncols, svec->nrows, DTfromHeader); |
519 |
|
/* multiply vector through */ |
520 |
< |
ivec = cm_multiply(Dmat, tvec); |
521 |
< |
cm_free(Dmat); cm_free(tvec); |
522 |
< |
Tmat = cm_bsdf(btdf); /* convert BTDF to matrix */ |
523 |
< |
free_BSDF(btdf); |
524 |
< |
cvec = cm_multiply(Tmat, ivec); /* cvec = component vector */ |
525 |
< |
cm_free(Tmat); cm_free(ivec); |
526 |
< |
if (hasDecimalSpec(argv[1])) { /* generating image */ |
520 |
> |
ivec = cm_multiply(Dmat, svec); |
521 |
> |
cm_free(Dmat); cm_free(svec); |
522 |
> |
cvec = cm_multiply(Tmat, ivec); /* cvec = component vector */ |
523 |
> |
cm_free(Tmat); cm_free(ivec); |
524 |
> |
} else { /* else just use sky vector */ |
525 |
> |
cvec = cm_load(argv[2], 0, 1, DTascii); |
526 |
> |
} |
527 |
> |
|
528 |
> |
if (hasNumberFormat(argv[1])) { /* generating image */ |
529 |
|
SET_FILE_BINARY(stdout); |
530 |
|
newheader("RADIANCE", stdout); |
531 |
|
printargs(argc, argv, stdout); |
532 |
|
fputnow(stdout); |
533 |
< |
if (!sum_images(argv[1], cvec)) |
533 |
> |
if (!sum_images(argv[1], cvec, stdout)) |
534 |
|
return(1); |
535 |
|
} else { /* generating vector */ |
536 |
|
CMATRIX *Vmat = cm_load(argv[1], 0, cvec->nrows, DTfromHeader); |