86 lines
2.4 KiB
C#
86 lines
2.4 KiB
C#
#region Copyright & License Information
|
|
/*
|
|
* Copyright 2007-2015 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 System.Drawing;
|
|
using System.Linq;
|
|
using OpenRA.Graphics;
|
|
using OpenRA.Mods.Common.Graphics;
|
|
using OpenRA.Traits;
|
|
|
|
namespace OpenRA.Mods.Common.Traits
|
|
{
|
|
[Desc("Draw a circle indicating my weapon's range.")]
|
|
class RenderRangeCircleInfo : ITraitInfo, IPlaceBuildingDecorationInfo, Requires<AttackBaseInfo>
|
|
{
|
|
public readonly string RangeCircleType = null;
|
|
|
|
[Desc("Range to draw if no armaments are available")]
|
|
public readonly WDist FallbackRange = WDist.Zero;
|
|
|
|
public IEnumerable<IRenderable> Render(WorldRenderer wr, World w, ActorInfo ai, WPos centerPosition)
|
|
{
|
|
var armaments = ai.TraitInfos<ArmamentInfo>()
|
|
.Where(a => a.UpgradeMinEnabledLevel == 0);
|
|
var range = FallbackRange;
|
|
|
|
if (armaments.Any())
|
|
range = armaments.Select(a => w.Map.Rules.Weapons[a.Weapon.ToLowerInvariant()].Range).Max();
|
|
|
|
if (range == WDist.Zero)
|
|
yield break;
|
|
|
|
yield return new RangeCircleRenderable(
|
|
centerPosition,
|
|
range,
|
|
0,
|
|
Color.FromArgb(128, Color.Yellow),
|
|
Color.FromArgb(96, Color.Black));
|
|
|
|
foreach (var a in w.ActorsWithTrait<RenderRangeCircle>())
|
|
if (a.Actor.Owner.IsAlliedWith(w.RenderPlayer))
|
|
if (a.Actor.Info.TraitInfo<RenderRangeCircleInfo>().RangeCircleType == RangeCircleType)
|
|
foreach (var r in a.Trait.RenderAfterWorld(wr))
|
|
yield return r;
|
|
}
|
|
|
|
public object Create(ActorInitializer init) { return new RenderRangeCircle(init.Self); }
|
|
}
|
|
|
|
class RenderRangeCircle : IPostRenderSelection
|
|
{
|
|
readonly Actor self;
|
|
readonly AttackBase attack;
|
|
|
|
public RenderRangeCircle(Actor self)
|
|
{
|
|
this.self = self;
|
|
attack = self.Trait<AttackBase>();
|
|
}
|
|
|
|
public IEnumerable<IRenderable> RenderAfterWorld(WorldRenderer wr)
|
|
{
|
|
if (!self.Owner.IsAlliedWith(self.World.RenderPlayer))
|
|
yield break;
|
|
|
|
var range = attack.GetMaximumRange();
|
|
if (range == WDist.Zero)
|
|
yield break;
|
|
|
|
yield return new RangeCircleRenderable(
|
|
self.CenterPosition,
|
|
range,
|
|
0,
|
|
Color.FromArgb(128, Color.Yellow),
|
|
Color.FromArgb(96, Color.Black));
|
|
}
|
|
}
|
|
}
|