76 lines
2.2 KiB
C#
76 lines
2.2 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 OpenRA.Mods.RA.Activities;
|
|
using OpenRA.Mods.RA.Render;
|
|
using OpenRA.Traits;
|
|
using OpenRA.Traits.Activities;
|
|
using System.Drawing;
|
|
using OpenRA.Mods.RA.Move;
|
|
|
|
namespace OpenRA.Mods.RA
|
|
{
|
|
class OreRefineryDockActionInfo : TraitInfo<OreRefineryDockAction> {}
|
|
|
|
class OreRefineryDockAction : IAcceptOreDockAction, INotifyCapture
|
|
{
|
|
|
|
Actor dockedHarv = null;
|
|
public void OnDock(Actor self, Actor harv, DeliverResources dockOrder)
|
|
{
|
|
var harvester = harv.Trait<Harvester>();
|
|
|
|
if (harv.Trait<IFacing>().Facing != 64)
|
|
harv.QueueActivity (new Turn (64));
|
|
|
|
harv.QueueActivity (new CallFunc (() =>
|
|
{
|
|
dockedHarv = harv;
|
|
var renderUnit = harv.Trait<RenderUnit> ();
|
|
if (renderUnit.anim.CurrentSequence.Name != "empty")
|
|
renderUnit.PlayCustomAnimation (harv, "empty", () =>
|
|
{
|
|
harvester.Deliver(harv, self);
|
|
harv.QueueActivity( new CallFunc( () => dockedHarv = null, false ) );
|
|
|
|
if (harvester.LastHarvestedCell != int2.Zero)
|
|
{
|
|
var mobile = harv.Trait<Mobile>();
|
|
harv.QueueActivity( mobile.MoveTo(harvester.LastHarvestedCell, 5) );
|
|
if (harv.Owner == self.World.LocalPlayer)
|
|
self.World.AddFrameEndTask( w =>
|
|
{
|
|
if (harv.Destroyed) return;
|
|
var line = harv.TraitOrDefault<DrawLineToTarget>();
|
|
if (line != null)
|
|
line.SetTargetSilently(harv, Target.FromCell(harvester.LastHarvestedCell), Color.Green);
|
|
});
|
|
}
|
|
harv.QueueActivity( new Harvest() );
|
|
});
|
|
}));
|
|
}
|
|
|
|
public void OnCapture (Actor self, Actor captor, Player oldOwner, Player newOwner)
|
|
{
|
|
if (dockedHarv == null)
|
|
return;
|
|
|
|
dockedHarv.World.AddFrameEndTask(w =>
|
|
{
|
|
// momentarily remove from world so the ownership queries don't get confused
|
|
w.Remove(dockedHarv);
|
|
dockedHarv.Owner = captor.Owner;
|
|
w.Add(dockedHarv);
|
|
});
|
|
}
|
|
}
|
|
}
|