rmatrices.py 文件源码

def quadaf(ev, y):
    L = ev[1]      # length
    k = ev[4]      # quadrupole strength
    if k == 0:
        R = drift(ev, y)
    else:
        wrzlk = sqrt(abs(k))
        Omega = wrzlk*L
        coshom = cosh(Omega)
        sinhom = sinh(Omega)
        cosom = cos(Omega)
        sinom = sin(Omega)
        R = array([
        [coshom,       sinhom/wrzlk, 0.,           0.,          0., 0],
        [wrzlk*sinhom, coshom,       0.,           0.,          0., 0],
        [0.,           0.,           cosom,        sinom/wrzlk, 0., 0],
        [0.,           0.,           -wrzlk*sinom, cosom,       0., 0],
        [0.,           0.,           0.,           0.,          1., L/(y**2)],
        [0.,           0.,           0.,           0.,          0., 1.]
        ])
    return R