using UnityEngine;

[AddComponentMenu("Modifiers/Warps/FFD 4x4x4")]
public class MegaFFD4x4x4Warp : MegaFFDWarp
{
	public override string WarpName() { return "FFD4x4x4"; }

	public override int GridSize()
	{
		return 4;
	}

	public override Vector3 Map(int ii, Vector3 p)
	{
		Vector3 q = Vector3.zero;

		Vector3 pp = tm.MultiplyPoint3x4(p);

		if ( inVol )
		{
			for ( int i = 0; i < 3; i++ )
			{
				if ( pp[i] < -EPSILON || pp[i] > 1.0f + EPSILON )
					return p;
			}
		}

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

		float ip, jp, kp;

		float sx = 1.0f - pp.x;
		float sy = 1.0f - pp.y;
		float sz = 1.0f - pp.z;
		float sy2 = sy * sy;
		float py = pp.y * pp.y;
		float sz2 = sz * sz;
		float pz = pp.z * pp.z;

		for ( int i = 0; i < 4; i++ )
		{
			if ( i == 0 )
				ip = (sx * sx * sx);
			else
			{
				if ( i == 1 )
					ip = (3.0f * pp.x * sx * sx);
				else
				{
					if ( i == 2 )
						ip = (3.0f * pp.x * pp.x * sx);
					else
						ip = (pp.x * pp.x * pp.x);
				}
			}

			for ( int j = 0; j < 4; j++ )
			{
				if ( j == 0 )
					jp = ip * (sy * sy2);
				else
				{
					if ( j == 1 )
						jp = ip * (3.0f * pp.y * sy2);
					else
					{
						if ( j == 2 )
							jp = ip * (3.0f * py * sy);
						else
							jp = ip * (py * pp.y);
					}
				}

				for ( int k = 0; k < 4; k++ )
				{
					if ( k == 0 )
						kp = jp * (sz * sz2);
					else
					{
						if ( k == 1 )
							kp = jp * (3.0f * pp.z * sz2);
						else
						{
							if ( k == 2 )
								kp = jp * (3.0f * pz * sz);
							else
								kp = jp * (pz * pp.z);
						}
					}

					int ix = (i << 4) + (j << 2) + k;
					q.x += pt[ix].x * kp;
					q.y += pt[ix].y * kp;
					q.z += pt[ix].z * kp;
				}
			}
		}

		q = Vector3.Lerp(ipp, q, dcy);

		return invtm.MultiplyPoint3x4(q);
	}

	public override int GridIndex(int i, int j, int k)
	{
		return (i << 4) + (j << 2) + k;
	}
}