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

autodetect uniform dxes, and fix nonuniform dxes setup

Browse Source
This commit is contained in:
Jan Petykiewicz 2025-01-31 20:17:15 -08:00
parent 5b16295769
commit 2f37d97c3e
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
opencl_fdtd/simulation.py
View File

@ -143,9 +143,12 @@ class Simulation:
if isinstance(dxes, float | int):
uniform_dx = dxes
min_dx = dxes
elif all((dxes[0][0][0] == dx).all() for dx in dxes[0] + dxes[1]):
uniform_dx = dxes[0][0][0]
min_dx = dxes[0][0][0]
else:
uniform_dx = False
self.inv_dxes = [self._create_field(1 / dxn) for dxn in dxes[0] + dxes[1]]
self.inv_dxes = self._create_inv_dxes(dxes)
min_dx = min(min(dxn) for dxn in dxes[0] + dxes[1])
max_dt = min_dx * .99 / numpy.sqrt(3)
@ -371,6 +374,17 @@ class Simulation:
raise FDTDError('Initial field list elements must have same shape as epsilon elements')
return pyopencl.array.to_device(self.queue, vec(initial_value).astype(self.arg_type))
def _create_inv_dxes(self, dxes: list[list[NDArray]]) -> pyopencl.array.Array:
if len(dxes[0]) or len(dxes[1]) != 3:
raise FDTDError('dxes must be two lists of 3 1D ndarrays each')
if self.shape != tuple(len(dx) for dx in dxes[0]):
raise FDTDError('dxes must be compatible with shape of epsilon')
if self.shape != tuple(len(dx) for dx in dxes[1]):
raise FDTDError('dxes must be compatible with shape of epsilon')
self.inv_dxes = [
pyopencl.array.to_device(self.queue, (1 / dxn).astype(self.arg_type))
for dxn in chain(dxes[0], dxes[1])]
def type_to_C(
float_type: type,