ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/util/wrapBSDF.c
Revision: 2.22
Committed: Tue Feb 14 19:58:37 2017 UTC (7 years, 1 month ago) by greg
Content type: text/plain
Branch: MAIN
CVS Tags: rad5R2, rad5R1
Changes since 2.21: +3 -2 lines
Log Message:
Added "d" nickname for device type

File Contents

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