bbd54cb32f
Upgradeable traits are notified whenever an upgrade of their declared types are granted or revoked. The traits maintain their own internal level counter, which is then used to enable or disable the trait functionality. A trait can register for multiple upgrade types which then all affect the internal level counter. IDisabledTrait for identifying (and filtering) disabled traits UpgradableTrait provides an abstract base for traits to support upgrade levels Added IDisabledTrait support to GlobalButtonOrderGenerator Includes rework by pchote with alterations.
90 lines
2.5 KiB
C#
90 lines
2.5 KiB
C#
#region Copyright & License Information
|
|
/*
|
|
* Copyright 2007-2014 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.Linq;
|
|
using OpenRA.Mods.Common;
|
|
using OpenRA.Traits;
|
|
|
|
namespace OpenRA.Mods.RA.Crates
|
|
{
|
|
[Desc("Grants an upgrade to the collector.")]
|
|
public class GrantUpgradeCrateActionInfo : CrateActionInfo
|
|
{
|
|
[Desc("The upgrades to apply.")]
|
|
public readonly string[] Upgrades = { };
|
|
|
|
[Desc("Duration of the upgrade (in ticks). Set to 0 for a permanent upgrade.")]
|
|
public readonly int Duration = 0;
|
|
|
|
[Desc("The range to search for extra collectors in.", "Extra collectors will also be granted the crate action.")]
|
|
public readonly WRange Range = new WRange(3);
|
|
|
|
[Desc("The maximum number of extra collectors to grant the crate action to.", "-1 = no limit")]
|
|
public readonly int MaxExtraCollectors = 4;
|
|
|
|
public override object Create(ActorInitializer init) { return new GrantUpgradeCrateAction(init.self, this); }
|
|
}
|
|
|
|
public class GrantUpgradeCrateAction : CrateAction
|
|
{
|
|
readonly Actor self;
|
|
readonly GrantUpgradeCrateActionInfo info;
|
|
|
|
public GrantUpgradeCrateAction(Actor self, GrantUpgradeCrateActionInfo info)
|
|
: base(self, info)
|
|
{
|
|
this.self = self;
|
|
this.info = info;
|
|
}
|
|
|
|
bool AcceptsUpgrade(Actor a)
|
|
{
|
|
var um = a.TraitOrDefault<UpgradeManager>();
|
|
return um != null && info.Upgrades.Any(u => um.AcceptsUpgrade(a, u));
|
|
}
|
|
|
|
public override int GetSelectionShares(Actor collector)
|
|
{
|
|
return AcceptsUpgrade(collector) ? info.SelectionShares : 0;
|
|
}
|
|
|
|
public override void Activate(Actor collector)
|
|
{
|
|
var actorsInRange = self.World.FindActorsInCircle(self.CenterPosition, info.Range)
|
|
.Where(a => a != self && a != collector && a.Owner == collector.Owner && AcceptsUpgrade(a));
|
|
|
|
if (info.MaxExtraCollectors > -1)
|
|
actorsInRange = actorsInRange.Take(info.MaxExtraCollectors);
|
|
|
|
collector.World.AddFrameEndTask(w =>
|
|
{
|
|
foreach (var a in actorsInRange.Append(collector))
|
|
{
|
|
if (!a.IsInWorld || a.IsDead)
|
|
continue;
|
|
|
|
var um = a.TraitOrDefault<UpgradeManager>();
|
|
foreach (var u in info.Upgrades)
|
|
{
|
|
if (!um.AcceptsUpgrade(a, u))
|
|
continue;
|
|
|
|
if (info.Duration > 0)
|
|
um.GrantTimedUpgrade(a, u, info.Duration);
|
|
else
|
|
um.GrantUpgrade(a, u, this);
|
|
}
|
|
}
|
|
});
|
|
|
|
base.Activate(collector);
|
|
}
|
|
}
|
|
} |