Initial commit

Library/PackageCache/com.unity.2d.animation@3.2.6/Editor/SpriteSkin/BoneGizmoToggle.cs

@@ -0,0 +1,58 @@
+using UnityEngine;
+using UnityEngine.U2D.Animation;
+
+namespace UnityEditor.U2D.Animation
+{
+    internal interface IBoneGizmoToggle
+    {
+        bool enableGizmos { get; }
+        void OnGUI();
+    }
+
+    internal class BoneGizmoToggle : IBoneGizmoToggle
+    {
+        private bool m_EnableGizmos;
+        private bool m_CurrentEnableGizmoState;
+
+        public bool enableGizmos
+        {
+            get { return m_CurrentEnableGizmoState; }
+        }
+
+        public BoneGizmoToggle()
+        {
+            SpriteSkin.onDrawGizmos.AddListener(OnDrawGizmos);
+        }
+
+        ~BoneGizmoToggle()
+        {
+            SpriteSkin.onDrawGizmos.RemoveListener(OnDrawGizmos);
+        }
+
+        //This callback will be called before OnSceneGUI in a Repaint event
+        private void OnDrawGizmos()
+        {
+            m_EnableGizmos = true;
+
+            //One time is enough
+            SpriteSkin.onDrawGizmos.RemoveListener(OnDrawGizmos);
+        }
+
+        public void OnGUI()
+        {
+            //Ignore events other than Repaint
+            if (Event.current.type != EventType.Repaint)
+                return;
+
+            if (m_CurrentEnableGizmoState != m_EnableGizmos)
+                SceneView.RepaintAll();
+
+            m_CurrentEnableGizmoState = m_EnableGizmos;
+
+            //Assume the Gizmo toggle is disabled and listen to the event again
+            m_EnableGizmos = false;
+            SpriteSkin.onDrawGizmos.RemoveListener(OnDrawGizmos);
+            SpriteSkin.onDrawGizmos.AddListener(OnDrawGizmos);
+        }
+    }
+}