CircleSelectManager.cs 5.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125
  1. /*
  2. Copyright (c) 2020 Omar Duarte
  3. Unauthorized copying of this file, via any medium is strictly prohibited.
  4. Writen by Omar Duarte, 2020.
  5. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  6. IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  7. FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  8. AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  9. LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  10. OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  11. THE SOFTWARE.
  12. */
  13. using UnityEngine;
  14. using System.Linq;
  15. namespace PluginMaster
  16. {
  17. #region DATA & SETTIGNS
  18. [System.Serializable]
  19. public class CircleSelectSettings : CircleToolBase, ISelectionBrushTool
  20. {
  21. [SerializeField] private ModifierToolSettings _modifierTool = new ModifierToolSettings();
  22. public CircleSelectSettings() => _modifierTool.OnDataChanged += DataChanged;
  23. public ModifierToolSettings.Command command { get => _modifierTool.command; set => _modifierTool.command = value; }
  24. public bool onlyTheClosest
  25. {
  26. get => _modifierTool.onlyTheClosest;
  27. set => _modifierTool.onlyTheClosest = value;
  28. }
  29. public bool outermostPrefabFilter
  30. {
  31. get => _modifierTool.outermostPrefabFilter;
  32. set => _modifierTool.outermostPrefabFilter = value;
  33. }
  34. public override void Copy(IToolSettings other)
  35. {
  36. var otherCircleSelectSettings = other as CircleSelectSettings;
  37. if (otherCircleSelectSettings == null) return;
  38. base.Copy(other);
  39. _modifierTool.Copy(otherCircleSelectSettings);
  40. }
  41. }
  42. [System.Serializable]
  43. public class CircleSelectManager : ToolManagerBase<CircleSelectSettings> { }
  44. #endregion
  45. #region PWBIO
  46. public static partial class PWBIO
  47. {
  48. private static Material _transparentBlueMaterial = null;
  49. public static Material transparentBlueMaterial
  50. {
  51. get
  52. {
  53. if (_transparentBlueMaterial == null)
  54. _transparentBlueMaterial = new Material(Shader.Find("PluginMaster/TransparentBlue"));
  55. return _transparentBlueMaterial;
  56. }
  57. }
  58. private static System.Collections.Generic.HashSet<GameObject> _toSelect
  59. = new System.Collections.Generic.HashSet<GameObject>();
  60. private static void CircleSelectDuringSceneGUI(UnityEditor.SceneView sceneView)
  61. {
  62. CircleSelectMouseEvents();
  63. var mousePos = Event.current.mousePosition;
  64. var mouseRay = UnityEditor.HandleUtility.GUIPointToWorldRay(mousePos);
  65. var center = mouseRay.GetPoint(_lastHitDistance);
  66. if (MouseRaycast(mouseRay, out RaycastHit mouseHit, out GameObject collider,
  67. float.MaxValue, -1, paintOnPalettePrefabs: true, castOnMeshesWithoutCollider: true))
  68. {
  69. _lastHitDistance = mouseHit.distance;
  70. center = mouseHit.point;
  71. PWBCore.UpdateTempCollidersIfHierarchyChanged();
  72. }
  73. DrawCircleTool(center, sceneView.camera, new Color(0.455f, 0.596f, 0.8f, 1f), CircleSelectManager.settings.radius);
  74. GetCircleToolTargets(mouseRay, sceneView.camera, CircleSelectManager.settings,
  75. CircleSelectManager.settings.radius, _toSelect);
  76. DrawObjectsToSelect(sceneView.camera);
  77. }
  78. private static void DrawObjectsToSelect(Camera camera)
  79. {
  80. foreach (var obj in _toSelect)
  81. {
  82. var filters = obj.GetComponentsInChildren<MeshFilter>();
  83. foreach (var filter in filters)
  84. {
  85. var mesh = filter.sharedMesh;
  86. if (mesh == null) continue;
  87. for (int subMeshIdx = 0; subMeshIdx < mesh.subMeshCount; ++subMeshIdx)
  88. Graphics.DrawMesh(mesh, filter.transform.localToWorldMatrix,
  89. transparentBlueMaterial, 0, camera, subMeshIdx);
  90. }
  91. }
  92. }
  93. private static void CircleSelectMouseEvents()
  94. {
  95. if (Event.current.button == 0 && !Event.current.alt
  96. && (Event.current.type == EventType.MouseDown
  97. || (Event.current.type == EventType.MouseDrag && !Event.current.control)))
  98. {
  99. var selectedObjects = new System.Collections.Generic.HashSet<Object>();
  100. if (Event.current.shift || Event.current.control)
  101. {
  102. selectedObjects.UnionWith(UnityEditor.Selection.objects);
  103. if (Event.current.control)
  104. {
  105. selectedObjects.ExceptWith(_toSelect);
  106. var selectedGameObjects = UnityEditor.Selection.objects.Select(o => o as GameObject);
  107. _toSelect.ExceptWith(selectedGameObjects);
  108. }
  109. }
  110. selectedObjects.UnionWith(_toSelect);
  111. UnityEditor.Selection.objects = selectedObjects.ToArray();
  112. Event.current.Use();
  113. }
  114. }
  115. }
  116. #endregion
  117. }