1078: Added ally repairing for CNC and RA

OpenRA.Mods.RA/Buildings/RepairableBuilding.cs

@@ -22,10 +22,10 @@ namespace OpenRA.Mods.RA.Buildings
 		public object Create(ActorInitializer init) { return new RepairableBuilding(init.self, this); }
 	}
 
-	public class RepairableBuilding : ITick, IResolveOrder, ISync
+	public class RepairableBuilding : ITick, ISync
 	{
 		[Sync]
-		bool isRepairing = false;
+		public Player Repairer = null;
 
 		Health Health;
 		RepairableBuildingInfo Info;
@@ -35,41 +35,55 @@ namespace OpenRA.Mods.RA.Buildings
 			Info = info;
 		}
 
-		public void ResolveOrder(Actor self, Order order)
+		public void RepairBuilding(Actor self, Player p)
 		{
-			if (order.OrderString == "Repair")
+			if (self.HasTrait<RepairableBuilding>())
 			{
-				isRepairing = !isRepairing;
-				if (isRepairing)
-					Sound.PlayToPlayer(self.Owner, self.World.WorldActor.Info.Traits.Get<EvaAlertsInfo>().Repairing);
+				if (self.AppearsFriendlyTo(p.PlayerActor))
+				{
+					if (Repairer == p)
+						Repairer = null;
+
+					else
+					{
+						Repairer = p;
+						Sound.PlayToPlayer(Repairer, p.World.WorldActor.Info.Traits.Get<EvaAlertsInfo>().Repairing);
+
+						self.World.AddFrameEndTask(
+							w => w.Add(new RepairIndicator(self, p)));
+					}
+				}
 			}
 		}
 
 		int remainingTicks;
 		public void Tick(Actor self)
 		{
-			if (!isRepairing) return;
+			if (Repairer == null) return;
 
 			if (remainingTicks == 0)
 			{
+				if (Repairer.WinState != WinState.Undefined)
+				{
+					Repairer = null;
+					return;
+				}
+
 				var buildingValue = self.GetSellValue();
 
 				var hpToRepair = Math.Min(Info.RepairStep, Health.MaxHP - Health.HP);
 				var cost = (hpToRepair * Info.RepairPercent * buildingValue) / (Health.MaxHP * 100);
-				if (!self.Owner.PlayerActor.Trait<PlayerResources>().TakeCash(cost))
+				if (!Repairer.PlayerActor.Trait<PlayerResources>().TakeCash(cost))
 				{
 					remainingTicks = 1;
 					return;
 				}
 
-				self.World.AddFrameEndTask(
-                    w => w.Add(new RepairIndicator(self, Info.RepairInterval / 2)));
-
 				self.InflictDamage(self, -hpToRepair, null);
 
 				if (Health.DamageState == DamageState.Undamaged)
 				{
-					isRepairing = false;
+					Repairer = null;
 					return;
 				}
 
@@ -79,7 +93,7 @@ namespace OpenRA.Mods.RA.Buildings
 				--remainingTicks;
 		}
 	}
-	public class AllowsBuildingRepairInfo : TraitInfo<AllowsBuildingRepair> {}
-	public class AllowsBuildingRepair {}
+	public class AllowsBuildingRepairInfo : TraitInfo<AllowsBuildingRepair> { }
+	public class AllowsBuildingRepair { }
 
 }

OpenRA.Mods.RA/Effects/RepairIndicator.cs

@@ -18,27 +18,37 @@ namespace OpenRA.Mods.RA.Effects
 {
 	class RepairIndicator : IEffect
 	{
-		int framesLeft;
-		Actor a;
-		Animation anim = new Animation("select");
+		Actor building;
+		Player player;
+		Animation anim = new Animation("allyrepair");
 
-		public RepairIndicator(Actor a, int frames) 
+		public RepairIndicator(Actor building, Player player)
 		{
-			this.a = a; anim.PlayRepeating("repair");
-            framesLeft = frames;
+			this.building = building;
+			this.player = player;
+			anim.PlayRepeating("repair");
 		}
 
-		public void Tick( World world )
+		public void Tick(World world)
 		{
-			if (--framesLeft == 0 || !a.IsInWorld || a.IsDead())
+			if (!building.IsInWorld ||
+				building.IsDead() ||
+				building.Trait<RepairableBuilding>().Repairer == null ||
+				building.Trait<RepairableBuilding>().Repairer != player)
 				world.AddFrameEndTask(w => w.Remove(this));
+
+			anim.Tick();
 		}
 
 		public IEnumerable<Renderable> Render()
 		{
-			if (!a.Destroyed)
+			if (!building.Destroyed)
+			{
+				var palette = building.Trait<RenderSimple>().Palette(player);
+
 				yield return new Renderable(anim.Image,
-					a.CenterLocation - .5f * anim.Image.size, "chrome", (int)a.CenterLocation.Y);
+					building.CenterLocation - .5f * anim.Image.size, palette, (int)building.CenterLocation.Y);
+			}
 		}
 	}
 }

OpenRA.Mods.RA/OpenRA.Mods.RA.csproj

@@ -72,6 +72,7 @@
     <Compile Include="Lint\LintBuildablePrerequisites.cs" />
     <Compile Include="Lint\CheckAutotargetWiring.cs" />
     <Compile Include="Lint\CheckSyncAnnotations.cs" />
+    <Compile Include="Player\AllyRepair.cs" />
     <Compile Include="ProductionBar.cs" />
     <Compile Include="Render\RenderEditorOnly.cs" />
     <Compile Include="SmokeTrailWhenDamaged.cs" />

OpenRA.Mods.RA/Orders/RepairOrderGenerator.cs

@@ -31,31 +31,31 @@ namespace OpenRA.Mods.RA.Orders
 			if (mi.Button == MouseButton.Left)
 			{
 				var underCursor = world.FindUnitsAtMouse(mi.Location)
-					.Where(a => a.Owner == world.LocalPlayer && a.HasTrait<RepairableBuilding>()).FirstOrDefault();
+					.Where(a => a.AppearsFriendlyTo(world.LocalPlayer.PlayerActor) && a.HasTrait<RepairableBuilding>()).FirstOrDefault();
 
 				if (underCursor == null)
 					yield break;
 
 				if (underCursor.Info.Traits.Contains<RepairableBuildingInfo>()
-				    && underCursor.GetDamageState() > DamageState.Undamaged)
-					yield return new Order("Repair", underCursor, false);
+					&& underCursor.GetDamageState() > DamageState.Undamaged)
+					yield return new Order("RepairBuilding", world.LocalPlayer.PlayerActor, false) { TargetActor = underCursor };
 			}
 		}
 
-		public void Tick( World world )
+		public void Tick(World world)
 		{
-			if( !PlayerIsAllowedToRepair( world ) )
+			if (!PlayerIsAllowedToRepair(world))
 				world.CancelInputMode();
 		}
 
-		public static bool PlayerIsAllowedToRepair( World world )
+		public static bool PlayerIsAllowedToRepair(World world)
 		{
 			return world.ActorsWithTrait<AllowsBuildingRepair>()
-                .Any(a => a.Actor.Owner == world.LocalPlayer);
+				 .Any(a => a.Actor.AppearsFriendlyTo(world.LocalPlayer.PlayerActor));
 		}
 
-		public void RenderAfterWorld( WorldRenderer wr, World world ) { }
-		public void RenderBeforeWorld( WorldRenderer wr, World world ) { }
+		public void RenderAfterWorld(WorldRenderer wr, World world) { }
+		public void RenderBeforeWorld(WorldRenderer wr, World world) { }
 
 		public string GetCursor(World world, int2 xy, MouseInput mi)
 		{

OpenRA.Mods.RA/Player/AllyRepair.cs

@@ -0,0 +1,35 @@
+#region Copyright & License Information
+/*
+ * Copyright 2007-2011 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;
+using OpenRA.Mods.RA.Buildings;
+using OpenRA.Mods.RA.Effects;
+using OpenRA.Traits;
+
+namespace OpenRA.Mods.RA
+{
+	class AllyRepairInfo : TraitInfo<AllyRepair> { }
+
+	class AllyRepair : IResolveOrder
+	{
+		public void ResolveOrder(Actor self, Order order)
+		{
+			if (order.OrderString == "RepairBuilding")
+			{
+				var building = order.TargetActor;
+
+				if (building.HasTrait<RepairableBuilding>())
+					if (building.AppearsFriendlyTo(self))
+						building.Trait<RepairableBuilding>().RepairBuilding(building, self.Owner);
+			}
+		}
+
+	}
+}

mods/cnc/rules/system.yaml

@@ -6,6 +6,7 @@ Player:
 		WinNotification:accom1.aud
 		LoseNotification:fail1.aud
 	PowerManager:
+	AllyRepair:
 	PlayerResources:
 		InitialCash: 5000
 	ActorGroupProxy:

mods/cnc/sequences/misc.yaml

@@ -111,6 +111,12 @@ select:
 	repair:
 		Start: 2
 
+allyrepair:
+	repair:
+		Start: 0
+		Length: *
+		Tick: 160
+
 crate:
 	idle: wcrate
 		Start: 0

mods/ra/rules/system.yaml

@@ -34,6 +34,7 @@ Player:
 		WinNotification:misnwon1.aud
 		LoseNotification:misnlst1.aud
 	PowerManager:
+	AllyRepair:
 	PlayerResources:
 		InitialCash: 5000
 	ActorGroupProxy:

mods/ra/sequences.yaml

@@ -1546,6 +1546,12 @@ poweroff:
 		Length: *
 		Tick: 160
 
+allyrepair:
+	repair:
+		Start: 0
+		Length: *
+		Tick: 160
+
 tabs:
 	left-normal:
 		Start: 0