117 lines
3.5 KiB
C#
117 lines
3.5 KiB
C#
#region Copyright & License Information
|
|
/*
|
|
* Copyright 2007-2022 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.Collections.Generic;
|
|
using OpenRA.Activities;
|
|
using OpenRA.Mods.Common.Traits;
|
|
using OpenRA.Primitives;
|
|
using OpenRA.Traits;
|
|
|
|
namespace OpenRA.Mods.Common.Activities
|
|
{
|
|
public class DeliverUnit : Activity
|
|
{
|
|
readonly Carryall carryall;
|
|
readonly BodyOrientation body;
|
|
readonly bool assignTargetOnFirstRun;
|
|
readonly WDist deliverRange;
|
|
readonly Color? targetLineColor;
|
|
|
|
Target destination;
|
|
|
|
public DeliverUnit(Actor self, WDist deliverRange, Color? targetLineColor)
|
|
: this(self, Target.Invalid, deliverRange, targetLineColor)
|
|
{
|
|
assignTargetOnFirstRun = true;
|
|
}
|
|
|
|
public DeliverUnit(Actor self, in Target destination, WDist deliverRange, Color? targetLineColor)
|
|
{
|
|
this.destination = destination;
|
|
this.deliverRange = deliverRange;
|
|
this.targetLineColor = targetLineColor;
|
|
|
|
carryall = self.Trait<Carryall>();
|
|
body = self.Trait<BodyOrientation>();
|
|
}
|
|
|
|
protected override void OnFirstRun(Actor self)
|
|
{
|
|
// In case this activity was queued (either via queued order of via AutoCarryall)
|
|
// something might have happened to the cargo in the time between the activity being
|
|
// queued and being run, so short out if it is no longer valid.
|
|
if (carryall.Carryable == null)
|
|
return;
|
|
|
|
if (assignTargetOnFirstRun)
|
|
destination = Target.FromCell(self.World, self.Location);
|
|
|
|
QueueChild(new Land(self, destination, deliverRange));
|
|
QueueChild(new Wait(carryall.Info.BeforeUnloadDelay, false));
|
|
QueueChild(new ReleaseUnit(self));
|
|
QueueChild(new TakeOff(self));
|
|
}
|
|
|
|
public override IEnumerable<TargetLineNode> TargetLineNodes(Actor self)
|
|
{
|
|
if (targetLineColor != null)
|
|
yield return new TargetLineNode(destination, targetLineColor.Value);
|
|
}
|
|
|
|
class ReleaseUnit : Activity
|
|
{
|
|
readonly Carryall carryall;
|
|
readonly BodyOrientation body;
|
|
readonly IFacing facing;
|
|
|
|
public ReleaseUnit(Actor self)
|
|
{
|
|
facing = self.Trait<IFacing>();
|
|
carryall = self.Trait<Carryall>();
|
|
body = self.Trait<BodyOrientation>();
|
|
}
|
|
|
|
protected override void OnFirstRun(Actor self)
|
|
{
|
|
// HACK: Activities still tick between the actor being killed and being disposed
|
|
// Thus the carryable might have changed since queuing because the death handler set it to null
|
|
if (carryall.Carryable == null)
|
|
return;
|
|
|
|
self.Trait<Aircraft>().RemoveInfluence();
|
|
|
|
var localOffset = carryall.CarryableOffset.Rotate(body.QuantizeOrientation(self.Orientation));
|
|
var targetPosition = self.CenterPosition + body.LocalToWorld(localOffset);
|
|
var targetLocation = self.World.Map.CellContaining(targetPosition);
|
|
carryall.Carryable.Trait<IPositionable>().SetPosition(carryall.Carryable, targetLocation, SubCell.FullCell);
|
|
carryall.Carryable.Trait<IFacing>().Facing = facing.Facing;
|
|
|
|
// Put back into world
|
|
self.World.AddFrameEndTask(w =>
|
|
{
|
|
if (self.IsDead)
|
|
return;
|
|
|
|
var cargo = carryall.Carryable;
|
|
if (cargo == null)
|
|
return;
|
|
|
|
var carryable = cargo.Trait<Carryable>();
|
|
w.Add(cargo);
|
|
carryall.DetachCarryable(self);
|
|
carryable.UnReserve(cargo);
|
|
carryable.Detached(cargo);
|
|
});
|
|
}
|
|
}
|
|
}
|
|
}
|