From 0450cc7ef6b877c93448a63128b86b0c94d9e5ae Mon Sep 17 00:00:00 2001
From: metalgearsloth <comedian_vs_clown@hotmail.com>
Date: Mon, 16 Dec 2024 16:11:04 +1100
Subject: [PATCH] LightOnCollide entityquery

---
 .../Light/EntitySystems/LightCollideSystem.cs       | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/Content.Shared/Light/EntitySystems/LightCollideSystem.cs b/Content.Shared/Light/EntitySystems/LightCollideSystem.cs
index f09ae6824ea8..2de7c5591fdb 100644
--- a/Content.Shared/Light/EntitySystems/LightCollideSystem.cs
+++ b/Content.Shared/Light/EntitySystems/LightCollideSystem.cs
@@ -9,9 +9,14 @@ public sealed class LightCollideSystem : EntitySystem
     [Dependency] private readonly SharedPhysicsSystem _physics = default!;
     [Dependency] private readonly SlimPoweredLightSystem _lights = default!;
 
+    private EntityQuery<LightOnCollideComponent> _lightQuery;
+
     public override void Initialize()
     {
         base.Initialize();
+
+        _lightQuery = GetEntityQuery<LightOnCollideComponent>();
+
         SubscribeLocalEvent<LightOnCollideColliderComponent, PreventCollideEvent>(OnPreventCollide);
         SubscribeLocalEvent<LightOnCollideColliderComponent, StartCollideEvent>(OnStart);
         SubscribeLocalEvent<LightOnCollideColliderComponent, EndCollideEvent>(OnEnd);
@@ -35,7 +40,7 @@ private void OnCollideShutdown(Entity<LightOnCollideColliderComponent> ent, ref
 
             var other = contact.OtherEnt(ent.Owner);
 
-            if (HasComp<LightOnCollideComponent>(other))
+            if (_lightQuery.HasComp(other))
             {
                 _physics.RegenerateContacts(other);
             }
@@ -46,7 +51,7 @@ private void OnCollideShutdown(Entity<LightOnCollideColliderComponent> ent, ref
     // At the moment there's no easy way to do collision whitelists based on components.
     private void OnPreventCollide(Entity<LightOnCollideColliderComponent> ent, ref PreventCollideEvent args)
     {
-        if (!HasComp<LightOnCollideComponent>(args.OtherEntity))
+        if (!_lightQuery.HasComp(args.OtherEntity))
         {
             args.Cancelled = true;
         }
@@ -57,7 +62,7 @@ private void OnEnd(Entity<LightOnCollideColliderComponent> ent, ref EndCollideEv
         if (args.OurFixtureId != ent.Comp.FixtureId)
             return;
 
-        if (!HasComp<LightOnCollideComponent>(args.OtherEntity))
+        if (!_lightQuery.HasComp(args.OtherEntity))
             return;
 
         // TODO: Engine bug IsTouching box2d yay.
@@ -74,7 +79,7 @@ private void OnStart(Entity<LightOnCollideColliderComponent> ent, ref StartColli
         if (args.OurFixtureId != ent.Comp.FixtureId)
             return;
 
-        if (!HasComp<LightOnCollideComponent>(args.OtherEntity))
+        if (!_lightQuery.HasComp(args.OtherEntity))
             return;
 
         _lights.SetEnabled(args.OtherEntity, true);