Cuda图像数据结构

CudaImage.h

struct CudaImage

{

int width;

int height;

float *imData;

float *cuData;

size_t pitch;

size_t nBytes;

void setPixel(int x, int y, float val)

设置(x,y)处的像素值为val

float getPixel(int x, int y)

取得（x,y)处坐标的像素值

float getPixelBI(float col, float row)

也是取得（x,y)处坐标的像素值，不过此时x,y都是浮点数，这个像素值是由它周围4个整数点的像素值通过距离上的权值得到的。

void InitCuData(int _width, int _height)

初始化显存

InitCuData函数用到了一个  

cudaMallocPitch()，

说明  向设备分配至少widthInBytes*height字节的线性存储器，并以*devPtr的形式返回指向所分配存储器的指针。该函数可以填充所分配的存储器，以确保在地址从一行更新到另一行时，给定行的对应指针依然满足对齐要求。 cudaMallocPitch((void**)(&cuData),&pitch, sizeof(float) * width,height)： *devPtr的形式返回指向所分配存储器的指针。 以*pitch的形式返回间距，即所分配存储器的宽度，以字节为单位。 分配至少widthInBytes*height字节的线性存储器。

void InitImData(int _width, int _height)

imData是一个 大小width*height的浮点数组，用来在主机上存储图像数据的结构

void CopyToDevice()

把数据从cpu上的imData复制到gpu上，返回指针cuData。使用了cudaMemcpy2D这个api,用于分配内存。cudaMemcpy2D(cuData,pitch, imData, sizeof(float) * width, sizeof(float) * width,height, cudaMemcpyHostToDevice));

void CopyToHost()

把数据从gpu上cuData指针复制到主机上imData

int LoadImage(const char *filename)

加载图片文件， 数据存入imData。首先把图片用opencv的函数cvLoadImage加载到opencv的结构，再从opencv的结构中把数据转成我们的CudaImage。

void toIplImage(IplImage *&img)

把imData 数据转成opencv格式IplImage

void write(const char * filename)

把imData的数据写入文件中，比如写成.jpg的文件

void show(const char * windowname)

把imData显示在屏幕上

}