901e604cf3
In places where arrays were being treated as a set, just create a set directly. This reveals the intention of such collections better, and also improves performance by allowing set based methods to be used.
112 lines
3.5 KiB
C#
112 lines
3.5 KiB
C#
#region Copyright & License Information
|
|
/*
|
|
* Copyright 2007-2015 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;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using OpenRA.Primitives;
|
|
using OpenRA.Traits;
|
|
|
|
namespace OpenRA.Mods.Common.Traits
|
|
{
|
|
[Desc("Creates duplicates of the actor that collects the crate.")]
|
|
class DuplicateUnitCrateActionInfo : CrateActionInfo
|
|
{
|
|
[Desc("The maximum number of duplicates to make.")]
|
|
public readonly int MaxAmount = 2;
|
|
|
|
[Desc("The minimum number of duplicates to make. Overrules MaxDuplicatesWorth.")]
|
|
public readonly int MinAmount = 1;
|
|
|
|
[Desc("The maximum total value allowed for the duplicates.", "Duplication stops if the total worth will exceed this number.", "-1 = no limit")]
|
|
public readonly int MaxDuplicateValue = -1;
|
|
|
|
[Desc("The maximum radius (in cells) that duplicates can be spawned.")]
|
|
public readonly int MaxRadius = 4;
|
|
|
|
[Desc("The list of unit target types we are allowed to duplicate.")]
|
|
public readonly HashSet<string> ValidTargets = new HashSet<string> { "Ground", "Water" };
|
|
|
|
[Desc("Which factions this crate action can occur for.")]
|
|
public readonly string[] ValidFactions = { };
|
|
|
|
[Desc("Is the new duplicates given to a specific owner, regardless of whom collected it?")]
|
|
public readonly string Owner = null;
|
|
|
|
public override object Create(ActorInitializer init) { return new DuplicateUnitCrateAction(init.Self, this); }
|
|
}
|
|
|
|
class DuplicateUnitCrateAction : CrateAction
|
|
{
|
|
readonly DuplicateUnitCrateActionInfo info;
|
|
|
|
public DuplicateUnitCrateAction(Actor self, DuplicateUnitCrateActionInfo info)
|
|
: base(self, info)
|
|
{
|
|
this.info = info;
|
|
}
|
|
|
|
public bool CanGiveTo(Actor collector)
|
|
{
|
|
if (info.ValidFactions.Any() && !info.ValidFactions.Contains(collector.Owner.Faction.InternalName))
|
|
return false;
|
|
|
|
var targetable = collector.TraitsImplementing<ITargetable>();
|
|
if (!info.ValidTargets.Overlaps(targetable.SelectMany(t => t.TargetTypes)))
|
|
return false;
|
|
|
|
var positionable = collector.TraitOrDefault<IPositionable>();
|
|
if (positionable == null)
|
|
return false;
|
|
|
|
return collector.World.Map.FindTilesInCircle(collector.Location, info.MaxRadius)
|
|
.Any(c => positionable.CanEnterCell(c));
|
|
}
|
|
|
|
public override int GetSelectionShares(Actor collector)
|
|
{
|
|
if (!CanGiveTo(collector))
|
|
return 0;
|
|
|
|
return base.GetSelectionShares(collector);
|
|
}
|
|
|
|
public override void Activate(Actor collector)
|
|
{
|
|
var positionable = collector.Trait<IPositionable>();
|
|
var candidateCells = collector.World.Map.FindTilesInCircle(collector.Location, info.MaxRadius)
|
|
.Where(c => positionable.CanEnterCell(c)).Shuffle(collector.World.SharedRandom)
|
|
.ToArray();
|
|
|
|
var duplicates = Math.Min(candidateCells.Length, info.MaxAmount);
|
|
|
|
// Restrict duplicate count to a maximum value
|
|
if (info.MaxDuplicateValue > 0)
|
|
{
|
|
var vi = collector.Info.Traits.GetOrDefault<ValuedInfo>();
|
|
if (vi != null && vi.Cost > 0)
|
|
duplicates = Math.Min(duplicates, info.MaxDuplicateValue / vi.Cost);
|
|
}
|
|
|
|
for (var i = 0; i < duplicates; i++)
|
|
{
|
|
var cell = candidateCells[i]; // Avoid modified closure bug
|
|
collector.World.AddFrameEndTask(w => w.CreateActor(collector.Info.Name, new TypeDictionary
|
|
{
|
|
new LocationInit(cell),
|
|
new OwnerInit(info.Owner ?? collector.Owner.InternalName)
|
|
}));
|
|
}
|
|
|
|
base.Activate(collector);
|
|
}
|
|
}
|
|
}
|