--- ray/src/util/rttree_reduce.c	2011/06/01 16:51:03	2.3
+++ ray/src/util/rttree_reduce.c	2014/03/17 01:59:48	2.9
@@ -1,5 +1,5 @@
 #ifndef lint
-static const char RCSid[] = "$Id: rttree_reduce.c,v 2.3 2011/06/01 16:51:03 greg Exp $";
+static const char RCSid[] = "$Id: rttree_reduce.c,v 2.9 2014/03/17 01:59:48 greg Exp $";
 #endif
 /*
  *  A utility called by genBSDF.pl to reduce tensor tree samples and output
@@ -17,7 +17,7 @@ float	*datarr;		/* our loaded BSDF data array */
 int	ttrank = 4;		/* tensor tree rank */
 int	log2g = 4;		/* log2 of grid resolution */
 int	infmt = 'a';		/* input format ('a','f','d') */
-double	pctcull = 99.;		/* target culling percentile */
+double	pctcull = 95.;		/* target culling percentile */
 
 #define dval3(ix,ox,oy)		datarr[((((ix)<<log2g)+(ox))<<log2g)+(oy)]
 #define dval4(ix,iy,ox,oy)	datarr[((((((ix)<<log2g)+(iy))<<log2g)+(ox))<<log2g)+(oy)]
@@ -152,7 +152,7 @@ print_tree(const TNODE *tp, const int bmin[], int l2s)
 		for (i = 0; i < 1<<ttrank; i++) {
 			float	val;
 			for (j = ttrank; j--; )
-				bkmin[j] = bmin[j] + (i>>j & 1);
+				bkmin[j] = bmin[j] + (i>>(ttrank-1-j) & 1);
 			val = (ttrank == 3) ? dval3(bkmin[0],bkmin[1],bkmin[2])
 				: dval4(bkmin[0],bkmin[1],bkmin[2],bkmin[3]);
 			printf(" %.4e", val);
@@ -260,17 +260,17 @@ load_data()
 		error(SYSTEM, "out of memory in load_data");
 	if (ttrank == 3) {
 		int	ix, ox;
-		for (ix = 0; ix < 1<<log2g; ix++)
+		for (ix = 0; ix < 1<<(log2g-1); ix++)
 			for (ox = 0; ox < 1<<log2g; ox++)
-				(*readf)(datarr+((((ix)<<log2g)+(ox))<<log2g),
-						1<<(log2g-1));
+				(*readf)(datarr+(((ix<<log2g)+ox)<<log2g),
+						1<<log2g);
 	} else /* ttrank == 4 */ {
 		int	ix, iy, ox;
 		for (ix = 0; ix < 1<<log2g; ix++)
 		    for (iy = 0; iy < 1<<log2g; iy++)
 			for (ox = 0; ox < 1<<log2g; ox++)
 				(*readf)(datarr +
-				((((((ix)<<log2g)+(iy))<<log2g)+(ox))<<log2g),
+				(((((ix<<log2g)+iy)<<log2g)+ox)<<log2g),
 						1<<log2g);
 	}
 	(*readf)(NULL, 0);	/* releases any buffers */
@@ -291,17 +291,50 @@ load_data()
 		error(WARNING, "binary data past end of expected input");
 }
 
+/* Enforce reciprocity by averaging data values */
+static void
+do_reciprocity()
+{
+	const int	siz = 1<<log2g;
+	float		*v1p, *v2p;
+
+	if (ttrank == 3) {
+		int	ix, ox, oy;
+		for (ix = 0; ix < siz>>1; ix++)
+		    for (ox = 0; ox < siz; ox++)
+		        for (oy = 0; oy < siz>>1; oy++) {
+				v1p = &dval3(ix,ox,oy);
+				v2p = &dval3(ix,ox,siz-1-oy);
+				*v1p = *v2p = .5f*( *v1p + *v2p );
+			}
+	} else /* ttrank == 4 */ {
+		int	ix, iy, ox, oy;
+		for (ix = 1; ix < siz; ix++)
+		    for (iy = 1; iy < siz; iy++)
+			for (ox = 0; ox < ix; ox++)
+			    for (oy = 0; oy < iy; oy++) {
+				v1p = &dval4(siz-1-ix,siz-1-iy,ox,oy);
+				v2p = &dval4(siz-1-ox,siz-1-oy,ix,iy);
+				*v1p = *v2p = .5f*( *v1p + *v2p );
+			    }
+	}
+}
+
 /* Load BSDF array, coalesce uniform regions and format as tensor tree */
 int
 main(int argc, char *argv[])
 {
 	int	doheader = 1;
+	int	recipavg = 0;
 	int	bmin[4];
 	TNODE	gtree;
 	int	i;
 					/* get options and parameters */
 	for (i = 1; i < argc && argv[i][0] == '-'; i++)
 		switch (argv[i][1]) {
+		case 'a':
+			recipavg = !recipavg;
+			break;
 		case 'h':
 			doheader = !doheader;
 			break;
@@ -331,13 +364,18 @@ main(int argc, char *argv[])
 	if (i < argc-1)
 		goto userr;
 					/* load input data */
-	if (i == argc-1 && freopen(argv[i], "rb", stdin) == NULL) {
+	if (i == argc-1 && freopen(argv[i], "r", stdin) == NULL) {
 		sprintf(errmsg, "cannot open input file '%s'", argv[i]);
 		error(SYSTEM, errmsg);
 	}
 	if (infmt != 'a')
 		SET_FILE_BINARY(stdin);
+#ifdef getc_unlocked			/* avoid lock/unlock overhead */
+	flockfile(stdin);
+#endif
 	load_data();
+	if (recipavg)
+		do_reciprocity();
 	if (doheader) {
 		for (i = 0; i < argc; i++) {
 			fputs(argv[i], stdout);
@@ -359,7 +397,7 @@ main(int argc, char *argv[])
 	*/
 	return(0);
 userr:
-	fprintf(stderr, "Usage: %s [-h][-f{a|f|d}][-r {3|4}][-g log2grid][-t trim%%] [input]\n",
+	fprintf(stderr, "Usage: %s [-h][-a][-f{a|f|d}][-r {3|4}][-g log2grid][-t trim%%] [input]\n",
 			argv[0]);
 	return(1);
 }