ElectricalAmplifierTimeStepper

ElectricalAmplifierTimeStepper(
    *,
    gain: Annotated[float, units='dB'] = 0,
    f_3dB: Annotated[
        Annotated[float, minimum=0, units='Hz'] | None, _
    ] = None,
    saturation_power: Annotated[float, units='dBm'] | None = None,
    compression_power: Annotated[float, units='dBm'] | None = None,
    ip3: Annotated[Annotated[float, units='dBm'] | None, _] = None,
    noise_figure: Annotated[float, units='dB'] | None = None,
    r0: Annotated[float, maximum=1, minimum=-1] = 0,
    r1: Annotated[float, maximum=1, minimum=-1] = 0,
    ports: Annotated[collections.abc.Sequence[str], maxItems=2, minItems=2]
    | None = None,
    seed: Annotated[int, minimum=0] | None = None,
) -> None

Time-stepper for a broadband amplifier or modulator driver.

This model implements an electrical amplifier with linear gain, noise, bandwidth limiting, port reflections, and common nonlinear effects: 1 dB compression, third-order intercept point (IP3), and saturation.

Args

• gain: The amplifier’s power gain.
• f_3dB: -3 dB frequency cutoff for bandwidth limiting. None or 0 disables filtering.
• saturation_power: Output saturation power.
• compression_power: Output 1 dB compression power for the fundamental of a real single tone.
• ip3: Output third-order intercept point from a two-tone measurement.
• noise_figure: The amplifier’s noise figure (NF). If None, noise is disabled.
• r0: Reflection coefficient for the input port.
• r1: Reflection coefficient for the output port.
• ports: Input and output port names. If not set, the sorted list of port names from the component is used.
• seed: Random number generator seed to ensure reproducibility.

Note

The forward path signal is processed through the following stages at each time step:

1. Low-pass filter for bandwidth limiting:

   $$\begin{aligned} y_0\![n] &= \alpha y_0\![n-1] + (1-\alpha) G_a a_\text{in} \\ \alpha &= e^{-2\pi f_\text{3dB} \Delta t} \\ G_a &= 10^\frac{G_\text{dB}}{20} \end{aligned}$$

2. Soft-knee compression:

   $$\begin{aligned} y_1 &= y_0 \left( 1 + \frac{k_s c_s}{P_\text{1dB}} y_0^2 \right)^{-\frac{1}{2}} \\ c_s &= 10^\frac{1}{10} - 1 \\ k_s &= \frac{10^{-\frac{1}{10}} q_\star}{2c_s} \approx 0.5395 \end{aligned}$$

   The calibration parameter $q_\star$ is the root of

   $$\frac{4}{\pi} \int_0^{\frac{\pi}{2}} \frac{\cos^2\theta}{\sqrt{1 + q_\star\cos^2\theta}}\,d\theta = 10^{-\frac{1}{20}}.$$

   This calibrates $P_\text{1dB}$ to the average power in the compressed fundamental rather than the instantaneous amplitude.

3. Cubic IP3 nonlinearity:

   $$\begin{aligned} y_2 &= y_1 \left(1 - c_i y_1^2\right) \\ c_i &= \frac{2}{3P_\text{IP3}} \end{aligned}$$

   For equal tones with peak amplitude $A$, the IM3 amplitude is $3c_i A^3/4$, which gives $P_\text{OIP3}=2/(3c_i)$.

4. Smooth saturation: A tanh function smoothly clamps the output amplitude to approach $\sqrt{P_\text{sat}}$.

5. Additive noise: if enabled, white Gaussian noise is added to the output. The noise power spectral density is derived from the noise figure $F$ as $S_P = k_B T_0 G F$, where $G = 10^{G_\text{dB}/10}$ is the linear power gain.

Methods

Method	Description
bind_native	Bind a compiled step function.
reset	Reset any internal state of the time stepper.
setup	Initialize the time stepper.
setup_state	Initialize internal state.
step	Compute the outputs of this time stepper, given inputs.
step_single	Take a single time step on the given inputs.
update	Update this time stepper.
write_verilog_a	Write this configured time stepper as one Verilog-A module.

bind_native

bind_native(self, step, keys, state=None, cstate=None, context=None)

Bind a compiled step function.

Instead of implementing step_single, a time stepper can hand the simulator the address of a C function with the signature:

int64 step(inputs, outputs, time_index, update_state, shutdown,
           context, state, cstate)

inputs and outputs are complex128 arrays with one entry per key, state and cstate are the float64 and complex128 arrays, and a nonzero return signals failure. The function must overwrite every element of outputs and only modify the state arrays when update_state is true. The arrays must not be reallocated after binding; reset restores them to the values given here, so a subclass overriding reset must call super().reset().

Calls with update_state false must give the same outputs: circuits may call the function several times in a time step, and setup checks this for steppers that declare path_delays. Random numbers must therefore come from a generator whose state lives in the arrays (e.g., a seed and a counter): output a draw kept there and draw the next one only when update_state is true, which also makes reset replay the stream. Do not use numpy.random in the function: numba draws it from a global stream that differs on every call and survives a reset.

The function can be compiled with numba (when available).

Args

• step (int | object): Address of the step function, or an object with an address attribute such as a numba cfunc.
• keys (Sequence[str]): Input/output keys.
• state (ndarray | None): Contiguous float64 array the function may use.
• cstate (ndarray | None): Contiguous complex128 array the function may use.
• context (int | None): Opaque pointer passed to the function.

Example

A two-port element that leaks the field entering its first port out of its second one:

>>> from numba import cfunc
>>> from photonforge.time_steppers import STEP_SIGNATURE
>>>
>>> @cfunc(STEP_SIGNATURE, cache=True)  # doctest: +SKIP
... def leaky_step(inputs, outputs, time_index, update_state,
...                shutdown, context, state, cstate):
...     # state[0]: decay per step
...     # cstate[0]: field leaving the second port
...     outputs[0] = 0j
...     outputs[1] = cstate[0] * state[0] + inputs[0]
...     if update_state:
...         cstate[0] = outputs[1]
...     return 0
...
>>> class LeakyTimeStepper(pf.TimeStepper):
...     def __init__(self, *, k=0.5):
...         super().__init__(k=k)
...
...     def setup_state(self, *, component, time_step, **kwargs):
...         keys = tuple(f"{p}@0" for p in sorted(component.ports))
...         state = numpy.array([
...             self.parametric_kwargs["k"] ** time_step
...         ])
...         self.bind_native(
...             leaky_step, keys, state, numpy.zeros(1, complex)
...         )
...
>>> pf.register_time_stepper_class(LeakyTimeStepper)

reset

reset(self)

Reset any internal state of the time stepper.

The default implementation restores the arrays given to bind_native, if any, and performs no other action. Derived classes override it to reset the state they hold themselves.

setup

setup(
    self,
    component,
    time_step,
    *,
    carrier_frequency=0,
    show_progress=True,
    **kwargs,
)

Initialize the time stepper.

Args

• component (Component): Component for the time stepper.
• time_step (float): The interval between time steps (in seconds).
• carrier_frequency (float): The carrier frequency used to construct the time stepper. The carrier should be omitted from the input signals, as it is handled automatically by the time stepper.
• show_progress (bool): If True, show setup progress.
• **kwargs (object): Keyword args forwarded to setup_state.

Returns

TimeStepper: This time stepper.

setup_state

setup_state(
    self,
    *,
    component: photonforge.Component,
    time_step: Annotated[float, minimum=0, units='s'],
    **kwargs: object,
) -> None

Initialize internal state.

Args

• component: Component representing the amplifier.
• time_step: The interval between time steps (in seconds).
• kwargs: Unused.

step

step(self, inputs=None, steps=0, time_step=0, show_progress=True)

Compute the outputs of this time stepper, given inputs.

Args

• inputs (TimeSeries): Inputs to the time stepper.
• steps (int): The number of time steps to execute. If inputs is defined, this value is not used.
• time_step (float): The interval between time steps (in seconds). If inputs is a defined, this value is not used.
• show_progress (bool): If True, show computation progress.

Returns

TimeSeries: Computed outputs.

step_single

step_single(self, inputs, outputs, time_index, update_state, shutdown)

Take a single time step on the given inputs.

Important

Derived classes must implement this method unless they bind a compiled step with bind_native, in which case the default implementation calls it and, when shutdown is set, the optional _after_native_run hook.

The method takes the inputs as a numpy array of type complex with values in the order defined by keys. It advances a single time step, updating its internal state (if any) according to the update_state flag. It writes the outputs in-place to the pre-allocated, zero-initialized outputs array in the same order.

Args

• inputs (ndarray): Input values at the current time step. Must be a 1D array of complex values ordered according to keys.
• outputs (ndarray): Pre-allocated output array where results will be stored. Same size and type as inputs.
• time_index (int): Time series index for the current input.
• update_state (bool): Whether to update the internal stepper state.
• shutdown (bool): Whether this is the last call to the single stepping function for the original TimeSeries.

update

update(self, *args, **kwargs)

Update this time stepper.

All contents of the time stepper are updated with the contents from the updated version. Object contents or modifications introduced after the time stepper creation (outside of the original parametric function) are not propagated.

Args

• *args (object): Positional arguments to the parametric time stepper function.
• **kwargs (object): Keyword arguments to the parametric time stepper function.

Returns

TimeStepper: This time stepper.

Note

The original keyword arguments used to generate the time stepper are stored and updated with the passed kwargs. This updated version is used in the function call and will be stored for future updates. Positional arguments are not stored.

write_verilog_a

write_verilog_a(
    self,
    path,
    *,
    backend='openvaf',
    port_convention='travelling_wave',
    module_name=None,
    z0=50.0,
    field_impedance=None,
)

Write this configured time stepper as one Verilog-A module.

The time stepper must first be configured with setup or Component.setup_time_stepper. The writer creates source only; it does not compile the model or create a simulator deck. Requires a PhotonForge Photonic Canvas license.

Args

• path (str | os.PathLike): Destination .va file.
• backend (Literal[“openvaf”, “spectre”]): Compiler capability target recorded in the source.
• port_convention (Literal[“travelling_wave”, “gf_45spclo_npe”]): External optical port convention. GF packed ports require the Spectre backend.
• module_name (str | None): Module identifier derived from path when omitted.
• z0 (float): Travelling-wave optical reference impedance.
• field_impedance (float | None): GF electric-field scaling.

Returns

pathlib.Path: Absolute destination path.

Attributes

Attribute	Description
parametric_function	str: Function used to update the time stepper.
parametric_kwargs	dict[str, object]: Keyword arguments used to update the time stepper.
path_delays	dict[tuple[str, str], int] | None: Minimum delay in time steps of each (input key, output key) signal path; missing pairs have no path and None means every input reaches every output within the time step.
properties	Properties: Object properties.
random_variables	list[RandomVariable]: Random variables associated to the time stepper’s parameters.

parametric_function

str: Function used to update the time stepper.

parametric_kwargs

dict[str, object]: Keyword arguments used to update the time stepper.

path_delays

dict[tuple[str, str], int] | None: Minimum delay in time steps of each (input key, output key) signal path; missing pairs have no path and None means every input reaches every output within the time step.

properties

Properties: Object properties.

random_variables

list[RandomVariable]: Random variables associated to the time stepper’s parameters.