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

use fdmath derivatives where possible

Browse Source
This commit is contained in:
Jan Petykiewicz 2019-11-25 00:04:53 -08:00
parent e69b0f61ec
commit e5db39fd49
1 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
meanas/fdtd/energy.py
View File

@ -2,8 +2,7 @@
from typing import List, Callable, Tuple, Dict from typing import List, Callable, Tuple, Dict
import numpy import numpy
from .. import dx_lists_t, field_t, field_updater from .. import dx_lists_t, field_t, field_updater, fdmath
def poynting(e: field_t, def poynting(e: field_t,
h: field_t, h: field_t,
@ -35,10 +34,8 @@ def poynting_divergence(s: field_t = None,
if s is None: if s is None:
s = poynting(e, h, dxes=dxes) s = poynting(e, h, dxes=dxes)
#TODO use deriv operators Dx, Dy, Dz = fdmath.functional.deriv_back()
ds = ((s[0] - numpy.roll(s[0], 1, axis=0)) + ds = Dx(s[0]) + Dy(s[1]) + Dz(s[2])
(s[1] - numpy.roll(s[1], 1, axis=1)) +
(s[2] - numpy.roll(s[2], 1, axis=2)))
return ds return ds