Add FreqPotentialDataArray

Author: damiano-flex

tidy3d/components/data/data_array.py

@@ -532,6 +532,24 @@ class FreqDataArray(DataArray):
     _dims = ("f",)
 
 
+class FreqPotentialDataArray(DataArray):
+    """Frequency-domain array.
+
+    Example
+    -------
+    >>> f = [2e14, 3e14]
+    >>> v = [0.1, 0.2, 0.3]
+    >>> coords = dict(f=f, v=v)
+    >>> fd = FreqPotentialDataArray((1+1j) * np.random.random((2, 3)), coords=coords)
+    """
+
+    __slots__ = ()
+    _dims = (
+        "f",
+        "v",
+    )
+
+
 class FreqModeDataArray(DataArray):
     """Array over frequency and mode index.
 
@@ -1357,6 +1375,7 @@ class PerturbationCoefficientDataArray(DataArray):
     FreqDataArray,
     TimeDataArray,
     FreqModeDataArray,
+    FreqPotentialDataArray,
     TriangleMeshDataArray,
     HeatDataArray,
     EMEScalarFieldDataArray,

tidy3d/components/tcad/data/sim_data.py

@@ -11,6 +11,7 @@
 from tidy3d.components.base import Tidy3dBaseModel
 from tidy3d.components.base_sim.data.sim_data import AbstractSimulationData
 from tidy3d.components.data.data_array import (
+    FreqPotentialDataArray,
     SpatialDataArray,
     SteadyVoltageDataArray,
 )
@@ -84,6 +85,12 @@ class DeviceCharacteristics(Tidy3dBaseModel):
         "is given in Ohms. Note that in 2D the resistance is given in :math:`\\Omega \\mu`.",
     )
 
+    ac_current_voltage: Optional[FreqPotentialDataArray] = pd.Field(
+        None,
+        title="AC current vs voltage and frequency",
+        description="Complex small-signal current I(v, f) as a FreqPotentialDataArray.",
+    )
+
 
 class AbstractHeatChargeSimulationData(AbstractSimulationData, ABC):
     """Abstract class for HeatChargeSimulation results, or VolumeMesher results."""