1 files changed, 17 insertions, 0 deletions
diff --git a/Assets/Scripts/Unit/Components/UnitRender.cs b/Assets/Scripts/Unit/Components/UnitRender.cs
index b3fd18c8..435c517d 100644
--- a/Assets/Scripts/Unit/Components/UnitRender.cs
+++ b/Assets/Scripts/Unit/Components/UnitRender.cs
@@ -7,6 +7,23 @@ using UnityEngine;
 public class UnitRender : UnitComponent
 {
 
+    public override void Initialize()
+    {
+        base.Initialize();
+    }
 
+    public void SetVisibilityInMainCamera(bool isVisible)
+    {
+        LayerMask mask = LayerMask.GetMask("PlayerCharacter");
+
+        if(!isVisible)
+        {
+            SceneManager.Instance.mainCamera.cullingMask &= ~mask.value;
+        }
+        else
+        {
+            SceneManager.Instance.mainCamera.cullingMask |= mask.value;
+        }
+    }
 
 }