Added GrantConditionOnMinelaying trait

Uses INotifyMineLaying callbacks
This commit is contained in:
michaeldgg2
2023-07-27 17:52:42 +02:00
committed by Gustas
parent 74f8db0578
commit d907192be0

View File

@@ -0,0 +1,44 @@
using OpenRA.Traits;
namespace OpenRA.Mods.Common.Traits
{
public class GrantConditionOnMinelayingInfo : ConditionalTraitInfo, Requires<MinelayerInfo>
{
[FieldLoader.Require]
[GrantedConditionReference]
[Desc("Condition to grant.")]
public readonly string Condition = null;
public override object Create(ActorInitializer init)
{
return new GrantConditionOnMinelaying(this);
}
}
public class GrantConditionOnMinelaying : ConditionalTrait<GrantConditionOnMinelayingInfo>, INotifyMineLaying
{
int conditionToken = Actor.InvalidConditionToken;
public GrantConditionOnMinelaying(GrantConditionOnMinelayingInfo info)
: base(info)
{
}
void INotifyMineLaying.MineLaid(Actor self, Actor mine)
{
if (conditionToken != Actor.InvalidConditionToken)
conditionToken = self.RevokeCondition(conditionToken);
}
void INotifyMineLaying.MineLaying(Actor self, CPos location)
{
conditionToken = self.GrantCondition(Info.Condition);
}
void INotifyMineLaying.MineLayingCanceled(Actor self, CPos location)
{
if (conditionToken != Actor.InvalidConditionToken)
conditionToken = self.RevokeCondition(conditionToken);
}
}
}