mirror/kissfft
1
0
Fork 0
mirror of https://github.com/mborgerding/kissfft.git synced 2025-06-03 17:18:11 -04:00
Code Issues Packages Projects Releases Wiki Activity

fast conv filtering cleaner ifc

Browse Source
This commit is contained in:
Mark Borgerding 2004-01-24 04:25:19 +00:00
parent 5005a2a99d
commit e335546c3e
1 changed files with 42 additions and 75 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

117
tools/kiss_fastfir.c
View File

@ -30,18 +30,16 @@ typedef kiss_fft_cpx kffsamp_t;
#define FFTINV kiss_fft #define FFTINV kiss_fft
#endif #endif
static int verbose=0;
void * kiss_fastfir_alloc(const kffsamp_t * imp_resp,size_t n_imp_resp, void * kiss_fastfir_alloc(const kffsamp_t * imp_resp,size_t n_imp_resp,
size_t * nfft,void * mem,size_t*lenmem); size_t * nfft,void * mem,size_t*lenmem);
/* n : the size of inbuf and outbuf in samples /* see do_file_filter for usage */
return value: the number of samples completely processed size_t kiss_fastfir( void * cfg, kffsamp_t * inbuf, kffsamp_t * outbuf, size_t n, int do_flush, size_t *offset);
n-retval samples should be copied to the front of the next input buffer */
size_t kff_nocopy( void *st, const kffsamp_t * inbuf, kffsamp_t * outbuf, size_t n);
static int verbose=0;
typedef struct { typedef struct {
int nfft; int nfft;
size_t ngood; size_t ngood;
@ -79,6 +77,8 @@ void * kiss_fastfir_alloc(
nfft<<=1; nfft<<=1;
}while (i>>=1); }while (i>>=1);
} }
if (pnfft)
*pnfft = nfft;
#ifdef REAL_FASTFIR #ifdef REAL_FASTFIR
n_freq_bins = nfft/2 + 1; n_freq_bins = nfft/2 + 1;
@ -148,8 +148,6 @@ void * kiss_fastfir_alloc(
st->fir_freq_resp[i].r *= scale; st->fir_freq_resp[i].r *= scale;
st->fir_freq_resp[i].i *= scale; st->fir_freq_resp[i].i *= scale;
} }
if (pnfft)
*pnfft = nfft;
return st; return st;
} }
@ -172,7 +170,7 @@ static void fastconv1buf(const kiss_fastfir_state *st,const kffsamp_t * in,kffsa
/* n : the size of inbuf and outbuf in samples /* n : the size of inbuf and outbuf in samples
return value: the number of samples completely processed return value: the number of samples completely processed
n-retval samples should be copied to the front of the next input buffer */ n-retval samples should be copied to the front of the next input buffer */
size_t kff_nocopy( static size_t kff_nocopy(
void *vst, void *vst,
const kffsamp_t * inbuf, const kffsamp_t * inbuf,
kffsamp_t * outbuf, kffsamp_t * outbuf,
@ -189,6 +187,7 @@ size_t kff_nocopy(
return norig - n; return norig - n;
} }
static
size_t kff_flush(void *vst,const kffsamp_t * inbuf,kffsamp_t * outbuf,size_t n) size_t kff_flush(void *vst,const kffsamp_t * inbuf,kffsamp_t * outbuf,size_t n)
{ {
size_t zpad=0,ntmp; size_t zpad=0,ntmp;
@ -209,55 +208,30 @@ size_t kff_flush(void *vst,const kffsamp_t * inbuf,kffsamp_t * outbuf,size_t n)
return ntmp + st->ngood - zpad; return ntmp + st->ngood - zpad;
} }
size_t kiss_fastfir(
void * vst,
kffsamp_t * inbuf,
kffsamp_t * outbuf,
size_t n,
int do_flush,
size_t *offset)
{
if (do_flush==0) {
size_t nwritten = kff_nocopy(vst,inbuf,outbuf,n);
*offset = n - nwritten;
memcpy( inbuf , inbuf+nwritten , *offset * sizeof(kffsamp_t) );
return nwritten;
}else{
/*flush*/
return kff_flush(vst,inbuf,outbuf,n);
}
}
#ifdef FAST_FILT_UTIL #ifdef FAST_FILT_UTIL
#include <unistd.h> #include <unistd.h>
#include <sys/types.h> #include <sys/types.h>
#include <sys/mman.h> #include <sys/mman.h>
int do_mmap_filter(int fd_src,int fd_dst,const kffsamp_t * imp_resp, size_t n_imp_resp, size_t nfft )
{
int error_code=1;
off_t src_len=0,dst_len=0;
void *psrc=NULL;
void *pdst=NULL;
src_len = lseek(fd_src,0,SEEK_END );
if (src_len == (off_t)-1 ) goto nommap;
if ( ftruncate(fd_dst,src_len) ) goto nommap;
if ( lseek(fd_src,0,SEEK_SET) == (off_t)-1 ) goto nommap;
if ( lseek(fd_dst,0,SEEK_SET) == (off_t)-1 ) goto nommap;
psrc = mmap(0,src_len,PROT_READ,MAP_SHARED,fd_src,0);
if (psrc==NULL) goto nommap;
pdst = mmap(0,src_len,PROT_WRITE,MAP_SHARED,fd_dst,0);
if (psrc==NULL) goto nommap;
{
const kffsamp_t *inbuf= (const kffsamp_t *)psrc;
kffsamp_t *outbuf= (kffsamp_t *)pdst;
size_t noutbuf;
size_t nsamps_in = src_len / sizeof(kffsamp_t);
void * cfg = kiss_fastfir_alloc(imp_resp,n_imp_resp,&nfft,0,0);
noutbuf = kff_nocopy( cfg, (kffsamp_t*)psrc, (kffsamp_t*)pdst, nsamps_in );
noutbuf += kff_flush( cfg,inbuf+noutbuf,outbuf+noutbuf,nsamps_in-noutbuf);
dst_len = noutbuf* sizeof(kffsamp_t);
free(cfg);
}
error_code=0;
nommap:
if (psrc) munmap(psrc,src_len);
if (pdst) {
munmap(pdst,src_len);
ftruncate(fd_dst,dst_len);
}
if (error_code)
perror("mapping");
return error_code;
}
void do_file_filter( void do_file_filter(
FILE * fin, FILE * fin,
FILE * fout, FILE * fout,
@ -265,27 +239,25 @@ void do_file_filter(
size_t n_imp_resp, size_t n_imp_resp,
size_t nfft ) size_t nfft )
{ {
void * cfg = kiss_fastfir_alloc(imp_resp,n_imp_resp,&nfft,0,0); size_t memcfg,membuf;
size_t max_buf=4*nfft; void * cfg;
kffsamp_t *inbuf = (kffsamp_t*)malloc(max_buf*sizeof(kffsamp_t)); size_t max_samps;
kffsamp_t *outbuf = (kffsamp_t*)malloc(max_buf*sizeof(kffsamp_t)); kffsamp_t *inbuf,*outbuf;
size_t ninbuf,noutbuf; size_t ninbuf,noutbuf;
size_t idx_inbuf=0; size_t idx_inbuf=0;
do{ kiss_fastfir_alloc(imp_resp,n_imp_resp,&nfft,0,&memcfg);
ninbuf = fread(&inbuf[idx_inbuf],sizeof(inbuf[0]),max_buf-idx_inbuf,fin ); max_samps = 4*nfft;
if (ninbuf) { membuf = max_samps*sizeof(kffsamp_t);
/* add new input to saved input */ cfg = malloc(membuf*2+memcfg);
ninbuf += idx_inbuf; kiss_fastfir_alloc(imp_resp,n_imp_resp,&nfft,cfg,&memcfg);
noutbuf = kff_nocopy( cfg, inbuf, outbuf, ninbuf ); inbuf = (kffsamp_t*)((char*)cfg+memcfg);
outbuf = (kffsamp_t*)((char*)cfg+memcfg+membuf);
/* move the unconsumed input samples to the front of the input buffer */ do{
idx_inbuf = ninbuf - noutbuf; ninbuf = fread( &inbuf[idx_inbuf], sizeof(inbuf[0]), max_samps-idx_inbuf , fin );
memmove( inbuf , &inbuf[noutbuf] , sizeof( inbuf[0] )*( ninbuf - noutbuf ) );
}else{ noutbuf = kiss_fastfir(cfg, inbuf, outbuf,idx_inbuf+ninbuf,ninbuf==0,&idx_inbuf);
/* last read -- flush */
noutbuf = kff_flush(cfg, inbuf,outbuf,idx_inbuf);
}
if ( fwrite( outbuf, sizeof(outbuf[0]), noutbuf, fout) != noutbuf ) { if ( fwrite( outbuf, sizeof(outbuf[0]), noutbuf, fout) != noutbuf ) {
perror("short write"); perror("short write");
@ -293,8 +265,6 @@ void do_file_filter(
} }
}while ( ninbuf ); }while ( ninbuf );
free(cfg); free(cfg);
free(inbuf);
free(outbuf);
} }
int main(int argc,char**argv) int main(int argc,char**argv)
@ -356,10 +326,7 @@ int main(int argc,char**argv)
fread(h,sizeof(kffsamp_t),nh,filtfile); fread(h,sizeof(kffsamp_t),nh,filtfile);
fclose(filtfile); fclose(filtfile);
#if 0 do_file_filter( fin, fout, h,nh,nfft);
if (do_mmap_filter( fileno(fin), fileno(fout), h,nh,nfft ) )
#endif
do_file_filter( fin, fout, h,nh,nfft);
if (fout!=stdout) fclose(fout); if (fout!=stdout) fclose(fout);
if (fin!=stdin) fclose(fin); if (fin!=stdin) fclose(fin);