Sovereignty/jecs
1
0
Fork 0
mirror of https://github.com/Ukendio/jecs.git synced 2025-04-24 17:10:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Prepared query

Browse source
This commit is contained in:
Ukendio 2024-04-28 20:33:45 +02:00
parent b75dc91a6a
commit 52cf531764
1 changed files with 50 additions and 36 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

86
lib/init.lua
View file

@ -361,44 +361,14 @@ local function getSmallestMap(componentIndex, components)
return s.sparse
end
function World.query(world: World, ...: i53): (() -> (number, ...any)) | () -> ()
local compatibleArchetypes = {}
local components = { ... }
local archetypes = world.archetypes
local queryLength = #components
local firstArchetypeMap = getSmallestMap(world.componentIndex, components)
local PreparedQuery = {}
PreparedQuery.__index = PreparedQuery
if not firstArchetypeMap then
return noop()
end
function PreparedQuery:__iter()
local compatibleArchetypes = self.compatibleArchetypes
local queryLength = self.queryLength
local components = self.components
for id in firstArchetypeMap do
local archetype = archetypes[id]
local columns = archetype.columns
local archetypeRecords = archetype.records
local indices = {}
local skip = false
for i, componentId in components do
local index = archetypeRecords[componentId]
if not index then
skip = true
break
end
indices[i] = columns[index]
end
if skip then
continue
end
table.insert(compatibleArchetypes, {
archetype = archetype,
indices = indices
})
end
local lastArchetype, compatibleArchetype = next(compatibleArchetypes)
if not compatibleArchetype then
return noop()
@ -482,6 +452,50 @@ function World.query(world: World, ...: i53): (() -> (number, ...any)) | () -> (
end
end
function World.query(world: World, ...: i53)
local compatibleArchetypes = {}
local components = { ... }
local archetypes = world.archetypes
local queryLength = #components
local firstArchetypeMap = getSmallestMap(world.componentIndex, components)
if not firstArchetypeMap then
return noop()
end
for id in firstArchetypeMap do
local archetype = archetypes[id]
local columns = archetype.columns
local archetypeRecords = archetype.records
local indices = {}
local skip = false
for i, componentId in components do
local index = archetypeRecords[componentId]
if not index then
skip = true
break
end
indices[i] = columns[index]
end
if skip then
continue
end
table.insert(compatibleArchetypes, {
archetype = archetype,
indices = indices
})
end
return setmetatable({
queryLength = queryLength,
compatibleArchetypes = compatibleArchetypes,
components = components
}, PreparedQuery)
end
function World.component(world: World)
local id = world.nextId + 1
if id > HI_COMPONENT_ID then