Use OnEnabled event to initialise UnderwaterEnvironmentalLighting

wave-harmonic · Aug 24, 2020 · 41fea27 · 41fea27
1 parent 0254195
commit 41fea27
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/crest/Assets/Crest/Crest-Examples/Shared/Scripts/UnderwaterEnvironmentalLighting.cs b/crest/Assets/Crest/Crest-Examples/Shared/Scripts/UnderwaterEnvironmentalLighting.cs
@@ -27,8 +27,11 @@ public class UnderwaterEnvironmentalLighting : MonoBehaviour
 
         public const float DEPTH_OUTSCATTER_CONSTANT = 0.25f;
 
-        void Start()
+        void OnEnable()
         {
+            OceanRenderer.OnOceanRendererEnabled -= OnOceanRendererEnabled;
+            OceanRenderer.OnOceanRendererEnabled += OnOceanRendererEnabled;
+
             if (OceanRenderer.Instance == null)
             {
                 enabled = false;
@@ -69,6 +72,18 @@ void OnDisable()
             RenderSettings.fogDensity = _fogDensity;
         }
 
+        private void OnDestroy()
+        {
+            // We need this event registered even when this component is disabled. So we unregister only when the
+            // component is destroyed.
+            OceanRenderer.OnOceanRendererEnabled -= OnOceanRendererEnabled;
+        }
+
+        void OnOceanRendererEnabled()
+        {
+            enabled = true;
+        }
+
         void LateUpdate()
         {
             if (OceanRenderer.Instance == null)