您的位置:首页 > 理论基础 > 数据结构算法

Cuda图像数据结构

2012-11-07 22:22 78 查看
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显示在屏幕上
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: