fb0cab7481
If a class is caching the TraitsImplementing enumerable, instead cache the results of enumerating it to an array. The avoids having to enumerate the sequence each time it is needed.
70 lines
1.6 KiB
C#
70 lines
1.6 KiB
C#
#region Copyright & License Information
|
|
/*
|
|
* Copyright 2007-2015 The OpenRA Developers (see AUTHORS)
|
|
* This file is part of OpenRA, which is free software. It is made
|
|
* available to you under the terms of the GNU General Public License
|
|
* as published by the Free Software Foundation. For more information,
|
|
* see COPYING.
|
|
*/
|
|
#endregion
|
|
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using OpenRA.Activities;
|
|
using OpenRA.Traits;
|
|
|
|
namespace OpenRA.Mods.Common.Activities
|
|
{
|
|
public class Drag : Activity
|
|
{
|
|
readonly IPositionable positionable;
|
|
readonly IMove movement;
|
|
readonly IDisableMove[] moveDisablers;
|
|
WPos start, end;
|
|
int length;
|
|
int ticks = 0;
|
|
|
|
public Drag(Actor self, WPos start, WPos end, int length)
|
|
{
|
|
positionable = self.Trait<IPositionable>();
|
|
movement = self.TraitOrDefault<IMove>();
|
|
moveDisablers = self.TraitsImplementing<IDisableMove>().ToArray();
|
|
this.start = start;
|
|
this.end = end;
|
|
this.length = length;
|
|
}
|
|
|
|
public override Activity Tick(Actor self)
|
|
{
|
|
if (moveDisablers.Any(d => d.MoveDisabled(self)))
|
|
return this;
|
|
|
|
var pos = length > 1
|
|
? WPos.Lerp(start, end, ticks, length - 1)
|
|
: end;
|
|
|
|
positionable.SetVisualPosition(self, pos);
|
|
if (++ticks >= length)
|
|
{
|
|
if (movement != null)
|
|
movement.IsMoving = false;
|
|
|
|
return NextActivity;
|
|
}
|
|
|
|
if (movement != null)
|
|
movement.IsMoving = true;
|
|
|
|
return this;
|
|
}
|
|
|
|
public override IEnumerable<Target> GetTargets(Actor self)
|
|
{
|
|
yield return Target.FromPos(end);
|
|
}
|
|
|
|
// Cannot be cancelled
|
|
public override void Cancel(Actor self) { }
|
|
}
|
|
}
|