| # Builds Matlab (or Octave) interface. In case of Matlab caffe must be | |
| # compield as shared library. Octave can link static or shared caffe library | |
| # To install octave run: sudo apt-get install liboctave-dev | |
| if(NOT BUILD_matlab) | |
| return() | |
| endif() | |
| if(HAVE_MATLAB AND Octave_compiler) | |
| set(build_using ${Matlab_build_mex_using}) | |
| elseif(HAVE_MATLAB AND NOT Octave_compiler) | |
| set(build_using "Matlab") | |
| elseif(NOT HAVE_MATLAB AND Octave_compiler) | |
| set(build_using "Octave") | |
| else() | |
| return() | |
| endif() | |
| if(NOT BUILD_SHARED_LIBS AND build_using MATCHES Matlab) | |
| message(FATAL_ERROR "Matlab MEX interface (with default mex options file) can only be built if caffe is compiled as shared library. Please enable 'BUILD_SHARED_LIBS' in CMake. Aternativelly you can switch to Octave compiler.") | |
| endif() | |
| # helper function to set proper mex file extension | |
| function(caffe_fetch_and_set_proper_mexext mexfile_variable) | |
| execute_process(COMMAND ${Matlab_mexext} OUTPUT_STRIP_TRAILING_WHITESPACE RESULT_VARIABLE res OUTPUT_VARIABLE ext) | |
| if(res MATCHES 0) | |
| get_filename_component(folder ${${mexfile_variable}} PATH) | |
| get_filename_component(name_we ${${mexfile_variable}} NAME_WE) | |
| set(${mexfile_variable} ${folder}/${name_we}.${ext} PARENT_SCOPE) | |
| endif() | |
| endfunction() | |
| # global settings | |
| file(GLOB Matlab_srcs +caffe/private/caffe_.cpp) | |
| set(Matlab_caffe_mex ${PROJECT_SOURCE_DIR}/matlab/+caffe/private/caffe_.mex) | |
| caffe_get_current_cflags(cflags) | |
| caffe_parse_linker_libs(Caffe_LINKER_LIBS folders libflags macos_frameworks) | |
| set(folders $<TARGET_LINKER_FILE_DIR:caffe> ${folders}) | |
| # prepare linker flag lists | |
| string(REPLACE ";" ";-L" link_folders "-L${folders}") | |
| string(REPLACE ";" ":" rpath_folders "${folders}") | |
| if(build_using MATCHES "Matlab") | |
| set(libflags -lcaffe${Caffe_POSTFIX} ${libflags}) # Matlab R2014a complans for -Wl,--whole-archive | |
| caffe_fetch_and_set_proper_mexext(Matlab_caffe_mex) | |
| add_custom_command(OUTPUT ${Matlab_caffe_mex} COMMAND ${Matlab_mex} | |
| ARGS -output ${Matlab_caffe_mex} ${Matlab_srcs} ${cflags} ${link_folders} ${libflags} | |
| DEPENDS caffe COMMENT "Building Matlab interface: ${Matlab_caffe_mex}" VERBATIM) | |
| add_custom_target(matlab ALL DEPENDS ${Matlab_caffe_mex} SOURCES ${Matlab_srcs}) | |
| elseif(build_using MATCHES "Octave") | |
| if("${CMAKE_CXX_COMPILER_ID}" STREQUAL "Clang") | |
| set(libflags -Wl,-force_load,$<TARGET_LINKER_FILE:caffe> ${libflags}) | |
| elseif("${CMAKE_CXX_COMPILER_ID}" STREQUAL "GNU") | |
| set(libflags -Wl,--whole-archive -lcaffe${Caffe_POSTFIX} -Wl,--no-whole-archive ${libflags}) | |
| endif() | |
| add_custom_command(OUTPUT ${Matlab_caffe_mex} COMMAND ${Octave_compiler} | |
| ARGS --mex -o ${Matlab_caffe_mex} ${Matlab_srcs} ${cflags} ${link_folders} ${libflags} -Wl,-rpath,${rpath_folders} | |
| DEPENDS caffe COMMENT "Building Octave interface: ${Matlab_caffe_mex}" VERBATIM) | |
| add_custom_target(octave ALL DEPENDS ${Matlab_caffe_mex} SOURCES ${Matlab_srcs}) | |
| endif() | |
| # ---[ Install | |
| file(GLOB mfiles caffe/*.m) | |
| install(FILES ${mfiles} ${Matlab_caffe_mex} DESTINATION matlab) | |