diff --git a/OpenRA.Mods.D2k/Activities/SwallowActor.cs b/OpenRA.Mods.D2k/Activities/SwallowActor.cs index b9ee1a0a32..64371e4803 100644 --- a/OpenRA.Mods.D2k/Activities/SwallowActor.cs +++ b/OpenRA.Mods.D2k/Activities/SwallowActor.cs @@ -70,7 +70,23 @@ namespace OpenRA.Mods.D2k.Activities sandworm.IsAttacking = true; foreach (var actor in lunch) - actor.World.AddFrameEndTask(_ => actor.Destroy()); + { + var actor1 = actor; // loop variable in closure hazard + + actor.World.AddFrameEndTask(_ => + { + actor1.Destroy(); + + // Harvester insurance + if (!actor1.HasTrait()) + return; + + var insurance = actor1.Owner.PlayerActor.TraitOrDefault(); + + if (insurance != null) + actor1.World.AddFrameEndTask(__ => insurance.TryActivate()); + }); + } positionable.SetPosition(worm, targetLocation); foreach (var notify in worm.TraitsImplementing()) diff --git a/OpenRA.Mods.D2k/OpenRA.Mods.D2k.csproj b/OpenRA.Mods.D2k/OpenRA.Mods.D2k.csproj index d4eb67f98e..5981c7e3c2 100644 --- a/OpenRA.Mods.D2k/OpenRA.Mods.D2k.csproj +++ b/OpenRA.Mods.D2k/OpenRA.Mods.D2k.csproj @@ -79,6 +79,7 @@ + diff --git a/OpenRA.Mods.D2k/Traits/Player/HarvesterInsurance.cs b/OpenRA.Mods.D2k/Traits/Player/HarvesterInsurance.cs new file mode 100644 index 0000000000..2174de20ca --- /dev/null +++ b/OpenRA.Mods.D2k/Traits/Player/HarvesterInsurance.cs @@ -0,0 +1,48 @@ +#region Copyright & License Information +/* + * Copyright 2007-2015 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. For more information, + * see COPYING. + */ +#endregion + +using System.Linq; +using OpenRA.Mods.Common.Traits; +using OpenRA.Traits; + +namespace OpenRA.Mods.D2k.Traits +{ + [Desc("A player with this trait will receive a free harvester when his last one gets eaten by a sandworm, provided he has at least one refinery.")] + public class HarvesterInsuranceInfo : ITraitInfo + { + public object Create(ActorInitializer init) { return new HarvesterInsurance(init.Self); } + } + + public class HarvesterInsurance + { + readonly Actor self; + + public HarvesterInsurance(Actor self) + { + this.self = self; + } + + public void TryActivate() + { + var harvesters = self.World.ActorsWithTrait().Where(x => x.Actor.Owner == self.Owner); + if (harvesters.Any()) + return; + + var refineries = self.World.ActorsWithTrait().Where(x => x.Actor.Owner == self.Owner); + if (!refineries.Any()) + return; + + var refinery = refineries.First().Actor; + var delivery = refinery.Trait(); + delivery.DoDelivery(refinery.Location + delivery.Info.DeliveryOffset, delivery.Info.Actor, + delivery.Info.DeliveringActor, delivery.Info.InitialActivity); + } + } +} diff --git a/mods/d2k/rules/player.yaml b/mods/d2k/rules/player.yaml index 68585dcb96..4779cbd49f 100644 --- a/mods/d2k/rules/player.yaml +++ b/mods/d2k/rules/player.yaml @@ -73,3 +73,4 @@ Player: Name: Unrestricted Prerequisites: techlevel.low, techlevel.medium, techlevel.high, techlevel.superweapons EnemyWatcher: + HarvesterInsurance: