55 lines
1.9 KiB
C#
55 lines
1.9 KiB
C#
using System.Collections;
|
|
using UnityEngine;
|
|
using UnityEngine.UI;
|
|
|
|
namespace Michsky.UI.Heat
|
|
{
|
|
[ExecuteInEditMode]
|
|
[DisallowMultipleComponent]
|
|
[AddComponentMenu("Heat UI/Layout/Layout Group Fix")]
|
|
public class LayoutGroupFix : MonoBehaviour
|
|
{
|
|
[SerializeField] private bool fixOnEnable = true;
|
|
[SerializeField] private bool fixWithDelay = true;
|
|
[SerializeField] private RebuildMethod rebuildMethod;
|
|
float fixDelay = 0.025f;
|
|
|
|
public enum RebuildMethod { ForceRebuild, MarkRebuild }
|
|
|
|
void OnEnable()
|
|
{
|
|
#if UNITY_EDITOR
|
|
LayoutRebuilder.ForceRebuildLayoutImmediate(GetComponent<RectTransform>());
|
|
if (Application.isPlaying == false) { return; }
|
|
#endif
|
|
if (fixWithDelay == false && fixOnEnable == true && rebuildMethod == RebuildMethod.ForceRebuild) { ForceRebuild(); }
|
|
else if (fixWithDelay == false && fixOnEnable == true && rebuildMethod == RebuildMethod.MarkRebuild) { MarkRebuild(); }
|
|
else if (fixWithDelay == true) { StartCoroutine(FixDelay()); }
|
|
}
|
|
|
|
public void FixLayout()
|
|
{
|
|
if (fixWithDelay == false && rebuildMethod == RebuildMethod.ForceRebuild) { ForceRebuild(); }
|
|
else if (fixWithDelay == false && rebuildMethod == RebuildMethod.MarkRebuild) { MarkRebuild(); }
|
|
else { StartCoroutine(FixDelay()); }
|
|
}
|
|
|
|
void ForceRebuild()
|
|
{
|
|
LayoutRebuilder.ForceRebuildLayoutImmediate(GetComponent<RectTransform>());
|
|
}
|
|
|
|
void MarkRebuild()
|
|
{
|
|
LayoutRebuilder.MarkLayoutForRebuild(GetComponent<RectTransform>());
|
|
}
|
|
|
|
IEnumerator FixDelay()
|
|
{
|
|
yield return new WaitForSecondsRealtime(fixDelay);
|
|
|
|
if (rebuildMethod == RebuildMethod.ForceRebuild) { ForceRebuild(); }
|
|
else if (rebuildMethod == RebuildMethod.MarkRebuild) { MarkRebuild(); }
|
|
}
|
|
}
|
|
} |