Definition at line 47 of file rk4.py. 00047 : 00048 00049 v=zeros(nvar, Float) 00050 vout=zeros(nvar,Float) 00051 dv=zeros(nvar,Float) 00052 00053 y=zeros((nstep+1, nvar),Float) 00054 xx=zeros(nstep+1,Float) 00055 00056 y[0,:]=vstart[:] 00057 v[:]=vstart[:] 00058 00059 xx[0]=x1 00060 x=x1 00061 h=(x2-x1)/nstep 00062 00063 for k in range(nstep): 00064 derivs(x, v, dv) 00065 rk4(v, dv, nvar, x, h, vout, derivs) 00066 x += h 00067 xx[k+1]=x 00068 v[:]=vout[:] 00069 y[k+1,:]=v[:] 00070 00071 return (xx,y) 00072
|