221 lines
5.9 KiB
C
221 lines
5.9 KiB
C
|
#include "includes.h"
|
|||
|
|
|
|||
|
|
/*FFT模块资源互斥方式配置*/
|
|||
|
|
#define CRITICAL_NULL 0
|
|||
|
|
#define CRITICAL_IRQ 1
|
|||
|
|
#define CRITICAL_MUTEX 2
|
|||
|
|
#define CRITICAL_SPIN_LOCK 3
|
|||
|
|
#define FFT_CRITICAL_MODE CRITICAL_SPIN_LOCK
|
|||
|
|
|
|||
|
|
#if (FFT_CRITICAL_MODE == CRITICAL_MUTEX)
|
|||
|
|
static OS_MUTEX fft_mutex;
|
|||
|
|
#define FFT_CRITICAL_INIT() os_mutex_create(&fft_mutex)
|
|||
|
|
#define FFT_ENTER_CRITICAL() os_mutex_pend(&fft_mutex, 0)
|
|||
|
|
#define FFT_EXIT_CRITICAL() os_mutex_post(&fft_mutex)
|
|||
|
|
#elif (FFT_CRITICAL_MODE == CRITICAL_IRQ)
|
|||
|
|
#define FFT_CRITICAL_INIT(...)
|
|||
|
|
#define FFT_ENTER_CRITICAL() local_irq_disable()
|
|||
|
|
#define FFT_EXIT_CRITICAL() local_irq_enable()
|
|||
|
|
#elif (FFT_CRITICAL_MODE == CRITICAL_SPIN_LOCK)
|
|||
|
|
static spinlock_t fft_lock;
|
|||
|
|
#define FFT_CRITICAL_INIT() spin_lock_init(&fft_lock)
|
|||
|
|
#define FFT_ENTER_CRITICAL() spin_lock(&fft_lock)
|
|||
|
|
#define FFT_EXIT_CRITICAL() spin_unlock(&fft_lock)
|
|||
|
|
|
|||
|
|
#endif /*FFT_CRITICAL_MODE*/
|
|||
|
|
|
|||
|
|
static volatile u8 fft_init = 0;
|
|||
|
|
|
|||
|
|
typedef struct {
|
|||
|
|
unsigned int fft_config;
|
|||
|
|
const int *in;
|
|||
|
|
int *out;
|
|||
|
|
} pi32v2_hw_fft_ctx;
|
|||
|
|
|
|||
|
|
#define FFT_ISR_IE 0
|
|||
|
|
void hw_fft_wrap(pi32v2_hw_fft_ctx *ctx)
|
|||
|
|
{
|
|||
|
|
if (fft_init == 0) {
|
|||
|
|
fft_init = 1;
|
|||
|
|
FFT_CRITICAL_INIT();
|
|||
|
|
}
|
|||
|
|
FFT_ENTER_CRITICAL();
|
|||
|
|
|
|||
|
|
JL_FFT->CON = 0;
|
|||
|
|
JL_FFT->CON |= 1 << 8;
|
|||
|
|
JL_FFT->CADR = (unsigned int)ctx;
|
|||
|
|
JL_FFT->CON = (1 << 8) | (0 << 6) | (FFT_ISR_IE << 2) | (0 << 1) | (1 << 0); //((1<<8)|(1<<0))
|
|||
|
|
while ((JL_FFT->CON & (1 << 7)) == 0);
|
|||
|
|
JL_FFT->CON |= (1 << 6);
|
|||
|
|
|
|||
|
|
FFT_EXIT_CRITICAL();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
static int REAL_FFT_CONFIG[6] = {
|
|||
|
|
809501970, 813696546, 822085426, 838862898, 872417602, 939526722,
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
/**** Real IFFT *******/
|
|||
|
|
static int REAL_IFFT_CONFIG[6] = {
|
|||
|
|
809501958, 1082131974, 822085382, 1107298310, 872417542, 1207962118,
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
/****************************
|
|||
|
|
*
|
|||
|
|
* Complex Support points
|
|||
|
|
* 32\64\128\256\512\1024\2048
|
|||
|
|
*
|
|||
|
|
* *************************/
|
|||
|
|
/**** Complex FFT *******/
|
|||
|
|
static int COMPLEX_FFT_CONFIG[7] = {
|
|||
|
|
538969360, 541066784, 545261344, 553650224, 570427696, 603982400, 671091520,
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
/**** Complex IFFT *******/
|
|||
|
|
static int COMPLEX_IFFT_CONFIG[7] = {
|
|||
|
|
1075840276, 809502212, 1082132244, 822085636, 1107298580, 872417796, 1207962388,
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
/*********************************************************************
|
|||
|
|
* hw_fft_config
|
|||
|
|
* Description: 根据配置生成 FFT_config
|
|||
|
|
* Arguments : N 运算数据量；
|
|||
|
|
log2N 运算数据量的对数值
|
|||
|
|
is_same_addr 输入输出是否同一个地址，0:否，1:是
|
|||
|
|
is_ifft 运算类型 0:FFT运算, 1:IFFT运算
|
|||
|
|
is_real 运算数据的类型 1:实数, 0:复数
|
|||
|
|
* Return : ConfgPars 写入FFT寄存器
|
|||
|
|
* Note(s) : None.
|
|||
|
|
*********************************************************************/
|
|||
|
|
unsigned int hw_fft_config(int N, int log2N, int is_same_addr, int is_ifft, int is_real)
|
|||
|
|
{
|
|||
|
|
unsigned int ConfgPars;
|
|||
|
|
ConfgPars = 0;
|
|||
|
|
if (is_real == 1) {
|
|||
|
|
if (is_ifft == 0) {
|
|||
|
|
ConfgPars = REAL_FFT_CONFIG[log2N - 6];
|
|||
|
|
ConfgPars |= is_same_addr;
|
|||
|
|
} else {
|
|||
|
|
ConfgPars = REAL_IFFT_CONFIG[log2N - 6];
|
|||
|
|
ConfgPars |= is_same_addr;
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
if (is_ifft == 0) {
|
|||
|
|
ConfgPars = COMPLEX_FFT_CONFIG[log2N - 5];
|
|||
|
|
ConfgPars |= is_same_addr;
|
|||
|
|
} else {
|
|||
|
|
ConfgPars = COMPLEX_IFFT_CONFIG[log2N - 5];
|
|||
|
|
ConfgPars |= is_same_addr;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
/* printf("ConfgPars%x\n",ConfgPars); */
|
|||
|
|
return ConfgPars;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/*********************************************************************
|
|||
|
|
* hw_fft_run
|
|||
|
|
* Description: fft/ifft运算函数
|
|||
|
|
* Arguments :fft_config FFT运算配置寄存器值
|
|||
|
|
in 输入数据地址(满足4byte对其，以及非const类型)
|
|||
|
|
out 输出数据地址
|
|||
|
|
* Return : void
|
|||
|
|
* Note(s) : None.
|
|||
|
|
*********************************************************************/
|
|||
|
|
void hw_fft_run(unsigned int fft_config, const int *in, int *out)
|
|||
|
|
{
|
|||
|
|
pi32v2_hw_fft_ctx ctx;
|
|||
|
|
ctx.fft_config = fft_config;
|
|||
|
|
ctx.in = in;
|
|||
|
|
ctx.out = out;
|
|||
|
|
hw_fft_wrap(&ctx);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//////////////////////////////////////////////////////////////////////////
|
|||
|
|
// VECTOR
|
|||
|
|
//////////////////////////////////////////////////////////////////////////
|
|||
|
|
|
|||
|
|
typedef struct {
|
|||
|
|
unsigned long vector_con;
|
|||
|
|
unsigned long vector_xadr;
|
|||
|
|
unsigned long vector_yadr;
|
|||
|
|
unsigned long vector_zadr;
|
|||
|
|
unsigned long vector_config0;
|
|||
|
|
unsigned long vector_config1;
|
|||
|
|
unsigned long vector_config2;
|
|||
|
|
unsigned long null;
|
|||
|
|
} hwvec_ctx_t;
|
|||
|
|
|
|||
|
|
static hwvec_ctx_t g_vector_core_set __attribute__((aligned(16)));
|
|||
|
|
|
|||
|
|
//__attribute__ ((always_inline)) inline
|
|||
|
|
void hwvec_exec(
|
|||
|
|
void *xptr,
|
|||
|
|
void *yptr,
|
|||
|
|
void *zptr,
|
|||
|
|
short x_inc,
|
|||
|
|
short y_inc,
|
|||
|
|
short z_inc,
|
|||
|
|
short nlen,
|
|||
|
|
short nloop,
|
|||
|
|
char q,
|
|||
|
|
long config,
|
|||
|
|
long const_dat
|
|||
|
|
)
|
|||
|
|
{
|
|||
|
|
|
|||
|
|
if (fft_init == 0) {
|
|||
|
|
fft_init = 1;
|
|||
|
|
FFT_CRITICAL_INIT();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
FFT_ENTER_CRITICAL();
|
|||
|
|
|
|||
|
|
g_vector_core_set.vector_con = config;
|
|||
|
|
g_vector_core_set.vector_config0 = (q << 24) | (nloop << 12) | (nlen);
|
|||
|
|
g_vector_core_set.vector_config1 = (z_inc << 20) | (y_inc << 10) | x_inc;
|
|||
|
|
//printf("nlen0:%d,nloop:%d,q:%d,config:%d,%d\n",nlen,nloop,q,config,const_dat);
|
|||
|
|
|
|||
|
|
g_vector_core_set.vector_xadr = (unsigned long)xptr;
|
|||
|
|
g_vector_core_set.vector_yadr = (unsigned long)yptr;
|
|||
|
|
g_vector_core_set.vector_zadr = (unsigned long)zptr;
|
|||
|
|
|
|||
|
|
JL_FFT->CONST = const_dat;
|
|||
|
|
|
|||
|
|
JL_FFT->CADR = (unsigned long)(&g_vector_core_set);
|
|||
|
|
|
|||
|
|
|
|||
|
|
// nu clr vector ie en
|
|||
|
|
JL_FFT->CON = (1 << 8) | (1 << 6) | (1 << 3) | (1 << 2) | (1 << 0);
|
|||
|
|
|
|||
|
|
|
|||
|
|
lp_waiting((int *)&JL_FFT->CON, BIT(7), BIT(6), 116);
|
|||
|
|
|
|||
|
|
FFT_EXIT_CRITICAL();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
u8 fft_wrap_for_ldac_enable(void)
|
|||
|
|
{
|
|||
|
|
return 1;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int _FFT_wrap_(int cfg, int *in, int *out)
|
|||
|
|
{
|
|||
|
|
//init pi32v2 fft hardware
|
|||
|
|
pi32v2_hw_fft_ctx ctx;
|
|||
|
|
if (in == out) {
|
|||
|
|
cfg |= 1;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
ctx.fft_config = cfg;
|
|||
|
|
ctx.in = in;
|
|||
|
|
ctx.out = out;
|
|||
|
|
hw_fft_wrap(&ctx);
|
|||
|
|
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
|