Merge pull request #7292 from RoosterDragon/upgradeoverlay-modifyrender

Adjust UpgradeOverlay.ModifyRender to be more efficient.
This commit is contained in:
reaperrr
2015-01-08 08:48:55 +01:00

View File

@@ -29,12 +29,19 @@ namespace OpenRA.Mods.Common.Traits
: 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 (!IsTraitDisabled && !a.IsDecoration)
if (!a.IsDecoration)
yield return a.WithPalette(wr.Palette(Info.Palette))
.WithZOffset(a.ZOffset + 1)
.AsDecoration();