修改水

Packages/com.waveharmonic.crest/Runtime/Scripts/Utility/Shared/Component/CustomBehaviour.cs

@@ -0,0 +1,81 @@
+// Crest Water System
+// Copyright © 2024 Wave Harmonic. All rights reserved.
+
+#if UNITY_EDITOR
+using UnityEditor;
+using UnityEngine;
+using MonoBehaviour = WaveHarmonic.Crest.Internal.EditorBehaviour;
+#else
+using MonoBehaviour = UnityEngine.MonoBehaviour;
+#endif
+
+namespace WaveHarmonic.Crest.Internal
+{
+    /// <summary>
+    /// Implements logic to smooth out Unity's wrinkles.
+    /// </summary>
+    public abstract class CustomBehaviour : MonoBehaviour
+    {
+        bool _AfterStart;
+
+#pragma warning disable 114
+        private protected virtual void Awake()
+        {
+#if UNITY_EDITOR
+            base.Awake();
+#endif
+        }
+
+        /// <summary>
+        /// Unity's Start method. Make sure to call base if overriden.
+        /// </summary>
+        protected void Start()
+        {
+            _AfterStart = true;
+
+#if UNITY_EDITOR
+            base.Start();
+            if (!enabled) return;
+#endif
+
+            OnStart();
+        }
+#pragma warning restore 114
+
+        /// <summary>
+        /// Called in OnEnable only after Start has ran.
+        /// </summary>
+        private protected virtual void Initialize()
+        {
+
+        }
+
+        /// <summary>
+        /// Replaces Start. Only called in the editor if passes validation.
+        /// </summary>
+        private protected virtual void OnStart()
+        {
+            Initialize();
+        }
+
+        /// <summary>
+        /// Unity's OnEnable method. Make sure to call base if overriden.
+        /// </summary>
+        private protected virtual void OnEnable()
+        {
+            if (!_AfterStart) return;
+            Initialize();
+        }
+
+#if UNITY_EDITOR
+        [InitializeOnEnterPlayMode]
+        static void OnEnterPlayModeInEditor(EnterPlayModeOptions options)
+        {
+            foreach (var @object in FindObjectsByType<CustomBehaviour>(FindObjectsInactive.Include, FindObjectsSortMode.None))
+            {
+                @object._AfterStart = false;
+            }
+        }
+#endif
+    }
+}