PDSCH

This module contains classes related to the Physical Downlink Shared Channel, PDSCH.

class aerial.phy5g.pdsch.pdsch_tx.PdschTx

PDSCH transmitter.

This class implements the whole PDSCH transmission pipeline from the transmitted transport block to the transmitted frequency-domain symbols.

__init__(cell_id, num_rx_ant, num_tx_ant, num_ul_bwp=273, num_dl_bwp=273, mu=1)

Initialize PdschTx.

Parameters

• cell_id (int) – Physical cell ID.

• num_rx_ant (int) – Number of receive antennas.

• num_tx_ant (int) – Number of transmit antennas.

• num_ul_bwp (int) – Number of PRBs in a uplink bandwidth part. Default: 273.

• num_dl_bwp (int) – Number of PRBs in a downlink bandwidth part. Default: 273.

• mu (int) – Numerology. Values in [0, 3]. Default: 1.

Return type

None

run(tb_inputs, num_ues, slot, num_dmrs_cdm_grps_no_data=2, dmrs_scrm_id=41, resource_alloc=1, prb_bitmap=None, start_prb=0, num_prbs=273, dmrs_syms=None, start_sym=2, num_symbols=12, scids=None, layers=None, dmrs_ports=None, bwp_starts=None, ref_points=None, rntis=None, data_scids=None, precoding_matrices=None, mcs_tables=None, mcs_indices=None, code_rates=None, mod_orders=None, rvs=None, num_prb_lbrms=None, max_layers=None, max_qms=None)

Run PDSCH transmission.

Set dynamic PDSCH parameters and call cuPHY to run the PDSCH transmission.

Parameters

• tb_inputs (List[np.ndarray]) – Transport blocks in bytes for each UE.

• num_ues (int) – Number of UEs.

• slot (int) – Slot number.

• num_dmrs_cdm_grps_no_data (int) – Number of DMRS CDM groups without data [3GPP TS 38.212, sec 7.3.1.1]. Value: 1->3.

• dmrs_scrm_id (int) – Downlink DMRS scrambling ID.

• resource_alloc (int) – Resource allocation type.

• prb_bitmap (List[int]) – Array of bytes indicating bitmask for allocated RBs.

• start_prb (int) – Start PRB index for the UE group.

• num_prbs (int) – Number of allocated PRBs for the UE group.

• dmrs_syms (List[int]) – For the UE group, a list of binary numbers each indicating whether the corresponding symbol is a DMRS symbol.

• start_sym (int) – Start OFDM symbol index of the UE group allocation.

• num_symbols (int) – Number of symbols in the allocation, starting from start_sym.

• scids (List[int]) – DMRS sequence initialization for each UE [TS38.211, sec 7.4.1.1.2].

• layers (List[int]) – Number of layers for each UE.

• dmrs_ports (List[int]) – DMRS ports for each UE. The format of each entry is in the SCF FAPI format as follows: A bitmap (mask) starting from the LSB where each bit indicates whether the corresponding DMRS port index is used.

• bwp_starts (List[int]) – Bandwidth part start (PRB number starting from 0). Used only if reference point is 1.

• ref_points (List[int]) – DMRS reference point per UE. Value 0 or 1.

• rntis (List[int]) –

• data_scids (List[int]) – Data scrambling IDs for each UE, more precisely dataScramblingIdentityPdsch [TS38.211, sec 7.3.1.1].

• precoding_matrices (List[np.ndarray]) – Precoding matrices, one per UE. The shape of each precoding matrix is number of layers x number of Tx antennas. If set to None, precoding is disabled.

• mcs_tables (List[int]) – MCS table per UE.

• mcs_indices (List[int]) – MCS index per UE.

• code_rates (List[int]) – Code rate for each UE in 3GPP format, i.e. code rate x 1024.

• mod_orders (List[int]) – Modulation order for each UE.

• rvs (List[int]) – Redundancy version per UE (default: 0 for each UE).

• num_prb_lbrms (List[int]) – Number of PRBs used for LBRM TB size computation. Possible values: {32, 66, 107, 135, 162, 217, 273}.

• max_layers (List[int]) – Number of layers used for LBRM TB size computation (at most 4).

• max_qms (List[int]) – Modulation order used for LBRM TB size computation. Value: 6 or 8.

Returns

A tuple containing:

• np.ndarray: Transmitted OFDM symbols in a frequency x time x antenna tensor.

• np.ndarray: Coded bits in a num_codewords x num_bits_per_codeword tensor.

Return type

np.ndarray, np.ndarray

classmethod cuphy_to_tx(tx_slot, num_ues, dmrs_ports, scids, precoding_matrices=None)

Map cuPHY outputs to Tx antenna ports.

Parameters

• tx_slot (numpy.ndarray) – Transmit buffer from cuPHY.

• num_ues (int) – Number of UEs.

• dmrs_ports (List[int]) – DMRS ports for each UE. The format of each entry is in the SCF FAPI format as follows: A bitmap (mask) starting from the LSB where each bit indicates whether the corresponding DMRS port index is used.

• scids (List[int]) – DMRS sequence initialization for each UE [TS38.211, sec 7.4.1.1.2].

• precoding_matrices (List[np.ndarray]) – Precoding matrices, one per UE. The shape of each precoding matrix is number of layers x number of Tx antennas. If set to None, precoding is disabled.

Returns

Transmitted OFDM symbols in a frequency x time x antenna tensor.

Return type

np.ndarray