Add symmetry planes for simulation to filter TE and TM for optics, or even and odd modes for RF about femwell HOT 3 CLOSED

from collections import OrderedDict

import matplotlib.pyplot as plt
import numpy as np
import scipy.constants
import shapely
from shapely.ops import clip_by_rect
from shapely.geometry import LineString, box
from skfem import Basis, ElementTriP0, ElementTriP1
from skfem.io.meshio import from_meshio
from tqdm import tqdm

from femwell.maxwell.waveguide import compute_modes
from femwell.mesh import mesh_from_OrderedDict
from femwell.visualization import plot_domains


wavelength = 1.55 #um
num_modes = 8
width = 4 #um

#Find modes with full mesh
core = box(0,0,width/2, 1)
polygons = OrderedDict(
    core = core,
    box = clip_by_rect(core.buffer(2, resolution = 5),
                       -np.inf,
                      -np.inf,
                      np.inf,
                      0),
    clad = clip_by_rect(core.buffer(2, resolution = 5),
                       -np.inf,
                      0,
                      np.inf,
                      np.inf),

)

resolutions = {"core": {"resolution": 0.1, "distance": 1}}

mesh = from_meshio(mesh_from_OrderedDict(polygons, resolutions, default_resolution_max=0.6))

basis0 = Basis(mesh, ElementTriP0())
epsilon = basis0.zeros(dtype = complex)

for subdomain, n in {"core": 1.9963, "box": 1.444, "clad": 1}.items():
    epsilon[basis0.get_dofs(elements = subdomain)] = n**2
    
modes = compute_modes(basis0, 
                      epsilon, 
                      wavelength=wavelength, 
                      num_modes=num_modes,
                      metallic_boundaries=True)

modes_full = modes
### Find modes with symmetry plane
polygons = OrderedDict(
    core = core,
    box = clip_by_rect(core.buffer(2, resolution = 5),
                       0,
                      -np.inf,
                      np.inf,
                      0),
    clad = clip_by_rect(core.buffer(2, resolution = 5),
                       0,
                      0,
                      np.inf,
                      np.inf),

)

resolutions = {"core": {"resolution": 0.1, "distance": 1}}

mesh = from_meshio(mesh_from_OrderedDict(polygons, resolutions, default_resolution_max=0.6))

basis0 = Basis(mesh, ElementTriP0())
epsilon = basis0.zeros(dtype = complex)

for subdomain, n in {"core": 1.9963, "box": 1.444, "clad": 1}.items():
    epsilon[basis0.get_dofs(elements = subdomain)] = n**2
    
modes = compute_modes(basis0, 
                      epsilon, 
                      wavelength=wavelength, 
                      num_modes=num_modes,
                      metallic_boundaries=True)

modes_sym = modes

print(modes_full.n_effs[0])
print(modes_sym.n_effs[0])

modes_full[0].plot_component('E', 'x')
modes_sym[0].plot_component('E', 'x')