Move RA World traits, FrozenUnderFog Modifier and various other traits into Traits

OpenRA.Mods.RA/Traits/Buildings/PrimaryBuilding.cs

@@ -0,0 +1,86 @@
+#region Copyright & License Information
+/*
+ * Copyright 2007-2014 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.Linq;
+using System.Collections.Generic;
+using OpenRA.Traits;
+using OpenRA.Mods.Common.Orders;
+
+namespace OpenRA.Mods.RA.Traits
+{
+	static class PrimaryExts
+	{
+		public static bool IsPrimaryBuilding(this Actor a)
+		{
+			var pb = a.TraitOrDefault<PrimaryBuilding>();
+			return pb != null && pb.IsPrimary;
+		}
+	}
+
+	[Desc("Used together with ClassicProductionQueue.")]
+	class PrimaryBuildingInfo : TraitInfo<PrimaryBuilding> { }
+
+	class PrimaryBuilding : IIssueOrder, IResolveOrder, ITags
+	{
+		bool isPrimary = false;
+		public bool IsPrimary { get { return isPrimary; } }
+
+		public IEnumerable<TagType> GetTags()
+		{
+			yield return isPrimary ? TagType.Primary : TagType.None;
+		}
+
+		public IEnumerable<IOrderTargeter> Orders
+		{
+			get { yield return new DeployOrderTargeter("PrimaryProducer", 1); }
+		}
+
+		public Order IssueOrder(Actor self, IOrderTargeter order, Target target, bool queued)
+		{
+			if (order.OrderID == "PrimaryProducer")
+				return new Order(order.OrderID, self, false);
+
+			return null;
+		}
+
+		public void ResolveOrder(Actor self, Order order)
+		{
+			if (order.OrderString == "PrimaryProducer")
+				SetPrimaryProducer(self, !isPrimary);
+		}
+
+		public void SetPrimaryProducer(Actor self, bool state)
+		{
+			if (state == false)
+			{
+				isPrimary = false;
+				return;
+			}
+
+			// TODO: THIS IS SHIT
+			// Cancel existing primaries
+			foreach (var p in self.Info.Traits.Get<ProductionInfo>().Produces)
+			{
+				var productionType = p;		// benign closure hazard
+				foreach (var b in self.World
+					.ActorsWithTrait<PrimaryBuilding>()
+					.Where(a =>
+						a.Actor.Owner == self.Owner &&
+						a.Trait.IsPrimary &&
+						a.Actor.Info.Traits.Get<ProductionInfo>().Produces.Contains(productionType)))
+					b.Trait.SetPrimaryProducer(b.Actor, false);
+			}
+
+			isPrimary = true;
+
+			Sound.PlayNotification(self.World.Map.Rules, self.Owner, "Speech", "PrimaryBuildingSelected", self.Owner.Country.Race);
+		}
+	}
+}