jan/fdfd_tools
2
2
Fork 1
Code Issues Pull Requests Releases Wiki Activity

Add test code for PEC

Browse Source
This commit is contained in:
jan 2016-07-03 14:23:51 -07:00
parent 28f85712ce
commit a512a88930
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
examples/test.py
View File

@ -177,7 +177,12 @@ def test1():
J = waveguide_mode.compute_source(**wg_args, **wg_results) J = waveguide_mode.compute_source(**wg_args, **wg_results)
H_overlap = waveguide_mode.compute_overlap_e(**wg_args, **wg_results) H_overlap = waveguide_mode.compute_overlap_e(**wg_args, **wg_results)
A = fdfd_tools.operators.e_full(omega, dxes, vec(grid.grids)).tocsr() pecg = gridlock.Grid(edge_coords, initial=0, num_grids=3)
# pecg.draw_cuboid(center=[700, 0, 0], dimensions=[80, 1e8, 1e8], eps=1)
pecg.grids = [numpy.sign(r) for r in pecg.grids]
# pecg.visualize_isosurface()
A = fdfd_tools.operators.e_full(omega, dxes, vec(grid.grids), pec=vec(pecg.grids)).tocsr()
b = -1j * omega * vec(J) b = -1j * omega * vec(J)
x = solve_A(A, b) x = solve_A(A, b)
E = unvec(x, grid.shape) E = unvec(x, grid.shape)