jens
e93e7927cb
git-svn-id: http://moon:8086/svn/software/trunk/libsrc/avpflms@1 b431acfa-c32f-4a4a-93f1-934dc6c82436
154 lines
3.8 KiB
C
Executable File
154 lines
3.8 KiB
C
Executable File
/*-------------------------------------------------------------------------
|
|
* pfft.h Partioned Fast Fourier Transform
|
|
* $Id: $
|
|
*-------------------------------------------------------------------------
|
|
*
|
|
* Copyright (C) 2001 Algo Vision Systems GmbH
|
|
*
|
|
*-------------------------------------------------------------------------
|
|
*/
|
|
|
|
#ifndef _PFFT_H
|
|
#define _PFFT_H
|
|
#include "avtypes.h"
|
|
#include "avfft.h"
|
|
|
|
/*-----------------------------------------------------------------------*/
|
|
/* PFLMS-Parameter */
|
|
/*-----------------------------------------------------------------------*/
|
|
|
|
typedef struct _sCOMPLEX
|
|
{
|
|
avfloat_t *pReal, *pImag;
|
|
} COMPLEX;
|
|
|
|
typedef struct _sREALBUF
|
|
{
|
|
avfloat_t *pData;
|
|
struct _sREALBUF *pNext, *pLast, *pLastPS;
|
|
UINT32 user;
|
|
} REALBUF;
|
|
|
|
typedef struct _sCMPXBUF
|
|
{
|
|
struct _sCOMPLEX cmpxData;
|
|
struct _sCMPXBUF *pNext, *pLast, *pLastPS;
|
|
UINT32 user;
|
|
} CMPXBUF;
|
|
|
|
typedef struct _sPFFT
|
|
{
|
|
UINT32 N, P, S, L, C; /* PFLMS-Parameter N, P, S, L, C */
|
|
CMPXBUF *pBufX; /* komplexer Puffer X[P*S][C] */
|
|
CMPXBUF *pX; /* aktuelle Zeiger auf Puffer[p][] */
|
|
COMPLEX *pY; /* temporaerer Puffer fuer Y[] und E[] */
|
|
COMPLEX BufXsave; /* Overlap-Save Puffer xs[S*L] */
|
|
FFT *pFFT; /* FFT-Objekt */
|
|
} PFFT;
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
AVERR PfftInit(PFFT *pObj,UINT32 N, UINT32 P, UINT32 S, UINT32 L, UINT32 C);
|
|
AVERR PfftFilterAlloc(PFFT *pObj, CMPXBUF **ppH);
|
|
AVERR PfftFilterInit(PFFT *pObj, COMPLEX WTD, CMPXBUF *pH);
|
|
AVERR PfftFilter(PFFT *pObj, CMPXBUF *pH, COMPLEX x, COMPLEX y);
|
|
AVERR PfftFilterFast(PFFT *pObj, CMPXBUF *pH);
|
|
avfloat_t *PfftGetBufInRe(PFFT *pObj);
|
|
avfloat_t *PfftGetBufInIm(PFFT *pObj);
|
|
avfloat_t *PfftGetBufOutRe(PFFT *pObj);
|
|
avfloat_t *PfftGetBufOutIm(PFFT *pObj);
|
|
|
|
/*-----------------------------------------------------------------*/
|
|
/* Complex-Funktionen
|
|
/*-----------------------------------------------------------------*/
|
|
_inline void CmpxVectMul(
|
|
struct _sCOMPLEX *pA,
|
|
struct _sCOMPLEX *pB,
|
|
struct _sCOMPLEX *pAB,
|
|
UINT32 len)
|
|
{
|
|
UINT32 i;
|
|
|
|
for (i=0; i < len; i++)
|
|
{
|
|
pAB->pReal[i] = pA->pReal[i]*pB->pReal[i]
|
|
- pA->pImag[i]*pB->pImag[i];
|
|
pAB->pImag[i] = pA->pReal[i]*pB->pImag[i]
|
|
+ pA->pImag[i]*pB->pReal[i];
|
|
}
|
|
}
|
|
|
|
_inline void CmpxVectAdd(
|
|
struct _sCOMPLEX *pA,
|
|
struct _sCOMPLEX *pB,
|
|
struct _sCOMPLEX *pAB,
|
|
UINT32 len)
|
|
{
|
|
UINT32 i;
|
|
|
|
for (i=0; i < len; i++)
|
|
{
|
|
pAB->pReal[i] = pA->pReal[i] + pB->pReal[i];
|
|
pAB->pImag[i] = pA->pImag[i] + pB->pImag[i];
|
|
}
|
|
}
|
|
|
|
_inline void CmpxVectMac(
|
|
struct _sCOMPLEX *pA,
|
|
struct _sCOMPLEX *pB,
|
|
struct _sCOMPLEX *pAB,
|
|
UINT32 len)
|
|
{
|
|
UINT32 i;
|
|
|
|
for (i=0; i < len; i++)
|
|
{
|
|
pAB->pReal[i] += pA->pReal[i]*pB->pReal[i]
|
|
- pA->pImag[i]*pB->pImag[i];
|
|
pAB->pImag[i] += pA->pReal[i]*pB->pImag[i]
|
|
+ pA->pImag[i]*pB->pReal[i];
|
|
}
|
|
}
|
|
|
|
_inline void CmpxVectMulS(
|
|
struct _sCOMPLEX *pA,
|
|
struct _sCOMPLEX *pB,
|
|
struct _sCOMPLEX *pAB,
|
|
UINT32 len)
|
|
{
|
|
UINT32 i;
|
|
|
|
for (i=0; i < len; i++)
|
|
{
|
|
pAB->pReal[i] = (pA->pReal[i]*pB->pReal[i]
|
|
- pA->pImag[i]*pB->pImag[i])*len;
|
|
pAB->pImag[i] = (pA->pReal[i]*pB->pImag[i]
|
|
+ pA->pImag[i]*pB->pReal[i])*len;
|
|
}
|
|
}
|
|
|
|
_inline void CmpxVectMacS(
|
|
struct _sCOMPLEX *pA,
|
|
struct _sCOMPLEX *pB,
|
|
struct _sCOMPLEX *pAB,
|
|
UINT32 len)
|
|
{
|
|
UINT32 i;
|
|
|
|
for (i=0; i < len; i++)
|
|
{
|
|
pAB->pReal[i] += (pA->pReal[i]*pB->pReal[i]
|
|
- pA->pImag[i]*pB->pImag[i])*len;
|
|
pAB->pImag[i] += (pA->pReal[i]*pB->pImag[i]
|
|
+ pA->pImag[i]*pB->pReal[i])*len;
|
|
}
|
|
}
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
#endif /* _PFFT_H */
|
|
|