Inverse Fast Fourier Transform 2D

二维快速傅里叶逆变换(Inverse Fast Fourier Transform 2D,以下简称IFFT2D)是基于快速傅里叶逆变换(以下简称IFFT)开发的快速算法, 实现了二维数据从频域到时域的快速转换,通常被用来将图像等二维频域数据复原为时域数据。

算子效果

时域输入数据参数频域输出数据
imagep_size = HB_HPL_FFT16
normalize = 0
dataType = HB_HPL_DATA_TYPE_I16
imFormat = HB_IM_FORMAT_SEPARATE
numDimensionSize = 2
image

原理

IFFT2D是基于IFFT算法开发的扩展方法,对二维数据分别从y方向和x方向执行IFFT操作,实现频域数据到时域数据的映射。具体流程如下:

  1. 对二维数据中每一列进行IFFT计算,其中ny为接口参数中设置的y轴FFT点数。
HPL5IFFT2D_y
  1. 在步骤1的基础上,对二维数据中每一行进行IFFT计算,其中nx为接口参数中设置的x轴FFT点数。
HPL5IFFT2D_x
  1. 返回步骤2的计算结果。

API接口

int32_t hbIFFT2D(hbUCPTaskHandle_t *taskHandle, hbHPLImaginaryData *dst,
                 hbHPLImaginaryData const *src, hbIFFT2DParam const *param);

详细接口信息请查看 hbIFFT2D

使用方法

// Include the header
#include "hobot/hb_ucp.h"
#include "hobot/hpl/hb_hpl.h"
#include "hobot/hpl/hb_ifft_2d.h"

// init, allocate memory for data
src_length = 1024 * 1024 * 5;
hbUCPMallocCached(&src_re_mem, src_length, 0);
hbUCPMallocCached(&src_im_mem, src_length, 0);
hbHPLImaginaryData src;
src.realDataVirAddr = src_re_mem.virAddr;
src.realDataPhyAddr = src_re_mem.phyAddr;
src.imDataVirAddr = src_im_mem.virAddr;
src.imDataPhyAddr = src_im_mem.phyAddr;
src.numDimensionSize = 2;
src.dataType = HB_HPL_DATA_TYPE_I16;
src.imFormat = HB_IM_FORMAT_SEPARATE;
src.dimensionSize[0] = 16 * 11;
src.dimensionSize[1] = 32 * 2;

hbUCPMallocCached(&dst_re_mem, src_length, 0);
hbUCPMallocCached(&dst_im_mem, src_length, 0);
hbHPLImaginaryData dst;
dst.realDataVirAddr = dst_re_mem.virAddr;
dst.realDataPhyAddr = dst_re_mem.phyAddr;
dst.imDataVirAddr = dst_im_mem.virAddr;
dst.imDataPhyAddr = dst_im_mem.phyAddr;
dst.numDimensionSize = 2;
dst.dataType = HB_HPL_DATA_TYPE_I16;
dst.imFormat = HB_IM_FORMAT_SEPARATE;
dst.dimensionSize[0] = 16 * 11;
dst.dimensionSize[1] = 32 * 2;

// init param
hbIFFT2DParam param;
param.nx = HB_FFT_POINT_SIZE_16;
param.ny = HB_FFT_POINT_SIZE_32;
param.normalize = 0;

// init task handle and schedule param
hbUCPTaskHandle_t task_handle{nullptr};
hbUCPSchedParam sched_param{...};

// create task
hbIFFT2D(&task_handle, &dst, &src, &param);

// submit task
hbUCPSubmitTask(task_handle, &sched_param);

// wait for task done 
hbUCPWaitTaskDone(task_handle, 0);

// release task handle
hbUCPReleaseTask(task_handle);

// release memory
hbUCPFree(&src_re_mem);
hbUCPFree(&src_im_mem);
hbUCPFree(&dst_re_mem);
hbUCPFree(&dst_im_mem);