spacecruft
/
strf
1
0
Fork 0
Code Releases Activity

Updated

pull/10/head
Cees Bassa 2014-06-09 12:24:38 +02:00
parent 5b70f48ae8
commit a82530df08
1 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
rffft.c
View File

@ -17,6 +17,7 @@ void usage(void)
printf("-c <chansize> Channel size [100Hz]\n"); printf("-c <chansize> Channel size [100Hz]\n");
printf("-t <tint> Integration time [1s]\n"); printf("-t <tint> Integration time [1s]\n");
printf("-n <nsub> Number of integrations per file [60]\n"); printf("-n <nsub> Number of integrations per file [60]\n");
printf("-m <use> Use every mth integration [1]\n");
printf("-h This help\n"); printf("-h This help\n");
return; return;
@ -24,12 +25,11 @@ void usage(void)
int main(int argc,char *argv[]) int main(int argc,char *argv[])
{ {
int i,j,k,l,m,nchan,nint=1,arg=0,nbytes,nsub=60,flag; int i,j,k,l,m,nchan,nint=1,arg=0,nbytes,nsub=60,flag,nuse=1;
fftwf_complex *c,*d; fftwf_complex *c,*d;
fftwf_plan fft; fftwf_plan fft;
FILE *infile,*outfile; FILE *infile,*outfile;
char infname[128],outfname[128],path[64]="/data/record",prefix[32]=""; char infname[128],outfname[128],path[64]="/data/record",prefix[32]="";
float *fbuf;
int16_t *ibuf; int16_t *ibuf;
float *z,length,fchan=100.0,tint=1.0; float *z,length,fchan=100.0,tint=1.0;
double freq,samp_rate; double freq,samp_rate;
@ -38,7 +38,7 @@ int main(int argc,char *argv[])
// Read arguments // Read arguments
if (argc>1) { if (argc>1) {
while ((arg=getopt(argc,argv,"i:f:s:c:t:p:n:h"))!=-1) { while ((arg=getopt(argc,argv,"i:f:s:c:t:p:n:hm:"))!=-1) {
switch(arg) { switch(arg) {
case 'i': case 'i':
@ -65,6 +65,10 @@ int main(int argc,char *argv[])
nsub=atoi(optarg); nsub=atoi(optarg);
break; break;
case 'm':
nuse=atoi(optarg);
break;
case 't': case 't':
tint=atof(optarg); tint=atof(optarg);
break; break;
@ -102,7 +106,6 @@ int main(int argc,char *argv[])
// Allocate // Allocate
c=fftwf_malloc(sizeof(fftwf_complex)*nchan); c=fftwf_malloc(sizeof(fftwf_complex)*nchan);
d=fftwf_malloc(sizeof(fftwf_complex)*nchan); d=fftwf_malloc(sizeof(fftwf_complex)*nchan);
fbuf=(float *) malloc(sizeof(float)*2*nchan);
ibuf=(int16_t *) malloc(sizeof(int16_t)*2*nchan); ibuf=(int16_t *) malloc(sizeof(int16_t)*2*nchan);
z=(float *) malloc(sizeof(float)*nchan); z=(float *) malloc(sizeof(float)*nchan);
@ -138,6 +141,10 @@ int main(int argc,char *argv[])
if (nbytes==0) if (nbytes==0)
break; break;
// Skip buffer
if (j%nuse!=0)
continue;
// Unpack // Unpack
for (i=0;i<nchan;i++) { for (i=0;i<nchan;i++) {
c[i][0]=(float) ibuf[2*i]/32768.0; c[i][0]=(float) ibuf[2*i]/32768.0;
@ -163,7 +170,7 @@ int main(int argc,char *argv[])
// Scale // Scale
for (i=0;i<nchan;i++) for (i=0;i<nchan;i++)
z[i]/=(float) nchan; z[i]*=(float) nuse/(float) nchan;
// Time stats // Time stats
length=(end.tv_sec-start.tv_sec)+(end.tv_usec-start.tv_usec)*1e-6; length=(end.tv_sec-start.tv_sec)+(end.tv_usec-start.tv_usec)*1e-6;
@ -199,7 +206,6 @@ int main(int argc,char *argv[])
fftwf_destroy_plan(fft); fftwf_destroy_plan(fft);
// Deallocate // Deallocate
free(fbuf);
free(ibuf); free(ibuf);
fftwf_free(c); fftwf_free(c);
fftwf_free(d); fftwf_free(d);