eigenmath/subst.cpp

56 lines
1.0 KiB
C++

#include "stdafx.h"
//-----------------------------------------------------------------------------
//
// Substitute new expr for old expr in expr.
//
// Input: tos-3 expr
//
// tos-2 old expr
//
// tos-1 new expr
//
// Output: Result on stack
//
//-----------------------------------------------------------------------------
#include "defs.h"
void
subst(void)
{
int i;
save();
p3 = pop(); // new expr
p2 = pop(); // old expr
p1 = pop(); // expr
if (p1->k == TENSOR) {
p4 = alloc_tensor(p1->u.tensor->nelem);
p4->u.tensor->ndim = p1->u.tensor->ndim;
for (i = 0; i < p1->u.tensor->ndim; i++)
p4->u.tensor->dim[i] = p1->u.tensor->dim[i];
for (i = 0; i < p1->u.tensor->nelem; i++) {
push(p1->u.tensor->elem[i]);
push(p2);
push(p3);
subst();
p4->u.tensor->elem[i] = pop();
}
push(p4);
} else if (equal(p1, p2))
push(p3);
else if (iscons(p1)) {
push(car(p1));
push(p2);
push(p3);
subst();
push(cdr(p1));
push(p2);
push(p3);
subst();
cons();
} else
push(p1);
restore();
}