//Create minimal LFSR that outputs vec using Berlekamp-Massey algorithm
void LFSR::BerlekampMassey(const NTuple<symbol>& vec)
{
m_Coeff.SetSize(0);
m_State.SetSize(0); // for computation fm is stored in m_State
m_Coeff.Allocate(vec.GetSize()/2+6); // to minimize reallocations!
m_State.Allocate(vec.GetSize()/2+6);
// arraytype lprofil; lprofil.SetSize(0);
int m=-1;
symbol dm=1;
int n=vec.GetSize();
for (int k=0;k<n;k++) {
// Invariant: fk (m_Coeff) is feedback polynom with minimal degree
// calculate next discrepancy d
symbol d=0;
int len=0; if (k) len=m_Coeff.GetSize();
for (int i=0;i<len;i++) {
d=d+m_Coeff[i]*vec[k-i-1];
d=mod(d,m_Modulus);
}
d=mod(d-vec[k],m_Modulus);
// now adjust feedback polynom, if neccessary
if (d) {
int m2=-1;
arraytype fm2;
if ((k+1)<=(2*len)) ;
else {
// m=k; fm=m_Coeff; (part I)
m2=k;
fm2.Copy(m_Coeff);
// Lk+1=k+1-lk
m_Coeff.SetSize(k+1-len);
}
// fk+1 = fk - (dk dm^-1) x^(k-m)fm
int /*symbol*/ c=(m_Modulus==2) ? 1 : mod(-(int)(symbol)(d*dm),m_Modulus);
m_Coeff[k-m-1]=mod(m_Coeff[k-m-1]-c,m_Modulus);
for (int i=0;i<m_State.GetSize();i++)
m_Coeff[i+k-m]=mod(m_Coeff[i+k-m]+c*m_State[i],m_Modulus);
if (m2>=0) { // m=k; fm=m_Coeff; (part II)
m=m2;
m_State.Copy(fm2);
dm=minverse(d,m_Modulus);
}
} /* if(d) */
//lrprofil.Add(GetDegree()); // TODO Linearitaetsprofil
} /* for i */
// finally initialize m_State with the first Lk elements of vec
SetState(vec);
}
int as154_seas(double *inp, int N, int optmethod, int p, int d, int q, int s, int P, int D, int Q,
double *phi, double *theta, double *PHI, double *THETA, double *wmean,double *var,double *loglik,double *hess) {
int i, pq, retval, length, offset,ret;
double *b, *tf, *x,*inp2,*dx,*thess;
int *ipiv;
double maxstep;
alik_seas_object obj;
custom_function as154_min;
obj = alik_seas_init(p, d, q, s, P, D, Q, N);
inp2 = (double*)malloc(sizeof(double)* (N - s*D));
pq = obj->pq;
b = (double*)malloc(sizeof(double)* pq);
tf = (double*)malloc(sizeof(double)* pq);
thess = (double*)malloc(sizeof(double)* pq*pq);
dx = (double*)malloc(sizeof(double)* pq);
ipiv = (int*)malloc(sizeof(int)* pq);
length = N;
maxstep = 1.0;
css_seas(inp, N, optmethod, p, d, q, s, P, D, Q, phi, theta, PHI, THETA, wmean, var,loglik,hess);
/*
*/
if (D > 0) {
N = diffs(inp, N, D, s, inp2);
}
else {
for (i = 0; i < N; ++i) {
inp2[i] = inp[i];
}
}
x = (double*)malloc(sizeof(double)* (N - d));
if (d > 0) {
N = diff(inp2, N, d, x); // No need to demean x
}
else {
for (i = 0; i < N; ++i) {
x[i] = inp2[i];
}
}
obj->N = N;
offset = obj->offset;
for (i = 0; i < p; ++i) {
b[i] = phi[i];
}
for (i = 0; i < q; ++i) {
b[p + i] = -theta[i];
}
for (i = 0; i < P; ++i) {
b[p + q + i] = PHI[i];
}
for (i = 0; i < Q; ++i) {
b[p + q + P + i] = -THETA[i];
}
if (obj->M == 1) {
b[p + q + P + Q] = *wmean;
}
obj->mean = *wmean;
//mdisplay(b, 1, p + q + P + Q);
for (i = 0; i < N; ++i) {
obj->x[offset + i] = obj->x[offset + 2 * N + i] = x[i];
}
for (i = N; i < 2 * N; ++i) {
obj->x[offset + i] = 0.0;
}
//printf("\n %d %g ", pq,maxstep);
// custom_function as154_min = { fas154_seas, obj };
as154_min.funcpt = fas154_seas;
as154_min.params = obj;
retval = fminunc(&as154_min, NULL, pq, b, maxstep, optmethod, tf);
if (retval == 0) {
ret = 0;
}
else if (retval == 15) {
ret = 15;
}
else if (retval == 4) {
ret = 4;
}
else {
ret = 1;
}
for (i = 0; i < pq; ++i) {
dx[i] = 1.0;
}
hessian_fd(&as154_min, tf, pq, dx, obj->eps, hess);
mtranspose(hess, pq, pq, thess);
for (i = 0; i < pq*pq; ++i) {
thess[i] = (N - d - s*D) * 0.5 * (hess[i] + thess[i]);
}
ludecomp(thess, pq, ipiv);
minverse(thess, pq, ipiv, hess);
for (i = 0; i < p; ++i) {
phi[i] = tf[i];
}
for (i = 0; i < q; ++i) {
theta[i] = -tf[p + i];
}
for (i = 0; i < P; ++i) {
PHI[i] = tf[p + q + i];
}
for (i = 0; i < Q; ++i) {
THETA[i] = -tf[p + q + P + i];
}
if (obj->M == 1) {
*wmean = tf[p + q + Q + P];
}
else {
*wmean = 0.0;
}
*var = (obj->ssq) / (double) N;
*loglik = obj->loglik;
//printf("MEAN %g \n", mean(obj->x+N,N));
//mdisplay(obj->x + N, 1, N);
free(b);
free(tf);
free(inp2);
free(x);
free(dx);
free(thess);
free(ipiv);
free_alik_seas(obj);
return ret;
}
int as154(double *inp, int N, int optmethod, int p, int d, int q, double *phi, double *theta, double *wmean, double *var,double *resid,double *loglik,double *hess) {
int i,pq,retval,length,ret;
double *b,*tf,*x,*dx,*thess;
int *ipiv;
double maxstep;
alik_object obj;
custom_function as154_min;
x = (double*)malloc(sizeof(double)* (N - d));
length = N;
maxstep = 1.0;
obj = alik_init(p, d, q, N);
css(inp, N, optmethod, p, d, q, phi, theta, wmean, var,resid,loglik,hess);
if (d > 0) {
N = diff(inp, N, d, x); // No need to demean x
}
else {
for (i = 0; i < N; ++i) {
x[i] = inp[i];
}
}
obj->N = N;
obj->mean = *wmean;
pq = obj->pq;
b = (double*)malloc(sizeof(double)* pq);
tf = (double*)malloc(sizeof(double)* pq);
thess = (double*)malloc(sizeof(double)* pq*pq);
dx = (double*)malloc(sizeof(double)* pq);
ipiv = (int*)malloc(sizeof(int)* pq);
for (i = 0; i < p; ++i) {
b[i] = phi[i];
}
for (i = 0; i < q; ++i) {
b[p + i] = -theta[i];
}
if (obj->M == 1) {
b[p + q] = obj->mean;
}
for (i = 0; i < N; ++i) {
obj->x[i] = obj->x[2 * N + i] = x[i];
}
for (i = N; i < 2 * N; ++i) {
obj->x[i] = 0.0;
}
//20141121ÐÞ¸Ä
//custom_function as154_min = { fas154, obj };
as154_min.funcpt = fas154;
as154_min.params = obj;
retval = fminunc(&as154_min, NULL, pq, b,maxstep, optmethod, tf);
if (retval == 0) {
ret = 0;
}
else if (retval == 15) {
ret = 15;
}
else if (retval == 4) {
ret = 4;
}
else {
ret = 1;
}
for (i = 0; i < pq; ++i) {
dx[i] = 1.0;
}
hessian_fd(&as154_min, tf, pq, dx, obj->eps, hess);
mtranspose(hess, pq, pq, thess);
for (i = 0; i < pq*pq; ++i) {
thess[i] = (N-d) * 0.5 * (hess[i] + thess[i]);
}
ludecomp(thess, pq, ipiv);
minverse(thess, pq, ipiv, hess);
for (i = 0; i < p; ++i) {
phi[i] = tf[i];
}
for (i = 0; i < q; ++i) {
theta[i] = -tf[p + i];
}
if (obj->M == 1) {
*wmean = tf[p + q];
}
else {
*wmean = 0.0;
}
/*
wmean = 0.0;
for (i = 0; i < N; ++i) {
wmean += (obj->x[N + i] * obj->x[N + i]);
}*/
*var = (obj->ssq) / (double) N;
for (i = 0; i < N - d; ++i) {
resid[i] = obj->x[N + i];
}
*loglik = obj->loglik;
//printf("MEAN %g \n", mean(obj->x+N,N));
//mdisplay(obj->x + N, 1, N);
free(b);
free(tf);
free(x);
free(thess);
free(ipiv);
free(dx);
free_alik(obj);
return ret;
}
