src/rt/pmapio.c

#ifndef lint
static const char RCSid[] = "$Id: pmapio.c,v 2.9 2016/05/17 17:39:47 rschregle Exp $";
#endif

/* 
   ======================================================================
   Photon map file I/O

   Roland Schregle (roland.schregle@{hslu.ch, gmail.com})
   (c) Fraunhofer Institute for Solar Energy Systems,
   (c) Lucerne University of Applied Sciences and Arts,
       supported by the Swiss National Science Foundation (SNSF, #147053)
   ======================================================================
   
   $Id: pmapio.c,v 2.9 2016/05/17 17:39:47 rschregle Exp $    
*/


#include "pmapio.h"
#include "pmapdiag.h"
#include "resolu.h"


void savePhotonMap (const PhotonMap *pmap, const char *fname,
                    int argc, char **argv)
{
   unsigned long  i, j;
   FILE           *file;

   if (!pmap || !pmap -> numPhotons || !validPmapType(pmap -> type)) {
      error(INTERNAL, "attempt to save empty or invalid photon map");
      return;
   }
      
   if (verbose) {
      if (pmap -> numPrimary)
         sprintf(errmsg, "Saving %s (%ld photons, %d primaries)\n", 
                 fname, pmap -> numPhotons, pmap -> numPrimary);
      else sprintf(errmsg, "Saving %s (%ld photons)\n", fname, 
                   pmap -> numPhotons);
                   
      eputs(errmsg);
      fflush(stderr);
   }
   
   if (!(file = fopen(fname, "wb"))) {
      sprintf(errmsg, "can't open photon map file %s", fname);
      error(SYSTEM, errmsg);
   }
      
   /* Write header */
   newheader("RADIANCE", file);
   
   /* Write command line */
   printargs(argc, argv, file);
   
   /* Include statistics in info text */
   fprintf(file, "NumPhotons\t= %ld\n"
           "AvgFlux\t\t= [%.2e, %.2e, %.2e]\n"
           "Bbox\t\t= [%.3f, %.3f, %.3f] [%.3f, %.3f, %.3f]\n"
           "CoG\t\t= [%.3f, %.3f, %.3f]\n"
           "MaxDist^2\t= %.3f\n",
           pmap -> numPhotons, pmap -> photonFlux [0], 
           pmap -> photonFlux [1], pmap -> photonFlux [2],
           pmap -> minPos [0], pmap -> minPos [1], pmap -> minPos [2],
           pmap -> maxPos [0], pmap -> maxPos [1], pmap -> maxPos [2],
           pmap -> CoG [0], pmap -> CoG [1], pmap -> CoG [2], 
           pmap -> CoGdist);
           
   if (pmap -> primaries)
      fprintf(file, "%d primary rays\n", pmap -> numPrimary);
   
   /* Write format */
   fputformat((char*)pmapFormat [pmap -> type], file);
   fprintf(file, "VERSION=%s\n", PMAP_FILEVER);
   
   /* Empty line = end of header */
   putc('\n', file);
   
   /* Write file format version */
   putstr(PMAP_FILEVER, file); 
   
   /* Write number of photons */
   putint(pmap -> numPhotons, sizeof(pmap -> numPhotons), file);
   
   /* Write average photon flux */
   for (j = 0; j < 3; j++) 
      putflt(pmap -> photonFlux [j], file);

   /* Write max and min photon positions */
   for (j = 0; j < 3; j++) {
      putflt(pmap -> minPos [j], file);
      putflt(pmap -> maxPos [j], file);
   }

   /* Write centre of gravity */
   for (j = 0; j < 3; j++)
      putflt(pmap -> CoG [j], file);
      
   /* Write avg distance to centre of gravity */
   putflt(pmap -> CoGdist, file);
   
   /* Write out primary photon rays (or just zero count if none) */
   if (pmap -> primaries) {
      putint(pmap -> numPrimary, sizeof(pmap -> numPrimary), file);
      
      for (i = 0; i < pmap -> numPrimary; i++) {
         PhotonPrimary *prim = pmap -> primaries + i;      

         putint(prim -> srcIdx, sizeof(prim -> srcIdx), file);
         putint(prim -> dir, sizeof(prim -> dir), file);
#ifdef PMAP_PRIMARYPOS         
         for (j = 0; j < 3; j++)
            putflt(prim -> pos [j], file);
#endif
         if (ferror(file))
            error(SYSTEM, "error writing primary photon rays");
      }
   }
   else putint(0, sizeof(pmap -> numPrimary), file);
   
   /* Save photon storage */
#ifdef PMAP_OOC
   if (OOC_SavePhotons(pmap, file)) {
#else
   if (kdT_SavePhotons(pmap, file)) {
#endif
      sprintf(errmsg, "error writing photon map file %s", fname);
      error(SYSTEM, errmsg);
   }

   fclose(file);
}


PhotonMapType loadPhotonMap (PhotonMap *pmap, const char *fname)
{
   PhotonMapType  ptype = PMAP_TYPE_NONE;
   char           format [128];
   unsigned long  i, j;
   FILE           *file;

   if (!pmap)
      return PMAP_TYPE_NONE;
   
   if ((file = fopen(fname, "rb")) == NULL) {
      sprintf(errmsg, "can't open photon map file %s", fname);
      error(SYSTEM, errmsg);
   }

   /* Get format string */
   strcpy(format, PMAP_FORMAT_GLOB);
   if (checkheader(file, format, NULL) != 1) {
      sprintf(errmsg, "photon map file %s has unknown format %s", 
              fname, format);
      error(USER, errmsg);
   }
   
   /* Identify photon map type from format string */
   for (ptype = 0; 
        ptype < NUM_PMAP_TYPES && strcmp(pmapFormat [ptype], format);
        ptype++);
      
   if (!validPmapType(ptype)) {
      sprintf(errmsg, "file %s contains an unknown photon map type", fname);
      error(USER, errmsg);
   }

   initPhotonMap(pmap, ptype);
   
   /* Get file format version and check for compatibility */
   if (strcmp(getstr(format, file), PMAP_FILEVER))
      error(USER, "incompatible photon map file format");
      
   /* Get number of photons */
   pmap -> numPhotons = getint(sizeof(pmap -> numPhotons), file);
      
   /* Get average photon flux */
   for (j = 0; j < 3; j++) 
      pmap -> photonFlux [j] = getflt(file);
   
   /* Get max and min photon positions */
   for (j = 0; j < 3; j++) {
      pmap -> minPos [j] = getflt(file);
      pmap -> maxPos [j] = getflt(file);
   }
   
   /* Get centre of gravity */
   for (j = 0; j < 3; j++)
      pmap -> CoG [j] = getflt(file);
      
   /* Get avg distance to centre of gravity */
   pmap -> CoGdist = getflt(file);

   /* Get primary photon rays */
   pmap -> numPrimary = getint(sizeof(pmap -> numPrimary), file);
   
   if (pmap -> numPrimary) {
      pmap -> primaries = calloc(pmap -> numPrimary, sizeof(PhotonPrimary));      
      if (!pmap -> primaries)
         error(INTERNAL, "can't allocate primary photon rays");
         
      for (i = 0; i < pmap -> numPrimary; i++) {
         PhotonPrimary *prim = pmap -> primaries + i;
         
         prim -> srcIdx = getint(sizeof(prim -> srcIdx), file);
         
         prim -> dir = getint(sizeof(prim -> dir), file);
#ifdef PMAP_PRIMARYPOS            
         for (j = 0; j < 3; j++)
            prim -> pos [j] = getflt(file);
#endif            
         if (feof(file))
            error(SYSTEM, "error reading primary photon rays");
      }
   }      
   
   /* Load photon storage */
#ifdef PMAP_OOC
   if (OOC_LoadPhotons(pmap, file)) {
#else
   if (kdT_LoadPhotons(pmap, file)) {
#endif
      sprintf(errmsg, "error reading photon map file %s", fname);
      error(SYSTEM, errmsg);
   }

   fclose(file);
   return ptype;
}
Revision:	2.10
Committed:	Mon Aug 14 21:12:10 2017 UTC (8 years, 2 months ago) by rschregle
Content type:	text/plain
Branch:	MAIN
CVS Tags:	rad5R1
Changes since 2.9:	+5 -5 lines
Log Message:	Updated photon map code for Windows; no multproc or ooC for now
#	User	Rev	Content
1	greg	2.6	#ifndef lint
2	rschregle	2.10	static const char RCSid[] = "$Id: pmapio.c,v 2.9 2016/05/17 17:39:47 rschregle Exp $";
3	greg	2.6	#endif
4	rschregle	2.9
5	greg	2.1	/*
6	rschregle	2.9	======================================================================
7	greg	2.1	Photon map file I/O
8
9			Roland Schregle (roland.schregle@{hslu.ch, gmail.com})
10			(c) Fraunhofer Institute for Solar Energy Systems,
11	rschregle	2.2	(c) Lucerne University of Applied Sciences and Arts,
12	rschregle	2.9	supported by the Swiss National Science Foundation (SNSF, #147053)
13			======================================================================
14	greg	2.1
15	rschregle	2.10	$Id: pmapio.c,v 2.9 2016/05/17 17:39:47 rschregle Exp $
16	greg	2.1	*/
17
18
19
20			#include "pmapio.h"
21			#include "pmapdiag.h"
22			#include "resolu.h"
23
24
25
26			void savePhotonMap (const PhotonMap pmap, const char fname,
27	greg	2.5	int argc, char **argv)
28	greg	2.1	{
29	rschregle	2.9	unsigned long i, j;
30			FILE *file;
31	greg	2.1
32	rschregle	2.9	if (!pmap \|\| !pmap -> numPhotons \|\| !validPmapType(pmap -> type)) {
33	greg	2.1	error(INTERNAL, "attempt to save empty or invalid photon map");
34			return;
35			}
36
37	rschregle	2.10	if (verbose) {
38	rschregle	2.9	if (pmap -> numPrimary)
39	rschregle	2.10	sprintf(errmsg, "Saving %s (%ld photons, %d primaries)\n",
40	rschregle	2.9	fname, pmap -> numPhotons, pmap -> numPrimary);
41	rschregle	2.10	else sprintf(errmsg, "Saving %s (%ld photons)\n", fname,
42	rschregle	2.9	pmap -> numPhotons);
43
44	greg	2.1	eputs(errmsg);
45			fflush(stderr);
46			}
47
48			if (!(file = fopen(fname, "wb"))) {
49			sprintf(errmsg, "can't open photon map file %s", fname);
50			error(SYSTEM, errmsg);
51			}
52
53			/* Write header */
54			newheader("RADIANCE", file);
55
56			/* Write command line */
57			printargs(argc, argv, file);
58
59			/* Include statistics in info text */
60	rschregle	2.9	fprintf(file, "NumPhotons\t= %ld\n"
61			"AvgFlux\t\t= [%.2e, %.2e, %.2e]\n"
62			"Bbox\t\t= [%.3f, %.3f, %.3f] [%.3f, %.3f, %.3f]\n"
63			"CoG\t\t= [%.3f, %.3f, %.3f]\n"
64			"MaxDist^2\t= %.3f\n",
65			pmap -> numPhotons, pmap -> photonFlux [0],
66	greg	2.1	pmap -> photonFlux [1], pmap -> photonFlux [2],
67			pmap -> minPos [0], pmap -> minPos [1], pmap -> minPos [2],
68	rschregle	2.9	pmap -> maxPos [0], pmap -> maxPos [1], pmap -> maxPos [2],
69			pmap -> CoG [0], pmap -> CoG [1], pmap -> CoG [2],
70			pmap -> CoGdist);
71
72			if (pmap -> primaries)
73			fprintf(file, "%d primary rays\n", pmap -> numPrimary);
74	greg	2.1
75			/* Write format */
76	greg	2.5	fputformat((char*)pmapFormat [pmap -> type], file);
77	rschregle	2.9	fprintf(file, "VERSION=%s\n", PMAP_FILEVER);
78	greg	2.1
79			/* Empty line = end of header */
80			putc('\n', file);
81
82			/* Write file format version */
83	rschregle	2.9	putstr(PMAP_FILEVER, file);
84	greg	2.1
85			/* Write number of photons */
86	rschregle	2.9	putint(pmap -> numPhotons, sizeof(pmap -> numPhotons), file);
87	greg	2.1
88			/* Write average photon flux */
89			for (j = 0; j < 3; j++)
90			putflt(pmap -> photonFlux [j], file);
91
92			/* Write max and min photon positions */
93			for (j = 0; j < 3; j++) {
94			putflt(pmap -> minPos [j], file);
95			putflt(pmap -> maxPos [j], file);
96			}
97
98			/* Write centre of gravity */
99			for (j = 0; j < 3; j++)
100			putflt(pmap -> CoG [j], file);
101
102			/* Write avg distance to centre of gravity */
103			putflt(pmap -> CoGdist, file);
104
105			/* Write out primary photon rays (or just zero count if none) */
106	rschregle	2.9	if (pmap -> primaries) {
107			putint(pmap -> numPrimary, sizeof(pmap -> numPrimary), file);
108	greg	2.1
109	rschregle	2.9	for (i = 0; i < pmap -> numPrimary; i++) {
110			PhotonPrimary *prim = pmap -> primaries + i;
111
112	greg	2.1	putint(prim -> srcIdx, sizeof(prim -> srcIdx), file);
113	greg	2.4	putint(prim -> dir, sizeof(prim -> dir), file);
114	rschregle	2.9	#ifdef PMAP_PRIMARYPOS
115	greg	2.1	for (j = 0; j < 3; j++)
116	greg	2.3	putflt(prim -> pos [j], file);
117	rschregle	2.9	#endif
118	greg	2.1	if (ferror(file))
119			error(SYSTEM, "error writing primary photon rays");
120			}
121			}
122	rschregle	2.9	else putint(0, sizeof(pmap -> numPrimary), file);
123
124			/* Save photon storage */
125			#ifdef PMAP_OOC
126			if (OOC_SavePhotons(pmap, file)) {
127			#else
128			if (kdT_SavePhotons(pmap, file)) {
129			#endif
130			sprintf(errmsg, "error writing photon map file %s", fname);
131			error(SYSTEM, errmsg);
132			}
133	greg	2.1
134			fclose(file);
135			}
136
137
138
139			PhotonMapType loadPhotonMap (PhotonMap pmap, const char fname)
140			{
141	rschregle	2.9	PhotonMapType ptype = PMAP_TYPE_NONE;
142			char format [128];
143			unsigned long i, j;
144			FILE *file;
145	greg	2.1
146			if (!pmap)
147			return PMAP_TYPE_NONE;
148
149			if ((file = fopen(fname, "rb")) == NULL) {
150			sprintf(errmsg, "can't open photon map file %s", fname);
151			error(SYSTEM, errmsg);
152			}
153
154			/* Get format string */
155			strcpy(format, PMAP_FORMAT_GLOB);
156			if (checkheader(file, format, NULL) != 1) {
157			sprintf(errmsg, "photon map file %s has unknown format %s",
158			fname, format);
159			error(USER, errmsg);
160			}
161
162			/* Identify photon map type from format string */
163			for (ptype = 0;
164	rschregle	2.8	ptype < NUM_PMAP_TYPES && strcmp(pmapFormat [ptype], format);
165	greg	2.1	ptype++);
166
167			if (!validPmapType(ptype)) {
168			sprintf(errmsg, "file %s contains an unknown photon map type", fname);
169			error(USER, errmsg);
170			}
171
172			initPhotonMap(pmap, ptype);
173
174			/* Get file format version and check for compatibility */
175	rschregle	2.9	if (strcmp(getstr(format, file), PMAP_FILEVER))
176	greg	2.1	error(USER, "incompatible photon map file format");
177
178			/* Get number of photons */
179	rschregle	2.9	pmap -> numPhotons = getint(sizeof(pmap -> numPhotons), file);
180	greg	2.1
181			/* Get average photon flux */
182			for (j = 0; j < 3; j++)
183			pmap -> photonFlux [j] = getflt(file);
184
185			/* Get max and min photon positions */
186			for (j = 0; j < 3; j++) {
187			pmap -> minPos [j] = getflt(file);
188			pmap -> maxPos [j] = getflt(file);
189			}
190
191			/* Get centre of gravity */
192			for (j = 0; j < 3; j++)
193			pmap -> CoG [j] = getflt(file);
194
195			/* Get avg distance to centre of gravity */
196			pmap -> CoGdist = getflt(file);
197
198			/* Get primary photon rays */
199	rschregle	2.9	pmap -> numPrimary = getint(sizeof(pmap -> numPrimary), file);
200	greg	2.1
201	rschregle	2.9	if (pmap -> numPrimary) {
202			pmap -> primaries = calloc(pmap -> numPrimary, sizeof(PhotonPrimary));
203			if (!pmap -> primaries)
204	greg	2.1	error(INTERNAL, "can't allocate primary photon rays");
205
206	rschregle	2.9	for (i = 0; i < pmap -> numPrimary; i++) {
207			PhotonPrimary *prim = pmap -> primaries + i;
208	greg	2.1
209			prim -> srcIdx = getint(sizeof(prim -> srcIdx), file);
210
211	greg	2.4	prim -> dir = getint(sizeof(prim -> dir), file);
212	rschregle	2.9	#ifdef PMAP_PRIMARYPOS
213	greg	2.1	for (j = 0; j < 3; j++)
214	greg	2.3	prim -> pos [j] = getflt(file);
215	rschregle	2.9	#endif
216	greg	2.1	if (feof(file))
217			error(SYSTEM, "error reading primary photon rays");
218			}
219	rschregle	2.9	}
220
221			/* Load photon storage */
222			#ifdef PMAP_OOC
223			if (OOC_LoadPhotons(pmap, file)) {
224			#else
225			if (kdT_LoadPhotons(pmap, file)) {
226			#endif
227			sprintf(errmsg, "error reading photon map file %s", fname);
228			error(SYSTEM, errmsg);
229	greg	2.1	}
230	rschregle	2.9
231	greg	2.1	fclose(file);
232			return ptype;
233			}