patchkernel: implement automatic mesh partitioning using METIS

Author: lucask81

.travis.yml

@@ -20,6 +20,7 @@ addons:
       - libopenmpi-dev
       - libpetsc3.6
       - libpetsc3.6.2-dev
+      - metis
 
 env:
   global:

CMakeLists.txt

@@ -595,7 +595,11 @@ set(LA_EXTERNAL_DEPS "PETSc")
 set(SA_EXTERNAL_DEPS "")
 set(CG_EXTERNAL_DEPS "")
 set(PABLO_EXTERNAL_DEPS "")
-set(PATCHKERNEL_EXTERNAL_DEPS "")
+if (BITPIT_ENABLE_MPI)
+    set(PATCHKERNEL_EXTERNAL_DEPS "METIS")
+else()
+    set(PATCHKERNEL_EXTERNAL_DEPS "")
+endif()
 set(SURFUNSTRUCTURED_EXTERNAL_DEPS "")
 set(VOLCARTESIAN_EXTERNAL_DEPS "")
 set(VOLOCTREE_EXTERNAL_DEPS "")
@@ -730,6 +734,7 @@ set(BLAS_LANGUAGES "Fortran")
 set(CBLAS_LANGUAGES "C")
 set(LAPACK_LANGUAGES "Fortran")
 set(LAPACKE_LANGUAGES "C")
+set(METIS_LANGUAGES "C")
 set(PETSC_LANGUAGES "Fortran;C")
 
 foreach(DEP_NAME IN LISTS EXTERNAL_DEPS)
@@ -901,6 +906,24 @@ if (${_BoostTest_index} GREATER -1)
 endif()
 unset(_BoostTest_index)
 
+list(FIND EXTERNAL_DEPS "METIS" _METIS_index)
+if (${_METIS_index} GREATER -1)
+    list(APPEND CMAKE_MODULE_PATH "${CMAKE_SOURCE_DIR}/external/METIS/cmake")
+
+    find_package(METIS)
+
+    if (METIS_FOUND)
+        list (APPEND BITPIT_EXTERNAL_DEPENDENCIES "METIS")
+        list (APPEND BITPIT_EXTERNAL_VARIABLES_LIBRARIES "METIS_LIBRARIES")
+        list (APPEND BITPIT_EXTERNAL_VARIABLES_INCLUDE_DIRS "METIS_INCLUDE_DIRS")
+        addPrivateDefinitions("BITPIT_ENABLE_METIS=1")
+    else()
+        addPrivateDefinitions("BITPIT_ENABLE_METIS=0")
+        message(STATUS "METIS library not found, default patch partitioning support will be disabled.")
+    endif()
+endif()
+unset(_METIS_index)
+
 set(BITPIT_EXTERNAL_LIBRARIES "")
 foreach (VARIABLE_NAME IN LISTS BITPIT_EXTERNAL_VARIABLES_LIBRARIES)
     list (APPEND BITPIT_EXTERNAL_LIBRARIES "${${VARIABLE_NAME}}")

external/CMakeLists.txt

@@ -9,3 +9,6 @@ add_subdirectory(LAPACKE)
 
 # PETSc
 add_subdirectory(PETSc)
+
+# METIS
+add_subdirectory(METIS)

external/METIS/CMakeLists.txt

@@ -0,0 +1,9 @@
+# Specify the version being used as well as the language
+cmake_minimum_required(VERSION 3.10)
+
+#------------------------------------------------------------------------------------#
+# CMake targets
+#------------------------------------------------------------------------------------#
+add_custom_target(clean-external-metis COMMAND ${CMAKE_MAKE_PROGRAM} clean WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR})
+
+install(FILES "cmake/FindMETIS.cmake" DESTINATION "${CMAKE_INSTALL_PREFIX}/${BITPIT_INSTALL_CMAKE_DIR}")