src/gen/gensurf.c

#ifndef lint
static const char RCSid[] = "$Id: gensurf.c,v 2.32 2025/04/18 23:02:53 greg Exp $";
#endif
/*
 *  gensurf.c - program to generate functional surfaces
 *
 *      Parametric functions x(s,t), y(s,t) and z(s,t)
 *  specify the surface, which is tesselated into an m by n
 *  array of paired triangles.
 *      The surface normal is defined by the right hand
 *  rule applied to (s,t).
 *
 *      4/3/87
 *
 *      4/16/02 Added conditional vertex output
 */

#include  "standard.h"

#include  "paths.h"
#include  "resolu.h"
#include  "rterror.h"
#include  "calcomp.h"

char  XNAME[] =         "X`SYS";                /* x function name */
char  YNAME[] =         "Y`SYS";                /* y function name */
char  ZNAME[] =         "Z`SYS";                /* z function name */

char  VNAME[] =         "valid";                /* valid vertex name */

#define  ABS(x)         ((x)>=0 ? (x) : -(x))

#define  ZEROVECT(v)    (DOT(v,v) <= FTINY*FTINY)

#define  pvect(p)       printf(vformat, (p)[0], (p)[1], (p)[2])

char  vformat[] = "%18.12g %18.12g %18.12g\n";
char  tsargs[] = "4 surf_dx surf_dy surf_dz surf.cal";
char  texname[] = "Phong";

int  smooth = 0;                /* apply smoothing? */
int  objout = 0;                /* output .OBJ format? */
int  rev = 0;                   /* invert normal directions? */

char  *modname, *surfname;

                                /* recorded data flags */
#define  HASBORDER      01
#define  TRIPLETS       02
                                /* a data structure */
struct {
        int     flags;                  /* data type */
        short   m, n;                   /* number of s and t values */
        RREAL   *data;                  /* the data itself, s major sort */
} datarec;                      /* our recorded data */

/* XXX this is redundant with rt/noise3.c, should go to a library */
double  l_hermite(char *), l_bezier(char *),
                l_bspline(char *), l_dataval(char *);

typedef struct {
        int  valid;     /* point is valid (vertex number) */
        int  nvalid;    /* normal is valid (normal number) */
        FVECT  p;       /* vertex position */
        FVECT  n;       /* average normal */
        RREAL  uv[2];   /* (u,v) position */
} POINT;

int  nverts = 0;                /* vertex output count */
int  nnorms = 0;                /* normal output count */

void loaddata(char *file, int m, int n, int pointsize);
double l_dataval(char *nam);
void putobjrow(POINT *rp, int n);
void putobjvert(POINT *p);
void putsquare(POINT *p0, POINT *p1, POINT *p2, POINT *p3);
void comprow(double s, POINT *row, int siz);
void compnorms(POINT *r0, POINT *r1, POINT *r2, int siz);
int norminterp(FVECT resmat[4], POINT *p0, POINT *p1, POINT *p2, POINT *p3);


int
main(int argc, char *argv[])
{
        POINT  *row0, *row1, *row2, *rp;
        int  i, j, m, n;
        char  stmp[256];

        esupport |= E_VARIABLE|E_FUNCTION|E_RCONST;
        esupport &= ~(E_OUTCHAN|E_INCHAN);
        varset("PI", ':', PI);
        funset("hermite", 5, ':', l_hermite);
        funset("bezier", 5, ':', l_bezier);
        funset("bspline", 5, ':', l_bspline);

        if (argc < 8)
                goto userror;

        for (i = 8; i < argc; i++)
                if (!strcmp(argv[i], "-e"))
                        scompile(argv[++i], NULL, 0);
                else if (!strcmp(argv[i], "-f")) {
                        char  *fpath = getpath(argv[++i], getrlibpath(), 0);
                        if (fpath == NULL) {
                                fprintf(stderr, "%s: cannot find file '%s'\n",
                                                argv[0], argv[i]);
                                quit(1);
                        }
                        fcompile(fpath);
                } else if (!strcmp(argv[i], "-s"))
                        smooth++;
                else if (!strcmp(argv[i], "-o"))
                        objout++;
                else if (!strcmp(argv[i], "-i"))
                        rev = 1;
                else
                        goto userror;

        modname = argv[1];
        surfname = argv[2];
        m = eval(argv[6]) + .5;
        n = eval(argv[7]) + .5;
        if (m <= 0 || n <= 0)
                goto userror;
        if (!strcmp(argv[5], "-") || access(argv[5], 4) == 0) { /* file? */
                funset(ZNAME, 2, ':', l_dataval);
                if (!strcmp(argv[5],argv[3]) && !strcmp(argv[5],argv[4])) {
                        loaddata(argv[5], m, n, 3);
                        funset(XNAME, 2, ':', l_dataval);
                        funset(YNAME, 2, ':', l_dataval);
                } else {
                        loaddata(argv[5], m, n, 1);
                        sprintf(stmp, "%s(s,t)=%s;", XNAME, argv[3]);
                        scompile(stmp, NULL, 0);
                        sprintf(stmp, "%s(s,t)=%s;", YNAME, argv[4]);
                        scompile(stmp, NULL, 0);
                }
        } else {
                sprintf(stmp, "%s(s,t)=%s;", XNAME, argv[3]);
                scompile(stmp, NULL, 0);
                sprintf(stmp, "%s(s,t)=%s;", YNAME, argv[4]);
                scompile(stmp, NULL, 0);
                sprintf(stmp, "%s(s,t)=%s;", ZNAME, argv[5]);
                scompile(stmp, NULL, 0);
        }
        row0 = (POINT *)malloc((n+3)*sizeof(POINT));
        row1 = (POINT *)malloc((n+3)*sizeof(POINT));
        row2 = (POINT *)malloc((n+3)*sizeof(POINT));
        if (row0 == NULL || row1 == NULL || row2 == NULL) {
                fprintf(stderr, "%s: out of memory\n", argv[0]);
                quit(1);
        }
        row0++; row1++; row2++;
                                                /* print header */
        fputs("# ", stdout);
        printargs(argc, argv, stdout);
        doptimize(1);
        eclock++;
                                                /* initialize */
        comprow(-1.0/m, row0, n);
        comprow(0.0, row1, n);
        comprow(1.0/m, row2, n);
        compnorms(row0, row1, row2, n);
        if (objout) {
                printf("\nusemtl %s\n\n", modname);
                printf("o %s\n\n", surfname);
                putobjrow(row1, n);
        }
                                                /* for each row */
        for (i = 0; i < m; i++) {
                                                /* compute next row */
                rp = row0;
                row0 = row1;
                row1 = row2;
                row2 = rp;
                comprow((double)(i+2)/m, row2, n);
                compnorms(row0, row1, row2, n);
                if (objout)
                        putobjrow(row1, n);

                for (j = 0; j < n; j++) {
                        int  orient = (j & 1);
                                                        /* put polygons */
                        if (!(row0[j].valid && row1[j+1].valid))
                                orient = 1;
                        else if (!(row1[j].valid && row0[j+1].valid))
                                orient = 0;
                        if (orient)
                                putsquare(&row0[j], &row1[j],
                                                &row0[j+1], &row1[j+1]);
                        else
                                putsquare(&row1[j], &row1[j+1],
                                                &row0[j], &row0[j+1]);
                }
        }

        return 0;

userror:
        fprintf(stderr, "Usage: %s material name ", argv[0]);
        fprintf(stderr, "x(s,t) y(s,t) z(s,t) m n [-s][-o][-e expr][-f file]\n");
        return 1;
}


void
loaddata(               /* load point data from file */
        char  *file,
        int  m,
        int  n,
        int  pointsize
)
{
        FILE  *fp;
        char  word[64];
        int  size;
        RREAL  *dp;

        datarec.flags = HASBORDER;              /* assume border values */
        datarec.m = m+1;
        datarec.n = n+1;
        size = datarec.m*datarec.n*pointsize;
        if (pointsize == 3)
                datarec.flags |= TRIPLETS;
        dp = (RREAL *)malloc(size*sizeof(RREAL));
        if ((datarec.data = dp) == NULL) {
                fputs("Out of memory\n", stderr);
                exit(1);
        }
        if (!strcmp(file, "-")) {
                file = "<stdin>";
                fp = stdin;
        } else if ((fp = fopen(file, "r")) == NULL) {
                fputs(file, stderr);
                fputs(": cannot open\n", stderr);
                exit(1);
        }
        while (size > 0 && fgetword(word, sizeof(word), fp) != NULL) {
                if (!isflt(word)) {
                        fprintf(stderr, "%s: garbled data value: %s\n",
                                        file, word);
                        exit(1);
                }
                *dp++ = atof(word);
                size--;
        }
        if (size == (m+n+1)*pointsize) {        /* no border after all */
                dp = (RREAL *)realloc(datarec.data,
                                m*n*pointsize*sizeof(RREAL));
                if (dp != NULL)
                        datarec.data = dp;
                datarec.flags &= ~HASBORDER;
                datarec.m = m;
                datarec.n = n;
                size = 0;
        }
        if (datarec.m < 2 || datarec.n < 2 || size != 0 ||
                        fgetword(word, sizeof(word), fp) != NULL) {
                fputs(file, stderr);
                fputs(": bad number of data points\n", stderr);
                exit(1);
        }
        fclose(fp);
}


double
l_dataval(                              /* return recorded data value */
        char  *nam
)
{
        double  u, v;
        int  i, j;
        RREAL  *dp;
        double  d00, d01, d10, d11;
                                                /* compute coordinates */
        u = argument(1); v = argument(2);
        if (datarec.flags & HASBORDER) {
                i = u *= datarec.m-1;
                j = v *= datarec.n-1;
        } else {
                i = u = u*datarec.m - .5;
                j = v = v*datarec.n - .5;
        }
        if (i < 0) i = 0;
        else if (i > datarec.m-2) i = datarec.m-2;
        if (j < 0) j = 0;
        else if (j > datarec.n-2) j = datarec.n-2;
                                                /* compute value */
        if (datarec.flags & TRIPLETS) {
                dp = datarec.data + 3*(j*datarec.m + i);
                if (nam == ZNAME)
                        dp += 2;
                else if (nam == YNAME)
                        dp++;
                d00 = dp[0]; d01 = dp[3];
                dp += 3*datarec.m;
                d10 = dp[0]; d11 = dp[3];
        } else {
                dp = datarec.data + j*datarec.m + i;
                d00 = dp[0]; d01 = dp[1];
                dp += datarec.m;
                d10 = dp[0]; d11 = dp[1];
        }
                                                /* bilinear interpolation */
        return((j+1-v)*((i+1-u)*d00+(u-i)*d01)+(v-j)*((i+1-u)*d10+(u-i)*d11));
}


void
putobjrow(                      /* output vertex row to .OBJ */
        POINT  *rp,
        int  n
)
{
        static FVECT  prevNorm;

        for ( ; n-- >= 0; rp++) {
                if (!rp->valid)
                        continue;
                fputs("v ", stdout);
                pvect(rp->p);
                rp->valid = ++nverts;
                printf("\tvt %.9g %.9g\n", rp->uv[0], rp->uv[1]);
                if (!smooth || ZEROVECT(rp->n))
                        rp->nvalid = 0;
                else if (VABSEQ(rp->n, prevNorm))
                        rp->nvalid = nnorms;
                else {
                        printf("\tvn %.9g %.9g %.9g\n",
                                        rp->n[0], rp->n[1], rp->n[2]);
                        rp->nvalid = ++nnorms;
                        VCOPY(prevNorm, rp->n);
                }
        }
}


void
putobjvert(             /* put out OBJ vertex index triplet */
        POINT *p
)
{
        int     pti = p->valid ? p->valid-nverts-1 : 0;
        int     ni = p->nvalid ? p->nvalid-nnorms-1 : 0;
        
        printf(" %d/%d/%d", pti, pti, ni);
}


void
putsquare(              /* put out a square */
        POINT *p0,
        POINT *p1,
        POINT *p2,
        POINT *p3
)
{
        static int  nout = 0;
        FVECT  norm[4];
        int  axis;
        FVECT  v1, v2, vc1, vc2;
        int  ok1, ok2;

        if (rev) {                      /* reverse normals? */
                POINT  *pt = p1; p1 = p2; p2 = pt;
        }
                                        /* compute exact normals */
        ok1 = (p0->valid && p1->valid && p2->valid);
        if (ok1) {
                VSUB(v1, p1->p, p0->p);
                VSUB(v2, p2->p, p0->p);
                fcross(vc1, v1, v2);
                ok1 = (normalize(vc1) != 0.0);
        }
        ok2 = (p1->valid && p2->valid && p3->valid);
        if (ok2) {
                VSUB(v1, p2->p, p3->p);
                VSUB(v2, p1->p, p3->p);
                fcross(vc2, v1, v2);
                ok2 = (normalize(vc2) != 0.0);
        }
        if (!(ok1 | ok2))
                return;
        if (objout) {                   /* output .OBJ faces */
                if (ok1 & ok2 && fdot(vc1,vc2) >= 1.0-FTINY*FTINY) {
                        putc('f', stdout);
                        putobjvert(p0); putobjvert(p1);
                        putobjvert(p3); putobjvert(p2);
                        putc('\n', stdout);
                        return;
                }
                if (ok1) {
                        putc('f', stdout);
                        putobjvert(p0); putobjvert(p1); putobjvert(p2);
                        putc('\n', stdout);
                }
                if (ok2) {
                        putc('f', stdout);
                        putobjvert(p2); putobjvert(p1); putobjvert(p3);
                        putc('\n', stdout);
                }
                return;
        }
                                        /* compute normal interpolation */
        axis = norminterp(norm, p0, p1, p2, p3);

                                        /* put out quadrilateral? */
        if (ok1 & ok2 && fdot(vc1,vc2) >= 1.0-FTINY*FTINY) {
                printf("\n%s ", modname);
                if (axis != -1) {
                        printf("texfunc %s\n%s\n", texname, tsargs);
                        printf("0\n13\t%d\n", axis);
                        pvect(norm[0]);
                        pvect(norm[1]);
                        pvect(norm[2]);
                        fvsum(v1, norm[3], vc1, -0.5);
                        fvsum(v1, v1, vc2, -0.5);
                        pvect(v1);
                        printf("\n%s ", texname);
                }
                printf("polygon %s.%d\n", surfname, ++nout);
                printf("0\n0\n12\n");
                pvect(p0->p);
                pvect(p1->p);
                pvect(p3->p);
                pvect(p2->p);
                return;
        }
                                        /* put out triangles? */
        if (ok1) {
                printf("\n%s ", modname);
                if (axis != -1) {
                        printf("texfunc %s\n%s\n", texname, tsargs);
                        printf("0\n13\t%d\n", axis);
                        pvect(norm[0]);
                        pvect(norm[1]);
                        pvect(norm[2]);
                        fvsum(v1, norm[3], vc1, -1.0);
                        pvect(v1);
                        printf("\n%s ", texname);
                }
                printf("polygon %s.%d\n", surfname, ++nout);
                printf("0\n0\n9\n");
                pvect(p0->p);
                pvect(p1->p);
                pvect(p2->p);
        }
        if (ok2) {
                printf("\n%s ", modname);
                if (axis != -1) {
                        printf("texfunc %s\n%s\n", texname, tsargs);
                        printf("0\n13\t%d\n", axis);
                        pvect(norm[0]);
                        pvect(norm[1]);
                        pvect(norm[2]);
                        fvsum(v2, norm[3], vc2, -1.0);
                        pvect(v2);
                        printf("\n%s ", texname);
                }
                printf("polygon %s.%d\n", surfname, ++nout);
                printf("0\n0\n9\n");
                pvect(p2->p);
                pvect(p1->p);
                pvect(p3->p);
        }
}


void
comprow(                        /* compute row of values */
        double  s,
        POINT  *row,
        int  siz
)
{
        double  st[2];
        int  end;
        int  checkvalid;
        int  i;
        
        if (smooth) {
                i = -1;                 /* compute one past each end */
                end = siz+1;
        } else {
                if (s < -FTINY || s > 1.0+FTINY)
                        return;
                i = 0;
                end = siz;
        }
        st[0] = s;
        checkvalid = (fundefined(VNAME) == 2);
        while (i <= end) {
                st[1] = (double)i/siz;
                if (checkvalid && funvalue(VNAME, 2, st) <= 0.0) {
                        row[i].valid = 0;
                        row[i].p[0] = row[i].p[1] = row[i].p[2] = 0.0;
                        row[i].uv[0] = row[i].uv[1] = 0.0;
                } else {
                        row[i].valid = 1;
                        row[i].p[0] = funvalue(XNAME, 2, st);
                        row[i].p[1] = funvalue(YNAME, 2, st);
                        row[i].p[2] = funvalue(ZNAME, 2, st);
                        row[i].uv[0] = st[0];
                        row[i].uv[1] = st[1];
                }
                i++;
        }
}


void
compnorms(              /* compute row of averaged normals */
        POINT  *r0,
        POINT  *r1,
        POINT  *r2,
        int  siz
)
{
        FVECT  v1, v2;

        if (!smooth)                    /* not needed if no smoothing */
                return;
                                        /* compute row 1 normals */
        while (siz-- >= 0) {
                if (!r1[0].valid)
                        goto skip;
                if (!r0[0].valid) {
                        if (!r2[0].valid) {
                                r1[0].n[0] = r1[0].n[1] = r1[0].n[2] = 0.0;
                                goto skip;
                        }
                        fvsum(v1, r2[0].p, r1[0].p, -1.0);
                } else if (!r2[0].valid)
                        fvsum(v1, r1[0].p, r0[0].p, -1.0);
                else
                        fvsum(v1, r2[0].p, r0[0].p, -1.0);
                if (!r1[-1].valid) {
                        if (!r1[1].valid) {
                                r1[0].n[0] = r1[0].n[1] = r1[0].n[2] = 0.0;
                                goto skip;
                        }
                        fvsum(v2, r1[1].p, r1[0].p, -1.0);
                } else if (!r1[1].valid)
                        fvsum(v2, r1[0].p, r1[-1].p, -1.0);
                else
                        fvsum(v2, r1[1].p, r1[-1].p, -1.0);
                if (rev)
                        fcross(r1[0].n, v2, v1);
                else
                        fcross(r1[0].n, v1, v2);
                normalize(r1[0].n);
        skip:
                r0++; r1++; r2++;
        }
}


int
norminterp(     /* compute normal interpolation */
        FVECT  resmat[4],
        POINT  *p0,
        POINT  *p1,
        POINT  *p2,
        POINT  *p3
)
{
#define u  ((ax+1)%3)
#define v  ((ax+2)%3)

        int  ax;
        MAT4  eqnmat;
        FVECT  v1;
        int  i, j;

        if (!smooth)                    /* no interpolation if no smoothing */
                return(-1);
                                        /* find dominant axis */
        VCOPY(v1, p0->n);
        fvsum(v1, v1, p1->n, 1.0);
        fvsum(v1, v1, p2->n, 1.0);
        fvsum(v1, v1, p3->n, 1.0);
        ax = ABS(v1[0]) > ABS(v1[1]) ? 0 : 1;
        ax = ABS(v1[ax]) > ABS(v1[2]) ? ax : 2;
                                        /* assign equation matrix */
        eqnmat[0][0] = p0->p[u]*p0->p[v];
        eqnmat[0][1] = p0->p[u];
        eqnmat[0][2] = p0->p[v];
        eqnmat[0][3] = 1.0;
        eqnmat[1][0] = p1->p[u]*p1->p[v];
        eqnmat[1][1] = p1->p[u];
        eqnmat[1][2] = p1->p[v];
        eqnmat[1][3] = 1.0;
        eqnmat[2][0] = p2->p[u]*p2->p[v];
        eqnmat[2][1] = p2->p[u];
        eqnmat[2][2] = p2->p[v];
        eqnmat[2][3] = 1.0;
        eqnmat[3][0] = p3->p[u]*p3->p[v];
        eqnmat[3][1] = p3->p[u];
        eqnmat[3][2] = p3->p[v];
        eqnmat[3][3] = 1.0;
                                        /* invert matrix (solve system) */
        if (!invmat4(eqnmat, eqnmat))
                return(-1);                     /* no solution */
                                        /* compute result matrix */
        for (j = 0; j < 4; j++)
                for (i = 0; i < 3; i++)
                        resmat[j][i] =  eqnmat[j][0]*p0->n[i] +
                                        eqnmat[j][1]*p1->n[i] +
                                        eqnmat[j][2]*p2->n[i] +
                                        eqnmat[j][3]*p3->n[i];
        return(ax);

#undef u
#undef v
}


double
l_hermite(char *nm)
{
        double  t;
        
        t = argument(5);
        return( argument(1)*((2.0*t-3.0)*t*t+1.0) +
                argument(2)*(-2.0*t+3.0)*t*t +
                argument(3)*((t-2.0)*t+1.0)*t +
                argument(4)*(t-1.0)*t*t );
}


double
l_bezier(char *nm)
{
        double  t;

        t = argument(5);
        return( argument(1) * (1.+t*(-3.+t*(3.-t))) +
                argument(2) * 3.*t*(1.+t*(-2.+t)) +
                argument(3) * 3.*t*t*(1.-t) +
                argument(4) * t*t*t );
}


double
l_bspline(char *nm)
{
        double  t;

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