Add Type Support for InfiltrateFor* traits

This commit is contained in:
Mustafa Alperen Seki
2017-11-07 10:32:11 +03:00
committed by Paul Chote
parent f7de5d46be
commit fc07391c8c
10 changed files with 65 additions and 11 deletions

View File

@@ -9,6 +9,7 @@
*/
#endregion
using System.Collections.Generic;
using System.Linq;
using OpenRA.Mods.Common.Traits;
using OpenRA.Traits;
@@ -16,12 +17,27 @@ using OpenRA.Traits;
namespace OpenRA.Mods.Cnc.Traits
{
[Desc("Steal and reset the owner's exploration.")]
class InfiltrateForExplorationInfo : TraitInfo<InfiltrateForExploration> { }
class InfiltrateForExplorationInfo : ITraitInfo
{
public readonly HashSet<string> Types = new HashSet<string>();
public object Create(ActorInitializer init) { return new InfiltrateForExploration(init.Self, this); }
}
class InfiltrateForExploration : INotifyInfiltrated
{
void INotifyInfiltrated.Infiltrated(Actor self, Actor infiltrator)
readonly InfiltrateForExplorationInfo info;
public InfiltrateForExploration(Actor self, InfiltrateForExplorationInfo info)
{
this.info = info;
}
void INotifyInfiltrated.Infiltrated(Actor self, Actor infiltrator, HashSet<string> types)
{
if (!info.Types.Overlaps(types))
return;
infiltrator.Owner.Shroud.Explore(self.Owner.Shroud);
var preventReset = self.Owner.PlayerActor.TraitsImplementing<IPreventsShroudReset>()
.Any(p => p.PreventShroudReset(self));