91 lines
2.4 KiB
C#
91 lines
2.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 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
|
|
{
|
|
public class OreRefineryDockActionInfo : TraitInfo<OreRefineryDockAction> {}
|
|
|
|
public class OreRefineryDockAction : IAcceptOreDockAction, INotifyCapture
|
|
{
|
|
public virtual IActivity DockSequence(Actor harv, Actor self)
|
|
{
|
|
return new RAHarvesterDockSequence(harv, self);
|
|
}
|
|
|
|
Actor dockedHarv = null;
|
|
bool preventDock = false;
|
|
public void OnDock(Actor self, Actor harv, DeliverResources dockOrder)
|
|
{
|
|
var mobile = harv.Trait<Mobile>();
|
|
var harvester = harv.Trait<Harvester>();
|
|
|
|
if (!preventDock)
|
|
{
|
|
harv.QueueActivity( new CallFunc( () => dockedHarv = harv, false ) );
|
|
harv.QueueActivity( DockSequence(harv, self) );
|
|
harv.QueueActivity( new CallFunc( () => dockedHarv = null, false ) );
|
|
}
|
|
|
|
// Tell the harvester to start harvesting
|
|
// TODO: This belongs on the harv idle activity
|
|
harv.QueueActivity( new CallFunc( () =>
|
|
{
|
|
if (harvester.LastHarvestedCell != int2.Zero)
|
|
{
|
|
harv.QueueActivity( mobile.MoveTo(harvester.LastHarvestedCell, 5) );
|
|
harv.SetTargetLine(Target.FromCell(harvester.LastHarvestedCell), Color.Red, false);
|
|
}
|
|
harv.QueueActivity( new Harvest() );
|
|
}));
|
|
}
|
|
|
|
public void Tick(Actor self)
|
|
{
|
|
// Harvester was killed while unloading
|
|
if (dockedHarv != null && dockedHarv.IsDead())
|
|
{
|
|
self.Trait<RenderBuilding>().CancelCustomAnim(self);
|
|
dockedHarv = null;
|
|
}
|
|
}
|
|
|
|
void CancelDock(Actor self)
|
|
{
|
|
preventDock = true;
|
|
|
|
// Cancel the dock sequence
|
|
if (dockedHarv != null && !dockedHarv.IsDead())
|
|
dockedHarv.CancelActivity();
|
|
}
|
|
|
|
public void Selling (Actor self) { CancelDock(self); }
|
|
public void Sold (Actor self) {}
|
|
|
|
public void Damaged (Actor self, AttackInfo e)
|
|
{
|
|
if (e.DamageState == DamageState.Dead)
|
|
CancelDock(self);
|
|
}
|
|
|
|
public void OnCapture (Actor self, Actor captor, Player oldOwner, Player newOwner)
|
|
{
|
|
if (dockedHarv != null)
|
|
dockedHarv.ChangeOwner(newOwner);
|
|
}
|
|
}
|
|
}
|