Merge the 3 turret rendering traits into WithTurret.

2013-04-08 03:04:20 +12:00
parent b7ea36b858
commit c91f0dfd2f
14 changed files with 137 additions and 149 deletions
--- a/OpenRA.Mods.RA/Render/WithTurret.cs
+++ b/OpenRA.Mods.RA/Render/WithTurret.cs
@@ -0,0 +1,81 @@
+#region Copyright & License Information
+/*
+ * Copyright 2007-2013 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;
+using System.Collections.Generic;
+using System.Linq;
+using OpenRA.FileFormats;
+using OpenRA.Graphics;
+using OpenRA.Traits;
+
+namespace OpenRA.Mods.RA.Render
+{
+	class WithTurretInfo : ITraitInfo, Requires<RenderSimpleInfo>, Requires<TurretedInfo>
+	{
+		[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";
+
+		public object Create(ActorInitializer init) { return new WithTurret(init.self, this); }
+	}
+
+	class WithTurret : ITick
+	{
+		WithTurretInfo info;
+		RenderSimple rs;
+		AttackBase ab;
+		Turreted t;
+		IEnumerable<Armament> arms;
+		Animation anim;
+
+		public WithTurret(Actor self, WithTurretInfo info)
+		{
+			this.info = info;
+			rs = self.Trait<RenderSimple>();
+			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);
+
+			anim = new Animation(rs.GetImage(self), () => t.turretFacing);
+			anim.Play(info.Sequence);
+			rs.anims.Add("turret_{0}".F(info.Turret), new AnimationWithOffset(
+				anim,
+				wr => TurretPosition(self, wr),
+				null) { ZOffset = 1 });
+		}
+
+		int2 TurretPosition(Actor self, WorldRenderer wr)
+		{
+			var recoil = arms.Aggregate(WRange.Zero, (a,b) => a + b.Recoil);
+			var localOffset = new WVec(-recoil, WRange.Zero, WRange.Zero);
+			var bodyOrientation = rs.QuantizeOrientation(self, self.Orientation);
+			var turretOrientation = rs.QuantizeOrientation(self, t.LocalOrientation(self));
+			var worldPos = t.Position(self) + rs.LocalToWorld(localOffset.Rotate(turretOrientation).Rotate(bodyOrientation));
+
+			return wr.ScreenPxOffset(worldPos);
+		}
+
+		public void Tick(Actor self)
+		{
+			if (info.AimSequence == null)
+				return;
+
+			var sequence = ab.IsAttacking ? info.AimSequence : info.Sequence;
+			rs.anims["turret_{0}".F(info.Turret)].Animation.ReplaceAnim(sequence);
+		}
+	}
+}