jecs/benches/query.bench.lua

--!optimize 2
--!native

local ReplicatedStorage = game:GetService("ReplicatedStorage")
local rgb = require(ReplicatedStorage.rgb)
local Matter = require(ReplicatedStorage.DevPackages.Matter)
local jecs = require(ReplicatedStorage.Lib)
local ecr = require(ReplicatedStorage.DevPackages.ecr)
local newWorld = Matter.World.new()
local ecs = jecs.World.new()

local A1 = Matter.component()
local A2 = Matter.component()
local A3 = Matter.component()
local A4 = Matter.component()
local A5 = Matter.component()
local A6 = Matter.component()
local A7 = Matter.component()
local A8 = Matter.component()

local B1 = ecr.component()
local B2 = ecr.component()
local B3 = ecr.component()
local B4 = ecr.component()
local B5 = ecr.component()
local B6 = ecr.component()
local B7 = ecr.component()
local B8 = ecr.component()

local C1 = ecs:entity()
local C2 = ecs:entity()
local C3 = ecs:entity()
local C4 = ecs:entity()
local C5 = ecs:entity()
local C6 = ecs:entity()
local C7 = ecs:entity()
local C8 = ecs:entity()

local registry2 = ecr.registry()

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

local common = 0
local N = 2^16-2
local archetypes = {}
for i = 1, N do 
	local id = registry2.create()
	local combination = ""
	local n = newWorld:spawn()
	
	local entity = ecs:entity()

	if flip() then 
		combination ..= "B"
		registry2:set(id, B2, {value = true}) 
		ecs:add(entity, C2, { value = true})
		newWorld:insert(n, A2({value = true}))		
	end
	if flip() then 
		combination ..= "C"
		registry2:set(id, B3, {value = true}) 
		ecs:add(entity, C3, { value = true})
		newWorld:insert(n, A3({value = true}))
	end
	if flip() then 
		combination ..= "D"
		registry2:set(id, B4, {value = true}) 
		ecs:add(entity, C4, { value = true})
		newWorld:insert(n, A4({value = true}))		
	end
	if flip() then 
		combination ..= "E"
		registry2:set(id, B5, {value = true}) 
		ecs:add(entity, C5, { value = true})
		newWorld:insert(n, A5({value = true}))		
	end
	if flip() then 
		combination ..= "F"
		registry2:set(id, B6, {value = true}) 
		ecs:add(entity, C6, { value = true})
		newWorld:insert(n, A6({value = true}))		
	end
	if flip() then 
		combination ..= "G"
		registry2:set(id, B7, {value = true}) 
		ecs:add(entity, C7, { value = true})
		newWorld:insert(n, A7({value = true}))		
	end
	if flip() then 
		combination ..= "H"
		registry2:set(id, B8, {value = true}) 
		ecs:add(entity, C8, { value = true})
		newWorld:insert(n, A8({value = true}))		
	end

	if #combination == 7 then 
		combination = "A" .. combination
		common += 1
		registry2:set(id, B1, {value = true}) 
		ecs:add(entity, C1, { value = true})
		newWorld:insert(n, A1({value = true}))	
	end

	archetypes[combination] = true
end

local white = rgb.white
local yellow = rgb.yellow
local gray = rgb.gray
local green = rgb.green

local WALL = gray(" │ ")

local numberOfArchetypes = 0
for _  in archetypes do 
	numberOfArchetypes += 1
end
print(common)

print(
	"N entities "..yellow(N)
	..WALL
	.."with common components: "
	..yellow(tostring(common).."/"..tostring(N)).." "
	..yellow("("..string.format("%.2f", (common / (2^16 - 2)* 100)).."%)")
	..WALL
	..yellow("Total Archetypes: "..numberOfArchetypes)
)

return {
	ParameterGenerator = function()
		return
	end,

	Functions = {
		Matter = function() 
			local matched = 0
			for entityId, firstComponent in newWorld:query(A1, A2, A3, A4) do
				matched += 1
			end
		end,


		ECR = function() 
			local matched = 0
			for entityId, firstComponent in registry2:view(B1, B2, B3, B4) do
				matched += 1
			end
		end,


		Jecs = function() 
			local matched = 0
			for entityId, firstComponent in ecs:query(C1, C2, C3, C4) do
				matched += 1
			end
		end

	},
}