57d3321d0f
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.
74 lines
2.2 KiB
C#
74 lines
2.2 KiB
C#
#region Copyright & License Information
|
|
/*
|
|
* Copyright 2007-2021 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 OpenRA.Mods.Common.Traits;
|
|
using OpenRA.Mods.Common.Traits.Render;
|
|
|
|
namespace OpenRA.Mods.Common.Activities
|
|
{
|
|
public class SpriteHarvesterDockSequence : HarvesterDockSequence
|
|
{
|
|
readonly WithSpriteBody wsb;
|
|
readonly WithDockingAnimationInfo wda;
|
|
protected bool dockAnimPlayed;
|
|
|
|
public SpriteHarvesterDockSequence(Actor self, Actor refinery, WAngle dockAngle, bool isDragRequired, in WVec dragOffset, int dragLength)
|
|
: base(self, refinery, dockAngle, isDragRequired, dragOffset, dragLength)
|
|
{
|
|
wsb = self.Trait<WithSpriteBody>();
|
|
wda = self.Info.TraitInfoOrDefault<WithDockingAnimationInfo>();
|
|
}
|
|
|
|
public override void OnStateDock(Actor self)
|
|
{
|
|
foreach (var trait in self.TraitsImplementing<INotifyHarvesterAction>())
|
|
trait.Docked();
|
|
|
|
foreach (var nd in Refinery.TraitsImplementing<INotifyDocking>())
|
|
nd.Docked(Refinery, self);
|
|
|
|
if (wda != null)
|
|
wsb.PlayCustomAnimation(self, wda.DockSequence, () => wsb.PlayCustomAnimationRepeating(self, wda.DockLoopSequence));
|
|
|
|
dockAnimPlayed = true;
|
|
dockingState = DockingState.Loop;
|
|
}
|
|
|
|
public override void OnStateUndock(Actor self)
|
|
{
|
|
// If dock animation hasn't played, we didn't actually dock and have to skip the undock anim and notification
|
|
if (!dockAnimPlayed)
|
|
{
|
|
dockingState = DockingState.Complete;
|
|
return;
|
|
}
|
|
|
|
dockingState = DockingState.Wait;
|
|
|
|
if (wda == null)
|
|
NotifyUndock(self);
|
|
else
|
|
wsb.PlayCustomAnimationBackwards(self, wda.DockSequence, () => NotifyUndock(self));
|
|
}
|
|
|
|
void NotifyUndock(Actor self)
|
|
{
|
|
dockingState = DockingState.Complete;
|
|
foreach (var trait in self.TraitsImplementing<INotifyHarvesterAction>())
|
|
trait.Undocked();
|
|
|
|
if (Refinery.IsInWorld && !Refinery.IsDead)
|
|
foreach (var nd in Refinery.TraitsImplementing<INotifyDocking>())
|
|
nd.Undocked(Refinery, self);
|
|
}
|
|
}
|
|
}
|