emerge 0.6.9__tar.gz → 0.6.11__tar.gz

{emerge-0.6.9 → emerge-0.6.11}/.bumpversion.toml

@@ -1,5 +1,5 @@
 [tool.bumpversion]
-current_version = "0.6.9"
+current_version = "0.6.11"
 parse = "(?P<major>\\d+)\\.(?P<minor>\\d+)\\.(?P<patch>\\d+)"
 serialize = ["{major}.{minor}.{patch}"]
 search = "{current_version}"

{emerge-0.6.9 → emerge-0.6.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: emerge
-Version: 0.6.9
+Version: 0.6.11
 Summary: An open source EM FEM simulator in Python
 Project-URL: Homepage, https://github.com/FennisRobert/EMerge
 Project-URL: Issues, https://github.com/FennisRobert/EMerge/issues

{emerge-0.6.9 → emerge-0.6.11}/emerge/__init__.py

@@ -18,7 +18,7 @@ along with this program; if not, see
 """
 import os
 
-__version__ = "0.6.9"
+__version__ = "0.6.11"
 
 ############################################################
 #               HANDLE ENVIRONMENT VARIABLES              #
@@ -43,12 +43,13 @@ from loguru import logger
 
 LOG_CONTROLLER.set_default()
 logger.debug('Importing modules')
+LOG_CONTROLLER._set_log_buffer()
 
 from ._emerge.simmodel import Simulation
 from ._emerge.material import Material, FreqCoordDependent, FreqDependent, CoordDependent
 from ._emerge import bc
 from ._emerge.solver import SolverBicgstab, SolverGMRES, SolveRoutine, ReverseCuthillMckee, Sorter, SolverPardiso, SolverUMFPACK, SolverSuperLU, EMSolver
-from ._emerge.cs import CoordinateSystem, CS, GCS, Plane, Axis, XAX, YAX, ZAX, XYPLANE, XZPLANE, YZPLANE, YXPLANE, ZXPLANE, ZYPLANE
+from ._emerge.cs import CoordinateSystem, CS, GCS, Plane, Axis, XAX, YAX, ZAX, XYPLANE, XZPLANE, YZPLANE, YXPLANE, ZXPLANE, ZYPLANE, cs
 from ._emerge.coord import Line
 from ._emerge import geo
 from ._emerge.selection import Selection, FaceSelection, DomainSelection, EdgeSelection
@@ -62,4 +63,12 @@ from ._emerge.howto import _HowtoClass
 
 howto = _HowtoClass()
 
-logger.debug('Importing complete!')
+logger.debug('Importing complete!')
+
+
+############################################################
+#                         CONSTANTS                        #
+############################################################
+
+CENTER = geo.Alignment.CENTER
+CORNER = geo.Alignment.CORNER

{emerge-0.6.9 → emerge-0.6.11}/emerge/_emerge/bc.py

@@ -53,7 +53,7 @@ class BoundaryCondition:
         
         
         if isinstance(assignment, GeoObject):
-            assignment = assignment.select
+            assignment = assignment.selection
         
         self.selection: Selection = assignment
         self.tags: list[int] = self.selection.tags

{emerge-0.6.9 → emerge-0.6.11}/emerge/_emerge/cs.py

@@ -530,3 +530,39 @@ CS = CoordinateSystem
 # The global coordinate system
 GCS = CoordinateSystem(XAX, YAX, ZAX, np.zeros(3), _is_global=True)
 
+def cs(axes: str, origin: tuple[float, float, float] = (0.,0.,0.,)) -> CoordinateSystem:
+    """Generate a coordinate system based on a simple string
+    The string must contain the letters x, X, y, Y, z and/or Z. 
+    Small letters refer to positive axes and capitals to negative axes.
+    
+    The string must be 3 characters long.
+    
+    The first position indices which global axis gets assigned to the new local X-axis
+    The second position indicates the Y-axis
+    The third position indicates the Z-axis
+    
+    Thus, rotating the global XYZ coordinate system 90 degrees around the Z axis would yield: yXz
+
+    Args:
+        axes (str): The axis description
+        origin (tuple[float, float, float], optional): The origin of the coordinate system. Defaults to (0.,0.,0.,).
+
+    Returns:
+        CoordinateSystem: The resultant coordinate system
+    """
+    if len(axes) != 3:
+        raise ValueError('Axis object must be of length 3')
+    
+    axlib = {
+        'x': Axis(np.array([1.0,0.,0.])),
+        'X': Axis(np.array([-1.0,0.,0.])),
+        'y': Axis(np.array([0.,1.0,0.])),
+        'Y': Axis(np.array([0.,-1.0,0.])),
+        'z': Axis(np.array([0.,0.,1.0])),
+        'Z': Axis(np.array([0.,0.,-1.0])),
+    }
+    ax_obj = [axlib[ax] for ax in axes]
+    
+    return (ax_obj[0]*ax_obj[1]*ax_obj[2]).displace(*origin)
+
+

{emerge-0.6.9 → emerge-0.6.11}/emerge/_emerge/geo/polybased.py

@@ -206,6 +206,7 @@ def orthonormalize(axis: np.ndarray) -> tuple[np.ndarray, np.ndarray, np.ndarray
     Zaxis = np.abs(Zaxis/np.linalg.norm(Zaxis))
     return Xaxis, Yaxis, Zaxis
 
+
 class GeoPrism(GeoVolume):
     """The GepPrism class generalizes the GeoVolume for extruded convex polygons.
     Besides having a volumetric definitions, the class offers a .front_face 
@@ -216,35 +217,65 @@ class GeoPrism(GeoVolume):
     """
     def __init__(self,
                  volume_tag: int,
-                 front_tag: int,
-                 side_tags: list[int],):
+                 front_tag: int | None = None,
+                 side_tags: list[int] | None = None,
+                 _axis: Axis | None = None):
         super().__init__(volume_tag)
-        self.front_tag: int = front_tag
-        self.back_tag: int = None
+        
+        
+        
+        if front_tag is not None and side_tags is not None:
+            self.front_tag: int = front_tag
+            self.back_tag: int = None
 
-        gmsh.model.occ.synchronize()
-        o1 = gmsh.model.occ.get_center_of_mass(2, self.front_tag)
-        n1 = gmsh.model.get_normal(self.front_tag, (0,0))
-        self._add_face_pointer('back', o1, n1)
+            gmsh.model.occ.synchronize()
+            self._add_face_pointer('back', tag=self.front_tag)
 
-        tags = gmsh.model.get_boundary(self.dimtags, oriented=False)
-        
-        for dim, tag in tags:
-            if (dim,tag) in side_tags:
-                continue
-            o2 = gmsh.model.occ.get_center_of_mass(2, tag)
-            n2 = gmsh.model.get_normal(tag, (0,0))
-            self._add_face_pointer('front', o2, n2)
-            self.back_tag = tag
-            break
+            tags = gmsh.model.get_boundary(self.dimtags, oriented=False)
+            
+            for dim, tag in tags:
+                if (dim,tag) in side_tags:
+                    continue
+                self._add_face_pointer('front',tag=tag)
+                self.back_tag = tag
+                break
 
-        self.side_tags: list[int] = [dt[1] for dt in tags if dt[1]!=self.front_tag and dt[1]!=self.back_tag]
+            self.side_tags: list[int] = [dt[1] for dt in tags if dt[1]!=self.front_tag and dt[1]!=self.back_tag]
 
-        for tag in self.side_tags:
-            o2 = gmsh.model.occ.get_center_of_mass(2, tag)
-            n2 = gmsh.model.get_normal(tag, (0,0))
-            self._add_face_pointer(f'side{tag}', o2, n2)
-            self.back_tag = tag
+            for tag in self.side_tags:
+    
+                self._add_face_pointer(f'side{tag}', tag=tag)
+                self.back_tag = tag
+                
+        elif _axis is not None:
+            _axis = _parse_axis(_axis)
+            gmsh.model.occ.synchronize()
+            tags = gmsh.model.get_boundary(self.dimtags, oriented=False)
+            faces = []
+            for dim, tag in tags:
+                o1 = np.array(gmsh.model.occ.get_center_of_mass(2, tag))
+                n1 = np.array(gmsh.model.get_normal(tag, (0,0)))
+                if abs(np.sum(n1*_axis.np)) > 0.99:
+                    dax = sum(o1 * _axis.np)
+                    faces.append((o1, n1, dax, tag))
+            
+            faces = sorted(faces, key=lambda x: x[2])
+            ftags = []
+            if len(faces) >= 2:
+                ftags.append(faces[0][3])
+                ftags.append(faces[-1][3])
+                self._add_face_pointer('front',faces[0][0], faces[0][1])
+                self._add_face_pointer('back', faces[-1][0], faces[-1][1])
+            elif len(faces)==1:
+                ftags.append(faces[0][3])
+                self._add_face_pointer('cap',faces[0][0], faces[0][1])
+            
+            ictr = 1
+            for dim, tag in tags:
+                if tag in ftags:
+                    continue
+                self._add_face_pointer(f'side{ictr}', tag=tag)
+                ictr += 1
 
     def outside(self, *exclude: Literal['front','back']) -> FaceSelection:
         """Select all outside faces except for the once specified by outside
@@ -462,9 +493,10 @@ class XYPolygon:
         ax, ay, az = axis
         
         volume = gmsh.model.occ.revolve(poly_fin.dimtags, x,y,z, ax, ay, az, angle*np.pi/180)
+        
         tags = [t for d,t in volume if d==3]
-        surftags = [t for d,t in volume if d==2]
-        return GeoPrism(tags, surftags[0], surftags)
+        poly_fin.remove()
+        return GeoPrism(tags, _axis=axis)
 
     @staticmethod
     def circle(radius: float, 

{emerge-0.6.9 → emerge-0.6.11}/emerge/_emerge/geo/shapes.py

@@ -27,6 +27,12 @@ from typing import Literal
 from functools import reduce
 
 class Alignment(Enum):
+    """The alignment Enum describes if a box, cube or rectangle location
+    is specified for the center or the bottom - front - left corner (min X Y and Z)
+
+    Args:
+        Enum (_type_): _description_
+    """
     CENTER = 1
     CORNER = 2
 

{emerge-0.6.9 → emerge-0.6.11}/emerge/_emerge/geometry.py

@@ -250,7 +250,7 @@ class GeoObject:
         return self.material._metal
     
     @property
-    def select(self) -> Selection:
+    def selection(self) -> Selection:
         '''Returns a corresponding Face/Domain or Edge Selection object'''
         if self.dim==1:
             return EdgeSelection(self.tags)
@@ -285,9 +285,18 @@ class GeoObject:
     
     def _add_face_pointer(self, 
                           name: str,
-                          origin: np.ndarray,
-                          normal: np.ndarray):
-        self._face_pointers[name] = _FacePointer(origin, normal)
+                          origin: np.ndarray | None = None,
+                          normal: np.ndarray | None = None,
+                          tag: int | None = None):
+        if tag is not None:
+            o = gmsh.model.occ.get_center_of_mass(2, tag)
+            n = gmsh.model.get_normal(tag, (0,0))
+            self._face_pointers[name] = _FacePointer(o, n)
+            return
+        if origin is not None and normal is not None:
+            self._face_pointers[name] = _FacePointer(origin, normal)
+            return
+        raise ValueError('Eitehr a tag or an origin + normal must be provided!')
     
     def make_copy(self) -> GeoObject:
         new_dimtags = gmsh.model.occ.copy(self.dimtags)
@@ -508,14 +517,14 @@ class GeoVolume(GeoObject):
             self.tags = [tag,]
 
     @property
-    def select(self) -> DomainSelection:
+    def selection(self) -> DomainSelection:
         return DomainSelection(self.tags)
     
 class GeoPoint(GeoObject):
     dim = 0
 
     @property
-    def select(self) -> PointSelection:
+    def selection(self) -> PointSelection:
         return PointSelection(self.tags)
     
     def __init__(self, tag: int | list[int]):
@@ -531,7 +540,7 @@ class GeoEdge(GeoObject):
     dim = 1
 
     @property
-    def select(self) -> EdgeSelection:
+    def selection(self) -> EdgeSelection:
         return EdgeSelection(self.tags)
     
     def __init__(self, tag: int | list[int]):
@@ -549,7 +558,7 @@ class GeoSurface(GeoObject):
     dim = 2
 
     @property
-    def select(self) -> FaceSelection:
+    def selection(self) -> FaceSelection:
         return FaceSelection(self.tags)
     
     def __init__(self, tag: int | list[int]):

{emerge-0.6.9 → emerge-0.6.11}/emerge/_emerge/logsettings.py

@@ -18,11 +18,14 @@
 from loguru import logger
 import sys
 from typing import Literal
-from enum import Enum
 from pathlib import Path
 import os
-import gmsh
+from collections import deque
 
+_LOG_BUFFER = deque()
+
+def _log_sink(message):
+    _LOG_BUFFER.append(message)
 ############################################################
 #                          FORMATS                         #
 ############################################################
@@ -82,6 +85,14 @@ class LogController:
                 format=FORMAT_DICT.get(loglevel, INFO_FORMAT))
         self.std_handlers.append(handle_id)
 
+    def _set_log_buffer(self):
+        logger.add(_log_sink)
+        
+    def _flush_log_buffer(self):
+        for msg in list(_LOG_BUFFER):
+            logger.opt(depth=6).log(msg.record["level"].name, msg.record["message"])
+        _LOG_BUFFER.clear()
+        
     def set_std_loglevel(self, loglevel: str):
         handler = {"sink": sys.stdout, 
                    "level": loglevel, 

{emerge-0.6.9 → emerge-0.6.11}/emerge/_emerge/material.py

@@ -310,6 +310,8 @@ class Material:
 
         self.color: str = color
         self.opacity: float = opacity
+        self._hash_key: int = -1
+        
         if _neff is None:
             self._neff: Callable = lambda f: np.sqrt(self.ur._fmax(f)*self.er._fmax(f))
         else:
@@ -318,8 +320,14 @@ class Material:
         self._color_rgb = tuple(int(hex_str[i:i+2], 16)/255.0 for i in (0, 2, 4))
         self._metal: bool = _metal
 
+    def __hash__(self):
+        return self._hash_key
+    
     def __str__(self) -> str:
-        return f'Material({self.name})'
+        return f'Material({self.name}, {self._hash_key})'
+    
+    def __repr__(self):
+        return f'Material({self.name}, {self._hash_key})'
     
     def initialize(self, xs: np.ndarray, ys: np.ndarray, zs: np.ndarray, ids: np.ndarray):
         """Initializes the Material properties to be evaluated at xyz-coordinates for
@@ -343,6 +351,7 @@ class Material:
         self.ur.reset()
         self.tand.reset()
         self.cond.reset()
+        self._hash_key = -1
         
     @property
     def frequency_dependent(self) -> bool:

{emerge-0.6.9 → emerge-0.6.11}/emerge/_emerge/mesh3d.py

@@ -544,22 +544,33 @@ class Mesh3D(Mesh):
         #arry = np.zeros((3,3,self.n_tets,), dtype=np.complex128)
         for vol in volumes:
             vol.material.reset()
-            
+        
         materials = []
+        i = 0
+        for vol in volumes:
+            if vol.material not in materials:
+                materials.append(vol.material)
+                vol.material._hash_key = i
+                i += 1
         
         xs = self.centers[0,:]
         ys = self.centers[1,:]
         zs = self.centers[2,:]
         
+        matassign = -1*np.ones((self.n_tets,), dtype=np.int64)
+        
         for volume in sorted(volumes, key=lambda x: x._priority):
         
             for dimtag in volume.dimtags:
                 etype, etag_list, ntags = gmsh.model.mesh.get_elements(*dimtag)
                 for etags in etag_list:
                     tet_ids = np.array([self.tet_t2i[t] for t in etags])
-                    volume.material.initialize(xs[tet_ids], ys[tet_ids], zs[tet_ids], tet_ids)
-                    if volume.material not in materials:
-                        materials.append(volume.material)
+                    matassign[tet_ids] = volume.material._hash_key
+        
+        for mat in materials:
+            ids = np.argwhere(matassign==mat._hash_key).flatten()
+            mat.initialize(xs[ids], ys[ids], zs[ids], ids)
+                    
         
         return materials
     

{emerge-0.6.9 → emerge-0.6.11}/emerge/_emerge/physics/microwave/microwave_3d.py

@@ -53,7 +53,7 @@ def run_job_multi(job: SimJob) -> SimJob:
     Returns:
         SimJob: The solved SimJob
     """
-    routine = DEFAULT_ROUTINE.duplicate()._configure_routine('MP')
+    routine = DEFAULT_ROUTINE._configure_routine('MP')
     for A, b, ids, reuse, aux in job.iter_Ab():
         solution, report = routine.solve(A, b, ids, reuse, id=job.id)
         report.add(**aux)
@@ -311,22 +311,25 @@ class Microwave3D:
 
         if points.size==0:
             raise SimulationError(f'The lumped port {port} has no nodes associated with it')
+        
         xs = self.mesh.nodes[0,points]
         ys = self.mesh.nodes[1,points]
         zs = self.mesh.nodes[2,points]
 
         dotprod = xs*field_axis[0] + ys*field_axis[1] + zs*field_axis[2]
 
-        start_id = points[np.argwhere(dotprod == np.min(dotprod))]
+        start_id = np.argwhere(dotprod == np.min(dotprod)).flatten()
+        
+        xs = xs[start_id]
+        ys = ys[start_id]
+        zs = zs[start_id]
         
-        start = _pick_central(self.mesh.nodes[:,start_id.flatten()])
-        logger.info(f'Starting node = {_dimstring(start)}')
-        end = start + port.Vdirection.np*port.height
-
-
-        port.vintline = Line.from_points(start, end, 21)
 
-        logger.info(f'Ending node = {_dimstring(end)}')
+        for x,y,z in zip(xs, ys, zs):
+            start = np.array([x,y,z])
+            end = start + port.Vdirection.np*port.height
+            port.vintline.append(Line.from_points(start, end, 21))
+            logger.trace(f'Port[{port.port_number}] integration line {start} -> {end}.')
         
         port.v_integration = True
     
@@ -1003,7 +1006,8 @@ class Microwave3D:
             if bc.Z0 is None:
                 raise SimulationError('Trying to compute the impedance of a boundary condition with no characteristic impedance.')
             
-            V = bc.vintline.line_integral(fieldfunction)
+            Voltages = [line.line_integral(fieldfunction) for line in bc.vintline]
+            V = sum(Voltages)/len(Voltages)
             
             if bc.active:
                 if bc.voltage is None:

{emerge-0.6.9 → emerge-0.6.11}/emerge/_emerge/physics/microwave/microwave_bc.py

@@ -306,11 +306,17 @@ class AbsorbingBoundary(RobinBC):
         Returns:
             complex: The γ-constant
         """
-        return 1j*self.get_beta(k0)
+        if self.order == 2:
+            
+            p0 = 1.06103
+            p2 = -0.84883
+            ky = k0*0.5
+            return 1j*k0*p0 - 1j*p2*ky**2/k0
+        else:
+            Factor = 1
+            return 1j*self.get_beta(k0)*Factor
     
-    def get_Uinc(self, x_local: np.ndarray, y_local: np.ndarray, k0: float) -> np.ndarray:
-        return np.zeros((3, len(x_local)), dtype=np.complex128)
-
+   
 @dataclass
 class PortMode:
     modefield: np.ndarray
@@ -825,10 +831,10 @@ class LumpedPort(PortBC):
         self.Vdirection: Axis = direction # type: ignore
         self.type = 'TEM'
         
-        logger.info('Constructing coordinate system from normal port')
-        self.cs = Axis(self.selection.normal).construct_cs()  # type: ignore
-
-        self.vintline: Line | None = None
+        # logger.info('Constructing coordinate system from normal port')
+        # self.cs = Axis(self.selection.normal).construct_cs()  # type: ignore
+        self.cs = GCS
+        self.vintline: list[Line] = []
         self.v_integration = True
 
     @property
@@ -881,14 +887,7 @@ class LumpedPort(PortBC):
                      k0: float,
                      which: Literal['E','H'] = 'E') -> np.ndarray:
         ''' Compute the port mode E-field in local coordinates (XY) + Z out of plane.'''
-
-        px, py, pz = self.cs.in_local_basis(*self.Vdirection.np)
-        
-        Ex = px*np.ones_like(x_local)
-        Ey = py*np.ones_like(x_local)
-        Ez = pz*np.ones_like(x_local)
-        Exyz = np.array([Ex, Ey, Ez])
-        return Exyz
+        raise RuntimeError('This function should never be called in this context.')
 
     def port_mode_3d_global(self, 
                             x_global: np.ndarray,
@@ -911,10 +910,9 @@ class LumpedPort(PortBC):
         Returns:
             np.ndarray: The E-field in (3,N) indexing.
         """
-        xl, yl, _ = self.cs.in_local_cs(x_global, y_global, z_global)
-        Ex, Ey, Ez = self.port_mode_3d(xl, yl, k0)
-        Exg, Eyg, Ezg = self.cs.in_global_basis(Ex, Ey, Ez)
-        return np.array([Exg, Eyg, Ezg])
+        ON = np.ones_like(x_global)
+        Ex, Ey, Ez = self.Vdirection.np
+        return np.array([Ex*ON, Ey*ON, Ez*ON])
 
 
 class LumpedElement(RobinBC):

{emerge-0.6.9 → emerge-0.6.11}/emerge/_emerge/plot/matplotlib/mpldisplay.py

@@ -145,7 +145,7 @@ def _norm(x, y, z):
 
 def _select(obj: GeoObject | Selection) -> Selection:
     if isinstance(obj, GeoObject):
-        return obj.select
+        return obj.selection
     return obj
 
 def _merge(lst: list[GeoObject | Selection]) -> Selection:

{emerge-0.6.9 → emerge-0.6.11}/emerge/_emerge/plot/pyvista/display.py

@@ -186,7 +186,7 @@ def _norm(x, y, z):
 
 def _select(obj: GeoObject | Selection) -> Selection:
     if isinstance(obj, GeoObject):
-        return obj.select
+        return obj.selection
     return obj
 
 def _merge(lst: Iterable[GeoObject | Selection]) -> Selection:
@@ -523,18 +523,19 @@ class PVDisplay(BaseDisplay):
         d = _min_distance(xf, yf, zf)
 
         if port.vintline is not None:
-            xs, ys, zs = port.vintline.cpoint
-            p_line = pv.Line(
-                pointa=(xs[0], ys[0], zs[0]),
-                pointb=(xs[-1], ys[-1], zs[-1]),
-            )
-            self._plot.add_mesh(
-                p_line,
-                color='red',
-                pickable=False,
-                line_width=3.0,
-            )
-        
+            for line in port.vintline:
+                xs, ys, zs = line.cpoint
+                p_line = pv.Line(
+                    pointa=(xs[0], ys[0], zs[0]),
+                    pointb=(xs[-1], ys[-1], zs[-1]),
+                )
+                self._plot.add_mesh(
+                    p_line,
+                    color='red',
+                    pickable=False,
+                    line_width=3.0,
+                )
+            
         if k0 is None:
             if isinstance(port, ModalPort):
                 k0 = port.get_mode(0).k0
@@ -596,7 +597,7 @@ class PVDisplay(BaseDisplay):
 
 
         if scale=='log':
-            T = lambda x: np.log10(np.abs(x))
+            T = lambda x: np.log10(np.abs(x+1e-12))
         elif scale=='symlog':
             T = lambda x: np.sign(x) * np.log10(1 + np.abs(x*np.log(10)))
         else:

{emerge-0.6.9 → emerge-0.6.11}/emerge/_emerge/plot/simple_plots.py

@@ -422,9 +422,14 @@ def plot_sp(f: np.ndarray | list[np.ndarray], S: list[np.ndarray] | np.ndarray,
             show_plot: bool = True,
             figdata: tuple | None = None) -> tuple[plt.Figure, plt.Axes, plt.Axes]:
     """Plot S-parameters in dB and phase
+    
+    One may provide:
+     - A single frequency with a single S-parameter
+     - A single frequency with a list of S-parameters
+     - A list of frequencies with a list of S-parameters
 
     Args:
-        f (np.ndarray): Frequency vector
+        f (np.ndarray | list[np.ndarray]): Frequency vector or list of frequencies
         S (list[np.ndarray] | np.ndarray): S-parameters to plot (list or single array)
         dblim (list, optional): Decibel y-axis limit. Defaults to [-80, 5].
         xunit (str, optional): Frequency unit. Defaults to "GHz".
@@ -535,7 +540,7 @@ def plot_sp(f: np.ndarray | list[np.ndarray], S: list[np.ndarray] | np.ndarray,
 
     
 def plot_ff(
-    theta: np.ndarray,
+    theta: np.ndarray | list[np.ndarray],
     E: Union[np.ndarray, Sequence[np.ndarray]],
     grid: bool = True,
     dB: bool = False,
@@ -554,7 +559,7 @@ def plot_ff(
 
     Parameters
     ----------
-    theta : np.ndarray
+    theta : np.ndarray | list[np.ndarray]
         Angle array (radians).
     E : np.ndarray or sequence of np.ndarray
         Complex E-field samples; magnitude will be plotted.
@@ -575,6 +580,12 @@ def plot_ff(
         E_list = [E]
     else:
         E_list = list(E)
+        
+    if not isinstance(theta, list):
+        thetas = [theta for _ in E_list]
+    else:
+        thetas = theta
+        
     n_series = len(E_list)
 
     # Style broadcasting
@@ -591,6 +602,7 @@ def plot_ff(
 
     fig, ax = plt.subplots()
     for i, Ei in enumerate(E_list):
+        theta = thetas[i]
         mag = np.abs(Ei)
         if dB:
             mag = 20*np.log10(mag)