Merge Mods.TS into Mods.Cnc

OpenRA.Mods.Cnc/Activities/VoxelHarvesterDockSequence.cs

@@ -0,0 +1,70 @@
+#region Copyright & License Information
+/*
+ * Copyright 2007-2017 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.Activities;
+using OpenRA.Mods.Cnc.Traits.Render;
+using OpenRA.Mods.Common.Activities;
+
+namespace OpenRA.Mods.Cnc.Activities
+{
+	public class VoxelHarvesterDockSequence : HarvesterDockSequence
+	{
+		readonly WithVoxelUnloadBody body;
+		readonly WithDockingOverlay spriteOverlay;
+
+		public VoxelHarvesterDockSequence(Actor self, Actor refinery, int dockAngle, bool isDragRequired, WVec dragOffset, int dragLength)
+			: base(self, refinery, dockAngle, isDragRequired, dragOffset, dragLength)
+		{
+			body = self.Trait<WithVoxelUnloadBody>();
+			spriteOverlay = refinery.TraitOrDefault<WithDockingOverlay>();
+		}
+
+		public override Activity OnStateDock(Actor self)
+		{
+			body.Docked = true;
+
+			if (spriteOverlay != null && !spriteOverlay.Visible)
+			{
+				spriteOverlay.Visible = true;
+				spriteOverlay.WithOffset.Animation.PlayThen(spriteOverlay.Info.Sequence, () => {
+					dockingState = DockingState.Loop;
+					spriteOverlay.Visible = false;
+				});
+			}
+			else
+				dockingState = DockingState.Loop;
+
+			return this;
+		}
+
+		public override Activity OnStateUndock(Actor self)
+		{
+			dockingState = DockingState.Wait;
+
+			if (spriteOverlay != null && !spriteOverlay.Visible)
+			{
+				spriteOverlay.Visible = true;
+				spriteOverlay.WithOffset.Animation.PlayBackwardsThen(spriteOverlay.Info.Sequence, () => {
+					dockingState = DockingState.Complete;
+					body.Docked = false;
+					spriteOverlay.Visible = false;
+				});
+			}
+			else
+			{
+				dockingState = DockingState.Complete;
+				body.Docked = false;
+			}
+
+			return this;
+		}
+	}
+}