auto-link special tracked parts on creation

mainly to fix temporary damagezones (from projectiles) which we want to use with damage_zone_hit events for example

the caching method, while better than looking through local parts, had its flaws

Author: pingu7867

lua/pac3/core/client/part_pool.lua

@@ -645,6 +645,7 @@ function pac.EnablePartsByClass(classname, enable)
 	end
 end
 
+local known_special_link_parts = {}
 function pac.LinkSpecialTrackedPartsForEvent(part, ply)
 	part.erroring_cached_parts = {}
 	part.found_cached_parts = {}
@@ -654,11 +655,27 @@ function pac.LinkSpecialTrackedPartsForEvent(part, ply)
 		["damage_zone"] = true,
 		["lock"] = true
 	}
+	known_special_link_parts[ply] = known_special_link_parts[ply] or {}
+	known_special_link_parts[ply][part] = part.specialtrackedparts
 	for _,part2 in pairs(all_parts) do
 		if ply == part2:GetPlayerOwner() and tracked_classes[part2.ClassName] then
 			table.insert(part.specialtrackedparts,part2)
 		end
 	end
+	known_special_link_parts[ply][part] = part.specialtrackedparts
+end
+function pac.InsertSpecialTrackedPart(ply, append_part, remove)
+	if append_part then
+		if known_special_link_parts[ply] then
+			for part,tbl in pairs(known_special_link_parts[ply]) do
+				if remove then table.RemoveByValue(part.specialtrackedparts, append_part) continue end
+				if part:IsValid() then
+					table.insert(part.specialtrackedparts, append_part)
+				end
+			end
+		end
+		return
+	end
 end
 
 --a centralized function to cache a part in a prebuilt list so we can access relevant parts already narrowed down instead of searching through all parts / localparts

lua/pac3/core/client/parts/damage_zone.lua

@@ -998,6 +998,8 @@ function PART:OnRemove()
 		pac.RemoveHook(v, "pace_draw_hitbox")
 	end
 	self:ClearHitMarkers()
+	--remove itself
+	pac.InsertSpecialTrackedPart(self:GetPlayerOwner(), self, true)
 end
 
 local previousRenderingHook
@@ -1295,6 +1297,7 @@ function PART:Initialize()
 		end
 	end)
 
+	pac.InsertSpecialTrackedPart(self:GetPlayerOwner(), self)
 end
 
 function PART:SetRadius(val)

lua/pac3/core/client/parts/lock.lua

@@ -478,6 +478,7 @@ function PART:reset_ent_ang()
 end
 
 function PART:OnRemove()
+	pac.InsertSpecialTrackedPart(self:GetPlayerOwner(), self, true)
 end
 
 function PART:DecideTarget()
@@ -595,6 +596,7 @@ function PART:Initialize()
 		end
 	end
 	if not convar_lock:GetBool() then self:SetError("lock parts are disabled on this server!") end
+	pac.InsertSpecialTrackedPart(self:GetPlayerOwner(), self)
 end