HeavenStudio/Assets/Scripts/LevelEditor/Selections.cs

using System.Collections;
using System.Collections.Generic;
using System.Linq;
using UnityEngine;

namespace RhythmHeavenMania.Editor
{
    public class Selections : MonoBehaviour
    {
        public List<TimelineEventObj> eventsSelected = new List<TimelineEventObj>();

        public static Selections instance { get; private set; }

        private void Awake()
        {
            instance = this;
        }

        private void Update()
        {
            var buggedSelections = eventsSelected.FindAll(c => c == null);
            if (buggedSelections.Count > 0)
            {
                for (int i = 0; i < buggedSelections.Count; i++)
                Deselect(buggedSelections[i]);
            }
        }

        public void ClickSelect(TimelineEventObj eventToAdd)
        {
            DeselectAll();
            eventsSelected.Add(eventToAdd);

            // CommandManager.instance.Execute(new Commands.Selection(new List<TimelineEventObj>() { eventToAdd } ));
        }

        public void ShiftClickSelect(TimelineEventObj eventToAdd)
        {
            if (!eventsSelected.Contains(eventToAdd))
            {
                eventsSelected.Add(eventToAdd);
            }
            else
            {
                eventsSelected.Remove(eventToAdd);
            }
        }

        public void DragSelect(TimelineEventObj eventToAdd)
        {
            if (!eventsSelected.Contains(eventToAdd))
            {
                eventsSelected.Add(eventToAdd);
            }
        }

        public void DeselectAll()
        {
            eventsSelected.Clear();
        }

        public void Deselect(TimelineEventObj eventToDeselect)
        {
            if (eventsSelected.Contains(eventToDeselect))
            {
                eventsSelected.Remove(eventToDeselect);
            }
        }
    }
}