diff --git a/OpenRA.Mods.RA/Render/WithIdleOverlay.cs b/OpenRA.Mods.RA/Render/WithIdleOverlay.cs index b4e1eb2012..ef855eeb46 100644 --- a/OpenRA.Mods.RA/Render/WithIdleOverlay.cs +++ b/OpenRA.Mods.RA/Render/WithIdleOverlay.cs @@ -49,7 +49,7 @@ namespace OpenRA.Mods.RA.Render } } - public class WithIdleOverlay : INotifyDamageStateChanged, INotifyBuildComplete, INotifySold + public class WithIdleOverlay : INotifyDamageStateChanged, INotifyBuildComplete, INotifySold, INotifyTransform { Animation overlay; bool buildComplete; @@ -83,6 +83,13 @@ namespace OpenRA.Mods.RA.Render buildComplete = false; } + public void BeforeTransform(Actor self) + { + buildComplete = false; + } + public void OnTransform(Actor self) { } + public void AfterTransform(Actor self) { } + public void DamageStateChanged(Actor self, AttackInfo e) { overlay.ReplaceAnim(RenderSprites.NormalizeSequence(overlay, e.DamageState, overlay.CurrentSequence.Name));