CrowdControl/Assets/Feel/MMTools/Tools/MMUtilities/MMTransformRandomizer.cs

using System.Collections;
using System.Collections.Generic;
using UnityEngine;

namespace MoreMountains.Tools
{
	/// <summary>
	/// Add this component to an object to randomize its position/rotation/scale on demand or automatically
	/// </summary>
	#if UNITY_EDITOR
	[ExecuteAlways]
	#endif
	public class MMTransformRandomizer : MonoBehaviour
	{
		/// the possible ways to automatically randomize
		public enum AutoExecutionModes { Never, OnAwake, OnStart, OnEnable }

		[Header("Position")]
		/// whether or not to randomize position
		public bool RandomizePosition = true;
		/// the minimum position to apply when randomizing 
		[MMCondition("RandomizePosition", true)]
		public Vector3 MinRandomPosition;
		/// the maximum position to apply when randomizing
		[MMCondition("RandomizePosition", true)]
		public Vector3 MaxRandomPosition;

		[Header("Rotation")]
		/// whether or not to randomize rotation
		public bool RandomizeRotation = true;
		/// the minimum rotation to apply when randomizing (in degrees)
		[MMCondition("RandomizeRotation", true)]
		public Vector3 MinRandomRotation;
		/// the maximum rotation to apply when randomizing (in degrees)
		[MMCondition("RandomizeRotation", true)]
		public Vector3 MaxRandomRotation;

		[Header("Scale")]
		/// whether or not to randomize scale
		public bool RandomizeScale = true;
		/// the minimum scale to apply when randomizing
		[MMCondition("RandomizeScale", true)]
		public Vector3 MinRandomScale;
		/// the maximum scale to apply when randomizing
		[MMCondition("RandomizeScale", true)]
		public Vector3 MaxRandomScale;

		[Header("Settings")]
		/// whether or not to remove this component after randomizing its attributes
		public bool AutoRemoveAfterRandomize = false;
		/// whether or not to remove all colliders attached to this object
		public bool RemoveAllColliders = false;
		/// the selected auto execution mode
		public AutoExecutionModes AutoExecutionMode = AutoExecutionModes.Never;

		/// <summary>
		/// On Awake we randomize if needed
		/// </summary>
		protected virtual void Awake()
		{
			if (Application.isPlaying && (AutoExecutionMode == AutoExecutionModes.OnAwake))
			{
				Randomize();
			}
		}

		/// <summary>
		/// On Start we randomize if needed
		/// </summary>
		protected virtual void Start()
		{
			if (Application.isPlaying && (AutoExecutionMode == AutoExecutionModes.OnStart))
			{
				Randomize();
			}
		}

		/// <summary>
		/// On Enable we randomize if needed
		/// </summary>
		protected virtual void OnEnable()
		{
			if (Application.isPlaying && (AutoExecutionMode == AutoExecutionModes.OnEnable))
			{
				Randomize();
			}
		}

		/// <summary>
		/// Randomizes position, rotation, scale, and cleanups if necessary
		/// </summary>
		public virtual void Randomize()
		{
			ProcessRandomizePosition();
			ProcessRandomizeRotation();
			ProcessRandomizeScale();
			RemoveColliders();
			Cleanup();
		}
        
		/// <summary>
		/// Randomizes the position
		/// </summary>
		protected virtual void ProcessRandomizePosition()
		{
			if (!RandomizePosition)
			{
				return;
			}
			Vector3 randomPosition = MMMaths.RandomVector3(MinRandomPosition, MaxRandomPosition);
			this.transform.localPosition += randomPosition;
		}

		/// <summary>
		/// Randomizes the rotation
		/// </summary>
		protected virtual void ProcessRandomizeRotation()
		{
			if (!RandomizeRotation)
			{
				return;
			}
			Vector3 randomRotation = MMMaths.RandomVector3(MinRandomRotation, MaxRandomRotation);
			this.transform.localRotation = Quaternion.Euler(randomRotation);
		}

		/// <summary>
		/// Randomizes the scale
		/// </summary>
		protected virtual void ProcessRandomizeScale()
		{
			if (!RandomizeScale)
			{
				return;
			}
			Vector3 randomScale = MMMaths.RandomVector3(MinRandomScale, MaxRandomScale);
			this.transform.localScale = randomScale;
		}

		/// <summary>
		/// Removes all colliders attached to this object or its children
		/// </summary>
		protected virtual void RemoveColliders()
		{
			if (RemoveAllColliders)
			{
				#if UNITY_EDITOR
				Collider[] colliders = this.gameObject.GetComponentsInChildren<Collider>();
				foreach (Collider collider in colliders)
				{
					DestroyImmediate(collider);
				}
				Collider2D[] colliders2D = this.gameObject.GetComponentsInChildren<Collider2D>();
				foreach (Collider2D collider2D in colliders2D)
				{
					DestroyImmediate(collider2D);
				}
				#endif
			}
		}

		/// <summary>
		/// Destroys this component
		/// </summary>
		protected virtual void Cleanup()
		{
			if (AutoRemoveAfterRandomize)
			{
				#if UNITY_EDITOR
				DestroyImmediate(this);
				#endif
			}
		}
	}
}