129 lines
3.7 KiB
C#
129 lines
3.7 KiB
C#
#region Copyright & License Information
|
|
/*
|
|
* Copyright 2007-2019 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, either version 3 of
|
|
* the License, or (at your option) any later version. For more
|
|
* information, see COPYING.
|
|
*/
|
|
#endregion
|
|
|
|
using System.Linq;
|
|
using OpenRA.Activities;
|
|
|
|
namespace OpenRA.Mods.Common.Traits
|
|
{
|
|
[Desc("Can be carried by units with the trait `Carryall`.")]
|
|
public class AutoCarryableInfo : CarryableInfo
|
|
{
|
|
[Desc("Required distance away from destination before requesting a pickup. Default is 6 cells.")]
|
|
public readonly WDist MinDistance = WDist.FromCells(6);
|
|
|
|
public override object Create(ActorInitializer init) { return new AutoCarryable(init.Self, this); }
|
|
}
|
|
|
|
public class AutoCarryable : Carryable, ICallForTransport
|
|
{
|
|
readonly AutoCarryableInfo info;
|
|
Activity afterLandActivity;
|
|
|
|
public AutoCarryable(Actor self, AutoCarryableInfo info)
|
|
: base(self, info)
|
|
{
|
|
this.info = info;
|
|
}
|
|
|
|
public WDist MinimumDistance { get { return info.MinDistance; } }
|
|
|
|
// No longer want to be carried
|
|
void ICallForTransport.MovementCancelled(Actor self) { MovementCancelled(self); }
|
|
void ICallForTransport.RequestTransport(Actor self, CPos destination, Activity afterLandActivity) { RequestTransport(self, destination, afterLandActivity); }
|
|
|
|
void MovementCancelled(Actor self)
|
|
{
|
|
if (state == State.Locked)
|
|
return;
|
|
|
|
Destination = null;
|
|
afterLandActivity = null;
|
|
|
|
// TODO: We could implement something like a carrier.Trait<Carryall>().CancelTransportNotify(self) and call it here
|
|
}
|
|
|
|
void RequestTransport(Actor self, CPos destination, Activity afterLandActivity)
|
|
{
|
|
var delta = self.World.Map.CenterOfCell(destination) - self.CenterPosition;
|
|
if (delta.HorizontalLengthSquared < info.MinDistance.LengthSquared)
|
|
{
|
|
Destination = null;
|
|
return;
|
|
}
|
|
|
|
Destination = destination;
|
|
this.afterLandActivity = afterLandActivity;
|
|
|
|
if (state != State.Free)
|
|
return;
|
|
|
|
// Inform all idle carriers
|
|
var carriers = self.World.ActorsWithTrait<Carryall>()
|
|
.Where(c => c.Trait.State == Carryall.CarryallState.Idle && !c.Actor.IsDead && c.Actor.Owner == self.Owner && c.Actor.IsInWorld)
|
|
.OrderBy(p => (self.Location - p.Actor.Location).LengthSquared);
|
|
|
|
// Enumerate idle carriers to find the first that is able to transport us
|
|
foreach (var carrier in carriers)
|
|
if (carrier.Trait.RequestTransportNotify(carrier.Actor, self, destination))
|
|
return;
|
|
}
|
|
|
|
// This gets called by carrier after we touched down
|
|
public override void Detached(Actor self)
|
|
{
|
|
if (!attached)
|
|
return;
|
|
|
|
Destination = null;
|
|
|
|
if (afterLandActivity != null)
|
|
self.QueueActivity(false, afterLandActivity);
|
|
|
|
base.Detached(self);
|
|
}
|
|
|
|
public override bool Reserve(Actor self, Actor carrier)
|
|
{
|
|
if (Reserved || !WantsTransport)
|
|
return false;
|
|
|
|
var delta = self.World.Map.CenterOfCell(Destination.Value) - self.CenterPosition;
|
|
if (delta.HorizontalLengthSquared < info.MinDistance.LengthSquared)
|
|
{
|
|
// Cancel pickup
|
|
MovementCancelled(self);
|
|
return false;
|
|
}
|
|
|
|
return base.Reserve(self, carrier);
|
|
}
|
|
|
|
// Prepare for transport pickup
|
|
public override bool LockForPickup(Actor self, Actor carrier)
|
|
{
|
|
if (state == State.Locked || !WantsTransport)
|
|
return false;
|
|
|
|
// Last chance to change our mind...
|
|
var delta = self.World.Map.CenterOfCell(Destination.Value) - self.CenterPosition;
|
|
if (delta.HorizontalLengthSquared < info.MinDistance.LengthSquared)
|
|
{
|
|
// Cancel pickup
|
|
MovementCancelled(self);
|
|
return false;
|
|
}
|
|
|
|
return base.LockForPickup(self, carrier);
|
|
}
|
|
}
|
|
}
|