ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/gen/gensurf.c
Revision: 2.32
Committed: Fri Apr 18 23:02:53 2025 UTC (3 weeks, 4 days ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.31: +3 -2 lines
Log Message:
perf(gensurf,genrev,genworm): Added call to doptimize() for expressions

File Contents

# Content
1 #ifndef lint
2 static const char RCSid[] = "$Id: gensurf.c,v 2.31 2025/01/21 01:56:27 greg Exp $";
3 #endif
4 /*
5 * gensurf.c - program to generate functional surfaces
6 *
7 * Parametric functions x(s,t), y(s,t) and z(s,t)
8 * specify the surface, which is tesselated into an m by n
9 * array of paired triangles.
10 * The surface normal is defined by the right hand
11 * rule applied to (s,t).
12 *
13 * 4/3/87
14 *
15 * 4/16/02 Added conditional vertex output
16 */
17
18 #include "standard.h"
19
20 #include "paths.h"
21 #include "resolu.h"
22 #include "rterror.h"
23 #include "calcomp.h"
24
25 char XNAME[] = "X`SYS"; /* x function name */
26 char YNAME[] = "Y`SYS"; /* y function name */
27 char ZNAME[] = "Z`SYS"; /* z function name */
28
29 char VNAME[] = "valid"; /* valid vertex name */
30
31 #define ABS(x) ((x)>=0 ? (x) : -(x))
32
33 #define ZEROVECT(v) (DOT(v,v) <= FTINY*FTINY)
34
35 #define pvect(p) printf(vformat, (p)[0], (p)[1], (p)[2])
36
37 char vformat[] = "%18.12g %18.12g %18.12g\n";
38 char tsargs[] = "4 surf_dx surf_dy surf_dz surf.cal";
39 char texname[] = "Phong";
40
41 int smooth = 0; /* apply smoothing? */
42 int objout = 0; /* output .OBJ format? */
43 int rev = 0; /* invert normal directions? */
44
45 char *modname, *surfname;
46
47 /* recorded data flags */
48 #define HASBORDER 01
49 #define TRIPLETS 02
50 /* a data structure */
51 struct {
52 int flags; /* data type */
53 short m, n; /* number of s and t values */
54 RREAL *data; /* the data itself, s major sort */
55 } datarec; /* our recorded data */
56
57 /* XXX this is redundant with rt/noise3.c, should go to a library */
58 double l_hermite(), l_bezier(), l_bspline(), l_dataval();
59
60 typedef struct {
61 int valid; /* point is valid (vertex number) */
62 int nvalid; /* normal is valid (normal number) */
63 FVECT p; /* vertex position */
64 FVECT n; /* average normal */
65 RREAL uv[2]; /* (u,v) position */
66 } POINT;
67
68 int nverts = 0; /* vertex output count */
69 int nnorms = 0; /* normal output count */
70
71 void loaddata(char *file, int m, int n, int pointsize);
72 double l_dataval(char *nam);
73 void putobjrow(POINT *rp, int n);
74 void putobjvert(POINT *p);
75 void putsquare(POINT *p0, POINT *p1, POINT *p2, POINT *p3);
76 void comprow(double s, POINT *row, int siz);
77 void compnorms(POINT *r0, POINT *r1, POINT *r2, int siz);
78 int norminterp(FVECT resmat[4], POINT *p0, POINT *p1, POINT *p2, POINT *p3);
79
80
81 int
82 main(int argc, char *argv[])
83 {
84 POINT *row0, *row1, *row2, *rp;
85 int i, j, m, n;
86 char stmp[256];
87
88 esupport |= E_VARIABLE|E_FUNCTION|E_RCONST;
89 esupport &= ~(E_OUTCHAN|E_INCHAN);
90 varset("PI", ':', PI);
91 funset("hermite", 5, ':', l_hermite);
92 funset("bezier", 5, ':', l_bezier);
93 funset("bspline", 5, ':', l_bspline);
94
95 if (argc < 8)
96 goto userror;
97
98 for (i = 8; i < argc; i++)
99 if (!strcmp(argv[i], "-e"))
100 scompile(argv[++i], NULL, 0);
101 else if (!strcmp(argv[i], "-f")) {
102 char *fpath = getpath(argv[++i], getrlibpath(), 0);
103 if (fpath == NULL) {
104 fprintf(stderr, "%s: cannot find file '%s'\n",
105 argv[0], argv[i]);
106 quit(1);
107 }
108 fcompile(fpath);
109 } else if (!strcmp(argv[i], "-s"))
110 smooth++;
111 else if (!strcmp(argv[i], "-o"))
112 objout++;
113 else if (!strcmp(argv[i], "-i"))
114 rev = 1;
115 else
116 goto userror;
117
118 modname = argv[1];
119 surfname = argv[2];
120 m = eval(argv[6]) + .5;
121 n = eval(argv[7]) + .5;
122 if (m <= 0 || n <= 0)
123 goto userror;
124 if (!strcmp(argv[5], "-") || access(argv[5], 4) == 0) { /* file? */
125 funset(ZNAME, 2, ':', l_dataval);
126 if (!strcmp(argv[5],argv[3]) && !strcmp(argv[5],argv[4])) {
127 loaddata(argv[5], m, n, 3);
128 funset(XNAME, 2, ':', l_dataval);
129 funset(YNAME, 2, ':', l_dataval);
130 } else {
131 loaddata(argv[5], m, n, 1);
132 sprintf(stmp, "%s(s,t)=%s;", XNAME, argv[3]);
133 scompile(stmp, NULL, 0);
134 sprintf(stmp, "%s(s,t)=%s;", YNAME, argv[4]);
135 scompile(stmp, NULL, 0);
136 }
137 } else {
138 sprintf(stmp, "%s(s,t)=%s;", XNAME, argv[3]);
139 scompile(stmp, NULL, 0);
140 sprintf(stmp, "%s(s,t)=%s;", YNAME, argv[4]);
141 scompile(stmp, NULL, 0);
142 sprintf(stmp, "%s(s,t)=%s;", ZNAME, argv[5]);
143 scompile(stmp, NULL, 0);
144 }
145 row0 = (POINT *)malloc((n+3)*sizeof(POINT));
146 row1 = (POINT *)malloc((n+3)*sizeof(POINT));
147 row2 = (POINT *)malloc((n+3)*sizeof(POINT));
148 if (row0 == NULL || row1 == NULL || row2 == NULL) {
149 fprintf(stderr, "%s: out of memory\n", argv[0]);
150 quit(1);
151 }
152 row0++; row1++; row2++;
153 /* print header */
154 fputs("# ", stdout);
155 printargs(argc, argv, stdout);
156 doptimize(1);
157 eclock++;
158 /* initialize */
159 comprow(-1.0/m, row0, n);
160 comprow(0.0, row1, n);
161 comprow(1.0/m, row2, n);
162 compnorms(row0, row1, row2, n);
163 if (objout) {
164 printf("\nusemtl %s\n\n", modname);
165 printf("o %s\n\n", surfname);
166 putobjrow(row1, n);
167 }
168 /* for each row */
169 for (i = 0; i < m; i++) {
170 /* compute next row */
171 rp = row0;
172 row0 = row1;
173 row1 = row2;
174 row2 = rp;
175 comprow((double)(i+2)/m, row2, n);
176 compnorms(row0, row1, row2, n);
177 if (objout)
178 putobjrow(row1, n);
179
180 for (j = 0; j < n; j++) {
181 int orient = (j & 1);
182 /* put polygons */
183 if (!(row0[j].valid && row1[j+1].valid))
184 orient = 1;
185 else if (!(row1[j].valid && row0[j+1].valid))
186 orient = 0;
187 if (orient)
188 putsquare(&row0[j], &row1[j],
189 &row0[j+1], &row1[j+1]);
190 else
191 putsquare(&row1[j], &row1[j+1],
192 &row0[j], &row0[j+1]);
193 }
194 }
195
196 return 0;
197
198 userror:
199 fprintf(stderr, "Usage: %s material name ", argv[0]);
200 fprintf(stderr, "x(s,t) y(s,t) z(s,t) m n [-s][-o][-e expr][-f file]\n");
201 return 1;
202 }
203
204
205 void
206 loaddata( /* load point data from file */
207 char *file,
208 int m,
209 int n,
210 int pointsize
211 )
212 {
213 FILE *fp;
214 char word[64];
215 int size;
216 RREAL *dp;
217
218 datarec.flags = HASBORDER; /* assume border values */
219 datarec.m = m+1;
220 datarec.n = n+1;
221 size = datarec.m*datarec.n*pointsize;
222 if (pointsize == 3)
223 datarec.flags |= TRIPLETS;
224 dp = (RREAL *)malloc(size*sizeof(RREAL));
225 if ((datarec.data = dp) == NULL) {
226 fputs("Out of memory\n", stderr);
227 exit(1);
228 }
229 if (!strcmp(file, "-")) {
230 file = "<stdin>";
231 fp = stdin;
232 } else if ((fp = fopen(file, "r")) == NULL) {
233 fputs(file, stderr);
234 fputs(": cannot open\n", stderr);
235 exit(1);
236 }
237 while (size > 0 && fgetword(word, sizeof(word), fp) != NULL) {
238 if (!isflt(word)) {
239 fprintf(stderr, "%s: garbled data value: %s\n",
240 file, word);
241 exit(1);
242 }
243 *dp++ = atof(word);
244 size--;
245 }
246 if (size == (m+n+1)*pointsize) { /* no border after all */
247 dp = (RREAL *)realloc(datarec.data,
248 m*n*pointsize*sizeof(RREAL));
249 if (dp != NULL)
250 datarec.data = dp;
251 datarec.flags &= ~HASBORDER;
252 datarec.m = m;
253 datarec.n = n;
254 size = 0;
255 }
256 if (datarec.m < 2 || datarec.n < 2 || size != 0 ||
257 fgetword(word, sizeof(word), fp) != NULL) {
258 fputs(file, stderr);
259 fputs(": bad number of data points\n", stderr);
260 exit(1);
261 }
262 fclose(fp);
263 }
264
265
266 double
267 l_dataval( /* return recorded data value */
268 char *nam
269 )
270 {
271 double u, v;
272 int i, j;
273 RREAL *dp;
274 double d00, d01, d10, d11;
275 /* compute coordinates */
276 u = argument(1); v = argument(2);
277 if (datarec.flags & HASBORDER) {
278 i = u *= datarec.m-1;
279 j = v *= datarec.n-1;
280 } else {
281 i = u = u*datarec.m - .5;
282 j = v = v*datarec.n - .5;
283 }
284 if (i < 0) i = 0;
285 else if (i > datarec.m-2) i = datarec.m-2;
286 if (j < 0) j = 0;
287 else if (j > datarec.n-2) j = datarec.n-2;
288 /* compute value */
289 if (datarec.flags & TRIPLETS) {
290 dp = datarec.data + 3*(j*datarec.m + i);
291 if (nam == ZNAME)
292 dp += 2;
293 else if (nam == YNAME)
294 dp++;
295 d00 = dp[0]; d01 = dp[3];
296 dp += 3*datarec.m;
297 d10 = dp[0]; d11 = dp[3];
298 } else {
299 dp = datarec.data + j*datarec.m + i;
300 d00 = dp[0]; d01 = dp[1];
301 dp += datarec.m;
302 d10 = dp[0]; d11 = dp[1];
303 }
304 /* bilinear interpolation */
305 return((j+1-v)*((i+1-u)*d00+(u-i)*d01)+(v-j)*((i+1-u)*d10+(u-i)*d11));
306 }
307
308
309 void
310 putobjrow( /* output vertex row to .OBJ */
311 POINT *rp,
312 int n
313 )
314 {
315 static FVECT prevNorm;
316
317 for ( ; n-- >= 0; rp++) {
318 if (!rp->valid)
319 continue;
320 fputs("v ", stdout);
321 pvect(rp->p);
322 rp->valid = ++nverts;
323 printf("\tvt %.9g %.9g\n", rp->uv[0], rp->uv[1]);
324 if (!smooth || ZEROVECT(rp->n))
325 rp->nvalid = 0;
326 else if (VABSEQ(rp->n, prevNorm))
327 rp->nvalid = nnorms;
328 else {
329 printf("\tvn %.9g %.9g %.9g\n",
330 rp->n[0], rp->n[1], rp->n[2]);
331 rp->nvalid = ++nnorms;
332 VCOPY(prevNorm, rp->n);
333 }
334 }
335 }
336
337
338 void
339 putobjvert( /* put out OBJ vertex index triplet */
340 POINT *p
341 )
342 {
343 int pti = p->valid ? p->valid-nverts-1 : 0;
344 int ni = p->nvalid ? p->nvalid-nnorms-1 : 0;
345
346 printf(" %d/%d/%d", pti, pti, ni);
347 }
348
349
350 void
351 putsquare( /* put out a square */
352 POINT *p0,
353 POINT *p1,
354 POINT *p2,
355 POINT *p3
356 )
357 {
358 static int nout = 0;
359 FVECT norm[4];
360 int axis;
361 FVECT v1, v2, vc1, vc2;
362 int ok1, ok2;
363
364 if (rev) { /* reverse normals? */
365 POINT *pt = p1; p1 = p2; p2 = pt;
366 }
367 /* compute exact normals */
368 ok1 = (p0->valid && p1->valid && p2->valid);
369 if (ok1) {
370 VSUB(v1, p1->p, p0->p);
371 VSUB(v2, p2->p, p0->p);
372 fcross(vc1, v1, v2);
373 ok1 = (normalize(vc1) != 0.0);
374 }
375 ok2 = (p1->valid && p2->valid && p3->valid);
376 if (ok2) {
377 VSUB(v1, p2->p, p3->p);
378 VSUB(v2, p1->p, p3->p);
379 fcross(vc2, v1, v2);
380 ok2 = (normalize(vc2) != 0.0);
381 }
382 if (!(ok1 | ok2))
383 return;
384 if (objout) { /* output .OBJ faces */
385 if (ok1 & ok2 && fdot(vc1,vc2) >= 1.0-FTINY*FTINY) {
386 putc('f', stdout);
387 putobjvert(p0); putobjvert(p1);
388 putobjvert(p3); putobjvert(p2);
389 putc('\n', stdout);
390 return;
391 }
392 if (ok1) {
393 putc('f', stdout);
394 putobjvert(p0); putobjvert(p1); putobjvert(p2);
395 putc('\n', stdout);
396 }
397 if (ok2) {
398 putc('f', stdout);
399 putobjvert(p2); putobjvert(p1); putobjvert(p3);
400 putc('\n', stdout);
401 }
402 return;
403 }
404 /* compute normal interpolation */
405 axis = norminterp(norm, p0, p1, p2, p3);
406
407 /* put out quadrilateral? */
408 if (ok1 & ok2 && fdot(vc1,vc2) >= 1.0-FTINY*FTINY) {
409 printf("\n%s ", modname);
410 if (axis != -1) {
411 printf("texfunc %s\n%s\n", texname, tsargs);
412 printf("0\n13\t%d\n", axis);
413 pvect(norm[0]);
414 pvect(norm[1]);
415 pvect(norm[2]);
416 fvsum(v1, norm[3], vc1, -0.5);
417 fvsum(v1, v1, vc2, -0.5);
418 pvect(v1);
419 printf("\n%s ", texname);
420 }
421 printf("polygon %s.%d\n", surfname, ++nout);
422 printf("0\n0\n12\n");
423 pvect(p0->p);
424 pvect(p1->p);
425 pvect(p3->p);
426 pvect(p2->p);
427 return;
428 }
429 /* put out triangles? */
430 if (ok1) {
431 printf("\n%s ", modname);
432 if (axis != -1) {
433 printf("texfunc %s\n%s\n", texname, tsargs);
434 printf("0\n13\t%d\n", axis);
435 pvect(norm[0]);
436 pvect(norm[1]);
437 pvect(norm[2]);
438 fvsum(v1, norm[3], vc1, -1.0);
439 pvect(v1);
440 printf("\n%s ", texname);
441 }
442 printf("polygon %s.%d\n", surfname, ++nout);
443 printf("0\n0\n9\n");
444 pvect(p0->p);
445 pvect(p1->p);
446 pvect(p2->p);
447 }
448 if (ok2) {
449 printf("\n%s ", modname);
450 if (axis != -1) {
451 printf("texfunc %s\n%s\n", texname, tsargs);
452 printf("0\n13\t%d\n", axis);
453 pvect(norm[0]);
454 pvect(norm[1]);
455 pvect(norm[2]);
456 fvsum(v2, norm[3], vc2, -1.0);
457 pvect(v2);
458 printf("\n%s ", texname);
459 }
460 printf("polygon %s.%d\n", surfname, ++nout);
461 printf("0\n0\n9\n");
462 pvect(p2->p);
463 pvect(p1->p);
464 pvect(p3->p);
465 }
466 }
467
468
469 void
470 comprow( /* compute row of values */
471 double s,
472 POINT *row,
473 int siz
474 )
475 {
476 double st[2];
477 int end;
478 int checkvalid;
479 int i;
480
481 if (smooth) {
482 i = -1; /* compute one past each end */
483 end = siz+1;
484 } else {
485 if (s < -FTINY || s > 1.0+FTINY)
486 return;
487 i = 0;
488 end = siz;
489 }
490 st[0] = s;
491 checkvalid = (fundefined(VNAME) == 2);
492 while (i <= end) {
493 st[1] = (double)i/siz;
494 if (checkvalid && funvalue(VNAME, 2, st) <= 0.0) {
495 row[i].valid = 0;
496 row[i].p[0] = row[i].p[1] = row[i].p[2] = 0.0;
497 row[i].uv[0] = row[i].uv[1] = 0.0;
498 } else {
499 row[i].valid = 1;
500 row[i].p[0] = funvalue(XNAME, 2, st);
501 row[i].p[1] = funvalue(YNAME, 2, st);
502 row[i].p[2] = funvalue(ZNAME, 2, st);
503 row[i].uv[0] = st[0];
504 row[i].uv[1] = st[1];
505 }
506 i++;
507 }
508 }
509
510
511 void
512 compnorms( /* compute row of averaged normals */
513 POINT *r0,
514 POINT *r1,
515 POINT *r2,
516 int siz
517 )
518 {
519 FVECT v1, v2;
520
521 if (!smooth) /* not needed if no smoothing */
522 return;
523 /* compute row 1 normals */
524 while (siz-- >= 0) {
525 if (!r1[0].valid)
526 goto skip;
527 if (!r0[0].valid) {
528 if (!r2[0].valid) {
529 r1[0].n[0] = r1[0].n[1] = r1[0].n[2] = 0.0;
530 goto skip;
531 }
532 fvsum(v1, r2[0].p, r1[0].p, -1.0);
533 } else if (!r2[0].valid)
534 fvsum(v1, r1[0].p, r0[0].p, -1.0);
535 else
536 fvsum(v1, r2[0].p, r0[0].p, -1.0);
537 if (!r1[-1].valid) {
538 if (!r1[1].valid) {
539 r1[0].n[0] = r1[0].n[1] = r1[0].n[2] = 0.0;
540 goto skip;
541 }
542 fvsum(v2, r1[1].p, r1[0].p, -1.0);
543 } else if (!r1[1].valid)
544 fvsum(v2, r1[0].p, r1[-1].p, -1.0);
545 else
546 fvsum(v2, r1[1].p, r1[-1].p, -1.0);
547 if (rev)
548 fcross(r1[0].n, v2, v1);
549 else
550 fcross(r1[0].n, v1, v2);
551 normalize(r1[0].n);
552 skip:
553 r0++; r1++; r2++;
554 }
555 }
556
557
558 int
559 norminterp( /* compute normal interpolation */
560 FVECT resmat[4],
561 POINT *p0,
562 POINT *p1,
563 POINT *p2,
564 POINT *p3
565 )
566 {
567 #define u ((ax+1)%3)
568 #define v ((ax+2)%3)
569
570 int ax;
571 MAT4 eqnmat;
572 FVECT v1;
573 int i, j;
574
575 if (!smooth) /* no interpolation if no smoothing */
576 return(-1);
577 /* find dominant axis */
578 VCOPY(v1, p0->n);
579 fvsum(v1, v1, p1->n, 1.0);
580 fvsum(v1, v1, p2->n, 1.0);
581 fvsum(v1, v1, p3->n, 1.0);
582 ax = ABS(v1[0]) > ABS(v1[1]) ? 0 : 1;
583 ax = ABS(v1[ax]) > ABS(v1[2]) ? ax : 2;
584 /* assign equation matrix */
585 eqnmat[0][0] = p0->p[u]*p0->p[v];
586 eqnmat[0][1] = p0->p[u];
587 eqnmat[0][2] = p0->p[v];
588 eqnmat[0][3] = 1.0;
589 eqnmat[1][0] = p1->p[u]*p1->p[v];
590 eqnmat[1][1] = p1->p[u];
591 eqnmat[1][2] = p1->p[v];
592 eqnmat[1][3] = 1.0;
593 eqnmat[2][0] = p2->p[u]*p2->p[v];
594 eqnmat[2][1] = p2->p[u];
595 eqnmat[2][2] = p2->p[v];
596 eqnmat[2][3] = 1.0;
597 eqnmat[3][0] = p3->p[u]*p3->p[v];
598 eqnmat[3][1] = p3->p[u];
599 eqnmat[3][2] = p3->p[v];
600 eqnmat[3][3] = 1.0;
601 /* invert matrix (solve system) */
602 if (!invmat4(eqnmat, eqnmat))
603 return(-1); /* no solution */
604 /* compute result matrix */
605 for (j = 0; j < 4; j++)
606 for (i = 0; i < 3; i++)
607 resmat[j][i] = eqnmat[j][0]*p0->n[i] +
608 eqnmat[j][1]*p1->n[i] +
609 eqnmat[j][2]*p2->n[i] +
610 eqnmat[j][3]*p3->n[i];
611 return(ax);
612
613 #undef u
614 #undef v
615 }
616
617
618 double
619 l_hermite(char *nm)
620 {
621 double t;
622
623 t = argument(5);
624 return( argument(1)*((2.0*t-3.0)*t*t+1.0) +
625 argument(2)*(-2.0*t+3.0)*t*t +
626 argument(3)*((t-2.0)*t+1.0)*t +
627 argument(4)*(t-1.0)*t*t );
628 }
629
630
631 double
632 l_bezier(char *nm)
633 {
634 double t;
635
636 t = argument(5);
637 return( argument(1) * (1.+t*(-3.+t*(3.-t))) +
638 argument(2) * 3.*t*(1.+t*(-2.+t)) +
639 argument(3) * 3.*t*t*(1.-t) +
640 argument(4) * t*t*t );
641 }
642
643
644 double
645 l_bspline(char *nm)
646 {
647 double t;
648
649 t = argument(5);
650 return( argument(1) * (1./6.+t*(-1./2.+t*(1./2.-1./6.*t))) +
651 argument(2) * (2./3.+t*t*(-1.+1./2.*t)) +
652 argument(3) * (1./6.+t*(1./2.+t*(1./2.-1./2.*t))) +
653 argument(4) * (1./6.*t*t*t) );
654 }