--- ray/src/rt/rtmain.c	2005/06/02 04:47:27	2.11
+++ ray/src/rt/rtmain.c	2019/08/14 20:07:20	2.30
@@ -1,5 +1,5 @@
 #ifndef lint
-static const char	RCSid[] = "$Id: rtmain.c,v 2.11 2005/06/02 04:47:27 greg Exp $";
+static const char	RCSid[] = "$Id: rtmain.c,v 2.30 2019/08/14 20:07:20 greg Exp $";
 #endif
 /*
  *  rtmain.c - main for rtrace per-ray calculation program
@@ -7,7 +7,6 @@ static const char	RCSid[] = "$Id: rtmain.c,v 2.11 2005
 
 #include "copyright.h"
 
-#include  <sys/types.h>
 #include  <signal.h>
 
 #include  "platform.h"
@@ -18,7 +17,12 @@ static const char	RCSid[] = "$Id: rtmain.c,v 2.11 2005
 #include  "ambient.h"
 #include  "random.h"
 #include  "paths.h"
+#include  "pmapray.h"
 
+extern char	*progname;		/* global argv[0] */
+
+extern char	*shm_boundary;		/* boundary of shared memory */
+
 					/* persistent processes define */
 #ifdef  F_SETLKW
 #define  PERSIST	1		/* normal persist */
@@ -26,26 +30,32 @@ static const char	RCSid[] = "$Id: rtmain.c,v 2.11 2005
 #define  PCHILD		3		/* child of normal persist */
 #endif
 
-char  *progname;			/* argv[0] */
-char  *octname;				/* octree name */
 char  *sigerr[NSIG];			/* signal error messages */
-char  *shm_boundary = NULL;		/* boundary of shared memory */
 char  *errfile = NULL;			/* error output file */
 
+int  nproc = 1;				/* number of processes */
+
 extern char  *formstr();		/* string from format */
-extern int  inform;			/* input format */
-extern int  outform;			/* output format */
-extern char  *outvals;			/* output values */
+int  inform = 'a';			/* input format */
+int  outform = 'a';			/* output format */
+char  *outvals = "v";			/* output specification */
 
-extern int  hresolu;			/* horizontal resolution */
-extern int  vresolu;			/* vertical resolution */
+int  hresolu = 0;			/* horizontal (scan) size */
+int  vresolu = 0;			/* vertical resolution */
 
-extern int  imm_irrad;			/* compute immediate irradiance? */
-extern int  lim_dist;			/* limit distance? */
+int  imm_irrad = 0;			/* compute immediate irradiance? */
+int  lim_dist = 0;			/* limit distance? */
 
-extern char  *tralist[];		/* list of modifers to trace (or no) */
-extern int  traincl;			/* include == 1, exclude == 0 */
+#ifndef	MAXMODLIST
+#define	MAXMODLIST	1024		/* maximum modifiers we'll track */
+#endif
 
+extern void  (*addobjnotify[])();	/* object notification calls */
+extern void  tranotify(OBJECT obj);
+
+char  *tralist[MAXMODLIST];		/* list of modifers to trace (or no) */
+int  traincl = -1;			/* include == 1, exclude == 0 */
+
 static int  loadflags = ~IO_FILES;	/* what to load from octree */
 
 static void onsig(int  signo);
@@ -59,20 +69,26 @@ main(int  argc, char  *argv[])
 #define	 check(ol,al)		if (argv[i][ol] || \
 				badarg(argc-i-1,argv+i+1,al)) \
 				goto badopt
-#define	 bool(olen,var)		switch (argv[i][olen]) { \
+#define	 check_bool(olen,var)		switch (argv[i][olen]) { \
 				case '\0': var = !var; break; \
 				case 'y': case 'Y': case 't': case 'T': \
 				case '+': case '1': var = 1; break; \
 				case 'n': case 'N': case 'f': case 'F': \
 				case '-': case '0': var = 0; break; \
 				default: goto badopt; }
+	extern char  *octname;
 	int  persist = 0;
-	char  **tralp;
-	int  duped1;
+	char  *octnm = NULL;
+	char  **tralp = NULL;
+	int  duped1 = -1;
 	int  rval;
 	int  i;
 					/* global program name */
 	progname = argv[0] = fixargv0(argv[0]);
+					/* add trace notify function */
+	for (i = 0; addobjnotify[i] != NULL; i++)
+		;
+	addobjnotify[i] = tranotify;
 					/* option city */
 	for (i = 1; i < argc; i++) {
 						/* expand arguments */
@@ -99,6 +115,12 @@ main(int  argc, char  *argv[])
 			continue;
 		}
 		switch (argv[i][1]) {
+		case 'n':				/* number of cores */
+			check(2,"i");
+			nproc = atoi(argv[++i]);
+			if (nproc <= 0)
+				error(USER, "bad number of processes");
+			break;
 		case 'x':				/* x resolution */
 			check(2,"i");
 			hresolu = atoi(argv[++i]);
@@ -109,7 +131,7 @@ main(int  argc, char  *argv[])
 			break;
 		case 'w':				/* warnings */
 			rval = erract[WARNING].pf != NULL;
-			bool(2,rval);
+			check_bool(2,rval);
 			if (rval) erract[WARNING].pf = wputs;
 			else erract[WARNING].pf = NULL;
 			break;
@@ -120,10 +142,10 @@ main(int  argc, char  *argv[])
 		case 'l':				/* limit distance */
 			if (argv[i][2] != 'd')
 				goto badopt;
-			bool(3,lim_dist);
+			check_bool(3,lim_dist);
 			break;
 		case 'I':				/* immed. irradiance */
-			bool(2,imm_irrad);
+			check_bool(2,imm_irrad);
 			break;
 		case 'f':				/* format i/o */
 			switch (argv[i][2]) {
@@ -155,7 +177,7 @@ main(int  argc, char  *argv[])
 			break;
 		case 'h':				/* header output */
 			rval = loadflags & IO_INFO;
-			bool(2,rval);
+			check_bool(2,rval);
 			loadflags = rval ? loadflags | IO_INFO :
 					loadflags & ~IO_INFO;
 			break;
@@ -169,7 +191,7 @@ main(int  argc, char  *argv[])
 					tralp = tralist;
 				}
 				if (argv[i][2] == 'I') {	/* file */
-					rval = wordfile(tralp,
+					rval = wordfile(tralp, MAXMODLIST-(tralp-tralist),
 					getpath(argv[++i],getrlibpath(),R_OK));
 					if (rval < 0) {
 						sprintf(errmsg,
@@ -191,7 +213,7 @@ main(int  argc, char  *argv[])
 					tralp = tralist;
 				}
 				if (argv[i][2] == 'E') {	/* file */
-					rval = wordfile(tralp,
+					rval = wordfile(tralp, MAXMODLIST-(tralp-tralist),
 					getpath(argv[++i],getrlibpath(),R_OK));
 					if (rval < 0) {
 						sprintf(errmsg,
@@ -225,10 +247,24 @@ main(int  argc, char  *argv[])
 			goto badopt;
 		}
 	}
+	if (nproc > 1) {
+		if (persist)
+			error(USER, "multiprocessing incompatible with persist file");
+		if (!vresolu && hresolu > 0 && hresolu < nproc)
+			error(WARNING, "number of cores should not exceed horizontal resolution");
+		if (trace != NULL)
+			error(WARNING, "multiprocessing does not work properly with trace mode");
+	}
 					/* initialize object types */
 	initotypes();
 					/* initialize urand */
-	initurand(2048);
+	if (rand_samp) {
+		srandom((long)time(0));
+		initurand(0);
+	} else {
+		srandom(0L);
+		initurand(2048);
+	}
 					/* set up signal handling */
 	sigdie(SIGINT, "Interrupt");
 #ifdef SIGHUP
@@ -260,12 +296,12 @@ main(int  argc, char  *argv[])
 #endif
 					/* get octree */
 	if (i == argc)
-		octname = NULL;
+		octnm = NULL;
 	else if (i == argc-1)
-		octname = argv[i];
+		octnm = argv[i];
 	else
 		goto badopt;
-	if (octname == NULL)
+	if (octnm == NULL)
 		error(USER, "missing octree argument");
 					/* set up output */
 #ifdef  PERSIST
@@ -276,21 +312,25 @@ main(int  argc, char  *argv[])
 #endif
 	if (outform != 'a')
 		SET_FILE_BINARY(stdout);
-	readoct(octname, loadflags, &thescene, NULL);
+	readoct(octname = octnm, loadflags, &thescene, NULL);
 	nsceneobjs = nobjects;
 
 	if (loadflags & IO_INFO) {	/* print header */
 		printargs(i, argv, stdout);
 		printf("SOFTWARE= %s\n", VersionID);
 		fputnow(stdout);
+		if ((outform == 'f') | (outform == 'd'))
+			fputendian(stdout);
 		fputformat(formstr(outform), stdout);
 		putchar('\n');
 	}
-
+	
+	ray_init_pmap();     /* PMAP: set up & load photon maps */
+	
 	marksources();			/* find and mark sources */
 
 	setambient();			/* initialize ambient calculation */
-
+	
 #ifdef  PERSIST
 	if (persist) {
 		fflush(stdout);
@@ -304,10 +344,11 @@ main(int  argc, char  *argv[])
 			while ((rval=fork()) == 0) {	/* keep on forkin' */
 				pflock(1);
 				pfhold();
+				ambsync();		/* load new values */
 			}
 			if (rval < 0)
 				error(SYSTEM, "cannot fork child for persist function");
-			pfdetach();		/* parent exits */
+			pfdetach();		/* parent will run then exit */
 		}
 	}
 runagain:
@@ -315,7 +356,7 @@ runagain:
 		dupheader();			/* send header to stdout */
 #endif
 					/* trace rays */
-	rtrace(NULL);
+	rtrace(NULL, nproc);
 					/* flush ambient file */
 	ambsync();
 #ifdef  PERSIST
@@ -330,12 +371,14 @@ runagain:
 		}
 	}
 	if (persist == PCHILD) {	/* wait for a signal then go again */
-		close(duped1);			/* release output handle */
 		pfhold();
 		raynum = nrays = 0;		/* reinitialize */
 		goto runagain;
 	}
 #endif
+
+	ray_done_pmap();           /* PMAP: free photon maps */
+	
 	quit(0);
 
 badopt:
@@ -344,7 +387,7 @@ badopt:
 	return 1; /* pro forma return */
 
 #undef	check
-#undef	bool
+#undef	check_bool
 }
 
 
@@ -420,8 +463,10 @@ printdefaults(void)			/* print default values to stdou
 
 	if (imm_irrad)
 		printf("-I+\t\t\t\t# immediate irradiance on\n");
-	printf("-x  %-9d\t\t\t# x resolution (flush interval)\n", hresolu);
-	printf("-y  %-9d\t\t\t# y resolution\n", vresolu);
+	printf("-n %-2d\t\t\t\t# number of rendering processes\n", nproc);
+	printf("-x %-9d\t\t\t# %s\n", hresolu,
+			vresolu && hresolu ? "x resolution" : "flush interval");
+	printf("-y %-9d\t\t\t# y resolution\n", vresolu);
 	printf(lim_dist ? "-ld+\t\t\t\t# limit distance on\n" :
 			"-ld-\t\t\t\t# limit distance off\n");
 	printf("-h%c\t\t\t\t# %s header\n", loadflags & IO_INFO ? '+' : '-',
@@ -434,7 +479,12 @@ printdefaults(void)			/* print default values to stdou
 		case 't': case 'T': printf(" trace"); break;
 		case 'o': printf(" origin"); break;
 		case 'd': printf(" direction"); break;
+		case 'r': printf(" reflect_contrib"); break;
+		case 'R': printf(" reflect_length"); break;
+		case 'x': printf(" unreflect_contrib"); break;
+		case 'X': printf(" unreflect_length"); break;
 		case 'v': printf(" value"); break;
+		case 'V': printf(" contribution"); break;
 		case 'l': printf(" length"); break;
 		case 'L': printf(" first_length"); break;
 		case 'p': printf(" point"); break;
@@ -442,9 +492,9 @@ printdefaults(void)			/* print default values to stdou
 		case 'N': printf(" unperturbed_normal"); break;
 		case 's': printf(" surface"); break;
 		case 'w': printf(" weight"); break;
+		case 'W': printf(" coefficient"); break;
 		case 'm': printf(" modifier"); break;
 		case 'M': printf(" material"); break;
-		case 'W': printf(" contribution"); break;
 		case '-': printf(" stroke"); break;
 		}
 	putchar('\n');