Unit.h

/*
 * Copyright 2010-2014 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/>.
 */
#ifndef OPENXCOM_UNIT_H
#define OPENXCOM_UNIT_H

#include <string>
#include <yaml-cpp/yaml.h>

namespace OpenXcom
{

enum SpecialAbility { SPECAB_NONE = 0, SPECAB_EXPLODEONDEATH, SPECAB_BURNFLOOR, SPECAB_RESPAWN };
struct UnitStats
{
    int tu, stamina, health, bravery, reactions, firing, throwing, strength, psiStrength, psiSkill, melee;
public:
    UnitStats() : tu(0), stamina(0), health(0), bravery(0), reactions(0), firing(0), throwing(0), strength(0), psiStrength(0), psiSkill(0), melee(0) {};
    UnitStats(int tu_, int stamina_, int health_, int bravery_, int reactions_, int firing_, int throwing_, int strength_, int psiStrength_, int psiSkill_, int melee_) : tu(tu_), stamina(stamina_), health(health_), bravery(bravery_), reactions(reactions_), firing(firing_), throwing(throwing_), strength(strength_), psiStrength(psiStrength_), psiSkill(psiSkill_), melee(melee_) {};
    UnitStats& operator+=(const UnitStats& stats) { tu += stats.tu; stamina += stats.stamina; health += stats.health; bravery += stats.bravery; reactions += stats.reactions; firing += stats.firing; throwing += stats.throwing; strength += stats.strength; psiStrength += stats.psiStrength; psiSkill += stats.psiSkill; melee += stats.melee; return *this; }
    UnitStats operator+(const UnitStats& stats) const { return UnitStats(tu + stats.tu, stamina + stats.stamina, health + stats.health, bravery + stats.bravery, reactions + stats.reactions, firing + stats.firing, throwing + stats.throwing, strength + stats.strength, psiStrength + stats.psiStrength, psiSkill + stats.psiSkill, melee + stats.melee); }
    UnitStats& operator-=(const UnitStats& stats) { tu -= stats.tu; stamina -= stats.stamina; health -= stats.health; bravery -= stats.bravery; reactions -= stats.reactions; firing -= stats.firing; throwing -= stats.throwing; strength -= stats.strength; psiStrength -= stats.psiStrength; psiSkill -= stats.psiSkill; melee -= stats.melee; return *this; }
    UnitStats operator-(const UnitStats& stats) const { return UnitStats(tu - stats.tu, stamina - stats.stamina, health - stats.health, bravery - stats.bravery, reactions - stats.reactions, firing - stats.firing, throwing - stats.throwing, strength - stats.strength, psiStrength - stats.psiStrength, psiSkill - stats.psiSkill, melee - stats.melee); }
    UnitStats operator-() const { return UnitStats(-tu, -stamina, -health, -bravery, -reactions, -firing, -throwing, -strength, -psiStrength, -psiSkill, -melee); }
    void merge(const UnitStats& stats) { tu = (stats.tu ? stats.tu : tu); stamina = (stats.stamina ? stats.stamina : stamina); health = (stats.health ? stats.health : health); bravery = (stats.bravery ? stats.bravery : bravery); reactions = (stats.reactions ? stats.reactions : reactions); firing = (stats.firing ? stats.firing : firing); throwing = (stats.throwing ? stats.throwing : throwing); strength = (stats.strength ? stats.strength : strength); psiStrength = (stats.psiStrength ? stats.psiStrength : psiStrength); psiSkill = (stats.psiSkill ? stats.psiSkill : psiSkill); melee = (stats.melee ? stats.melee : melee); };
};

class Unit
{
private:
    std::string _type;
    std::string _race;
    std::string _rank;
    UnitStats _stats;
    std::string _armor;
    int _standHeight, _kneelHeight, _floatHeight;
    int _value, _deathSound, _aggroSound, _moveSound;
    int _intelligence, _aggression, _energyRecovery;
    SpecialAbility _specab;
    std::string _spawnUnit;
    bool _livingWeapon;
public:
    Unit(const std::string &type);
    ~Unit();
    void load(const YAML::Node& node, int modIndex);
    std::string getType() const;
    UnitStats *getStats();
    int getStandHeight() const;
    int getKneelHeight() const;
    int getFloatHeight() const;
    std::string getArmor() const;
    std::string getRace() const;
    std::string getRank() const;
    int getValue() const;
    int getDeathSound() const;
    int getMoveSound() const;
    int getIntelligence() const;
    int getAggression() const;
    int getSpecialAbility() const;
    std::string getSpawnUnit() const;
    int getAggroSound() const;
    bool isLivingWeapon() const;
    int getEnergyRecovery() const;
};

}

namespace YAML
{
    template<>
    struct convert<OpenXcom::UnitStats>
    {
        static Node encode(const OpenXcom::UnitStats& rhs)
        {
            Node node;
            node["tu"] = rhs.tu;
            node["stamina"] = rhs.stamina;
            node["health"] = rhs.health;
            node["bravery"] = rhs.bravery;
            node["reactions"] = rhs.reactions;
            node["firing"] = rhs.firing;
            node["throwing"] = rhs.throwing;
            node["strength"] = rhs.strength;
            node["psiStrength"] = rhs.psiStrength;
            node["psiSkill"] = rhs.psiSkill;
            node["melee"] = rhs.melee;
            return node;
        }

        static bool decode(const Node& node, OpenXcom::UnitStats& rhs)
        {
            if (!node.IsMap())
                return false;

            rhs.tu = node["tu"].as<int>(rhs.tu);
            rhs.stamina = node["stamina"].as<int>(rhs.stamina);
            rhs.health = node["health"].as<int>(rhs.health);
            rhs.bravery = node["bravery"].as<int>(rhs.bravery);
            rhs.reactions = node["reactions"].as<int>(rhs.reactions);
            rhs.firing = node["firing"].as<int>(rhs.firing);
            rhs.throwing = node["throwing"].as<int>(rhs.throwing);
            rhs.strength = node["strength"].as<int>(rhs.strength);
            rhs.psiStrength = node["psiStrength"].as<int>(rhs.psiStrength);
            rhs.psiSkill = node["psiSkill"].as<int>(rhs.psiSkill);
            rhs.melee = node["melee"].as<int>(rhs.melee);
            return true;
        }
    };
}

#endif