Make WithDockingAnimation optional

The `Refinery` trait has a hardcoded usage of `SpriteHarvesterDockSequence`, which requires the harvester to have `WithDockingAnimation`, making it inconvenient-at-best to NOT have a docking/unloading animation.
This commit is contained in:
penev92
2022-02-23 19:47:18 +02:00
committed by abcdefg30
parent fa8bfc6ca0
commit 57d3321d0f

View File

@@ -24,17 +24,20 @@ namespace OpenRA.Mods.Common.Activities
: base(self, refinery, dockAngle, isDragRequired, dragOffset, dragLength) : base(self, refinery, dockAngle, isDragRequired, dragOffset, dragLength)
{ {
wsb = self.Trait<WithSpriteBody>(); wsb = self.Trait<WithSpriteBody>();
wda = self.Info.TraitInfo<WithDockingAnimationInfo>(); wda = self.Info.TraitInfoOrDefault<WithDockingAnimationInfo>();
} }
public override void OnStateDock(Actor self) public override void OnStateDock(Actor self)
{ {
foreach (var trait in self.TraitsImplementing<INotifyHarvesterAction>()) foreach (var trait in self.TraitsImplementing<INotifyHarvesterAction>())
trait.Docked(); trait.Docked();
foreach (var nd in Refinery.TraitsImplementing<INotifyDocking>()) foreach (var nd in Refinery.TraitsImplementing<INotifyDocking>())
nd.Docked(Refinery, self); nd.Docked(Refinery, self);
if (wda != null)
wsb.PlayCustomAnimation(self, wda.DockSequence, () => wsb.PlayCustomAnimationRepeating(self, wda.DockLoopSequence)); wsb.PlayCustomAnimation(self, wda.DockSequence, () => wsb.PlayCustomAnimationRepeating(self, wda.DockLoopSequence));
dockAnimPlayed = true; dockAnimPlayed = true;
dockingState = DockingState.Loop; dockingState = DockingState.Loop;
} }
@@ -49,8 +52,14 @@ namespace OpenRA.Mods.Common.Activities
} }
dockingState = DockingState.Wait; dockingState = DockingState.Wait;
wsb.PlayCustomAnimationBackwards(self, wda.DockSequence,
() => if (wda == null)
NotifyUndock(self);
else
wsb.PlayCustomAnimationBackwards(self, wda.DockSequence, () => NotifyUndock(self));
}
void NotifyUndock(Actor self)
{ {
dockingState = DockingState.Complete; dockingState = DockingState.Complete;
foreach (var trait in self.TraitsImplementing<INotifyHarvesterAction>()) foreach (var trait in self.TraitsImplementing<INotifyHarvesterAction>())
@@ -59,7 +68,6 @@ namespace OpenRA.Mods.Common.Activities
if (Refinery.IsInWorld && !Refinery.IsDead) if (Refinery.IsInWorld && !Refinery.IsDead)
foreach (var nd in Refinery.TraitsImplementing<INotifyDocking>()) foreach (var nd in Refinery.TraitsImplementing<INotifyDocking>())
nd.Undocked(Refinery, self); nd.Undocked(Refinery, self);
});
} }
} }
} }