GraphicConnector.cs 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106
  1. using System;
  2. using System.Collections.Generic;
  3. using UnityEngine;
  4. using UnityEngine.UI;
  5. namespace Coffee.UISoftMask
  6. {
  7. internal static class GraphicConnectorExtension
  8. {
  9. public static void SetVerticesDirtyEx(this Graphic graphic)
  10. {
  11. GraphicConnector.FindConnector(graphic).SetVerticesDirty(graphic);
  12. }
  13. public static void SetMaterialDirtyEx(this Graphic graphic)
  14. {
  15. GraphicConnector.FindConnector(graphic).SetMaterialDirty(graphic);
  16. }
  17. public static T GetComponentInParentEx<T>(this Component component, bool includeInactive = false) where T : MonoBehaviour
  18. {
  19. if (!component) return null;
  20. var trans = component.transform;
  21. while (trans)
  22. {
  23. var c = trans.GetComponent<T>();
  24. if (c && (includeInactive || c.isActiveAndEnabled)) return c;
  25. trans = trans.parent;
  26. }
  27. return null;
  28. }
  29. }
  30. public class GraphicConnector
  31. {
  32. private static readonly List<GraphicConnector> s_Connectors = new List<GraphicConnector>();
  33. private static readonly Dictionary<Type, GraphicConnector> s_ConnectorMap = new Dictionary<Type, GraphicConnector>();
  34. private static readonly GraphicConnector s_EmptyConnector = new GraphicConnector();
  35. #if UNITY_EDITOR
  36. [UnityEditor.InitializeOnLoadMethod]
  37. #endif
  38. [RuntimeInitializeOnLoadMethod(RuntimeInitializeLoadType.BeforeSceneLoad)]
  39. private static void Init()
  40. {
  41. AddConnector(new GraphicConnector());
  42. }
  43. protected static void AddConnector(GraphicConnector connector)
  44. {
  45. s_Connectors.Add(connector);
  46. s_Connectors.Sort((x, y) => y.priority - x.priority);
  47. }
  48. public static GraphicConnector FindConnector(Graphic graphic)
  49. {
  50. if (!graphic) return s_EmptyConnector;
  51. var type = graphic.GetType();
  52. GraphicConnector connector = null;
  53. if (s_ConnectorMap.TryGetValue(type, out connector)) return connector;
  54. foreach (var c in s_Connectors)
  55. {
  56. if (!c.IsValid(graphic)) continue;
  57. s_ConnectorMap.Add(type, c);
  58. return c;
  59. }
  60. return s_EmptyConnector;
  61. }
  62. /// <summary>
  63. /// Connector priority.
  64. /// </summary>
  65. protected virtual int priority
  66. {
  67. get { return -1; }
  68. }
  69. /// <summary>
  70. /// The connector is valid for the component.
  71. /// </summary>
  72. protected virtual bool IsValid(Graphic graphic)
  73. {
  74. return true;
  75. }
  76. public virtual void SetVerticesDirty(Graphic graphic)
  77. {
  78. if (graphic)
  79. graphic.SetVerticesDirty();
  80. }
  81. public virtual void SetMaterialDirty(Graphic graphic)
  82. {
  83. if (graphic)
  84. graphic.SetMaterialDirty();
  85. }
  86. }
  87. }