fb0cab7481
If a class is caching the TraitsImplementing enumerable, instead cache the results of enumerating it to an array. The avoids having to enumerate the sequence each time it is needed.
117 lines
4.0 KiB
C#
117 lines
4.0 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.Linq;
|
|
using OpenRA.Graphics;
|
|
using OpenRA.Mods.Common.Graphics;
|
|
using OpenRA.Traits;
|
|
|
|
namespace OpenRA.Mods.Common.Traits
|
|
{
|
|
[Desc("Renders turrets for units with the Turreted trait.")]
|
|
public class WithTurretInfo : UpgradableTraitInfo, ITraitInfo, IRenderActorPreviewSpritesInfo,
|
|
Requires<RenderSpritesInfo>, Requires<TurretedInfo>, Requires<IBodyOrientationInfo>
|
|
{
|
|
[Desc("Sequence name to use")]
|
|
public readonly string Sequence = "turret";
|
|
|
|
[Desc("Sequence name to use when prepared to fire")]
|
|
public readonly string AimSequence = null;
|
|
|
|
[Desc("Turreted 'Turret' key to display")]
|
|
public readonly string Turret = "primary";
|
|
|
|
[Desc("Render recoil")]
|
|
public readonly bool Recoils = true;
|
|
|
|
public object Create(ActorInitializer init) { return new WithTurret(init.Self, this); }
|
|
|
|
public IEnumerable<IActorPreview> RenderPreviewSprites(ActorPreviewInitializer init, RenderSpritesInfo rs, string image, int facings, PaletteReference p)
|
|
{
|
|
if (UpgradeMinEnabledLevel > 0)
|
|
yield break;
|
|
|
|
var body = init.Actor.Traits.Get<BodyOrientationInfo>();
|
|
var t = init.Actor.Traits.WithInterface<TurretedInfo>()
|
|
.First(tt => tt.Turret == Turret);
|
|
|
|
var ifacing = init.Actor.Traits.GetOrDefault<IFacingInfo>();
|
|
var bodyFacing = ifacing != null ? init.Contains<FacingInit>() ? init.Get<FacingInit, int>() : ifacing.GetInitialFacing() : 0;
|
|
var turretFacing = init.Contains<TurretFacingInit>() ? init.Get<TurretFacingInit, int>() : t.InitialFacing;
|
|
|
|
var anim = new Animation(init.World, image, () => turretFacing);
|
|
anim.Play(Sequence);
|
|
|
|
var orientation = body.QuantizeOrientation(new WRot(WAngle.Zero, WAngle.Zero, WAngle.FromFacing(bodyFacing)), facings);
|
|
var offset = body.LocalToWorld(t.Offset.Rotate(orientation));
|
|
yield return new SpriteActorPreview(anim, offset, offset.Y + offset.Z + 1, p, rs.Scale);
|
|
}
|
|
}
|
|
|
|
public class WithTurret : UpgradableTrait<WithTurretInfo>, ITick
|
|
{
|
|
RenderSprites rs;
|
|
IBodyOrientation body;
|
|
AttackBase ab;
|
|
Turreted t;
|
|
Armament[] arms;
|
|
Animation anim;
|
|
|
|
public WithTurret(Actor self, WithTurretInfo info)
|
|
: base(info)
|
|
{
|
|
rs = self.Trait<RenderSprites>();
|
|
body = self.Trait<IBodyOrientation>();
|
|
|
|
ab = self.TraitOrDefault<AttackBase>();
|
|
t = self.TraitsImplementing<Turreted>()
|
|
.First(tt => tt.Name == info.Turret);
|
|
arms = self.TraitsImplementing<Armament>()
|
|
.Where(w => w.Info.Turret == info.Turret).ToArray();
|
|
|
|
anim = new Animation(self.World, rs.GetImage(self), () => t.TurretFacing);
|
|
anim.Play(info.Sequence);
|
|
rs.Add("turret_{0}_{1}".F(info.Turret, info.Sequence), new AnimationWithOffset(
|
|
anim, () => TurretOffset(self), () => IsTraitDisabled, () => false, p => ZOffsetFromCenter(self, p, 1)));
|
|
|
|
// Restrict turret facings to match the sprite
|
|
t.QuantizedFacings = anim.CurrentSequence.Facings;
|
|
}
|
|
|
|
WVec TurretOffset(Actor self)
|
|
{
|
|
if (!Info.Recoils)
|
|
return t.Position(self);
|
|
|
|
var recoil = arms.Aggregate(WRange.Zero, (a, b) => a + b.Recoil);
|
|
var localOffset = new WVec(-recoil, WRange.Zero, WRange.Zero);
|
|
var bodyOrientation = body.QuantizeOrientation(self, self.Orientation);
|
|
var turretOrientation = body.QuantizeOrientation(self, t.LocalOrientation(self));
|
|
return t.Position(self) + body.LocalToWorld(localOffset.Rotate(turretOrientation).Rotate(bodyOrientation));
|
|
}
|
|
|
|
public void Tick(Actor self)
|
|
{
|
|
if (Info.AimSequence == null)
|
|
return;
|
|
|
|
var sequence = ab.IsAttacking ? Info.AimSequence : Info.Sequence;
|
|
anim.ReplaceAnim(sequence);
|
|
}
|
|
|
|
public static int ZOffsetFromCenter(Actor self, WPos pos, int offset)
|
|
{
|
|
var delta = self.CenterPosition - pos;
|
|
return delta.Y + delta.Z + offset;
|
|
}
|
|
}
|
|
}
|