using UnityEngine;
using System.IO;

[AddComponentMenu("Modifiers/Warps/Ripple")]
public class MegaRippleWarp : MegaWarp
{
	public float	amp		= 0.0f;
	public float	amp2	= 0.0f;
	public float	flex	= 1.0f;
	public float	wave	= 1.0f;
	public float	phase	= 0.0f;
	public bool		animate	= false;
	public float	Speed	= 1.0f;
	public float	radius	= 1.0f;
	public int		segments = 10;
	public int		circles	= 4;
	float time	= 0.0f;
	float dy	= 0.0f;
	float t		= 0.0f;

	public override string WarpName()	{ return "Ripple"; }
	public override string GetIcon()	{ return "MegaRipple icon.png"; }
	public override string GetHelpURL() { return "?page_id=2587"; }

	public override Vector3 Map(int i, Vector3 p)
	{
		p = tm.MultiplyPoint3x4(p);

		Vector3 ip = p;
		float dist = p.magnitude;
		float dcy = Mathf.Exp(-totaldecay * Mathf.Abs(dist));

		float a;

		if ( amp != amp2 )
		{
			float len  = p.magnitude;
			if ( len == 0.0f )
				a = amp;
			else
			{
				float u = (Mathf.Acos(p.x / len)) / Mathf.PI;
				u = (u > 0.5f) ? (1.0f - u) : u;
				u *= 2.0f;
				u = Mathf.SmoothStep(0.0f, 1.0f, u);
				a = amp * (1.0f - u) + amp2 * u;
			}
		}
		else
			a = amp;

		float oldZ = p.y;
		p.y = 0.0f;
		float r = p.magnitude;
		p.y = oldZ + flex * MegaUtils.WaveFunc(r, time, a, wave, phase, dy);

		p = Vector3.Lerp(ip, p, dcy);

		return invtm.MultiplyPoint3x4(p);
	}

	void Update()
	{
		if ( animate )
		{
			if ( Application.isPlaying )
				t += Time.deltaTime * Speed;
			phase = t;
		}
	}

	public override bool Prepare(float decay)
	{
		tm = transform.worldToLocalMatrix;
		invtm = tm.inverse;

		dy = Decay / 1000.0f;

		totaldecay = dy + decay;
		if ( totaldecay < 0.0f )
			totaldecay = 0.0f;

		return true;
	}

	Vector3 GetPos(float u, float rad)
	{
		Vector3 pos = Vector3.zero;

		pos.x = rad * Mathf.Cos(u * Mathf.PI * 2.0f);
		pos.z = rad * Mathf.Sin(u * Mathf.PI * 2.0f);

		float u2 = (u > 0.5f) ? (u - 0.5f) : u;
		u2 = (u2 > 0.25f) ? (0.5f - u2) : u2;
		u2 = u2 * 4.0f;
		u2 = u2 * u2;
		pos.y = MegaUtils.WaveFunc(rad, t, amp * (1.0f - u2) + amp2 * u2, wave, phase, dy);

		return pos;
	}

	void MakeCircle(float t, float rad, float r1, float a1, float a2, float w, float s, float d, int numCircleSegs)
	{
		Vector3 last = Vector3.zero;
		Vector3 pos = Vector3.zero;
		Vector3 pos1 = Vector3.zero;
		Vector3 first = Vector3.zero;

		Gizmos.color = gCol2;

		for ( int i = 0; i < numCircleSegs; i++ )
		{
			float u = (float)i / (float)numCircleSegs;
			pos = GetPos(u, rad);
			pos1 = GetPos(u, r1);

			if ( i > 0 )
				Gizmos.DrawLine(last, pos);
			else
				first = pos;

			Gizmos.color = gCol1;
			Gizmos.DrawLine(pos1, pos);

			if ( (i & 1) == 0 )
				Gizmos.color = gCol1;
			else
				Gizmos.color = gCol2;

			last = pos;
		}

		Gizmos.DrawLine(last, first);
	}

	public override void DrawGizmo(Color col)
	{
		SetGizCols(col.a);
		tm = Matrix4x4.identity;
		invtm = tm.inverse;

		if ( !Prepare(0.0f) )
			return;

		tm = tm * transform.localToWorldMatrix;	// * tm;
		invtm = tm.inverse;

		Gizmos.matrix = transform.localToWorldMatrix;

		float r1 = 0.0f;
		for ( int i = 0; i < circles; i++ )
		{
			float r = ((float)i / (float)circles) * radius;
			MakeCircle(t, r, r1, amp, amp2, wave, phase, dy, segments);
			r1 = r;
		}
	}
}