Add FireClusterWarhead

2018-07-08 13:35:39 +02:00
parent 0e3d343f15
commit 2648764ee3
1 changed files with 115 additions and 0 deletions
--- a/OpenRA.Mods.Common/Warheads/FireClusterWarhead.cs
+++ b/OpenRA.Mods.Common/Warheads/FireClusterWarhead.cs
@@ -0,0 +1,115 @@
+#region Copyright & License Information
+/*
+ * Copyright 2007-2019 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.GameRules;
+using OpenRA.Mods.Common.Traits;
+using OpenRA.Traits;
+
+namespace OpenRA.Mods.Common.Warheads
+{
+	public class FireClusterWarhead : Warhead, IRulesetLoaded<WeaponInfo>
+	{
+		[WeaponReference, FieldLoader.Require]
+		[Desc("Has to be defined in weapons.yaml as well.")]
+		public readonly string Weapon = null;
+
+		[Desc("Number of weapons fired at random 'x' cells. Negative values will result in a number equal to 'x' footprint cells fired.")]
+		public readonly int RandomClusterCount = -1;
+
+		[FieldLoader.Require]
+		[Desc("Size of the cluster footprint")]
+		public readonly CVec Dimensions = CVec.Zero;
+
+		[FieldLoader.Require]
+		[Desc("Cluster footprint. Cells marked as X will be attacked.",
+			"Cells marked as x will be attacked randomly until RandomClusterCount is reached.")]
+		public readonly string Footprint = string.Empty;
+
+		WeaponInfo weapon;
+
+		public void RulesetLoaded(Ruleset rules, WeaponInfo info)
+		{
+			if (!rules.Weapons.TryGetValue(Weapon.ToLowerInvariant(), out weapon))
+				throw new YamlException("Weapons Ruleset does not contain an entry '{0}'".F(Weapon.ToLowerInvariant()));
+		}
+
+		public override void DoImpact(Target target, Actor firedBy, IEnumerable<int> damageModifiers)
+		{
+			if (!target.IsValidFor(firedBy))
+				return;
+
+			var map = firedBy.World.Map;
+			var targetCell = map.CellContaining(target.CenterPosition);
+
+			var targetCells = CellsMatching(targetCell, false);
+			foreach (var c in targetCells)
+				FireProjectileAtCell(map, firedBy, target, c, damageModifiers);
+
+			if (RandomClusterCount != 0)
+			{
+				var randomTargetCells = CellsMatching(targetCell, true);
+				var clusterCount = RandomClusterCount < 0 ? randomTargetCells.Count() : RandomClusterCount;
+				if (randomTargetCells.Any())
+					for (var i = 0; i < clusterCount; i++)
+						FireProjectileAtCell(map, firedBy, target, randomTargetCells.Random(firedBy.World.SharedRandom), damageModifiers);
+			}
+		}
+
+		void FireProjectileAtCell(Map map, Actor firedBy, Target target, CPos targetCell, IEnumerable<int> damageModifiers)
+		{
+			var tc = Target.FromCell(firedBy.World, targetCell);
+
+			if (!weapon.IsValidAgainst(tc, firedBy.World, firedBy))
+				return;
+
+			var args = new ProjectileArgs
+			{
+				Weapon = weapon,
+				Facing = (map.CenterOfCell(targetCell) - target.CenterPosition).Yaw.Facing,
+
+				DamageModifiers = damageModifiers.ToArray(),
+				InaccuracyModifiers = new int[0],
+				RangeModifiers = new int[0],
+
+				Source = target.CenterPosition,
+				CurrentSource = () => target.CenterPosition,
+				SourceActor = firedBy,
+				PassiveTarget = map.CenterOfCell(targetCell),
+				GuidedTarget = tc
+			};
+
+			if (args.Weapon.Projectile != null)
+			{
+				var projectile = args.Weapon.Projectile.Create(args);
+				if (projectile != null)
+					firedBy.World.AddFrameEndTask(w => w.Add(projectile));
+
+				if (args.Weapon.Report != null && args.Weapon.Report.Any())
+					Game.Sound.Play(SoundType.World, args.Weapon.Report, firedBy.World, target.CenterPosition);
+			}
+		}
+
+		IEnumerable<CPos> CellsMatching(CPos location, bool random)
+		{
+			var cellType = !random ? 'X' : 'x';
+			var index = 0;
+			var footprint = Footprint.Where(c => !char.IsWhiteSpace(c)).ToArray();
+			var x = location.X - (Dimensions.X - 1) / 2;
+			var y = location.Y - (Dimensions.Y - 1) / 2;
+			for (var j = 0; j < Dimensions.Y; j++)
+				for (var i = 0; i < Dimensions.X; i++)
+					if (footprint[index++] == cellType)
+						yield return new CPos(x + i, y + j);
+		}
+	}
+}