srsLTE/lte/phy/lib/resampling/test/resample_arb_test.c

64 lines
1.4 KiB
C

#include <stdio.h>
#include <stdlib.h>
#include <strings.h>
#include <unistd.h>
#include <math.h>
#include <time.h>
#include "liblte/phy/phy.h"
#include "liblte/phy/resampling/resample_arb.h"
typedef _Complex float cf_t;
int main(int argc, char **argv) {
int N = 100; // Number of sinwave samples
int delay = 5; // Delay of our resampling filter
float down = 25.0; // Downsampling rate
for(int up=1;up<down;up++)
{
float rate = up/down;
printf("Testing rate %f\n", rate);
cf_t *in = malloc(N*sizeof(cf_t));
if(!in) {
perror("malloc");
exit(-1);
}
cf_t *out = malloc(N*sizeof(cf_t));
if(!out) {
perror("malloc");
exit(-1);
}
for(int i=0;i<N;i++)
in[i] = sin(i*2*M_PI/N);
// Resample
resample_arb_t r;
resample_arb_init(&r, rate);
int n_out = resample_arb_compute(&r, in, out, N);
// Check interp values
for(int i=delay+1;i<n_out;i++){
float idx = i/rate;
int pre = floor(idx)-delay;
int post = ceil(idx)-delay;
int round = roundf(idx)-delay;
float diff = fabs(creal(in[pre])-creal(in[post]));
float diff2 = fabs(creal(out[i])-creal(in[round]));
if(diff2 > diff && pre != post){
printf("Interpolation failed at index %f", idx);
exit(-1);
}
}
free(in);
free(out);
}
printf("Ok\n");
exit(0);
}