118 lines
3.4 KiB
C#
118 lines
3.4 KiB
C#
#region Copyright & License Information
|
|
/*
|
|
* Copyright 2007-2016 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 OpenRA.Mods.Common.Activities;
|
|
using OpenRA.Traits;
|
|
|
|
namespace OpenRA.Mods.Common.Traits
|
|
{
|
|
[Desc("Can be carried by actors with the `Carryall` trait.")]
|
|
public class CarryableInfo : UpgradableTraitInfo
|
|
{
|
|
[GrantedConditionReference]
|
|
[Desc("The condition to grant to self while a carryall has been reserved.")]
|
|
public readonly string ReservedCondition = null;
|
|
|
|
[GrantedConditionReference]
|
|
[Desc("The condition to grant to self while being carried.")]
|
|
public readonly string CarriedCondition = null;
|
|
|
|
[Desc("Carryall attachment point relative to body.")]
|
|
public readonly WVec LocalOffset = WVec.Zero;
|
|
|
|
public override object Create(ActorInitializer init) { return new Carryable(init.Self, this); }
|
|
}
|
|
|
|
public class Carryable : UpgradableTrait<CarryableInfo>
|
|
{
|
|
ConditionManager upgradeManager;
|
|
int reservedToken = ConditionManager.InvalidConditionToken;
|
|
int carriedToken = ConditionManager.InvalidConditionToken;
|
|
|
|
public Actor Carrier { get; private set; }
|
|
public bool Reserved { get { return state != State.Free; } }
|
|
public CPos? Destination { get; protected set; }
|
|
public bool WantsTransport { get { return Destination != null && !IsTraitDisabled; } }
|
|
|
|
protected enum State { Free, Reserved, Locked }
|
|
protected State state = State.Free;
|
|
protected bool attached;
|
|
|
|
public Carryable(Actor self, CarryableInfo info)
|
|
: base(info) { }
|
|
|
|
protected override void Created(Actor self)
|
|
{
|
|
upgradeManager = self.Trait<ConditionManager>();
|
|
|
|
base.Created(self);
|
|
}
|
|
|
|
public virtual void Attached(Actor self)
|
|
{
|
|
if (attached)
|
|
return;
|
|
|
|
attached = true;
|
|
|
|
if (carriedToken == ConditionManager.InvalidConditionToken && !string.IsNullOrEmpty(Info.CarriedCondition))
|
|
carriedToken = upgradeManager.GrantCondition(self, Info.CarriedCondition);
|
|
}
|
|
|
|
// This gets called by carrier after we touched down
|
|
public virtual void Detached(Actor self)
|
|
{
|
|
if (!attached)
|
|
return;
|
|
|
|
attached = false;
|
|
|
|
if (carriedToken != ConditionManager.InvalidConditionToken)
|
|
carriedToken = upgradeManager.RevokeCondition(self, carriedToken);
|
|
}
|
|
|
|
public virtual bool Reserve(Actor self, Actor carrier)
|
|
{
|
|
if (Reserved || IsTraitDisabled)
|
|
return false;
|
|
|
|
state = State.Reserved;
|
|
Carrier = carrier;
|
|
|
|
if (reservedToken == ConditionManager.InvalidConditionToken && !string.IsNullOrEmpty(Info.ReservedCondition))
|
|
reservedToken = upgradeManager.GrantCondition(self, Info.ReservedCondition);
|
|
|
|
return true;
|
|
}
|
|
|
|
public virtual void UnReserve(Actor self)
|
|
{
|
|
state = State.Free;
|
|
Carrier = null;
|
|
|
|
if (reservedToken != ConditionManager.InvalidConditionToken)
|
|
reservedToken = upgradeManager.RevokeCondition(self, reservedToken);
|
|
}
|
|
|
|
// Prepare for transport pickup
|
|
public virtual bool LockForPickup(Actor self, Actor carrier)
|
|
{
|
|
if (state == State.Locked)
|
|
return false;
|
|
|
|
state = State.Locked;
|
|
Carrier = carrier;
|
|
self.QueueActivity(false, new WaitFor(() => state != State.Locked, false));
|
|
return true;
|
|
}
|
|
}
|
|
}
|