demo/src/ReplicatedStorage/systems/receive_replication.luau

@@ -74,12 +74,11 @@ return function(world: types.World)
             local removed = map.removed
 
             if removed then
-                for _, entity in removed do
-                    entity = ecs_map_get(world, entity)
-                    if not world:contains(entity) then
+                for i, e in removed do
+                    if not world:contains(e) then
                         continue
                     end
-                    world:remove(entity, id)
+                    world:remove(e, id)
                 end
             end
         end

jecs.d.ts

@@ -105,7 +105,7 @@ export class World {
 	/**
 	 * Creates a new World.
 	 */
-	private constructor();
+	constructor();
 
 	/**
 	 * Enforces a check for entities to be created within a desired range.
@@ -249,8 +249,6 @@ export class World {
 	query<T extends Id[]>(...components: T): Query<InferComponents<T>>;
 }
 
-export function world(): World;
-
 export function component<T>(): Entity<T>;
 
 export function tag(): Tag;

jecs.luau

@@ -52,8 +52,6 @@ export type Query<T...> = typeof(setmetatable(
 	}
 ))
 
-type QueryArm<T...> = () -> ()
-
 export type Observer = {
 	callback: (archetype: Archetype) -> (),
 	query: QueryInner,
@@ -439,7 +437,6 @@ end
 
 local function archetype_move(
 	entity_index: EntityIndex,
-	entity: Entity,
 	to: Archetype,
 	dst_row: i24,
 	from: Archetype,
@@ -453,9 +450,6 @@ local function archetype_move(
 	local id_types = from.types
 	local columns_map = to.columns_map
 
-	if src_row ~= last then
-		-- If the entity is the last row in the archetype then swapping it would be meaningless.
-
 	for i, column in src_columns do
 		if column == NULL_ARRAY then
 			continue
@@ -469,42 +463,35 @@ local function archetype_move(
 			dst_column[dst_row] = column[src_row]
 		end
 
+		-- If the entity is the last row in the archetype then swapping it would be meaningless.
+		if src_row ~= last then
 			-- Swap rempves columns to ensure there are no holes in the archetype.
 			column[src_row] = column[last]
+		end
 		column[last] = nil
 	end
 
+	local moved = #src_entities
 
 	-- Move the entity from the source to the destination archetype.
 	-- Because we have swapped columns we now have to update the records
 	-- corresponding to the entities' rows that were swapped.
+	local e1 = src_entities[src_row]
+	local e2 = src_entities[moved]
 
-		local e2 = src_entities[last]
+	if src_row ~= moved then
 		src_entities[src_row] = e2
+	end
+
+	src_entities[moved] = nil :: any
+	dst_entities[dst_row] = e1
 
 	local sparse_array = entity_index.sparse_array
+
+	local record1 = sparse_array[ECS_ENTITY_T_LO(e1 :: number)]
 	local record2 = sparse_array[ECS_ENTITY_T_LO(e2 :: number)]
+	record1.row = dst_row
 	record2.row = src_row
-	else
-		for i, column in src_columns do
-			if column == NULL_ARRAY then
-				continue
-			end
-			-- Retrieves the new column index from the source archetype's record from each component
-			-- We have to do this because the columns are tightly packed and indexes may not correspond to each other.
-			local dst_column = columns_map[id_types[i]]
-
-			-- Sometimes target column may not exist, e.g. when you remove a component.
-			if dst_column then
-				dst_column[dst_row] = column[src_row]
-			end
-
-			column[last] = nil
-		end
-	end
-
-	src_entities[last] = nil :: any
-	dst_entities[dst_row] = entity
 end
 
 local function archetype_append(
@@ -537,7 +524,7 @@ local function entity_move(
 	local sourceRow = record.row
 	local from = record.archetype
 	local dst_row = archetype_append(entity, to)
-	archetype_move(entity_index, entity, to, dst_row, from, sourceRow)
+	archetype_move(entity_index, to, dst_row, from, sourceRow)
 	record.archetype = to
 	record.row = dst_row
 end
@@ -576,7 +563,7 @@ local function world_get(world: World, entity: Entity,
 	if not b then
 		return va
 	elseif not c then
-		return va, fetch(b, columns_map, row)
+		return va, fetch(a, columns_map, row)
 	elseif not d then
 		return va, fetch(b, columns_map, row), fetch(c, columns_map, row)
 	elseif not e then
@@ -671,9 +658,6 @@ local function id_record_ensure(world: World, id: Entity): ComponentRecord
 	local relation = id
 	local target = 0
 	local is_pair = ECS_IS_PAIR(id :: number)
-
-	local has_delete = false
-
 	if is_pair then
 		relation = entity_index_get_alive(entity_index, ECS_PAIR_FIRST(id :: number)) :: i53
 		ecs_assert(relation and entity_index_is_alive(
@@ -681,19 +665,16 @@ local function id_record_ensure(world: World, id: Entity): ComponentRecord
 		target = entity_index_get_alive(entity_index, ECS_PAIR_SECOND(id :: number)) :: i53
 		ecs_assert(target and entity_index_is_alive(
 			entity_index, target), ECS_INTERNAL_ERROR)
+	end
 
+	local cleanup_policy = world_target(world, relation, EcsOnDelete, 0)
 	local cleanup_policy_target = world_target(world, relation, EcsOnDeleteTarget, 0)
 
-		if cleanup_policy_target == EcsDelete then
-			has_delete = true
-		end
-	else
-		local cleanup_policy = world_target(world, relation, EcsOnDelete, 0)
+	local has_delete = false
 
-		if cleanup_policy == EcsDelete then
+	if cleanup_policy == EcsDelete or cleanup_policy_target == EcsDelete then
 		has_delete = true
 	end
-	end
 
 	local on_add, on_change, on_remove = world_get(world,
 		relation, EcsOnAdd, EcsOnChange, EcsOnRemove)
@@ -2045,16 +2026,18 @@ local function ecs_bulk_insert(world: World, entity: Entity, ids: { Entity }, va
 		local value = values[i] :: any
 
 		local on_add = idr.hooks.on_add
-
-		if value ~= nil then
-			columns_map[id][row] = value
 		local on_change = idr.hooks.on_change
+
+		if value then
+			columns_map[id][row] = value
 			local hook = if set then on_change else on_add
 			if hook then
 				hook(entity, id, value :: any)
 			end
-		elseif on_add then
-			on_add(entity, id)
+		else
+			if on_add then
+				on_add(entity, id, value)
+			end
 		end
 	end
 end
@@ -2157,41 +2140,38 @@ local function world_new()
 		return r
 	end
 
-	-- local function inner_entity_index_try_get(entity: number): Record?
-	-- 	local r = inner_entity_index_try_get_any(entity)
+	-- local function entity_index_try_get_safe(entity: number): Record?
+	-- 	local r = entity_index_try_get_any_fast(entity_index, entity)
 	-- 	if r then
 	-- 		local r_dense = r.dense
 	-- 		if r_dense > entity_index.alive_count then
 	-- 			return nil
 	-- 		end
-	-- 		if eindex_dense_array[r_dense] ~= entity then
+	-- 		if entity_index.dense_array[r_dense] ~= entity then
 	-- 			return nil
 	-- 		end
 	-- 	end
 	-- 	return r
 	-- end
 
-	local function inner_entity_index_try_get_unsafe(entity: number): Record?
-		local r = inner_entity_index_try_get_any(entity)
+	local function inner_entity_index_try_get(entity: number): Record?
+		local r = eindex_sparse_array[ECS_ENTITY_T_LO(entity)]
 		if r then
-			local r_dense = r.dense
-			-- if r_dense > entity_index.alive_count then
-			-- 	return nil
-			-- end
-			if eindex_dense_array[r_dense] ~= entity then
+			if eindex_dense_array[r.dense] ~= entity then
 				return nil
 			end
 		end
 		return r
 	end
 
+
 	local function inner_world_add<T, a>(
 		world: World,
 		entity: Entity<T>,
 		id: Id<a>
 	): ()
 		local entity_index = world.entity_index
-		local record = inner_entity_index_try_get_unsafe(entity :: number)
+		local record = inner_entity_index_try_get(entity :: number)
 		if not record then
 			return
 		end
@@ -2219,7 +2199,7 @@ local function world_new()
 
 	local function inner_world_get(world: World, entity: Entity,
 		a: Id, b: Id?, c: Id?, d: Id?, e: Id?): ...any
-		local record = inner_entity_index_try_get_unsafe(entity::number)
+		local record = inner_entity_index_try_get(entity::number)
 		if not record then
 			return nil
 		end
@@ -2237,7 +2217,7 @@ local function world_new()
 		if not b then
 			return va
 		elseif not c then
-			return va, fetch(b, columns_map, row)
+			return va, fetch(a, columns_map, row)
 		elseif not d then
 			return va, fetch(b, columns_map, row), fetch(c, columns_map, row)
 		elseif not e then
@@ -2250,7 +2230,7 @@ local function world_new()
 	local function inner_world_has(world: World, entity: i53,
 		a: i53, b: i53?, c: i53?, d: i53?, e: i53?): boolean
 
-		local record = inner_entity_index_try_get_unsafe(entity)
+		local record = inner_entity_index_try_get(entity)
 		if not record then
 			return false
 		end
@@ -2270,7 +2250,7 @@ local function world_new()
 	end
 
 	local function inner_world_target<T, a>(world: World, entity: Entity<T>, relation: Id<a>, index: number?): Entity?
-		local record = inner_entity_index_try_get_unsafe(entity :: number)
+		local record = inner_entity_index_try_get(entity :: number)
 		if not record then
 			return nil
 		end
@@ -2332,7 +2312,7 @@ local function world_new()
 	end
 
 	local function inner_world_set<T, a>(world: World, entity: Entity<T>, id: Id<a>, data: a): ()
-		local record = inner_entity_index_try_get_unsafe(entity :: number)
+		local record = inner_entity_index_try_get(entity :: number)
 		if not record then
 			return
 		end
@@ -2433,7 +2413,7 @@ local function world_new()
 	end
 
 	local function inner_world_remove<T, a>(world: World, entity: Entity<T>, id: Id<a>)
-		local record = inner_entity_index_try_get_unsafe(entity :: number)
+		local record = inner_entity_index_try_get(entity :: number)
 		if not record then
 			return
 		end
@@ -2470,8 +2450,8 @@ local function world_new()
 				local idr_archetype = archetypes[archetype_id]
 				local entities = idr_archetype.entities
 				local n = #entities
-				table.move(entities, 1, n, count + 1, queue)
 				count += n
+				table.move(entities, 1, n, #queue + 1, queue)
 			end
 			for _, e in queue do
 				inner_world_remove(world, e, entity)
@@ -2553,12 +2533,11 @@ local function world_new()
 				end
 			end
 		end
-
 	end
 
 	local function inner_world_delete<T>(world: World, entity: Entity<T>)
 		local entity_index = world.entity_index
-		local record = inner_entity_index_try_get_unsafe(entity::number)
+		local record = inner_entity_index_try_get(entity::number)
 		if not record then
 			return
 		end
@@ -2596,42 +2575,16 @@ local function world_new()
 					archetype_destroy(world, idr_archetype)
 				end
 			else
-				local on_remove = idr.hooks.on_remove
-				if on_remove then
 				for archetype_id in idr.records do
 					local idr_archetype = archetypes[archetype_id]
-						local to = archetype_traverse_remove(world, entity, idr_archetype)
 					local entities = idr_archetype.entities
 					local n = #entities
 					for i = n, 1, -1 do
-							local e = entities[i]
-							on_remove(e, entity)
-							local r = eindex_sparse_array[ECS_ID(e :: number)]
-							local from = r.archetype
-							if from ~= idr_archetype then
-								-- unfortunately the on_remove hook allows a window where `e` can have changed archetype
-								-- this is hypothetically not that expensive of an operation anyways
-								to = archetype_traverse_remove(world, entity, from)
-							end
-							entity_move(entity_index, e, r, to)
+						inner_world_remove(world, entities[i], entity)
 					end
 
 					archetype_destroy(world, idr_archetype)
 				end
-				else
-					for archetype_id in idr.records do
-						local idr_archetype = archetypes[archetype_id]
-						local to = archetype_traverse_remove(world, entity, idr_archetype)
-						local entities = idr_archetype.entities
-						local n = #entities
-						for i = n, 1, -1 do
-							local e = entities[i]
-							entity_move(entity_index, e, eindex_sparse_array[ECS_ID(e :: number)], to)
-						end
-
-						archetype_destroy(world, idr_archetype)
-					end
-				end
 			end
 		end
 
@@ -2850,7 +2803,7 @@ local function world_new()
 			if value == NULL then
 				inner_world_add(world, i, ty)
 			else
-				inner_world_set(world, i, ty, value)
+				inner_world_add(world, i, ty, value)
 			end
 		end
 	end

package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@rbxts/jecs",
-	"version": "0.7.3",
+	"version": "0.7.1",
 	"description": "Stupidly fast Entity Component System",
 	"main": "jecs.luau",
 	"repository": {

test/tests.luau

@@ -22,13 +22,9 @@ type Entity<T=nil> = jecs.Entity<T>
 type Id<T=unknown> = jecs.Id<T>
 
 local entity_visualiser = require("@tools/entity_visualiser")
+local lifetime_tracker_add = require("@tools/lifetime_tracker")
 local dwi = entity_visualiser.stringify
 
-TEST("repro", function()
-
-
-end)
-
 TEST("bulk", function()
 	local world = jecs.world()
 	local A = world:component()
@@ -46,10 +42,7 @@ TEST("bulk", function()
 	CHECK(world:get(e, B) == 2)
 	CHECK(world:get(e, C) == 3)
 
-	jecs.bulk_insert(world, e,
-		{ D, E, F },
-		{ 4, nil, 5 }
-	)
+	jecs.bulk_insert(world, e, { D, E, F }, { 4, nil, 5 })
 	CHECK(world:get(e, A) == 1)
 	CHECK(world:get(e, B) == 2)
 	CHECK(world:get(e, C) == 3)
@@ -58,10 +51,7 @@ TEST("bulk", function()
 	CHECK(world:get(e, E) == nil and world:has(e, E))
 	CHECK(world:get(e, F) == 5)
 
-	jecs.bulk_insert(world, e,
-		{ A, D, E, F, C },
-		{ 10, 40, nil, 50, 30 }
-	)
+	jecs.bulk_insert(world, e, { A, D, E, F, C }, { 10, 40, nil, 50, 30 })
 
 	CHECK(world:get(e, A) == 10)
 	CHECK(world:get(e, B) == 2)
@@ -386,51 +376,8 @@ TEST("world:contains()", function()
 end)
 
 TEST("world:delete()", function()
-	do CASE "pair(OnDelete, Delete)"
-	    local world = jecs.world()
-	    local ct = world:component()
-	    world:add(ct, jecs.pair(jecs.OnDelete, jecs.Delete))
-
-	    local e1 = world:entity()
-	    local e2 = world:entity()
-
-	    local dummy = world:entity()
-
-	    world:add(e1, ct)
-	    world:add(e2, jecs.pair(ct, dummy))
-
-	    world:delete(dummy)
-
-	    CHECK(world:contains(e2))
-
-	    world:delete(ct)
-
-	    CHECK(not world:contains(e1))
-	end
-
-	do CASE "pair(OnDeleteTarget, Delete)"
-	    local world = jecs.world()
-	    local ct = world:component()
-	    world:add(ct, jecs.pair(jecs.OnDeleteTarget, jecs.Delete))
-
-	    local e1 = world:entity()
-	    local e2 = world:entity()
-
-	    local dummy = world:entity()
-
-	    world:add(e1, ct)
-
-	    world:add(e2, jecs.pair(ct, dummy))
-
-	    world:delete(dummy)
-
-	    CHECK(not world:contains(e2))
-
-	    world:delete(ct)
-
-	    CHECK(world:contains(e1))
-	end
 	do CASE "remove (*, R) pairs when relationship is invalidated"
+		print("-------")
 		local world = jecs.world()
 		local e1 = world:entity()
 		local e2 = world:entity()
@@ -494,7 +441,7 @@ TEST("world:delete()", function()
 		local A = world:entity()
 		local B = world:entity()
 
-		world:add(Relation, pair(jecs.OnDeleteTarget, jecs.Delete))
+		world:add(Relation, pair(jecs.OnDelete, jecs.Delete))
 
 		local entity = world:entity()
 

wally.toml

@@ -1,6 +1,6 @@
 [package]
 name = "ukendio/jecs"
-version = "0.7.3"
+version = "0.7.1"
 registry = "https://github.com/UpliftGames/wally-index"
 realm = "shared"
 license = "MIT"