r""" Set the network topology. It is possible to set up a different network topology for each batch example. The batch size u

r""" Given a ``parameter_name`` used in the configuration file, returns a tensor with shape [batch size, number of BSs, number of UTs]

r""" Convert degree to radian Input ------ x : Tensor Angles in degree Output ------- y : Tensor

Compute the maximum relative error, ``r`` being the reference value, ``x`` an estimate of ``r``.

r""" Add three leading dimensions to array, with shape [1, num_tx, num_rx], to reshape it as expected by the channel impulse response

Field pattern in the vertical and horizontal polarization (TR 38.901 7.3-4/5) Inputs ------- theta: Zeni

Generate cluster delays. See step 5 of section 7.5 from TR 38.901 specification. Input ------ delay_spread :

r""" Generate vector on unit sphere (7.1-6) Input ------- theta : Arbitrary shape, tf.float Zenith [radia

Reference implementation: Unit to sphere vector

r""" Wrap ``angle`` to (0,360) Input ------ angle : Tensor Input to wrap Output ------- y : Tensor

Generate cluster powers. See step 6 of section 7.5 from TR 38.901 specification. Input ------ delays : [batc

r""" Transform field components from LCS to GCS (7.1-11) Input ------ f_prime : K-Dim Tensor of shape [...,2], tf.flo

Reference implementation: L2G response

Compute the azimuth angle of arrival (AoA) See step 7 of section 7.5 from TR 38.901 specification. Input ------

r""" Generate rays from ``angles`` (which could be ZoD, ZoA, AoD, or AoA) and the angle spread ``c`` using equation 7.7-0a of TR38.901

Randomly shuffle a tensor carrying azimuth/zenith angles of arrival/departure. Input ------ angles : [batch

Randomly shuffle a tensor carrying azimuth/zenith angles of arrival/departure. Input ------ angles : [batch

Compute the zenith angle of arrival (ZoA) See step 7 of section 7.5 from TR 38.901 specification. Input ------

Reference implementation: Forward rotation matrix

r""" Convert radian to degree Input ------ x : Tensor Angles in radian Output ------- y : Tensor

r""" Forward composite rotation matrix (7.1-4) Input ------ orientations : [...,3], tf.float Orie

r""" Compute the angles ``theta``, ``phi`` in LCS rotated according to ``orientations`` (7.1-7/8) Input ------

r""" Rotate the ``positions`` according to the ``orientations`` Input ------ orientations : [...,3], tf.float

r""" Compute matrix accounting for phases offsets between antenna elements Input ----- topology : Topology

r""" Compute matrix with phase shifts due to mobility in (7.5-22) Input ----- topology : Topology Topolog

r""" Compute matrix accounting for the element responses, random phases and xpr Input ----- topology : Topolo

Apply path loss and shadow fading ``sf`` to paths coefficients ``h``. Input ------ h : [batch size, num_tx, num_rx, num_paths

Reference implementation: GCS to LCS angles

r""" Sample UTs location from a sector or a cell Input -------- batch_size : int Batch size num_ut : int Number

r""" Sample a tensor with shape ``shape`` from a Bernoulli distribution with probability ``p`` Input -------- shape : Tensor shap

Generate pathlosses [dB] for each BS-UT link. Input ------ None Output ------- A tensor

Reference implementation: Array offset matrix

Reference implementation: Doppler matrix calculation

Reference implementation: Field matrix

Reference implemenrtation: Compute the channel matrix of the NLoS component

Reference implementation: Phase matrix

Generate departure or arrival azimuth angles (degrees). See step 7 of section 7.5 from TR 38.901 specification. Input

Compute the azimuth angle of departure (AoD) See step 7 of section 7.5 from TR 38.901 specification. Input ------

r"""Compute the maximum radian Doppler frequency [Hz] for a given speed [m/s]. The maximum radian Doppler frequency :math:`\omega_d`

Generate cross-polarization power ratios. Step 9 in TR 38.901 specification. Input ------ None Out

Randomly couples the angles within a cluster for both azimuth and elevation. #TODO adjusted for 811 #NOTE 5: For all

r""" Reverse composite rotation matrix (7.1-4) Input ------ orientations : [...,3], tf.float Orientations

r""" Generate random and uniformly distributed phases for all rays and polarization combinations Input -----

r""" Combine LOS and LOS components to compute (7.5-30) Input ----- phi: [batch size, num TXs, num RXs, num clusters,

r"""Compute d_bar_rx in (7.5-22), i.e., the positions in GCS of elements forming the receive antenna panel Input -----

r"""Compute d_bar_tx in (7.5-22), i.e., the positions in GCS of elements forming the transmit panel Input ----- topol

r"""Compute the LOS channels from (7.5-29) Intput ------ topology : Topology Network topology t : [num t

r""" Compute the full NLOS channel matrix (7.5-28) Input ----- phi: [batch size, num TXs, num RXs, num clusters, num

r""" Compute matrix with random phases in (7.5-22) Input ----- phi : [batch size, num TXs, num RXs, num clusters, num

r""" Compute the final NLOS matrix in (7.5-27) Input ------ h_full : [batch size, num_tx, num rx, num clusters, num r

Generate departure or arrival zenith angles (degrees). See step 7 of section 7.5 from TR 38.901 specification. Input

Compute the zenith angle of departure (ZoD) See step 7 of section 7.5 from TR 38.901 specification. Input ------

Reference implementation: Reverse rotation matrix

r""" Set valid parameters for a specified 3GPP system level ``scenario`` (DenseUrban, Urban, or SubUrban). If a parameter is given, then

Reference implementation: Positions of the RX array elements

Reference implementation: Positions of the TX array elements

Reference implementation: Compute the channel matrix of the NLoS component 2

Reference implementation: Compute the channel matrix of the NLoS component 2

Updates internal quantities. Must be called at every update of the scenario that changes the state of UTs or their locations.

Given a scenario (DenseUrban, Urban, or SubUrban), the number of clusters is different for different state of UT-BS links (LoS, NLoS,

Compute and store as attribute the square-root of the cross-LSPs correlation matrices for each BS-UT link, and then the correspondin

r""" Computes the following internal values: * 2D distances for all BS-UT pairs in the X-Y plane * 3D distances for all BS-UT

r"""Computes the mean and standard deviations of LSPs in log-domain

Compute the square root of the spatial correlation matrices of LSPs. The LSPs are correlated accross users according to the distance

r"""Computes the basic component of the pathloss [dB]

r"""Computes the building entry component of the pathloss [dB]

r"""Computes the atmospheric gasses component of the pathloss [dB]

r"""Computes the scintillation component of the pathloss [dB]

r""" Compute displacement angle :math:`Psi` for the transformation of LCS-GCS field components in (7.1-15) of TR38.901 specification

r"""Load parameters of a CDL model. The model parameters are stored as JSON files with the following keys: * los : boolean that indic

r"""Load parameters of a TDL model. The model parameters are stored as JSON files with the following keys: * los : boolean that indic

r"""Load the configuration files corresponding to the 2 possible states of UTs: LoS and NLoS

Randomly couples the angles within a cluster for both azimuth and elevation. Step 8 in TR 38.901 specification. Inp

r"""Sample 2D indoor distances for indoor devices, according to section 7.4.3.1 of TR 38.901.

r"""Set the LoS state of each UT randomly, following the procedure described in section 7.4.2 of TR 38.901. LoS state of each UT is ra

r"""Clip the carrier frequency ``fc`` in GHz for LSP calculation Input ----- fc : float Carrier frequency [GHz]

Reference implementation: Compute psi angle