epic/OpenRA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add support for removing segments when parent nodes are removed.

Browse Source
This commit is contained in:
Paul Chote
2017-03-27 18:14:24 +01:00
parent 304e3ef9f9
commit 227b80d6c2
1 changed files with 17 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
OpenRA.Mods.Common/Traits/Buildings/LineBuild.cs
View File

@@ -51,16 +51,21 @@ namespace OpenRA.Mods.Common.Traits
[Desc("Actor type for line-built segments (defaults to same actor).")] [Desc("Actor type for line-built segments (defaults to same actor).")]
public readonly string SegmentType = null; public readonly string SegmentType = null;
[Desc("Delete generated segments when destroyed or sold.")]
public readonly bool SegmentsRequireNode = false;
public object Create(ActorInitializer init) { return new LineBuild(init, this); } public object Create(ActorInitializer init) { return new LineBuild(init, this); }
} }
public class LineBuild : INotifyAddedToWorld, INotifyRemovedFromWorld, INotifyLineBuildSegmentsChanged public class LineBuild : INotifyKilled, INotifyAddedToWorld, INotifyRemovedFromWorld, INotifyLineBuildSegmentsChanged
{ {
readonly LineBuildInfo info;
readonly Actor[] parentNodes = new Actor[0]; readonly Actor[] parentNodes = new Actor[0];
HashSet<Actor> segments; HashSet<Actor> segments;
public LineBuild(ActorInitializer init, LineBuildInfo info) public LineBuild(ActorInitializer init, LineBuildInfo info)
{ {
this.info = info;
if (init.Contains<LineBuildParentInit>()) if (init.Contains<LineBuildParentInit>())
parentNodes = init.Get<LineBuildParentInit>().Value(init.World); parentNodes = init.Get<LineBuildParentInit>().Value(init.World);
} }
@@ -95,6 +100,17 @@ namespace OpenRA.Mods.Common.Traits
if (!parent.Disposed) if (!parent.Disposed)
foreach (var n in parent.TraitsImplementing<INotifyLineBuildSegmentsChanged>()) foreach (var n in parent.TraitsImplementing<INotifyLineBuildSegmentsChanged>())
n.SegmentRemoved(parent, self); n.SegmentRemoved(parent, self);
if (info.SegmentsRequireNode && segments != null)
foreach (var s in segments)
s.Dispose();
}
void INotifyKilled.Killed(Actor self, AttackInfo e)
{
if (info.SegmentsRequireNode && segments != null)
foreach (var s in segments)
s.Kill(e.Attacker);
} }
} }
} }