50 lines
1.4 KiB
C#
50 lines
1.4 KiB
C#
#region Copyright & License Information
|
|
/*
|
|
* Copyright 2007-2010 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 LICENSE.
|
|
*/
|
|
#endregion
|
|
|
|
using System.Linq;
|
|
using OpenRA.Traits;
|
|
using OpenRA.Traits.Activities;
|
|
|
|
namespace OpenRA.Mods.RA.Activities
|
|
{
|
|
class CaptureBuilding : CancelableActivity
|
|
{
|
|
Actor target;
|
|
|
|
public CaptureBuilding(Actor target) { this.target = target; }
|
|
|
|
public override IActivity Tick(Actor self)
|
|
{
|
|
if (IsCanceled) return NextActivity;
|
|
if (target == null || !target.IsInWorld || target.IsDead()) return NextActivity;
|
|
if (target.Owner == self.Owner) return NextActivity;
|
|
|
|
if( !target.Trait<IOccupySpace>().OccupiedCells().Any( x => x == self.Location ) )
|
|
return NextActivity;
|
|
|
|
// todo: clean this up
|
|
self.World.AddFrameEndTask(w =>
|
|
{
|
|
// momentarily remove from world so the ownership queries don't get confused
|
|
var oldOwner = target.Owner;
|
|
w.Remove(target);
|
|
target.Owner = self.Owner;
|
|
w.Add(target);
|
|
|
|
foreach (var t in target.TraitsImplementing<INotifyCapture>())
|
|
t.OnCapture(target, self, oldOwner, self.Owner);
|
|
|
|
self.Destroy();
|
|
});
|
|
return this;
|
|
}
|
|
}
|
|
}
|