cufftComplexに画素値を代入してみる

概要

• OpenCV C++ で読み込んだ画像を、cufftで使用するcufftComplexに渡す

関連記事

• cuFFTパラメータ編

コード

#include <opencv2/core/core.hpp>
#include <opencv2/videoio.hpp>
#include <opencv2/highgui.hpp>
#include <opencv2/core/core.hpp>
#include <opencv/cv.hpp>
#include <iostream>

#include "cufft.h"

#include <string>
#include <vector>
#include <sys/stat.h>
#include "helper_cuda.h" //省いてもok　省く場合はcheckCudaErrorsは使えない

using namespace std;

int main()
{
    //画像をグレースケールで読み込み（作業がしやすいので）
    cv::Mat img = cv::imread("test.jpg", 0);

    if(img.empty()==true){
      cout << "Error opening file" << "("<< __LINE__<< ")" << endl;
      return 0;
    }

    int width, height;

    width  = img.cols;  // 横幅を取得
    height = img.rows;  // 縦幅を取得

    // 入力
    cufftComplex *d_i_img;
    checkCudaErrors(cudaMalloc((void **)&d_i_img, sizeof(cufftComplex)*width*height));

    // 画素値代入用変数
    cufftComplex *h_complex;
    checkCudaErrors(cudaMallocHost((void **)&h_complex, sizeof(cufftComplex)*height*width));

    cout << "width : "<< width << " height : " << height << endl;

    // 画素値代入
    for( int icnt =0;  icnt<height*width; ++icnt){ 
        h_complex[icnt].x = (float)img.data[icnt];
        h_complex[icnt].y = 0.0f;
    }

    checkCudaErrors(cudaMemcpy(d_i_img, h_complex, sizeof(cufftComplex)*width*height, cudaMemcpyHostToDevice));

    // すべてのcopyが終わるまで待機
    if (cudaDeviceSynchronize() != cudaSuccess){
      cout << "Cuda error: Failed to synchronize\n";
      return 0;
    }

    cudaFree(d_i_img);
    cudaFree(h_complex);
    cout << "ok";

    return 0;
}

解説

コードの方を少し解説しますと

GPUに入力する画像の大きさ分のメモリをcudaMallocで確保

// 入力
cufftComplex *d_i_img;
checkCudaErrors(cudaMalloc((void **)&d_i_img, sizeof(cufftComplex)*width*height));

画素値を代入する用のcufftComplexをcudaMallocHostでホスト側にメモリを確保

• cudaMallocHostについて

// 画素値代入用変数
cufftComplex *h_complex;
cudaMallocHost((void **)&h_complex, height * width * sizeof(cufftComplex));

その後画素値を代入

// 画素値代入
for( int icnt =0;  icnt<height*width; ++icnt){ 
    h_complex[icnt].x = (float)img.data[icnt];
    h_complex[icnt].y = 0.0f;
}

これで画像を使ってcufftの入力に加えたりすることができます。