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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
if (ITK_USE_GPU)
find_package ( OpenCL REQUIRED )
if ( NOT ${OPENCL_FOUND} )
message ( FATAL "Could not find OpenCL" )
endif ()
macro(sourcefile_to_string SOURCE_FILE RESULT_CMAKE_VAR)
file(STRINGS ${SOURCE_FILE} FileStrings)
foreach(SourceLine ${FileStrings})
# replace all \ with \\ to make the c string constant work
string(REGEX REPLACE "\\\\" "\\\\\\\\" TempSourceLine "${SourceLine}")
# replace all " with \" to make the c string constant work
string(REGEX REPLACE "\"" "\\\\\"" EscapedSourceLine "${TempSourceLine}")
set(${RESULT_CMAKE_VAR} "${${RESULT_CMAKE_VAR}}\n\"${EscapedSourceLine}\\n\"")
endforeach()
endmacro()
macro(write_gpu_kernel_to_file OPENCL_FILE GPUFILTER_NAME GPUFILTER_KERNELNAME
OUTPUT_FILE SRC_VAR)
sourcefile_to_string(${OPENCL_FILE} ${GPUFILTER_KERNELNAME}_SourceString)
set(${GPUFILTER_KERNELNAME}_KernelString
"#include \"itk${GPUFILTER_NAME}.h\"\n\n")
set(${GPUFILTER_KERNELNAME}_KernelString
"${${GPUFILTER_KERNELNAME}_KernelString}namespace itk\n")
set(${GPUFILTER_KERNELNAME}_KernelString
"${${GPUFILTER_KERNELNAME}_KernelString}{\n\n")
set(${GPUFILTER_KERNELNAME}_KernelString
"${${GPUFILTER_KERNELNAME}_KernelString}const char* ${GPUFILTER_KERNELNAME}::GetOpenCLSource()\n")
set(${GPUFILTER_KERNELNAME}_KernelString
"${${GPUFILTER_KERNELNAME}_KernelString}{\n")
set(${GPUFILTER_KERNELNAME}_KernelString
"${${GPUFILTER_KERNELNAME}_KernelString} return ${${GPUFILTER_KERNELNAME}_SourceString};\n")
set(${GPUFILTER_KERNELNAME}_KernelString
"${${GPUFILTER_KERNELNAME}_KernelString}}\n\n")
set(${GPUFILTER_KERNELNAME}_KernelString
"${${GPUFILTER_KERNELNAME}_KernelString}}\n")
file(WRITE ${CMAKE_CURRENT_BINARY_DIR}/${OUTPUT_FILE}
"${${GPUFILTER_KERNELNAME}_KernelString}")
configure_file(${OPENCL_FILE}
${CMAKE_CURRENT_BINARY_DIR}/${OUTPUT_FILE}.cl COPYONLY)
add_custom_target(${GPUFILTER_KERNELNAME}_Target
DEPENDS
${CMAKE_CURRENT_BINARY_DIR}/${OUTPUT_FILE}
${CMAKE_CURRENT_SOURCE_DIR}/CMakeLists.txt
${CMAKE_CURRENT_BINARY_DIR}/${OUTPUT_FILE}.cl
)
set_source_files_properties(${CMAKE_CURRENT_BINARY_DIR}/${OUTPUT_FILE}
PROPERTIES GENERATED ON)
set(${SRC_VAR} ${${SRC_VAR}} ${OUTPUT_FILE})
endmacro()
macro(write_gpu_kernels GPUKernels GPU_SRC)
foreach(GPUKernel ${GPUKernels})
get_filename_component(FilterName ${GPUKernel} NAME_WE)
write_gpu_kernel_to_file(${GPUKernel} ${FilterName} ${FilterName}Kernel "${FilterName}Kernel.cxx" ${GPU_SRC})
endforeach()
endmacro()
endif()
|
