OpenRA/OpenRa.Game/Traits/Activities/Move.cs

using System;
using System.Collections.Generic;
using System.Linq;
using OpenRa.Game.GameRules;

namespace OpenRa.Game.Traits.Activities
{
	class Move : Activity
	{
		public Activity NextActivity { get; set; }

		int2? destination;
		public List<int2> path;
		Func<Actor, Mobile, List<int2>> getPath;

		MovePart move;

		public Move( int2 destination )
		{
			this.getPath = ( self, mobile ) => Game.PathFinder.FindUnitPath(
				self.Location, destination,
				mobile.GetMovementType() );
			this.destination = destination;
		}

		public Move( Actor target, int range )
		{
			this.getPath = ( self, mobile ) => Game.PathFinder.FindUnitPathToRange(
				self.Location, target.Location,
				mobile.GetMovementType(), range );
			this.destination = null;
		}

		static bool CanEnterCell( int2 c, Actor self )
		{
			var u = Game.UnitInfluence.GetUnitAt( c );
			var b = Game.BuildingInfluence.GetBuildingAt( c );
			return ( u == null || u == self ) && b == null;
		}

		public void Tick( Actor self, Mobile mobile )
		{
			if( move != null )
			{
				move.TickMove( self, mobile, this );
				return;
			}

			if( destination == self.Location )
			{
				mobile.InternalSetActivity( NextActivity );
				return;
			}

			if( path == null ) path = getPath( self, mobile ).TakeWhile( a => a != self.Location ).ToList();

			if( path.Count == 0 )
			{
				destination = mobile.toCell;
				return;
			}

			destination = path[ 0 ];

			var nextCell = PopPath( self, mobile );
			if( nextCell == null )
				return;

			int2 dir = nextCell.Value - mobile.fromCell;
			var firstFacing = Util.GetFacing( dir, mobile.facing );
			if( firstFacing != mobile.facing )
			{
				var t = new Turn( firstFacing ) { NextActivity = this };
				mobile.InternalSetActivity( t );
				path.Add( nextCell.Value );

				t.Tick( self, mobile );
			}
			else
			{
				mobile.toCell = nextCell.Value;
				move = new MoveFirstHalf(
					CenterOfCell( mobile.fromCell ),
					BetweenCells( mobile.fromCell, mobile.toCell ),
					mobile.facing,
					mobile.facing,
					0 );

				Game.UnitInfluence.Update( mobile );

				move.TickMove( self, mobile, this );
			}
		}

		int2? PopPath( Actor self, Mobile mobile )
		{
			if( path.Count == 0 ) return null;
			var nextCell = path[ path.Count - 1 ];
			if( !CanEnterCell( nextCell, self ) )
			{
				if( ( mobile.toCell - destination.Value ).LengthSquared <= 8 )
				{
					path.Clear();
					return null;
				}

				Game.UnitInfluence.Remove( mobile );
				var newPath = Game.PathFinder.FindPathToPath( self.Location, path, mobile.GetMovementType() )
					.TakeWhile( a => a != self.Location )
					.ToList();
				Game.UnitInfluence.Add( mobile );
				if( newPath.Count == 0 )
					return null;

				while( path[ path.Count - 1 ] != newPath[ 0 ] )
					path.RemoveAt( path.Count - 1 );
				for( int i = 1 ; i < newPath.Count ; i++ )
					path.Add( newPath[ i ] );

				if( path.Count == 0 )
					return null;
				nextCell = path[ path.Count - 1 ];
			}
			path.RemoveAt( path.Count - 1 );
			return nextCell;
		}

		static float2 CenterOfCell( int2 loc )
		{
			return new float2( 12, 12 ) + Game.CellSize * (float2)loc;
		}

		static float2 BetweenCells( int2 from, int2 to )
		{
			return 0.5f * ( CenterOfCell( from ) + CenterOfCell( to ) );
		}

		abstract class MovePart
		{
			public readonly float2 from, to;
			public readonly int fromFacing, toFacing;
			public int moveFraction;
			public readonly int moveFractionTotal;

			public MovePart( float2 from, float2 to, int fromFacing, int toFacing, int startingFraction )
			{
				this.from = from;
				this.to = to;
				this.fromFacing = fromFacing;
				this.toFacing = toFacing;
				this.moveFraction = startingFraction;
				this.moveFractionTotal = (int)( to - from ).Length * ( 25 / 6 );
			}

			public void TickMove( Actor self, Mobile mobile, Move parent )
			{
				var oldFraction = moveFraction;
				var oldTotal = moveFractionTotal;

				moveFraction += ( self.unitInfo as UnitInfo.MobileInfo ).Speed;
				UpdateCenterLocation( self, mobile );
				if( moveFraction >= moveFractionTotal )
				{
					parent.move = OnComplete( self, mobile, parent );
					if( parent.move == null )
						UpdateCenterLocation( self, mobile );
				}
			}

			void UpdateCenterLocation( Actor self, Mobile mobile )
			{
				var frac = (float)moveFraction / moveFractionTotal;

				self.CenterLocation = float2.Lerp( from, to, frac );
				if( moveFraction >= moveFractionTotal )
					mobile.facing = toFacing & 0xFF;
				else
					mobile.facing = ( fromFacing + ( toFacing - fromFacing ) * moveFraction / moveFractionTotal ) & 0xFF;
			}

			protected abstract MovePart OnComplete( Actor self, Mobile mobile, Move parent );
		}

		class MoveFirstHalf : MovePart
		{
			public MoveFirstHalf( float2 from, float2 to, int fromFacing, int toFacing, int startingFraction )
				: base( from, to, fromFacing, toFacing, startingFraction )
			{
			}

			protected override MovePart OnComplete( Actor self, Mobile mobile, Move parent )
			{
				var nextCell = parent.PopPath( self, mobile );
				if( nextCell != null )
				{
					if( ( nextCell - mobile.toCell ) != ( mobile.toCell - mobile.fromCell ) )
					{
						var ret = new MoveFirstHalf(
							BetweenCells( mobile.fromCell, mobile.toCell ),
							BetweenCells( mobile.toCell, nextCell.Value ),
							mobile.facing,
							Util.GetNearestFacing( mobile.facing, Util.GetFacing( nextCell.Value - mobile.toCell, mobile.facing ) ),
							moveFraction - moveFractionTotal );
						mobile.fromCell = mobile.toCell;
						mobile.toCell = nextCell.Value;
						Game.UnitInfluence.Update( mobile );
						return ret;
					}
					else
						parent.path.Add( nextCell.Value );
				}
				var ret2 = new MoveSecondHalf(
					BetweenCells( mobile.fromCell, mobile.toCell ),
					CenterOfCell( mobile.toCell ),
					mobile.facing,
					mobile.facing,
					moveFraction - moveFractionTotal );
				mobile.fromCell = mobile.toCell;
				Game.UnitInfluence.Update( mobile );
				return ret2;
			}
		}

		class MoveSecondHalf : MovePart
		{
			public MoveSecondHalf( float2 from, float2 to, int fromFacing, int toFacing, int startingFraction )
				: base( from, to, fromFacing, toFacing, startingFraction )
			{
			}

			protected override MovePart OnComplete( Actor self, Mobile mobile, Move parent )
			{
				self.CenterLocation = CenterOfCell( mobile.toCell );
				mobile.fromCell = mobile.toCell;
				return null;
			}
		}

		public void Cancel( Actor self, Mobile mobile )
		{
			path = new List<int2>();
			NextActivity = null;
		}
	}
}