From 0759d5ed2bfa1ecfc8f00ab955d8618db474f280 Mon Sep 17 00:00:00 2001
From: David Hirvonen <dhirvonen@elucideye.com>
Date: Sun, 7 Jan 2018 17:49:39 -0500
Subject: [PATCH] add cmake w/ relocatable pkgconfig installation (#53)

---
 .gitignore            |   2 +-
 CMakeLists.txt        | 120 ++++++++++++++++++++++++++++++++++++++++++
 cmake/Config.cmake.in |   4 ++
 3 files changed, 125 insertions(+), 1 deletion(-)
 create mode 100644 CMakeLists.txt
 create mode 100644 cmake/Config.cmake.in

diff --git a/.gitignore b/.gitignore
index 121caaafe..40a245d24 100644
--- a/.gitignore
+++ b/.gitignore
@@ -30,10 +30,10 @@
 *.pyc
 *.mpi
 *.exe
-*.txt
 *tmp*
 *.rabit
 *.mock
 dmlc-core
 recommonmark
 recom
+_*
diff --git a/CMakeLists.txt b/CMakeLists.txt
new file mode 100644
index 000000000..7676fe194
--- /dev/null
+++ b/CMakeLists.txt
@@ -0,0 +1,120 @@
+cmake_minimum_required(VERSION 3.0)
+
+project(rabit VERSION 0.0.0)
+
+option(RABIT_BUILD_TESTS "Build rabit tests" OFF)
+option(RABIT_BUILD_MPI "Build MPI" OFF)
+
+add_library(rabit src/allreduce_base.cc src/allreduce_robust.cc src/engine.cc src/c_api.cc)
+add_library(rabit_base src/allreduce_base.cc src/engine_base.cc src/c_api.cc)
+add_library(rabit_empty src/engine_empty.cc src/c_api.cc)
+
+set(rabit_libs rabit rabit_base rabit_empty)
+if(RABIT_BUILD_MPI)
+  add_library(rabit_mpi src/engine_mpi.cc)
+  list(APPEND rabit_libs rabit_mpi)
+endif()
+
+if(RABIT_BUILD_TESTS)
+  add_library(rabit_mock src/allreduce_base.cc src/allreduce_robust.cc src/engine_mock.cc src/c_api.cc)
+  list(APPEND rabit_libs rabit_mock) # add to list to apply build settings, then remove
+endif()
+
+foreach(lib ${rabit_libs})
+  #include "./internal/utils.h"  
+  target_include_directories(${lib} PUBLIC
+    "$<BUILD_INTERFACE:${CMAKE_CURRENT_LIST_DIR}/include>"    
+    "$<BUILD_INTERFACE:${CMAKE_CURRENT_LIST_DIR}/include/rabit>"
+    ) 
+endforeach()
+
+if(RABIT_BUILD_TESTS)
+  list(REMOVE_ITEM rabit_libs "rabit_mock") # remove here to avoid installing it
+  set(tests speed_test lazy_recover local_recover model_recover)
+  foreach(test ${tests})
+    add_executable(${test} test/${test}.cc)
+    target_link_libraries(${test} rabit_mock)
+    install(TARGETS ${test} DESTINATION bin)
+  endforeach()
+  if(RABIT_BUILD_MPI)
+    add_executable(speed_test_mpi test/speed_test.cc)
+    target_link_libraries(speed_test_mpi rabit_mpi)
+    install(TARGETS speed_test_mpi DESTINATION bin)    
+  endif()
+endif()
+
+# Installation (https://github.com/forexample/package-example) {
+
+# Layout. This works for all platforms:
+#   * <prefix>/lib/cmake/<PROJECT-NAME>
+#   * <prefix>/lib/
+#   * <prefix>/include/
+set(config_install_dir "lib/cmake/${PROJECT_NAME}")
+set(include_install_dir "include")
+
+set(generated_dir "${CMAKE_CURRENT_BINARY_DIR}/generated")
+
+# Configuration
+set(version_config "${generated_dir}/${PROJECT_NAME}ConfigVersion.cmake")
+set(project_config "${generated_dir}/${PROJECT_NAME}Config.cmake")
+set(TARGETS_EXPORT_NAME "${PROJECT_NAME}Targets")
+set(namespace "${PROJECT_NAME}::")
+
+# Include module with fuction 'write_basic_package_version_file'
+include(CMakePackageConfigHelpers)
+
+# Configure '<PROJECT-NAME>ConfigVersion.cmake'
+# Use:
+#   * PROJECT_VERSION
+write_basic_package_version_file(
+    "${version_config}" COMPATIBILITY SameMajorVersion
+)
+
+# Configure '<PROJECT-NAME>Config.cmake'
+# Use variables:
+#   * TARGETS_EXPORT_NAME
+#   * PROJECT_NAME
+configure_package_config_file(
+    "cmake/Config.cmake.in"
+    "${project_config}"
+    INSTALL_DESTINATION "${config_install_dir}"
+)
+
+# Targets:
+#   * <prefix>/lib/librabit.a
+#   * <prefix>/lib/librabit_base
+#   * <prefix>/lib/librabit_empty
+#   * header location after install: <prefix>/include/rabit/rabit.h
+#   * headers can be included by C++ code `#include <rabit/rabit.h>`
+install(
+    TARGETS ${rabit_libs}
+    EXPORT "${TARGETS_EXPORT_NAME}"
+    LIBRARY DESTINATION "lib"
+    ARCHIVE DESTINATION "lib"
+    RUNTIME DESTINATION "bin"
+    INCLUDES DESTINATION "${include_install_dir}"
+)
+
+# Headers:
+install(
+    DIRECTORY "include/"
+    DESTINATION "${include_install_dir}"
+    FILES_MATCHING PATTERN "*.h"
+)
+
+# Config
+#   * <prefix>/lib/cmake/rabit/rabitConfig.cmake
+#   * <prefix>/lib/cmake/rabit/rabitConfigVersion.cmake
+install(
+    FILES "${project_config}" "${version_config}"
+    DESTINATION "${config_install_dir}"
+)
+
+# Config
+#   * <prefix>/lib/cmake/Foo/FooTargets.cmake
+install(
+    EXPORT "${TARGETS_EXPORT_NAME}"
+    NAMESPACE "${namespace}"
+    DESTINATION "${config_install_dir}"
+)
+# }
diff --git a/cmake/Config.cmake.in b/cmake/Config.cmake.in
new file mode 100644
index 000000000..38bbde7b3
--- /dev/null
+++ b/cmake/Config.cmake.in
@@ -0,0 +1,4 @@
+@PACKAGE_INIT@
+
+include("${CMAKE_CURRENT_LIST_DIR}/@TARGETS_EXPORT_NAME@.cmake")
+check_required_components("@PROJECT_NAME@")