Fix a crash with support powers and units without selection decorations
This commit is contained in:
@@ -192,6 +192,7 @@ namespace OpenRA.Mods.Cnc.Traits
|
|||||||
if (unit.CanBeViewedByPlayer(manager.Self.Owner))
|
if (unit.CanBeViewedByPlayer(manager.Self.Owner))
|
||||||
{
|
{
|
||||||
var decorations = unit.TraitsImplementing<ISelectionDecorations>().FirstEnabledTraitOrDefault();
|
var decorations = unit.TraitsImplementing<ISelectionDecorations>().FirstEnabledTraitOrDefault();
|
||||||
|
if (decorations != null)
|
||||||
foreach (var d in decorations.RenderSelectionAnnotations(unit, wr, Color.Red))
|
foreach (var d in decorations.RenderSelectionAnnotations(unit, wr, Color.Red))
|
||||||
yield return d;
|
yield return d;
|
||||||
}
|
}
|
||||||
@@ -318,6 +319,7 @@ namespace OpenRA.Mods.Cnc.Traits
|
|||||||
if (unit.CanBeViewedByPlayer(manager.Self.Owner))
|
if (unit.CanBeViewedByPlayer(manager.Self.Owner))
|
||||||
{
|
{
|
||||||
var decorations = unit.TraitsImplementing<ISelectionDecorations>().FirstEnabledTraitOrDefault();
|
var decorations = unit.TraitsImplementing<ISelectionDecorations>().FirstEnabledTraitOrDefault();
|
||||||
|
if (decorations != null)
|
||||||
foreach (var d in decorations.RenderSelectionAnnotations(unit, wr, Color.Red))
|
foreach (var d in decorations.RenderSelectionAnnotations(unit, wr, Color.Red))
|
||||||
yield return d;
|
yield return d;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -155,6 +155,7 @@ namespace OpenRA.Mods.Common.Traits
|
|||||||
foreach (var unit in power.UnitsInRange(xy))
|
foreach (var unit in power.UnitsInRange(xy))
|
||||||
{
|
{
|
||||||
var decorations = unit.TraitsImplementing<ISelectionDecorations>().FirstEnabledTraitOrDefault();
|
var decorations = unit.TraitsImplementing<ISelectionDecorations>().FirstEnabledTraitOrDefault();
|
||||||
|
if (decorations != null)
|
||||||
foreach (var d in decorations.RenderSelectionAnnotations(unit, wr, Color.Red))
|
foreach (var d in decorations.RenderSelectionAnnotations(unit, wr, Color.Red))
|
||||||
yield return d;
|
yield return d;
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user