Create abstract class AffectsShroud[Info] with RevealsShroud core logic also used by CreatesShroud

2015-09-07 04:39:46 -05:00
parent aae3c8ef7d
commit 3e211232a3
4 changed files with 114 additions and 97 deletions
--- a/OpenRA.Mods.Common/Traits/AffectsShroud.cs
+++ b/OpenRA.Mods.Common/Traits/AffectsShroud.cs
@@ -0,0 +1,101 @@
+#region Copyright & License Information
+/*
+ * Copyright 2007-2015 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. For more information,
+ * see COPYING.
+ */
+#endregion
+
+using System;
+using System.Linq;
+using OpenRA.Traits;
+
+namespace OpenRA.Mods.Common.Traits
+{
+	public abstract class AffectsShroudInfo : ITraitInfo
+	{
+		public readonly WDist Range = WDist.Zero;
+
+		[Desc("Possible values are CenterPosition (measure range from the center) and ",
+			"Footprint (measure range from the footprint)")]
+		public readonly VisibilityType Type = VisibilityType.Footprint;
+
+		public abstract object Create(ActorInitializer init);
+	}
+
+	public abstract class AffectsShroud : ITick, ISync, INotifyAddedToWorld, INotifyRemovedFromWorld
+	{
+		static readonly PPos[] NoCells = { };
+
+		readonly AffectsShroudInfo info;
+		[Sync] CPos cachedLocation;
+		[Sync] bool cachedDisabled;
+
+		protected abstract void AddCellsToPlayerShroud(Actor self, Player player, PPos[] uv);
+		protected abstract void RemoveCellsFromPlayerShroud(Actor self, Player player);
+		protected virtual bool IsDisabled(Actor self) { return false; }
+
+		public AffectsShroud(Actor self, AffectsShroudInfo info) { this.info = info; }
+
+		PPos[] ProjectedCells(Actor self)
+		{
+			var map = self.World.Map;
+			var range = Range;
+			if (range == WDist.Zero)
+				return NoCells;
+
+			if (info.Type == VisibilityType.Footprint)
+				return self.OccupiesSpace.OccupiedCells()
+					.SelectMany(kv => Shroud.ProjectedCellsInRange(map, kv.First, range))
+					.Distinct().ToArray();
+
+			return Shroud.ProjectedCellsInRange(map, self.CenterPosition, range)
+				.ToArray();
+		}
+
+		public void Tick(Actor self)
+		{
+			if (!self.IsInWorld)
+				return;
+
+			var centerPosition = self.CenterPosition;
+			var projectedPos = centerPosition - new WVec(0, centerPosition.Z, centerPosition.Z);
+			var projectedLocation = self.World.Map.CellContaining(projectedPos);
+			var disabled = IsDisabled(self);
+
+			if (cachedLocation == projectedLocation && cachedDisabled == disabled)
+				return;
+
+			cachedLocation = projectedLocation;
+			cachedDisabled = disabled;
+
+			var cells = ProjectedCells(self);
+			foreach (var p in self.World.Players)
+			{
+				RemoveCellsFromPlayerShroud(self, p);
+				AddCellsToPlayerShroud(self, p, cells);
+			}
+		}
+
+		public void AddedToWorld(Actor self)
+		{
+			var centerPosition = self.CenterPosition;
+			var projectedPos = centerPosition - new WVec(0, centerPosition.Z, centerPosition.Z);
+			cachedLocation = self.World.Map.CellContaining(projectedPos);
+			cachedDisabled = IsDisabled(self);
+			var cells = ProjectedCells(self);
+			foreach (var p in self.World.Players)
+				AddCellsToPlayerShroud(self, p, cells);
+		}
+
+		public void RemovedFromWorld(Actor self)
+		{
+			foreach (var p in self.World.Players)
+				RemoveCellsFromPlayerShroud(self, p);
+		}
+
+		public WDist Range { get { return cachedDisabled ? WDist.Zero : info.Range; } }
+	}
+}