using System;
using System.Collections.Generic;
using UnityEngine;
namespace Unity.Linq
{
///
/// Set type of cloned child GameObject's localPosition/Scale/Rotation.
///
public enum TransformCloneType
{
/// Set to same as Original. This is default of Add methods.
KeepOriginal,
/// Set to same as Parent.
FollowParent,
/// Set to Position = zero, Scale = one, Rotation = identity.
Origin,
/// Position/Scale/Rotation as is.
DoNothing
}
///
/// Set type of moved child GameObject's localPosition/Scale/Rotation.
///
public enum TransformMoveType
{
/// Set to same as Parent.
FollowParent,
/// Set to Position = zero, Scale = one, Rotation = identity.
Origin,
/// Position/Scale/Rotation as is.
DoNothing
}
public static partial class GameObjectExtensions
{
static UnityEngine.GameObject GetGameObject(T obj)
where T : UnityEngine.Object
{
var gameObject = obj as GameObject;
if (gameObject == null)
{
var component = obj as Component;
if (component == null)
{
return null;
}
gameObject = component.gameObject;
}
return gameObject;
}
#region Add
///
/// Adds the GameObject/Component as children of this GameObject. Target is cloned.
///
/// Parent GameObject.
/// Clone Target.
/// Choose set type of cloned child GameObject's localPosition/Scale/Rotation.
/// Set activates/deactivates child GameObject. If null, doesn't set specified value.
/// Set name of child GameObject. If null, doesn't set specified value.
/// Set layer of child GameObject same with parent.
public static T Add(this GameObject parent, T childOriginal, TransformCloneType cloneType = TransformCloneType.KeepOriginal, bool? setActive = null, string specifiedName = null, bool setLayer = false)
where T : UnityEngine.Object
{
if (parent == null) throw new ArgumentNullException("parent");
if (childOriginal == null) throw new ArgumentNullException("childOriginal");
var child = UnityEngine.Object.Instantiate(childOriginal);
var childGameObject = GetGameObject(child);
// for uGUI, should use SetParent(parent, false)
var childTransform = childGameObject.transform;
#if !(UNITY_4_0 || UNITY_4_1 || UNITY_4_2 || UNITY_4_3 || UNITY_4_4 || UNITY_4_5)
var rectTransform = childTransform as RectTransform;
if (rectTransform != null)
{
rectTransform.SetParent(parent.transform, worldPositionStays: false);
}
else
{
#endif
var parentTransform = parent.transform;
childTransform.parent = parentTransform;
switch (cloneType)
{
case TransformCloneType.FollowParent:
childTransform.localPosition = parentTransform.localPosition;
childTransform.localScale = parentTransform.localScale;
childTransform.localRotation = parentTransform.localRotation;
break;
case TransformCloneType.Origin:
childTransform.localPosition = Vector3.zero;
childTransform.localScale = Vector3.one;
childTransform.localRotation = Quaternion.identity;
break;
case TransformCloneType.KeepOriginal:
var co = GetGameObject(childOriginal);
var childOriginalTransform = co.transform;
childTransform.localPosition = childOriginalTransform.localPosition;
childTransform.localScale = childOriginalTransform.localScale;
childTransform.localRotation = childOriginalTransform.localRotation;
break;
case TransformCloneType.DoNothing:
default:
break;
}
#if !(UNITY_4_0 || UNITY_4_1 || UNITY_4_2 || UNITY_4_3 || UNITY_4_4 || UNITY_4_5)
}
#endif
if (setLayer)
{
childGameObject.layer = parent.layer;
}
if (setActive != null)
{
childGameObject.SetActive(setActive.Value);
}
if (specifiedName != null)
{
child.name = specifiedName;
}
return child;
}
///
/// Adds the GameObject/Component as children of this GameObject. Target is cloned.
///
/// Parent GameObject.
/// Clone Target.
/// Choose set type of cloned child GameObject's localPosition/Scale/Rotation.
/// Set activates/deactivates child GameObject. If null, doesn't set specified value.
/// Set name of child GameObject. If null, doesn't set specified value.
public static T[] AddRange(this GameObject parent, IEnumerable childOriginals, TransformCloneType cloneType = TransformCloneType.KeepOriginal, bool? setActive = null, string specifiedName = null, bool setLayer = false)
where T : UnityEngine.Object
{
if (parent == null) throw new ArgumentNullException("parent");
if (childOriginals == null) throw new ArgumentNullException("childOriginals");
// iteration optimize
{
var array = childOriginals as T[];
if (array != null)
{
var result = new T[array.Length];
for (int i = 0; i < array.Length; i++)
{
var child = Add(parent, array[i], cloneType, setActive, specifiedName, setLayer);
result[i] = child;
}
return result;
}
}
{
var iterList = childOriginals as IList;
if (iterList != null)
{
var result = new T[iterList.Count];
for (int i = 0; i < iterList.Count; i++)
{
var child = Add(parent, iterList[i], cloneType, setActive, specifiedName, setLayer);
result[i] = child;
}
return result;
}
}
{
var result = new List();
foreach (var childOriginal in childOriginals)
{
var child = Add(parent, childOriginal, cloneType, setActive, specifiedName, setLayer);
result.Add(child);
}
return result.ToArray();
}
}
///
/// Adds the GameObject/Component as the first children of this GameObject. Target is cloned.
///
/// Parent GameObject.
/// Clone Target.
/// Choose set type of cloned child GameObject's localPosition/Scale/Rotation.
/// Set activates/deactivates child GameObject. If null, doesn't set specified value.
/// Set name of child GameObject. If null, doesn't set specified value.
/// Set layer of child GameObject same with parent.
public static T AddFirst(this GameObject parent, T childOriginal, TransformCloneType cloneType = TransformCloneType.KeepOriginal, bool? setActive = null, string specifiedName = null, bool setLayer = false)
where T : UnityEngine.Object
{
var child = Add(parent, childOriginal, cloneType, setActive, specifiedName, setLayer);
var go = GetGameObject(child);
if (go == null) return child;
go.transform.SetAsFirstSibling();
return child;
}
///
/// Adds the GameObject/Component as the first children of this GameObject. Target is cloned.
///
/// Parent GameObject.
/// Clone Target.
/// Choose set type of cloned child GameObject's localPosition/Scale/Rotation.
/// Set activates/deactivates child GameObject. If null, doesn't set specified value.
/// Set name of child GameObject. If null, doesn't set specified value.
/// Set layer of child GameObject same with parent.
public static T[] AddFirstRange(this GameObject parent, IEnumerable childOriginals, TransformCloneType cloneType = TransformCloneType.KeepOriginal, bool? setActive = null, string specifiedName = null, bool setLayer = false)
where T : UnityEngine.Object
{
var child = AddRange(parent, childOriginals, cloneType, setActive, specifiedName, setLayer);
for (int i = child.Length - 1; i >= 0; i--)
{
var go = GetGameObject(child[i]);
if (go == null) continue;
go.transform.SetAsFirstSibling();
}
return child;
}
///
/// Adds the GameObject/Component before this GameObject. Target is cloned.
///
/// Parent GameObject.
/// Clone Target.
/// Choose set type of cloned child GameObject's localPosition/Scale/Rotation.
/// Set activates/deactivates child GameObject. If null, doesn't set specified value.
/// Set name of child GameObject. If null, doesn't set specified value.
/// Set layer of child GameObject same with parent.
public static T AddBeforeSelf(this GameObject parent, T childOriginal, TransformCloneType cloneType = TransformCloneType.KeepOriginal, bool? setActive = null, string specifiedName = null, bool setLayer = false)
where T : UnityEngine.Object
{
var root = parent.Parent();
if (root == null) throw new InvalidOperationException("The parent root is null");
var sibilingIndex = parent.transform.GetSiblingIndex();
var child = Add(root, childOriginal, cloneType, setActive, specifiedName, setLayer);
var go = GetGameObject(child);
if (go == null) return child;
go.transform.SetSiblingIndex(sibilingIndex);
return child;
}
///
/// Adds the GameObject/Component before this GameObject. Target is cloned.
///
/// Parent GameObject.
/// Clone Target.
/// Choose set type of cloned child GameObject's localPosition/Scale/Rotation.
/// Set activates/deactivates child GameObject. If null, doesn't set specified value.
/// Set name of child GameObject. If null, doesn't set specified value.
/// Set layer of child GameObject same with parent.
public static T[] AddBeforeSelfRange(this GameObject parent, IEnumerable childOriginals, TransformCloneType cloneType = TransformCloneType.KeepOriginal, bool? setActive = null, string specifiedName = null, bool setLayer = false)
where T : UnityEngine.Object
{
var root = parent.Parent();
if (root == null) throw new InvalidOperationException("The parent root is null");
var sibilingIndex = parent.transform.GetSiblingIndex();
var child = AddRange(root, childOriginals, cloneType, setActive, specifiedName, setLayer);
for (int i = child.Length - 1; i >= 0; i--)
{
var go = GetGameObject(child[i]);
if (go == null) continue;
go.transform.SetSiblingIndex(sibilingIndex);
}
return child;
}
///
/// Adds the GameObject/Component after this GameObject. Target is cloned.
///
/// Parent GameObject.
/// Clone Target.
/// Choose set type of cloned child GameObject's localPosition/Scale/Rotation.
/// Set activates/deactivates child GameObject. If null, doesn't set specified value.
/// Set name of child GameObject. If null, doesn't set specified value.
/// Set layer of child GameObject same with parent.
public static T AddAfterSelf(this GameObject parent, T childOriginal, TransformCloneType cloneType = TransformCloneType.KeepOriginal, bool? setActive = null, string specifiedName = null, bool setLayer = false)
where T : UnityEngine.Object
{
var root = parent.Parent();
if (root == null) throw new InvalidOperationException("The parent root is null");
var sibilingIndex = parent.transform.GetSiblingIndex() + 1;
var child = Add(root, childOriginal, cloneType, setActive, specifiedName, setLayer);
var go = GetGameObject(child);
if (go == null) return child;
go.transform.SetSiblingIndex(sibilingIndex);
return child;
}
///
/// Adds the GameObject/Component after this GameObject. Target is cloned.
///
/// Parent GameObject.
/// Clone Target.
/// Choose set type of cloned child GameObject's localPosition/Scale/Rotation.
/// Set activates/deactivates child GameObject. If null, doesn't set specified value.
/// Set name of child GameObject. If null, doesn't set specified value.
/// Set layer of child GameObject same with parent.
public static T[] AddAfterSelfRange(this GameObject parent, IEnumerable childOriginals, TransformCloneType cloneType = TransformCloneType.KeepOriginal, bool? setActive = null, string specifiedName = null, bool setLayer = false)
where T : UnityEngine.Object
{
var root = parent.Parent();
if (root == null) throw new InvalidOperationException("The parent root is null");
var sibilingIndex = parent.transform.GetSiblingIndex() + 1;
var child = AddRange(root, childOriginals, cloneType, setActive, specifiedName, setLayer);
for (int i = child.Length - 1; i >= 0; i--)
{
var go = GetGameObject(child[i]);
if (go == null) continue;
go.transform.SetSiblingIndex(sibilingIndex);
}
return child;
}
#endregion
#region Move
///
/// Move the GameObject/Component as children of this GameObject.
///
/// Parent GameObject.
/// Target.
/// Choose set type of cloned child GameObject's localPosition/Scale/Rotation.
/// Set activates/deactivates child GameObject. If null, doesn't set specified value.
/// Set layer of child GameObject same with parent.
public static T MoveToLast(this GameObject parent, T child, TransformMoveType moveType = TransformMoveType.DoNothing, bool? setActive = null, bool setLayer = false)
where T : UnityEngine.Object
{
if (parent == null) throw new ArgumentNullException("parent");
if (child == null) throw new ArgumentNullException("child");
var childGameObject = GetGameObject(child);
if (child == null) return child;
// for uGUI, should use SetParent(parent, false)
var childTransform = childGameObject.transform;
#if !(UNITY_4_0 || UNITY_4_1 || UNITY_4_2 || UNITY_4_3 || UNITY_4_4 || UNITY_4_5)
var rectTransform = childTransform as RectTransform;
if (rectTransform != null)
{
rectTransform.SetParent(parent.transform, worldPositionStays: false);
}
else
{
#endif
var parentTransform = parent.transform;
childTransform.parent = parentTransform;
switch (moveType)
{
case TransformMoveType.FollowParent:
childTransform.localPosition = parentTransform.localPosition;
childTransform.localScale = parentTransform.localScale;
childTransform.localRotation = parentTransform.localRotation;
break;
case TransformMoveType.Origin:
childTransform.localPosition = Vector3.zero;
childTransform.localScale = Vector3.one;
childTransform.localRotation = Quaternion.identity;
break;
case TransformMoveType.DoNothing:
default:
break;
}
#if !(UNITY_4_0 || UNITY_4_1 || UNITY_4_2 || UNITY_4_3 || UNITY_4_4 || UNITY_4_5)
}
#endif
if (setLayer)
{
childGameObject.layer = parent.layer;
}
if (setActive != null)
{
childGameObject.SetActive(setActive.Value);
}
return child;
}
///
/// Move the GameObject/Component as children of this GameObject.
///
/// Parent GameObject.
/// Target.
/// Choose set type of moved child GameObject's localPosition/Scale/Rotation.
/// Set activates/deactivates child GameObject. If null, doesn't set specified value.
/// Set layer of child GameObject same with parent.
public static T[] MoveToLastRange(this GameObject parent, IEnumerable childs, TransformMoveType moveType = TransformMoveType.DoNothing, bool? setActive = null, bool setLayer = false)
where T : UnityEngine.Object
{
if (parent == null) throw new ArgumentNullException("parent");
if (childs == null) throw new ArgumentNullException("childs");
// iteration optimize
{
var array = childs as T[];
if (array != null)
{
var result = new T[array.Length];
for (int i = 0; i < array.Length; i++)
{
var child = MoveToLast(parent, array[i], moveType, setActive, setLayer);
result[i] = child;
}
return result;
}
}
{
var iterList = childs as IList;
if (iterList != null)
{
var result = new T[iterList.Count];
for (int i = 0; i < iterList.Count; i++)
{
var child = MoveToLast(parent, iterList[i], moveType, setActive, setLayer);
result[i] = child;
}
return result;
}
}
{
var result = new List();
foreach (var childOriginal in childs)
{
var child = MoveToLast(parent, childOriginal, moveType, setActive, setLayer);
result.Add(child);
}
return result.ToArray();
}
}
///
/// Move the GameObject/Component as the first children of this GameObject.
///
/// Parent GameObject.
/// Target.
/// Choose set type of cloned child GameObject's localPosition/Scale/Rotation.
/// Set activates/deactivates child GameObject. If null, doesn't set specified value.
/// Set layer of child GameObject same with parent.
public static T MoveToFirst(this GameObject parent, T child, TransformMoveType moveType = TransformMoveType.DoNothing, bool? setActive = null, bool setLayer = false)
where T : UnityEngine.Object
{
MoveToLast(parent, child, moveType, setActive, setLayer);
var go = GetGameObject(child);
if (go == null) return child;
go.transform.SetAsFirstSibling();
return child;
}
///
/// Move the GameObject/Component as the first children of this GameObject.
///
/// Parent GameObject.
/// Target.
/// Choose set type of cloned child GameObject's localPosition/Scale/Rotation.
/// Set activates/deactivates child GameObject. If null, doesn't set specified value.
/// Set layer of child GameObject same with parent.
public static T[] MoveToFirstRange(this GameObject parent, IEnumerable childs, TransformMoveType moveType = TransformMoveType.DoNothing, bool? setActive = null, bool setLayer = false)
where T : UnityEngine.Object
{
var child = MoveToLastRange(parent, childs, moveType, setActive, setLayer);
for (int i = child.Length - 1; i >= 0; i--)
{
var go = GetGameObject(child[i]);
if (go == null) continue;
go.transform.SetAsFirstSibling();
}
return child;
}
///
/// Move the GameObject/Component before this GameObject.
///
/// Parent GameObject.
/// Target.
/// Choose set type of cloned child GameObject's localPosition/Scale/Rotation.
/// Set activates/deactivates child GameObject. If null, doesn't set specified value.
/// Set layer of child GameObject same with parent.
public static T MoveToBeforeSelf(this GameObject parent, T child, TransformMoveType moveType = TransformMoveType.DoNothing, bool? setActive = null, bool setLayer = false)
where T : UnityEngine.Object
{
var root = parent.Parent();
if (root == null) throw new InvalidOperationException("The parent root is null");
var sibilingIndex = parent.transform.GetSiblingIndex();
MoveToLast(root, child, moveType, setActive, setLayer);
var go = GetGameObject(child);
if (go == null) return child;
go.transform.SetSiblingIndex(sibilingIndex);
return child;
}
///
/// Move the GameObject/Component before GameObject.
///
/// Parent GameObject.
/// Target.
/// Choose set type of cloned child GameObject's localPosition/Scale/Rotation.
/// Set activates/deactivates child GameObject. If null, doesn't set specified value.
/// Set layer of child GameObject same with parent.
public static T[] MoveToBeforeSelfRange(this GameObject parent, IEnumerable childs, TransformMoveType moveType = TransformMoveType.DoNothing, bool? setActive = null, bool setLayer = false)
where T : UnityEngine.Object
{
var root = parent.Parent();
if (root == null) throw new InvalidOperationException("The parent root is null");
var sibilingIndex = parent.transform.GetSiblingIndex();
var child = MoveToLastRange(root, childs, moveType, setActive, setLayer);
for (int i = child.Length - 1; i >= 0; i--)
{
var go = GetGameObject(child[i]);
if (go == null) continue;
go.transform.SetSiblingIndex(sibilingIndex);
}
return child;
}
///
/// Move the GameObject/Component after this GameObject.
///
/// Parent GameObject.
/// Target.
/// Choose set type of cloned child GameObject's localPosition/Scale/Rotation.
/// Set activates/deactivates child GameObject. If null, doesn't set specified value.
/// Set layer of child GameObject same with parent.
public static T MoveToAfterSelf(this GameObject parent, T child, TransformMoveType moveType = TransformMoveType.DoNothing, bool? setActive = null, bool setLayer = false)
where T : UnityEngine.Object
{
var root = parent.Parent();
if (root == null) throw new InvalidOperationException("The parent root is null");
var sibilingIndex = parent.transform.GetSiblingIndex() + 1;
MoveToLast(root, child, moveType, setActive, setLayer);
var go = GetGameObject(child);
if (go == null) return child;
go.transform.SetSiblingIndex(sibilingIndex);
return child;
}
///
/// Move the GameObject/Component after this GameObject.
///
/// Parent GameObject.
/// Target.
/// Choose set type of cloned child GameObject's localPosition/Scale/Rotation.
/// Set activates/deactivates child GameObject. If null, doesn't set specified value.
/// Set layer of child GameObject same with parent.
public static T[] MoveToAfterSelfRange(this GameObject parent, IEnumerable childs, TransformMoveType moveType = TransformMoveType.DoNothing, bool? setActive = null, bool setLayer = false)
where T : UnityEngine.Object
{
var root = parent.Parent();
if (root == null) throw new InvalidOperationException("The parent root is null");
var sibilingIndex = parent.transform.GetSiblingIndex() + 1;
var child = MoveToLastRange(root, childs, moveType, setActive, setLayer);
for (int i = child.Length - 1; i >= 0; i--)
{
var go = GetGameObject(child[i]);
if (go == null) continue;
go.transform.SetSiblingIndex(sibilingIndex);
}
return child;
}
#endregion
/// Destroy this GameObject safety(check null).
/// If in EditMode, should be true or pass !Application.isPlaying.
/// set to parent = null.
public static void Destroy(this GameObject self, bool useDestroyImmediate = false, bool detachParent = false)
{
if (self == null) return;
if (detachParent)
{
#if !(UNITY_4_0 || UNITY_4_1 || UNITY_4_2 || UNITY_4_3 || UNITY_4_4 || UNITY_4_5)
self.transform.SetParent(null);
#else
self.transform.parent = null;
#endif
}
if (useDestroyImmediate)
{
GameObject.DestroyImmediate(self);
}
else
{
GameObject.Destroy(self);
}
}
}
}