82bea961ba
- Made Array.IndexOf available via extension method. - Made ToHashSet extension method. - Change collections queried often via Contains into sets. - Avoid Count() extension if Count or Length property exist. - Made Count() > 0 checks and variations calls to Any() instead. - Don't call ToList/ToArray if there is no benefit to materializing the sequence. - If the sequence does benefit from materialization, follow this general pattern: - Collection queried often via Contains use ToHashSet to speed up lookups. - Short lived variables use ToList. This is because ToArray requires an extra copy to output the final size. - Collections persisted into fields or for a long time use ToArray to minimize memory overhead.
89 lines
2.6 KiB
C#
89 lines
2.6 KiB
C#
#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.Collections.Generic;
|
|
using System.Linq;
|
|
using OpenRA.Graphics;
|
|
using OpenRA.Primitives;
|
|
using OpenRA.Traits;
|
|
|
|
namespace OpenRA.Mods.Common.Traits
|
|
{
|
|
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();
|
|
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;
|
|
}
|
|
}
|
|
}
|