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

Add RequiresCondition to ProvidesPrerequisite

Browse Source
This commit is contained in:
Jean-Rémy Buchs
2017-08-13 15:47:30 +02:00
committed by Paul Chote
parent c9b4568117
commit 2b96c2ed78
1 changed files with 44 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

62
OpenRA.Mods.Common/Traits/Player/ProvidesPrerequisite.cs
View File

@@ -15,7 +15,7 @@ using OpenRA.Traits;
namespace OpenRA.Mods.Common.Traits namespace OpenRA.Mods.Common.Traits
{ {
public class ProvidesPrerequisiteInfo : ITechTreePrerequisiteInfo public class ProvidesPrerequisiteInfo : ConditionalTraitInfo, ITechTreePrerequisiteInfo
{ {
[Desc("The prerequisite type that this provides. If left empty it defaults to the actor's name.")] [Desc("The prerequisite type that this provides. If left empty it defaults to the actor's name.")]
public readonly string Prerequisite = null; public readonly string Prerequisite = null;
@@ -28,27 +28,26 @@ namespace OpenRA.Mods.Common.Traits
[Desc("Should it recheck everything when it is captured?")] [Desc("Should it recheck everything when it is captured?")]
public readonly bool ResetOnOwnerChange = false; public readonly bool ResetOnOwnerChange = false;
public object Create(ActorInitializer init) { return new ProvidesPrerequisite(init, this); } public override object Create(ActorInitializer init) { return new ProvidesPrerequisite(init, this); }
} }
public class ProvidesPrerequisite : ITechTreePrerequisite, INotifyOwnerChanged public class ProvidesPrerequisite : ConditionalTrait<ProvidesPrerequisiteInfo>, ITechTreePrerequisite, INotifyOwnerChanged, INotifyCreated
{ {
readonly ProvidesPrerequisiteInfo info;
readonly string prerequisite; readonly string prerequisite;
bool enabled = true; bool enabled;
TechTree techTree;
string faction;
public ProvidesPrerequisite(ActorInitializer init, ProvidesPrerequisiteInfo info) public ProvidesPrerequisite(ActorInitializer init, ProvidesPrerequisiteInfo info)
: base(info)
{ {
this.info = info;
prerequisite = info.Prerequisite; prerequisite = info.Prerequisite;
if (string.IsNullOrEmpty(prerequisite)) if (string.IsNullOrEmpty(prerequisite))
prerequisite = init.Self.Info.Name; prerequisite = init.Self.Info.Name;
var faction = init.Contains<FactionInit>() ? init.Get<FactionInit, string>() : init.Self.Owner.Faction.InternalName; faction = init.Contains<FactionInit>() ? init.Get<FactionInit, string>() : init.Self.Owner.Faction.InternalName;
Update(init.Self.Owner, faction);
} }
public IEnumerable<string> ProvidesPrerequisites public IEnumerable<string> ProvidesPrerequisites
@@ -62,21 +61,48 @@ namespace OpenRA.Mods.Common.Traits
} }
} }
public void OnOwnerChanged(Actor self, Player oldOwner, Player newOwner) protected override void Created(Actor self)
{ {
if (info.ResetOnOwnerChange) techTree = self.Owner.PlayerActor.Trait<TechTree>();
Update(newOwner, newOwner.Faction.InternalName);
Update();
base.Created(self);
} }
void Update(Player owner, string faction) public void OnOwnerChanged(Actor self, Player oldOwner, Player newOwner)
{ {
enabled = true; techTree = newOwner.PlayerActor.Trait<TechTree>();
if (info.Factions.Any()) if (Info.ResetOnOwnerChange)
enabled = info.Factions.Contains(faction); faction = newOwner.Faction.InternalName;
if (info.RequiresPrerequisites.Any() && enabled) Update();
enabled = owner.PlayerActor.Trait<TechTree>().HasPrerequisites(info.RequiresPrerequisites); }
void Update()
{
enabled = !IsTraitDisabled;
if (IsTraitDisabled)
return;
if (Info.Factions.Any())
enabled = Info.Factions.Contains(faction);
if (Info.RequiresPrerequisites.Any() && enabled)
enabled = techTree.HasPrerequisites(Info.RequiresPrerequisites);
}
protected override void TraitEnabled(Actor self)
{
Update();
techTree.ActorChanged(self);
}
protected override void TraitDisabled(Actor self)
{
Update();
techTree.ActorChanged(self);
} }
} }
} }