LightWave2D 0.0.6__tar.gz → 0.0.7__tar.gz

lightwave2d-0.0.7/LightWave2D/VERSION

@@ -0,0 +1 @@
+0.0.7

{lightwave2d-0.0.6 → lightwave2d-0.0.7}/LightWave2D/components.py

@@ -36,12 +36,14 @@ class BaseComponent():
         edgecolor (str): The color of the component's edge.
         alpha (float): Transparency level of the component.
         rotation (float): Rotation angle of the component.
+        sigma (float): Conductivity of the component.
     """
     grid: Grid
     facecolor: str = 'lightblue'
     edgecolor: str = 'blue'
     alpha: float = 0.3
     rotation: float = 0
+    sigma: float = 0
 
     def __post_init__(self):
         x0, y0 = self.position
@@ -63,14 +65,10 @@ class BaseComponent():
 
         self.path = Path(self.polygon.exterior.coords)
 
-        x_mesh, y_mesh = numpy.meshgrid(self.grid.x_stamp, self.grid.y_stamp)
-
-        coordinates = numpy.c_[x_mesh.T.flatten(), y_mesh.T.flatten()]
+        coordinates = numpy.c_[self.grid.x_mesh.T.flatten(), self.grid.y_mesh.T.flatten()]
 
         self.idx = self.path.contains_points(coordinates).astype(bool).reshape(self.epsilon_r_mesh.shape)
 
-        self.epsilon_r_mesh[self.idx] = self.epsilon_r
-
     def add_to_ax(self, ax: plt.axis) -> PatchCollection:
         """
         Add the component to the provided axis.
@@ -107,14 +105,23 @@ class BaseComponent():
 
         plt.show()
 
-    def add_to_mesh(self, epsilon_r_mesh: numpy.ndarray) -> NoReturn:
+    def add_to_epsilon_r_mesh(self, epsilon_r_mesh: numpy.ndarray) -> NoReturn:
         """
         Add the component's permittivity to the provided mesh.
 
         Args:
             epsilon_r_mesh (np.ndarray): The permittivity mesh to be updated.
         """
-        epsilon_r_mesh += self.epsilon_r_mesh
+        epsilon_r_mesh[self.idx] = self.epsilon_r
+
+    def add_to_sigma_mesh(self, sigma_mesh: numpy.ndarray) -> NoReturn:
+        """
+        Add the component's sigma to the provided mesh.
+
+        Args:
+            sigma_mesh (np.ndarray): The permittivity mesh to be updated.
+        """
+        sigma_mesh[self.idx] = self.sigma
 
     def add_non_linear_effect_to_field(self, field: numpy.ndarray) -> NoReturn:
         """
@@ -123,7 +130,8 @@ class BaseComponent():
         Args:
             field (np.ndarray): The field to which non-linear effects will be added.
         """
-        chi_2 = 1e10
+        return
+        chi_2 = 0
 
         field += self.idx * self.grid.dt**2 / (self.epsilon_r_mesh * Physics.epsilon_0 * Physics.mu_0) * chi_2 * field ** 2
 

{lightwave2d-0.0.6 → lightwave2d-0.0.7}/LightWave2D/experiment.py

@@ -211,6 +211,11 @@ class Experiment:
             sigma_x, sigma_y = self.pml.sigma_x, self.pml.sigma_y
         else:
             sigma_x = sigma_y = numpy.zeros(self.grid.shape)
+
+        for component in self.components:
+            component.add_to_sigma_mesh(sigma_x)
+            component.add_to_sigma_mesh(sigma_y)
+
         return sigma_x, sigma_y
 
     def get_epsilon(self) -> numpy.ndarray:
@@ -222,7 +227,7 @@ class Experiment:
         """
         epsilon_r_mesh = numpy.ones(self.grid.shape)
         for component in self.components:
-            component.add_to_mesh(epsilon_r_mesh)
+            component.add_to_epsilon_r_mesh(epsilon_r_mesh)
 
         return epsilon_r_mesh * Physics.epsilon_0
 
@@ -264,11 +269,15 @@ class Experiment:
 
             Ez[1:-1, 1:-1] += eps_factor[1:-1, 1:-1] * (dHy_dx - dHx_dy)
 
+            # Apply absorption (losses) to the electric field Ez
+            absorption_factor = 1 - (sigma_x + sigma_y) * eps_factor / 2
+            absorption_factor = numpy.clip(absorption_factor, 0, 1)  # Absorption cliped to [0, 1] to ensure stability
+
+            Ez *= absorption_factor
+
             for component in self.components:
                 component.add_non_linear_effect_to_field(Ez)
 
-            Ez *= (1 - (sigma_x + sigma_y) * eps_factor / 2)
-
             for source in self.sources:
                 source.add_source_to_field(Ez, time=t)
 
@@ -342,6 +351,7 @@ class Experiment:
             frame_number (int): The index of the frame to be saved.
             filename (str): The file path where the image will be saved.
             dpi (int): The resolution of the saved image in dots per inch.
+            show_intensity (bool): If True, the intensity is shown instead of amplitude
             colormap (Optional[Union[str, object]]): The colormap used for visualizing the data. Defaults to a predefined blue-black-red colormap.
 
         Returns:
@@ -394,6 +404,7 @@ class Experiment:
             skip_frame: int = 10,
             scale_max: float = 5,
             unit_size: int = 6,
+            auto_adjust_clim: bool = False,
             colormap: Optional[Union[str, object]] = colormaps.blue_black_red) -> animation.FuncAnimation:
         """
         Renders the propagation of a field as an animation using matplotlib.
@@ -403,6 +414,7 @@ class Experiment:
 
         Args:
             skip_frame (int): The number of time steps to skip between frames in the animation.
+            auto_adjust_clim (bool): TODO
             colormap (Optional[Union[str, object]]): The colormap used for visualizing the data. Defaults to a predefined blue-black-red colormap.
             unit_size (float): The base unit size for scaling the plot elements.
 
@@ -437,6 +449,11 @@ class Experiment:
             """
             field_t = self.Ez_t[frame].T
             field_artist.set_array(field_t)
+
+            if auto_adjust_clim:
+                max_amplitude = abs(field_t).max() / scale_max
+                field_artist.set_clim(vmin=-max_amplitude, vmax=max_amplitude)
+
             return [artist for artist in artist_list]
 
         def init_func():

{lightwave2d-0.0.6 → lightwave2d-0.0.7}/LightWave2D/grid.py

@@ -52,6 +52,8 @@ class Grid:
         self.x_stamp = numpy.arange(self.n_x) * self.dx
         self.y_stamp = numpy.arange(self.n_y) * self.dy
 
+        self.x_mesh, self.y_mesh = numpy.meshgrid(self.x_stamp, self.y_stamp)
+
         self.polygon = geo.Polygon([
             (self.x_stamp[0], self.y_stamp[0]),
             (self.x_stamp[0], self.y_stamp[-1]),

{lightwave2d-0.0.6 → lightwave2d-0.0.7}/LightWave2D/pml.py

@@ -47,7 +47,6 @@ class PML:
         opposite_width = f"{100 - opposite_width}%"
         self.width_start = self.grid.get_coordinate(x=self.width, y=self.width)
         self.width_stop = self.grid.get_coordinate(x=opposite_width, y=opposite_width)
-        self.width_stop_ = self.grid.get_coordinate(x="100%", y="100%")
 
         bottom_boundary = y_mesh < self.width_start.y
         left_boundary = x_mesh < self.width_start.x

{lightwave2d-0.0.6 → lightwave2d-0.0.7}/LightWave2D.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: LightWave2D
-Version: 0.0.6
+Version: 0.0.7
 Summary: A package for finite-difference time domain computation for light propagation.
 Home-page: https://github.com/MartinPdeS/LightWave2D
 Author: Martin Poinsinet de Sivry

{lightwave2d-0.0.6 → lightwave2d-0.0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: LightWave2D
-Version: 0.0.6
+Version: 0.0.7
 Summary: A package for finite-difference time domain computation for light propagation.
 Home-page: https://github.com/MartinPdeS/LightWave2D
 Author: Martin Poinsinet de Sivry

lightwave2d-0.0.6/LightWave2D/VERSION

@@ -1 +0,0 @@
-0.0.6