Function get_equiv_sources

python/source.py:780–812

Given the fields along a slice, returns the equivalent sources as meep source objects for the beam.

(field, normal_vec, time_src, center, size)

def get_equiv_sources(field, normal_vec, time_src, center, size):
    """Given the fields along a slice, returns the equivalent sources as meep source objects for the beam."""
    # Get fields
    Ex, Ey, Ez, Hx, Hy, Hz = field
    nHat = normal_vec

    # Electric current K = nHat x H
    Kx = nHat[1] * Hz - nHat[2] * Hy
    Ky = nHat[2] * Hx - nHat[0] * Hz
    Kz = nHat[0] * Hy - nHat[1] * Hx

    # Mangnetic current N = - nHat x E
    Nx = nHat[2] * Ey - nHat[1] * Ez
    Ny = nHat[0] * Ez - nHat[2] * Ex
    Nz = nHat[1] * Ex - nHat[0] * Ey

    # Source components
    components = {mp.Ex: Kx, mp.Ey: Ky, mp.Ez: Kz, mp.Hx: Nx, mp.Hy: Ny, mp.Hz: Nz}

    # Make sources
    sources = [
        mp.Source(
            time_src,
            field_comp,
            center=center,
            size=size,
            amp_data=source_comp,
        )
        for field_comp, source_comp in components.items()
        if np.sum(np.abs(source_comp))
    ]

    return sources


class GaussianBeam2DSource(GaussianBeam3DSource):