132 lines
1.6 KiB
C++
132 lines
1.6 KiB
C++
//-----------------------------------------------------------------------------
|
|
//
|
|
// Author : philippe.billet@noos.fr
|
|
//
|
|
// tchebychevT polynomial T(x,n+1)=2*x*T(n,x)-T(x,n-1) T(x,0)=1,T(x,1)=x
|
|
//
|
|
// Input: tos-2 x (can be a symbol or expr)
|
|
//
|
|
// tos-1 n
|
|
//
|
|
// Output: Result on stack
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
|
|
#include "stdafx.h"
|
|
#include "defs.h"
|
|
|
|
void
|
|
eval_tchebychevT(void)
|
|
{
|
|
push(cadr(p1));
|
|
eval();
|
|
push(caddr(p1));
|
|
eval();
|
|
tchebychevT();
|
|
}
|
|
|
|
void
|
|
tchebychevT(void)
|
|
{
|
|
save();
|
|
yytchebychevT();
|
|
restore();
|
|
}
|
|
|
|
#define X p1
|
|
#define N p2
|
|
#define Y p3
|
|
#define Y1 p4
|
|
#define Y0 p5
|
|
|
|
void
|
|
yytchebychevT(void)
|
|
{
|
|
int n;
|
|
|
|
N = pop();
|
|
X = pop();
|
|
|
|
push(N);
|
|
n = pop_integer();
|
|
|
|
if (n < 0) {
|
|
push_symbol(TCHEBYCHEVT);
|
|
push(X);
|
|
push(N);
|
|
list(3);
|
|
return;
|
|
}
|
|
if (n == 0) {
|
|
push_integer(1);
|
|
return;
|
|
}
|
|
if (issymbol(X))
|
|
yytchebychevT2(n-1);
|
|
else {
|
|
Y = X; // do this when X is an expr
|
|
X = symbol(SECRETX);
|
|
yytchebychevT2(n-1);
|
|
X = Y;
|
|
push(symbol(SECRETX));
|
|
push(X);
|
|
subst();
|
|
eval();
|
|
}
|
|
}
|
|
|
|
void
|
|
yytchebychevT2(int n)
|
|
{
|
|
int i;
|
|
|
|
push(X);
|
|
push_integer(1);
|
|
Y1 = pop();
|
|
for (i = 0; i < n; i++) {
|
|
Y0 = Y1;
|
|
|
|
Y1 = pop();
|
|
|
|
push(X);
|
|
push(Y1);
|
|
multiply();
|
|
|
|
push_integer(2);
|
|
multiply();
|
|
|
|
push(Y0);
|
|
|
|
subtract();
|
|
|
|
}
|
|
}
|
|
|
|
static char *s[] = {
|
|
|
|
|
|
"tchebychevT(x,n)",
|
|
"tchebychevT(x,n)",
|
|
|
|
"tchebychevT(x,0)-1",
|
|
"0",
|
|
|
|
"tchebychevT(x,1)-x",
|
|
"0",
|
|
|
|
"tchebychevT(x,2)-(2x^2-1)",
|
|
"0",
|
|
|
|
"tchebychevT(x,3)-(4*x^3-3*x)",
|
|
"0",
|
|
|
|
"tchebychevT(x,4)-(8*x^4-8*x^2+1)",
|
|
"0"
|
|
|
|
};
|
|
|
|
void
|
|
test_tchebychevT(void)
|
|
{
|
|
test(__FILE__, s, sizeof s / sizeof (char *));
|
|
}
|