srsLTE/srslte/lib/ch_estimation/chest_common.c

83 lines
2.3 KiB
C

/**
*
* \section COPYRIGHT
*
* Copyright 2013-2015 Software Radio Systems Limited
*
* \section LICENSE
*
* This file is part of the srsLTE library.
*
* srsLTE is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of
* the License, or (at your option) any later version.
*
* srsLTE is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* A copy of the GNU Affero General Public License can be found in
* the LICENSE file in the top-level directory of this distribution
* and at http://www.gnu.org/licenses/.
*
*/
#include <stdio.h>
#include <stdlib.h>
#include <strings.h>
#include <string.h>
#include <complex.h>
#include <math.h>
#include "srslte/ch_estimation/chest_common.h"
#include "srslte/utils/vector.h"
#include "srslte/utils/convolution.h"
void srslte_chest_set_triangle_filter(float *fil, int filter_len)
{
for (int i=0;i<filter_len/2;i++) {
fil[i] = i+1;
fil[i+filter_len/2+1]=filter_len/2-i;
}
fil[filter_len/2]=filter_len/2+1;
float s=0;
for (int i=0;i<filter_len;i++) {
s+=fil[i];
}
for (int i=0;i<filter_len;i++) {
fil[i]/=s;
}
}
/* Uses the difference between the averaged and non-averaged pilot estimates */
float srslte_chest_estimate_noise_pilots(cf_t *noisy, cf_t *noiseless, cf_t *noise_vec, uint32_t nof_pilots)
{
/* Substract noisy pilot estimates */
srslte_vec_sub_ccc(noiseless, noisy, noise_vec, nof_pilots);
/* Compute average power */
float power = srslte_vec_avg_power_cf(noise_vec, nof_pilots);
return power;
}
void srslte_chest_set_smooth_filter3_coeff(float *smooth_filter, float w)
{
smooth_filter[0] = w;
smooth_filter[2] = w;
smooth_filter[1] = 1-2*w;
}
void srslte_chest_average_pilots(cf_t *input, cf_t *output, float *filter,
uint32_t nof_ref, uint32_t nof_symbols, uint32_t filter_len)
{
for (int l=0;l<nof_symbols;l++) {
srslte_conv_same_cf(&input[l*nof_ref], filter, &output[l*nof_ref], nof_ref, filter_len);
}
}