diff --git a/OpenRA.Mods.Common/Warheads/FireClusterWarhead.cs b/OpenRA.Mods.Common/Warheads/FireClusterWarhead.cs new file mode 100644 index 0000000000..98360e1ef0 --- /dev/null +++ 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 + { + [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 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 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 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); + } + } +}