AIModule.h

#pragma once
/*
 * Copyright 2010-2016 OpenXcom Developers.
 *
 * This file is part of OpenXcom.
 *
 * OpenXcom is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * OpenXcom is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with OpenXcom.  If not, see <http://www.gnu.org/licenses/>.
 */
#include <yaml-cpp/yaml.h>
#include "BattlescapeGame.h"
#include "Position.h"
#include "../Savegame/BattleUnit.h"
#include <vector>


namespace OpenXcom
{

class SavedBattleGame;
class BattleUnit;
struct BattleAction;
class BattlescapeState;
class Node;

enum AIMode { AI_PATROL, AI_AMBUSH, AI_COMBAT, AI_ESCAPE };
class AIModule
{
private:
    SavedBattleGame *_save;
    BattleUnit *_unit;
    BattleUnit *_aggroTarget;
    int _knownEnemies, _visibleEnemies, _spottingEnemies;
    int _escapeTUs, _ambushTUs;
    BattleAction *_escapeAction, *_ambushAction, *_attackAction, *_patrolAction, *_psiAction;
    bool _rifle, _melee, _blaster;
    bool _traceAI, _didPsi;
    int _AIMode, _intelligence, _closestDist;
    Node *_fromNode, *_toNode;
    std::vector<int> _reachable, _reachableWithAttack, _wasHitBy;
    BattleActionType _reserve;
    UnitFaction _targetFaction;
public:
    AIModule(SavedBattleGame *save, BattleUnit *unit, Node *node);
    ~AIModule();
    void load(const YAML::Node& node);
    YAML::Node save() const;
    void think(BattleAction *action);
    void setWasHitBy(BattleUnit *attacker);
    bool getWasHitBy(int attacker) const;
    void setupPatrol();
    void setupAmbush();
    void setupAttack();
    void setupEscape();
    int countKnownTargets() const;
    int getSpottingUnits(const Position& pos) const;
    int selectNearestTarget();
    bool selectClosestKnownEnemy();
    bool selectRandomTarget();
    bool selectPointNearTarget(BattleUnit *target, int maxTUs) const;
    void evaluateAIMode();
    bool findFirePoint();
    bool explosiveEfficacy(Position targetPos, BattleUnit *attackingUnit, int radius, int diff, bool grenade = false) const;
    bool getNodeOfBestEfficacy(BattleAction *action);
    void meleeAction();
    void wayPointAction();
    void projectileAction();
    void selectFireMethod();
    void grenadeAction();
    bool psiAction();
    void meleeAttack();
    bool validTarget(BattleUnit* unit, bool assessDanger, bool includeCivs) const;
    BattleActionType getReserveMode();
    void selectMeleeOrRanged();
    BattleUnit* getTarget();
};

}