Add a trait PlayerRadarTerrain to track explored terrain

OpenRA.Mods.Common/Traits/Player/PlayerRadarTerrain.cs

@@ -0,0 +1,119 @@
+#region Copyright & License Information
+/*
+ * Copyright 2007-2019 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 System;
+using System.Collections.Generic;
+using OpenRA.Graphics;
+using OpenRA.Primitives;
+using OpenRA.Traits;
+
+namespace OpenRA.Mods.Common.Traits
+{
+	public class PlayerRadarTerrainInfo : ITraitInfo, Requires<ShroudInfo>
+	{
+		public object Create(ActorInitializer init)
+		{
+			return new PlayerRadarTerrain(init.Self);
+		}
+	}
+
+	public class PlayerRadarTerrain : IWorldLoaded
+	{
+		public bool IsInitialized { get; private set; }
+
+		readonly World world;
+		CellLayer<Pair<int, int>> terrainColor;
+		readonly HashSet<MPos> dirtyTerrainCells = new HashSet<MPos>();
+		readonly Shroud shroud;
+
+		public event Action<MPos> CellTerrainColorChanged = null;
+
+		public PlayerRadarTerrain(Actor self)
+		{
+			world = self.World;
+			shroud = self.Trait<Shroud>();
+			shroud.CellsChanged += OnShroudCellsChanged;
+		}
+
+		void OnShroudCellsChanged(IEnumerable<PPos> puvs)
+		{
+			foreach (var puv in puvs)
+			{
+				foreach (var uv in world.Map.Unproject(puv))
+				{
+					if (dirtyTerrainCells.Contains(uv))
+					{
+						UpdateTerrainCellColor(uv);
+						dirtyTerrainCells.Remove(uv);
+					}
+				}
+			}
+		}
+
+		void UpdateTerrainCell(CPos cell)
+		{
+			var uv = cell.ToMPos(world.Map);
+			if (!world.Map.CustomTerrain.Contains(uv))
+				return;
+
+			if (!shroud.IsVisible(uv))
+				dirtyTerrainCells.Add(uv);
+			else
+				UpdateTerrainCellColor(uv);
+		}
+
+		void UpdateTerrainCellColor(MPos uv)
+		{
+			terrainColor[uv] = GetColor(world.Map, uv);
+
+			if (CellTerrainColorChanged != null)
+				CellTerrainColorChanged(uv);
+		}
+
+		public void WorldLoaded(World w, WorldRenderer wr)
+		{
+			terrainColor = new CellLayer<Pair<int, int>>(w.Map);
+
+			w.AddFrameEndTask(_ =>
+			{
+				// Set initial terrain data
+				foreach (var uv in world.Map.AllCells.MapCoords)
+					UpdateTerrainCellColor(uv);
+
+				world.Map.Tiles.CellEntryChanged += UpdateTerrainCell;
+				world.Map.CustomTerrain.CellEntryChanged += UpdateTerrainCell;
+
+				IsInitialized = true;
+			});
+		}
+
+		public Pair<int, int> this[MPos uv]
+		{
+			get { return terrainColor[uv]; }
+		}
+
+		public static Pair<int, int> GetColor(Map map, MPos uv)
+		{
+			var custom = map.CustomTerrain[uv];
+			int leftColor, rightColor;
+			if (custom == byte.MaxValue)
+			{
+				var type = map.Rules.TileSet.GetTileInfo(map.Tiles[uv]);
+				leftColor = type != null ? type.LeftColor.ToArgb() : Color.Black.ToArgb();
+				rightColor = type != null ? type.RightColor.ToArgb() : Color.Black.ToArgb();
+			}
+			else
+				leftColor = rightColor = map.Rules.TileSet[custom].Color.ToArgb();
+
+			return Pair.New(leftColor, rightColor);
+		}
+	}
+}