ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/gen/gensurf.c
Revision: 2.14
Committed: Tue Mar 23 15:18:09 2004 UTC (20 years, 1 month ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.13: +2 -4 lines
Log Message:
Fixed missing declaration for atof()

File Contents

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