78 lines
2.2 KiB
C#
78 lines
2.2 KiB
C#
#region Copyright & License Information
|
|
/*
|
|
* Copyright 2007-2021 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.Mods.Common.Effects;
|
|
using OpenRA.Traits;
|
|
|
|
namespace OpenRA.Mods.Common.Traits
|
|
{
|
|
[Desc("Blinks the actor and captor when it is being captured.")]
|
|
class CapturableProgressBlinkInfo : ConditionalTraitInfo, Requires<CapturableInfo>
|
|
{
|
|
[Desc("Number of ticks to wait between repeating blinks.")]
|
|
public readonly int Interval = 50;
|
|
|
|
public override object Create(ActorInitializer init) { return new CapturableProgressBlink(this); }
|
|
}
|
|
|
|
class CapturableProgressBlink : ConditionalTrait<CapturableProgressBlinkInfo>, ITick, ICaptureProgressWatcher
|
|
{
|
|
readonly List<Player> captorOwners = new List<Player>();
|
|
readonly HashSet<Actor> captors = new HashSet<Actor>();
|
|
int tick = 0;
|
|
|
|
public CapturableProgressBlink(CapturableProgressBlinkInfo info)
|
|
: base(info) { }
|
|
|
|
void ICaptureProgressWatcher.Update(Actor self, Actor captor, Actor target, int current, int total)
|
|
{
|
|
if (self != target)
|
|
return;
|
|
|
|
if (total == 0)
|
|
{
|
|
captors.Remove(captor);
|
|
if (!captors.Any(c => c.Owner == captor.Owner))
|
|
captorOwners.Remove(captor.Owner);
|
|
}
|
|
else if (captors.Add(captor) && !captorOwners.Contains(captor.Owner))
|
|
captorOwners.Add(captor.Owner);
|
|
}
|
|
|
|
void ITick.Tick(Actor self)
|
|
{
|
|
if (IsTraitDisabled)
|
|
return;
|
|
|
|
if (captorOwners.Count == 0)
|
|
{
|
|
tick = 0;
|
|
return;
|
|
}
|
|
|
|
// Separate the blinks from different players by 4 ticks
|
|
if (tick / 4 < captorOwners.Count && tick % 4 == 0)
|
|
{
|
|
var captorOwner = captorOwners[tick / 4];
|
|
self.World.Add(new FlashTarget(self, captorOwner.Color));
|
|
foreach (var captor in captors)
|
|
if (captor.Owner == captorOwner)
|
|
self.World.Add(new FlashTarget(captor, captorOwner.Color));
|
|
}
|
|
|
|
if (++tick >= Info.Interval)
|
|
tick = 0;
|
|
}
|
|
}
|
|
}
|