-
Notifications
You must be signed in to change notification settings - Fork 0
/
imageDenoising_copy_kernel.cuh
41 lines (35 loc) · 1.2 KB
/
imageDenoising_copy_kernel.cuh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
/*
* Copyright 1993-2015 NVIDIA Corporation. All rights reserved.
*
* Please refer to the NVIDIA end user license agreement (EULA) associated
* with this source code for terms and conditions that govern your use of
* this software. Any use, reproduction, disclosure, or distribution of
* this software and related documentation outside the terms of the EULA
* is strictly prohibited.
*
*/
__global__ void Copy(
TColor *dst,
int imageW,
int imageH,
cudaTextureObject_t texImage
)
{
const int ix = blockDim.x * blockIdx.x + threadIdx.x;
const int iy = blockDim.y * blockIdx.y + threadIdx.y;
//Add half of a texel to always address exact texel centers
const float x = (float)ix + 0.5f;
const float y = (float)iy + 0.5f;
if (ix < imageW && iy < imageH)
{
float4 fresult = tex2D<float4>(texImage, x, y);
dst[imageW * iy + ix] = make_color(fresult.x, fresult.y, fresult.z, 0);
}
}
extern "C" void
cuda_Copy(TColor *d_dst, int imageW, int imageH, cudaTextureObject_t texImage)
{
dim3 threads(BLOCKDIM_X, BLOCKDIM_Y);
dim3 grid(iDivUp(imageW, BLOCKDIM_X), iDivUp(imageH, BLOCKDIM_Y));
Copy<<<grid, threads>>>(d_dst, imageW, imageH, texImage);
}