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

prettyify example plots

Browse Source
This commit is contained in:
Jan Petykiewicz 2019-10-27 12:46:12 -07:00
parent 2a1fa1df7b
commit 589d658853
1 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
examples/fdfd.py
View File

@ -188,25 +188,29 @@ def test1(solver=generic_solver):
pcolor(numpy.real(E[1][center[0], :, :]).T) pcolor(numpy.real(E[1][center[0], :, :]).T)
pyplot.subplot(2, 2, 2) pyplot.subplot(2, 2, 2)
pyplot.plot(numpy.log10(numpy.abs(E[1][:, center[1], center[2]]) + 1e-10)) pyplot.plot(numpy.log10(numpy.abs(E[1][:, center[1], center[2]]) + 1e-10))
pyplot.grid(alpha=0.6)
pyplot.ylabel('log10 of field')
pyplot.subplot(2, 2, 3) pyplot.subplot(2, 2, 3)
pcolor(numpy.real(E[1][:, :, center[2]]).T) pcolor(numpy.real(E[1][:, :, center[2]]).T)
pyplot.subplot(2, 2, 4) pyplot.subplot(2, 2, 4)
def poyntings(E): def poyntings(E):
H = functional.e2h(omega, dxes)(E) H = functional.e2h(omega, dxes)(E)
poynting = 0.5 * fdtd.poynting(e=E, h=H.conj()) * dx * dx poynting = fdtd.poynting(e=E, h=H.conj(), dxes=dxes)
cross1 = operators.poynting_e_cross(vec(E), dxes) @ vec(H).conj() cross1 = operators.poynting_e_cross(vec(E), dxes) @ vec(H).conj()
cross2 = operators.poynting_h_cross(vec(H), dxes) @ vec(E).conj() * -1 cross2 = operators.poynting_h_cross(vec(H), dxes) @ vec(E).conj() * -1
s1 = unvec(0.5 * numpy.real(cross1), grid.shape) s1 = 0.5 * unvec(numpy.real(cross1), grid.shape)
s2 = unvec(0.5 * numpy.real(cross2), grid.shape) s2 = 0.5 * unvec(numpy.real(cross2), grid.shape)
s0 = poynting.real s0 = 0.5 * poynting.real
# s2 = poynting.imag # s2 = poynting.imag
return s0, s1, s2 return s0, s1, s2
s0x, s1x, s2x = poyntings(E) s0x, s1x, s2x = poyntings(E)
pyplot.plot(s0x[0].sum(axis=2).sum(axis=1), label='s0') pyplot.plot(s0x[0].sum(axis=2).sum(axis=1), label='s0', marker='.')
pyplot.plot(s1x[0].sum(axis=2).sum(axis=1), label='s1') pyplot.plot(s1x[0].sum(axis=2).sum(axis=1), label='s1', marker='.')
pyplot.plot(s2x[0].sum(axis=2).sum(axis=1), label='s2') pyplot.plot(s2x[0].sum(axis=2).sum(axis=1), label='s2', marker='.')
pyplot.plot(E[1][:, center[1], center[2]].real.T, label='Ey', marker='x')
pyplot.grid(alpha=0.6)
pyplot.legend() pyplot.legend()
pyplot.show() pyplot.show()
@ -215,7 +219,8 @@ def test1(solver=generic_solver):
e_ovl_rolled = numpy.roll(e_overlap, i, axis=1) e_ovl_rolled = numpy.roll(e_overlap, i, axis=1)
q += [numpy.abs(vec(E) @ vec(e_ovl_rolled).conj())] q += [numpy.abs(vec(E) @ vec(e_ovl_rolled).conj())]
pyplot.figure() pyplot.figure()
pyplot.plot(q) pyplot.plot(q, marker='.')
pyplot.grid(alpha=0.6)
pyplot.title('Overlap with mode') pyplot.title('Overlap with mode')
pyplot.show() pyplot.show()
print('Average overlap with mode:', sum(q)/len(q)) print('Average overlap with mode:', sum(q)/len(q))
@ -227,6 +232,7 @@ def module_available(name):
if __name__ == '__main__': if __name__ == '__main__':
#test0() #test0()
# test1()
if module_available('opencl_fdfd'): if module_available('opencl_fdfd'):
from opencl_fdfd import cg_solver as opencl_solver from opencl_fdfd import cg_solver as opencl_solver