ViewVC Help

View File | Revision Log | Show Annotations | Download File | Root Listing

root/Development/ray/src/rt/pmapdump.c

Comparing ray/src/rt/pmapdump.c (file contents):
Revision 2.7 by rschregle, Tue May 17 17:39:47 2016 UTC vs.
Revision 2.14 by rschregle, Tue Jan 22 18:28:23 2019 UTC

#	Line 4 | Line 4 | static const char RCSid[] = "$Id$";
4
5		/*
6		======================================================================
7	<	Dump photon maps as RADIANCE scene description to stdout
7	>	Dump photon maps as RADIANCE scene description or ASCII point list
8	>	to stdout
9
10		Roland Schregle (roland.schregle@{hslu.ch, gmail.com})
11		(c) Fraunhofer Institute for Solar Energy Systems,
#	Line 17 | Line 18 | static const char RCSid[] = "$Id$";
18
19
20
21	+	#include "pmap.h"
22		#include "pmapio.h"
21	–	#include "pmapparm.h"
22	–	#include "pmaptype.h"
23		#include "rtio.h"
24		#include "resolu.h"
25		#include "random.h"
26		#include "math.h"
27
28	–	#define PMAPDUMP_REC "$Revision$"
28
30	–
29		/* Defaults */
30		/*    Sphere radius as fraction of avg. intersphere dist */
31		/*    Relative scale for sphere radius (fudge factor) */
#	Line 36 | Line 34 | static const char RCSid[] = "$Id$";
34		#define RADSCALE 1.0
35		#define NSPHERES 10000
36
37	+	/* Format for optional ASCII output as XYZ RGB points */
38	+	#define POINTFMT "%g\t%g\t%g\t%g\t%g\t%g\n"
39
40		/* RADIANCE material and object defs for each photon type */
41		typedef struct {
42		char *mat, *obj;
43		} RadianceDef;
44
45	–
46	–
47	–	/* Colour code is as follows:    global         = blue
48	–	precomp global = cyan
49	–	caustic        = red
50	–	volume         = green
51	–	direct         = magenta
52	–	contrib        = yellow */
45		const RadianceDef radDefs [] = {
46	<	{  "void plastic mat.global\n0\n0\n5 0 0 1 0 0\n",
46	>	{  "void glow mat.global\n0\n0\n4 %g %g %g 0\n",
47		"mat.global sphere obj.global\n0\n0\n4 %g %g %g %g\n"
48		},
49	<	{  "void plastic mat.pglobal\n0\n0\n5 0 1 1 0 0\n",
50	<	"mat.pglobal sphere obj.global\n0\n0\n4 %g %g %g %g\n"
49	>	{  "void glow mat.pglobal\n0\n0\n4 %g %g %g 0\n",
50	>	"mat.pglobal sphere obj.pglobal\n0\n0\n4 %g %g %g %g\n"
51		},
52	<	{  "void plastic mat.caustic\n0\n0\n5 1 0 0 0 0\n",
52	>	{  "void glow mat.caustic\n0\n0\n4 %g %g %g 0\n",
53		"mat.caustic sphere obj.caustic\n0\n0\n4 %g %g %g %g\n"
54		},
55	<	{  "void plastic mat.volume\n0\n0\n5 0 1 0 0 0\n",
55	>	{  "void glow mat.volume\n0\n0\n4 %g %g %g 0\n",
56		"mat.volume sphere obj.volume\n0\n0\n4 %g %g %g %g\n"
57		},
58	<	{  "void plastic mat.direct\n0\n0\n5 1 0 1 0 0\n",
58	>	{  "void glow mat.direct\n0\n0\n4 %g %g %g 0\n",
59		"mat.direct sphere obj.direct\n0\n0\n4 %g %g %g %g\n"
60		},
61	<	{  "void plastic mat.contrib\n0\n0\n5 1 1 0 0 0\n",
61	>	{  "void glow mat.contrib\n0\n0\n4 %g %g %g 0\n",
62		"mat.contrib sphere obj.contrib\n0\n0\n4 %g %g %g %g\n"
63		}
64		};
65
66
67	+	/* Default colour codes are as follows:   global         = blue
68	+	precomp global = cyan
69	+	caustic        = red
70	+	volume         = green
71	+	direct         = magenta
72	+	contrib        = yellow */
73	+	const COLOR colDefs [] = {
74	+	{0.25, 0.25, 2}, {0.1, 1, 1}, {1, 0.1, 0.1},
75	+	{0.1, 1, 0.1}, {1, 0.1, 1}, {1, 1, 0.1}
76	+	};
77
78	+
79	+	static int setBool(char *str, unsigned pos, unsigned *var)
80	+	{
81	+	switch ((str) [pos]) {
82	+	case '\0':
83	+	*var = !*var;
84	+	break;
85	+	case 'y': case 'Y': case 't': case 'T': case '+': case '1':
86	+	*var = 1;
87	+	break;
88	+	case 'n': case 'N': case 'f': case 'F': case '-': case '0':
89	+	*var = 0;
90	+	break;
91	+	default:
92	+	return 0;
93	+	}
94	+
95	+	return 1;
96	+	}
97	+
98	+
99		int main (int argc, char** argv)
100		{
101	<	char           format [128];
102	<	RREAL          rad, radScale = RADSCALE, vol, dumpRatio;
103	<	unsigned       arg, j, ptype;
101	>	char           format [MAXFMTLEN];
102	>	RREAL          rad, radScale = RADSCALE, extent, dumpRatio;
103	>	unsigned       arg, j, ptype, dim, fluxCol = 0, points = 0;
104		long           numSpheres = NSPHERES;
105	+	COLOR          col = {0, 0, 0};
106		FILE           *pmapFile;
107		PhotonMap      pm;
108		PhotonPrimary  pri;
#	Line 86 | Line 110 | int main (int argc, char** argv)
110		#ifdef PMAP_OOC
111		char           leafFname [1024];
112		#endif
113	<
113	>
114		if (argc < 2) {
115	<	puts("Dump photon maps as RADIANCE scene description\n");
116	<	printf("Usage: %s [-r radscale1] [-n nspheres1] pmap1 "
117	<	"[-r radscale2] [-n nspheres2] pmap2 ...\n", argv [0]);
115	>	puts("Dump photon maps as RADIANCE scene description "
116	>	"or ASCII point list\n");
117	>	printf("Usage: %s "
118	>	"[-a] [-r radscale1] [-n num1] "
119	>	"[-f | -c rcol1 gcol1 bcol1] pmap1 "
120	>	"[-a] [-r radscale2] [-n num2] "
121	>	"[-f | -c rcol2 gcol2 bcol2] pmap2 "
122	>	"...\n", argv [0]);
123		return 1;
124		}
125
#	Line 98 | Line 127 | int main (int argc, char** argv)
127		/* Parse options */
128		if (argv [arg][0] == '-') {
129		switch (argv [arg][1]) {
130	+	case 'a':
131	+	if (!setBool(argv [arg], 2, &points))
132	+	error(USER, "invalid option syntax at -a");
133	+	break;
134		case 'r':
135		if ((radScale = atof(argv [++arg])) <= 0)
136		error(USER, "invalid radius scale");
#	Line 105 | Line 138 | int main (int argc, char** argv)
138
139		case 'n':
140		if ((numSpheres = parseMultiplier(argv [++arg])) <= 0)
141	<	error(USER, "invalid number of spheres");
141	>	error(USER, "invalid number of points/spheres");
142		break;
143
144	+	case 'c':
145	+	if (fluxCol)
146	+	error(USER, "-f and -c are mutually exclusive");
147	+
148	+	if (badarg(argc - arg - 1, &argv [arg + 1], "fff"))
149	+	error(USER, "invalid RGB colour");
150	+
151	+	for (j = 0; j < 3; j++)
152	+	col [j] = atof(argv [++arg]);
153	+	break;
154	+
155	+	case 'f':
156	+	if (intens(col) > 0)
157	+	error(USER, "-f and -c are mutually exclusive");
158	+
159	+	if (!setBool(argv [arg], 2, &fluxCol))
160	+	error(USER, "invalid option syntax at -f");
161	+	break;
162	+
163		default:
164		sprintf(errmsg, "unknown option %s", argv [arg]);
165		error(USER, errmsg);
#	Line 116 | Line 168 | int main (int argc, char** argv)
168
169		continue;
170		}
171	<
172	<	/* Dump photon map */
171	>
172	>	/* Open next photon map file */
173		if (!(pmapFile = fopen(argv [arg], "rb"))) {
174		sprintf(errmsg, "can't open %s", argv [arg]);
175		error(SYSTEM, errmsg);
176		}
177	<
177	>
178		/* Get format string */
179		strcpy(format, PMAP_FORMAT_GLOB);
180		if (checkheader(pmapFile, format, NULL) != 1) {
#	Line 130 | Line 182 | int main (int argc, char** argv)
182		argv [arg], format);
183		error(USER, errmsg);
184		}
185	<
185	>
186		/* Identify photon map type from format string */
187		for (ptype = 0;
188		ptype < NUM_PMAP_TYPES && strcmp(pmapFormat [ptype], format);
189		ptype++);
190	<
190	>
191		if (!validPmapType(ptype)) {
192		sprintf(errmsg, "file %s contains an unknown photon map type",
193		argv [arg]);
#	Line 145 | Line 197 | int main (int argc, char** argv)
197		/* Get file format version and check for compatibility */
198		if (strcmp(getstr(format, pmapFile), PMAP_FILEVER))
199		error(USER, "incompatible photon map file format");
200	+
201	+	if (!points) {
202	+	/* Dump command line as comment */
203	+	fputs("# ", stdout);
204	+	printargs(argc, argv, stdout);
205	+	fputc('\n', stdout);
206	+	}
207
208	<	/* Dump command line as comment */
209	<	fputs("# ", stdout);
210	<	printargs(argc, argv, stdout);
211	<	fputc('\n', stdout);
208	>	/* Set point/sphere colour if independent of photon flux,
209	>	output RADIANCE material def if required */
210	>	if (!fluxCol) {
211	>	if (intens(col) <= 0)
212	>	copycolor(col, colDefs [ptype]);
213	>	if (!points) {
214	>	printf(radDefs [ptype].mat, col [0], col [1], col [2]);
215	>	fputc('\n', stdout);
216	>	}
217	>	}
218
154	–	/* Dump material def */
155	–	fputs(radDefs [ptype].mat, stdout);
156	–	fputc('\n', stdout);
157	–
219		/* Get number of photons */
220		pm.numPhotons = getint(sizeof(pm.numPhotons), pmapFile);
221	<
221	>
222		/* Skip avg photon flux */
223		for (j = 0; j < 3; j++)
224		getflt(pmapFile);
225	<
225	>
226		/* Get distribution extent (min & max photon positions) */
227		for (j = 0; j < 3; j++) {
228		pm.minPos [j] = getflt(pmapFile);
#	Line 172 | Line 233 | int main (int argc, char** argv)
233		for (j = 0; j < 4; j++)
234		getflt(pmapFile);
235
236	<	/* Sphere radius based on avg intersphere dist
237	<	(= sphere distrib density ^-1/3) */
238	<	vol = (pm.maxPos [0] - pm.minPos [0]) * (pm.maxPos [1] - pm.minPos [1]) *
239	<	(pm.maxPos [2] - pm.minPos [2]);
240	<	rad = radScale * RADCOEFF * pow(vol / numSpheres, 1./3.);
236	>	/* Sphere radius based on avg intersphere dist depending on
237	>	whether the distribution occupies a 1D line (!), a 2D plane,
238	>	or 3D volume (= sphere distrib density ^-1/d, where d is the
239	>	dimensionality of the distribution) */
240	>	for (j = 0, extent = 1.0, dim = 0; j < 3; j++) {
241	>	rad = pm.maxPos [j] - pm.minPos [j];
242	>
243	>	if (rad > FTINY) {
244	>	dim++;
245	>	extent *= rad;
246	>	}
247	>	}
248	>
249	>	rad = radScale * RADCOEFF * pow(extent / numSpheres, 1./dim);
250
251		/* Photon dump probability to satisfy target sphere count */
252	<	dumpRatio = numSpheres < pm.numPhotons
183	<	? (float)numSpheres / pm.numPhotons : 1;
252	>	dumpRatio = min(1, (float)numSpheres / pm.numPhotons);
253
254		/* Skip primary rays */
255		pm.numPrimary = getint(sizeof(pm.numPrimary), pmapFile);
256		while (pm.numPrimary-- > 0) {
257	<	getint(sizeof(pri.srcIdx) + sizeof(pri.dir), pmapFile);
257	>	/* Skip source index & incident dir */
258	>	getint(sizeof(pri.srcIdx), pmapFile);
259	>	#ifdef PMAP_PRIMARYDIR
260	>	/* Skip primary incident dir */
261	>	getint(sizeof(pri.dir), pmapFile);
262	>	#endif
263	>	#ifdef PMAP_PRIMARYPOS
264	>	/* Skip primary hitpoint */
265		for (j = 0; j < 3; j++)
266		getflt(pmapFile);
267	+	#endif
268		}
269
270		#ifdef PMAP_OOC
#	Line 202 | Line 279 | int main (int argc, char** argv)
279		}
280		#endif
281
282	<	/* Load photons */
282	>	/* Read photons */
283		while (pm.numPhotons-- > 0) {
284		#ifdef PMAP_OOC
285	<	/* Get entire photon record
285	>	/* Get entire photon record from ooC octree leaf file
286		!!! OOC PMAP FILES CURRENTLY DON'T USE PORTABLE I/O !!! */
287		if (!fread(&p, sizeof(p), 1, pmapFile)) {
288		sprintf(errmsg, "error reading OOC leaf file %s", leafFname);
289		error(SYSTEM, errmsg);
290		}
291	<	#else
292	<	/* Get photon position */
291	>	#else /* kd-tree */
292	>	/* Get photon position */
293		for (j = 0; j < 3; j++)
294		p.pos [j] = getflt(pmapFile);
295	<	#endif
296	<	/* Dump photon probabilistically acc. to target sphere count */
220	<	if (frandom() <= dumpRatio) {
221	<	printf(radDefs [ptype].obj, p.pos [0], p.pos [1], p.pos [2], rad);
222	<	fputc('\n', stdout);
223	<	}
224	<
225	<	#ifndef PMAP_OOC
226	<	/* Skip photon normal and flux */
295	>
296	>	/* Get photon normal (currently not used) */
297		for (j = 0; j < 3; j++)
298	<	getint(sizeof(p.norm [j]), pmapFile);
299	<
300	<	#ifdef PMAP_FLOAT_FLUX
298	>	p.norm [j] = getint(1, pmapFile);
299	>
300	>	/* Get photon flux */
301	>	#ifdef PMAP_FLOAT_FLUX
302		for (j = 0; j < 3; j++)
303	<	getflt(pmapFile);
304	<	#else
303	>	p.flux [j] = getflt(pmapFile);
304	>	#else
305		for (j = 0; j < 4; j++)
306	<	getint(1, pmapFile);
307	<	#endif
306	>	p.flux [j] = getint(1, pmapFile);
307	>	#endif
308
309		/* Skip primary ray index */
310		getint(sizeof(p.primary), pmapFile);
#	Line 241 | Line 312 | int main (int argc, char** argv)
312		/* Skip flags */
313		getint(sizeof(p.flags), pmapFile);
314		#endif
315	+
316	+	/* Dump photon probabilistically acc. to target sphere count */
317	+	if (frandom() <= dumpRatio) {
318	+	if (fluxCol)
319	+	/* Get photon flux */
320	+	getPhotonFlux(&p, col);
321	+
322	+	if (!points) {
323	+	if (fluxCol) {
324	+	/* Dump material def if variable (depends on flux) */
325	+	printf(radDefs [ptype].mat, col [0], col [1], col [2]);
326	+	fputc('\n', stdout);
327	+	}
328	+	printf(radDefs [ptype].obj, p.pos [0], p.pos [1], p.pos [2],
329	+	rad);
330	+	fputc('\n', stdout);
331	+	}
332	+	else /* Dump as XYZ RGB point */
333	+	printf(POINTFMT, p.pos [0], p.pos [1], p.pos [2],
334	+	col [0], col [1] ,col [2]);
335	+	}
336
337		if (ferror(pmapFile) || feof(pmapFile)) {
338		sprintf(errmsg, "error reading %s", argv [arg]);
#	Line 253 | Line 345 | int main (int argc, char** argv)
345		/* Reset defaults for next dump */
346		radScale = RADSCALE;
347		numSpheres = NSPHERES;
348	+	col [0] = col [1] = col [2] = 0;
349	+	fluxCol = points = 0;
350		}
351
352		return 0;

Diff Legend

–	Removed lines
+	Added lines
<	Changed lines (old)
>	Changed lines (new)