new-horizons/NewHorizons/Components/TornadoFix.cs

namespace NewHorizons.Components
{
    public class TornadoFix : SectoredMonoBehaviour
    {
        public TornadoController tornadoController;

        public new void Awake()
        {
            base.Awake();
            tornadoController = GetComponent<TornadoController>();
            tornadoController._tornadoRoot.SetActive(false);

            tornadoController._formationDuration = 1f;
            tornadoController._collapseDuration = 1f;
            if (_sector != null)
            {
                _sector.OnOccupantEnterSector += OnOccupantEnterSector;
                _sector.OnOccupantExitSector += OnOccupantExitSector;
            }
        }

        public new void SetSector(Sector sector)
        {
            if (_sector != null)
            {
                _sector.OnOccupantEnterSector -= OnOccupantEnterSector;
                _sector.OnOccupantExitSector -= OnOccupantExitSector;
            }

            base.SetSector(sector);
            _sector.OnOccupantEnterSector += OnOccupantEnterSector;
            _sector.OnOccupantExitSector += OnOccupantExitSector;
        }

        public new void OnDestroy()
        {
            base.OnDestroy();
            _sector.OnOccupantEnterSector -= OnOccupantEnterSector;
            _sector.OnOccupantExitSector -= OnOccupantExitSector;
        }

        public void OnOccupantEnterSector(SectorDetector _)
        {
            // Only form if not active and not forming
            if (tornadoController._tornadoRoot.activeInHierarchy || tornadoController._tornadoForming) return;

            tornadoController.StartFormation();
        }

        public void OnOccupantExitSector(SectorDetector _)
        {

            if (_sector.ContainsAnyOccupants(DynamicOccupant.Player | DynamicOccupant.Probe | DynamicOccupant.Ship)) return;

            // If the root is inactive it has collapsed. Also don't collapse if we're already doing it
            if (!tornadoController._tornadoRoot.activeInHierarchy || tornadoController._tornadoCollapsing) return;

            tornadoController.StartCollapse();
        }
    }
}