void CBuilderCAI::ExecuteResurrect(Command& c)
{
// not all builders are resurrect-capable by default
if (!owner->unitDef->canResurrect)
return;
if (c.params.size() == 1) {
unsigned int id = (unsigned int) c.params[0];
if (id >= unitHandler->MaxUnits()) { // resurrect feature
CFeature* feature = featureHandler->GetFeature(id - unitHandler->MaxUnits());
if (feature && feature->udef != NULL) {
if (((c.options & INTERNAL_ORDER) && !(c.options & CONTROL_KEY) && IsFeatureBeingReclaimed(feature->id, owner)) ||
!ResurrectObject(feature)) {
StopMove();
RemoveUnitFromResurrecters(owner);
FinishCommand();
}
else {
AddUnitToResurrecters(owner);
}
} else {
RemoveUnitFromResurrecters(owner);
if (ownerBuilder->lastResurrected && unitHandler->GetUnitUnsafe(ownerBuilder->lastResurrected) != NULL && owner->unitDef->canRepair) {
// resurrection finished, start repair (by overwriting the current order)
c = Command(CMD_REPAIR, c.options | INTERNAL_ORDER, ownerBuilder->lastResurrected);
ownerBuilder->lastResurrected = 0;
inCommand = false;
SlowUpdate();
return;
}
StopMove();
FinishCommand();
}
} else { // resurrect unit
RemoveUnitFromResurrecters(owner);
FinishCommand();
}
} else if (c.params.size() == 4) {
// area resurrect
const float3 pos = c.GetPos(0);
const float radius = c.params[3];
if (FindResurrectableFeatureAndResurrect(pos, radius, c.options, (c.options & META_KEY))) {
inCommand = false;
SlowUpdate();
return;
}
if (!(c.options & ALT_KEY)) {
FinishCommand();
}
} else {
// wrong number of parameters
RemoveUnitFromResurrecters(owner);
FinishCommand();
}
}
bool CBuilderCAI::IsFeatureBeingResurrected(int featureId, CUnit *friendUnit)
{
bool retval = false;
std::list<CUnit*> rm;
for (CUnitSet::iterator it = resurrecters.begin(); it != resurrecters.end(); ++it) {
if ((*it)->commandAI->commandQue.empty()) {
rm.push_back(*it);
continue;
}
const Command& c = (*it)->commandAI->commandQue.front();
if (c.GetID() != CMD_RESURRECT || c.params.size() != 1) {
rm.push_back(*it);
continue;
}
const int cmdFeatureId = (int)c.params[0];
if (cmdFeatureId-uh->MaxUnits() == featureId && (!friendUnit || teamHandler->Ally(friendUnit->allyteam, (*it)->allyteam))) {
retval = true;
break;
}
}
for (std::list<CUnit*>::iterator it = rm.begin(); it != rm.end(); ++it)
RemoveUnitFromResurrecters(*it);
return retval;
}
CBuilderCAI::~CBuilderCAI()
{
// if uh == NULL then all pointers to units should be considered dangling pointers
if (uh) {
RemoveUnitFromReclaimers(owner);
RemoveUnitFromFeatureReclaimers(owner);
RemoveUnitFromResurrecters(owner);
uh->RemoveBuilderCAI(this);
}
}