Add WithHarvesterSpriteBody

And move PrefixByFullness there.
Also put it into Mods.Cnc, as RA is the only shipping mod
using this.

OpenRA.Mods.Common/OpenRA.Mods.Common.csproj

@@ -615,6 +615,7 @@
     <Compile Include="UpdateRules\Rules\20190314\MakeMobilePausableConditional.cs" />
     <Compile Include="UpdateRules\Rules\20190314\MultipleDeploySounds.cs" />
     <Compile Include="UpdateRules\Rules\20190314\RemoveSimpleBeacon.cs" />
+    <Compile Include="UpdateRules\Rules\20190314\SplitHarvesterSpriteBody.cs" />
     <Compile Include="UtilityCommands\CheckRuntimeAssembliesCommand.cs" />
     <Compile Include="UtilityCommands\CheckYaml.cs" />
     <Compile Include="UtilityCommands\ConvertPngToShpCommand.cs" />

OpenRA.Mods.Common/Traits/Render/WithHarvestAnimation.cs

@@ -17,9 +17,6 @@ namespace OpenRA.Mods.Common.Traits.Render
 {
 	public class WithHarvestAnimationInfo : ITraitInfo, Requires<WithSpriteBodyInfo>, Requires<HarvesterInfo>
 	{
-		[Desc("Prefix added to idle and harvest sequences depending on fullness of harvester.")]
-		[SequenceReference(null, true)] public readonly string[] PrefixByFullness = { "" };
-
 		[Desc("Displayed while harvesting.")]
 		[SequenceReference] public readonly string HarvestSequence = "harvest";
 
@@ -29,64 +26,26 @@ namespace OpenRA.Mods.Common.Traits.Render
 		public object Create(ActorInitializer init) { return new WithHarvestAnimation(init, this); }
 	}
 
-	public class WithHarvestAnimation : ITick, INotifyHarvesterAction
+	public class WithHarvestAnimation : INotifyHarvesterAction
 	{
 		public readonly WithHarvestAnimationInfo Info;
 		readonly WithSpriteBody wsb;
-		readonly Harvester harv;
-
-		// TODO: Remove this once WithSpriteBody has its own replacement
-		public bool IsModifying;
 
 		public WithHarvestAnimation(ActorInitializer init, WithHarvestAnimationInfo info)
 		{
 			Info = info;
-			harv = init.Self.Trait<Harvester>();
 			wsb = init.Self.TraitsImplementing<WithSpriteBody>().Single(w => w.Info.Name == Info.Body);
 		}
 
-		protected virtual string NormalizeHarvesterSequence(Actor self, string baseSequence)
-		{
-			var desiredState = harv.Fullness * (Info.PrefixByFullness.Length - 1) / 100;
-			var desiredPrefix = Info.PrefixByFullness[desiredState];
-
-			if (wsb.DefaultAnimation.HasSequence(desiredPrefix + baseSequence))
-				return desiredPrefix + baseSequence;
-			else
-				return baseSequence;
-		}
-
-		void ITick.Tick(Actor self)
-		{
-			var baseSequence = wsb.NormalizeSequence(self, wsb.Info.Sequence);
-			var sequence = NormalizeHarvesterSequence(self, baseSequence);
-			if (!IsModifying && wsb.DefaultAnimation.HasSequence(sequence) && wsb.DefaultAnimation.CurrentSequence.Name != sequence)
-				wsb.DefaultAnimation.ReplaceAnim(sequence);
-		}
-
 		void INotifyHarvesterAction.Harvested(Actor self, ResourceType resource)
 		{
-			var baseSequence = wsb.NormalizeSequence(self, Info.HarvestSequence);
-			var sequence = NormalizeHarvesterSequence(self, baseSequence);
-			if (!IsModifying && wsb.DefaultAnimation.HasSequence(sequence))
-			{
-				IsModifying = true;
-				wsb.PlayCustomAnimation(self, sequence, () => IsModifying = false);
-			}
-		}
-
-		// If IsModifying isn't set to true, the docking animation
-		// will be overridden by the WithHarvestAnimation fullness modifier.
-		void INotifyHarvesterAction.Docked()
-		{
-			IsModifying = true;
-		}
-
-		void INotifyHarvesterAction.Undocked()
-		{
-			IsModifying = false;
+			var sequence = wsb.NormalizeSequence(self, Info.HarvestSequence);
+			if (wsb.DefaultAnimation.HasSequence(sequence) && wsb.DefaultAnimation.CurrentSequence.Name != sequence)
+				wsb.PlayCustomAnimation(self, sequence);
 		}
 
+		void INotifyHarvesterAction.Docked() { }
+		void INotifyHarvesterAction.Undocked() { }
 		void INotifyHarvesterAction.MovingToResources(Actor self, CPos targetCell, Activity next) { }
 		void INotifyHarvesterAction.MovingToRefinery(Actor self, Actor refineryActor, Activity next) { }
 		void INotifyHarvesterAction.MovementCancelled(Actor self) { }

OpenRA.Mods.Common/UpdateRules/Rules/20190314/SplitHarvesterSpriteBody.cs

@@ -0,0 +1,64 @@
+#region Copyright & License Information
+/*
+ * Copyright 2007-2019 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;
+using System.Collections.Generic;
+using System.Linq;
+
+namespace OpenRA.Mods.Common.UpdateRules.Rules
+{
+	public class SplitHarvesterSpriteBody : UpdateRule
+	{
+		public override string Name { get { return "Split fullness display from WithHarvestAnimation to new WithHarvesterSpriteBody"; } }
+		public override string Description
+		{
+			get
+			{
+				return "WithHarvestAnimation.PrefixByFullness logic was moved to a dedicated WithHarvesterSpriteBody.";
+			}
+		}
+
+		readonly List<Tuple<string, string>> fullnessPrefixes = new List<Tuple<string, string>>();
+
+		public override IEnumerable<string> AfterUpdate(ModData modData)
+		{
+			var message = "PrefixByFullness has been removed from WithHarvestAnimation.\n"
+				+ "To display fullness levels, use the new WithHarvesterSpriteBody\n"
+				+ "to switch between separate image sprites instead (see RA mod harvester for reference).\n"
+				+ "The following places most likely need manual changes:\n"
+				+ UpdateUtils.FormatMessageList(fullnessPrefixes.Select(n => n.Item1 + " (" + n.Item2 + ")"));
+
+			if (fullnessPrefixes.Any())
+				yield return message;
+
+			fullnessPrefixes.Clear();
+		}
+
+		public override IEnumerable<string> UpdateActorNode(ModData modData, MiniYamlNode actorNode)
+		{
+			var harvAnim = actorNode.LastChildMatching("WithHarvestAnimation");
+			if (harvAnim != null)
+			{
+				var fullnessPrefix = harvAnim.LastChildMatching("PrefixByFullness");
+
+				// If PrefixByFullness is empty, no changes are needed.
+				if (fullnessPrefix == null)
+					yield break;
+
+				harvAnim.RemoveNode(fullnessPrefix);
+
+				fullnessPrefixes.Add(Tuple.Create(actorNode.Key, actorNode.Location.Filename));
+			}
+
+			yield break;
+		}
+	}
+}

OpenRA.Mods.Common/UpdateRules/UpdatePath.cs

@@ -124,6 +124,7 @@ namespace OpenRA.Mods.Common.UpdateRules
 				new StreamlineRepairableTraits(),
 				new ReplaceSpecialMoveConsiderations(),
 				new RefactorHarvesterIdle(),
+				new SplitHarvesterSpriteBody(),
 			})
 		};