mirror of
https://github.com/bakpakin/tiny-ecs.git
synced 2024-11-17 04:44:23 +00:00
Rename World:free to World:remove.
Fix World:free bug. Add __tostring method for Aspect and System. Add a few more tests. Add more comments. Give systems ids. Change injected "_id" field in Entities to "id" field.
This commit is contained in:
parent
f4c6ce1004
commit
b98cd5a14d
191
jojo.lua
191
jojo.lua
@ -6,19 +6,19 @@ local jojo = {
|
||||
-- Simplified class implementation with no inheritance or polymorphism.
|
||||
local setmetatable = setmetatable
|
||||
local function class()
|
||||
local c = {}
|
||||
local mt = {}
|
||||
setmetatable(c, mt)
|
||||
c.__index = c
|
||||
function mt.__call(_, ...)
|
||||
local newobj = {}
|
||||
setmetatable(newobj, c)
|
||||
if c.init then
|
||||
c.init(newobj, ...)
|
||||
end
|
||||
return newobj
|
||||
end
|
||||
return c
|
||||
local c = {}
|
||||
local mt = {}
|
||||
setmetatable(c, mt)
|
||||
c.__index = c
|
||||
function mt.__call(_, ...)
|
||||
local newobj = {}
|
||||
setmetatable(newobj, c)
|
||||
if c.init then
|
||||
c.init(newobj, ...)
|
||||
end
|
||||
return newobj
|
||||
end
|
||||
return c
|
||||
end
|
||||
|
||||
local World = class()
|
||||
@ -30,7 +30,6 @@ jojo.Aspect = Aspect
|
||||
jojo.System = System
|
||||
|
||||
local tinsert = table.insert
|
||||
local tremove = table.remove
|
||||
local tconcat = table.concat
|
||||
local pairs = pairs
|
||||
local ipairs = ipairs
|
||||
@ -82,7 +81,7 @@ function Aspect:init(required, excluded, oneRequired)
|
||||
-- Iterate through one-required Components
|
||||
for _, v in ipairs(oneRequired or {}) do
|
||||
if requiredSet[v] then -- If one-required Comp. is also required,
|
||||
-- don't need one required Components
|
||||
-- don't need one required Components
|
||||
self[3] = {}
|
||||
return
|
||||
end
|
||||
@ -97,17 +96,23 @@ end
|
||||
-- Composes multiple Aspects into one Aspect. The resulting Aspect will match
|
||||
-- any Entity that matches all sub Aspects.
|
||||
function Aspect.compose(...)
|
||||
|
||||
local newa = {{}, {}, {}}
|
||||
|
||||
for _, a in ipairs{...} do
|
||||
if a[4] then
|
||||
|
||||
if a[4] then -- Aspect must be empty Aspect
|
||||
return Aspect()
|
||||
end
|
||||
|
||||
for i = 1, 3 do
|
||||
for _, c in ipairs(a[i]) do
|
||||
tinsert(newa[i], c)
|
||||
end
|
||||
end
|
||||
|
||||
end
|
||||
|
||||
return Aspect(newa[1], newa[2], newa[3])
|
||||
end
|
||||
|
||||
@ -115,16 +120,26 @@ end
|
||||
|
||||
-- Returns boolean indicating if an Entity matches the Aspect.
|
||||
function Aspect:matches(entity)
|
||||
|
||||
-- Aspect is the empty Aspect
|
||||
if self[4] then return false end
|
||||
|
||||
local rs, es, os = self[1], self[2], self[3]
|
||||
|
||||
-- Assert Entity has all required Components
|
||||
for i = 1, #rs do
|
||||
local r = rs[i]
|
||||
if entity[r] == nil then return false end
|
||||
end
|
||||
|
||||
-- Assert Entity has no excluded Components
|
||||
for i = 1, #es do
|
||||
local e = es[i]
|
||||
if entity[e] ~= nil then return false end
|
||||
end
|
||||
|
||||
-- if Aspect has at least one Component in the one-required
|
||||
-- field, assert that the Entity has at least one of these.
|
||||
if #os >= 1 then
|
||||
for i = 1, #os do
|
||||
local o = os[i]
|
||||
@ -132,24 +147,28 @@ function Aspect:matches(entity)
|
||||
end
|
||||
return false
|
||||
end
|
||||
|
||||
return true
|
||||
end
|
||||
|
||||
-- Aspect:trace()
|
||||
|
||||
-- Prints out a description of this Aspect.
|
||||
function Aspect:trace()
|
||||
function Aspect:__tostring()
|
||||
if self[4] then
|
||||
print("Empty Aspect.")
|
||||
return "JojoAspect<>"
|
||||
else
|
||||
print("Required Components:", tconcat(self[1], ", "))
|
||||
print("Excluded Components:", tconcat(self[2], ", "))
|
||||
print("One Req. Components:", tconcat(self[3], ", "))
|
||||
return "JojoAspect<Required: {" ..
|
||||
tconcat(self[1], ", ") ..
|
||||
"}, Excluded: {" ..
|
||||
tconcat(self[2], ", ") ..
|
||||
"}, One Req.: {" ..
|
||||
tconcat(self[3], ", ") ..
|
||||
"}>"
|
||||
end
|
||||
end
|
||||
|
||||
----- System -----
|
||||
|
||||
System.nextID = 0
|
||||
|
||||
-- System(preupdate, update, [aspect])
|
||||
|
||||
-- Creates a new System with the given aspect and update callback. The update
|
||||
@ -162,6 +181,23 @@ function System:init(preupdate, update, aspect)
|
||||
self.update = update
|
||||
self.aspect = aspect or Aspect()
|
||||
self.active = true
|
||||
local id = System.nextID
|
||||
self.id = id
|
||||
System.nextID = id + 1
|
||||
end
|
||||
|
||||
function System:__tostring()
|
||||
return "JojoSystem<id: "..
|
||||
self.id ..
|
||||
"preupdate: " ..
|
||||
self.preupdate ..
|
||||
", update: " ..
|
||||
self.update ..
|
||||
", aspect: " ..
|
||||
self.aspect ..
|
||||
", active: " ..
|
||||
self.active ..
|
||||
">"
|
||||
end
|
||||
|
||||
----- World -----
|
||||
@ -172,35 +208,39 @@ end
|
||||
-- Systems after creation.
|
||||
function World:init(...)
|
||||
|
||||
local args = {...}
|
||||
local args = {...}
|
||||
|
||||
-- Table of Entity IDs to status
|
||||
self.status = {}
|
||||
-- Table of Entity IDs to status
|
||||
self.status = {}
|
||||
|
||||
-- Table of Entity IDs to Entities
|
||||
self.entities = {}
|
||||
-- Table of Entity IDs to Entities
|
||||
self.entities = {}
|
||||
|
||||
-- List of Systems
|
||||
self.systems = args
|
||||
-- List of Systems
|
||||
self.systems = args
|
||||
|
||||
-- Accumulated time for each System.
|
||||
self.times = {}
|
||||
-- Table of System IDs to System Indices
|
||||
local systemIndices = {}
|
||||
self.systemIndices = systemIndices
|
||||
|
||||
-- Next available entity ID
|
||||
self.nextID = 0
|
||||
-- Next available entity ID
|
||||
self.nextID = 0
|
||||
|
||||
-- Table of System indices to Sets of matching Entity IDs
|
||||
local aspectEntities = {}
|
||||
self.aspectEntities = aspectEntities
|
||||
for i, sys in ipairs(args) do
|
||||
aspectEntities[i] = {}
|
||||
end
|
||||
-- Table of System indices to Sets of matching Entity IDs
|
||||
local systemEntities = {}
|
||||
self.systemEntities = systemEntities
|
||||
|
||||
for i, sys in ipairs(args) do
|
||||
systemEntities[i] = {}
|
||||
systemIndices[sys.id] = i
|
||||
end
|
||||
|
||||
end
|
||||
|
||||
-- World:add(...)
|
||||
|
||||
-- Adds Entities to the World. An Entity is just a table of Components.
|
||||
-- Adds Entities to the World. Entities will enter the World the next time
|
||||
-- World:update(dt) is called.
|
||||
function World:add(...)
|
||||
local args = {...}
|
||||
local status = self.status
|
||||
@ -208,7 +248,7 @@ function World:add(...)
|
||||
for _, e in ipairs(args) do
|
||||
local id = self.nextID
|
||||
self.nextID = id + 1
|
||||
e._id = id
|
||||
e.id = id
|
||||
entities[id] = e
|
||||
status[id] = "add"
|
||||
end
|
||||
@ -217,65 +257,88 @@ end
|
||||
-- World:changed(...)
|
||||
|
||||
-- Call this function on any Entities that have changed such that they would
|
||||
-- now match different systems.
|
||||
function World:changed(...)
|
||||
-- now match different systems. Entities will be updated in the world the next
|
||||
-- time World:update(dt) is called.
|
||||
function World:change(...)
|
||||
local args = {...}
|
||||
local status = self.status
|
||||
for _, e in ipairs(args) do
|
||||
status[e._id] = "add"
|
||||
status[e.id] = "add"
|
||||
end
|
||||
end
|
||||
|
||||
-- World:free(...)
|
||||
|
||||
-- Frees Entities from the World.
|
||||
function World:free(...)
|
||||
-- Removes Entities from the World. Entities will exit the World the next time
|
||||
-- World:update(dt) is called.
|
||||
function World:remove(...)
|
||||
local args = {...}
|
||||
local status = self.status
|
||||
for _, e in ipairs(args) do
|
||||
status[e._id] = "free"
|
||||
status[e.id] = "remove"
|
||||
end
|
||||
end
|
||||
|
||||
-- World:update()
|
||||
|
||||
-- Updates the World.
|
||||
-- Updates the World, frees Entities that have been marked for freeing, adds
|
||||
-- entities that have been marked for adding, etc.
|
||||
function World:update(dt)
|
||||
|
||||
local statuses = self.status
|
||||
local aspectEntities = self.aspectEntities
|
||||
local systemEntities = self.systemEntities
|
||||
local entities = self.entities
|
||||
local systems = self.systems
|
||||
|
||||
for eid, s in pairs(statuses) do
|
||||
if s == "add" then
|
||||
local e = entities[eid]
|
||||
for sysi, set in pairs(aspectEntities) do
|
||||
local a = systems[sysi].aspect
|
||||
set[eid] = a:matches(e) and true or nil
|
||||
for sysid, eids in pairs(systemEntities) do
|
||||
local a = systems[sysid].aspect
|
||||
eids[eid] = a:matches(e) and true or nil
|
||||
end
|
||||
statuses[eid] = nil
|
||||
end
|
||||
end
|
||||
|
||||
for sysi, s in ipairs(self.systems) do
|
||||
for sysid, s in ipairs(self.systems) do
|
||||
|
||||
local preupdate = s.preupdate
|
||||
if s.active and preupdate then
|
||||
local eids = systemEntities[sysid]
|
||||
local active = s.active
|
||||
|
||||
-- Preupdate
|
||||
if active and preupdate then
|
||||
preupdate(dt)
|
||||
end
|
||||
local eids = aspectEntities[sysi]
|
||||
local u = s.update
|
||||
for eid in pairs(eids) do
|
||||
local status = statuses[eid]
|
||||
if status == "free" then
|
||||
eids[eid] = nil
|
||||
|
||||
if active then -- Free freed entities and update the others.
|
||||
|
||||
local u = s.update
|
||||
|
||||
for eid in pairs(eids) do
|
||||
local status = statuses[eid]
|
||||
if status == "remove" then
|
||||
eids[eid] = nil
|
||||
else
|
||||
u(entities[eid], dt)
|
||||
end
|
||||
end
|
||||
u(entities[eid], dt)
|
||||
|
||||
else -- Just free freed Entities
|
||||
|
||||
for eid in pairs(eids) do
|
||||
if statuses[eid] == "remove" then
|
||||
eids[eid] = nil
|
||||
end
|
||||
end
|
||||
|
||||
end
|
||||
end
|
||||
|
||||
-- Reset all statuses for next update
|
||||
for eid, s in pairs(statuses) do
|
||||
if s == "free" then
|
||||
if s == "remove" then
|
||||
entities[eid].id = nil
|
||||
entities[eid] = nil
|
||||
end
|
||||
|
10
jojotest.lua
10
jojotest.lua
@ -80,16 +80,20 @@ local world = World(moves)
|
||||
world:add(e1, e2, e3)
|
||||
world:update(21)
|
||||
assert(e1.xform.x == 21, "e1.xform.x should be 21, but is " .. e1.xform.x)
|
||||
assert(e2.xform.x == -19, "e2.xform.x should be -19, but is " .. e2.xform.x)
|
||||
assert(e3.xform.y == 68, "e3.xform.y should be 68, but is " .. e3.xform.y)
|
||||
|
||||
world:free(e3, e2)
|
||||
world:remove(e3, e2)
|
||||
world:update(20)
|
||||
assert(e1.xform.x == 41, "e1.xform.x should be 41, but is " .. e1.xform.x)
|
||||
assert(e2.xform.x == -19, "e2.xform.x should be -19, but is " .. e2.xform.x)
|
||||
assert(e3.xform.y == 68, "e3.xform.y should be 68, but is " .. e3.xform.y)
|
||||
|
||||
world:add(e3, e2)
|
||||
world:update(19)
|
||||
world:free(e3, e2)
|
||||
world:remove(e3, e2)
|
||||
e1.vel = nil
|
||||
world:changed(e1)
|
||||
world:change(e1)
|
||||
world:update(11)
|
||||
|
||||
assert(e1.xform.x == 60, "e1.xform.x should be 60, but is " .. e1.xform.x)
|
||||
|
Loading…
Reference in New Issue
Block a user