CODEX Play TIme

ascii.txt

@@ -0,0 +1,27 @@
+
+        @@@@@@@: @@@@@@@   =@@@    @@@#   @@@@        
+        @@@ :@@% @@@       @@@@@   @@@@. %@@@@        
+        @@@@@@@- @@@@@@*  @@@-@@.  @@=@@+@@%@@        
+        @@@  @@@ @@@     @@@@@@@@  @@ %@@@:%@@        
+        @@@@@@@: @@@@@@@+@@    @@@.@@  @@@ #@@        
+
+        *@@@@@@@@@%*=::::---:::---::---+%@*=*+        
+        -%#-=@@@@@@#+=------==-=%%+::::..-@@#         
+        *@+-*@+=#++*#***+===++%@@@=    .*@=-%-        
+        #@@@+.=--#@:         -+-+.   .=%:  -@@        
+        #@@---:=@-          #=#=  ..-%-   @@@         
+        #@@*=+@=          #+**:..-=*-   *%*           
+        *@@%@+          +*##-::-+#+   +#+             
+         @@+          *%#%*+==+**   =+=               
+                   -%@%@@%#***#.  :+=.                
+                 .@@@@@@%%%#%=   #=-                  
+                 %@@@@@@@@@#  .*==                    
+               +%@@@@@@@@@- .**+.          @*-        
+             =*%@@@@@@@@*. +%#-          *@@@#        
+           =%%@#%@@@@@#..+#%=          =**@@--        
+         *@@@%+%@@@@@:.:*%*          -#=@+ =::        
+        @@@@:-*@@@@=.:*#*          -%=%%   +::        
+        =#@--+%@@=..@%%%**+=--==*##%@@.   .**-        
+        -#@#%@@@#*+=#@*=++==-==+**#%=:.:=+%@@=        
+        @@*@@%+*#%@@@%+-:.......::-+=:=#@#@@@@        
+

beamz/__init__.py

@@ -15,16 +15,19 @@
     Rectangle, Circle, Ring, 
     CircularBend, Polygon, Taper
 )
-from beamz.design.sources import ModeSource, GaussianSource
-from beamz.design.monitors import Monitor
-from beamz.design.signals import ramped_cosine, plot_signal
-from beamz.design.mode import solve_modes, slab_mode_source
+from beamz.devices.sources import ModeSource, GaussianSource
+from beamz.devices.monitors import Monitor
+from beamz.devices.signals import ramped_cosine, plot_signal
+from beamz.devices.mode import solve_modes, slab_mode_source
 
 # Import simulation-related classes and functions
 from beamz.simulation.meshing import RegularGrid
 from beamz.simulation.fdtd import FDTD
 from beamz.simulation.backends import get_backend
 
+from beamz.optimization.optimizers import Optimizer
+from beamz.optimization.topology import compute_overlap_gradient
+
 # Import optimization-related classes
 # (Currently empty, to be filled as the module grows)
 
@@ -81,6 +84,10 @@
     'RegularGrid': RegularGrid,
     'FDTD': FDTD,
     'get_backend': get_backend,
+
+    # Optimization
+    'Optimizer': Optimizer,
+    'compute_overlap_gradient': compute_overlap_gradient,
 
     # UI helpers
     'display_header': display_header,

beamz/design/__init__.py

@@ -8,10 +8,10 @@
     Rectangle, Circle, Ring, CircularBend, Polygon, Taper
 )
 from beamz.design.pml import PML
-from beamz.design.sources import ModeSource, GaussianSource
-from beamz.design.monitors import Monitor
-from beamz.design.signals import ramped_cosine, plot_signal
-from beamz.design.mode import solve_modes, slab_mode_source
+from beamz.devices.sources import ModeSource, GaussianSource
+from beamz.devices.monitors import Monitor
+from beamz.devices.signals import ramped_cosine, plot_signal
+from beamz.devices.mode import solve_modes, slab_mode_source
 
 __all__ = [
     'Material', 'CustomMaterial',

beamz/design/core.py

@@ -7,8 +7,8 @@
 from beamz.helpers import display_status
 
 from beamz.design.materials import Material
-from beamz.design.sources import ModeSource, GaussianSource
-from beamz.design.monitors import Monitor
+from beamz.devices.sources import ModeSource, GaussianSource
+from beamz.devices.monitors import Monitor
 from beamz.design.structures import Polygon, Rectangle, Circle, Ring, CircularBend, Taper
 from beamz.design.pml import PML
 
@@ -178,8 +178,8 @@ def unify_polygons(self):
                 non_polygon_structures.append(structure)
                 continue
             # Import at function level to avoid circular imports
-            from beamz.design.sources import ModeSource, GaussianSource
-            from beamz.design.monitors import Monitor
+            from beamz.devices.sources import ModeSource, GaussianSource
+            from beamz.devices.monitors import Monitor
             if isinstance(structure, (ModeSource, GaussianSource, Monitor)):
                 non_polygon_structures.append(structure)
                 continue

beamz/design/structures.py

@@ -23,7 +23,7 @@ def _process_vertices(self, vertices, z=0):
             vertices_3d = [(x, y, vertices_3d[i][2] if len(vertices_3d[i]) > 2 else z) 
                           for i, (x, y) in enumerate(vertices_2d)]
         return vertices_3d
-    
+
     def _process_vertices_preserve_orientation(self, vertices, z=0):
         if not vertices: return []
         vertices_3d = self._ensure_3d_vertices(vertices)
@@ -345,7 +345,7 @@ def add_to_plot(self, ax, facecolor=None, edgecolor="black", alpha=None, linesty
         if alpha is None: alpha = 1
         if linestyle is None: linestyle = '-'
         return super().add_to_plot(ax, facecolor=facecolor, edgecolor=edgecolor, alpha=alpha, linestyle=linestyle)
-    
+
     def copy(self):
         return Ring(position=self.position, 
                     inner_radius=self.inner_radius, 
@@ -412,7 +412,7 @@ def add_to_plot(self, ax, facecolor=None, edgecolor="black", alpha=None, linesty
         if linestyle is None: linestyle = '-'
         # Use parent polygon drawing
         return super().add_to_plot(ax, facecolor=facecolor, edgecolor=edgecolor, alpha=alpha, linestyle=linestyle)
-
+        
     def copy(self):
         return CircularBend(self.position, self.inner_radius, self.outer_radius, 
                             self.angle, self.rotation, self.material, self.color, self.optimize, 

beamz/design/monitors.py → beamz/devices/monitors.py

@@ -1,17 +1,24 @@
+from typing import Callable, Optional
+
 import numpy as np
 import matplotlib.pyplot as plt
 from matplotlib.patches import Rectangle as MatplotlibRectangle
 
-class Monitor():    
-    def __init__(self, design=None, start=(0,0), end=None, plane_normal=None, plane_position=0, 
-                 size=None, record_fields=True, accumulate_power=True, live_update=False, 
-                 record_interval=1, max_history_steps=None):
+class Monitor():
+    def __init__(self, design=None, start=(0,0), end=None, plane_normal=None, plane_position=0,
+                 size=None, record_fields=True, accumulate_power=True, live_update=False,
+                 record_interval=1, max_history_steps=None,
+                 objective_function: Optional[Callable[["Monitor"], float]] = None,
+                 name: Optional[str] = None):
         self.design = design
         self.should_record_fields = record_fields
         self.accumulate_power = accumulate_power
         self.live_update = live_update
         self.record_interval = record_interval
         self.max_history_steps = max_history_steps
+        self.objective_function = objective_function
+        self.objective_value: Optional[float] = None
+        self.name = name
 
         # Determine if this is a 3D monitor based on input parameters
         self.is_3d = self._determine_3d_mode(start, end, design)
@@ -32,6 +39,7 @@ def __init__(self, design=None, start=(0,0), end=None, plane_normal=None, plane_
         self.power_accumulated = None
         self.energy_history = []
         self.power_history = []
+        self.power_timestamps = []
         self.power_accumulation_count = 0
         # Recording control
         self.step_count = 0
@@ -44,8 +52,26 @@ def __init__(self, design=None, start=(0,0), end=None, plane_normal=None, plane_
 
         if self.is_3d: 
             self._init_3d_monitor(start, end, plane_normal, plane_position, size)
-        else: 
+        else:
             self._init_2d_monitor(start, end)
+
+    def evaluate_objective(self) -> Optional[float]:
+        """Evaluate the objective function associated with this monitor, if any."""
+        if self.objective_function is None:
+            return None
+        try:
+            value = self.objective_function(self)
+        except Exception as exc:
+            print(f"Warning: monitor objective evaluation failed: {exc}")
+            return None
+        if value is None:
+            return None
+        try:
+            self.objective_value = float(value)
+        except (TypeError, ValueError):
+            print(f"Warning: monitor objective returned non-numeric value: {value}")
+            return None
+        return self.objective_value
 
     def _determine_3d_mode(self, start, end, design):
         """Determine if this should be a 3D monitor based on inputs."""
@@ -280,7 +306,7 @@ def record_fields_2d(self, Ez, Hx, Hy, t, dx, dy, step=0):
             self.fields['t'].append(t)
 
         if self.accumulate_power:
-            self._calculate_power_2d(Ez_values, Hx_values, Hy_values)
+            self._calculate_power_2d(Ez_values, Hx_values, Hy_values, t)
 
         self.last_record_step = step
         self._manage_memory()
@@ -361,7 +387,7 @@ def slice_field(arr):
             self.fields['t'].append(t)
 
         if self.accumulate_power:
-            self._calculate_power_3d(Ex_slice, Ey_slice, Ez_slice, Hx_slice, Hy_slice, Hz_slice)
+            self._calculate_power_3d(Ex_slice, Ey_slice, Ez_slice, Hx_slice, Hy_slice, Hz_slice, t)
 
         self.last_record_step = step
         self._manage_memory()
@@ -378,7 +404,7 @@ def record_fields(self, *args, **kwargs):
             # 2D: Ez, Hx, Hy, t, dx, dy, step
             self.record_fields_2d(*args, **kwargs)
 
-    def _calculate_power_2d(self, Ez_values, Hx_values, Hy_values):
+    def _calculate_power_2d(self, Ez_values, Hx_values, Hy_values, t):
         """Calculate Poynting vector and power for 2D fields."""
         Ez_array = np.array(Ez_values)
         Hx_array = np.array(Hx_values)
@@ -394,9 +420,10 @@ def _calculate_power_2d(self, Ez_values, Hx_values, Hy_values):
         else:
             self.power_accumulated += power_mag
         self.power_history.append(total_power)
+        self.power_timestamps.append(float(t))
         self.power_accumulation_count += 1
-    
-    def _calculate_power_3d(self, Ex, Ey, Ez, Hx, Hy, Hz):
+
+    def _calculate_power_3d(self, Ex, Ey, Ez, Hx, Hy, Hz, t):
         """Calculate Poynting vector and power for 3D fields."""
         # Poynting vector S = E × H
         Sx = Ey * Hz - Ez * Hy
@@ -408,6 +435,7 @@ def _calculate_power_3d(self, Ex, Ey, Ez, Hx, Hy, Hz):
         if self.power_accumulated is None: self.power_accumulated = power_mag.copy()
         else: self.power_accumulated += power_mag
         self.power_history.append(total_power)
+        self.power_timestamps.append(float(t))
         self.power_accumulation_count += 1
 
     def _manage_memory(self):
@@ -422,6 +450,7 @@ def _manage_memory(self):
         if len(self.power_history) > self.max_history_steps:
             excess = len(self.power_history) - self.max_history_steps
             self.power_history = self.power_history[excess:]
+            self.power_timestamps = self.power_timestamps[excess:]
 
     def start_live_visualization(self, field_component='Ez'):
         """Start live field visualization."""
@@ -540,17 +569,24 @@ def get_field_statistics(self):
     def save_data(self, filename, format='npz'):
         """Save recorded data to file."""
         if format == 'npz':
-            np.savez(filename, 
-                    fields=self.fields,
-                    power_history=self.power_history,
-                    monitor_info={'type': self.monitor_type, 'is_3d': self.is_3d})
+            np.savez(
+                filename,
+                fields=self.fields,
+                power_history=self.power_history,
+                power_timestamps=self.power_timestamps,
+                monitor_info={'type': self.monitor_type, 'is_3d': self.is_3d}
+            )
         else: raise ValueError(f"Unsupported format: {format}")
 
     def load_data(self, filename):
         """Load data from file."""
         data = np.load(filename, allow_pickle=True)
         self.fields = data['fields'].item()
         self.power_history = list(data['power_history'])
+        if 'power_timestamps' in data:
+            self.power_timestamps = list(data['power_timestamps'])
+        else:
+            self.power_timestamps = list(range(len(self.power_history)))
 
     def add_to_plot(self, ax, facecolor="navy", edgecolor="navy", alpha=1, linestyle="-"):
         """Add monitor visualization to 2D plot."""

beamz/design/sources.py → beamz/devices/sources.py

@@ -2,7 +2,7 @@
 from beamz.const import LIGHT_SPEED, µm
 import matplotlib.pyplot as plt
 import matplotlib.patches as patches
-from beamz.design.mode import solve_modes
+from beamz.devices.mode import solve_modes
 
 class GaussianSource():
     """A Gaussian current distribution in space.
@@ -126,7 +126,7 @@ def __init__(self, design, position=None, width=None, height=None, wavelength=1.
             # This assumes a simple rectangular waveguide structure
             # Extract core/cladding indices for analytical solver
             try:
-                from beamz.design.mode import slab_mode_source
+                from beamz.devices.mode import slab_mode_source
 
                 # Sample x coordinates along the cross-section
                 num_points = eps_1d.size

beamz/optimization/__init__.py

@@ -1,8 +1,6 @@
-"""
-Optimization module for BEAMZ - Contains adjoint optimization functionality.
-"""
+"""Adjoint-based optimization helpers for BEAMZ."""
 
-# Currently empty as the module is under development
-# Will be populated as optimization features are added
+from . import topology
+from .optimizers import Optimizer
 
-__all__ = [] 
+__all__ = ["topology", "Optimizer"]