test_linear_wakefield.py 文件源码

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

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

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

    return( er )

# ---------------------------
# Comparison plots
# ---------------------------