71 lines
1.8 KiB
C#
71 lines
1.8 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Text;
|
|
|
|
namespace OpenRa.Game.Traits.Activities
|
|
{
|
|
class Harvest : IActivity
|
|
{
|
|
public IActivity NextActivity { get; set; }
|
|
bool isHarvesting = false;
|
|
|
|
public void Tick(Actor self, Mobile mobile)
|
|
{
|
|
if( isHarvesting ) return;
|
|
|
|
if( NextActivity != null )
|
|
{
|
|
mobile.InternalSetActivity( NextActivity );
|
|
NextActivity.Tick( self, mobile );
|
|
return;
|
|
}
|
|
|
|
var harv = self.traits.Get<Harvester>();
|
|
|
|
if( harv.IsFull )
|
|
{
|
|
mobile.QueueActivity( new DeliverOre() );
|
|
mobile.InternalSetActivity( NextActivity );
|
|
}
|
|
|
|
var isGem = false;
|
|
if (Game.map.ContainsResource(self.Location) &&
|
|
Game.map.Harvest(self.Location, out isGem))
|
|
{
|
|
var harvestAnim = "harvest" + Util.QuantizeFacing(mobile.facing, 8);
|
|
var renderUnit = self.traits.WithInterface<RenderUnit>().First(); /* better have one of these! */
|
|
if( harvestAnim != renderUnit.anim.CurrentSequence.Name )
|
|
{
|
|
isHarvesting = true;
|
|
renderUnit.PlayCustomAnimation( self, harvestAnim, () => isHarvesting = false );
|
|
}
|
|
harv.AcceptResource(isGem);
|
|
return;
|
|
}
|
|
|
|
PlanMoreHarvesting( self, mobile );
|
|
}
|
|
|
|
void PlanMoreHarvesting(Actor self, Mobile mobile)
|
|
{
|
|
mobile.QueueActivity(new Move(
|
|
() =>
|
|
{
|
|
var search = new PathSearch
|
|
{
|
|
heuristic = loc => (Game.map.ContainsResource(loc) ? 0 : 1),
|
|
umt = UnitMovementType.Wheel,
|
|
checkForBlocked = true
|
|
};
|
|
search.AddInitialCell(self.Location);
|
|
return Game.PathFinder.FindPath(search);
|
|
}));
|
|
mobile.QueueActivity(new Harvest());
|
|
mobile.InternalSetActivity( NextActivity );
|
|
}
|
|
|
|
public void Cancel(Actor self, Mobile mobile) { }
|
|
}
|
|
}
|