dependencies.cmake - LightGBM dependency was added

Author: hudlijac

cmake/dependencies.cmake

@@ -1,6 +1,7 @@
 # Project dependencies
 find_package(Armadillo REQUIRED)
 find_package(Boost REQUIRED COMPONENTS regex serialization)
+find_package(LIGHTGBM REQUIRED)
 find_package(OpenMP REQUIRED)
 find_package(Python3 REQUIRED COMPONENTS Development NumPy)
 

cmake/modules/FindLIGHTGBM.cmake

@@ -0,0 +1,40 @@
+# Find the LightGBM includes and library
+#
+# This module defines the following IMPORTED targets:
+#
+#  LightGBM::lightgbm   - The LightGBM library, if found.
+#
+# This module will set the following variables in your project:
+#
+#  LIGHTGBM_INCLUDE_DIRS - where to find <LightGBM/c_api.h>, etc.
+#  LIGHTGBM_FOUND        - True if the LightGBM library has been found.
+
+find_package(PkgConfig QUIET)
+if(PKG_CONFIG_FOUND)
+    pkg_check_modules(PC_LIGHTGBM QUIET lightgbm)
+endif()
+
+find_path(
+    LIGHTGBM_INCLUDE_DIR lightgbm
+    NAMES LightGBM/c_api.h
+    HINTS ${PC_LIGHTGBM_INCLUDEDIR} ${PC_LIGHTGBM_INCLUDE_DIRS}
+    PATH_SUFFIXES include
+)
+
+if(PC_LIGHTGBM_VERSION)
+    # Version extracted from pkg-config
+    set(LIGHTGBM_VERSION_STRING ${PC_LIGHTGBM_VERSION})
+endif()
+
+# Handle find_package() arguments (i.e. QUIETLY and REQUIRED) and set
+# LIGHTGBM_FOUND to TRUE if all listed variables are filled.
+include(FindPackageHandleStandardArgs)
+find_package_handle_standard_args(
+    LIGHTGBM
+    REQUIRED_VARS LIGHTGBM_INCLUDE_DIR
+    VERSION_VAR LIGHTGBM_VERSION_STRING
+)
+
+set(LIGHTGBM_INCLUDE_DIRS ${LIGHTGBM_INCLUDE_DIR})
+mark_as_advanced(LIGHTGBM_INCLUDE_DIR)
+

src/wif/CMakeLists.txt

@@ -1,6 +1,7 @@
 set(LIBWIF_SOURCES
 	classifiers/classifier.cpp
 	classifiers/ipPrefixClassifier.cpp
+	classifiers/lightGBMClassifier.cpp
 	classifiers/mlpackClassifier.cpp
 	classifiers/regexClassifier.cpp
 	classifiers/scikitMlClassifier.cpp
@@ -9,6 +10,7 @@ set(LIBWIF_SOURCES
 	combinators/majorityCombinator.cpp
 	combinators/sumCombinator.cpp
 	filesystem/fileModificationChecker.cpp
+	ml/lightGBMWrapper.cpp
 	ml/mlpackModels/decisionTreeModel.cpp
 	ml/mlpackModels/hoeffdingTreeModel.cpp
 	ml/mlpackModels/linearSVMModel.cpp
@@ -45,13 +47,6 @@ if(BUILD_WITH_UNIREC)
 	)
 endif()
 
-if(BUILD_WITH_LIGHTGBM)
-	list(APPEND LIBWIF_SOURCES
-		classifiers/lightGBMClassifier.cpp
-		ml/lightGBMWrapper.cpp
-	)
-endif()
-
 add_library(wif SHARED ${LIBWIF_SOURCES})
 target_link_libraries(wif PRIVATE ${LIBWIF_LIBS})
 target_include_directories(wif PUBLIC ${CMAKE_SOURCE_DIR}/include)