FiE-Game/src/Fie.Utility/Wiggler.cs

using UnityEngine;

namespace Fie.Utility
{
	public class Wiggler
	{
		public enum WiggleTemplate
		{
			WIGGLE_TYPE_MINIMUM,
			WIGGLE_TYPE_SMALL,
			WIGGLE_TYPE_MIDDLE,
			WIGGLE_TYPE_BIG
		}

		private const float MIN_RANDOM_SHAKE_ANGLE_RANGE = 120f;

		private const float MAX_RANDOM_SHAKE_ANGLE_RANGE = 240f;

		private float nowAngle;

		private float nowTime;

		private float totalTime;

		private float wiggleSegmentTime;

		private int totalWiggleCount = 1;

		private int nowWiggleCount;

		private Vector3 wiggleScale = Vector3.zero;

		private Vector3 wiggleNormal = Vector3.zero;

		private Vector3 nowWigglePoint = Vector3.zero;

		private Vector3 nextWigglePoint = Vector3.zero;

		public bool isEnd => nowWiggleCount >= totalWiggleCount;

		public Wiggler(Vector3 normal, float initTotalTime, int initWiggleCount, Vector3 initWiggleScale)
		{
			totalTime = Mathf.Max(initTotalTime, 0f);
			totalWiggleCount = Mathf.Max(initWiggleCount, 1);
			wiggleScale = initWiggleScale;
			wiggleNormal = normal.normalized;
			InitializeGeneralParams();
		}

		public Wiggler(Vector3 normal, WiggleTemplate template)
		{
			float num = 0f;
			int num2 = 1;
			Vector3 zero = Vector3.zero;
			switch (template)
			{
			default:
				num = 0.15f;
				num2 = 4;
				zero = new Vector3(0.015f, 0.015f);
				break;
			case WiggleTemplate.WIGGLE_TYPE_SMALL:
				num = 0.2f;
				num2 = 5;
				zero = new Vector3(0.03f, 0.03f);
				break;
			case WiggleTemplate.WIGGLE_TYPE_MIDDLE:
				num = 0.3f;
				num2 = 7;
				zero = new Vector3(0.1f, 0.1f);
				break;
			case WiggleTemplate.WIGGLE_TYPE_BIG:
				num = 0.5f;
				num2 = 10;
				zero = new Vector3(0.1f, 0.1f);
				break;
			}
			totalTime = Mathf.Max(num, 0f);
			totalWiggleCount = Mathf.Max(num2, 1);
			wiggleScale = zero;
			wiggleNormal = normal.normalized;
			InitializeGeneralParams();
		}

		public Vector3 UpdateWiggler(float updateTime)
		{
			if (nowTime >= totalTime)
			{
				return Vector3.zero;
			}
			Vector3 result = (!(wiggleSegmentTime > 0f)) ? Vector3.zero : Vector3.Lerp(nowWigglePoint, nextWigglePoint, Mathf.Min(nowTime / wiggleSegmentTime, 1f));
			nowTime += updateTime;
			if (nowTime > wiggleSegmentTime && nowWiggleCount < totalWiggleCount)
			{
				nowWiggleCount++;
				SetNextWigglePoint(nowWiggleCount);
				nowTime = 0f;
			}
			return result;
		}

		private void InitializeGeneralParams()
		{
			nowTime = 0f;
			wiggleSegmentTime = totalTime / (float)totalWiggleCount;
			nowWiggleCount = 0;
			SetNextWigglePoint(nowWiggleCount);
		}

		private void SetNextWigglePoint(int count)
		{
			nowWigglePoint = new Vector3(nextWigglePoint.x, nextWigglePoint.y);
			float num = ((float)totalWiggleCount - (float)count) / (float)totalWiggleCount;
			float angle = nowAngle + Random.Range(120f, 240f);
			Quaternion rotation = Quaternion.AngleAxis(angle, wiggleNormal);
			Vector3 vector = rotation * Vector3.up;
			vector.Normalize();
			nextWigglePoint = new Vector3(wiggleScale.x * (vector.x * num), wiggleScale.y * (vector.y * num));
			nowAngle = angle;
		}
	}
}