Move ShroudPalette, ShroudRenderer and StartGameNotification to Common

Move World into Traits folder
2014-11-09 16:04:23 +01:00
parent 3372914216
commit 41f3ee1080
15 changed files with 17 additions and 16 deletions
--- a/OpenRA.Mods.Common/Traits/World/MPStartLocations.cs
+++ b/OpenRA.Mods.Common/Traits/World/MPStartLocations.cs
@@ -0,0 +1,89 @@
+#region Copyright & License Information
+/*
+ * Copyright 2007-2014 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.Collections.Generic;
+using System.Linq;
+using OpenRA.Graphics;
+using OpenRA.Primitives;
+using OpenRA.Traits;
+
+namespace OpenRA.Mods.Common
+{
+	public class MPStartLocationsInfo : TraitInfo<MPStartLocations>
+	{
+		public readonly WRange InitialExploreRange = WRange.FromCells(5);
+	}
+
+	public class MPStartLocations : IWorldLoaded
+	{
+		public Dictionary<Player, CPos> Start = new Dictionary<Player, CPos>();
+
+		public void WorldLoaded(World world, WorldRenderer wr)
+		{
+			var spawns = world.Map.GetSpawnPoints().ToList();
+			var taken = world.LobbyInfo.Clients.Where(c => c.SpawnPoint != 0 && c.Slot != null)
+					.Select(c => spawns[c.SpawnPoint-1]).ToList();
+			var available = spawns.Except(taken).ToList();
+
+			// Set spawn
+			foreach (var kv in world.LobbyInfo.Slots)
+			{
+				var player = FindPlayerInSlot(world, kv.Key);
+				if (player == null) continue;
+
+				var client = world.LobbyInfo.ClientInSlot(kv.Key);
+				var spid = (client == null || client.SpawnPoint == 0)
+					? ChooseSpawnPoint(world, available, taken)
+					: spawns[client.SpawnPoint-1];
+
+				Start.Add(player, spid);
+
+				player.SpawnPoint = (client == null || client.SpawnPoint == 0)
+					? spawns.IndexOf(spid) + 1
+					: client.SpawnPoint;
+			}
+
+			// Explore allied shroud
+			var explore = world.WorldActor.Info.Traits.Get<MPStartLocationsInfo>().InitialExploreRange;
+			foreach (var p in Start.Keys)
+				foreach (var q in world.Players)
+					if (p.IsAlliedWith(q))
+						q.Shroud.Explore(world, Start[p], explore);
+
+			// Set viewport
+			if (world.LocalPlayer != null && Start.ContainsKey(world.LocalPlayer))
+				wr.Viewport.Center(world.Map.CenterOfCell(Start[world.LocalPlayer]));
+		}
+
+		static Player FindPlayerInSlot(World world, string pr)
+		{
+			return world.Players.FirstOrDefault(p => p.PlayerReference.Name == pr);
+		}
+
+		static CPos ChooseSpawnPoint(World world, List<CPos> available, List<CPos> taken)
+		{
+			if (available.Count == 0)
+				throw new InvalidOperationException("No free spawnpoint.");
+
+			var n = taken.Count == 0
+				? world.SharedRandom.Next(available.Count)
+				: available			// pick the most distant spawnpoint from everyone else
+					.Select((k, i) => Pair.New(k, i))
+					.MaxBy(a => taken.Sum(t => (t - a.First).LengthSquared)).Second;
+
+			var sp = available[n];
+			available.RemoveAt(n);
+			taken.Add(sp);
+			return sp;
+		}
+	}
+}
+