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

poynting divergence doesn't use dt, and can have default dxes

Browse Source
This commit is contained in:
Jan Petykiewicz 2019-07-17 00:50:49 -07:00
parent dd4e6f294f
commit 79e14af4db
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
fdfd_tools/fdtd.py
View File

@ -263,9 +263,13 @@ def poynting(e, h):
return numpy.array(s)
def poynting_divergence(dt, dxes, s=None, *, e=None, h=None): # TODO dxes
def poynting_divergence(s=None, *, e=None, h=None, dxes=None): # TODO dxes
if dxes is None:
dxes = tuple(tuple(numpy.ones(1) for _ in range(3)) for _ in range(2))
if s is None:
s = poynting(e, h)
ds = ((s[0] - numpy.roll(s[0], 1, axis=0)) / numpy.sqrt(dxes[0][0] * dxes[1][0])[:, None, None] +
(s[1] - numpy.roll(s[1], 1, axis=1)) / numpy.sqrt(dxes[0][1] * dxes[1][1])[None, :, None] +
(s[2] - numpy.roll(s[2], 1, axis=2)) / numpy.sqrt(dxes[0][2] * dxes[1][2])[None, None, :] )