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.11 by greg, Wed Dec 9 20:33:48 2009 UTC vs.
Revision 2.19 by greg, Tue Feb 21 20:09:25 2012 UTC

# Line 10 | Line 10 | static const char RCSid[] = "$Id$";
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  
# Line 108 | Line 110 | cm_load(const char *fname, int nrows, int ncols, int d
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) {
# Line 289 | Line 293 | cm_print(const CMATRIX *cm, FILE *fp)
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);
301          int     nbadohm = 0;
302          int     nneg = 0;
303          int     r, c;
304 <        
304 >                                        /* reciprocity is "transparent" */
305          for (c = 0; c < cm->ncols; c++) {
306 <                float   dom = getBSDF_incohm(bsdf,c);
307 <                FVECT   v;
308 <                
305 <                if (dom <= .0) {
306 <                        nbadohm++;
307 <                        continue;
308 <                }
309 <                if (!getBSDF_incvec(v,bsdf,c) || v[2] > FTINY)
310 <                        error(USER, "illegal incoming BTDF direction");
311 <                dom *= -v[2];
306 >                const double    dom = mBSDF_incohm(bsdf,c);
307 >                                        /* projected solid angle */
308 >                nbadohm += (dom <= 0);
309  
310                  for (r = 0; r < cm->nrows; r++) {
311 <                        float   f = BSDF_value(bsdf,c,r);
311 >                        float   f = mBSDF_value(bsdf,c,r);
312                          COLORV  *mp = cm_lval(cm,r,c);
313 <
314 <                        if (f <= .0) {
313 >                                        /* check BSDF value */
314 >                        if ((f <= 0) | (dom <= 0)) {
315                                  nneg += (f < -FTINY);
316                                  f = .0f;
317                          }
318 <                        mp[0] = mp[1] = mp[2] = f * dom;
318 >                        copycolor(mp, specCol);
319 >                        scalecolor(mp, f);
320 >                        addcolor(mp, bsdfLamb);
321 >                        scalecolor(mp, dom);
322                  }
323          }
324 <        if (nneg || nbadohm) {
324 >        if (nneg | nbadohm) {
325                  sprintf(errmsg,
326                      "BTDF has %d negatives and %d bad incoming solid angles",
327                                  nneg, nbadohm);
# Line 330 | Line 330 | cm_bsdf(const struct BSDF_data *bsdf)
330          return(cm);
331   }
332  
333 + /* Load and convert a matrix BSDF from the given XML file */
334 + static CMATRIX *
335 + cm_loadBSDF(char *fname)
336 + {
337 +        CMATRIX *Tmat;
338 +        char    *fpath;
339 +        SDError ec;
340 +        SDData  myBSDF;
341 +        COLOR   bsdfLamb, specCol;
342 +                                        /* find path to BSDF file */
343 +        fpath = getpath(fname, getrlibpath(), R_OK);
344 +        if (fpath == NULL) {
345 +                sprintf(errmsg, "cannot find BSDF file '%s'", fname);
346 +                error(USER, errmsg);
347 +        }
348 +        SDclearBSDF(&myBSDF, fname);    /* load XML and check type */
349 +        ec = SDloadFile(&myBSDF, fpath);
350 +        if (ec)
351 +                error(USER, transSDError(ec));
352 +        if (myBSDF.tf == NULL || myBSDF.tf->ncomp != 1 ||
353 +                        myBSDF.tf->comp[0].func != &SDhandleMtx) {
354 +                sprintf(errmsg, "unsupported BSDF '%s'", fpath);
355 +                error(USER, errmsg);
356 +        }
357 +                                        /* convert BTDF to matrix */
358 +        ccy2rgb(&myBSDF.tLamb.spec, myBSDF.tLamb.cieY/PI, bsdfLamb);
359 +        ccy2rgb(&myBSDF.tf->comp[0].cspec[0], 1., specCol);
360 +        Tmat = cm_bsdf(bsdfLamb, specCol, (SDMat *)myBSDF.tf->comp[0].dist);
361 +                                        /* free BSDF and return */
362 +        SDfreeBSDF(&myBSDF);
363 +        return(Tmat);
364 + }
365 +
366   /* Sum together a set of images and write result to stdout */
367   static int
368 < sum_images(const char *fspec, const CMATRIX *cv)
368 > sum_images(const char *fspec, const CMATRIX *cv, FILE *fout)
369   {
370          int     myDT = DTfromHeader;
371          CMATRIX *pmat;
# Line 348 | Line 381 | sum_images(const char *fspec, const CMATRIX *cv)
381                  FILE            *fp;
382                  int             dt, xr, yr;
383                  COLORV          *psp;
384 +                                                        /* check for zero */
385 +                if ((scv[RED] == 0) & (scv[GRN] == 0) & (scv[BLU] == 0) &&
386 +                                (myDT != DTfromHeader) | (i < cv->nrows-1))
387 +                        continue;
388                                                          /* open next picture */
389                  sprintf(fname, fspec, i);
390                  if ((fp = fopen(fname, "r")) == NULL) {
# Line 370 | Line 407 | sum_images(const char *fspec, const CMATRIX *cv)
407                          pmat = cm_alloc(myYR, myXR);
408                          memset(pmat->cmem, 0, sizeof(COLOR)*myXR*myYR);
409                                                          /* finish header */
410 <                        fputformat(myDT==DTrgbe ? COLRFMT : CIEFMT, stdout);
411 <                        fputc('\n', stdout);
412 <                        fprtresolu(myXR, myYR, stdout);
413 <                        fflush(stdout);
410 >                        fputformat(myDT==DTrgbe ? COLRFMT : CIEFMT, fout);
411 >                        fputc('\n', fout);
412 >                        fprtresolu(myXR, myYR, fout);
413 >                        fflush(fout);
414                  } else if ((dt != myDT) | (xr != myXR) | (yr != myYR)) {
415                          sprintf(errmsg, "picture '%s' format/size mismatch",
416                                          fname);
# Line 398 | Line 435 | sum_images(const char *fspec, const CMATRIX *cv)
435          free(scanline);
436                                                          /* write scanlines */
437          for (y = 0; y < myYR; y++)
438 <                if (fwritescan((COLOR *)cm_lval(pmat, y, 0), myXR, stdout) < 0)
438 >                if (fwritescan((COLOR *)cm_lval(pmat, y, 0), myXR, fout) < 0)
439                          return(0);
440          cm_free(pmat);                                  /* all done */
441 <        return(fflush(stdout) == 0);
441 >        return(fflush(fout) == 0);
442   }
443  
444   /* check to see if a string contains a %d or %o specification */
445   static int
446   hasNumberFormat(const char *s)
447   {
448 <        while (*s && *s != '%')
449 <                s++;
450 <        if (!*s)
451 <                return(0);
452 <        do
453 <                ++s;
454 <        while (isdigit(*s));
455 <
456 <        return(*s == 'd' | *s == 'i' | *s == 'o' | *s == 'x' | *s == 'X');
448 >        while (*s) {
449 >                while (*s != '%')
450 >                        if (!*s++)
451 >                                return(0);
452 >                if (*++s == '%') {              /* ignore "%%" */
453 >                        ++s;
454 >                        continue;
455 >                }
456 >                while (isdigit(*s))             /* field length */
457 >                        ++s;
458 >                                                /* field we'll use? */
459 >                if ((*s == 'd') | (*s == 'i') | (*s == 'o') |
460 >                                        (*s == 'x') | (*s == 'X'))
461 >                        return(1);
462 >        }
463 >        return(0);                              /* didn't find one */
464   }
465  
466   int
467   main(int argc, char *argv[])
468   {
469 <        CMATRIX                 *tvec, *Dmat, *Tmat, *ivec, *cvec;
426 <        struct BSDF_data        *btdf;
469 >        CMATRIX                 *cvec;
470  
471          progname = argv[0];
472  
473 <        if ((argc < 4) | (argc > 5)) {
474 <                fprintf(stderr, "Usage: %s Vspec Tbsdf.xml Dmat.dat [tregvec]\n",
475 <                                progname);
473 >        if ((argc < 2) | (argc > 5)) {
474 >                fprintf(stderr, "Usage: %s DCspec [tregvec]\n", progname);
475 >                fprintf(stderr, "   or: %s Vspec Tbsdf.xml Dmat.dat [tregvec]\n",
476 >                                        progname);
477                  return(1);
478          }
479 <                                                /* load Tregenza vector */
480 <        tvec = cm_load(argv[4], 0, 1, DTascii); /* argv[4]==NULL iff argc==4 */
481 <                                                /* load BTDF */
482 <        btdf = load_BSDF(argv[2]);
483 <        if (btdf == NULL)
484 <                return(1);
479 >
480 >        if (argc > 3) {                         /* VTDs expression */
481 >                CMATRIX *svec, *Dmat, *Tmat, *ivec;
482 >                                                /* get sky vector */
483 >                svec = cm_load(argv[4], 0, 1, DTascii);
484 >                                                /* load BSDF */
485 >                Tmat = cm_loadBSDF(argv[2]);
486                                                  /* load Daylight matrix */
487 <        Dmat = cm_load(argv[3], btdf->ninc, tvec->nrows, DTfromHeader);
487 >                Dmat = cm_load(argv[3], Tmat->ncols, svec->nrows, DTfromHeader);
488                                                  /* multiply vector through */
489 <        ivec = cm_multiply(Dmat, tvec);
490 <        cm_free(Dmat); cm_free(tvec);
491 <        Tmat = cm_bsdf(btdf);                   /* convert BTDF to matrix */
492 <        free_BSDF(btdf);
493 <        cvec = cm_multiply(Tmat, ivec);         /* cvec = component vector */
494 <        cm_free(Tmat); cm_free(ivec);
489 >                ivec = cm_multiply(Dmat, svec);
490 >                cm_free(Dmat); cm_free(svec);
491 >                cvec = cm_multiply(Tmat, ivec); /* cvec = component vector */
492 >                cm_free(Tmat); cm_free(ivec);
493 >        } else {                                /* else just use sky vector */
494 >                cvec = cm_load(argv[2], 0, 1, DTascii);
495 >        }
496 >
497          if (hasNumberFormat(argv[1])) {         /* generating image */
498                  SET_FILE_BINARY(stdout);
499                  newheader("RADIANCE", stdout);
500                  printargs(argc, argv, stdout);
501                  fputnow(stdout);
502 <                if (!sum_images(argv[1], cvec))
502 >                if (!sum_images(argv[1], cvec, stdout))
503                          return(1);
504          } else {                                /* generating vector */
505                  CMATRIX *Vmat = cm_load(argv[1], 0, cvec->nrows, DTfromHeader);

Diff Legend

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