ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/util/wrapBSDF.c
Revision: 2.23
Committed: Wed Apr 3 23:50:25 2019 UTC (3 years, 7 months ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.22: +19 -2 lines
Log Message:
Added check for duplicate component input

File Contents

# User Rev Content
1 greg 2.1 #ifndef lint
2 greg 2.23 static const char RCSid[] = "$Id: wrapBSDF.c,v 2.22 2017/02/14 19:58:37 greg Exp $";
3 greg 2.1 #endif
4     /*
5 greg 2.2 * Wrap BSDF data in valid WINDOW XML file
6 greg 2.1 *
7 greg 2.2 * G. Ward February 2015
8 greg 2.1 */
9    
10 schorsch 2.20 #include "platform.h"
11 greg 2.1 #include <ctype.h>
12     #include "rtio.h"
13 greg 2.17 #include "paths.h"
14 greg 2.1 #include "ezxml.h"
15     #include "bsdf.h"
16     #include "bsdf_m.h"
17     /* XML template file names */
18     const char def_template[] = "minimalBSDFt.xml";
19 greg 2.3 const char win6_template[] = "WINDOW6BSDFt.xml";
20 greg 2.1
21     const char stdin_name[] = "<stdin>";
22     /* input files (can be stdin_name) */
23     const char *xml_input = NULL;
24     /* unit for materials & geometry */
25 greg 2.2 const char *attr_unit = "meter";
26 greg 2.1 const char legal_units[] = "meter|foot|inch|centimeter|millimeter";
27     /* system materials & geometry */
28     const char *mgf_geometry = NULL;
29 greg 2.16 /* comment list */
30     #define MAXCOMM 30
31     const char *commlist[MAXCOMM];
32     int ncomm = 0;
33 greg 2.6 /* angle bases */
34 greg 2.1 enum { ABdefault=-1, ABklemsFull=0, ABklemsHalf, ABklemsQuarter,
35     ABtensorTree3, ABtensorTree4, ABend };
36    
37     int angle_basis = ABdefault;
38    
39 greg 2.7 int correct_solid_angle = 0;
40    
41     const char *klems_basis_name[] = {
42     "LBNL/Klems Full",
43     "LBNL/Klems Half",
44     "LBNL/Klems Quarter",
45     };
46 greg 2.1 /* field IDs and nicknames */
47     struct s_fieldID {
48     char nickName[4];
49 greg 2.3 short has_unit;
50     short win_need;
51 greg 2.1 const char *fullName;
52     } XMLfieldID[] = {
53 greg 2.3 {"m", 0, 1, "Manufacturer"},
54     {"n", 0, 1, "Name"},
55 greg 2.22 {"d", 0, 0, "DeviceType"},
56 greg 2.3 {"c", 0, 0, "ThermalConductivity"},
57     {"ef", 0, 0, "EmissivityFront"},
58     {"eb", 0, 0, "EmissivityBack"},
59     {"tir", 0, 0, "TIR"},
60     {"eo", 0, 0, "EffectiveOpennessFraction"},
61     {"t", 1, 1, "Thickness"},
62     {"h", 1, 0, "Height"},
63     {"w", 1, 0, "Width"},
64     {"\0", 0, 0, NULL} /* terminator */
65 greg 2.1 };
66     /* field assignments */
67 greg 2.22 #define MAXASSIGN 16
68 greg 2.1 const char *field_assignment[MAXASSIGN];
69     int nfield_assign = 0;
70     #define FASEP ';'
71     /* data file(s) & spectra */
72     enum { DTtransForward, DTtransBackward, DTreflForward, DTreflBackward };
73    
74 greg 2.11 enum { DSsolar=-1, DSnir=-2, DSxbar31=-3, DSvisible=-4, DSzbar31=-5,
75     DSuprime=-6, DSvprime=-7 };
76 greg 2.1
77     #define MAXFILES 20
78    
79     struct s_dfile {
80     const char *fname; /* input data file name */
81     short type; /* BSDF data type */
82     short spectrum; /* BSDF sensor spectrum */
83     } data_file[MAXFILES];
84    
85     int ndataf = 0; /* number of data files */
86    
87 greg 2.18 int unlink_datafiles = 0; /* unlink data files when done */
88    
89 greg 2.1 const char *spectr_file[MAXFILES]; /* custom spectral curve input */
90    
91     const char top_level_name[] = "WindowElement";
92    
93 greg 2.3 static char basis_definition[][256] = {
94 greg 2.1
95 greg 2.4 "\t<DataDefinition>\n"
96 greg 2.1 "\t\t<IncidentDataStructure>Columns</IncidentDataStructure>\n"
97     "\t\t<AngleBasis>\n"
98     "\t\t\t<AngleBasisName>LBNL/Klems Full</AngleBasisName>\n"
99 greg 2.4 "\t\t\t</AngleBasis>\n"
100     "\t</DataDefinition>\n",
101 greg 2.1
102 greg 2.4 "\t<DataDefinition>\n"
103 greg 2.1 "\t\t<IncidentDataStructure>Columns</IncidentDataStructure>\n"
104     "\t\t<AngleBasis>\n"
105     "\t\t\t<AngleBasisName>LBNL/Klems Half</AngleBasisName>\n"
106 greg 2.4 "\t\t\t</AngleBasis>\n"
107     "\t</DataDefinition>\n",
108 greg 2.1
109 greg 2.4 "\t<DataDefinition>\n"
110 greg 2.1 "\t\t<IncidentDataStructure>Columns</IncidentDataStructure>\n"
111     "\t\t<AngleBasis>\n"
112     "\t\t\t<AngleBasisName>LBNL/Klems Quarter</AngleBasisName>\n"
113 greg 2.4 "\t\t\t</AngleBasis>\n"
114     "\t</DataDefinition>\n",
115 greg 2.1
116 greg 2.4 "\t<DataDefinition>\n"
117     "\t\t<IncidentDataStructure>TensorTree3</IncidentDataStructure>\n"
118     "\t</DataDefinition>\n",
119    
120     "\t<DataDefinition>\n"
121     "\t\t<IncidentDataStructure>TensorTree4</IncidentDataStructure>\n"
122     "\t</DataDefinition>\n",
123 greg 2.1 };
124    
125 greg 2.23 /* Check that the last-added data file is unique */
126     static int
127     check_new_data_file()
128     {
129     int i = ndataf;
130    
131     while (i-- > 0)
132     if ((data_file[i].spectrum == data_file[ndataf].spectrum) &
133     (data_file[i].type == data_file[ndataf].type)) {
134     fprintf(stderr,
135     "%s: warning - ignoring duplicate component\n",
136     data_file[ndataf].fname);
137     return 0;
138     }
139     return 1;
140     }
141    
142 greg 2.1 /* Copy data from file descriptor to stdout and close */
143     static int
144     copy_and_close(int fd)
145     {
146     int ok = 1;
147     char buf[8192];
148     int n;
149    
150     if (fd < 0)
151     return 0;
152     while ((n = read(fd, buf, sizeof(buf))) > 0)
153     if (write(fileno(stdout), buf, n) != n) {
154     ok = 0;
155     break;
156     }
157     ok &= (n == 0);
158     close(fd);
159     return ok;
160     }
161    
162     /* Allocate and assign string from file or stream */
163     static char *
164     input2str(const char *inpspec)
165     {
166     FILE *fp = NULL;
167     char *str;
168     int len, pos, n;
169    
170     if (inpspec == NULL || !*inpspec)
171     return "";
172     if (inpspec == stdin_name) { /* read from stdin */
173     fp = stdin;
174 greg 2.2 } else if (inpspec[0] == '!') { /* read from command */
175 greg 2.1 fp = popen(inpspec+1, "r");
176     if (fp == NULL) {
177     fprintf(stderr, "Cannot start process '%s'\n",
178 greg 2.2 inpspec);
179 greg 2.1 return "";
180     }
181     } else { /* else load file */
182     int fd = open(inpspec, O_RDONLY);
183     if (fd < 0) {
184     fprintf(stderr, "%s: cannot open\n", inpspec);
185     return "";
186     }
187 schorsch 2.20 #if !defined(_WIN32) && !defined(_WIN64)
188     /* XXX somehow broken on Windows */
189 greg 2.1 len = lseek(fd, 0L, SEEK_END);
190 greg 2.3 if (len > 0) {
191     lseek(fd, 0L, SEEK_SET);
192     str = (char *)malloc(len+1);
193     if (str == NULL) {
194     close(fd);
195     goto memerr;
196     }
197     if (read(fd, str, len) != len) {
198     fprintf(stderr, "%s: read error\n", inpspec);
199     free(str);
200     close(fd);
201     return "";
202     }
203     str[len] = '\0';
204 greg 2.1 close(fd);
205 greg 2.3 return str;
206 greg 2.1 }
207 greg 2.12 #endif
208 greg 2.3 fp = fdopen(fd, "r"); /* not a regular file */
209 greg 2.1 }
210     /* reading from stream */
211     str = (char *)malloc((len=8192)+1);
212     if (str == NULL)
213     goto memerr;
214     pos = 0;
215     while ((n = read(fileno(fp), str+pos, len-pos)) > 0)
216     if ((pos += n) >= len) { /* need more space? */
217     str = (char *)realloc(str, (len += len>>2) + 1);
218     if (str == NULL)
219     goto memerr;
220     }
221     if (n < 0) {
222     fprintf(stderr, "%s: read error\n", inpspec);
223     free(str);
224     str = "";
225     } else { /* tidy up result */
226     str[pos] = '\0';
227     str = (char *)realloc(str, (len=pos)+1);
228     if (str == NULL)
229     goto memerr;
230     }
231 greg 2.2 if (inpspec[0] != '!')
232 greg 2.1 fclose(fp);
233     else if (pclose(fp))
234 greg 2.2 fprintf(stderr, "Error running command '%s'\n", inpspec);
235 greg 2.1 return str;
236     memerr:
237     fprintf(stderr, "%s: error allocating memory\n", inpspec);
238     if (fp != NULL)
239 greg 2.2 (inpspec[0] == '!') ? pclose(fp) : fclose(fp);
240 greg 2.1 return "";
241     }
242    
243     /* Make material assignments in field_assignment to XML fields */
244     static int
245     mat_assignments(const char *caller, const char *fn, ezxml_t wtl)
246     {
247     int i;
248    
249     wtl = ezxml_child(wtl, "Material");
250     if (wtl == NULL) {
251     fprintf(stderr, "%s: missing <Material> tag\n", fn);
252     return 0;
253     }
254     for (i = 0; i < nfield_assign; i++) {
255     const char *fnext = field_assignment[i];
256     for ( ; ; ) {
257 greg 2.5 int added = 0;
258 greg 2.1 ezxml_t fld;
259     char sbuf[512];
260     int j;
261    
262     while (isspace(*fnext))
263     ++fnext;
264     if (!*fnext)
265     break;
266 greg 2.3 for (j = 0; *fnext != '=' && !isspace(*fnext); ) {
267 greg 2.1 if (!*fnext | (*fnext == FASEP) |
268     (j >= sizeof(sbuf)-1)) {
269     fprintf(stderr,
270     "%s: bad tag name in assignment '%s'\n",
271     caller, field_assignment[i]);
272     return 0;
273     }
274     sbuf[j++] = *fnext++;
275     }
276 greg 2.2 sbuf[j] = '\0'; /* check known field */
277 greg 2.1 for (j = 0; XMLfieldID[j].nickName[0]; j++)
278 greg 2.2 if (!strcasecmp(sbuf, XMLfieldID[j].nickName) ||
279     !strcasecmp(sbuf, XMLfieldID[j].fullName)) {
280 greg 2.1 strcpy(sbuf, XMLfieldID[j].fullName);
281     break;
282     }
283     /* check if tag exists */
284     fld = ezxml_child(wtl, sbuf);
285     if (fld == NULL) { /* otherwise, create one */
286 greg 2.3 if (!XMLfieldID[j].nickName[0])
287     fprintf(stderr,
288     "%s: warning - adding tag <%s>\n",
289     fn, sbuf);
290 greg 2.5 ezxml_add_txt(wtl, "\t");
291 greg 2.1 fld = ezxml_add_child_d(wtl, sbuf, strlen(wtl->txt));
292 greg 2.5 ++added;
293 greg 2.1 }
294     if (XMLfieldID[j].has_unit)
295     ezxml_set_attr(fld, "unit", attr_unit);
296 greg 2.3 XMLfieldID[j].win_need = 0;
297 greg 2.1 while (isspace(*fnext))
298     ++fnext;
299     if (*fnext++ != '=') {
300     fprintf(stderr,
301     "%s: missing '=' in assignment '%s'\n",
302     caller, field_assignment[i]);
303     return 0;
304     }
305 greg 2.3 for (j = 0; *fnext && *fnext != FASEP; ) {
306 greg 2.1 if (j >= sizeof(sbuf)-1) {
307     fprintf(stderr,
308     "%s: field too long in '%s'\n",
309     caller, field_assignment[i]);
310     return 0;
311     }
312     sbuf[j++] = *fnext++;
313     }
314     sbuf[j] = '\0';
315     ezxml_set_txt_d(fld, sbuf);
316 greg 2.5 if (added)
317     ezxml_add_txt(wtl, "\n\t");
318 greg 2.1 fnext += (*fnext == FASEP);
319     }
320     }
321 greg 2.3 /* check required WINDOW settings */
322     if (xml_input == win6_template)
323     for (i = 0; XMLfieldID[i].nickName[0]; i++)
324     if (XMLfieldID[i].win_need &&
325 greg 2.10 !ezxml_txt(ezxml_child(wtl,XMLfieldID[i].fullName))[0])
326 greg 2.3 fprintf(stderr,
327 greg 2.10 "%s: warning - missing '%s' assignment for WINDOW <%s>\n",
328 greg 2.3 caller, XMLfieldID[i].nickName,
329     XMLfieldID[i].fullName);
330 greg 2.10 return 1;
331 greg 2.1 }
332    
333     /* Complete angle basis specification */
334     static int
335     finish_angle_basis(ezxml_t ab)
336     {
337     const char *bn = ezxml_txt(ezxml_child(ab, "AngleBasisName"));
338     int i, n = nabases;
339     char buf[32];
340    
341     if (!*bn) {
342     fputs("Internal error - missing <AngleBasisName>!\n", stderr);
343     return 0;
344     }
345     while (n-- > 0)
346     if (!strcasecmp(bn, abase_list[n].name))
347     break;
348     if (n < 0) {
349     fprintf(stderr, "Internal error - unknown angle basis '%s'", bn);
350     return 0;
351     }
352     for (i = 0; abase_list[n].lat[i].nphis; i++) {
353     ezxml_t tb, abb = ezxml_add_child(ab, "AngleBasisBlock",
354     strlen(ab->txt));
355     sprintf(buf, "%g", i ?
356     .5*(abase_list[n].lat[i].tmin + abase_list[n].lat[i+1].tmin) :
357     .0);
358 greg 2.4 ezxml_add_txt(abb, "\n\t\t\t\t");
359 greg 2.1 ezxml_set_txt_d(ezxml_add_child(abb,"Theta",strlen(abb->txt)), buf);
360     sprintf(buf, "%d", abase_list[n].lat[i].nphis);
361 greg 2.4 ezxml_add_txt(abb, "\n\t\t\t\t");
362 greg 2.1 ezxml_set_txt_d(ezxml_add_child(abb,"nPhis",strlen(abb->txt)), buf);
363 greg 2.4 ezxml_add_txt(abb, "\n\t\t\t\t");
364 greg 2.1 tb = ezxml_add_child(abb, "ThetaBounds", strlen(abb->txt));
365 greg 2.4 ezxml_add_txt(tb, "\n\t\t\t\t\t");
366 greg 2.1 sprintf(buf, "%g", abase_list[n].lat[i].tmin);
367 greg 2.4 ezxml_set_txt_d(ezxml_add_child(tb,"LowerTheta",strlen(tb->txt)), buf);
368     ezxml_add_txt(tb, "\n\t\t\t\t\t");
369 greg 2.1 sprintf(buf, "%g", abase_list[n].lat[i+1].tmin);
370 greg 2.4 ezxml_set_txt_d(ezxml_add_child(tb,"UpperTheta",strlen(tb->txt)), buf);
371     ezxml_add_txt(tb, "\n\t\t\t\t");
372     ezxml_add_txt(abb, "\n\t\t\t");
373     ezxml_add_txt(ab, "\n\t\t\t");
374 greg 2.1 }
375     return 1;
376     }
377    
378     /* Determine our angle basis from current tags */
379     static int
380     determine_angle_basis(const char *fn, ezxml_t wtl)
381     {
382     const char *ids;
383     int i;
384    
385     wtl = ezxml_child(wtl, "DataDefinition");
386     if (wtl == NULL)
387     return -1;
388     ids = ezxml_txt(ezxml_child(wtl, "IncidentDataStructure"));
389 greg 2.4 if (!ids[0])
390 greg 2.1 return -1;
391     for (i = 0; i < ABend; i++) {
392     ezxml_t parsed = ezxml_parse_str(basis_definition[i],
393     strlen(basis_definition[i]));
394     int match = 0;
395     if (!strcmp(ids, ezxml_txt(ezxml_child(parsed,
396     "IncidentDataStructure")))) {
397     const char *abn0 = ezxml_txt(
398     ezxml_child(ezxml_child(wtl,
399     "AngleBasis"), "AngleBasisName"));
400     const char *abn1 = ezxml_txt(
401     ezxml_child(ezxml_child(parsed,
402     "AngleBasis"), "AngleBasisName"));
403     match = !strcmp(abn0, abn1);
404     }
405     ezxml_free(parsed);
406     if (match)
407     return i;
408     }
409     return -1;
410     }
411    
412 greg 2.7 /* Filter Klems angle basis, factoring out incident projected solid angle */
413 greg 2.6 static int
414     filter_klems_matrix(FILE *fp)
415     {
416 greg 2.7 const char *bn = klems_basis_name[angle_basis];
417 greg 2.6 int i, j, n = nabases;
418     /* get angle basis */
419     while (n-- > 0)
420     if (!strcasecmp(bn, abase_list[n].name))
421     break;
422     if (n < 0)
423     return 0;
424     /* read/correct/write matrix */
425     for (i = 0; i < abase_list[n].nangles; i++) {
426 greg 2.21 const double corr = 1./io_getohm(i, &abase_list[n]);
427 greg 2.6 for (j = 0; j < abase_list[n].nangles; j++) {
428     double d;
429     if (fscanf(fp, "%lf", &d) != 1)
430     return 0;
431 greg 2.7 if (d < -1e-3) {
432     fputs("Negative BSDF data!\n", stderr);
433     return 0;
434     }
435 greg 2.21 printf(" %.3e", d*corr*(d > 0));
436 greg 2.6 }
437     fputc('\n', stdout);
438     }
439 greg 2.7 while ((i = getc(fp)) != EOF)
440     if (!isspace(i)) {
441     fputs("Unexpected data past EOF\n", stderr);
442     return 0;
443     }
444     return 1; /* all is good */
445 greg 2.6 }
446    
447 greg 2.1 /* Write out BSDF data block with surrounding tags */
448     static int
449     writeBSDFblock(const char *caller, struct s_dfile *df)
450     {
451 greg 2.7 int correct_klems = correct_solid_angle;
452 greg 2.1 char *cp;
453    
454     puts("\t<WavelengthData>");
455     puts("\t\t<LayerNumber>System</LayerNumber>");
456     switch (df->spectrum) {
457     case DSvisible:
458     puts("\t\t<Wavelength unit=\"Integral\">Visible</Wavelength>");
459 greg 2.7 puts("\t\t<SourceSpectrum>CIE Illuminant D65 1nm.ssp</SourceSpectrum>");
460 greg 2.1 puts("\t\t<DetectorSpectrum>ASTM E308 1931 Y.dsp</DetectorSpectrum>");
461     break;
462     case DSxbar31:
463     puts("\t\t<Wavelength unit=\"Integral\">CIE-X</Wavelength>");
464 greg 2.14 puts("\t\t<SourceSpectrum>CIE Illuminant D65 1nm.ssp</SourceSpectrum>");
465 greg 2.1 puts("\t\t<DetectorSpectrum>ASTM E308 1931 X.dsp</DetectorSpectrum>");
466     break;
467     case DSzbar31:
468     puts("\t\t<Wavelength unit=\"Integral\">CIE-Z</Wavelength>");
469 greg 2.14 puts("\t\t<SourceSpectrum>CIE Illuminant D65 1nm.ssp</SourceSpectrum>");
470 greg 2.1 puts("\t\t<DetectorSpectrum>ASTM E308 1931 Z.dsp</DetectorSpectrum>");
471     break;
472 greg 2.11 case DSuprime:
473 greg 2.13 puts("\t\t<Wavelength unit=\"Integral\">CIE-u</Wavelength>");
474 greg 2.14 puts("\t\t<SourceSpectrum>CIE Illuminant D65 1nm.ssp</SourceSpectrum>");
475 greg 2.11 break;
476     case DSvprime:
477 greg 2.13 puts("\t\t<Wavelength unit=\"Integral\">CIE-v</Wavelength>");
478 greg 2.14 puts("\t\t<SourceSpectrum>CIE Illuminant D65 1nm.ssp</SourceSpectrum>");
479 greg 2.11 break;
480 greg 2.1 case DSsolar:
481     puts("\t\t<Wavelength unit=\"Integral\">Solar</Wavelength>");
482 greg 2.14 puts("\t\t<SourceSpectrum>CIE Illuminant D65 1nm.ssp</SourceSpectrum>");
483 greg 2.1 puts("\t\t<DetectorSpectrum>None</DetectorSpectrum>");
484     break;
485     case DSnir:
486     puts("\t\t<Wavelength unit=\"Integral\">NIR</Wavelength>");
487 greg 2.14 puts("\t\t<SourceSpectrum>PLACE_HOLDER</SourceSpectrum>");
488 greg 2.1 puts("\t\t<DetectorSpectrum>PLACE_HOLDER</DetectorSpectrum>");
489     break;
490     default:
491     cp = strrchr(spectr_file[df->spectrum], '.');
492     if (cp != NULL)
493     *cp = '\0';
494     printf("\t\t<Wavelength unit=\"Integral\">%s</Wavelength>\n",
495     spectr_file[df->spectrum]);
496     if (cp != NULL)
497     *cp = '.';
498 greg 2.14 puts("\t\t<SourceSpectrum>CIE Illuminant D65 1nm.ssp</SourceSpectrum>");
499 greg 2.1 printf("\t\t<DetectorSpectrum>%s</DetectorSpectrum>\n",
500     spectr_file[df->spectrum]);
501     break;
502     }
503     puts("\t\t<WavelengthDataBlock>");
504     fputs("\t\t\t<WavelengthDataDirection>", stdout);
505     switch (df->type) {
506     case DTtransForward:
507     fputs("Transmission Front", stdout);
508     break;
509     case DTtransBackward:
510     fputs("Transmission Back", stdout);
511     break;
512     case DTreflForward:
513     fputs("Reflection Front", stdout);
514     break;
515     case DTreflBackward:
516     fputs("Reflection Back", stdout);
517     break;
518     default:
519     fprintf(stderr, "%s: internal - bad BSDF type (%d)\n", caller, df->type);
520     return 0;
521     }
522     puts("</WavelengthDataDirection>");
523     switch (angle_basis) {
524     case ABklemsFull:
525     case ABklemsHalf:
526     case ABklemsQuarter:
527 greg 2.7 fputs("\t\t\t<ColumnAngleBasis>", stdout);
528     fputs(klems_basis_name[angle_basis], stdout);
529     puts("</ColumnAngleBasis>");
530     fputs("\t\t\t<RowAngleBasis>", stdout);
531     fputs(klems_basis_name[angle_basis], stdout);
532     puts("</RowAngleBasis>");
533 greg 2.1 break;
534     case ABtensorTree3:
535     case ABtensorTree4:
536     puts("\t\t\t<AngleBasis>LBNL/Shirley-Chiu</AngleBasis>");
537 greg 2.7 correct_klems = 0;
538 greg 2.1 break;
539     default:
540     fprintf(stderr, "%s: bad angle basis (%d)\n", caller, angle_basis);
541     return 0;
542     }
543     puts("\t\t\t<ScatteringDataType>BTDF</ScatteringDataType>");
544     puts("\t\t\t<ScatteringData>");
545     fflush(stdout);
546 greg 2.7 if (correct_klems) { /* correct Klems matrix data */
547 greg 2.6 FILE *fp = stdin;
548     if (df->fname[0] == '!')
549     fp = popen(df->fname+1, "r");
550     else if (df->fname != stdin_name)
551     fp = fopen(df->fname, "r");
552     if (fp == NULL) {
553     fprintf(stderr, "%s: cannot open '%s'\n",
554     caller, df->fname);
555     return 0;
556     }
557     if (!filter_klems_matrix(fp)) {
558     fprintf(stderr, "%s: Klems data error from '%s'\n",
559     caller, df->fname);
560     return 0;
561     }
562     if (df->fname[0] != '!') {
563     fclose(fp);
564     } else if (pclose(fp)) {
565     fprintf(stderr, "%s: error running '%s'\n",
566     caller, df->fname);
567     return 0;
568     }
569     } else if (df->fname == stdin_name) {
570 greg 2.1 copy_and_close(fileno(stdin));
571 greg 2.2 } else if (df->fname[0] != '!') {
572 greg 2.1 if (!copy_and_close(open(df->fname, O_RDONLY))) {
573     fprintf(stderr, "%s: error reading from '%s'\n",
574     caller, df->fname);
575     return 0;
576     }
577     } else if (system(df->fname+1)) {
578 greg 2.2 fprintf(stderr, "%s: error running '%s'\n", caller, df->fname);
579 greg 2.1 return 0;
580     }
581     puts("\t\t\t</ScatteringData>");
582     puts("\t\t</WavelengthDataBlock>");
583     puts("\t</WavelengthData>");
584     return 1;
585     }
586    
587     /* Write out XML, interpolating BSDF data block(s) */
588     static int
589     writeBSDF(const char *caller, ezxml_t fl)
590     {
591 greg 2.2 char *xml = ezxml_toxml(fl); /* store XML in string */
592 greg 2.1 int ei, i;
593     /* locate trailer */
594     for (ei = strlen(xml)-strlen("</Layer></Optical></WindowElement>");
595     ei >= 0; ei--)
596     if (!strncmp(xml+ei, "</Layer>", 8))
597     break;
598     if (ei < 0) {
599     fprintf(stderr, "%s: internal - cannot find trailer\n",
600     caller);
601     free(xml);
602     return 0;
603     }
604 greg 2.15 puts("<?xml version=\"1.0\" encoding=\"UTF-8\"?>");
605 greg 2.16 for (i = 0; i < ncomm; i++)
606     printf("<!-- %s -->\n", commlist[i]);
607 greg 2.1 fflush(stdout); /* write previous XML info. */
608     if (write(fileno(stdout), xml, ei) != ei) {
609     free(xml);
610     return 0;
611     }
612     for (i = 0; i < ndataf; i++) /* interpolate new data */
613     if (!writeBSDFblock(caller, &data_file[i])) {
614     free(xml);
615     return 0;
616     }
617     fputs(xml+ei, stdout); /* write trailer */
618 greg 2.2 free(xml); /* free string */
619 greg 2.4 fputc('\n', stdout);
620 greg 2.18 if (fflush(stdout) != 0)
621     return 0;
622     /* unlink data files if req. */
623     for (i = ndataf*(unlink_datafiles != 0); i--; )
624     if (data_file[i].fname != stdin_name &&
625     data_file[i].fname[0] != '!')
626     unlink(data_file[i].fname);
627 greg 2.19 if (unlink_datafiles > 1 && mgf_geometry != NULL &&
628     mgf_geometry != stdin_name &&
629     mgf_geometry[0] != '!')
630     unlink(mgf_geometry);
631 greg 2.18 return 1;
632 greg 2.1 }
633    
634     /* Insert BSDF data into XML wrapper */
635     static int
636     wrapBSDF(const char *caller)
637     {
638 greg 2.2 const char *xml_path = xml_input;
639 greg 2.1 ezxml_t fl, wtl;
640     /* load previous XML/template */
641 greg 2.2 if (xml_input == stdin_name) {
642     fl = ezxml_parse_fp(stdin);
643     } else if (xml_input[0] == '!') {
644     FILE *pfp = popen(xml_input+1, "r");
645     if (pfp == NULL) {
646     fprintf(stderr, "%s: cannot start process '%s'\n",
647     caller, xml_input);
648     return 0;
649     }
650     fl = ezxml_parse_fp(pfp);
651     if (pclose(pfp)) {
652     fprintf(stderr, "%s: error running '%s'\n",
653     caller, xml_input);
654     return 0;
655     }
656     } else {
657     xml_path = getpath((char *)xml_input, getrlibpath(), R_OK);
658     if (xml_path == NULL) {
659     fprintf(stderr, "%s: cannot find XML file named '%s'\n",
660     caller, xml_input==NULL ? "NULL" : xml_input);
661     return 0;
662     }
663     fl = ezxml_parse_file(xml_path);
664 greg 2.1 }
665     if (fl == NULL) {
666 greg 2.2 fprintf(stderr, "%s: cannot load XML '%s'\n", caller, xml_path);
667 greg 2.1 return 0;
668     }
669     if (ezxml_error(fl)[0]) {
670 greg 2.2 fprintf(stderr, "%s: error in XML '%s': %s\n", caller, xml_path,
671 greg 2.1 ezxml_error(fl));
672     goto failure;
673     }
674     if (strcmp(ezxml_name(fl), top_level_name)) {
675     fprintf(stderr, "%s: top level in XML '%s' not '%s'\n",
676     caller, xml_path, top_level_name);
677     goto failure;
678     }
679     wtl = ezxml_child(fl, "FileType");
680     if (wtl != NULL && strcmp(ezxml_txt(wtl), "BSDF")) {
681     fprintf(stderr, "%s: wrong FileType in XML '%s' (must be 'BSDF')",
682     caller, xml_path);
683     goto failure;
684     }
685     wtl = ezxml_child(ezxml_child(fl, "Optical"), "Layer");
686     if (wtl == NULL) {
687     fprintf(stderr, "%s: no optical layers in XML '%s'",
688     caller, xml_path);
689     goto failure;
690     }
691     /* make material assignments */
692 greg 2.3 if (!mat_assignments(caller, xml_path, wtl))
693 greg 2.1 goto failure;
694     if (mgf_geometry != NULL) { /* add geometry if specified */
695     ezxml_t geom = ezxml_child(wtl, "Geometry");
696     if (geom == NULL)
697 greg 2.3 geom = ezxml_add_child(wtl, "Geometry", strlen(wtl->txt));
698 greg 2.1 ezxml_set_attr(geom, "format", "MGF");
699     geom = ezxml_child(geom, "MGFblock");
700     if (geom == NULL) {
701     geom = ezxml_child(wtl, "Geometry");
702 greg 2.3 geom = ezxml_add_child(geom, "MGFblock", 0);
703 greg 2.1 }
704     ezxml_set_attr(geom, "unit", attr_unit);
705     ezxml_set_txt(geom, input2str(mgf_geometry));
706     if (geom->txt[0])
707     ezxml_set_flag(geom, EZXML_TXTM);
708     }
709     /* check basis */
710     if (angle_basis != ABdefault) {
711 greg 2.4 size_t offset = 0;
712 greg 2.1 ezxml_t ab, dd = ezxml_child(wtl, "DataDefinition");
713 greg 2.4 if (dd != NULL) {
714     offset = dd->off;
715 greg 2.5 if (dd->child != NULL)
716     fprintf(stderr,
717 greg 2.1 "%s: warning - replacing existing <DataDefinition> in '%s'\n",
718     caller, xml_path);
719 greg 2.4 ezxml_remove(dd);
720     } else
721     offset = strlen(wtl->txt);
722     dd = ezxml_insert(ezxml_parse_str(basis_definition[angle_basis],
723 greg 2.1 strlen(basis_definition[angle_basis])),
724 greg 2.4 wtl, offset);
725 greg 2.1 if ((ab = ezxml_child(dd, "AngleBasis")) != NULL &&
726     !finish_angle_basis(ab))
727     goto failure;
728     } else if ((angle_basis = determine_angle_basis(xml_path, wtl)) < 0) {
729     fprintf(stderr, "%s: need -a option to set angle basis\n",
730     caller);
731     goto failure;
732     }
733     /* write & add BSDF data blocks */
734     if (!writeBSDF(caller, fl))
735     goto failure;
736     ezxml_free(fl); /* all done */
737     return 1;
738     failure:
739     ezxml_free(fl);
740     return 0;
741     }
742    
743     /* Report usage and exit */
744     static void
745     UsageExit(const char *pname)
746     {
747     fputs("Usage: ", stderr);
748     fputs(pname, stderr);
749 greg 2.11 fputs(" [-W][-c][-a {kf|kh|kq|t3|t4}][-u unit][-g geom][-f 'x=string;y=string']", stderr);
750 greg 2.16 fputs(" [-s spectr][-tb inp][-tf inp][-rb inp][-rf inp][-C comm]", stderr);
751 greg 2.2 fputs(" [input.xml]\n", stderr);
752 greg 2.1 exit(1);
753     }
754    
755     /* Load XML file and use to wrap BSDF data (or modify fields) */
756     int
757     main(int argc, char *argv[])
758     {
759     int cur_spectrum = DSvisible;
760     int ncust_spec = 0;
761     int used_stdin = 0;
762     int units_set = 0;
763     int i;
764     /* get/check arguments */
765     for (i = 1; i < argc && argv[i][0] == '-'; i++) {
766     switch (argv[i][1]) {
767     case 'W': /* customize for WINDOW 6 output */
768     xml_input = win6_template;
769     continue;
770     case 'f': /* field assignment(s) */
771     if (++i >= argc)
772     UsageExit(argv[0]);
773     if (nfield_assign >= MAXASSIGN) {
774     fprintf(stderr, "%s: too many -f options",
775     argv[0]);
776     return 1;
777     }
778     field_assignment[nfield_assign++] = argv[i];
779     continue;
780     case 'u': /* unit */
781     if (++i >= argc)
782     UsageExit(argv[0]);
783     if (units_set++) {
784     fprintf(stderr, "%s: only one -u option allowed\n",
785     argv[0]);
786     return 1;
787     }
788 greg 2.2 if (strstr(legal_units, argv[i]) == NULL) {
789     fprintf(stderr, "%s: -u unit must be one of (%s)\n",
790 greg 2.1 argv[0], legal_units);
791     return 1;
792     }
793     attr_unit = argv[i];
794     continue;
795 greg 2.18 case 'U': /* unlink data files when done */
796 greg 2.19 unlink_datafiles = 1 + (argv[i][2] == 'U');
797 greg 2.18 continue;
798 greg 2.1 case 'a': /* angle basis */
799     if (++i >= argc)
800     UsageExit(argv[0]);
801     if (angle_basis != ABdefault) {
802     fprintf(stderr, "%s: only one -a option allowed\n",
803     argv[0]);
804     return 1;
805     }
806     if (!strcasecmp(argv[i], "kf"))
807     angle_basis = ABklemsFull;
808     else if (!strcasecmp(argv[i], "kh"))
809     angle_basis = ABklemsHalf;
810     else if (!strcasecmp(argv[i], "kq"))
811     angle_basis = ABklemsQuarter;
812     else if (!strcasecmp(argv[i], "t3"))
813     angle_basis = ABtensorTree3;
814     else if (!strcasecmp(argv[i], "t4"))
815     angle_basis = ABtensorTree4;
816     else
817     UsageExit(argv[0]);
818     continue;
819 greg 2.7 case 'c': /* correct solid angle */
820 greg 2.9 correct_solid_angle = 1;
821 greg 2.7 continue;
822 greg 2.16 case 'C': /* comment */
823     if (ncomm >= MAXCOMM) {
824     fprintf(stderr, "%s: too many comments\n",
825     argv[0]);
826     return 1;
827     }
828     if (strchr(argv[++i], '>') != NULL) {
829     fprintf(stderr, "%s: illegal character in comment\n",
830     argv[0]);
831     return 1;
832     }
833     commlist[ncomm++] = argv[i];
834     continue;
835 greg 2.1 case 't': /* transmission */
836     if (i >= argc-1)
837     UsageExit(argv[0]);
838     if (ndataf >= MAXFILES) {
839     fprintf(stderr, "%s: too many data files\n",
840     argv[0]);
841     return 1;
842     }
843     if (!strcmp(argv[i], "-tf"))
844     data_file[ndataf].type = DTtransForward;
845     else if (!strcmp(argv[i], "-tb"))
846     data_file[ndataf].type = DTtransBackward;
847     else
848     UsageExit(argv[0]);
849     if (!strcmp(argv[++i], "-")) {
850     if (used_stdin++) UsageExit(argv[i]);
851     argv[i] = (char *)stdin_name;
852     }
853     data_file[ndataf].fname = argv[i];
854     data_file[ndataf].spectrum = cur_spectrum;
855 greg 2.23 ndataf += check_new_data_file();
856 greg 2.1 continue;
857     case 'r': /* reflection */
858     if (i >= argc-1)
859     UsageExit(argv[0]);
860     if (ndataf >= MAXFILES) {
861     fprintf(stderr, "%s: too many data files\n",
862     argv[0]);
863     return 1;
864     }
865     if (!strcmp(argv[i], "-rf"))
866     data_file[ndataf].type = DTreflForward;
867     else if (!strcmp(argv[i], "-rb"))
868     data_file[ndataf].type = DTreflBackward;
869     else
870     UsageExit(argv[0]);
871     if (!strcmp(argv[++i], "-")) {
872     if (used_stdin++) UsageExit(argv[i]);
873     argv[i] = (char *)stdin_name;
874     }
875     data_file[ndataf].fname = argv[i];
876 greg 2.8 data_file[ndataf].spectrum = cur_spectrum;
877     ndataf++;
878 greg 2.1 continue;
879     case 's': /* spectrum name or input file */
880     if (++i >= argc)
881     UsageExit(argv[0]);
882     if (!strcasecmp(argv[i], "Solar"))
883     cur_spectrum = DSsolar;
884     else if (!strcasecmp(argv[i], "Visible") ||
885     !strcasecmp(argv[i], "CIE-Y"))
886     cur_spectrum = DSvisible;
887     else if (!strcasecmp(argv[i], "CIE-X"))
888     cur_spectrum = DSxbar31;
889     else if (!strcasecmp(argv[i], "CIE-Z"))
890     cur_spectrum = DSzbar31;
891 greg 2.11 else if (!strcasecmp(argv[i], "CIE-u"))
892     cur_spectrum = DSuprime;
893     else if (!strcasecmp(argv[i], "CIE-v"))
894     cur_spectrum = DSvprime;
895 greg 2.1 else if (!strcasecmp(argv[i], "NIR"))
896     cur_spectrum = DSnir;
897     else {
898     if (!strcmp(argv[i], "-")) {
899     fprintf(stderr,
900     "%s: cannot read spectra from stdin",
901     argv[0]);
902     return 1;
903     }
904     cur_spectrum = ncust_spec;
905     spectr_file[ncust_spec++] = argv[i];
906     }
907     continue;
908     case 'g': /* MGF geometry file */
909     if (i >= argc-1)
910     UsageExit(argv[0]);
911     if (mgf_geometry != NULL) {
912     fprintf(stderr, "%s: only one -g option allowed\n",
913     argv[0]);
914     return 1;
915     }
916     if (!strcmp(argv[++i], "-")) {
917     if (used_stdin++) UsageExit(argv[i]);
918     argv[i] = (char *)stdin_name;
919     }
920     mgf_geometry = argv[i];
921     continue;
922     case '\0': /* input XML from stdin */
923     break;
924     default:
925     UsageExit(argv[0]);
926     break;
927     }
928     break;
929     }
930     doneOptions: /* get XML input */
931     if (i >= argc) {
932     if (xml_input == NULL)
933     xml_input = def_template;
934     } else if ((i < argc-1) | (xml_input != NULL)) {
935     fprintf(stderr, "%s: only one XML input allowed\n", argv[0]);
936     UsageExit(argv[0]);
937     } else if (!strcmp(argv[i], "-")) {
938     if (used_stdin++) UsageExit(argv[0]);
939     xml_input = stdin_name;
940     } else {
941     xml_input = argv[i];
942     }
943     /* wrap it! */
944     return !wrapBSDF(argv[0]);
945     }