35 |
|
|
36 |
|
#define isdecimal(c) (isdigit(c) | ((c) == '.')) |
37 |
|
|
38 |
+ |
#define envalue(ep) ((ep)->type==NUM ? (ep)->v.num : evalue(ep)) |
39 |
+ |
|
40 |
|
static double euminus(EPNODE *), eargument(EPNODE *), enumber(EPNODE *); |
41 |
|
static double echannel(EPNODE *); |
42 |
|
static double eadd(EPNODE *), esubtr(EPNODE *), |
211 |
|
} |
212 |
|
|
213 |
|
|
214 |
+ |
static void |
215 |
+ |
epflatten( /* flatten hierarchies for '+', '*' */ |
216 |
+ |
EPNODE *epar |
217 |
+ |
) |
218 |
+ |
{ |
219 |
+ |
EPNODE *ep; |
220 |
+ |
|
221 |
+ |
if (epar->nkids < 0) /* can't handle array allocations */ |
222 |
+ |
return; |
223 |
+ |
|
224 |
+ |
for (ep = epar->v.kid; ep != NULL; ep = ep->sibling) |
225 |
+ |
while (ep->type == epar->type && ep->nkids > 0) { |
226 |
+ |
EPNODE *ep1 = ep->v.kid; |
227 |
+ |
while (ep1->sibling != NULL) |
228 |
+ |
ep1 = ep1->sibling; |
229 |
+ |
ep1->sibling = ep->sibling; |
230 |
+ |
epar->nkids += ep->nkids - 1; |
231 |
+ |
ep1 = ep->v.kid; |
232 |
+ |
*ep = *ep1; |
233 |
+ |
efree(ep1); /* not epfree()! */ |
234 |
+ |
} |
235 |
+ |
} |
236 |
+ |
|
237 |
+ |
|
238 |
|
void |
239 |
< |
epoptimize( /* realloc lists as arrays if > length 3 */ |
239 |
> |
epoptimize( /* flatten operations, lists -> arrays */ |
240 |
|
EPNODE *epar |
241 |
|
) |
242 |
|
{ |
243 |
|
EPNODE *ep; |
244 |
|
|
245 |
< |
if (epar->nkids > 3) { /* do this node if > 3 kids */ |
245 |
> |
if ((epar->type == '+') | (epar->type == '*')) |
246 |
> |
epflatten(epar); /* flatten associative operations */ |
247 |
> |
|
248 |
> |
if (epar->nkids) /* do children if any */ |
249 |
> |
for (ep = epar->v.kid; ep != NULL; ep = ep->sibling) |
250 |
> |
epoptimize(ep); |
251 |
> |
|
252 |
> |
if (epar->nkids > 4) { /* make list into array if > 4 kids */ |
253 |
|
int n = 1; |
254 |
|
epar->v.kid = (EPNODE *)erealloc(epar->v.kid, |
255 |
|
sizeof(EPNODE)*epar->nkids); |
260 |
|
epar->v.kid[n-1].sibling = epar->v.kid + n; |
261 |
|
n++; |
262 |
|
} |
263 |
< |
epar->nkids = -epar->nkids; |
263 |
> |
epar->nkids = -n; |
264 |
|
} |
232 |
– |
if (epar->nkids) /* do children if any */ |
233 |
– |
for (ep = epar->v.kid; ep != NULL; ep = ep->sibling) |
234 |
– |
epoptimize(ep); |
265 |
|
} |
266 |
|
|
267 |
|
/* the following used to be a switch */ |
304 |
|
EPNODE *ep |
305 |
|
) |
306 |
|
{ |
307 |
+ |
double sum = 0; |
308 |
|
EPNODE *ep1 = ep->v.kid; |
309 |
|
|
310 |
< |
return(evalue(ep1) + evalue(ep1->sibling)); |
310 |
> |
do |
311 |
> |
sum += envalue(ep1); |
312 |
> |
while ((ep1 = ep1->sibling) != NULL); |
313 |
> |
|
314 |
> |
return(sum); |
315 |
|
} |
316 |
|
|
317 |
|
static double |
320 |
|
) |
321 |
|
{ |
322 |
|
EPNODE *ep1 = ep->v.kid; |
323 |
+ |
EPNODE *ep2 = ep1->sibling; |
324 |
|
|
325 |
< |
return(evalue(ep1) - evalue(ep1->sibling)); |
325 |
> |
return(envalue(ep1) - envalue(ep2)); |
326 |
|
} |
327 |
|
|
328 |
|
static double |
330 |
|
EPNODE *ep |
331 |
|
) |
332 |
|
{ |
333 |
+ |
double prod = 1; |
334 |
|
EPNODE *ep1 = ep->v.kid; |
335 |
|
|
336 |
< |
return(evalue(ep1) * evalue(ep1->sibling)); |
336 |
> |
do |
337 |
> |
prod *= envalue(ep1); |
338 |
> |
while ((ep1 = ep1->sibling) != NULL); |
339 |
> |
|
340 |
> |
return(prod); |
341 |
|
} |
342 |
|
|
343 |
|
static double |
346 |
|
) |
347 |
|
{ |
348 |
|
EPNODE *ep1 = ep->v.kid; |
349 |
< |
double d; |
349 |
> |
double den = evalue(ep1->sibling); |
350 |
|
|
351 |
< |
d = evalue(ep1->sibling); |
311 |
< |
if (d == 0.0) { |
351 |
> |
if (den == 0.0) { |
352 |
|
wputs("Division by zero\n"); |
353 |
|
errno = ERANGE; |
354 |
|
return(0.0); |
355 |
|
} |
356 |
< |
return(evalue(ep1) / d); |
356 |
> |
return(envalue(ep1) / den); |
357 |
|
} |
358 |
|
|
359 |
|
static double |
561 |
|
) |
562 |
|
{ |
563 |
|
if (ep->nkids < 0) { |
564 |
< |
eputs("Cannot add child after optimization\n"); |
564 |
> |
eputs("Cannot add kid to EPNODE array\n"); |
565 |
|
quit(1); |
566 |
|
} |
567 |
|
ep->nkids++; |
572 |
|
; |
573 |
|
ep->sibling = ek; |
574 |
|
} |
575 |
< |
ek->sibling = NULL; |
575 |
> |
ek->sibling = NULL; /* shouldn't be necessary */ |
576 |
|
} |
577 |
|
|
578 |
|
|
802 |
|
scan(); |
803 |
|
return(ep1); |
804 |
|
} |
765 |
– |
|
805 |
|
if (esupport&E_INCHAN && nextc == '$') { |
806 |
|
scan(); |
807 |
|
ep1 = newnode(); |
809 |
|
ep1->v.chan = getinum(); |
810 |
|
return(ep1); |
811 |
|
} |
773 |
– |
|
812 |
|
if (esupport&(E_VARIABLE|E_FUNCTION) && |
813 |
|
(isalpha(nextc) | (nextc == CNTXMARK))) { |
814 |
|
nam = getname(); |
846 |
|
ep1 = rconst(ep1); |
847 |
|
return(ep1); |
848 |
|
} |
811 |
– |
|
849 |
|
if (isdecimal(nextc)) { |
850 |
|
ep1 = newnode(); |
851 |
|
ep1->type = NUM; |