mirror of
https://github.com/wiremod/wire.git
synced 2025-03-04 03:03:04 -05:00
e1b30fb964
* Prevent an error on disabling cd lock * Fix styling issue * Add self isvalid check too --------- Co-authored-by: thegrb93 <grbrown93@sbcglobal.net>
125 lines
3.3 KiB
Lua
125 lines
3.3 KiB
Lua
AddCSLuaFile()
|
|
DEFINE_BASECLASS( "base_wire_entity" )
|
|
ENT.PrintName = "Wire CD Lock"
|
|
ENT.WireDebugName = "CD Lock"
|
|
|
|
if CLIENT then return end -- No more client
|
|
|
|
--Time after losing one disk to search for another
|
|
local NEW_DISK_WAIT_TIME = 2
|
|
local DISK_IN_SOCKET_CONSTRAINT_POWER = 5000
|
|
local DISK_IN_ATTACH_RANGE = 16
|
|
|
|
function ENT:Initialize()
|
|
self:PhysicsInit(SOLID_VPHYSICS)
|
|
self:SetMoveType(MOVETYPE_VPHYSICS)
|
|
self:SetSolid(SOLID_VPHYSICS)
|
|
|
|
self.Const = nil
|
|
self.Disk = nil
|
|
self.DisableLinking = 0
|
|
|
|
self.Inputs = WireLib.CreateSpecialInputs(self, { "Disable" }, { "NORMAL" })
|
|
self.Outputs = WireLib.CreateSpecialOutputs(self, { "Locked", "DiskEntity" }, { "NORMAL", "ENTITY" })
|
|
|
|
self:NextThink(CurTime() + 0.25)
|
|
end
|
|
|
|
function ENT:TriggerInput(iname, value)
|
|
if (iname == "Disable") then
|
|
self.DisableLinking = value
|
|
if (value >= 1) and (self.Const) then
|
|
self.Const:Remove()
|
|
--self.NoCollideConst:Remove()
|
|
|
|
self.Const = nil
|
|
self.Disk.Lock = nil
|
|
self.Disk = nil
|
|
--self.NoCollideConst = nil
|
|
|
|
WireLib.TriggerOutput(self, "Locked", 0)
|
|
WireLib.TriggerOutput(self, "DiskEntity", nil)
|
|
self:NextThink(CurTime() + NEW_DISK_WAIT_TIME)
|
|
end
|
|
end
|
|
end
|
|
|
|
function ENT:Think()
|
|
BaseClass.Think(self)
|
|
if self.DoNextThink then
|
|
self:NextThink( self.DoNextThink )
|
|
self.DoNextThink = nil
|
|
return true
|
|
end
|
|
|
|
if not IsValid(self.Disk) and self.DisableLinking < 1 then -- if we're not linked
|
|
-- Find entities near us
|
|
local lockCenter = self:LocalToWorld(Vector(0, 0, 0))
|
|
local local_ents = ents.FindInSphere(lockCenter, DISK_IN_ATTACH_RANGE)
|
|
for key, disk in pairs(local_ents) do
|
|
-- If we find a disk, try to attach it to us
|
|
if (disk:IsValid() and disk:GetClass() == "gmod_wire_cd_disk") then
|
|
if (disk.Lock == nil) then
|
|
self:AttachDisk(disk)
|
|
end
|
|
end
|
|
end
|
|
else
|
|
self:NextThink(CurTime() + 1)
|
|
return true
|
|
end
|
|
end
|
|
|
|
function ENT:AttachDisk(disk)
|
|
--Position disk
|
|
|
|
local newpos = self:LocalToWorld(Vector(0, 0, 0))
|
|
local lockAng = self:GetAngles()
|
|
disk:SetPos(newpos)
|
|
disk:SetAngles(lockAng)
|
|
|
|
self.NoCollideConst = constraint.NoCollide(self, disk, 0, 0)
|
|
if (not self.NoCollideConst) then
|
|
WireLib.TriggerOutput(self, "Locked", 0)
|
|
WireLib.TriggerOutput(self, "DiskEntity", nil)
|
|
return
|
|
end
|
|
|
|
--Constrain together
|
|
self.Const = constraint.Weld(self, disk, 0, 0, DISK_IN_SOCKET_CONSTRAINT_POWER, true)
|
|
if (not self.Const) then
|
|
self.NoCollideConst:Remove()
|
|
self.NoCollideConst = nil
|
|
WireLib.TriggerOutput(self, "Locked", 0)
|
|
WireLib.TriggerOutput(self, "DiskEntity", nil)
|
|
return
|
|
end
|
|
|
|
self.Const:CallOnRemove("wire_cd_remove_on_weld",function()
|
|
if not self:IsValid() then return end
|
|
self.Const = nil
|
|
if IsValid(self.Disk) then
|
|
self.Disk.Lock = nil
|
|
end
|
|
self.Disk = nil
|
|
self.NoCollideConst = nil
|
|
|
|
WireLib.TriggerOutput(self, "Locked", 0)
|
|
WireLib.TriggerOutput(self, "DiskEntity", nil)
|
|
|
|
self.DoNextThink = CurTime() + NEW_DISK_WAIT_TIME --Give time before next grabbing a disk.
|
|
end)
|
|
|
|
--Prepare clearup incase one is removed
|
|
disk:DeleteOnRemove(self.Const)
|
|
self:DeleteOnRemove(self.Const)
|
|
self.Const:DeleteOnRemove(self.NoCollideConst)
|
|
|
|
disk.Lock = self
|
|
self.Disk = disk
|
|
WireLib.TriggerOutput(self, "Locked", 1)
|
|
WireLib.TriggerOutput(self, "DiskEntity", disk)
|
|
end
|
|
|
|
duplicator.RegisterEntityClass("gmod_wire_cd_lock", WireLib.MakeWireEnt, "Data")
|