-
Notifications
You must be signed in to change notification settings - Fork 2.6k
/
SpawnMapActors.cs
78 lines (64 loc) · 2.4 KB
/
SpawnMapActors.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
#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 System.Collections.Generic;
using System.Linq;
using OpenRA.Graphics;
using OpenRA.Traits;
namespace OpenRA.Mods.Common.Traits
{
[TraitLocation(SystemActors.World)]
[Desc("Spawns the initial units for each player upon game start.")]
public class SpawnMapActorsInfo : TraitInfo<SpawnMapActors> { }
public class SpawnMapActors : IWorldLoaded
{
public Dictionary<string, Actor> Actors = new();
public uint LastMapActorID { get; private set; }
public void WorldLoaded(World world, WorldRenderer wr)
{
var preventMapSpawns = world.WorldActor.TraitsImplementing<IPreventMapSpawn>()
.Concat(world.WorldActor.Owner.PlayerActor.TraitsImplementing<IPreventMapSpawn>())
.ToArray();
foreach (var kv in world.Map.ActorDefinitions)
{
var actorReference = new ActorReference(kv.Value.Value, kv.Value.ToDictionary());
// If an actor's doesn't have a valid owner transfer ownership to neutral
var ownerInit = actorReference.Get<OwnerInit>();
if (!world.Players.Any(p => p.InternalName == ownerInit.InternalName))
{
actorReference.Remove(ownerInit);
actorReference.Add(new OwnerInit(world.WorldActor.Owner));
}
actorReference.Add(new SkipMakeAnimsInit());
actorReference.Add(new SpawnedByMapInit());
if (PreventMapSpawn(world, actorReference, preventMapSpawns))
continue;
var actor = world.CreateActor(true, actorReference);
Actors[kv.Key] = actor;
LastMapActorID = actor.ActorID;
}
}
static bool PreventMapSpawn(World world, ActorReference actorReference, IEnumerable<IPreventMapSpawn> preventMapSpawns)
{
foreach (var pms in preventMapSpawns)
if (pms.PreventMapSpawn(world, actorReference))
return true;
return false;
}
}
public class SkipMakeAnimsInit : RuntimeFlagInit { }
public class SpawnedByMapInit : ActorInit, ISuppressInitExport, ISingleInstanceInit
{
protected SpawnedByMapInit(string instanceName)
: base(instanceName) { }
public SpawnedByMapInit() { }
public override MiniYaml Save() => null;
}
}