Disabled DrawGizmosEditor when objects are hidden in het SceneVisibil…

…ityManager
jeffreylanters · May 9, 2023 · 6aff5f4 · 6aff5f4
1 parent 7d1690e
commit 6aff5f4
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/Runtime/Controller.cs b/Runtime/Controller.cs
@@ -1,6 +1,9 @@
 using System.Collections.Generic;
 using System.Linq;
 using UnityEngine;
+#if UNITY_EDITOR
+using UnityEditor;
+#endif
 
 namespace ElRaccoone.EntityComponentSystem {
   /// <summary>
@@ -134,7 +137,13 @@ void OnDrawGizmos () {
         // interface.
         var monoBehaviours = FindObjectsOfType<MonoBehaviour> ();
         var entities = monoBehaviours.OfType<IEntityComponent> ();
-        OnDrawEditorGizmos (entities.ToArray ());
+        var visibleEntities = entities.Where (entity => {
+          var monoBehaviour = entity as MonoBehaviour;
+          return monoBehaviour.enabled
+            && monoBehaviour.gameObject.activeInHierarchy
+            && !SceneVisibilityManager.instance.IsHidden (monoBehaviour.gameObject);
+        });
+        OnDrawEditorGizmos (visibleEntities.ToArray ());
         return;
       }
       // Invoking draw gizmos method on each system and service.