crop.cuh
1.8 KB
template<typename T>
__global__ void cuda_crop2d(T* dest, T* src, size_t sx, size_t sy, size_t x, size_t y, size_t dx, size_t dy){
size_t xi = blockIdx.x * blockDim.x + threadIdx.x; //calculate the current working position within the destination image
size_t yi = blockIdx.y * blockDim.y + threadIdx.y;
if(xi >= dx || yi >= dy) return; //if this thread is outside of the destination image, return
size_t di = yi * dx + xi; //calculate the 1D index into the destination image
size_t si = (y + yi) * sx + (x + xi); //calculate the 1D index into the source image
dest[di] = src[si]; //copy the corresponding source pixel to the destination image
}
/// Crops a 2D image composed of elements of type T
/// @param dest is a device pointer to memory of size dx*dy that will store the cropped image
/// @param src is a device pointer to memory of size sx*sy that stores the original image
/// @param sx is the size of the source image along x
/// @param sy is the size of the source image along y
/// @param x is the x-coordinate of the start position of the cropped region within the source image
/// @param y is the y-coordinate of the start position of the cropped region within the source image
/// @param dx is the size of the destination image along x
/// @param dy is the size of the destination image along y
template<typename T>
void gpu_crop2d(T* dest, T* src, size_t sx, size_t sy, size_t x, size_t y, size_t dx, size_t dy){
int max_threads = stim::maxThreadsPerBlock(); //get the maximum number of threads per block for the CUDA device
dim3 threads( sqrt(max_threads), sqrt(max_threads) );
dim3 blocks( dx / sqrt(threads.x) + 1, dy / sqrt(threads.y) + 1); //calculate the optimal number of blocks
cuda_crop2d<T> <<< blocks, threads >>>(dest, src, sx, sy, x, y, dx, dy);
}