-- Copyright (C) 2025 kbz_8
-- This file is part of "Pulse"
-- For conditions of distribution and use, see copyright notice in LICENSE

add_repositories("local-repo Xmake")

local backends = {
	Vulkan = {
		option = "vulkan",
		default = not is_plat("wasm"),
		packages = { "vulkan-headers", "vulkan-memory-allocator" },
		custom = function()
			add_defines("VK_NO_PROTOTYPES")
			add_files("Sources/Backends/Vulkan/**.cpp")
		end
	},
	Metal = {
		option = "metal",
		default = is_plat("macosx", "iphoneos"),
		custom = function()
			add_frameworks("Metal")
			add_files("Sources/Backends/Metal/**.m")
		end
	},
	WebGPU = {
		option = "webgpu",
		default = is_plat("wasm"),
		custom = function()
			if not is_plat("wasm") then
				add_packages("wgpu-native")
			end
			add_cxflags("cl::/experimental:c11atomics")
		end
	},
	D3D11 = {
		option = "d3d11",
		default = is_plat("windows", "msys", "mingw"),
		custom = function()
			if is_plat("linux") then
				add_sysincludedirs("/usr/include/dxvk")
				add_syslinks("dxvk_d3d11", "dxvk_dxgi")
			else
				add_syslinks("d3d11", "dxgi", "windowscodecs")
			end
		end
	},
	Software = {
		option = "software",
		default = true,
		packages = { "spirv-vm", "cpuinfo", "spirv-reflect" },
		custom = function()
			add_cxflags("cl::/experimental:c11atomics")
		end
	},
	OpenGL = {
		option = "opengl",
		default = not is_plat("macosx", "iphoneos", "wasm", "appletvos"),
		packages = { "opengl-headers", "egl-headers" },
		custom = function()
			add_defines("EGL_NO_X11")
			if is_plat("windows", "mingw") then
				add_syslinks("User32")
			else
				remove_files("Sources/Backends/OpenGL/WGL/**.c")
			end
		end,
		before_build = function(target, os)
			local gles_dir = target:pkg("opengl-headers"):installdir()
			os.runv("python3", {"Scripts/GenerateOpenGLDefs.py", "Sources/Backends/OpenGL/OpenGLFunctions.h", gles_dir .. "/include/GLES3/gl32.h", "Sources/Backends/OpenGL/OpenGLWraps.h"})
		end
	},
}

local sanitizers = {
	asan = "address",
	lsan = "leak",
	tsan = "thread",
}

for opt, policy in table.orderpairs(sanitizers) do
	option(opt, { description = "Enable " .. opt, default = false })
	if has_config(opt) then
		set_policy("build.sanitizer." .. policy, true)
	end
end

for name, module in table.orderpairs(backends) do
	if module.option then
		option(module.option, { description = "Enables the " .. name .. " backend", default = module.default })
	end
end

add_rules("mode.debug", "mode.release")

add_includedirs("Includes")
set_languages("c17", "cxx20")
set_encodings("utf-8")

set_objectdir("build/Objs/$(os)_$(arch)")
set_targetdir("build/Bin/$(os)_$(arch)")
set_rundir("build/Bin/$(os)_$(arch)")
set_dependir("build/.deps")

set_optimize("fastest")

includes("Xmake/**.lua")

option("unitybuild", { description = "Build the library using unity build", default = false })

if is_plat("wasm") then
	backends.Vulkan = nil
else
	add_requires("tiny-c-thread")
	if has_config("webgpu") then
		add_requires("wgpu-native") -- Adding it manually here because wgpu-native is not available in wasm
	end
end

for name, module in pairs(backends) do
	if has_config(module.option) then
		if module.packages then
			add_requires(table.unpack(module.packages))
		end
	end
end

target("pulse_gpu")
	set_kind("$(kind)")
	add_defines("PULSE_BUILD")
	add_headerfiles("Sources/*.h", { prefixdir = "private", install = false })
	add_headerfiles("Sources/*.inl", { prefixdir = "private", install = false })

	if not is_plat("wasm") then
		add_packages("tiny-c-thread")
	end

	add_files("Sources/*.c")

	if has_config("unitybuild") then
		add_rules("c.unity_build", { batchsize = 6 })
	end

	if is_plat("wasm") then
		add_defines("PULSE_PLAT_WASM")
	end

	before_build(function(target)
		for name, module in pairs(backends) do
			if module ~= nil and has_config(module.option) and module.before_build then
				module.before_build(target, os)
			end
		end
	end)

	for name, module in pairs(backends) do
		if module ~= nil and has_config(module.option) then
			if module.packages then
				add_packages(table.unpack(module.packages))
			end

			add_defines("PULSE_ENABLE_" .. name:upper() .. "_BACKEND")

			add_headerfiles("Sources/Backends/" .. name .. "/**.h", { prefixdir = "private", install = false })
			add_headerfiles("Sources/Backends/" .. name .. "/**.inl", { prefixdir = "private", install = false })

			-- Checks if there are C files in the backend directory and add them if so
			local cfiles = os.files("Sources/Backends/" .. name .. "/**.c")
			if #cfiles > 0 then
				add_files("Sources/Backends/" .. name .. "/**.c")
			end

			if module.custom then
				module.custom()
			end
		end
	end

	on_load(function(target)
		if target:kind() == "static" then
			target:add("defines", "PULSE_STATIC", { public = true })
		end
	end)
target_end()

includes("Tests/xmake.lua")
includes("Examples/*.lua")