Add IsEGP and adjust the ValidEGP function

wiremod · Dec 31, 2023 · 2c3fdb1 · 2c3fdb1
1 parent 1a9c6d3
commit 2c3fdb1
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 5 deletions.
diff --git a/lua/entities/gmod_wire_egp/init.lua b/lua/entities/gmod_wire_egp/init.lua
@@ -2,6 +2,7 @@ AddCSLuaFile("cl_init.lua")
 AddCSLuaFile("shared.lua")
 include('shared.lua')
 
+ENT.IsEGP = true
 ENT.WireDebugName = "E2 Graphics Processor"
 
 function ENT:Initialize()

diff --git a/lua/entities/gmod_wire_egp/lib/egplib/usefulfunctions.lua b/lua/entities/gmod_wire_egp/lib/egplib/usefulfunctions.lua
@@ -228,10 +228,13 @@ end
 --------------------------------------------------------
 -- Other
 --------------------------------------------------------
-function EGP:ValidEGP( Ent )
-	if not IsValid( Ent ) then return false end
-	local class = Ent:GetClass()
-	return class == "gmod_wire_egp" or class == "gmod_wire_egp_hud" or class == "gmod_wire_egp_emitter"
+do
+	local GetVar = FindMetaTable( "Entity" ).GetVar
+
+	function EGP:ValidEGP( Ent )
+		if not IsValid( Ent ) then return false end
+		return GetVar( Ent, "IsEGP", false )
+	end
 end
 
 

diff --git a/lua/entities/gmod_wire_egp_emitter.lua b/lua/entities/gmod_wire_egp_emitter.lua
@@ -4,6 +4,7 @@ ENT.PrintName       = "Wire E2 Graphics Processor Emitter"
 ENT.WireDebugName	= "E2 Graphics Processor Emitter"
 ENT.RenderGroup    = RENDERGROUP_BOTH
 
+ENT.IsEGP = true
 ENT.gmod_wire_egp_emitter = true
 
 local DrawOffsetPos = Vector(0, 0, 71)
@@ -174,4 +175,4 @@ if CLIENT then
 	end
 end
 
-function ENT:UpdateTransmitState() return TRANSMIT_ALWAYS end
+function ENT:UpdateTransmitState() return TRANSMIT_ALWAYS end
diff --git a/lua/entities/gmod_wire_egp_hud/init.lua b/lua/entities/gmod_wire_egp_hud/init.lua
@@ -6,6 +6,7 @@ include("huddraw.lua")
 
 DEFINE_BASECLASS("base_wire_entity")
 
+ENT.IsEGP = true
 ENT.WireDebugName = "E2 Graphics Processor HUD"
 
 util.AddNetworkString("EGP_HUD_Use")