/*  drmod.c    CCMATH mathematics library source code.
 *
 *  Copyright (C)  2000   Daniel A. Atkinson    All rights reserved.
 *  This code may be redistributed under the terms of the GNU library
 *  public license (LGPL). ( See the lgpl.license file for details.)
 * ------------------------------------------------------------------------
 */
#include "arma.h"
#include <stdlib.h>
extern struct mcof *par,*pma;
extern int nar,nma,np;
static int max,mxm,kst,kd; static double *pz,*pm;
double drmod(double y,double *dr)
{ struct mcof *p,*q; double yp,*pf,*pd,*pl;
  yp=0.; pl=dr+np;
  for(pf=dr,p=par,q=p+nar; p<q ;++p){
    *pf= *(pz+(kst+p->lag)%max);
    yp+= p->cf * *pf++;
   }
  for(p=pma,q=p+nma; p<q ;++p){
    *pf= *(pm+(kst+p->lag)%max);
    yp+= p->cf * *pf++;
   }
  if(nma){
    for(pf=dr,pd=pz+2*max; pf<pl ;++pd,++pf)
      for(p=pma,q=p+nma; p<q ;++p)
        *pf+= p->cf * *(pd+((kd+p->lag)%mxm)*np);
    kd=(kd+mxm-1)%mxm; pd=pz+2*max+np*kd;
    for(pf=dr; pf<pl ;) *pd++ = *pf++;
   }
  kst=(kst+max-1)%max; *(pm+kst)=(yp-=y);
  *(pz+kst)=y; return -yp;
}
void setdr(int k)
{ if(k){ max=mxm=kst=kd=0; np=nma+nar;
    if(nar) max=(par+nar-1)->lag+1;
    if(nma && (mxm=(pma+nma-1)->lag+1)>max) max=mxm;
    pz=(double *)calloc(2*max+np*mxm,sizeof(double));
    pm=pz+max;
   }
  else free(pz);
}


syntax highlighted by Code2HTML, v. 0.9.1