SPAR/AIModule/SparAIModule/DecisionMaking/Layer1Reactive/FSM/Conditions/IsAttacking.h

Go to the documentation of this file.

#pragma once
#include "../../../../Utils/FSM/Condition.h"

class IsAttacking : public Condition
{
public:
  IsAttacking(BWAPI::Unit* unit): m_unit(unit) {}
  bool evaluate() const
  {
    BWAPI::Unit* target = m_unit->getOrderTarget();
    return target != NULL && target->getPlayer()->isEnemy(BWAPI::Broodwar->self());
  }
private:
  BWAPI::Unit* m_unit;
};

class GroupIsAttacking : public Condition
{
public:
  GroupIsAttacking(const PersistentUnitGroup* group): m_group(group) {}
  bool evaluate() const
  {
    for(PersistentUnitGroup::const_iterator it=m_group->begin(); it != m_group->end(); ++it)
    {
      BWAPI::Unit* target = (*it)->getOrderTarget();
      if(target != NULL && target->getPlayer()->isEnemy(BWAPI::Broodwar->self()))
        return true;
    }
    return false;
  }
private:
  const PersistentUnitGroup* m_group;
};