1 files changed, 45 insertions, 0 deletions
diff --git a/Assets/Scripts/GravityTypes/GravityPlane.cs b/Assets/Scripts/GravityTypes/GravityPlane.cs
new file mode 100644
index 0000000..81a7005
--- /dev/null
+++ b/Assets/Scripts/GravityTypes/GravityPlane.cs
@@ -0,0 +1,45 @@
+using System.Collections;
+using System.Collections.Generic;
+using UnityEngine;
+
+public class GravityPlane : GravitySource
+{
+	[SerializeField]
+	float gravity = 9.81f;
+
+	[SerializeField]
+	float range = 1f;
+
+	public override Vector3 GetGravity(Vector3 position)
+	{
+		Vector3 up = transform.up;
+		float distance = Vector3.Dot(up, position - transform.position);
+		if(distance > range)
+			return Vector3.zero;
+
+		float g = -gravity;
+		if (distance > 0f)
+			g *= 1f - distance / range;
+		return -gravity * up;
+	}
+
+	void OnDrawGizmos()
+	{
+		Vector3 scale = transform.localScale;
+		scale.y = range;
+		Gizmos.matrix = Matrix4x4.TRS(
+				transform.position,
+				transform.rotation,
+				scale
+				);
+		//transform.localToWorldMatrix;
+		Vector3 size = new Vector3(1f, 0f, 1f);
+		Gizmos.color = Color.yellow;
+		Gizmos.DrawWireCube(Vector3.zero, size);
+		if (range > 0f)
+		{
+			Gizmos.color = Color.cyan;
+			Gizmos.DrawWireCube(Vector3.up, size);
+		}
+	}
+}