df6956867a
When IsTraitDisabled is set, ModifyRender doesn't actually alter the sequence, so we can just return it directly and avoid adding another layer of enumerable. If it is unset, then the modifying enumerable has one less condition to check inside the loop.
51 lines
1.4 KiB
C#
51 lines
1.4 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.Collections.Generic;
|
|
using OpenRA.Graphics;
|
|
using OpenRA.Traits;
|
|
|
|
namespace OpenRA.Mods.Common.Traits
|
|
{
|
|
[Desc("Display a colored overlay when a timed upgrade is active.")]
|
|
public class UpgradeOverlayInfo : UpgradableTraitInfo, ITraitInfo
|
|
{
|
|
[Desc("Palette to use when rendering the overlay")]
|
|
public readonly string Palette = "invuln";
|
|
|
|
public object Create(ActorInitializer init) { return new UpgradeOverlay(this); }
|
|
}
|
|
|
|
public class UpgradeOverlay : UpgradableTrait<UpgradeOverlayInfo>, IRenderModifier
|
|
{
|
|
public UpgradeOverlay(UpgradeOverlayInfo info)
|
|
: base(info) { }
|
|
|
|
public IEnumerable<IRenderable> ModifyRender(Actor self, WorldRenderer wr, IEnumerable<IRenderable> r)
|
|
{
|
|
if (IsTraitDisabled)
|
|
return r;
|
|
return ModifiedRender(self, wr, r);
|
|
}
|
|
|
|
IEnumerable<IRenderable> ModifiedRender(Actor self, WorldRenderer wr, IEnumerable<IRenderable> r)
|
|
{
|
|
foreach (var a in r)
|
|
{
|
|
yield return a;
|
|
|
|
if (!a.IsDecoration)
|
|
yield return a.WithPalette(wr.Palette(Info.Palette))
|
|
.WithZOffset(a.ZOffset + 1)
|
|
.AsDecoration();
|
|
}
|
|
}
|
|
}
|
|
} |