#include <cstdio>
#include <cufft.h>
#include <cmath>
#include <cuda_runtime.h>
#include "BaseConstVariable.h"

#define cudaCheckError(ans) { gpuAssert((ans), __FILE__, __LINE__); }
inline void gpuAssert(cudaError_t code, const char* file, int line) {
    if (code != cudaSuccess) {
        fprintf(stderr, "CUDA Error: %s %s %d\n", cudaGetErrorString(code), file, line);
        exit(code);
    }
}

#define c LIGHTSPEED
 

struct GPUDATA {
    int Nfft, K, Np, nx, ny; // 傅里叶点数、频点数、脉冲数、图像列、图像行
    float* AntX, * AntY, * AntZ, * minF; // 天线坐标、起始频率
    float* x_mat, * y_mat, * z_mat;// 地面坐标
    float* r_vec; // 坐标范围
    cufftComplex* phdata;// 回波
    cufftComplex* im_final;// 图像
    float R0; // 参考斜距
};

extern "C" {
    void initGPUData(GPUDATA& h_data, GPUDATA& d_data);
    void freeGPUData(GPUDATA& d_data);
};