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

Add poynting halves

Browse Source 
maybe will remove this later... dunno if it's actually useful
This commit is contained in:
Jan Petykiewicz 2019-07-30 00:33:07 -07:00
parent cf0db63a3f
commit 23d65d6ebb
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
opencl_fdtd/simulation.py
View File

@ -83,6 +83,7 @@ class Simulation(object):
queue: pyopencl.CommandQueue = None,
float_type: numpy.float32 or numpy.float64 = numpy.float32,
do_poynting: bool = True,
do_poynting_halves = False,
do_fieldsrc: bool = False):
"""
Initialize the simulation.
@ -119,6 +120,7 @@ class Simulation(object):
in complexity.
* GPU memory requirements are approximately doubled, since S and the intermediate
products must be stored.
:param do_poynting_halves: TODO DOCUMENT
"""
if initial_fields is None:
initial_fields = {}
@ -202,6 +204,7 @@ class Simulation(object):
'common_header': common_source,
'pmls': pmls,
'do_poynting': do_poynting,
'do_poynting_halves': do_poynting_halves,
'bloch': bloch_boundaries,
'uniform_dx': uniform_dx,
}
@ -227,6 +230,12 @@ class Simulation(object):
if do_poynting:
self.S = pyopencl.array.zeros_like(self.E)
S_fields[ptr('S')] = self.S
if do_poynting_halves:
self.S0 = pyopencl.array.zeros_like(self.E)
self.S1 = pyopencl.array.zeros_like(self.E)
S_fields[ptr('S0')] = self.S0
S_fields[ptr('S1')] = self.S1
J_fields = OrderedDict()
if do_fieldsrc: