ObjectGroupDatabase.cs 7.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204
  1. #if UNITY_EDITOR
  2. using UnityEngine;
  3. using System;
  4. using System.Collections.Generic;
  5. namespace O3DWB
  6. {
  7. [Serializable]
  8. public class ObjectGroupDatabase : ScriptableObject
  9. {
  10. #region Private Variables
  11. [SerializeField]
  12. private ObjectGroupCollection _objectGroups = new ObjectGroupCollection();
  13. [SerializeField]
  14. private bool _preserveGroupChildren = true;
  15. [SerializeField]
  16. private ObjectGroupDatabaseView _view;
  17. #endregion
  18. #region Public Properties
  19. public bool IsEmpty { get { return _objectGroups.IsEmpty; } }
  20. public int NumberOfGroups { get { return _objectGroups.NumberOfEntities; } }
  21. public ObjectGroup ActiveGroup { get { return _objectGroups.MarkedEntity; } }
  22. public int IndexOfActiveGroup { get { return _objectGroups.IndexOfMarkedEntity; } }
  23. public bool PreserveGroupChildren { get { return _preserveGroupChildren; } set { _preserveGroupChildren = value; } }
  24. public ObjectGroupDatabaseView View { get { return _view; } }
  25. #endregion
  26. public static ObjectGroupDatabase Get()
  27. {
  28. return Octave3DWorldBuilder.ActiveInstance.PlacementObjectGroupDatabase;
  29. }
  30. #region Constructors
  31. public ObjectGroupDatabase()
  32. {
  33. _view = new ObjectGroupDatabaseView(this);
  34. }
  35. #endregion
  36. #region Public Methods
  37. public ObjectGroup CreateObjectGroup(GameObject gameObject)
  38. {
  39. if (ContainsObjectGroup(gameObject))
  40. {
  41. Debug.LogWarning("This object is already marked as an object group.");
  42. return GetObjectGroup(gameObject);
  43. }
  44. ObjectGroup newObjectGroup = ObjectGroupFactory.Create(gameObject, GetAllObjectGroupNames());
  45. _objectGroups.AddEntity(newObjectGroup);
  46. if (NumberOfGroups == 1) SetActiveObjectGroup(newObjectGroup);
  47. PrefabManagementWindow.Get().Repaint();
  48. return newObjectGroup;
  49. }
  50. public ObjectGroup CreateObjectGroup(string groupName)
  51. {
  52. if (!string.IsNullOrEmpty(groupName))
  53. {
  54. ObjectGroup newObjectGroup = ObjectGroupFactory.Create(groupName, GetAllObjectGroupNames());
  55. _objectGroups.AddEntity(newObjectGroup);
  56. if (NumberOfGroups == 1) SetActiveObjectGroup(newObjectGroup);
  57. PrefabManagementWindow.Get().Repaint();
  58. return newObjectGroup;
  59. }
  60. return null;
  61. }
  62. public void MakeNoLongerGroup(GameObject gameObject)
  63. {
  64. if (!ContainsObjectGroup(gameObject)) return;
  65. UndoEx.RecordForToolAction(this);
  66. var objectGroup = _objectGroups.GetEntityByPredicate(item => item.GroupObject == gameObject);
  67. _objectGroups.RemoveEntity(objectGroup);
  68. UndoEx.DestroyObjectImmediate(objectGroup);
  69. PrefabManagementWindow.Get().Repaint();
  70. }
  71. public void MakeAllNoLongerGroup()
  72. {
  73. UndoEx.RecordForToolAction(this);
  74. var groupsToRemove = _objectGroups.GetAllEntities();
  75. _objectGroups.RemoveAllEntities();
  76. foreach(var group in groupsToRemove)
  77. {
  78. UndoEx.DestroyObjectImmediate(group);
  79. }
  80. PrefabManagementWindow.Get().Repaint();
  81. }
  82. public bool ContainsObjectGroup(GameObject objectGroup)
  83. {
  84. return _objectGroups.ContainsEntityByPredicate(item => item.GroupObject == objectGroup);
  85. }
  86. public bool ContainsObjectGroup(ObjectGroup objectGroup)
  87. {
  88. return _objectGroups.ContainsEntity(objectGroup);
  89. }
  90. public ObjectGroup GetObjectGroup(GameObject groupParent)
  91. {
  92. if (!ContainsObjectGroup(groupParent)) return null;
  93. return _objectGroups.GetEntityByPredicate(item => item.GroupObject == groupParent);
  94. }
  95. public void RenameObjectGroup(ObjectGroup objectGroup, string newName)
  96. {
  97. if (ContainsObjectGroup(objectGroup)) _objectGroups.RenameEntity(objectGroup, newName);
  98. PrefabManagementWindow.Get().Repaint();
  99. }
  100. public void RemoveAndDestroyObjectGroup(ObjectGroup objectGroup)
  101. {
  102. if (ContainsObjectGroup(objectGroup))
  103. {
  104. _objectGroups.RemoveEntity(objectGroup);
  105. GameObject groupParent = objectGroup.GroupObject;
  106. if (_preserveGroupChildren) groupParent.MoveImmediateChildrenUpOneLevel(true);
  107. if (groupParent != null) UndoEx.DestroyObjectImmediate(groupParent);
  108. UndoEx.DestroyObjectImmediate(objectGroup);
  109. PrefabManagementWindow.Get().Repaint();
  110. }
  111. }
  112. public void RemoveAndDestroyAllObjectGroups()
  113. {
  114. List<ObjectGroup> allGroups = GetAllObjectGroups();
  115. var groupsToDestroy = new List<ObjectGroup>();
  116. foreach (ObjectGroup objectGroup in allGroups)
  117. {
  118. _objectGroups.RemoveEntity(objectGroup);
  119. groupsToDestroy.Add(objectGroup);
  120. }
  121. foreach (ObjectGroup objectGroup in allGroups)
  122. {
  123. GameObject groupParent = objectGroup.GroupObject;
  124. if (_preserveGroupChildren) groupParent.MoveImmediateChildrenUpOneLevel(true);
  125. if (groupParent != null) UndoEx.DestroyObjectImmediate(groupParent);
  126. UndoEx.DestroyObjectImmediate(objectGroup);
  127. }
  128. PrefabManagementWindow.Get().Repaint();
  129. }
  130. public List<ObjectGroup> GetAllObjectGroups()
  131. {
  132. return _objectGroups.GetAllEntities();
  133. }
  134. public ObjectGroup GetObjectGroupByIndex(int groupIndex)
  135. {
  136. return _objectGroups.GetEntityByIndex(groupIndex);
  137. }
  138. public ObjectGroup GetObjectGroupByName(string name)
  139. {
  140. if(string.IsNullOrEmpty(name)) return null;
  141. return _objectGroups.GetEntityByName(name);
  142. }
  143. public List<string> GetAllObjectGroupNames()
  144. {
  145. return _objectGroups.GetAllEntityNames();
  146. }
  147. public void SetActiveObjectGroup(ObjectGroup newActiveObjectGroup)
  148. {
  149. if (newActiveObjectGroup != null && !ContainsObjectGroup(newActiveObjectGroup)) return;
  150. _objectGroups.MarkEntity(newActiveObjectGroup);
  151. }
  152. public void RemoveGroupsWithNullParents()
  153. {
  154. List<ObjectGroup> groupsWithNullParents = _objectGroups.GetAllEntities();
  155. groupsWithNullParents = groupsWithNullParents.FindAll(item => item != null && item.GroupObject == null);
  156. _objectGroups.RemoveWithPredicate(item => item != null && item.GroupObject == null);
  157. foreach (var group in groupsWithNullParents) ScriptableObject.DestroyImmediate(group);
  158. if (!IsEmpty)
  159. {
  160. if (IndexOfActiveGroup < 0 || IndexOfActiveGroup >= NumberOfGroups) SetActiveObjectGroup(GetObjectGroupByIndex(0));
  161. }
  162. else SetActiveObjectGroup(null);
  163. }
  164. #endregion
  165. }
  166. }
  167. #endif