78 lines
2.0 KiB
C#
78 lines
2.0 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 OpenRA.Activities;
|
|
using OpenRA.Mods.Common.Traits;
|
|
using OpenRA.Traits;
|
|
|
|
namespace OpenRA.Mods.Common.Activities
|
|
{
|
|
public class Repair : Activity
|
|
{
|
|
readonly RepairsUnitsInfo repairsUnits;
|
|
readonly Actor host;
|
|
int remainingTicks;
|
|
Health health;
|
|
bool played = false;
|
|
|
|
public Repair(Actor host)
|
|
{
|
|
this.host = host;
|
|
repairsUnits = host.Info.Traits.Get<RepairsUnitsInfo>();
|
|
}
|
|
|
|
public override Activity Tick(Actor self)
|
|
{
|
|
if (IsCanceled) return NextActivity;
|
|
if (host == null || !host.IsInWorld) return NextActivity;
|
|
|
|
health = self.TraitOrDefault<Health>();
|
|
if (health == null) return NextActivity;
|
|
|
|
if (health.DamageState == DamageState.Undamaged)
|
|
{
|
|
Sound.PlayNotification(self.World.Map.Rules, self.Owner, "Speech", repairsUnits.FinishRepairingNotification, self.Owner.Country.Race);
|
|
return NextActivity;
|
|
}
|
|
|
|
if (remainingTicks == 0)
|
|
{
|
|
var unitCost = self.Info.Traits.Get<ValuedInfo>().Cost;
|
|
var hpToRepair = repairsUnits.HpPerStep;
|
|
var cost = Math.Max(1, (hpToRepair * unitCost * repairsUnits.ValuePercentage) / (health.MaxHP * 100));
|
|
|
|
if (!played)
|
|
{
|
|
played = true;
|
|
Sound.PlayNotification(self.World.Map.Rules, self.Owner, "Speech", repairsUnits.StartRepairingNotification, self.Owner.Country.Race);
|
|
}
|
|
|
|
if (!self.Owner.PlayerActor.Trait<PlayerResources>().TakeCash(cost))
|
|
{
|
|
remainingTicks = 1;
|
|
return this;
|
|
}
|
|
|
|
self.InflictDamage(self, -hpToRepair, null);
|
|
|
|
foreach (var depot in host.TraitsImplementing<INotifyRepair>())
|
|
depot.Repairing(self, host);
|
|
|
|
remainingTicks = repairsUnits.Interval;
|
|
}
|
|
else
|
|
--remainingTicks;
|
|
|
|
return this;
|
|
}
|
|
}
|
|
}
|