test_linear_wakefield.py 文件源码

def Ez( z, r, t) :
    """
    Get the 2d Ez field

    Parameters
    ----------
    z : 1darray
    t, r : float
    """
    Nz = len(z)
    Nr = len(r)
    window_zmax = z.max()

    ez = np.zeros((Nz, Nr))
    for iz in range(Nz) :
        for ir in range(Nr) :
          ez[iz, ir] = quad( kernel_Ez, z[iz]-c*t, window_zmax-c*t,
            args = ( z[iz]-c*t, r[ir] ), limit=30 )[0]
    return( ez )