Compactify args
This commit is contained in:
parent
10e275611d
commit
c92656bed8
@ -127,19 +127,16 @@ def test1(solver=generic_solver):
|
|||||||
grid.pos2ind(dims[1], which_shifts=None).astype(int))
|
grid.pos2ind(dims[1], which_shifts=None).astype(int))
|
||||||
src_axis = 0
|
src_axis = 0
|
||||||
wg_args = {
|
wg_args = {
|
||||||
'omega': omega,
|
|
||||||
'slices': [slice(i, f+1) for i, f in zip(*ind_dims)],
|
'slices': [slice(i, f+1) for i, f in zip(*ind_dims)],
|
||||||
'dxes': dxes,
|
'dxes': dxes,
|
||||||
'axis': src_axis,
|
'axis': src_axis,
|
||||||
'polarity': +1,
|
'polarity': +1,
|
||||||
'epsilon': grid.grids,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
wg_results = waveguide_mode.solve_waveguide_mode(mode_number=0, **wg_args)
|
wg_results = waveguide_mode.solve_waveguide_mode(mode_number=0, omega=omega, epsilon=grid.grids, **wg_args)
|
||||||
J = waveguide_mode.compute_source(**wg_args, E=wg_results['E'], wavenumber=wg_results['wavenumber'])
|
J = waveguide_mode.compute_source(E=wg_results['E'], wavenumber=wg_results['wavenumber'],
|
||||||
e_overlap = waveguide_mode.compute_overlap_e(E=wg_results['E'], wavenumber=wg_results['wavenumber'],
|
omega=omega, epsilon=grid.grids, **wg_args)
|
||||||
dxes=dxes, axis=src_axis, polarity=wg_args['polarity'],
|
e_overlap = waveguide_mode.compute_overlap_e(E=wg_results['E'], wavenumber=wg_results['wavenumber'], **wg_args)
|
||||||
slices=wg_args['slices'])
|
|
||||||
|
|
||||||
pecg = gridlock.Grid(edge_coords, initial=0.0, num_grids=3)
|
pecg = gridlock.Grid(edge_coords, initial=0.0, num_grids=3)
|
||||||
# pecg.draw_cuboid(center=[700, 0, 0], dimensions=[80, 1e8, 1e8], eps=1)
|
# pecg.draw_cuboid(center=[700, 0, 0], dimensions=[80, 1e8, 1e8], eps=1)
|
||||||
|
Loading…
Reference in New Issue
Block a user