using UnityEngine;

[AddComponentMenu("Modifiers/Squeeze")]
public class MegaSqueeze : MegaModifier
{
	public float			amount		= 0.0f;
	public float			crv			= 0.0f;
	public float			radialamount = 0.0f;
	public float			radialcrv	= 0.0f;
	public bool				doRegion	= false;
	public float			to			= 0.0f;
	public float			from		= 0.0f;
	public MegaAxis			axis		= MegaAxis.Y;
	Matrix4x4				mat			= new Matrix4x4();
	float k1;
	float k2;
	float k3;
	float k4;
	float l;
	float l2;
	float ovl;
	float ovl2;

	void SetK(float K1, float K2, float K3, float K4)
	{
		k1 = K1;
		k2 = K2;
		k3 = K3;
		k4 = K4;
	}

	public override string ModName() { return "Squeeze"; }
	public override string GetHelpURL() { return "?page_id=338"; }

	public override Vector3 Map(int i, Vector3 p)
	{
		float z;

		p = tm.MultiplyPoint3x4(p);

		if ( l != 0.0f )
		{
			if ( doRegion )
			{
				if ( p.y < from )
					z = from * ovl;
				else
				{
					if ( p.y > to )
						z = to * ovl;
					else
						z = p.y * ovl;
				}
			}
			else
				z = Mathf.Abs(p.y * ovl);


			float f =  1.0f + z * k1 + k2 * z * (1.0f - z);

			p.y *= f;
		}

		if ( l2 != 0.0f )
		{
			float dist = Mathf.Sqrt(p.x * p.x + p.z * p.z);
			float xy = dist * ovl2;
			float f1 =  1.0f + xy * k3 + k4 * xy * (1.0f - xy);
			p.x *= f1;
			p.z *= f1;
		}

		return invtm.MultiplyPoint3x4(p);
	}

	public override bool ModLateUpdate(MegaModContext mc)
	{
		return Prepare(mc);
	}

	public override bool Prepare(MegaModContext mc)
	{
		mat = Matrix4x4.identity;

		SetAxis(mat);
		SetK(amount, crv, radialamount, radialcrv);
		Vector3 size = bbox.Size();

		switch ( axis )
		{
			case MegaAxis.X:
				l = size[0];	//bbox.max[1] - bbox.min[1];
				l2 = Mathf.Sqrt(size[1] * size[1] + size[2] * size[2]);
				break;

			case MegaAxis.Y:
				l = size[1];	//bbox.max[1] - bbox.min[1];
				l2 = Mathf.Sqrt(size[0] * size[0] + size[2] * size[2]);
				break;

			case MegaAxis.Z:
				l = size[2];	//bbox.max[1] - bbox.min[1];
				l2 = Mathf.Sqrt(size[1] * size[1] + size[0] * size[0]);
				break;
		}

		if ( l != 0.0f )
			ovl = 1.0f / l;

		if ( l2 != 0.0f )
			ovl2 = 1.0f / l2;
		return true;
	}

	public override void ExtraGizmo(MegaModContext mc)
	{
		if ( doRegion )
			DrawFromTo(MegaAxis.Z, from, to, mc);
	}
}