src/common/testBSDF.c

#ifndef lint
static const char RCSid[] = "$Id: testBSDF.c,v 1.13 2018/05/10 01:41:33 greg Exp $";
#endif
/*
 * Simple test program to demonstrate BSDF operation.
 *
 *      G. Ward         June 2015
 */

#define _USE_MATH_DEFINES
#include <stdio.h>
#include <stdlib.h>
#include <math.h>
#include <ctype.h>
#include "rtio.h"
#include "bsdf.h"

static void
Usage(const char *prog)
{
        printf("Usage: %s [bsdf_directory]\n", prog);
        printf("Input commands:\n");
        printf("  L bsdf.xml\t\t\t Load (make active) given BSDF input file\n");
        printf("  i\t\t\t\t Report general information (metadata)\n");
        printf("  c\t\t\t\t Report diffuse and specular components\n");
        printf("  q theta_i phi_i theta_o phi_o\t Query BSDF for given path (CIE-XYZ)\n");
        printf("  s[r|t][s|d] N theta phi\t Generate N ray directions & colors at given incidence\n");
        printf("  h[s|d] theta phi\t\t Report hemispherical scattering at given incidence\n");
        printf("  r[s|d] theta phi\t\t Report hemispherical reflection at given incidence\n");
        printf("  t[s|d] theta phi\t\t Report hemispherical transmission at given incidence\n");
        printf("  a theta phi [t2 p2]\t\t Report resolution (in proj. steradians) for given direction(s)\n");
        printf("  ^D\t\t\t\t Quit program\n");
}

static void
vec_from_deg(FVECT v, double theta, double phi)
{
        const double    DEG = M_PI/180.;

        theta *= DEG; phi *= DEG;
        v[0] = v[1] = sin(theta);
        v[0] *= cos(phi);
        v[1] *= sin(phi);
        v[2] = cos(theta);
}

static void
printXYZ(const char *intro, const SDValue *vp)
{
        if (vp->cieY <= 1e-9) {
                printf("%s0 0 0\n", intro);
                return;
        }
        printf("%s%.3e %.3e %.3e\n", intro,
                        vp->spec.cx/vp->spec.cy*vp->cieY,
                        vp->cieY,
                        (1.-vp->spec.cx-vp->spec.cy)/
                        vp->spec.cy*vp->cieY);
}

int
main(int argc, char *argv[])
{
        const char      *directory = NULL;
        char            inp[512], path[512];
        const SDData    *bsdf = NULL;

        if (argc > 2 || (argc == 2 && argv[1][0] == '-')) {
                Usage(argv[0]);
                return 1;
        }
        if (argc == 2)
                directory = argv[1];
        
        SDretainSet = SDretainBSDFs;            /* keep BSDFs in memory */

                                                /* loop on command */
        while (fgets(inp, sizeof(inp), stdin)) {
                int     sflags = SDsampAll;
                char    *cp = inp;
                char    *cp2;
                FVECT   vin, vout;
                double  proja[2];
                int     n, i;
                SDValue val;
                
                while (isspace(*cp)) cp++;

                switch (toupper(*cp)) {
                case 'L':                       /* load/activate BSDF input */
                        cp2 = cp = sskip2(cp, 1);
                        if (!*cp)
                                break;
                        while (*cp) cp++;
                        while (isspace(*--cp)) *cp = '\0';
                        if (directory)
                                sprintf(path, "%s/%s", directory, cp2);
                        else
                                strcpy(path, cp2);
                        if (bsdf)
                                SDfreeCache(bsdf);
                        bsdf = SDcacheFile(path);
                        continue;
                case 'I':                       /* report general info. */
                        if (!bsdf)
                                goto noBSDFerr;
                        printf("Material: '%s'\n", bsdf->matn);
                        printf("Manufacturer: '%s'\n", bsdf->makr);
                        printf("Width, Height, Thickness (m): %.4e, %.4e, %.4e\n",
                                        bsdf->dim[0], bsdf->dim[1], bsdf->dim[2]);
                        if (bsdf->mgf)
                                printf("Has geometry: %lu bytes\n",
                                                (unsigned long)strlen(bsdf->mgf));
                        else
                                printf("Has geometry: no\n");
                        break;
                case 'C':                       /* report constant values */
                        if (!bsdf)
                                goto noBSDFerr;
                        if (bsdf->rf)
                                printf("Peak front hemispherical reflectance: %.3e\n",
                                                bsdf->rLambFront.cieY +
                                                bsdf->rf->maxHemi);
                        if (bsdf->rb)
                                printf("Peak back hemispherical reflectance: %.3e\n",
                                                bsdf->rLambBack.cieY +
                                                bsdf->rb->maxHemi);
                        if (bsdf->tf)
                                printf("Peak front hemispherical transmittance: %.3e\n",
                                                bsdf->tLamb.cieY + bsdf->tf->maxHemi);
                        if (bsdf->tb)
                                printf("Peak back hemispherical transmittance: %.3e\n",
                                                bsdf->tLamb.cieY + bsdf->tb->maxHemi);
                        printXYZ("Diffuse Front Reflectance: ", &bsdf->rLambFront);
                        printXYZ("Diffuse Back Reflectance: ", &bsdf->rLambBack);
                        printXYZ("Diffuse Transmittance: ", &bsdf->tLamb);
                        break;
                case 'Q':                       /* query BSDF value */
                        if (!bsdf)
                                goto noBSDFerr;
                        if (!*sskip2(cp,4))
                                break;
                        vec_from_deg(vin, atof(sskip2(cp,1)), atof(sskip2(cp,2)));
                        vec_from_deg(vout, atof(sskip2(cp,3)), atof(sskip2(cp,4)));
                        if (!SDreportError(SDevalBSDF(&val, vout, vin, bsdf), stderr))
                                printXYZ("", &val);
                        break;
                case 'S':                       /* sample BSDF */
                        if (!bsdf)
                                goto noBSDFerr;
                        if (!*sskip2(cp,3))
                                break;
                        if (toupper(cp[1]) == 'R') {
                                sflags &= ~SDsampT;
                                ++cp;
                        } else if (toupper(cp[1]) == 'T') {
                                sflags &= ~SDsampR;
                                ++cp;
                        }
                        if (toupper(cp[1]) == 'S')
                                sflags &= ~SDsampDf;
                        else if (toupper(cp[1]) == 'D')
                                sflags &= ~SDsampSp;
                        i = n = atoi(sskip2(cp,1));
                        vec_from_deg(vin, atof(sskip2(cp,2)), atof(sskip2(cp,3)));
                        while (i-- > 0) {
                                VCOPY(vout, vin);
                                if (SDreportError(SDsampBSDF(&val, vout,
                                                (i+rand()*(1./(RAND_MAX+.5)))/(double)n,
                                                sflags, bsdf), stderr))
                                        break;
                                printf("%.8f %.8f %.8f ", vout[0], vout[1], vout[2]);
                                printXYZ("", &val);
                        }
                        break;
                case 'H':                       /* hemispherical values */
                case 'R':
                case 'T':
                        if (!bsdf)
                                goto noBSDFerr;
                        if (!*sskip2(cp,2))
                                break;
                        if (toupper(cp[0]) == 'R')
                                sflags &= ~SDsampT;
                        else if (toupper(cp[0]) == 'T')
                                sflags &= ~SDsampR;
                        if (toupper(cp[1]) == 'S')
                                sflags &= ~SDsampDf;
                        else if (toupper(cp[1]) == 'D')
                                sflags &= ~SDsampSp;
                        vec_from_deg(vin, atof(sskip2(cp,1)), atof(sskip2(cp,2)));
                        printf("%.4e\n", SDdirectHemi(vin, sflags, bsdf));
                        break;
                case 'A':                       /* resolution in proj. steradians */
                        if (!bsdf)
                                goto noBSDFerr;
                        if (!*sskip2(cp,2))
                                break;
                        vec_from_deg(vin, atof(sskip2(cp,1)), atof(sskip2(cp,2)));
                        if (*sskip2(cp,4)) {
                                vec_from_deg(vout, atof(sskip2(cp,3)), atof(sskip2(cp,4)));
                                if (SDreportError(SDsizeBSDF(proja, vout, vin,
                                                SDqueryMin+SDqueryMax, bsdf), stderr))
                                        continue;
                        } else if (SDreportError(SDsizeBSDF(proja, vin, NULL,
                                                SDqueryMin+SDqueryMax, bsdf), stderr))
                                        continue;
                        printf("%.4e %.4e\n", proja[0], proja[1]);
                        break;
                default:
                        Usage(argv[0]);
                        break;
                }
                fflush(stdout);                 /* in case we're on remote */
                continue;
noBSDFerr:
                fprintf(stderr, "%s: First, use 'L' command to load BSDF\n", argv[0]);
        }
        return 0;
}
Revision:	1.14
Committed:	Tue Jun 11 14:26:29 2019 UTC (6 years, 4 months ago) by greg
Content type:	text/plain
Branch:	MAIN
Changes since 1.13:	+11 -8 lines
Log Message:	Added output flushing for remote operation
#	User	Rev	Content
1	greg	1.1	#ifndef lint
2	greg	1.14	static const char RCSid[] = "$Id: testBSDF.c,v 1.13 2018/05/10 01:41:33 greg Exp $";
3	greg	1.1	#endif
4			/*
5			* Simple test program to demonstrate BSDF operation.
6			*
7			* G. Ward June 2015
8			*/
9
10			#define _USE_MATH_DEFINES
11			#include <stdio.h>
12			#include <stdlib.h>
13			#include <math.h>
14			#include <ctype.h>
15			#include "rtio.h"
16			#include "bsdf.h"
17
18			static void
19			Usage(const char *prog)
20			{
21			printf("Usage: %s [bsdf_directory]\n", prog);
22			printf("Input commands:\n");
23	greg	1.5	printf(" L bsdf.xml\t\t\t Load (make active) given BSDF input file\n");
24	greg	1.6	printf(" i\t\t\t\t Report general information (metadata)\n");
25			printf(" c\t\t\t\t Report diffuse and specular components\n");
26	greg	1.5	printf(" q theta_i phi_i theta_o phi_o\t Query BSDF for given path (CIE-XYZ)\n");
27	greg	1.10	printf(" s[r\|t][s\|d] N theta phi\t Generate N ray directions & colors at given incidence\n");
28			printf(" h[s\|d] theta phi\t\t Report hemispherical scattering at given incidence\n");
29			printf(" r[s\|d] theta phi\t\t Report hemispherical reflection at given incidence\n");
30			printf(" t[s\|d] theta phi\t\t Report hemispherical transmission at given incidence\n");
31	greg	1.1	printf(" a theta phi [t2 p2]\t\t Report resolution (in proj. steradians) for given direction(s)\n");
32			printf(" ^D\t\t\t\t Quit program\n");
33			}
34
35			static void
36			vec_from_deg(FVECT v, double theta, double phi)
37			{
38			const double DEG = M_PI/180.;
39
40			theta = DEG; phi = DEG;
41			v[0] = v[1] = sin(theta);
42			v[0] *= cos(phi);
43			v[1] *= sin(phi);
44			v[2] = cos(theta);
45			}
46
47	greg	1.6	static void
48			printXYZ(const char intro, const SDValue vp)
49			{
50	greg	1.11	if (vp->cieY <= 1e-9) {
51			printf("%s0 0 0\n", intro);
52			return;
53			}
54	greg	1.6	printf("%s%.3e %.3e %.3e\n", intro,
55			vp->spec.cx/vp->spec.cy*vp->cieY,
56			vp->cieY,
57			(1.-vp->spec.cx-vp->spec.cy)/
58			vp->spec.cy*vp->cieY);
59			}
60
61	greg	1.1	int
62			main(int argc, char *argv[])
63			{
64	greg	1.3	const char *directory = NULL;
65	greg	1.1	char inp[512], path[512];
66			const SDData *bsdf = NULL;
67
68			if (argc > 2 \|\| (argc == 2 && argv[1][0] == '-')) {
69			Usage(argv[0]);
70			return 1;
71			}
72			if (argc == 2)
73			directory = argv[1];
74
75	greg	1.3	SDretainSet = SDretainBSDFs; /* keep BSDFs in memory */
76	greg	1.1
77			/* loop on command */
78	greg	1.8	while (fgets(inp, sizeof(inp), stdin)) {
79	greg	1.1	int sflags = SDsampAll;
80			char *cp = inp;
81			char *cp2;
82			FVECT vin, vout;
83			double proja[2];
84	greg	1.9	int n, i;
85	greg	1.1	SDValue val;
86
87			while (isspace(*cp)) cp++;
88
89	greg	1.4	switch (toupper(*cp)) {
90	greg	1.1	case 'L': /* load/activate BSDF input */
91			cp2 = cp = sskip2(cp, 1);
92			if (!*cp)
93			break;
94			while (*cp) cp++;
95			while (isspace(--cp)) cp = '\0';
96	greg	1.3	if (directory)
97	greg	1.1	sprintf(path, "%s/%s", directory, cp2);
98			else
99			strcpy(path, cp2);
100	greg	1.3	if (bsdf)
101	greg	1.2	SDfreeCache(bsdf);
102	greg	1.1	bsdf = SDcacheFile(path);
103			continue;
104	greg	1.6	case 'I': /* report general info. */
105	greg	1.8	if (!bsdf)
106	greg	1.6	goto noBSDFerr;
107			printf("Material: '%s'\n", bsdf->matn);
108			printf("Manufacturer: '%s'\n", bsdf->makr);
109			printf("Width, Height, Thickness (m): %.4e, %.4e, %.4e\n",
110			bsdf->dim[0], bsdf->dim[1], bsdf->dim[2]);
111	greg	1.13	if (bsdf->mgf)
112			printf("Has geometry: %lu bytes\n",
113			(unsigned long)strlen(bsdf->mgf));
114			else
115			printf("Has geometry: no\n");
116	greg	1.14	break;
117	greg	1.6	case 'C': /* report constant values */
118	greg	1.8	if (!bsdf)
119	greg	1.6	goto noBSDFerr;
120	greg	1.8	if (bsdf->rf)
121	greg	1.6	printf("Peak front hemispherical reflectance: %.3e\n",
122			bsdf->rLambFront.cieY +
123			bsdf->rf->maxHemi);
124	greg	1.8	if (bsdf->rb)
125	greg	1.6	printf("Peak back hemispherical reflectance: %.3e\n",
126			bsdf->rLambBack.cieY +
127			bsdf->rb->maxHemi);
128	greg	1.8	if (bsdf->tf)
129	greg	1.6	printf("Peak front hemispherical transmittance: %.3e\n",
130			bsdf->tLamb.cieY + bsdf->tf->maxHemi);
131	greg	1.8	if (bsdf->tb)
132	greg	1.6	printf("Peak back hemispherical transmittance: %.3e\n",
133			bsdf->tLamb.cieY + bsdf->tb->maxHemi);
134			printXYZ("Diffuse Front Reflectance: ", &bsdf->rLambFront);
135			printXYZ("Diffuse Back Reflectance: ", &bsdf->rLambBack);
136			printXYZ("Diffuse Transmittance: ", &bsdf->tLamb);
137	greg	1.14	break;
138	greg	1.1	case 'Q': /* query BSDF value */
139	greg	1.8	if (!bsdf)
140	greg	1.1	goto noBSDFerr;
141			if (!*sskip2(cp,4))
142			break;
143			vec_from_deg(vin, atof(sskip2(cp,1)), atof(sskip2(cp,2)));
144			vec_from_deg(vout, atof(sskip2(cp,3)), atof(sskip2(cp,4)));
145	greg	1.6	if (!SDreportError(SDevalBSDF(&val, vout, vin, bsdf), stderr))
146			printXYZ("", &val);
147	greg	1.14	break;
148	greg	1.1	case 'S': /* sample BSDF */
149	greg	1.8	if (!bsdf)
150	greg	1.1	goto noBSDFerr;
151			if (!*sskip2(cp,3))
152			break;
153	greg	1.9	if (toupper(cp[1]) == 'R') {
154			sflags &= ~SDsampT;
155			++cp;
156			} else if (toupper(cp[1]) == 'T') {
157			sflags &= ~SDsampR;
158			++cp;
159			}
160			if (toupper(cp[1]) == 'S')
161			sflags &= ~SDsampDf;
162			else if (toupper(cp[1]) == 'D')
163			sflags &= ~SDsampSp;
164			i = n = atoi(sskip2(cp,1));
165	greg	1.1	vec_from_deg(vin, atof(sskip2(cp,2)), atof(sskip2(cp,3)));
166	greg	1.9	while (i-- > 0) {
167			VCOPY(vout, vin);
168			if (SDreportError(SDsampBSDF(&val, vout,
169			(i+rand()*(1./(RAND_MAX+.5)))/(double)n,
170	greg	1.1	sflags, bsdf), stderr))
171			break;
172	greg	1.9	printf("%.8f %.8f %.8f ", vout[0], vout[1], vout[2]);
173			printXYZ("", &val);
174	greg	1.1	}
175	greg	1.14	break;
176	greg	1.9	case 'H': /* hemispherical values */
177	greg	1.1	case 'R':
178			case 'T':
179	greg	1.8	if (!bsdf)
180	greg	1.1	goto noBSDFerr;
181			if (!*sskip2(cp,2))
182			break;
183	greg	1.9	if (toupper(cp[0]) == 'R')
184	greg	1.3	sflags &= ~SDsampT;
185	greg	1.9	else if (toupper(cp[0]) == 'T')
186	greg	1.3	sflags &= ~SDsampR;
187	greg	1.9	if (toupper(cp[1]) == 'S')
188			sflags &= ~SDsampDf;
189			else if (toupper(cp[1]) == 'D')
190			sflags &= ~SDsampSp;
191	greg	1.1	vec_from_deg(vin, atof(sskip2(cp,1)), atof(sskip2(cp,2)));
192			printf("%.4e\n", SDdirectHemi(vin, sflags, bsdf));
193	greg	1.14	break;
194	greg	1.3	case 'A': /* resolution in proj. steradians */
195	greg	1.8	if (!bsdf)
196	greg	1.1	goto noBSDFerr;
197			if (!*sskip2(cp,2))
198			break;
199			vec_from_deg(vin, atof(sskip2(cp,1)), atof(sskip2(cp,2)));
200			if (*sskip2(cp,4)) {
201			vec_from_deg(vout, atof(sskip2(cp,3)), atof(sskip2(cp,4)));
202	greg	1.12	if (SDreportError(SDsizeBSDF(proja, vout, vin,
203	greg	1.1	SDqueryMin+SDqueryMax, bsdf), stderr))
204			continue;
205			} else if (SDreportError(SDsizeBSDF(proja, vin, NULL,
206			SDqueryMin+SDqueryMax, bsdf), stderr))
207			continue;
208			printf("%.4e %.4e\n", proja[0], proja[1]);
209	greg	1.14	break;
210			default:
211			Usage(argv[0]);
212			break;
213	greg	1.1	}
214	greg	1.14	fflush(stdout); /* in case we're on remote */
215	greg	1.1	continue;
216			noBSDFerr:
217	greg	1.6	fprintf(stderr, "%s: First, use 'L' command to load BSDF\n", argv[0]);
218	greg	1.1	}
219			return 0;
220			}