src/common/calfunc.c

#ifndef lint
static const char       RCSid[] = "$Id$";
#endif
/*
 *  calfunc.c - routines for calcomp using functions.
 *
 *      If VARIABLE is not set, only library functions
 *  can be accessed.
 *
 *  2/19/03     Eliminated conditional compiles in favor of esupport extern.
 */

#include "copyright.h"

#include  <stdio.h>

#include  <errno.h>

#include  <math.h>

#include  "calcomp.h"

                                /* bits in argument flag (better be right!) */
#define  AFLAGSIZ       (8*sizeof(unsigned long))
#define  ALISTSIZ       6       /* maximum saved argument list */

typedef struct activation {
    char  *name;                /* function name */
    struct activation  *prev;   /* previous activation */
    double  *ap;                /* argument list */
    unsigned long  an;          /* computed argument flags */
    EPNODE  *fun;               /* argument function */
}  ACTIVATION;          /* an activation record */

static ACTIVATION  *curact = NULL;

static double  libfunc();

#ifndef  MAXLIB
#define  MAXLIB         64      /* maximum number of library functions */
#endif

static double  l_if(), l_select(), l_rand();
static double  l_floor(), l_ceil();
static double  l_sqrt();
static double  l_sin(), l_cos(), l_tan();
static double  l_asin(), l_acos(), l_atan(), l_atan2();
static double  l_exp(), l_log(), l_log10();

                        /* functions must be listed alphabetically */
static LIBR  library[MAXLIB] = {
    { "acos", 1, ':', l_acos },
    { "asin", 1, ':', l_asin },
    { "atan", 1, ':', l_atan },
    { "atan2", 2, ':', l_atan2 },
    { "ceil", 1, ':', l_ceil },
    { "cos", 1, ':', l_cos },
    { "exp", 1, ':', l_exp },
    { "floor", 1, ':', l_floor },
    { "if", 3, ':', l_if },
    { "log", 1, ':', l_log },
    { "log10", 1, ':', l_log10 },
    { "rand", 1, ':', l_rand },
    { "select", 1, ':', l_select },
    { "sin", 1, ':', l_sin },
    { "sqrt", 1, ':', l_sqrt },
    { "tan", 1, ':', l_tan },
};

static int  libsize = 16;

#define  resolve(ep)    ((ep)->type==VAR?(ep)->v.ln:argf((ep)->v.chan))


int
fundefined(fname)               /* return # of arguments for function */
char  *fname;
{
    register LIBR  *lp;
    register VARDEF  *vp;

    if ((vp = varlookup(fname)) != NULL && vp->def != NULL
                && vp->def->v.kid->type == FUNC)
        return(nekids(vp->def->v.kid) - 1);
    lp = vp != NULL ? vp->lib : liblookup(fname);
    if (lp == NULL)
        return(0);
    return(lp->nargs);
}


double
funvalue(fname, n, a)           /* return a function value to the user */
char  *fname;
int  n;
double  *a;
{
    ACTIVATION  act;
    register VARDEF  *vp;
    double  rval;
                                        /* push environment */
    act.name = fname;
    act.prev = curact;
    act.ap = a;
    if (n >= AFLAGSIZ)
        act.an = ~0;
    else
        act.an = (1L<<n)-1;
    act.fun = NULL;
    curact = &act;

    if ((vp = varlookup(fname)) == NULL || vp->def == NULL
                || vp->def->v.kid->type != FUNC)
        rval = libfunc(fname, vp);
    else
        rval = evalue(vp->def->v.kid->sibling);

    curact = act.prev;                  /* pop environment */
    return(rval);
}


void
funset(fname, nargs, assign, fptr)      /* set a library function */
char  *fname;
int  nargs;
int  assign;
double  (*fptr)();
{
    int  oldlibsize = libsize;
    char *cp;
    register LIBR  *lp;
                                                /* check for context */
    for (cp = fname; *cp; cp++)
        ;
    if (cp == fname)
        return;
    if (cp[-1] == CNTXMARK)
        *--cp = '\0';
    if ((lp = liblookup(fname)) == NULL) {      /* insert */
        if (libsize >= MAXLIB) {
            eputs("Too many library functons!\n");
            quit(1);
        }
        for (lp = &library[libsize]; lp > library; lp--)
            if (strcmp(lp[-1].fname, fname) > 0) {
                lp[0].fname = lp[-1].fname;
                lp[0].nargs = lp[-1].nargs;
                lp[0].atyp = lp[-1].atyp;
                lp[0].f = lp[-1].f;
            } else
                break;
        libsize++;
    }
    if (fptr == NULL) {                         /* delete */
        while (lp < &library[libsize-1]) {
            lp[0].fname = lp[1].fname;
            lp[0].nargs = lp[1].nargs;
            lp[0].atyp = lp[1].atyp;
            lp[0].f = lp[1].f;
            lp++;
        }
        libsize--;
    } else {                                    /* or assign */
        lp[0].fname = fname;            /* string must be static! */
        lp[0].nargs = nargs;
        lp[0].atyp = assign;
        lp[0].f = fptr;
    }
    if (libsize != oldlibsize)
        libupdate(fname);                       /* relink library */
}


int
nargum()                        /* return number of available arguments */
{
    register int  n;

    if (curact == NULL)
        return(0);
    if (curact->fun == NULL) {
        for (n = 0; (1L<<n) & curact->an; n++)
            ;
        return(n);
    }
    return(nekids(curact->fun) - 1);
}


double
argument(n)                     /* return nth argument for active function */
register int  n;
{
    register ACTIVATION  *actp = curact;
    register EPNODE  *ep;
    double  aval;

    if (actp == NULL || --n < 0) {
        eputs("Bad call to argument!\n");
        quit(1);
    }
                                                /* already computed? */
    if (n < AFLAGSIZ && 1L<<n & actp->an)
        return(actp->ap[n]);

    if (actp->fun == NULL || (ep = ekid(actp->fun, n+1)) == NULL) {
        eputs(actp->name);
        eputs(": too few arguments\n");
        quit(1);
    }
    curact = actp->prev;                        /* pop environment */
    aval = evalue(ep);                          /* compute argument */
    curact = actp;                              /* push back environment */
    if (n < ALISTSIZ) {                         /* save value */
        actp->ap[n] = aval;
        actp->an |= 1L<<n;
    }
    return(aval);
}


VARDEF *
argf(n)                         /* return function def for nth argument */
int  n;
{
    register ACTIVATION  *actp;
    register EPNODE  *ep;

    for (actp = curact; actp != NULL; actp = actp->prev) {

        if (n <= 0)
            break;

        if (actp->fun == NULL)
            goto badarg;

        if ((ep = ekid(actp->fun, n)) == NULL) {
            eputs(actp->name);
            eputs(": too few arguments\n");
            quit(1);
        }
        if (ep->type == VAR)
            return(ep->v.ln);                   /* found it */

        if (ep->type != ARG)
            goto badarg;

        n = ep->v.chan;                         /* try previous context */
    }
    eputs("Bad call to argf!\n");
    quit(1);

badarg:
    eputs(actp->name);
    eputs(": argument not a function\n");
    quit(1);
}


char *
argfun(n)                       /* return function name for nth argument */
int  n;
{
    return(argf(n)->name);
}


double
efunc(ep)                               /* evaluate a function */
register EPNODE  *ep;
{
    ACTIVATION  act;
    double  alist[ALISTSIZ];
    double  rval;
    register VARDEF  *dp;
                                        /* push environment */
    dp = resolve(ep->v.kid);
    act.name = dp->name;
    act.prev = curact;
    act.ap = alist;
    act.an = 0;
    act.fun = ep;
    curact = &act;

    if (dp->def == NULL || dp->def->v.kid->type != FUNC)
        rval = libfunc(act.name, dp);
    else
        rval = evalue(dp->def->v.kid->sibling);
    
    curact = act.prev;                  /* pop environment */
    return(rval);
}


LIBR *
liblookup(fname)                /* look up a library function */
char  *fname;
{
    int  upper, lower;
    register int  cm, i;

    lower = 0;
    upper = cm = libsize;

    while ((i = (lower + upper) >> 1) != cm) {
        cm = strcmp(fname, library[i].fname);
        if (cm > 0)
            lower = i;
        else if (cm < 0)
            upper = i;
        else
            return(&library[i]);
        cm = i;
    }
    return(NULL);
}


/*
 *  The following routines are for internal use:
 */


static double
libfunc(fname, vp)                      /* execute library function */
char  *fname;
VARDEF  *vp;
{
    register LIBR  *lp;
    double  d;
    int  lasterrno;

    if (vp != NULL)
        lp = vp->lib;
    else
        lp = liblookup(fname);
    if (lp == NULL) {
        eputs(fname);
        eputs(": undefined function\n");
        quit(1);
    }
    lasterrno = errno;
    errno = 0;
    d = (*lp->f)(lp->fname);
#ifdef  IEEE
    if (errno == 0)
        if (isnan(d))
            errno = EDOM;
        else if (isinf(d))
            errno = ERANGE;
#endif
    if (errno) {
        wputs(fname);
        if (errno == EDOM)
                wputs(": domain error\n");
        else if (errno == ERANGE)
                wputs(": range error\n");
        else
                wputs(": error in call\n");
        return(0.0);
    }
    errno = lasterrno;
    return(d);
}


/*
 *  Library functions:
 */


static double
l_if()                  /* if(cond, then, else) conditional expression */
                        /* cond evaluates true if greater than zero */
{
    if (argument(1) > 0.0)
        return(argument(2));
    else
        return(argument(3));
}


static double
l_select()              /* return argument #(A1+1) */
{
        register int  n;

        n = argument(1) + .5;
        if (n == 0)
                return(nargum()-1);
        if (n < 1 || n > nargum()-1) {
                errno = EDOM;
                return(0.0);
        }
        return(argument(n+1));
}


static double
l_rand()                /* random function between 0 and 1 */
{
    double  x;

    x = argument(1);
    x *= 1.0/(1.0 + x*x) + 2.71828182845904;
    x += .785398163397447 - floor(x);
    x = 1e5 / x;
    return(x - floor(x));
}


static double
l_floor()               /* return largest integer not greater than arg1 */
{
    return(floor(argument(1)));
}


static double
l_ceil()                /* return smallest integer not less than arg1 */
{
    return(ceil(argument(1)));
}


static double
l_sqrt()
{
    return(sqrt(argument(1)));
}


static double
l_sin()
{
    return(sin(argument(1)));
}


static double
l_cos()
{
    return(cos(argument(1)));
}


static double
l_tan()
{
    return(tan(argument(1)));
}


static double
l_asin()
{
    return(asin(argument(1)));
}


static double
l_acos()
{
    return(acos(argument(1)));
}


static double
l_atan()
{
    return(atan(argument(1)));
}


static double
l_atan2()
{
    return(atan2(argument(1), argument(2)));
}


static double
l_exp()
{
    return(exp(argument(1)));
}


static double
l_log()
{
    return(log(argument(1)));
}


static double
l_log10()
{
    return(log10(argument(1)));
}
Revision:	2.9
Committed:	Tue Feb 25 02:47:21 2003 UTC (22 years, 8 months ago) by greg
Content type:	text/plain
Branch:	MAIN
Changes since 2.8:	+1 -56 lines
Log Message:	Replaced inline copyright notice with #include "copyright.h"
#	User	Rev	Content
1	greg	1.1	#ifndef lint
2	greg	2.8	static const char RCSid[] = "$Id$";
3	greg	1.1	#endif
4			/*
5			* calfunc.c - routines for calcomp using functions.
6			*
7	greg	2.8	* If VARIABLE is not set, only library functions
8			* can be accessed.
9			*
10			* 2/19/03 Eliminated conditional compiles in favor of esupport extern.
11			*/
12
13	greg	2.9	#include "copyright.h"
14	greg	1.1
15			#include <stdio.h>
16
17			#include <errno.h>
18
19	greg	2.5	#include <math.h>
20
21	greg	1.1	#include "calcomp.h"
22
23	greg	1.5	/* bits in argument flag (better be right!) */
24			#define AFLAGSIZ (8*sizeof(unsigned long))
25	greg	1.1	#define ALISTSIZ 6 /* maximum saved argument list */
26
27			typedef struct activation {
28			char name; / function name */
29			struct activation prev; / previous activation */
30			double ap; / argument list */
31			unsigned long an; /* computed argument flags */
32			EPNODE fun; / argument function */
33			} ACTIVATION; /* an activation record */
34
35			static ACTIVATION *curact = NULL;
36
37	greg	1.2	static double libfunc();
38	greg	1.1
39	greg	2.8	#ifndef MAXLIB
40	greg	1.1	#define MAXLIB 64 /* maximum number of library functions */
41	greg	2.8	#endif
42	greg	1.1
43	greg	1.2	static double l_if(), l_select(), l_rand();
44			static double l_floor(), l_ceil();
45			static double l_sqrt();
46			static double l_sin(), l_cos(), l_tan();
47			static double l_asin(), l_acos(), l_atan(), l_atan2();
48			static double l_exp(), l_log(), l_log10();
49	greg	1.1
50			/* functions must be listed alphabetically */
51			static LIBR library[MAXLIB] = {
52	greg	1.4	{ "acos", 1, ':', l_acos },
53			{ "asin", 1, ':', l_asin },
54			{ "atan", 1, ':', l_atan },
55			{ "atan2", 2, ':', l_atan2 },
56			{ "ceil", 1, ':', l_ceil },
57			{ "cos", 1, ':', l_cos },
58			{ "exp", 1, ':', l_exp },
59			{ "floor", 1, ':', l_floor },
60			{ "if", 3, ':', l_if },
61			{ "log", 1, ':', l_log },
62			{ "log10", 1, ':', l_log10 },
63			{ "rand", 1, ':', l_rand },
64			{ "select", 1, ':', l_select },
65			{ "sin", 1, ':', l_sin },
66			{ "sqrt", 1, ':', l_sqrt },
67			{ "tan", 1, ':', l_tan },
68	greg	1.1	};
69
70			static int libsize = 16;
71
72			#define resolve(ep) ((ep)->type==VAR?(ep)->v.ln:argf((ep)->v.chan))
73
74
75			int
76			fundefined(fname) /* return # of arguments for function */
77			char *fname;
78			{
79	greg	2.7	register LIBR *lp;
80	greg	1.1	register VARDEF *vp;
81
82	greg	2.7	if ((vp = varlookup(fname)) != NULL && vp->def != NULL
83			&& vp->def->v.kid->type == FUNC)
84	greg	1.1	return(nekids(vp->def->v.kid) - 1);
85	greg	2.7	lp = vp != NULL ? vp->lib : liblookup(fname);
86			if (lp == NULL)
87			return(0);
88			return(lp->nargs);
89	greg	1.1	}
90
91
92			double
93			funvalue(fname, n, a) /* return a function value to the user */
94			char *fname;
95			int n;
96			double *a;
97			{
98			ACTIVATION act;
99			register VARDEF *vp;
100			double rval;
101			/* push environment */
102			act.name = fname;
103			act.prev = curact;
104			act.ap = a;
105	greg	1.5	if (n >= AFLAGSIZ)
106			act.an = ~0;
107			else
108			act.an = (1L<<n)-1;
109	greg	1.1	act.fun = NULL;
110			curact = &act;
111
112			if ((vp = varlookup(fname)) == NULL \|\| vp->def == NULL
113			\|\| vp->def->v.kid->type != FUNC)
114			rval = libfunc(fname, vp);
115			else
116			rval = evalue(vp->def->v.kid->sibling);
117
118			curact = act.prev; /* pop environment */
119			return(rval);
120			}
121
122
123	greg	2.8	void
124	greg	1.4	funset(fname, nargs, assign, fptr) /* set a library function */
125	greg	1.1	char *fname;
126			int nargs;
127	greg	1.4	int assign;
128	greg	1.1	double (*fptr)();
129			{
130	greg	2.6	int oldlibsize = libsize;
131	greg	2.8	char *cp;
132	greg	1.1	register LIBR *lp;
133	greg	2.8	/* check for context */
134			for (cp = fname; *cp; cp++)
135			;
136			if (cp == fname)
137			return;
138			if (cp[-1] == CNTXMARK)
139			*--cp = '\0';
140	greg	2.2	if ((lp = liblookup(fname)) == NULL) { /* insert */
141	greg	1.1	if (libsize >= MAXLIB) {
142			eputs("Too many library functons!\n");
143			quit(1);
144			}
145			for (lp = &library[libsize]; lp > library; lp--)
146			if (strcmp(lp[-1].fname, fname) > 0) {
147			lp[0].fname = lp[-1].fname;
148			lp[0].nargs = lp[-1].nargs;
149	greg	1.4	lp[0].atyp = lp[-1].atyp;
150	greg	1.1	lp[0].f = lp[-1].f;
151			} else
152			break;
153			libsize++;
154			}
155	greg	2.2	if (fptr == NULL) { /* delete */
156			while (lp < &library[libsize-1]) {
157			lp[0].fname = lp[1].fname;
158			lp[0].nargs = lp[1].nargs;
159			lp[0].atyp = lp[1].atyp;
160			lp[0].f = lp[1].f;
161			lp++;
162			}
163			libsize--;
164			} else { /* or assign */
165			lp[0].fname = fname; /* string must be static! */
166			lp[0].nargs = nargs;
167			lp[0].atyp = assign;
168			lp[0].f = fptr;
169			}
170	greg	2.6	if (libsize != oldlibsize)
171			libupdate(fname); /* relink library */
172	greg	1.1	}
173
174
175			int
176			nargum() /* return number of available arguments */
177			{
178			register int n;
179
180			if (curact == NULL)
181			return(0);
182			if (curact->fun == NULL) {
183			for (n = 0; (1L<<n) & curact->an; n++)
184			;
185			return(n);
186			}
187			return(nekids(curact->fun) - 1);
188			}
189
190
191			double
192			argument(n) /* return nth argument for active function */
193			register int n;
194			{
195			register ACTIVATION *actp = curact;
196	greg	1.7	register EPNODE *ep;
197	greg	1.1	double aval;
198
199			if (actp == NULL \|\| --n < 0) {
200			eputs("Bad call to argument!\n");
201			quit(1);
202			}
203			/* already computed? */
204	greg	1.5	if (n < AFLAGSIZ && 1L<<n & actp->an)
205	greg	1.1	return(actp->ap[n]);
206
207			if (actp->fun == NULL \|\| (ep = ekid(actp->fun, n+1)) == NULL) {
208			eputs(actp->name);
209			eputs(": too few arguments\n");
210			quit(1);
211			}
212			curact = actp->prev; /* pop environment */
213			aval = evalue(ep); /* compute argument */
214			curact = actp; /* push back environment */
215			if (n < ALISTSIZ) { /* save value */
216			actp->ap[n] = aval;
217			actp->an \|= 1L<<n;
218			}
219			return(aval);
220			}
221
222
223			VARDEF *
224			argf(n) /* return function def for nth argument */
225			int n;
226			{
227			register ACTIVATION *actp;
228			register EPNODE *ep;
229
230			for (actp = curact; actp != NULL; actp = actp->prev) {
231
232			if (n <= 0)
233			break;
234
235			if (actp->fun == NULL)
236			goto badarg;
237
238			if ((ep = ekid(actp->fun, n)) == NULL) {
239			eputs(actp->name);
240			eputs(": too few arguments\n");
241			quit(1);
242			}
243			if (ep->type == VAR)
244			return(ep->v.ln); /* found it */
245
246			if (ep->type != ARG)
247			goto badarg;
248
249			n = ep->v.chan; /* try previous context */
250			}
251			eputs("Bad call to argf!\n");
252			quit(1);
253
254			badarg:
255			eputs(actp->name);
256			eputs(": argument not a function\n");
257			quit(1);
258			}
259
260
261			char *
262			argfun(n) /* return function name for nth argument */
263			int n;
264			{
265			return(argf(n)->name);
266			}
267
268
269			double
270			efunc(ep) /* evaluate a function */
271			register EPNODE *ep;
272			{
273			ACTIVATION act;
274			double alist[ALISTSIZ];
275			double rval;
276			register VARDEF *dp;
277			/* push environment */
278			dp = resolve(ep->v.kid);
279			act.name = dp->name;
280			act.prev = curact;
281			act.ap = alist;
282			act.an = 0;
283			act.fun = ep;
284			curact = &act;
285
286			if (dp->def == NULL \|\| dp->def->v.kid->type != FUNC)
287			rval = libfunc(act.name, dp);
288			else
289			rval = evalue(dp->def->v.kid->sibling);
290
291			curact = act.prev; /* pop environment */
292			return(rval);
293			}
294
295
296			LIBR *
297			liblookup(fname) /* look up a library function */
298			char *fname;
299			{
300			int upper, lower;
301			register int cm, i;
302
303			lower = 0;
304			upper = cm = libsize;
305
306			while ((i = (lower + upper) >> 1) != cm) {
307			cm = strcmp(fname, library[i].fname);
308			if (cm > 0)
309			lower = i;
310			else if (cm < 0)
311			upper = i;
312			else
313			return(&library[i]);
314			cm = i;
315			}
316			return(NULL);
317			}
318
319
320			/*
321			* The following routines are for internal use:
322			*/
323
324
325			static double
326			libfunc(fname, vp) /* execute library function */
327			char *fname;
328	greg	2.3	VARDEF *vp;
329	greg	1.1	{
330	greg	2.3	register LIBR *lp;
331	greg	1.1	double d;
332			int lasterrno;
333
334	greg	2.3	if (vp != NULL)
335			lp = vp->lib;
336			else
337			lp = liblookup(fname);
338			if (lp == NULL) {
339	greg	1.1	eputs(fname);
340			eputs(": undefined function\n");
341			quit(1);
342			}
343			lasterrno = errno;
344			errno = 0;
345	greg	2.3	d = (*lp->f)(lp->fname);
346	greg	1.1	#ifdef IEEE
347	greg	1.9	if (errno == 0)
348			if (isnan(d))
349			errno = EDOM;
350			else if (isinf(d))
351			errno = ERANGE;
352	greg	1.1	#endif
353			if (errno) {
354			wputs(fname);
355	greg	1.8	if (errno == EDOM)
356			wputs(": domain error\n");
357			else if (errno == ERANGE)
358			wputs(": range error\n");
359			else
360			wputs(": error in call\n");
361	greg	1.1	return(0.0);
362			}
363			errno = lasterrno;
364			return(d);
365			}
366
367
368			/*
369			* Library functions:
370			*/
371
372
373			static double
374			l_if() /* if(cond, then, else) conditional expression */
375			/* cond evaluates true if greater than zero */
376			{
377			if (argument(1) > 0.0)
378			return(argument(2));
379			else
380			return(argument(3));
381			}
382
383
384			static double
385			l_select() /* return argument #(A1+1) */
386			{
387			register int n;
388
389			n = argument(1) + .5;
390			if (n == 0)
391			return(nargum()-1);
392			if (n < 1 \|\| n > nargum()-1) {
393			errno = EDOM;
394			return(0.0);
395			}
396			return(argument(n+1));
397			}
398
399
400			static double
401			l_rand() /* random function between 0 and 1 */
402			{
403			double x;
404
405			x = argument(1);
406			x = 1.0/(1.0 + xx) + 2.71828182845904;
407			x += .785398163397447 - floor(x);
408			x = 1e5 / x;
409			return(x - floor(x));
410			}
411
412
413			static double
414			l_floor() /* return largest integer not greater than arg1 */
415			{
416			return(floor(argument(1)));
417			}
418
419
420			static double
421			l_ceil() /* return smallest integer not less than arg1 */
422			{
423			return(ceil(argument(1)));
424			}
425
426
427			static double
428			l_sqrt()
429			{
430			return(sqrt(argument(1)));
431			}
432
433
434			static double
435			l_sin()
436			{
437			return(sin(argument(1)));
438			}
439
440
441			static double
442			l_cos()
443			{
444			return(cos(argument(1)));
445			}
446
447
448			static double
449			l_tan()
450			{
451			return(tan(argument(1)));
452			}
453
454
455			static double
456			l_asin()
457			{
458			return(asin(argument(1)));
459			}
460
461
462			static double
463			l_acos()
464			{
465			return(acos(argument(1)));
466			}
467
468
469			static double
470			l_atan()
471			{
472			return(atan(argument(1)));
473			}
474
475
476			static double
477			l_atan2()
478			{
479			return(atan2(argument(1), argument(2)));
480			}
481
482
483			static double
484			l_exp()
485			{
486			return(exp(argument(1)));
487			}
488
489
490			static double
491			l_log()
492			{
493			return(log(argument(1)));
494			}
495
496
497			static double
498			l_log10()
499			{
500			return(log10(argument(1)));
501			}