detect file extension of mex function

18711b55 · Joel Nelson · 30ba39c9 · 18711b55
Commit 18711b55 authored Jun 29, 2018 by Joel Nelson
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 1 deletion

dlib/matlab/cmake_mex_wrapper dlib/matlab/cmake_mex_wrapper +8 -1

No files found.
--- a/dlib/matlab/cmake_mex_wrapper
+++ b/dlib/matlab/cmake_mex_wrapper
@@ -70,7 +70,14 @@ INCLUDE(InstallRequiredSystemLibraries)


 MACRO(add_mex_function name )
-    ADD_LIBRARY(${name} MODULE ${name}.cpp )
+    if (EXISTS "${CMAKE_CURRENT_SOURCE_DIR}/${name}.C")
+        ADD_LIBRARY(${name} MODULE ${name}.C)
+    elseif(EXISTS "${CMAKE_CURRENT_SOURCE_DIR}/${name}.c")
+        ADD_LIBRARY(${name} MODULE ${name}.c)
+    else()
+        ADD_LIBRARY(${name} MODULE ${name}.cpp )
+    endif()
+
    target_compile_definitions(${name} PRIVATE -DMEX_FILENAME=${name})
    if (UNIX)
        # Doing this prevents our mex function from exporting any symbols