138 lines
3.9 KiB
C#
138 lines
3.9 KiB
C#
#region Copyright & License Information
|
|
/*
|
|
* Copyright (c) The OpenRA Developers and Contributors
|
|
* 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, either version 3 of
|
|
* the License, or (at your option) any later version. For more
|
|
* information, see COPYING.
|
|
*/
|
|
#endregion
|
|
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using OpenRA.Activities;
|
|
using OpenRA.Mods.Common.Traits;
|
|
using OpenRA.Primitives;
|
|
using OpenRA.Traits;
|
|
|
|
namespace OpenRA.Mods.Common.Activities
|
|
{
|
|
public class MoveToDock : Activity
|
|
{
|
|
readonly DockClientManager dockClient;
|
|
Actor dockHostActor;
|
|
IDockHost dockHost;
|
|
readonly INotifyDockClientMoving[] notifyDockClientMoving;
|
|
readonly Color? dockLineColor;
|
|
readonly MoveCooldownHelper moveCooldownHelper;
|
|
readonly bool forceEnter;
|
|
readonly bool ignoreOccupancy;
|
|
|
|
public MoveToDock(Actor self, Actor dockHostActor = null, IDockHost dockHost = null,
|
|
bool forceEnter = false, bool ignoreOccupancy = false, Color? dockLineColor = null)
|
|
{
|
|
dockClient = self.Trait<DockClientManager>();
|
|
this.dockHostActor = dockHostActor;
|
|
this.dockHost = dockHost;
|
|
this.forceEnter = forceEnter;
|
|
this.ignoreOccupancy = ignoreOccupancy;
|
|
this.dockLineColor = dockLineColor;
|
|
notifyDockClientMoving = self.TraitsImplementing<INotifyDockClientMoving>().ToArray();
|
|
moveCooldownHelper = new MoveCooldownHelper(self.World, self.Trait<IMove>() as Mobile) { RetryIfDestinationBlocked = true };
|
|
}
|
|
|
|
protected override void OnFirstRun(Actor self)
|
|
{
|
|
if (IsCanceling || dockClient.IsTraitDisabled)
|
|
return;
|
|
|
|
// We were ordered to dock to an actor but host was unspecified.
|
|
if (dockHostActor != null && dockHost == null)
|
|
{
|
|
var link = dockClient.AvailableDockHosts(dockHostActor, default, forceEnter, ignoreOccupancy)
|
|
.ClosestDock(self, dockClient);
|
|
|
|
if (link.HasValue)
|
|
dockHost = link.Value.Trait;
|
|
}
|
|
}
|
|
|
|
public override bool Tick(Actor self)
|
|
{
|
|
if (IsCanceling)
|
|
return true;
|
|
|
|
if (dockClient.IsTraitDisabled)
|
|
{
|
|
Cancel(self, true);
|
|
return true;
|
|
}
|
|
|
|
// Find the nearest DockHost if not explicitly ordered to a specific dock.
|
|
if (dockHost == null || !dockHost.IsEnabledAndInWorld)
|
|
{
|
|
var host = dockClient.ClosestDock(null);
|
|
if (host.HasValue)
|
|
{
|
|
dockHost = host.Value.Trait;
|
|
dockHostActor = host.Value.Actor;
|
|
}
|
|
else
|
|
{
|
|
// No docks exist; check again after delay defined in dockClient.
|
|
QueueChild(new Wait(dockClient.Info.SearchForDockDelay));
|
|
return false;
|
|
}
|
|
}
|
|
|
|
var result = moveCooldownHelper.Tick(false);
|
|
if (result != null)
|
|
return result.Value;
|
|
|
|
if (dockClient.ReserveHost(dockHostActor, dockHost))
|
|
{
|
|
if (dockHost.QueueMoveActivity(this, dockHostActor, self, dockClient, moveCooldownHelper))
|
|
{
|
|
foreach (var ndcm in notifyDockClientMoving)
|
|
ndcm.MovingToDock(self, dockHostActor, dockHost);
|
|
|
|
return false;
|
|
}
|
|
|
|
dockHost.QueueDockActivity(this, dockHostActor, self, dockClient);
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
// The dock explicitly chosen by the user is currently occupied. Wait and check again.
|
|
QueueChild(new Wait(dockClient.Info.SearchForDockDelay));
|
|
return false;
|
|
}
|
|
}
|
|
|
|
public override void Cancel(Actor self, bool keepQueue = false)
|
|
{
|
|
dockClient.UnreserveHost();
|
|
foreach (var ndcm in notifyDockClientMoving)
|
|
ndcm.MovementCancelled(self);
|
|
|
|
base.Cancel(self, keepQueue);
|
|
}
|
|
|
|
public override IEnumerable<TargetLineNode> TargetLineNodes(Actor self)
|
|
{
|
|
if (!dockLineColor.HasValue)
|
|
yield break;
|
|
|
|
if (dockHostActor != null)
|
|
yield return new TargetLineNode(Target.FromActor(dockHostActor), dockLineColor.Value);
|
|
else
|
|
{
|
|
if (dockClient.ReservedHostActor != null)
|
|
yield return new TargetLineNode(Target.FromActor(dockClient.ReservedHostActor), dockLineColor.Value);
|
|
}
|
|
}
|
|
}
|
|
}
|