如何将图像坐标展平为一维数组?

时间:2013-07-30 18:01:23

标签: graphics opencl

我的源代码来自使用OpenCL进行异构计算第4章基本OpenCL示例>图像旋转。这本书遗漏了几个关键细节。

我的主要问题是我不知道如何初始化我提供给他们内核的数组(他们没有告诉你如何)。我所拥有的是:

int W = inImage.width();
int H = inImage.height();
float *myImage = new float[W*H];
    for(int row = 0; row < H; row++)
        for(int col = 0; col < W; col++)
            myImage[row*W+col] = col;

我提供给这个内核:

    __kernel void img_rotate(__global float* dest_data, __global float* src_data, int W, int H, float sinTheta, float cosTheta) 
{
    const int ix = get_global_id(0);
    const int iy = get_global_id(1);
    float x0 = W/2.0f;
    float y0 = H/2.0f;
    float xoff = ix-x0;
    float yoff = iy-y0;
    int xpos = (int)(xoff*cosTheta + yoff*sinTheta + x0);
    int ypos = (int)(yoff*cosTheta - xoff*sinTheta + y0);
    if(((int)xpos>=0) && ((int)xpos < W) && ((int)ypos>=0) && ((int)ypos<H))
    {
        dest_data[iy*W+ix] = src_data[ypos*W+xpos];
        //dest_data[iy*W+ix] = src_data[iy*W+ix];
    }
}

我也很难为theta找到合适的价值。一个整数对于theta来说是一个合适的值,对吗?

float theta = 45; // 45 degrees, right?
float cos_theta = cos(theta);
float sin_theta = sin(theta);

1 个答案:

答案 0 :(得分:2)

在编写OpenCL代码时,无论数据是1D,2D还是3D,我总是将每个内核视为读取一组3D数据:

 __kernel void TestKernel(__global float *Data){
      k = get_global_id(0); //also z
      j = get_global_id(1); //also y
      i = get_global_id(2); //also x

      //Convert 3D to 1D
      int linear_coord = i + get_global_size(0)*j + get_global_size(0)*get_global_size(1)*k;

      //do stuff
 }

在执行clEnqueueNDKernelRange(...)时,只需将维度设置为:

 int X = 500;
 int Y = 300;
 int Z = 1;

 size_t GlobalDim = {Z, Y, X};

这让我的所有内核都可以在各个方面轻松运行。