53 lines
1.8 KiB
C
53 lines
1.8 KiB
C
/*
|
|
* SPDX-License-Identifier: LicenseRef-CSSL-1.0
|
|
*/
|
|
|
|
#include "PHY/sse_intrin.h"
|
|
#include "PHY/NR_ESTIMATION/nr_ul_estimation.h"
|
|
|
|
// Reference of openair1/PHY/LTE_ESTIMATION/freq_equalization.c
|
|
|
|
// This is 4096/(1:4096) in simde__m128i format
|
|
static simde__m128i nr_inv_ch[4096]; /* = {0, 4096/1, 4096/2, 4096/3, 4096/4...}*/
|
|
|
|
void nr_init_fde()
|
|
{
|
|
for (int i = 1;i < 4096; i++)
|
|
nr_inv_ch[i] = simde_mm_set1_epi16(4096/i);
|
|
}
|
|
|
|
void nr_freq_equalization(NR_DL_FRAME_PARMS *frame_parms,
|
|
c16_t *rxdataF_comp,
|
|
c16_t *ul_ch_mag,
|
|
c16_t *ul_ch_magb,
|
|
uint8_t symbol,
|
|
uint16_t Msc_RS,
|
|
uint8_t Qm)
|
|
{
|
|
simde__m128i *rxdataF_comp128 = (simde__m128i *)rxdataF_comp;
|
|
simde__m128i *ul_ch_mag128 = (simde__m128i *)ul_ch_mag;
|
|
simde__m128i *ul_ch_magb128 = (simde__m128i *)ul_ch_magb;
|
|
|
|
AssertFatal(symbol < frame_parms->symbols_per_slot, "symbol %d >= %d\n",
|
|
symbol, frame_parms->symbols_per_slot);
|
|
AssertFatal(Msc_RS <= frame_parms->N_RB_UL*12, "Msc_RS %d >= %d\n",
|
|
Msc_RS, frame_parms->N_RB_UL*12);
|
|
|
|
for (uint16_t re = 0; re < (Msc_RS >> 2); re++) {
|
|
int16_t amp = (*((int16_t*)&ul_ch_mag128[re]));
|
|
|
|
if (amp > 4095)
|
|
amp = 4095;
|
|
|
|
rxdataF_comp128[re] = simde_mm_srai_epi16(simde_mm_mullo_epi16(rxdataF_comp128[re],nr_inv_ch[amp]),3);
|
|
|
|
if (Qm == 4)
|
|
ul_ch_mag128[re] = simde_mm_set1_epi16(324); // this is 512*2/sqrt(10)
|
|
else if (Qm == 6) {
|
|
ul_ch_mag128[re] = simde_mm_set1_epi16(316); // this is 512*4/sqrt(42)
|
|
ul_ch_magb128[re] = simde_mm_set1_epi16(158); // this is 512*2/sqrt(42)
|
|
} else if(Qm != 2)
|
|
AssertFatal(1, "nr_freq_equalization(), Qm = %d, should be 2, 4 or 6. symbol=%d, Msc_RS=%d\n", Qm, symbol, Msc_RS);
|
|
}
|
|
}
|