jecs/test/benches/query.luau

--!optimize 2
--!native

local testkit = require("@modules/testkit")
local BENCH, START = testkit.benchmark()
local function TITLE(title: string)
	print()
	print(testkit.color.white(title))
end

local jecs = require("@jecs")
local mirror = require("@mirror")

do
	TITLE(testkit.color.white_underline("Jecs query"))
	local ecs = jecs.world() :: jecs.World
	do
		TITLE("one component in common")

		local function view_bench(world: jecs.World,
			A: jecs.Id,
			B: jecs.Id,
			C: jecs.Id,
			D: jecs.Id,
			E: jecs.Id,
			F: jecs.Id,
			G: jecs.Id,
			H: jecs.Id
		)
			BENCH("4 component", function()
				for _ in world:query(D, C, B, A) do
				end
			end)

		end

		local D1 = ecs:component()
		local D2 = ecs:component()
		local D3 = ecs:component()
		local D4 = ecs:component()
		local D5 = ecs:component()
		local D6 = ecs:component()
		local D7 = ecs:component()
		local D8 = ecs:component()

		local function flip()
			return math.random() >= 0.15
		end

		local added = 0
		local archetypes = {}
		for i = 1, 2 ^ 12 - 2 do
			local entity = ecs:entity()
			ecs:add(entity, entity)

			local combination = ""

			if flip() then
				combination ..= "B"
				ecs:set(entity, D2, { value = true })
			end
			if flip() then
				combination ..= "C"
				ecs:set(entity, D3, { value = true })
			end
			if flip() then
				combination ..= "D"
				ecs:set(entity, D4, { value = true })
			end
			if flip() then
				combination ..= "E"
				ecs:set(entity, D5, { value = true })
			end
			if flip() then
				combination ..= "F"
				ecs:set(entity, D6, { value = true })
			end
			if flip() then
				combination ..= "G"
				ecs:set(entity, D7, { value = true })
			end
			if flip() then
				combination ..= "H"
				ecs:set(entity, D8, { value = true })
			end

			if #combination == 7 then
				added += 1
				ecs:set(entity, D1, { value = true })
			end
			archetypes[combination] = true
		end

		local a = 0
		for _ in archetypes do
			a += 1
		end

		view_bench(ecs, D1, D2, D3, D4, D5, D6, D7, D8)
	end
end

do
	TITLE(testkit.color.white_underline("Mirror query"))
	local ecs = mirror.World.new() :: jecs.World
	do
		TITLE("one component in common")

		local function view_bench(
			world: mirror.World,
			A: jecs.Id,
			B: jecs.Id,
			C: jecs.Id,
			D: jecs.Id,
			E: jecs.Id,
			F: jecs.Id,
			G: jecs.Id,
			H: jecs.Id
		)
			BENCH("4 component", function()
				for _ in world:query(D, C, B, A) do
				end
			end)
		end

		local D1 = ecs:component()
		local D2 = ecs:component()
		local D3 = ecs:component()
		local D4 = ecs:component()
		local D5 = ecs:component()
		local D6 = ecs:component()
		local D7 = ecs:component()
		local D8 = ecs:component()

		local function flip()
			return math.random() >= 0.5
		end

		local added = 0
		local archetypes = {}
		for i = 1, 2 ^ 12 - 2 do
			local entity = ecs:entity()
			ecs:add(entity, entity)

			local combination = ""

			if flip() then
				combination ..= "B"
				ecs:set(entity, D2, { value = true })
			end
			if flip() then
				combination ..= "C"
				ecs:set(entity, D3, { value = true })
			end
			if flip() then
				combination ..= "D"
				ecs:set(entity, D4, { value = true })
			end
			if flip() then
				combination ..= "E"
				ecs:set(entity, D5, { value = true })
			end
			if flip() then
				combination ..= "F"
				ecs:set(entity, D6, { value = true })
			end
			if flip() then
				combination ..= "G"
				ecs:set(entity, D7, { value = true })
			end
			if flip() then
				combination ..= "H"
				ecs:set(entity, D8, { value = true })
			end

			if flip() then
				ecs:set(entity, D1, { value = true })
			end
		end

		local a = 0
		for _ in archetypes do
			a += 1
		end

		view_bench(ecs, D1, D2, D3, D4, D5, D6, D7, D8)
	end
end