TouhouLS/TouhouStation/Assets/RealCode/Menu/GameTransition.cs

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

public enum GameState { Menu, SinglePlayer, Multiplayer, Lobby, SelectingRegion, ConnectionInProgress, InGame, Continue }

public class GameTransition : MonoBehaviour {

  public static GameTransition Instance { get; private set; }

  [SerializeField]
  private GameState _state;
  public GameState state { 
    get => _state;
    set {
      if (_state == value) return;

      _state = value;
      EventSystem.current.SetSelectedGameObject(null);
      var select = rDisplays.FirstOrDefault(d => d.states.Contains(value) && d.firstSelectedGameObject != null);
      if (select != null)
        EventSystem.current.SetSelectedGameObject(select.firstSelectedGameObject);
    }
  }

  public bool inMultiplayer;

  public float lerpSpeed = 10f;

  [System.Serializable]
  private class RectTransformDisplay{
    public RectTransform target = null;
    public GameState[] states = new GameState[] { GameState.Menu };

    [System.NonSerialized]
    public Vector2 basePosition = Vector2.zero;
    public Vector2 selectedPosition = Vector2.zero;
    public GameObject firstSelectedGameObject = null;
  }

  [System.Serializable]
  private class TransformDisplay : TransformDefault{
    public GameState[] states = new GameState[] { GameState.Menu };
    public int playerLoadedCount = 0;

    public TransformDisplay(Transform t, Vector3 p, Vector3 r, Vector3 s) : base(t, p, r, s) { }
  }

  private class TransformDefault{
    public Transform target = null;

    public Vector3 position = Vector3.zero;
    public Vector3 rotation = Vector3.zero;
    public Vector3 scale = Vector3.one;

    public TransformDefault(Transform t, Vector3 p, Vector3 r, Vector3 s){
      target = t;
      position = p;
      rotation = r;
      scale = s;
    }

    public void Set(float lerp){
      target.localPosition = Vector3.Lerp(target.localPosition, position, lerp);
      target.localRotation = Quaternion.Slerp(target.localRotation, Quaternion.Euler(rotation), lerp);
      target.localScale = Vector3.Lerp(target.localScale, scale, lerp);
    }
  }

  private TransformDefault[] tDefault;

  [SerializeField]
  private RectTransformDisplay[] rDisplays = new RectTransformDisplay[0];

  [SerializeField]
  private TransformDisplay[] tDisplays = new TransformDisplay[0];

  public GameState[] selectedOrthoStates;
  public float selectedCameraOrthoSize = 3f;
  private float baseCameraOrthoSize;
  private Camera baseCamera;

  private void Awake() {
    Instance = this;
  }

  private void Start() {
    baseCamera = Camera.main;
    baseCameraOrthoSize = baseCamera.orthographicSize;

    foreach(var r in rDisplays){
      r.basePosition = r.target.anchoredPosition;
    }

    tDefault = tDisplays.Select(t => t.target).Distinct().Select(t => new TransformDefault(t, t.localPosition, t.localEulerAngles, t.localScale)).ToArray();
  }

  private void Update() {

    if (inMultiplayer){
      switch(NetworkManager.net.State){
        case ExitGames.Client.Photon.LoadBalancing.ClientState.Disconnected:
        case ExitGames.Client.Photon.LoadBalancing.ClientState.PeerCreated:
          state = GameState.Menu;
          inMultiplayer = false;
          break;
        case ExitGames.Client.Photon.LoadBalancing.ClientState.ConnectedToNameServer:
          state = GameState.SelectingRegion;
          break;
        case ExitGames.Client.Photon.LoadBalancing.ClientState.JoinedLobby:
          state = GameState.Multiplayer;
          break;
        case ExitGames.Client.Photon.LoadBalancing.ClientState.Joined:
          state = GameState.Lobby;
          break;
      }
    }

    

    // transform 
    var dist = Time.deltaTime * lerpSpeed;
    var playersLoaded = NetworkManager.inRoom ? NetworkManager.net.CurrentRoom.PlayerCount : 2;

    foreach(var d in rDisplays){
      var selected = d.states.Contains(state);

      var item = d.target;
      if (item){
        item.anchoredPosition = Vector3.Lerp(item.anchoredPosition, selected ? d.selectedPosition : d.basePosition, dist);
      }
    }

    var set = new Dictionary<Transform, TransformDefault>();
    foreach(var t in tDefault)
      set.Add(t.target, t);

    foreach(var d in tDisplays){
      var selected = d.states.Contains(state);

      if (selected && d.playerLoadedCount <= playersLoaded){
        d.Set(dist);
        set.Remove(d.target);
      }
    }

    foreach(var d in set){
      d.Value.Set(dist);
    }

    baseCamera.orthographicSize = Mathf.Lerp(baseCamera.orthographicSize, selectedOrthoStates.Contains(state) ? selectedCameraOrthoSize : baseCameraOrthoSize, dist);

  }
}