Move JumpJet traits to the C&C library.

OpenRA.Mods.Cnc/Traits/World/JumpjetActorLayer.cs

@@ -0,0 +1,108 @@
+#region Copyright & License Information
+/*
+ * Copyright (c) The OpenRA Developers and Contributors
+ * 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.Pathfinder;
+using OpenRA.Mods.Common.Traits;
+using OpenRA.Traits;
+
+namespace OpenRA.Mods.Cnc.Traits
+{
+	[TraitLocation(SystemActors.World)]
+	public class JumpjetActorLayerInfo : TraitInfo, ICustomMovementLayerInfo
+	{
+		[Desc("Terrain type of the airborne layer.")]
+		public readonly string TerrainType = "Jumpjet";
+
+		[Desc("Height offset relative to the smoothed terrain for movement.")]
+		public readonly WDist HeightOffset = new WDist(3992);
+
+		[Desc("Cell radius for smoothing adjacent cell heights.")]
+		public readonly int SmoothingRadius = 2;
+
+		public override object Create(ActorInitializer init) { return new JumpjetActorLayer(init.Self, this); }
+	}
+
+	public class JumpjetActorLayer : ICustomMovementLayer
+	{
+		readonly Map map;
+
+		readonly byte terrainIndex;
+		readonly CellLayer<int> height;
+
+		public JumpjetActorLayer(Actor self, JumpjetActorLayerInfo info)
+		{
+			map = self.World.Map;
+			terrainIndex = self.World.Map.Rules.TerrainInfo.GetTerrainIndex(info.TerrainType);
+			height = new CellLayer<int>(map);
+			var cellHeight = self.World.Map.CellHeightStep.Length;
+			foreach (var c in map.AllCells)
+			{
+				var neighbourCount = 0;
+				var neighbourHeight = 0;
+				for (var dy = -info.SmoothingRadius; dy <= info.SmoothingRadius; dy++)
+				{
+					for (var dx = -info.SmoothingRadius; dx <= info.SmoothingRadius; dx++)
+					{
+						var neighbour = c + new CVec(dx, dy);
+						if (!map.AllCells.Contains(neighbour))
+							continue;
+
+						neighbourCount++;
+						neighbourHeight += map.Height[neighbour];
+					}
+				}
+
+				height[c] = info.HeightOffset.Length + neighbourHeight * cellHeight / neighbourCount;
+			}
+		}
+
+		bool ICustomMovementLayer.EnabledForLocomotor(LocomotorInfo li) { return li is JumpjetLocomotorInfo; }
+		byte ICustomMovementLayer.Index => CustomMovementLayerType.Jumpjet;
+		bool ICustomMovementLayer.InteractsWithDefaultLayer => true;
+		bool ICustomMovementLayer.ReturnToGroundLayerOnIdle => true;
+
+		WPos ICustomMovementLayer.CenterOfCell(CPos cell)
+		{
+			var pos = map.CenterOfCell(cell);
+			return pos + new WVec(0, 0, height[cell] - pos.Z);
+		}
+
+		bool ValidTransitionCell(CPos cell, LocomotorInfo li)
+		{
+			var terrainType = map.GetTerrainInfo(cell).Type;
+			var jli = (JumpjetLocomotorInfo)li;
+			if (!jli.JumpjetTransitionTerrainTypes.Contains(terrainType) && jli.JumpjetTransitionTerrainTypes.Count > 0)
+				return false;
+
+			if (jli.JumpjetTransitionOnRamps)
+				return true;
+
+			return map.Ramp[cell] == 0;
+		}
+
+		short ICustomMovementLayer.EntryMovementCost(LocomotorInfo li, CPos cell)
+		{
+			var jli = (JumpjetLocomotorInfo)li;
+			return ValidTransitionCell(cell, jli) ? jli.JumpjetTransitionCost : PathGraph.MovementCostForUnreachableCell;
+		}
+
+		short ICustomMovementLayer.ExitMovementCost(LocomotorInfo li, CPos cell)
+		{
+			var jli = (JumpjetLocomotorInfo)li;
+			return ValidTransitionCell(cell, jli) ? jli.JumpjetTransitionCost : PathGraph.MovementCostForUnreachableCell;
+		}
+
+		byte ICustomMovementLayer.GetTerrainIndex(CPos cell)
+		{
+			return terrainIndex;
+		}
+	}
+}