diff --git a/OpenRA.Mods.D2k/Traits/Render/WithBuildingPlacedOverlay.cs b/OpenRA.Mods.D2k/Traits/Render/WithBuildingPlacedOverlay.cs index 92fe2c792e..3b1a90aa6b 100644 --- a/OpenRA.Mods.D2k/Traits/Render/WithBuildingPlacedOverlay.cs +++ b/OpenRA.Mods.D2k/Traits/Render/WithBuildingPlacedOverlay.cs @@ -33,7 +33,7 @@ namespace OpenRA.Mods.D2k.Traits public object Create(ActorInitializer init) { return new WithBuildingPlacedOverlay(init.Self, this); } } - public class WithBuildingPlacedOverlay : INotifyBuildComplete, INotifySold, INotifyDamageStateChanged, INotifyBuildingPlaced + public class WithBuildingPlacedOverlay : INotifyBuildComplete, INotifySold, INotifyDamageStateChanged, INotifyBuildingPlaced, INotifyTransform { Animation overlay; bool buildComplete; @@ -65,6 +65,14 @@ namespace OpenRA.Mods.D2k.Traits 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));