fdtd.py

@@ -230,9 +230,9 @@ def main():
         Ectr[t] = sim.E[ind].get()
         u[t] = pyopencl.array.sum(sim.E * sim.E * sim.eps + h_old * sim.H).get() * dx * dx * dx
         ui[t] = (sim.E * sim.E * sim.eps + h_old * sim.H).reshape(epsilon.shape).get()[:, pml_thickness+m:-pml_thickness-m, :,
-                                                                                          pml_thickness+m:-pml_thickness-m].sum() * dx * dx * dx
+                                                                                            pml_thickness+m:-pml_thickness-m].sum() * dx * dx * dx
 #        ui[t] = (sim.E * sim.E * sim.eps + h_old * sim.H).reshape(epsilon.shape).get()[:, pml_thickness+m:-pml_thickness-m,
-#                                                                                          pml_thickness+m:-pml_thickness-m, :].sum() * dx * dx * dx
+#                                                                                            pml_thickness+m:-pml_thickness-m, :].sum() * dx * dx * dx
 
         if t % 100 == 0:
             avg = (t + 1) / (time.perf_counter() - start)

opencl_fdtd/__init__.py

@@ -1,5 +1,5 @@
 from .simulation import Simulation, type_to_C
 
 __author__ = 'Jan Petykiewicz'
-__version__ = '0.4'
+__vesion__ = '0.4'
 version = __version__

opencl_fdtd/simulation.py

@@ -2,7 +2,7 @@
 Class for constructing and holding the basic FDTD operations and fields
 """
 
-from typing import Callable, Type, Sequence
+from typing import List, Dict, Callable, Type, Union, Optional, Sequence
 from collections import OrderedDict
 import numpy
 from numpy.typing import NDArray
@@ -23,7 +23,7 @@ __author__ = 'Jan Petykiewicz'
 jinja_env = jinja2.Environment(loader=jinja2.PackageLoader(__name__.split('.')[0], 'kernels'))
 
 
-class Simulation:
+class Simulation(object):
     r"""
     Constructs and holds the basic FDTD operations and related fields
 
@@ -62,29 +62,29 @@ class Simulation:
     S: pyopencl.array.Array
     eps: pyopencl.array.Array
     dt: float
-    inv_dxes: list[pyopencl.array.Array]
+    inv_dxes: List[pyopencl.array.Array]
 
     arg_type: Type
 
     context: pyopencl.Context
     queue: pyopencl.CommandQueue
 
-    update_E: Callable[[list[pyopencl.Event]], pyopencl.Event]
-    update_H: Callable[[list[pyopencl.Event]], pyopencl.Event]
-    update_S: Callable[[list[pyopencl.Event]], pyopencl.Event]
-    update_J: Callable[[list[pyopencl.Event]], pyopencl.Event]
-    sources: dict[str, str]
+    update_E: Callable[[List[pyopencl.Event]], pyopencl.Event]
+    update_H: Callable[[List[pyopencl.Event]], pyopencl.Event]
+    update_S: Callable[[List[pyopencl.Event]], pyopencl.Event]
+    update_J: Callable[[List[pyopencl.Event]], pyopencl.Event]
+    sources: Dict[str, str]
 
     def __init__(
             self,
             epsilon: NDArray,
-            pmls: Sequence[dict[str, float]],
-            bloch_boundaries: Sequence[dict[str, float]] = (),
-            dxes: list[list[NDArray]] | float | None = None,
-            dt: float | None = None,
-            initial_fields: dict[str, NDArray] | None = None,
-            context: pyopencl.Context | None = None,
-            queue: pyopencl.CommandQueue | None = None,
+            pmls: Sequence[Dict[str, float]],
+            bloch_boundaries: Sequence[Dict[str, float]] = (),
+            dxes: Union[List[List[NDArray]], float, None] = None,
+            dt: Optional[float] = None,
+            initial_fields: Optional[Dict[str, NDArray]] = None,
+            context: Optional[pyopencl.Context] = None,
+            queue: Optional[pyopencl.CommandQueue] = None,
             float_type: Type = numpy.float32,
             do_poynting: bool = True,
             do_fieldsrc: bool = False,
@@ -331,8 +331,8 @@ class Simulation:
 
     def _create_context(
             self,
-            context: pyopencl.Context | None = None,
-            queue: pyopencl.CommandQueue | None = None,
+            context: Optional[pyopencl.Context] = None,
+            queue: Optional[pyopencl.CommandQueue] = None,
             ) -> None:
         if context is None:
             self.context = pyopencl.create_some_context()
@@ -353,7 +353,7 @@ class Simulation:
             raise Exception(f'Epsilon shape mismatch. Expected {self.shape}, got {epsilon[0].shape}')
         self.eps = pyopencl.array.to_device(self.queue, vec(epsilon).astype(self.arg_type))
 
-    def _create_field(self, initial_value: NDArray | None = None) -> pyopencl.array.Array:
+    def _create_field(self, initial_value: Optional[NDArray] = None) -> pyopencl.array.Array:
         if initial_value is None:
             return pyopencl.array.zeros_like(self.eps)
         else: