shuren/fdfd_tools
1
0
Fork 0
forked from jan/fdfd_tools
Code Issues Pull Requests Releases Wiki Activity

use opencl solver (for testing)

Browse Source
This commit is contained in:
jan 2016-07-04 16:35:28 -07:00
parent 8daab636ea
commit eb4d9be6cf
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
examples/test.py
View File

@ -10,6 +10,8 @@ import gridlock
from matplotlib import pyplot
from opencl_fdfd import cg_solver
__author__ = 'Jan Petykiewicz'
@ -185,16 +187,25 @@ def test1():
# pmcg.draw_cuboid(center=[700, 0, 0], dimensions=[80, 1e8, 1e8], eps=1)
# pmcg.visualize_isosurface()
'''
Solve!
'''
A = fdfd_tools.operators.e_full(omega, dxes,
epsilon=vec(grid.grids),
pec=vec(pecg.grids),
pmc=vec(pmcg.grids)).tocsr()
b = -1j * omega * vec(J)
x = solve_A(A, b)
# x = solve_A(A, b)
x = cg_solver(omega, dxes, J=vec(J), epsilon=vec(grid.grids),
pec=vec(pecg.grids), pmc=vec(pmcg.grids))
E = unvec(x, grid.shape)
print('Norm of the residual is ', numpy.linalg.norm(A @ x - b))
'''
Plot results
'''
def pcolor(v):
vmax = numpy.max(numpy.abs(v))
pyplot.pcolor(v, cmap='seismic', vmin=-vmax, vmax=vmax)