RenderBuildingRefinery -> WithResources.

2013-06-20 18:16:14 +12:00
parent f376eb9b2a
commit e99d0a0b43
7 changed files with 75 additions and 71 deletions
--- a/OpenRA.Mods.RA/Render/WithResources.cs
+++ b/OpenRA.Mods.RA/Render/WithResources.cs
@@ -0,0 +1,70 @@
+#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.Collections.Generic;
+using System.Linq;
+using OpenRA.FileFormats;
+using OpenRA.Graphics;
+using OpenRA.Traits;
+
+namespace OpenRA.Mods.RA.Render
+{
+	class WithResourcesInfo : ITraitInfo, Requires<RenderSimpleInfo>
+	{
+		[Desc("Sequence name to use")]
+		public readonly string Sequence = "resources";
+
+		public object Create(ActorInitializer init) { return new WithResources(init.self, this); }
+	}
+
+	class WithResources : INotifyBuildComplete, INotifySold, INotifyCapture, INotifyDamageStateChanged
+	{
+		WithResourcesInfo info;
+		Animation anim;
+		RenderSimple rs;
+		PlayerResources playerResources;
+		bool buildComplete;
+
+		public WithResources(Actor self, WithResourcesInfo info)
+		{
+			this.info = info;
+			rs = self.Trait<RenderSimple>();
+			playerResources = self.Owner.PlayerActor.Trait<PlayerResources>();
+
+			anim = new Animation(rs.GetImage(self));
+			anim.PlayFetchIndex(info.Sequence,
+			                    () => playerResources.OreCapacity != 0
+			                    ? ((10 * anim.CurrentSequence.Length - 1) * playerResources.Ore) / (10 * playerResources.OreCapacity)
+			                    : 0);
+
+			rs.anims.Add("resources_{0}".F(info.Sequence), new AnimationWithOffset(
+				anim, null, () => !buildComplete, 1024));
+		}
+
+		public void BuildingComplete( Actor self )
+		{
+			buildComplete = true;
+		}
+
+		public void DamageStateChanged(Actor self, AttackInfo e)
+		{
+			if (anim.CurrentSequence != null)
+				anim.ReplaceAnim(rs.NormalizeSequence(self, info.Sequence));
+		}
+
+		public void OnCapture (Actor self, Actor captor, Player oldOwner, Player newOwner)
+		{
+			playerResources = newOwner.PlayerActor.Trait<PlayerResources>();
+		}
+
+		public void Selling(Actor self) { rs.anims.Remove("resources_{0}".F(info.Sequence)); }
+		public void Sold(Actor self) { }
+	}
+}