package("opencv-mobile")
    set_homepage("https://github.com/nihui/opencv-mobile")
    set_description("The minimal opencv for Android, iOS, ARM Linux, Windows, Linux, MacOS, WebAssembly")
    set_license("Apache-2.0")

    -- parse version-tag: version -> ocv, tag -> ocv-mobile
    if on_source then
        on_source(function (package)
            local raw_ver = package:requireinfo().version
            if raw_ver and raw_ver:find('-') then
                local vers = raw_ver:split('-', {plain = true})
                if #vers == 2 then
                    local ver_map = {
                        ["v29"] = {
                            ["4.10.0"] = "e9209285ad4d682536db4505bc06e46b94b9e56d91896e16c2853c83a870f004",
                            ["3.4.20"] = "85c19b443454d3ae839d8f4f7a6a71c79f9ac38592a8a96e2f806fc0c68b64f4", 
                        },
                        ["v31"] = {
                            ["4.10.0"] = "0ad54eef6fbd34787a6b13e9486f97e99102b1dc478eb74f2e8d30008c35be4e",
                            ["3.4.20"] = "60d05862441e85072e4e45b5241d1799de18f4a755bf26a7a89583d968b00e42", 
                        },
                        ["v33"] = {
                            ["4.11.0"] = "394b0dc31b9fee48333b6f043b0f22942ed95cd9b5acd101995e3054f468f930",
                            ["3.4.20"] = "6c0b45d1df1165c798d8e051b049c10ec55fa84cba2a6fbd6f7cfe93ab36ebd0", 
                        },
                        ["v34"] = {
                            ["4.12.0"] = "34f2f113dd78392baaba97cae74226a0337e14b4d6b938f17e215c25d547dad6",
                            ["3.4.20"] = "c976c30d47c18a6307f55703079a9c80f3bb73c7ec0dd50759a7e37aab111aff", 
                        }
                    }
                    local ocv_ver = vers[1]
                    local ocv_mobile_ver = vers[2]
                    if ver_map[ocv_mobile_ver] and ver_map[ocv_mobile_ver][ocv_ver] then
                        local url = string.format("https://github.com/nihui/opencv-mobile/releases/download/%s/opencv-mobile-%s.zip", ocv_mobile_ver, ocv_ver)
                        package:add("urls", url)
                        -- package:version_set(ocv_ver)
                        package:requireinfo().version = ocv_ver
                        package:add("versions", ocv_ver, ver_map[ocv_mobile_ver][ocv_ver])
                    else
                        raise("package(opencv-mobile): The version that you are requiring is not supported or does not exist.")
                    end
                else
                    raise("package(opencv-mobile): Invalid version format! Please specify the version in the format of '<opencv_version>-<mobile_tag>' (like 4.12.0-v34).")
                end
            else
                -- this map stores the latest opencv-mobile version-tag mapping
                -- key = opencv version, value = the latest opencv-mobile release tag
                local lts_version_map = {
                    ["4.12.0"] = "v34",
                    ["4.11.0"] = "v33",
                    ["4.10.0"] = "v31",
                    ["3.4.20"] = "v34",
                }

                package:add("urls", "https://github.com/nihui/opencv-mobile/releases/download/$(version).zip", {version = function (version)
                    local tag = lts_version_map[tostring(version)]
                    if tag then
                        return string.format("%s/opencv-mobile-%s", tag, tostring(version))
                    else
                        raise("The version of opencv-mobile that you are requiring is not supported or exists.")
                    end
                end})

                package:add("versions", "4.12.0", "34f2f113dd78392baaba97cae74226a0337e14b4d6b938f17e215c25d547dad6")  -- v34
                package:add("versions", "4.11.0", "394b0dc31b9fee48333b6f043b0f22942ed95cd9b5acd101995e3054f468f930")  -- v33
                package:add("versions", "4.10.0", "0ad54eef6fbd34787a6b13e9486f97e99102b1dc478eb74f2e8d30008c35be4e")  -- v31
                package:add("versions", "3.4.20", "c976c30d47c18a6307f55703079a9c80f3bb73c7ec0dd50759a7e37aab111aff")  -- v34
            end
        end)
    end

    add_patches("*", "patches/msvc.patch", "6fa760ea58c8b90c87129f16c84b128a4447ea11cee7d6568ea4f5e7ae250971")

    add_deps("cmake")
    if not is_plat("android", "iphoneos") then
        add_deps("openmp")
    end

    if on_check then
        on_check("android", function (package)
            -- https://github.com/android/ndk/issues/1202#issuecomment-768524852
            if package:is_arch("armeabi-v7a") then
                local ndk = package:toolchain("ndk"):config("ndkver")
                assert(ndk and tonumber(ndk) >= 23, "package(opencv-mobile): armeabi-v7a requires NDK version >= r23")
            end
        end)
    end

    on_load("android", "iphoneos", "linux", "macosx", "windows", "mingw@windows,msys", function (package)
        if package:is_plat("windows") then
            local arch = "x64"
            if     package:is_arch("x86")   then arch = "x86"
            elseif package:is_arch("arm64") then arch = "ARM64"
            end
            local linkdir = (package:config("shared") and "lib" or "staticlib")
            local vs = package:toolchain("msvc"):config("vs")
            local vc_ver = "vc13"
            if     vs == "2015" then vc_ver = "vc14"
            elseif vs == "2017" then vc_ver = "vc15"
            elseif vs == "2019" then vc_ver = "vc16"
            elseif vs == "2022" then vc_ver = "vc17"
            elseif vs == "2026" then vc_ver = "vc18"
            end
            package:add("linkdirs", linkdir) -- fix path for 4.9.0/vs2022
            package:add("linkdirs", path.join(arch, vc_ver, linkdir))
        elseif package:is_plat("mingw") then
            local arch = (package:is_arch("x86_64") and "x64" or "x86")
            local linkdir = (package:config("shared") and "lib" or "staticlib")
            package:add("linkdirs", path.join(arch, "mingw", linkdir))
        elseif package:is_plat("android") then
            local linkdir = (package:config("shared") and "libs" or "staticlibs")
            package:add("linkdirs", path.join("sdk/native", linkdir, package:targetarch()))
            package:add("linkdirs", path.join("sdk/native/3rdparty/libs", package:targetarch()))
            package:add("includedirs", "sdk/native/jni/include")
        elseif package:version():ge("4.0") then
            package:add("includedirs", "include/opencv4")
        end
    end)

    on_install("android", "iphoneos", "linux", "macosx", "windows", "mingw@windows,msys", function (package)
        if package:is_plat("windows", "mingw") then
            -- fix for v30-v34
            io.replace("modules/highgui/src/display_win32.cpp", "#include <mutex>\n\n", "#include <mutex>\n#include <cstdio>\n\n", {plain = true})
            -- fix for v30, v31
            io.replace("modules/highgui/src/display_win32.cpp", "static LRESULT", "LRESULT", {plain = true})
            io.replace("modules/highgui/src/display_win32.h", "#include <Windows.h>", "#include <windows.h>", {plain = true})
            io.replace("modules/highgui/src/display_win32.h", "static LRESULT", "LRESULT", {plain = true})
        end
        -- no platforms folder https://github.com/nihui/opencv-mobile/blob/v32/.github/workflows/release.yml#L73
        if package:is_plat("linux") and package:is_arch("arm64") and package:version():eq("4.11.0") then
            io.replace("cmake/platforms/OpenCV-Linux.cmake", "AND NOT CMAKE_CROSSCOMPILING", "AND FALSE", {plain = true})
        end

        local configs = {}
        table.insert(configs, "-DCMAKE_BUILD_TYPE=" .. (package:debug() and "Debug" or "Release"))
        table.insert(configs, "-DBUILD_SHARED_LIBS=" .. (package:config("shared") and "ON" or "OFF"))
        if package:is_plat("windows") then
            table.insert(configs, "-DBUILD_WITH_STATIC_CRT=" .. (package:has_runtime("MT", "MTd") and "ON" or "OFF"))
            if package:is_arch("arm64") then
                table.insert(configs, "-DOPENCV_SKIP_SYSTEM_PROCESSOR_DETECTION=ON")
                table.insert(configs, "-DAARCH64=ON")
            end
        elseif package:is_plat("mingw") then
            table.insert(configs, "-DCMAKE_SYSTEM_NAME=Windows")
            table.insert(configs, "-DCMAKE_SYSTEM_PROCESSOR=" .. (package:is_arch("x86_64") and "AMD64" or "i686"))
        elseif package:is_plat("macosx") then
            table.insert(configs, "-DCMAKE_SYSTEM_PROCESSOR=" .. (package:is_arch("x86_64") and "AMD64" or "ARM64"))
        elseif package:is_plat("android") then
            table.insert(configs, "-DANDROID_CPP_FEATURES=no-rtti no-exceptions")
            table.insert(configs, "-DOPENCV_DISABLE_FILESYSTEM_SUPPORT=ON")
            table.insert(configs, "-DANDROID_SUPPORT_FLEXIBLE_PAGE_SIZES=ON")
            if package:is_arch("arm64-v8a") then
                -- https://github.com/Tencent/ncnn/pull/4362#issuecomment-2857174155
                local ndk = package:toolchain("ndk"):config("ndkver")
                if ndk and tonumber(ndk) >= 24 then
                    table.insert(configs, "-DOPENCV_EXTRA_FLAGS=-mno-outline-atomics")
                end
            elseif package:is_arch("armeabi-v7a") then
                table.insert(configs, "-DANDROID_ARM_NEON=ON")
            end
        elseif package:is_plat("iphoneos") or (package:is_plat("linux") and package:is_arch("arm.*")) then
            table.insert(configs, "-DCMAKE_CXX_FLAGS=-fno-rtti -fno-exceptions")
        end
        local options = string.split(io.readfile("options.txt"), "\n", {plain = true})
        table.remove_if(options, function (_, option)
            return option:startswith("-DCMAKE_BUILD_TYPE") or option:startswith("-DBUILD_SHARED_LIBS") or option:startswith("-DBUILD_WITH_STATIC_CRT")
        end)
        table.join2(configs, options)
        import("package.tools.cmake").install(package, configs)

        if package:is_plat("windows") then
            local arch = "x64"
            if     package:is_arch("x86")   then arch = "x86"
            elseif package:is_arch("arm64") then arch = "ARM64"
            end
            if package:is_arch("arm64") then
                os.trymv(path.join(package:installdir(), "x64"), path.join(package:installdir(), "ARM64"))
                os.trymv(path.join(package:installdir(), "x86"), path.join(package:installdir(), "ARM64"))
            end
            local linkdir = (package:config("shared") and "lib" or "staticlib")
            local vs = package:toolchain("msvc"):config("vs")
            local vc_ver = "vc13"
            if     vs == "2015" then vc_ver = "vc14"
            elseif vs == "2017" then vc_ver = "vc15"
            elseif vs == "2019" then vc_ver = "vc16"
            elseif vs == "2022" then vc_ver = "vc17"
            elseif vs == "2026" then vc_ver = "vc18"
            end

            local libfiles = {}
            table.join2(libfiles, os.files(package:installdir(linkdir, "*.lib")))
            table.join2(libfiles, os.files(package:installdir(arch, vc_ver, linkdir, "*.lib")))
            for _, f in ipairs(libfiles) do
                if not f:match("opencv_.+") then
                    package:add("links", path.basename(f))
                end
            end
            package:addenv("PATH", "bin") -- fix path for 4.9.0/vs2022
            package:addenv("PATH", path.join(arch, vc_ver, "bin"))
        elseif package:is_plat("mingw") then
            local arch = package:is_arch("x86_64") and "x64" or "x86"
            local linkdir = (package:config("shared") and "lib" or "staticlib")
            for _, f in ipairs(os.files(package:installdir(arch, "mingw", linkdir, "lib*.a"))) do
                if not f:match("libopencv_.+") then
                    package:add("links", path.basename(f):match("lib(.+)"))
                end
            end
            package:addenv("PATH", path.join(arch, "mingw", "bin"))
        elseif package:is_plat("android") then
            for _, suffix in ipairs({"*.a", "*.so"}) do
                local lib_name = package:config("shared") and "libs" or "staticlibs"
                local libdir_1 = package:installdir(path.join("sdk/native", lib_name, package:targetarch()))
                local libdir_2 = package:installdir(path.join("sdk/native/3rdparty/libs", package:targetarch()))
                local libfiles = {}
                table.join2(libfiles, os.files(path.join(libdir_1, suffix)))
                table.join2(libfiles, os.files(path.join(libdir_2, suffix)))
                for _, f in ipairs(libfiles) do
                    package:add("links", path.basename(f):match("lib(.+)"))
                end
            end
            package:add("syslinks", "omp")
        else
            package:addenv("PATH", "bin")
        end
    end)

    on_test(function (package)
        assert(package:check_cxxsnippets({test = [[
            #include <iostream>
            void test(int argc, char** argv) {
                cv::CommandLineParser parser(argc, argv, "{help h||show help message}");
                if (parser.has("help")) {
                    parser.printMessage();
                }
                cv::Mat image(3, 3, CV_8UC1);
                std::cout << CV_VERSION << std::endl;
            }
        ]]}, {configs = {languages = "c++11"},
              includes = package:version():ge("3.0") and "opencv2/opencv.hpp" or "opencv/cv.h"}))
    end)