Rename EmitInfantryOnSell and remove default actor type.

2018-06-01 10:15:11 +02:00
parent 3d79541148
commit 7dd64445fc
12 changed files with 76 additions and 29 deletions
--- a/OpenRA.Mods.Common/Traits/SpawnActorsOnSell.cs
+++ b/OpenRA.Mods.Common/Traits/SpawnActorsOnSell.cs
@@ -0,0 +1,93 @@
+#region Copyright & License Information
+/*
+ * Copyright 2007-2018 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, either version 3 of
+ * the License, or (at your option) any later version. For more
+ * information, see COPYING.
+ */
+#endregion
+
+using System.Collections.Generic;
+using System.Linq;
+using OpenRA.Primitives;
+using OpenRA.Traits;
+
+namespace OpenRA.Mods.Common.Traits
+{
+	[Desc("Spawn new actors when sold.")]
+	public class SpawnActorsOnSellInfo : ITraitInfo
+	{
+		public readonly int ValuePercent = 40;
+		public readonly int MinHpPercent = 30;
+
+		[ActorReference, FieldLoader.Require]
+		[Desc("Actor types to spawn on sell. Be sure to use lowercase.")]
+		public readonly string[] ActorTypes = null;
+
+		[Desc("Spawns actors only if the selling player's faction is in this list. " +
+			"Leave empty to allow all factions by default.")]
+		public readonly HashSet<string> Factions = new HashSet<string>();
+
+		public object Create(ActorInitializer init) { return new SpawnActorsOnSell(init.Self, this); }
+	}
+
+	public class SpawnActorsOnSell : INotifySold
+	{
+		readonly SpawnActorsOnSellInfo info;
+		readonly bool correctFaction;
+
+		public SpawnActorsOnSell(Actor self, SpawnActorsOnSellInfo info)
+		{
+			this.info = info;
+			var factionsList = info.Factions;
+			correctFaction = factionsList.Count == 0 || factionsList.Contains(self.Owner.Faction.InternalName);
+		}
+
+		void INotifySold.Selling(Actor self) { }
+
+		void Emit(Actor self)
+		{
+			if (!correctFaction)
+				return;
+
+			var csv = self.Info.TraitInfoOrDefault<CustomSellValueInfo>();
+			var valued = self.Info.TraitInfoOrDefault<ValuedInfo>();
+			var cost = csv != null ? csv.Value : (valued != null ? valued.Cost : 0);
+
+			var health = self.TraitOrDefault<Health>();
+			var dudesValue = info.ValuePercent * cost / 100;
+			if (health != null)
+			{
+				// Cast to long to avoid overflow when multiplying by the health
+				if (100L * health.HP >= info.MinHpPercent * (long)health.MaxHP)
+					dudesValue = (int)((long)health.HP * dudesValue / health.MaxHP);
+				else
+					dudesValue = 0;
+			}
+
+			var buildingInfo = self.Info.TraitInfoOrDefault<BuildingInfo>();
+
+			var eligibleLocations = buildingInfo != null ? buildingInfo.Tiles(self.Location).ToList() : new List<CPos>();
+			var actorTypes = info.ActorTypes.Select(a => new { Name = a, Cost = self.World.Map.Rules.Actors[a].TraitInfo<ValuedInfo>().Cost }).ToList();
+
+			while (eligibleLocations.Count > 0 && actorTypes.Any(a => a.Cost <= dudesValue))
+			{
+				var at = actorTypes.Where(a => a.Cost <= dudesValue).Random(self.World.SharedRandom);
+				var loc = eligibleLocations.Random(self.World.SharedRandom);
+
+				eligibleLocations.Remove(loc);
+				dudesValue -= at.Cost;
+
+				self.World.AddFrameEndTask(w => w.CreateActor(at.Name, new TypeDictionary
+				{
+					new LocationInit(loc),
+					new OwnerInit(self.Owner),
+				}));
+			}
+		}
+
+		void INotifySold.Sold(Actor self) { Emit(self); }
+	}
+}