biotracker_sampletracker/cmake/Locate.cmake at master · BioroboticsLab/biotracker_sampletracker · GitHub

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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81

function(locate_shared_library VAR)
    foreach(name ${ARGN})
        set(_filename "${CMAKE_SHARED_LIBRARY_PREFIX}${name}${CMAKE_SHARED_LIBRARY_SUFFIX}")
        message(STATUS "Locating ${_filename}")
        if("${CMAKE_SYSTEM_NAME}" STREQUAL "Windows")
            execute_process(
                COMMAND where.exe "${_filename}"
                OUTPUT_VARIABLE _matches
                OUTPUT_STRIP_TRAILING_WHITESPACE
                RESULT_VARIABLE _res
            )
            if(NOT "${_res}" STREQUAL "0")
                message(FATAL_ERROR "Could not locate ${_filename}")
            endif()
            string(REGEX REPLACE "\n" ";" _matches "${_matches}")
            list(GET _matches 0 _match)
            string(REGEX REPLACE "\\\\" "/" _match "${_match}")

            if(DEFINED ${VAR})
                list(APPEND ${VAR} "${_match}")
                set(${VAR} ${${VAR}} PARENT_SCOPE)
            else()
                set(${VAR} "${_match}" PARENT_SCOPE)
            endif()
        else()
            message(FATAL_ERROR "Platform not supported")
        endif()
    endforeach()
endfunction()

function(locate_opencv VAR)
    cmake_parse_arguments(ARG "" "" "COMPONENTS" ${ARGN})

    if("${CMAKE_SYSTEM_NAME}" STREQUAL "Windows")
        set(suffix "${OpenCV_VERSION_MAJOR}${OpenCV_VERSION_MINOR}${OpenCV_VERSION_PATCH}")

        foreach(component ${ARG_COMPONENTS})
            locate_shared_library(${VAR} "opencv_${component}${suffix}")
        endforeach()
        set(${VAR} ${${VAR}} PARENT_SCOPE)
    else()
        message(FATAL_ERROR "Platform not supported")
    endif()
endfunction()

function(locate_cuda VAR)
    cmake_parse_arguments(ARG "" "" "COMPONENTS" ${ARGN})

    if("${CMAKE_SYSTEM_NAME}" STREQUAL "Windows")
        if(${CMAKE_SIZEOF_VOID_P} EQUAL 8)
            set(suffix "64_${CUDA_VERSION_MAJOR}${CUDA_VERSION_MINOR}")
        else()
            set(suffix "32_${CUDA_VERSION_MAJOR}${CUDA_VERSION_MINOR}")
        endif()

        foreach(component ${ARG_COMPONENTS})
            locate_shared_library(${VAR} "${component}${suffix}")
        endforeach()
        set(${VAR} ${${VAR}} PARENT_SCOPE)
    else()
        message(FATAL_ERROR "Platform not supported")
    endif()
endfunction()

function(locate_cudnn VAR)
    cmake_parse_arguments(ARG "" "" "COMPONENTS" ${ARGN})

    if("${CMAKE_SYSTEM_NAME}" STREQUAL "Windows")
        if(${CMAKE_SIZEOF_VOID_P} EQUAL 8)
            set(suffix "64_${version}")
        else()
            set(suffix "32_${version}")
        endif()

        locate_shared_library(${VAR} "cudnn${suffix}")
        set(${VAR} ${${VAR}} PARENT_SCOPE)
    else()
        message(FATAL_ERROR "Platform not supported")
    endif()
endfunction()