xmake.lua

-- 设置项目信息
set_languages("cxx17")
set_allowedplats("windows", "linux")

-- 添加依赖
add_requires("opencv", {system = true})

-- 添加编译规则
add_rules("plugin.compile_commands.autoupdate", {outputdir = "build"})
add_rules("mode.release")

-- 定义选项
option("tensorrt")
    set_showmenu(true)
    set_description("TensorRT Path. Example: /usr/local/tensorrt")
    on_check(function (option)
        if not option:enabled() then 
            raise("TensorRT path is not set. Please specify the TensorRT path.")
        end 
    end)

option("deploy")
    set_showmenu(true)
    set_description("TensorRT-YOLO Project Path.")
    on_check(function (option)
        if not option:enabled() then 
            raise("TensorRT-YOLO project path is not set. Please specify the TensorRT-YOLO Project path.")
        end 
    end)

-- 定义目标
target("segment")
    -- 设置目标类型
    set_kind("binary")

    -- 设置编译路径
    set_targetdir("$(projectdir)/bin")

    -- 设置运行路径
    set_rundir("$(projectdir)")

    -- 添加依赖
    add_packages("opencv")

    -- 添加文件
    add_files("segment.cpp")

    -- 添加 cuda
    add_rules("cuda")
    add_cugencodes("native")
    add_cuflags("-allow-unsupported-compiler")

    -- 添加deploy链接目录和链接库
    if has_config("deploy") then
        add_includedirs(path.join("$(deploy)", "include"))
        add_linkdirs(path.join("$(deploy)", "lib"))
        add_links("deploy")

        -- 复制deploy库文件到项目目录
        after_build(function (target)

            -- 定义deploy库文件路径
            local deploy_lib_name = is_host("windows") and "deploy.dll" or "libdeploy.so"
            local deploy_lib_path = path.join("$(deploy)", "lib", deploy_lib_name)

            if os.isfile(deploy_lib_path) then

                -- 目标文件路径
                local deploy_lib_target_path = path.join(target:targetdir(), deploy_lib_name)
            
                -- 复制文件
                os.cp(deploy_lib_path, deploy_lib_target_path)
            end
        end)
    end

    -- 添加TensorRT链接目录和链接库
    if has_config("tensorrt") then
        local tensorrt_path = get_config("tensorrt")
        add_includedirs(path.join(tensorrt_path, "include"))
        add_linkdirs(path.join(tensorrt_path, "lib"))
        local libs = is_plat("windows") and os.exists(path.join(get_config("tensorrt"), "lib", "nvinfer_10.dll")) and "nvinfer_10 nvinfer_plugin_10 nvonnxparser_10" or "nvinfer nvinfer_plugin nvonnxparser"
        add_links(libs:split("%s+"))
    end