From ccfda9db0171ea1113ba0772249806c5f62e926f Mon Sep 17 00:00:00 2001
From: Hoildkv <42310255+xq114@users.noreply.github.com>
Date: Sun, 16 May 2021 13:27:34 +0800
Subject: [PATCH] add flann (#415)

* fix lz4

* add flann

* fix flann
---
 packages/f/flann/patches/1.9.1/cuda10.patch | 12 ++++
 packages/f/flann/xmake.lua                  | 65 +++++++++++++++++++++
 packages/l/lz4/xmake.lua                    |  8 ++-
 3 files changed, 84 insertions(+), 1 deletion(-)
 create mode 100644 packages/f/flann/patches/1.9.1/cuda10.patch
 create mode 100644 packages/f/flann/xmake.lua

diff --git a/packages/f/flann/patches/1.9.1/cuda10.patch b/packages/f/flann/patches/1.9.1/cuda10.patch
new file mode 100644
index 000000000..479312a01
--- /dev/null
+++ b/packages/f/flann/patches/1.9.1/cuda10.patch
@@ -0,0 +1,12 @@
+diff --git a/src/cpp/flann/algorithms/kdtree_cuda_3d_index.cu b/src/cpp/flann/algorithms/kdtree_cuda_3d_index.cu
+index e9b05080..8465322e 100644
+--- a/src/cpp/flann/algorithms/kdtree_cuda_3d_index.cu
++++ b/src/cpp/flann/algorithms/kdtree_cuda_3d_index.cu
+@@ -32,6 +32,7 @@
+ #include <flann/util/cuda/result_set.h>
+ // #define THRUST_DEBUG 1
+ #include <cuda.h>
++#include <thrust/gather.h>
+ #include <thrust/copy.h>
+ #include <thrust/device_vector.h>
+ #include <vector_types.h>
diff --git a/packages/f/flann/xmake.lua b/packages/f/flann/xmake.lua
new file mode 100644
index 000000000..737a61ed5
--- /dev/null
+++ b/packages/f/flann/xmake.lua
@@ -0,0 +1,65 @@
+package("flann")
+
+    set_homepage("https://github.com/flann-lib/flann/")
+    set_description("Fast Library for Approximate Nearest Neighbors")
+    set_license("BSD-3-Clause")
+
+    add_urls("https://github.com/flann-lib/flann/archive/refs/tags/$(version).tar.gz",
+             "https://github.com/flann-lib/flann.git")
+    add_versions("1.9.1", "b23b5f4e71139faa3bcb39e6bbcc76967fbaf308c4ee9d4f5bfbeceaa76cc5d3")
+
+    add_patches("1.9.1", path.join(os.scriptdir(), "patches", "1.9.1", "cuda10.patch"), "087492b422362f42c83b320296d9af7b22023cf20d81ea7bd99efabd1535f7d6")
+
+    add_configs("with_cuda", {description = "Enable cuda build.", default = false, type = "boolean"})
+
+    add_deps("cmake", "lz4")
+    on_load("windows", "linux", "macosx", function (package)
+        if package:is_plat("windows") and not package:config("shared") then
+            package:add("defines", "FLANN_STATIC")
+        end
+        if package:config("with_cuda") then
+            package:add("deps", "cuda", {system = true})
+        end
+    end)
+
+    on_install("windows", "linux", "macosx", function (package)
+        os.cd("src/cpp")
+        io.replace("flann/defines.h", "#ifdef WIN32", "#ifdef _WIN32", {plain = true})
+        io.writefile("xmake.lua", format([[
+            add_rules("mode.debug", "mode.release")
+            add_requires("lz4")
+            if is_plat("windows") then
+                if is_kind("static") then
+                    add_defines("FLANN_STATIC")
+                elseif is_kind("shared") then
+                    add_defines("FLANN_EXPORTS")
+                end
+                add_cxxflags("/bigobj")
+            end
+            target("flann")
+                set_kind("$(kind)")
+                add_files("flann/flann.cpp")
+                add_includedirs(".")
+                add_includedirs("flann")
+                add_packages("lz4")
+                add_headerfiles("(flann/config.h)", "(flann/defines.h)", "(flann/flann.h)")
+                add_headerfiles("(flann/flann.hpp)", "(flann/general.h)", "(flann/algorithms/*.h)", "(flann/io/*.h)", "(flann/nn/*.h)", "(flann/util/*.h)")
+            target("flann_cuda")
+                set_enabled(%s)
+                set_kind("$(kind)")
+                add_files("flann/algorithms/*.cu")
+                add_defines("FLANN_USE_CUDA")
+                add_includedirs(".")
+                add_includedirs("flann")
+                add_packages("lz4")
+        ]], package:config("with_cuda") and "true" or "false"))
+        local configs = {kind = package:config("shared") and "shared" or "static"}
+        if package:config("with_cuda") then
+            package:add("defines", "FLANN_USE_CUDA")
+        end
+        import("package.tools.xmake").install(package, configs)
+    end)
+
+    on_test(function (package)
+        assert(package:has_cfuncs("flann_find_nearest_neighbors_index", {includes = "flann/flann.h"}))
+    end)
diff --git a/packages/l/lz4/xmake.lua b/packages/l/lz4/xmake.lua
index 8863cb12b..957cd2a0c 100644
--- a/packages/l/lz4/xmake.lua
+++ b/packages/l/lz4/xmake.lua
@@ -7,13 +7,19 @@ package("lz4")
              "https://github.com/lz4/lz4.git")
     add_versions("v1.9.3", "030644df4611007ff7dc962d981f390361e6c97a34e5cbc393ddfbe019ffe2c1")
 
+    on_load("windows", function (package)
+        if package:config("shared") then
+            package:add("defines", "LZ4_DLL_IMPORT")
+        end
+    end)
+
     on_install(function (package)
         io.writefile("xmake.lua", [[
             add_rules("mode.debug", "mode.release")
             target("lz4")
                 set_kind("$(kind)")
                 add_files("lib/*.c")
-                add_headerfiles("lib/lz4.h")
+                add_headerfiles("lib/lz4.h", "lib/lz4hc.h", "lib/lz4frame.h")
                 if is_kind("shared") and is_plat("windows") then
                     add_defines("LZ4_DLL_EXPORT")
                 end