Refractored DirectivitData: radiation characteristics are computed vi…

…a properties

CHANGELOG.md

@@ -31,6 +31,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Added an option to simulate plane wave propagation at fixed angles by setting parameter `angular_spec=FixedAngleSpec()` when defining a `PlaneWave` source.
 - The universal `tidy3d.web` can now also be used to handle `ModeSolver` simulations.
 - Support for differentiation with respect to `PolySlab.slab_bounds`.
+- `DirectivityMonitor` can now be used to decompose fields into circular polarization states.
 
 ### Changed
 - Priority is given to `snapping_points` in `GridSpec` when close to structure boundaries, which reduces the chance of them being skipped.
@@ -51,6 +52,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - `estimate_cost` is now called at the end of every `web.upload` call.
 - Internal refactor of adjoint shape gradients using `GradientSurfaceMesh`.
 - Enhanced progress bar display in batch operations with better formatting, colors, and status tracking.
+- The behavior of `DirectivityData` was changed, so that only far fields and flux are stored. Radiation characteristics (e.g., directivity, axial ratio) can be accessed through properties.
 
 ### Fixed
 - Significant speedup for field projection computations.

tests/test_data/test_data_arrays.py

@@ -184,21 +184,16 @@ def make_mode_index_data_array():
     return td.ModeIndexDataArray(values, coords=dict(f=FS, mode_index=MODE_INDICES))
 
 
+def make_far_field_data_array():
+    values = (1 + 1j) * np.random.random((len(PD), len(THETAS), len(PHIS), len(FS)))
+    return td.FieldProjectionAngleDataArray(values, coords=dict(r=PD, theta=THETAS, phi=PHIS, f=FS))
+
+
 def make_flux_data_array():
     values = np.random.random(len(FS))
     return td.FluxDataArray(values, coords=dict(f=FS))
 
 
-def make_directivity_data_array():
-    values = np.random.random((len(PD), len(THETAS), len(PHIS), len(FS)))
-    return td.DirectivityDataArray(values, coords=dict(r=PD, theta=THETAS, phi=PHIS, f=FS))
-
-
-def make_axial_ratio_data_array():
-    values = np.random.random((len(PD), len(THETAS), len(PHIS), len(FS)))
-    return td.AxialRatioDataArray(values, coords=dict(r=PD, theta=THETAS, phi=PHIS, f=FS))
-
-
 def make_flux_time_data_array():
     values = np.random.random(len(TS))
     return td.FluxTimeDataArray(values, coords=dict(t=TS))
@@ -261,13 +256,8 @@ def test_flux_time_data_array():
     data = data.interp(t=1e-13)
 
 
-def test_directivity_data_array():
-    data = make_directivity_data_array()
-    data = data.sel(f=1e14, phi=0)
-
-
-def test_axial_ratio_data_array():
-    data = make_axial_ratio_data_array()
+def test_far_field_data_array():
+    data = make_far_field_data_array()
     data = data.sel(f=1e14, phi=0)
 
 

tests/test_data/test_monitor_data.py

@@ -33,9 +33,8 @@
     PERMITTIVITY_MONITOR,
     SIM,
     SIM_SYM,
-    make_axial_ratio_data_array,
     make_diffraction_data_array,
-    make_directivity_data_array,
+    make_far_field_data_array,
     make_flux_data_array,
     make_flux_time_data_array,
     make_mode_amps_data_array,
@@ -54,9 +53,6 @@
 GRID_CORRECTION = FreqModeDataArray(
     1 + 0.01 * np.random.rand(*N_COMPLEX.shape), coords=N_COMPLEX.coords
 )
-DIRECTIVITY = make_directivity_data_array()
-AXIALRATIO = make_axial_ratio_data_array()
-
 """ Make the montor data """
 
 
@@ -190,8 +186,16 @@ def make_flux_data():
 
 
 def make_directivity_data():
+    data = make_far_field_data_array()
     return DirectivityData(
-        monitor=DIRECTIVITY_MONITOR, directivity=DIRECTIVITY.copy(), axial_ratio=AXIALRATIO.copy()
+        monitor=DIRECTIVITY_MONITOR,
+        flux=FLUX.copy(),
+        Er=data,
+        Etheta=data,
+        Ephi=data,
+        Hr=data,
+        Htheta=data,
+        Hphi=data,
     )
 
 
@@ -337,6 +341,8 @@ def test_directivity_data():
     data = make_directivity_data()
     _ = data.directivity
     _ = data.axial_ratio
+    _ = data.left_polarization
+    _ = data.right_polarization
 
 
 def test_diffraction_data():

tidy3d/__init__.py

@@ -37,11 +37,9 @@
 
 # data
 from .components.data.data_array import (
-    AxialRatioDataArray,
     CellDataArray,
     ChargeDataArray,
     DiffractionDataArray,
-    DirectivityDataArray,
     EMECoefficientDataArray,
     EMEModeIndexDataArray,
     EMEScalarFieldDataArray,
@@ -417,8 +415,6 @@ def set_logging_level(level: str) -> None:
     "FieldProjectionCartesianDataArray",
     "FieldProjectionKSpaceDataArray",
     "DiffractionDataArray",
-    "DirectivityDataArray",
-    "AxialRatioDataArray",
     "HeatDataArray",
     "ChargeDataArray",
     "FieldDataset",

tidy3d/components/data/data_array.py

@@ -921,55 +921,6 @@ class FieldProjectionAngleDataArray(DataArray):
     _data_attrs = {"long_name": "radiation vectors"}
 
 
-class DirectivityDataArray(DataArray):
-    """Directivity in the frequency domain as a function of angles theta and phi.
-    Directivity is a dimensionless quantity defined as the ratio of the radiation
-    intensity in a given direction to the average radiation intensity over all directions.
-
-    Example
-    -------
-    >>> f = np.linspace(1e14, 2e14, 10)
-    >>> r = np.atleast_1d(5)
-    >>> theta = np.linspace(0, np.pi, 10)
-    >>> phi = np.linspace(0, 2*np.pi, 20)
-    >>> coords = dict(r=r, theta=theta, phi=phi, f=f)
-    >>> values = np.random.random((len(r), len(theta), len(phi), len(f)))
-    >>> data = DirectivityDataArray(values, coords=coords)
-    """
-
-    __slots__ = ()
-    _dims = ("r", "theta", "phi", "f")
-    _data_attrs = {"long_name": "radiation intensity"}
-
-
-class AxialRatioDataArray(DataArray):
-    """Axial Ratio (AR) in the frequency domain as a function of angles theta and phi.
-    AR is a dimensionless quantity defined as the ratio of the major axis to the minor
-    axis of the polarization ellipse.
-
-    Note
-    ----
-    The axial ratio computation is based on:
-
-    Balanis, Constantine A., "Antenna Theory: Analysis and Design,"
-    John Wiley & Sons, Chapter 2.12 (2016).
-
-    Example
-    -------
-    >>> f = np.linspace(1e14, 2e14, 10)
-    >>> r = np.atleast_1d(5)
-    >>> theta = np.linspace(0, np.pi, 10)
-    >>> phi = np.linspace(0, 2*np.pi, 20)
-    >>> coords = dict(r=r, theta=theta, phi=phi, f=f)
-    >>> values = np.random.random((len(r), len(theta), len(phi), len(f)))
-    >>> data = AxialRatioDataArray(values, coords=coords)
-    """
-
-    __slots__ = ()
-    _dims = ("r", "theta", "phi", "f")
-    _data_attrs = {"long_name": "axial ratio"}
-
-
 class FieldProjectionCartesianDataArray(DataArray):
     """Far fields in frequency domain as a function of local x and y coordinates.
 
@@ -1267,8 +1218,6 @@ class IndexedDataArray(DataArray):
     FieldProjectionCartesianDataArray,
     FieldProjectionKSpaceDataArray,
     DiffractionDataArray,
-    DirectivityDataArray,
-    AxialRatioDataArray,
     FreqModeDataArray,
     FreqDataArray,
     TimeDataArray,