ObjectSelectionGizmos.cs 8.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232
  1. #if UNITY_EDITOR
  2. using UnityEngine;
  3. using UnityEditor;
  4. using System;
  5. using System.Linq;
  6. using System.Collections.Generic;
  7. namespace O3DWB
  8. {
  9. [Serializable]
  10. public class ObjectSelectionGizmos: ScriptableObject
  11. {
  12. #region Private Variables
  13. [SerializeField]
  14. private bool _areGizmosActive = true;
  15. [SerializeField]
  16. private GizmoType _activeGizmoType = GizmoType.Move;
  17. [SerializeField]
  18. private TransformSpace _gizmoTransformSpace = TransformSpace.Global;
  19. [SerializeField]
  20. private TransformGizmoPivotPoint _gizmoTransformPivotPoint = TransformGizmoPivotPoint.Center;
  21. [SerializeField]
  22. private ObjectMoveGizmo _objectMoveGizmo;
  23. [SerializeField]
  24. private ObjectRotationGizmo _objectRotationGizmo;
  25. [SerializeField]
  26. private ObjectScaleGizmo _objectScaleGizmo;
  27. [SerializeField]
  28. private ObjectSelectionExtrudeGizmo _objectDuplicateGizmo;
  29. #endregion
  30. #region Public Properties
  31. public bool AreGizmosActive
  32. {
  33. get { return _areGizmosActive; }
  34. set
  35. {
  36. _areGizmosActive = value;
  37. SceneView.RepaintAll();
  38. }
  39. }
  40. public GizmoType ActiveGizmoType
  41. {
  42. get { return _activeGizmoType; }
  43. set
  44. {
  45. _activeGizmoType = value;
  46. AdjustActiveGizmoRotation();
  47. AdjustActiveGizmoPosition();
  48. SceneView.RepaintAll();
  49. }
  50. }
  51. public TransformSpace GizmoTransformSpace
  52. {
  53. get { return _gizmoTransformSpace; }
  54. set
  55. {
  56. _gizmoTransformSpace = value;
  57. AdjustActiveGizmoRotation();
  58. SceneView.RepaintAll();
  59. }
  60. }
  61. public TransformGizmoPivotPoint GizmoTransformPivotPoint
  62. {
  63. get { return _gizmoTransformPivotPoint; }
  64. set
  65. {
  66. _gizmoTransformPivotPoint = value;
  67. AdjustActiveGizmoPosition();
  68. SceneView.RepaintAll();
  69. }
  70. }
  71. public ObjectMoveGizmo ObjectMoveGizmo
  72. {
  73. get
  74. {
  75. if (_objectMoveGizmo == null) _objectMoveGizmo = Octave3DWorldBuilder.ActiveInstance.CreateScriptableObject<ObjectMoveGizmo>();
  76. return _objectMoveGizmo;
  77. }
  78. }
  79. public ObjectRotationGizmo ObjectRotationGizmo
  80. {
  81. get
  82. {
  83. if (_objectRotationGizmo == null) _objectRotationGizmo = Octave3DWorldBuilder.ActiveInstance.CreateScriptableObject<ObjectRotationGizmo>();
  84. return _objectRotationGizmo;
  85. }
  86. }
  87. public ObjectScaleGizmo ObjectScaleGizmo
  88. {
  89. get
  90. {
  91. if (_objectScaleGizmo == null) _objectScaleGizmo = Octave3DWorldBuilder.ActiveInstance.CreateScriptableObject<ObjectScaleGizmo>();
  92. return _objectScaleGizmo;
  93. }
  94. }
  95. public ObjectSelectionExtrudeGizmo ObjectDuplicateGizmo
  96. {
  97. get
  98. {
  99. if (_objectDuplicateGizmo == null) _objectDuplicateGizmo = Octave3DWorldBuilder.ActiveInstance.CreateScriptableObject<ObjectSelectionExtrudeGizmo>();
  100. return _objectDuplicateGizmo;
  101. }
  102. }
  103. #endregion
  104. #region Public Methods
  105. public bool OwnsGizmo(ObjectGizmo objectTransformGizmo)
  106. {
  107. if (objectTransformGizmo == ObjectMoveGizmo) return true;
  108. if (objectTransformGizmo == ObjectRotationGizmo) return true;
  109. if (objectTransformGizmo == ObjectScaleGizmo) return true;
  110. if (objectTransformGizmo == ObjectDuplicateGizmo) return true;
  111. return false;
  112. }
  113. public void RenderHandles(HashSet<GameObject> selectedGameObjects)
  114. {
  115. if (selectedGameObjects == null || selectedGameObjects.Count == 0 || !_areGizmosActive) return;
  116. ObjectGizmo activeGizmo = GetActiveGizmo();
  117. if (activeGizmo != null)
  118. {
  119. activeGizmo.TargetObjects = selectedGameObjects;
  120. activeGizmo.RenderHandles(_gizmoTransformPivotPoint);
  121. }
  122. }
  123. public void OnObjectSelectionUpdatedUsingMouseClick()
  124. {
  125. ObjectGizmo activeGizmo = GetActiveGizmo();
  126. if (activeGizmo != null)
  127. {
  128. /*GameObject lastSelectedGameObject = ObjectSelection.Get().GetLastSelectedGameObject();
  129. if (lastSelectedGameObject == null) return;
  130. Transform lastSelectedObjectTransform = lastSelectedGameObject.transform;
  131. if (_gizmoTransformPivotPoint == TransformGizmoPivotPoint.Pivot && lastSelectedGameObject != null
  132. && _activeGizmoType != GizmoType.Duplicate) activeGizmo.WorldPosition = lastSelectedObjectTransform.position;
  133. else activeGizmo.WorldPosition = ObjectSelection.Get().GetWorldCenter();
  134. if (_gizmoTransformSpace == TransformSpace.Local && lastSelectedGameObject != null
  135. && _activeGizmoType != GizmoType.Duplicate) activeGizmo.WorldRotation = lastSelectedObjectTransform.rotation;
  136. else activeGizmo.WorldRotation = Quaternion.identity;*/
  137. if (ObjectSelection.Get().NumberOfSelectedObjects != 0)
  138. {
  139. AdjustActiveGizmoPosition();
  140. AdjustActiveGizmoRotation();
  141. }
  142. }
  143. }
  144. public void OnObjectSelectionUpdatedUsingSelectionShape()
  145. {
  146. ObjectGizmo activeGizmo = GetActiveGizmo();
  147. if (activeGizmo != null)
  148. {
  149. /* GameObject firstSelectedGameObject = ObjectSelection.Get().GetFirstSelectedGameObject();
  150. if (firstSelectedGameObject == null) return;
  151. Transform firstSelectedObjectTransform = firstSelectedGameObject.transform;
  152. if (_gizmoTransformPivotPoint == TransformGizmoPivotPoint.Pivot && firstSelectedGameObject != null
  153. && _activeGizmoType != GizmoType.Duplicate) activeGizmo.WorldPosition = firstSelectedObjectTransform.position;
  154. else activeGizmo.WorldPosition = ObjectSelection.Get().GetWorldCenter();
  155. if (_gizmoTransformSpace == TransformSpace.Local && firstSelectedObjectTransform != null
  156. && _activeGizmoType != GizmoType.Duplicate) activeGizmo.WorldRotation = firstSelectedObjectTransform.rotation;
  157. else activeGizmo.WorldRotation = Quaternion.identity;*/
  158. if (ObjectSelection.Get().NumberOfSelectedObjects != 0)
  159. {
  160. AdjustActiveGizmoPosition();
  161. AdjustActiveGizmoRotation();
  162. }
  163. }
  164. }
  165. public void OnObjectSelectionUpdated()
  166. {
  167. OnObjectSelectionUpdatedUsingSelectionShape();
  168. }
  169. #endregion
  170. #region Private Methods
  171. private ObjectGizmo GetActiveGizmo()
  172. {
  173. if (_activeGizmoType == GizmoType.Move) return ObjectMoveGizmo;
  174. if (_activeGizmoType == GizmoType.Rotate) return ObjectRotationGizmo;
  175. if (_activeGizmoType == GizmoType.Scale) return ObjectScaleGizmo;
  176. if (_activeGizmoType == GizmoType.Duplicate) return ObjectDuplicateGizmo;
  177. return null;
  178. }
  179. private void AdjustActiveGizmoPosition()
  180. {
  181. ObjectGizmo activeGizmo = GetActiveGizmo();
  182. if (activeGizmo != null)
  183. {
  184. if (_gizmoTransformPivotPoint == TransformGizmoPivotPoint.Pivot && _activeGizmoType != GizmoType.Duplicate)
  185. {
  186. GameObject firstSelectedObject = ObjectSelection.Get().GetFirstSelectedGameObject();
  187. if (firstSelectedObject != null) activeGizmo.WorldPosition = firstSelectedObject.transform.position;
  188. }
  189. else activeGizmo.WorldPosition = ObjectSelection.Get().GetWorldCenter();
  190. }
  191. }
  192. private void AdjustActiveGizmoRotation()
  193. {
  194. ObjectGizmo activeGizmo = GetActiveGizmo();
  195. if (activeGizmo != null)
  196. {
  197. if ((_gizmoTransformSpace == TransformSpace.Local && _activeGizmoType != GizmoType.Duplicate) || _activeGizmoType == GizmoType.Scale)
  198. {
  199. GameObject firstSelectedObject = ObjectSelection.Get().GetFirstSelectedGameObject();
  200. if (firstSelectedObject != null) activeGizmo.WorldRotation = firstSelectedObject.transform.rotation;
  201. else activeGizmo.WorldRotation = Quaternion.identity;
  202. }
  203. else activeGizmo.WorldRotation = Quaternion.identity;
  204. }
  205. }
  206. #endregion
  207. }
  208. }
  209. #endif