Add five more update rules

2018-03-30 17:07:43 +02:00
parent 01684bc329
commit af1a2429b1
7 changed files with 312 additions and 1 deletions
--- a/OpenRA.Mods.Common/UpdateRules/Rules/DecoupleSelfReloading.cs
+++ b/OpenRA.Mods.Common/UpdateRules/Rules/DecoupleSelfReloading.cs
@@ -0,0 +1,85 @@
+#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;
+
+namespace OpenRA.Mods.Common.UpdateRules.Rules
+{
+	public class DecoupleSelfReloading : UpdateRule
+	{
+		public override string Name { get { return "Replace 'SelfReloads' with 'ReloadAmmoPool'"; } }
+		public override string Description
+		{
+			get
+			{
+				return "'SelfReloads', 'SelfReloadDelay', 'ReloadCount', 'ResetOnFire'\n" +
+					"and 'RearmSound' were renamed and moved from 'AmmoPool' to a new 'ReloadAmmoPool' trait.";
+			}
+		}
+
+		public override IEnumerable<string> UpdateActorNode(ModData modData, MiniYamlNode actorNode)
+		{
+			var poolNumber = 0;
+			var ammoPools = actorNode.ChildrenMatching("AmmoPool").ToList();
+			foreach (var pool in ammoPools)
+			{
+				var selfReloads = pool.LastChildMatching("SelfReloads");
+				if (selfReloads == null || !selfReloads.NodeValue<bool>())
+					continue;
+
+				poolNumber++;
+				var reloadOnCond = new MiniYamlNode("ReloadAmmoPool@" + poolNumber, "");
+
+				var name = pool.LastChildMatching("Name");
+				if (name != null)
+					reloadOnCond.AddNode("AmmoPool", name.NodeValue<string>());
+
+				var selfReloadDelay = pool.LastChildMatching("SelfReloadDelay");
+				if (selfReloadDelay != null)
+				{
+					selfReloadDelay.RenameKey("Delay");
+					reloadOnCond.AddNode(selfReloadDelay);
+					pool.RemoveNodes("SelfReloadDelay");
+				}
+
+				pool.RemoveNodes("SelfReloads");
+
+				var reloadCount = pool.LastChildMatching("ReloadCount");
+				if (reloadCount != null)
+				{
+					reloadCount.RenameKey("Count");
+					reloadOnCond.AddNode(reloadCount);
+					pool.RemoveNodes("ReloadCount");
+				}
+
+				var reset = pool.LastChildMatching("ResetOnFire");
+				if (reset != null)
+				{
+					reloadOnCond.AddNode(reset);
+					pool.RemoveNodes("ResetOnFire");
+				}
+
+				var rearmSound = pool.LastChildMatching("RearmSound");
+				if (rearmSound != null)
+				{
+					rearmSound.RenameKey("Sound");
+					reloadOnCond.AddNode(rearmSound);
+					pool.RemoveNodes("RearmSound");
+				}
+
+				actorNode.AddNode(reloadOnCond);
+			}
+
+			yield break;
+		}
+	}
+}