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.22 by greg, Tue Oct 23 22:24:09 2012 UTC vs.
Revision 2.25 by greg, Fri Jan 11 17:21:39 2013 UTC

# Line 107 | Line 107 | getDTfromHeader(FILE *fp)
107   static CMATRIX *
108   cm_load(const char *fname, int nrows, int ncols, int dtype)
109   {
110        CMATRIX *cm;
110          FILE    *fp = stdin;
111 +        CMATRIX *cm;
112  
113          if (ncols <= 0)
114                  error(USER, "Non-positive number of columns");
# Line 118 | Line 118 | cm_load(const char *fname, int nrows, int ncols, int d
118                  sprintf(errmsg, "cannot open file '%s'", fname);
119                  error(SYSTEM, errmsg);
120          }
121 + #ifdef getc_unlocked
122 +        flockfile(fp);
123 + #endif
124          if (dtype != DTascii)
125                  SET_FILE_BINARY(fp);
126          if (dtype == DTfromHeader)
# Line 158 | Line 161 | cm_load(const char *fname, int nrows, int ncols, int d
161                  cm = cm_alloc(guessrows, ncols);
162          } else
163                  cm = cm_alloc(nrows, ncols);
164 <        if (cm == NULL)
164 >        if (cm == NULL)                                 /* XXX never happens */
165                  return(NULL);
166          if (dtype == DTascii) {                         /* read text file */
167                  int     maxrow = (nrows > 0 ? nrows : 32000);
# Line 233 | Line 236 | cm_load(const char *fname, int nrows, int ncols, int d
236                                  cvp += 3;
237                          }
238                  }
239 <                if (getc(fp) != EOF) {
239 >                if (fgetc(fp) != EOF) {
240                                  sprintf(errmsg,
241                                  "unexpected data at end of binary file %s",
242                                                  fname);
# Line 242 | Line 245 | cm_load(const char *fname, int nrows, int ncols, int d
245          }
246          if (fp != stdin)
247                  fclose(fp);
248 + #ifdef getc_unlocked
249 +        else
250 +                funlockfile(fp);
251 + #endif
252          return(cm);
253   EOFerror:
254 <        sprintf(errmsg, "unexpected EOF reading %s",
248 <                        fname);
254 >        sprintf(errmsg, "unexpected EOF reading %s", fname);
255          error(USER, errmsg);
256   not_handled:
257          error(INTERNAL, "unhandled data size or length in cm_load()");
258          return(NULL);   /* gratis return */
259   }
260  
261 + /* Extract a column vector from a matrix */
262 + static CMATRIX *
263 + cm_column(const CMATRIX *cm, int c)
264 + {
265 +        CMATRIX *cvr;
266 +        int     dr;
267 +
268 +        if ((c < 0) | (c >= cm->ncols))
269 +                return(NULL);
270 +        cvr = cm_alloc(cm->nrows, 1);
271 +        if (cvr == NULL)
272 +                return(NULL);
273 +        for (dr = 0; dr < cm->nrows; dr++) {
274 +                const COLORV    *sp = cm_lval(cm,dr,c);
275 +                COLORV          *dp = cv_lval(cvr,dr);
276 +                dp[0] = sp[0];
277 +                dp[1] = sp[1];
278 +                dp[2] = sp[2];
279 +        }
280 +        return(cvr);
281 + }
282 +
283   /* Scale a matrix by a single value */
284   static CMATRIX *
285   cm_scale(const CMATRIX *cm1, const COLOR sca)
# Line 436 | Line 464 | cm_loadBSDF(char *fname, COLOR cLamb)
464          if (ec)
465                  error(USER, transSDError(ec));
466          ccy2rgb(&myBSDF.tLamb.spec, myBSDF.tLamb.cieY/PI, bsdfLamb);
467 <        recip = (myBSDF.tf == NULL);
468 <        tdf = recip ? myBSDF.tb : myBSDF.tf;
467 >        recip = (myBSDF.tb == NULL);
468 >        tdf = recip ? myBSDF.tf : myBSDF.tb;
469          if (tdf == NULL) {              /* no non-Lambertian transmission? */
470                  if (cLamb != NULL)
471                          copycolor(cLamb, bsdfLamb);
# Line 459 | Line 487 | cm_loadBSDF(char *fname, COLOR cLamb)
487          return(Tmat);
488   }
489  
490 < /* Sum together a set of images and write result to stdout */
490 > /* Sum together a set of images and write result to fout */
491   static int
492   sum_images(const char *fspec, const CMATRIX *cv, FILE *fout)
493   {
494          int     myDT = DTfromHeader;
495 <        CMATRIX *pmat;
496 <        COLOR   *scanline;
497 <        int     myXR, myYR;
495 >        COLOR   *scanline = NULL;
496 >        CMATRIX *pmat = NULL;
497 >        int     myXR=0, myYR=0;
498          int     i, y;
499  
500          if (cv->ncols != 1)
# Line 541 | Line 569 | sum_images(const char *fspec, const CMATRIX *cv, FILE
569   static int
570   hasNumberFormat(const char *s)
571   {
572 +        if (s == NULL)
573 +                return(0);
574 +
575          while (*s) {
576                  while (*s != '%')
577                          if (!*s++)
# Line 562 | Line 593 | hasNumberFormat(const char *s)
593   int
594   main(int argc, char *argv[])
595   {
596 <        CMATRIX                 *cvec;          /* component vector result */
596 >        int             skyfmt = DTascii;
597 >        int             nsteps = 1;
598 >        char            *ofspec = NULL;
599 >        FILE            *ofp = stdout;
600 >        CMATRIX         *cmtx;          /* component vector/matrix result */
601 >        char            fnbuf[256];
602 >        int             a, i;
603  
604          progname = argv[0];
605 +                                        /* get options */
606 +        for (a = 1; a < argc && argv[a][0] == '-'; a++)
607 +                switch (argv[a][1]) {
608 +                case 'n':
609 +                        nsteps = atoi(argv[++a]);
610 +                        if (nsteps <= 0)
611 +                                goto userr;
612 +                        break;
613 +                case 'o':
614 +                        ofspec = argv[++a];
615 +                        break;
616 +                case 'i':
617 +                        switch (argv[a][2]) {
618 +                        case 'f':
619 +                                skyfmt = DTfloat;
620 +                                break;
621 +                        case 'd':
622 +                                skyfmt = DTdouble;
623 +                                break;
624 +                        case 'a':
625 +                                skyfmt = DTascii;
626 +                                break;
627 +                        default:
628 +                                goto userr;
629 +                        }
630 +                        break;
631 +                default:
632 +                        goto userr;
633 +                }
634 +        if ((argc-a < 1) | (argc-a > 4))
635 +                goto userr;
636  
637 <        if ((argc < 2) | (argc > 5)) {
638 <                fprintf(stderr, "Usage: %s DCspec [tregvec]\n", progname);
571 <                fprintf(stderr, "   or: %s Vspec Tbsdf.xml Dmat.dat [tregvec]\n",
572 <                                        progname);
573 <                return(1);
574 <        }
575 <
576 <        if (argc > 3) {                         /* VTDs expression */
577 <                CMATRIX *svec, *Dmat, *Tmat, *ivec;
637 >        if (argc-a > 2) {                       /* VTDs expression */
638 >                CMATRIX *smtx, *Dmat, *Tmat, *imtx;
639                  COLOR   tLamb;
640 <                                                /* get sky vector */
641 <                svec = cm_load(argv[4], 0, 1, DTascii);
640 >                                                /* get sky vector/matrix */
641 >                smtx = cm_load(argv[a+3], 0, nsteps, skyfmt);
642                                                  /* load BSDF */
643 <                Tmat = cm_loadBSDF(argv[2], tLamb);
643 >                Tmat = cm_loadBSDF(argv[a+1], tLamb);
644                                                  /* load Daylight matrix */
645 <                Dmat = cm_load(argv[3], Tmat==NULL ? 0 : Tmat->ncols,
646 <                                        svec->nrows, DTfromHeader);
645 >                Dmat = cm_load(argv[a+2], Tmat==NULL ? 0 : Tmat->ncols,
646 >                                        smtx->nrows, DTfromHeader);
647                                                  /* multiply vector through */
648 <                ivec = cm_multiply(Dmat, svec);
649 <                cm_free(Dmat); cm_free(svec);
648 >                imtx = cm_multiply(Dmat, smtx);
649 >                cm_free(Dmat); cm_free(smtx);
650                  if (Tmat == NULL) {             /* diffuse only */
651 <                        cvec = cm_scale(ivec, tLamb);
651 >                        cmtx = cm_scale(imtx, tLamb);
652                  } else {                        /* else apply BTDF matrix */
653 <                        cvec = cm_multiply(Tmat, ivec);
653 >                        cmtx = cm_multiply(Tmat, imtx);
654                          cm_free(Tmat);
655                  }
656 <                cm_free(ivec);
657 <        } else {                                /* else just use sky vector */
658 <                cvec = cm_load(argv[2], 0, 1, DTascii);
656 >                cm_free(imtx);
657 >        } else {                                /* sky vector/matrix only */
658 >                cmtx = cm_load(argv[a+1], 0, nsteps, skyfmt);
659          }
660 <
661 <        if (hasNumberFormat(argv[1])) {         /* generating image */
662 <                SET_FILE_BINARY(stdout);
663 <                newheader("RADIANCE", stdout);
664 <                printargs(argc, argv, stdout);
665 <                fputnow(stdout);
666 <                if (!sum_images(argv[1], cvec, stdout))
660 >                                                /* prepare output stream */
661 >        if ((ofspec != NULL) & (nsteps == 1) && hasNumberFormat(ofspec)) {
662 >                sprintf(fnbuf, ofspec, 1);
663 >                ofspec = fnbuf;
664 >        }
665 >        if (ofspec != NULL && !hasNumberFormat(ofspec)) {
666 >                if ((ofp = fopen(ofspec, "w")) == NULL) {
667 >                        fprintf(stderr, "%s: cannot open '%s' for output\n",
668 >                                        progname, ofspec);
669                          return(1);
670 <        } else {                                /* generating vector */
671 <                CMATRIX *Vmat = cm_load(argv[1], 0, cvec->nrows, DTfromHeader);
672 <                CMATRIX *rvec = cm_multiply(Vmat, cvec);
670 >                }
671 >                ofspec = NULL;                  /* only need to open once */
672 >        }
673 >        if (hasNumberFormat(argv[a])) {         /* generating image(s) */
674 >                if (ofspec == NULL) {
675 >                        SET_FILE_BINARY(ofp);
676 >                        newheader("RADIANCE", ofp);
677 >                        printargs(argc, argv, ofp);
678 >                        fputnow(ofp);
679 >                }
680 >                if (nsteps > 1)                 /* multiple output frames? */
681 >                        for (i = 0; i < nsteps; i++) {
682 >                                CMATRIX *cvec = cm_column(cmtx, i);
683 >                                if (ofspec != NULL) {
684 >                                        sprintf(fnbuf, ofspec, i+1);
685 >                                        if ((ofp = fopen(fnbuf, "wb")) == NULL) {
686 >                                                fprintf(stderr,
687 >                                        "%s: cannot open '%s' for output\n",
688 >                                                        progname, fnbuf);
689 >                                                return(1);
690 >                                        }
691 >                                        newheader("RADIANCE", ofp);
692 >                                        printargs(argc, argv, ofp);
693 >                                        fputnow(ofp);
694 >                                }
695 >                                fprintf(ofp, "FRAME=%d\n", i+1);
696 >                                if (!sum_images(argv[a], cvec, ofp))
697 >                                        return(1);
698 >                                if (ofspec != NULL) {
699 >                                        if (fclose(ofp) == EOF) {
700 >                                                fprintf(stderr,
701 >                                                "%s: error writing to '%s'\n",
702 >                                                        progname, fnbuf);
703 >                                                return(1);
704 >                                        }
705 >                                        ofp = stdout;
706 >                                }
707 >                                cm_free(cvec);
708 >                        }
709 >                else if (!sum_images(argv[a], cmtx, ofp))
710 >                        return(1);
711 >        } else {                                /* generating vector/matrix */
712 >                CMATRIX *Vmat = cm_load(argv[a], 0, cmtx->nrows, DTfromHeader);
713 >                CMATRIX *rmtx = cm_multiply(Vmat, cmtx);
714                  cm_free(Vmat);
715 <                cm_print(rvec, stdout);
716 <                cm_free(rvec);
715 >                if (ofspec != NULL)             /* multiple vector files? */
716 >                        for (i = 0; i < nsteps; i++) {
717 >                                CMATRIX *rvec = cm_column(rmtx, i);
718 >                                sprintf(fnbuf, ofspec, i+1);
719 >                                if ((ofp = fopen(fnbuf, "w")) == NULL) {
720 >                                        fprintf(stderr,
721 >                                        "%s: cannot open '%s' for output\n",
722 >                                                        progname, fnbuf);
723 >                                        return(1);
724 >                                }
725 >                                cm_print(rvec, ofp);
726 >                                if (fclose(ofp) == EOF) {
727 >                                        fprintf(stderr,
728 >                                                "%s: error writing to '%s'\n",
729 >                                                        progname, fnbuf);
730 >                                        return(1);
731 >                                }
732 >                                ofp = stdout;
733 >                                cm_free(rvec);
734 >                        }
735 >                else
736 >                        cm_print(rmtx, ofp);
737 >                cm_free(rmtx);
738          }
739 <        cm_free(cvec);                          /* final clean-up */
739 >        if (fflush(ofp) == EOF) {               /* final clean-up */
740 >                fprintf(stderr, "%s: write error on output\n", progname);
741 >                return(1);
742 >        }
743 >        cm_free(cmtx);
744          return(0);
745 + userr:
746 +        fprintf(stderr, "Usage: %s [-n nsteps][-o ospec][-i{f|d}] DCspec [skyf]\n",
747 +                                progname);
748 +        fprintf(stderr, "   or: %s [-n nsteps][-o ospec][-i{f|d}] Vspec Tbsdf.xml Dmat.dat [skyf]\n",
749 +                                progname);
750 +        return(1);
751   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines