rmatrices.py 文件源码

def quadrf(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([
        [cosom,        sinom/wrzlk, 0.,           0.,           0., 0.],
        [-wrzlk*sinom, cosom,       0.,           0.,           0., 0.],
        [0.,            0.,         coshom,       sinhom/wrzlk, 0., 0.],
        [0.,            0.,         wrzlk*sinhom, coshom,       0., 0.],
        [0.,            0.,         0.,           0.,           1., L/(y**2)],
        [0.,            0.,         0.,           0.,           0., 1.]
        ])
    return R