84 lines
2.3 KiB
C#
84 lines
2.3 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Text;
|
|
|
|
namespace OpenRa.Game.Traits.Activities
|
|
{
|
|
class Harvest : Activity
|
|
{
|
|
public Activity NextActivity { get; set; }
|
|
bool isHarvesting = false;
|
|
|
|
public void Tick(Actor self, Mobile mobile)
|
|
{
|
|
var harv = self.traits.Get<Harvester>();
|
|
var isGem = false;
|
|
|
|
if (!harv.IsFull &&
|
|
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)
|
|
renderUnit.PlayCustomAnimation(self, harvestAnim, () => isHarvesting = false);
|
|
harv.AcceptResource(isGem);
|
|
return;
|
|
}
|
|
|
|
if (isHarvesting) return;
|
|
|
|
if (harv.IsFull)
|
|
PlanReturnToBase(self, mobile);
|
|
else
|
|
PlanMoreHarvesting(self, mobile);
|
|
}
|
|
|
|
/* maybe this doesnt really belong here, since it's the
|
|
* same as what UnitOrders has to do for an explicit return */
|
|
|
|
void PlanReturnToBase(Actor self, Mobile mobile)
|
|
{
|
|
/* find a proc */
|
|
var proc = ChooseReturnLocation(self);
|
|
if (proc == null)
|
|
{
|
|
Cancel(self, mobile); /* is this a sane way to cancel? */
|
|
return;
|
|
}
|
|
|
|
mobile.QueueActivity(new Move(proc.Location + new int2(1, 2)));
|
|
mobile.QueueActivity(new Turn(64));
|
|
mobile.QueueActivity(new DeliverOre());
|
|
|
|
mobile.InternalSetActivity(NextActivity);
|
|
}
|
|
|
|
static Actor ChooseReturnLocation(Actor self)
|
|
{
|
|
/* todo: compute paths to possible procs, taking into account enemy presence */
|
|
/* currently, we're good at choosing close, inaccessible procs */
|
|
|
|
return Game.world.Actors.Where(
|
|
a => a.Owner == self.Owner &&
|
|
a.traits.Contains<AcceptsOre>())
|
|
.OrderBy(p => (p.Location - self.Location).LengthSquared)
|
|
.FirstOrDefault();
|
|
}
|
|
|
|
void PlanMoreHarvesting(Actor self, Mobile mobile)
|
|
{
|
|
/* find a nearby patch */
|
|
/* todo: add the queries we need to support this! */
|
|
|
|
mobile.InternalSetActivity(NextActivity);
|
|
}
|
|
|
|
public void Cancel(Actor self, Mobile mobile)
|
|
{
|
|
mobile.InternalSetActivity(null); /* bob: anything else required? */
|
|
}
|
|
}
|
|
}
|