Space soccer polish and editor preview button

Assets/Plugins/BezierSolution/Editor/BezierWalkerLocomotionEditor.cs

@@ -0,0 +1,56 @@
+using System.Collections.Generic;
+using UnityEditor;
+using UnityEngine;
+
+namespace BezierSolution.Extras
+{
+	[CustomEditor( typeof( BezierWalkerLocomotion ) )]
+	[CanEditMultipleObjects]
+	public class BezierWalkerLocomotionEditor : BezierWalkerEditor
+	{
+		private int tailSaveDataStartIndex;
+
+		protected override void SaveInitialData()
+		{
+			base.SaveInitialData();
+			tailSaveDataStartIndex = initialPositions.Count;
+
+			for( int i = 0; i < walkers.Length; i++ )
+			{
+				List<Transform> tail = ( (BezierWalkerLocomotion) walkers[i] ).Tail;
+				for( int j = 0; j < tail.Count; j++ )
+				{
+					initialPositions.Add( tail[j].position );
+					initialRotations.Add( tail[j].rotation );
+				}
+			}
+		}
+
+		protected override void RestoreInitialData()
+		{
+			base.RestoreInitialData();
+
+			int index = tailSaveDataStartIndex;
+			for( int i = 0; i < walkers.Length; i++ )
+			{
+				if( walkers[i] )
+				{
+					List<Transform> tail = ( (BezierWalkerLocomotion) walkers[i] ).Tail;
+					for( int j = 0; j < tail.Count; j++, index++ )
+					{
+						tail[j].position = initialPositions[index];
+						tail[j].rotation = initialRotations[index];
+					}
+				}
+			}
+		}
+
+		protected override void Simulate( float deltaTime )
+		{
+			for( int i = 0; i < walkers.Length; i++ )
+				( (BezierWalkerLocomotion) walkers[i] ).walker.Execute( deltaTime );
+
+			base.Simulate( deltaTime );
+		}
+	}
+}