--- ray/src/gen/gendaymtx.c	2013/06/06 20:21:10	2.12
+++ ray/src/gen/gendaymtx.c	2019/06/25 00:09:45	2.27
@@ -1,5 +1,5 @@
 #ifndef lint
-static const char RCSid[] = "$Id: gendaymtx.c,v 2.12 2013/06/06 20:21:10 greg Exp $";
+static const char RCSid[] = "$Id: gendaymtx.c,v 2.27 2019/06/25 00:09:45 greg Exp $";
 #endif
 /*
  *  gendaymtx.c
@@ -86,7 +86,11 @@ static const char RCSid[] = "$Id: gendaymtx.c,v 2.12 2
 #include <string.h>
 #include <ctype.h>
 #include "rtmath.h"
+#include "rtio.h"
+#include "resolu.h"
+#include "platform.h"
 #include "color.h"
+#include "resolu.h"
 
 char *progname;								/* Program name */
 char errmsg[128];							/* Error message buffer */
@@ -292,16 +296,37 @@ extern int	rh_init(void);
 extern float *	resize_dmatrix(float *mtx_data, int nsteps, int npatch);
 extern void	AddDirect(float *parr);
 
+
+static const char *
+getfmtname(int fmt)
+{
+	switch (fmt) {
+	case 'a':
+		return("ascii");
+	case 'f':
+		return("float");
+	case 'd':
+		return("double");
+	}
+	return("unknown");
+}
+
+
 int
 main(int argc, char *argv[])
 {
 	char	buf[256];
+	int	doheader = 1;		/* output header? */
 	double	rotation = 0;		/* site rotation (degrees) */
 	double	elevation;		/* site elevation (meters) */
 	int	dir_is_horiz;		/* direct is meas. on horizontal? */
 	float	*mtx_data = NULL;	/* our matrix data */
-	int	ntsteps = 0;		/* number of rows in matrix */
+	int	ntsteps = 0;		/* number of time steps */
+	int	avgSky = 0;		/* compute average sky r.t. matrix? */
+	int	nrows = 0;		/* number of rows in matrix */
+	int	rows_alloc = 0;		/* number of allocated rows */
 	int	last_monthly = 0;	/* month of last report */
+	int	inconsistent = 0;	/* inconsistent options set? */
 	int	mo, da;			/* month (1-12) and day (1-31) */
 	double	hr;			/* hour (local standard time) */
 	double	dir, dif;		/* direct and diffuse values */
@@ -320,6 +345,9 @@ main(int argc, char *argv[])
 		case 'v':			/* verbose progress reports */
 			verbose++;
 			break;
+		case 'h':			/* turn off header */
+			doheader = 0;
+			break;
 		case 'o':			/* output format */
 			switch (argv[i][2]) {
 			case 'f':
@@ -349,19 +377,24 @@ main(int argc, char *argv[])
 			rhsubdiv = atoi(argv[++i]);
 			break;
 		case 'c':			/* sky color */
+			inconsistent |= (skycolor[1] <= 1e-4);
 			skycolor[0] = atof(argv[++i]);
 			skycolor[1] = atof(argv[++i]);
 			skycolor[2] = atof(argv[++i]);
 			break;
 		case 'd':			/* solar (direct) only */
 			skycolor[0] = skycolor[1] = skycolor[2] = 0;
-			if (suncolor[1] <= 1e-4)
+			if (suncolor[1] <= 1e-4) {
+				inconsistent = 1;
 				suncolor[0] = suncolor[1] = suncolor[2] = 1;
+			}
 			break;
 		case 's':			/* sky only (no direct) */
 			suncolor[0] = suncolor[1] = suncolor[2] = 0;
-			if (skycolor[1] <= 1e-4)
+			if (skycolor[1] <= 1e-4) {
+				inconsistent = 1;
 				skycolor[0] = skycolor[1] = skycolor[2] = 1;
+			}
 			break;
 		case 'r':			/* rotate distribution */
 			if (argv[i][2] && argv[i][2] != 'z')
@@ -370,13 +403,25 @@ main(int argc, char *argv[])
 			break;
 		case '5':			/* 5-phase calculation */
 			nsuns = 1;
-			fixed_sun_sa = 6.797e-05;
+			fixed_sun_sa = PI/360.*atof(argv[++i]);
+			if (fixed_sun_sa <= 0) {
+				fprintf(stderr, "%s: missing solar disk size argument for '-5' option\n",
+						argv[0]);
+				exit(1);
+			}
+			fixed_sun_sa *= fixed_sun_sa*PI;
 			break;
+		case 'A':			/* compute average sky */
+			avgSky = 1;
+			break;
 		default:
 			goto userr;
 		}
 	if (i < argc-1)
 		goto userr;
+	if (inconsistent)
+		fprintf(stderr, "%s: WARNING: inconsistent -s, -d, -c options!\n",
+				progname);
 	if (i == argc-1 && freopen(argv[i], "r", stdin) == NULL) {
 		fprintf(stderr, "%s: cannot open '%s' for input\n",
 				progname, argv[i]);
@@ -434,14 +479,23 @@ main(int argc, char *argv[])
 	s_latitude = DegToRad(s_latitude);
 	s_longitude = DegToRad(s_longitude);
 	s_meridian = DegToRad(s_meridian);
+					/* initial allocation */
+	mtx_data = resize_dmatrix(mtx_data, rows_alloc=2, nskypatch);
 					/* process each time step in tape */
 	while (scanf("%d %d %lf %lf %lf\n", &mo, &da, &hr, &dir, &dif) == 5) {
 		double		sda, sta;
-					/* make space for next time step */
-		mtx_offset = 3*nskypatch*ntsteps++;
-		mtx_data = resize_dmatrix(mtx_data, ntsteps, nskypatch);
+
+		mtx_offset = 3*nskypatch*nrows;
+		nrows += !avgSky | !nrows;
+					/* make space for next row */
+		if (nrows > rows_alloc) {
+			rows_alloc += (rows_alloc>>1) + nrows + 7;
+			mtx_data = resize_dmatrix(mtx_data, rows_alloc, nskypatch);
+		}
+		ntsteps++;		/* keep count of time steps */
 		if (dif <= 1e-4) {
-			memset(mtx_data+mtx_offset, 0, sizeof(float)*3*nskypatch);
+			if (!avgSky | !nrows)
+				memset(mtx_data+mtx_offset, 0, sizeof(float)*3*nskypatch);
 			continue;
 		}
 		if (verbose && mo != last_monthly)
@@ -466,6 +520,9 @@ main(int argc, char *argv[])
 					/* compute sky patch values */
 		ComputeSky(mtx_data+mtx_offset);
 		AddDirect(mtx_data+mtx_offset);
+					/* update cumulative sky? */
+		for (i = 3*nskypatch*(avgSky&(ntsteps>1)); i--; )
+			mtx_data[i] += mtx_data[mtx_offset+i];
 	}
 					/* check for junk at end */
 	while ((i = fgetc(stdin)) != EOF)
@@ -477,41 +534,61 @@ main(int argc, char *argv[])
 			fputs(buf, stderr); fputc('\n', stderr);
 			break;
 		}
+	if (!ntsteps) {
+		fprintf(stderr, "%s: no valid time steps on input\n", progname);
+		exit(1);
+	}
+	dif = 1./(double)ntsteps;	/* average sky? */
+	for (i = 3*nskypatch*(avgSky&(ntsteps>1)); i--; )
+		mtx_data[i] *= dif;
 					/* write out matrix */
+	if (outfmt != 'a')
+		SET_FILE_BINARY(stdout);
 #ifdef getc_unlocked
 	flockfile(stdout);
 #endif
 	if (verbose)
 		fprintf(stderr, "%s: writing %smatrix with %d time steps...\n",
-				progname, outfmt=='a' ? "" : "binary ", ntsteps);
+				progname, outfmt=='a' ? "" : "binary ", nrows);
+	if (doheader) {
+		newheader("RADIANCE", stdout);
+		printargs(argc, argv, stdout);
+		printf("LATLONG= %.8f %.8f\n", RadToDeg(s_latitude),
+					-RadToDeg(s_longitude));
+		printf("NROWS=%d\n", nskypatch);
+		printf("NCOLS=%d\n", nrows);
+		printf("NCOMP=3\n");
+		fputformat((char *)getfmtname(outfmt), stdout);
+		putchar('\n');
+	}
 					/* patches are rows (outer sort) */
 	for (i = 0; i < nskypatch; i++) {
 		mtx_offset = 3*i;
 		switch (outfmt) {
 		case 'a':
-			for (j = 0; j < ntsteps; j++) {
+			for (j = 0; j < nrows; j++) {
 				printf("%.3g %.3g %.3g\n", mtx_data[mtx_offset],
 						mtx_data[mtx_offset+1],
 						mtx_data[mtx_offset+2]);
 				mtx_offset += 3*nskypatch;
 			}
-			if (ntsteps > 1)
+			if (nrows > 1)
 				fputc('\n', stdout);
 			break;
 		case 'f':
-			for (j = 0; j < ntsteps; j++) {
-				fwrite(mtx_data+mtx_offset, sizeof(float), 3,
+			for (j = 0; j < nrows; j++) {
+				putbinary(mtx_data+mtx_offset, sizeof(float), 3,
 						stdout);
 				mtx_offset += 3*nskypatch;
 			}
 			break;
 		case 'd':
-			for (j = 0; j < ntsteps; j++) {
+			for (j = 0; j < nrows; j++) {
 				double	ment[3];
 				ment[0] = mtx_data[mtx_offset];
 				ment[1] = mtx_data[mtx_offset+1];
 				ment[2] = mtx_data[mtx_offset+2];
-				fwrite(ment, sizeof(double), 3, stdout);
+				putbinary(ment, sizeof(double), 3, stdout);
 				mtx_offset += 3*nskypatch;
 			}
 			break;
@@ -525,7 +602,7 @@ main(int argc, char *argv[])
 		fprintf(stderr, "%s: done.\n", progname);
 	exit(0);
 userr:
-	fprintf(stderr, "Usage: %s [-v][-d|-s][-r deg][-m N][-g r g b][-c r g b][-o{f|d}][-O{0|1}] [tape.wea]\n",
+	fprintf(stderr, "Usage: %s [-v][-h][-A][-d|-s][-r deg][-m N][-g r g b][-c r g b][-o{f|d}][-O{0|1}] [tape.wea]\n",
 			progname);
 	exit(1);
 fmterr:
@@ -622,6 +699,12 @@ ComputeSky(float *parr)
 	/* Calculate relative horizontal illuminance */
 	norm_diff_illum = CalcRelHorzIllum(parr);
 
+	/* Check for zero sky -- make uniform in that case */
+	if (norm_diff_illum <= FTINY) {
+		for (i = 1; i < nskypatch; i++)
+			setcolor(parr+3*i, 1., 1., 1.);
+		norm_diff_illum = PI;
+	}
 	/* Normalization coefficient */
 	norm_diff_illum = diff_illum / norm_diff_illum;
 
@@ -891,7 +974,9 @@ int CalcSkyParamFromIllum()
 		/* Convert illuminance to irradiance */
 		index = GetCategoryIndex();
 		diff_irrad = diff_illum / CalcDiffuseIllumRatio(index);
-		dir_irrad = dir_illum / CalcDirectIllumRatio(index);
+		dir_irrad = CalcDirectIllumRatio(index);
+		if (dir_irrad > 0.1)
+			dir_irrad = dir_illum / dir_irrad;
 	
 		/* Calculate sky brightness and clearness */
 		sky_brightness = CalcSkyBrightness();