--- ray/src/util/dctimestep.c	2009/06/19 06:49:42	2.2
+++ ray/src/util/dctimestep.c	2010/07/01 21:54:56	2.12
@@ -1,5 +1,5 @@
 #ifndef lint
-static const char RCSid[] = "$Id: dctimestep.c,v 2.2 2009/06/19 06:49:42 greg Exp $";
+static const char RCSid[] = "$Id: dctimestep.c,v 2.12 2010/07/01 21:54:56 greg Exp $";
 #endif
 /*
  * Compute time-step result using Daylight Coefficient method.
@@ -38,7 +38,7 @@ cm_alloc(int nrows, int ncols)
 	CMATRIX	*cm;
 
 	if ((nrows <= 0) | (ncols <= 0))
-		return(NULL);
+		error(USER, "attempt to create empty matrix");
 	cm = (CMATRIX *)malloc(sizeof(CMATRIX) +
 				3*sizeof(COLORV)*(nrows*ncols - 1));
 	if (cm == NULL)
@@ -108,6 +108,8 @@ cm_load(const char *fname, int nrows, int ncols, int d
 	CMATRIX	*cm;
 	FILE	*fp = stdin;
 
+	if (ncols <= 0)
+		error(USER, "Non-positive number of columns");
 	if (fname == NULL)
 		fname = "<stdin>";
 	else if ((fp = fopen(fname, "r")) == NULL) {
@@ -160,12 +162,12 @@ cm_load(const char *fname, int nrows, int ncols, int d
 		int	maxrow = (nrows > 0 ? nrows : 32000);
 		int	r, c;
 		for (r = 0; r < maxrow; r++) {
-		    if (r >= cm->nrows)		/* need more space? */
+		    if (r >= cm->nrows)			/* need more space? */
 			cm = cm_resize(cm, 2*cm->nrows);
 		    for (c = 0; c < ncols; c++) {
 		        COLORV	*cv = cm_lval(cm,r,c);
 			if (fscanf(fp, COLSPEC, cv, cv+1, cv+2) != 3)
-				if ((nrows <= 0) & (r > 0) & (c == 0)) {
+				if ((nrows <= 0) & (r > 0) & !c) {
 					cm = cm_resize(cm, maxrow=r);
 					break;
 				} else
@@ -193,7 +195,7 @@ cm_load(const char *fname, int nrows, int ncols, int d
 					if (nread == cm->nrows*cm->ncols)
 							/* need more space? */
 						cm = cm_resize(cm, 2*cm->nrows);
-					else if (nread % cm->ncols == 0)
+					else if (nread && !(nread % cm->ncols))
 							/* seem to be  done */
 						cm = cm_resize(cm, nread/cm->ncols);
 					else		/* ended mid-row */
@@ -294,18 +296,45 @@ static CMATRIX *
 cm_bsdf(const struct BSDF_data *bsdf)
 {
 	CMATRIX	*cm = cm_alloc(bsdf->nout, bsdf->ninc);
-	COLORV	*mp = cm->cmem;
+	int	nbadohm = 0;
+	int	nneg = 0;
 	int	r, c;
 	
-	for (r = 0; r < cm->nrows; r++)
-		for (c = 0; c < cm->ncols; c++, mp += 3)
-			mp[0] = mp[1] = mp[2] = BSDF_value(bsdf,c,r);
+	for (c = 0; c < cm->ncols; c++) {
+		float	dom = getBSDF_incohm(bsdf,c);
+		FVECT	v;
+		
+		if (dom <= .0) {
+			nbadohm++;
+			continue;
+		}
+		if (!getBSDF_incvec(v,bsdf,c) || v[2] > FTINY)
+			error(USER, "illegal incoming BTDF direction");
+		dom *= -v[2];
+
+		for (r = 0; r < cm->nrows; r++) {
+			float	f = BSDF_value(bsdf,c,r);
+			COLORV	*mp = cm_lval(cm,r,c);
+
+			if (f <= .0) {
+				nneg += (f < -FTINY);
+				f = .0f;
+			}
+			mp[0] = mp[1] = mp[2] = f * dom;
+		}
+	}
+	if (nneg || nbadohm) {
+		sprintf(errmsg,
+		    "BTDF has %d negatives and %d bad incoming solid angles",
+				nneg, nbadohm);
+		error(WARNING, errmsg);
+	}
 	return(cm);
 }
 
 /* Sum together a set of images and write result to stdout */
 static int
-sum_images(const char *fspec, const CMATRIX *cv)
+sum_images(const char *fspec, const CMATRIX *cv, FILE *fp)
 {
 	int	myDT = DTfromHeader;
 	CMATRIX	*pmat;
@@ -343,10 +372,10 @@ sum_images(const char *fspec, const CMATRIX *cv)
 			pmat = cm_alloc(myYR, myXR);
 			memset(pmat->cmem, 0, sizeof(COLOR)*myXR*myYR);
 							/* finish header */
-			fputformat(myDT==DTrgbe ? COLRFMT : CIEFMT, stdout);
-			fputc('\n', stdout);
-			fprtresolu(myXR, myYR, stdout);
-			fflush(stdout);
+			fputformat(myDT==DTrgbe ? COLRFMT : CIEFMT, fp);
+			fputc('\n', fp);
+			fprtresolu(myXR, myYR, fp);
+			fflush(fp);
 		} else if ((dt != myDT) | (xr != myXR) | (yr != myYR)) {
 			sprintf(errmsg, "picture '%s' format/size mismatch",
 					fname);
@@ -371,15 +400,15 @@ sum_images(const char *fspec, const CMATRIX *cv)
 	free(scanline);
 							/* write scanlines */
 	for (y = 0; y < myYR; y++)
-		if (fwritescan((COLOR *)cm_lval(pmat, y, 0), myXR, stdout) < 0)
+		if (fwritescan((COLOR *)cm_lval(pmat, y, 0), myXR, fp) < 0)
 			return(0);
 	cm_free(pmat);					/* all done */
-	return(fflush(stdout) == 0);
+	return(fflush(fp) == 0);
 }
 
-/* check to see if a string contains a %d specification */
-int
-hasDecimalSpec(const char *s)
+/* check to see if a string contains a %d or %o specification */
+static int
+hasNumberFormat(const char *s)
 {
 	while (*s && *s != '%')
 		s++;
@@ -389,45 +418,51 @@ hasDecimalSpec(const char *s)
 		++s;
 	while (isdigit(*s));
 
-	return(*s == 'd');
+	return(*s == 'd' | *s == 'i' | *s == 'o' | *s == 'x' | *s == 'X');
 }
 
 int
 main(int argc, char *argv[])
 {
-	CMATRIX			*tvec, *Dmat, *Tmat, *ivec, *cvec;
-	struct BSDF_data	*btdf;
+	CMATRIX			*cvec;
 
 	progname = argv[0];
 
-	if ((argc < 4) | (argc > 5)) {
-		fprintf(stderr, "Usage: %s Vspec Tbsdf.xml Dmat.dat [tregvec]\n",
-				progname);
+	if ((argc < 2) | (argc > 5)) {
+		fprintf(stderr, "Usage: %s DCspec [tregvec]\n", progname);
+		fprintf(stderr, "   or: %s Vspec Tbsdf.xml Dmat.dat [tregvec]\n",
+					progname);
 		return(1);
 	}
-	tvec = cm_load(argv[4], 0, 1, DTascii);	/* argv[4]==NULL iff argc==4 */
-	Dmat = cm_load(argv[3], 0, tvec->nrows, DTfromHeader);
-	btdf = load_BSDF(argv[2]);
-	if (btdf == NULL)
-		return(1);
-	if (btdf->ninc != Dmat->nrows) {
-		sprintf(errmsg, "Incoming BTDF dir (%d) mismatch to D (%d)",
-				btdf->ninc, Dmat->nrows);
-		error(USER, errmsg);
-	}
+
+	if (argc > 3) {				/* need to compute DC matrix? */
+		CMATRIX	*svec, *Dmat, *Tmat, *ivec;
+		struct BSDF_data	*btdf;
+						/* get sky vector */
+		svec = cm_load(argv[4], 0, 1, DTascii);
+						/* load BSDF */
+		btdf = load_BSDF(argv[2]);
+		if (btdf == NULL)
+			return(1);
+						/* load Daylight matrix */
+		Dmat = cm_load(argv[3], btdf->ninc, svec->nrows, DTfromHeader);
 						/* multiply vector through */
-	ivec = cm_multiply(Dmat, tvec);
-	cm_free(Dmat); cm_free(tvec);
-	Tmat = cm_bsdf(btdf);			/* convert BTDF to matrix */
-	free_BSDF(btdf);
-	cvec = cm_multiply(Tmat, ivec);		/* cvec = component vector */
-	cm_free(Tmat); cm_free(ivec);
-	if (hasDecimalSpec(argv[1])) {		/* generating image */
+		ivec = cm_multiply(Dmat, svec);
+		cm_free(Dmat); cm_free(svec);
+		Tmat = cm_bsdf(btdf);		/* convert BTDF to matrix */
+		free_BSDF(btdf);
+		cvec = cm_multiply(Tmat, ivec);	/* cvec = component vector */
+		cm_free(Tmat); cm_free(ivec);
+	} else {				/* else just use sky vector */
+		cvec = cm_load(argv[2], 0, 1, DTascii);
+	}
+
+	if (hasNumberFormat(argv[1])) {		/* generating image */
 		SET_FILE_BINARY(stdout);
 		newheader("RADIANCE", stdout);
 		printargs(argc, argv, stdout);
 		fputnow(stdout);
-		if (!sum_images(argv[1], cvec))
+		if (!sum_images(argv[1], cvec, stdout))
 			return(1);
 	} else {				/* generating vector */
 		CMATRIX	*Vmat = cm_load(argv[1], 0, cvec->nrows, DTfromHeader);