Added resonator FIR

This commit is contained in:
ha7ilm 2017-04-09 14:53:22 +02:00
parent 874d6b7c06
commit c1e953cd4e
3 changed files with 74 additions and 17 deletions

64
csdr.c
View file

@ -136,7 +136,8 @@ char usage[]=
" bpsk_costas_loop_cc <samples_per_bits>\n"
" binary_slicer_f_u8\n"
" simple_agc_cc <rate> [reference [max_gain]]\n"
" firdes_carrier_c <rate> <length> [window [--octave]]\n"
" firdes_resonator_c <rate> <length> [window [--octave]]\n"
" resonators_fir_cc <taps_length> [resonator_rate × N]\n"
" ?<search_the_function_list>\n"
" =<evaluate_python_expression>\n"
" \n"
@ -1054,6 +1055,7 @@ int main(int argc, char *argv[])
TRY_YIELD;
}
}
if(!strcmp(argv[1],"fir_decimate_cc"))
{
bigbufs=1;
@ -1111,7 +1113,6 @@ int main(int argc, char *argv[])
output_size=fir_decimate_cc((complexf*)input_buffer, (complexf*)output_buffer, the_bufsize, factor, taps, padded_taps_length);
//fprintf(stderr, "os %d\n",output_size);
fwrite(output_buffer, sizeof(complexf), output_size, stdout);
fflush(stdout);
TRY_YIELD;
input_skip=factor*output_size;
memmove((complexf*)input_buffer,((complexf*)input_buffer)+input_skip,(the_bufsize-input_skip)*sizeof(complexf)); //memmove lets the source and destination overlap
@ -1167,7 +1168,6 @@ int main(int argc, char *argv[])
output_size=fir_interpolate_cc((complexf*)input_buffer, (complexf*)interp_output_buffer, the_bufsize, factor, taps, taps_length);
//fprintf(stderr, "os %d\n",output_size);
fwrite(interp_output_buffer, sizeof(complexf), output_size, stdout);
fflush(stdout);
TRY_YIELD;
input_skip=output_size/factor;
memmove((complexf*)input_buffer,((complexf*)input_buffer)+input_skip,(the_bufsize-input_skip)*sizeof(complexf)); //memmove lets the source and destination overlap
@ -2389,7 +2389,6 @@ int main(int argc, char *argv[])
if(serial.input_used==0) { fprintf(stderr, "%s: error: serial_line_decoder_f_u8() got stuck.\n", argv[1]); return -3; }
//printf("now out %d | ", serial.output_size);
fwrite(output_buffer, sizeof(unsigned char), serial.output_size, stdout);
fflush(stdout);
TRY_YIELD;
}
}
@ -2468,7 +2467,6 @@ int main(int argc, char *argv[])
timing_recovery_cc((complexf*)input_buffer, (complexf*)output_buffer, the_bufsize, &state);
//fprintf(stderr, "trcc is=%d, os=%d, ip=%d\n",the_bufsize, state.output_size, state.input_processed);
fwrite(output_buffer, sizeof(complexf), state.output_size, stdout);
fflush(stdout);
TRY_YIELD;
//fprintf(stderr, "state.input_processed = %d\n", state.input_processed);
memmove((complexf*)input_buffer,((complexf*)input_buffer)+state.input_processed,(the_bufsize-state.input_processed)*sizeof(complexf)); //memmove lets the source and destination overlap
@ -2701,7 +2699,7 @@ int main(int argc, char *argv[])
}
}
if(!strcmp(argv[1],"firdes_carrier_c")) //<rate> <length> [window [--octave]]
if(!strcmp(argv[1],"firdes_resonator_c")) //<rate> <length> [window [--octave]]
{
//Process the params
if(argc<=3) return badsyntax("need required parameters (rate, length)");
@ -2717,14 +2715,14 @@ int main(int argc, char *argv[])
{
window=firdes_get_window_from_string(argv[4]);
}
else fprintf(stderr,"firdes_carrier_c: window = %s\n",firdes_get_string_from_window(window));
else fprintf(stderr,"firdes_resonator_c: window = %s\n",firdes_get_string_from_window(window));
int octave=(argc>=6 && !strcmp("--octave",argv[5]));
complexf* taps=(complexf*)calloc(sizeof(complexf),length);
//Make the filter
firdes_add_carrier_c(taps, length, rate, window);
firdes_add_resonator_c(taps, length, rate, window);
//Do the output
if(octave) printf("taps=[");
@ -2736,15 +2734,59 @@ int main(int argc, char *argv[])
"];figure(\"Position\",[0 0 1000 1000]);fser=fft([taps,zeros(1,%d)]);ampl=abs(fser).^2;halfindex=floor(1+size(ampl)(2)/2);\n"
"amplrev=[ampl(halfindex:end),ampl(1:halfindex)];\n" //we have to swap the output of FFT
"subplot(2,1,1);plot(amplrev);\n"
"subplot(2,1,2);plot(arg(fser));\n"
"#figure(2);freqz(taps);\n"
"#figur(3);plot3(taps);\n",fft_length-length);
"subplot(2,1,2);plot(arg(fser));\n",fft_length-length);
//Wait forever, so that octave won't close just after popping up the window.
//You can close it with ^C.
if(octave) { fflush(stdout); getchar(); }
return 0;
}
if(!strcmp(argv[1],"resonators_fir_cc")) //<taps_length> <resonator_rate × N>
{
if(argc<=2) return badsyntax("need required parameter (taps_length)");
int taps_length;
sscanf(argv[2],"%d",&taps_length);
int num_resonators = argc-3;
float* resonator_rate = (float*)malloc(sizeof(float)*num_resonators);
for(int i=0;i<num_resonators;i++)
sscanf(argv[3+i], "%f", resonator_rate+i);
if(num_resonators<=0) return badsyntax("need required parameter (resonator_rate) once or multiple times");
for(int i=0;i<num_resonators;i++)
fprintf(stderr, "%f\n", resonator_rate[i]);
fflush(stderr);
window_t window = WINDOW_DEFAULT;
if(!initialize_buffers()) return -2;
sendbufsize(the_bufsize);
complexf* taps = (complexf*)calloc(sizeof(complexf),taps_length);
for(int i=0; i<num_resonators; i++)
{
firdes_add_resonator_c(taps, taps_length, resonator_rate[i], window);
}
for(int i=0; i<taps_length; i++)
{
taps[i].i/=num_resonators;
taps[i].q/=num_resonators;
}
int output_size=0;
FREAD_C;
for(;;)
{
FEOF_CHECK;
output_size = apply_fir_cc((complexf*)input_buffer, (complexf*)output_buffer, the_bufsize, taps, taps_length);
fwrite(output_buffer, sizeof(complexf), output_size, stdout);
fprintf(stderr, "os = %d\n", output_size);
TRY_YIELD;
memmove((complexf*)input_buffer,((complexf*)input_buffer)+output_size,(the_bufsize-output_size)*sizeof(complexf));
fread(((complexf*)input_buffer)+(the_bufsize-output_size), sizeof(complexf), output_size, stdin);
}
}
if(!strcmp(argv[1],"none"))
{

View file

@ -2009,13 +2009,13 @@ void simple_agc_cc(complexf* input, complexf* output, int input_size, float rate
}
}
void firdes_add_carrier_c(complexf* output, int length, float rate, window_t window)
void firdes_add_resonator_c(complexf* output, int length, float rate, window_t window)
{
complexf* taps = (complexf*)malloc(sizeof(complexf)*length);
int middle=length/2;
float phase = 0, phase_addition = rate*M_PI*2;
float phase = 0, phase_addition = -rate*M_PI*2;
float (*window_function)(float) = firdes_get_window_kernel(window);
for(int i=0; i<length; i++) //@@firdes_carrier_c: calculate taps
for(int i=0; i<length; i++) //@@firdes_add_resonator_c: calculate taps
{
e_powj(&taps[i], phase);
float window_multiplier = window_function(fabs((float)(middle-i)/middle));
@ -2023,15 +2023,16 @@ void firdes_add_carrier_c(complexf* output, int length, float rate, window_t win
taps[i].q *= window_multiplier;
phase += phase_addition;
while(phase>2*M_PI) phase-=2*M_PI;
while(phase<0) phase+=2*M_PI;
}
//Normalize filter kernel
float sum=0;
for(int i=0;i<length;i++) //@firdes_carrier_c: normalize pass 1
for(int i=0;i<length;i++) //@firdes_add_resonator_c: normalize pass 1
{
sum+=sqrt(taps[i].i*taps[i].i + taps[i].q*taps[i].q);
}
for(int i=0;i<length;i++) //@firdes_carrier_c: normalize pass 2
for(int i=0;i<length;i++) //@firdes_add_resonator_c: normalize pass 2
{
taps[i].i/=sum;
taps[i].q/=sum;
@ -2043,6 +2044,19 @@ void firdes_add_carrier_c(complexf* output, int length, float rate, window_t win
}
}
int apply_fir_cc(complexf* input, complexf* output, int input_size, complexf* taps, int taps_length)
{
int i;
for(i=0; i<input_size-taps_length+1; i++)
{
csetnull(&output[i]);
for(int ti=0;ti<taps_length;ti++)
{
cmultadd(&output[i], &input[i+ti], &taps[ti]);
}
}
return i;
}

View file

@ -338,4 +338,5 @@ typedef struct bpsk_costas_loop_state_s
bpsk_costas_loop_state_t init_bpsk_costas_loop_cc(float samples_per_bits);
void bpsk_costas_loop_cc(complexf* input, complexf* output, int input_size, bpsk_costas_loop_state_t* state);
void simple_agc_cc(complexf* input, complexf* output, int input_size, float rate, float reference, float max_gain, float* current_gain);
void firdes_add_carrier_c(complexf* output, int length, float rate, window_t window);
void firdes_add_resonator_c(complexf* output, int length, float rate, window_t window);
int apply_fir_cc(complexf* input, complexf* output, int input_size, complexf* taps, int taps_length);