Ruleset.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_RULESET_H
#define OPENXCOM_RULESET_H

#include <map>
#include <vector>
#include <string>
#include <yaml-cpp/yaml.h>
#include "../Savegame/GameTime.h"

namespace OpenXcom
{

class SavedGame;
class SoldierNamePool;
class Soldier;
class RuleCountry;
class RuleRegion;
class RuleBaseFacility;
class RuleCraft;
class RuleCraftWeapon;
class RuleItem;
class RuleUfo;
class RuleTerrain;
class MapDataSet;
class ResourcePack;
class RuleSoldier;
class Unit;
class Armor;
class ArticleDefinition;
class RuleInventory;
class RuleResearch;
class RuleManufacture;
class AlienRace;
class AlienDeployment;
class UfoTrajectory;
class RuleAlienMission;
class City;
class Base;
class MCDPatch;
class ExtraSprites;
class ExtraSounds;
class ExtraStrings;
class StatString;

class Ruleset
{
protected:
    std::vector<SoldierNamePool*> _names;
    std::map<std::string, RuleCountry*> _countries;
    std::map<std::string, RuleRegion*> _regions;
    std::map<std::string, RuleBaseFacility*> _facilities;
    std::map<std::string, RuleCraft*> _crafts;
    std::map<std::string, RuleCraftWeapon*> _craftWeapons;
    std::map<std::string, RuleItem*> _items;
    std::map<std::string, RuleUfo*> _ufos;
    std::map<std::string, RuleTerrain*> _terrains;
    std::map<std::string, MapDataSet*> _mapDataSets;
    std::map<std::string, RuleSoldier*> _soldiers;
    std::map<std::string, Unit*> _units;
    std::map<std::string, AlienRace*> _alienRaces;
    std::map<std::string, AlienDeployment*> _alienDeployments;
    std::map<std::string, Armor*> _armors;
    std::map<std::string, ArticleDefinition*> _ufopaediaArticles;
    std::map<std::string, RuleInventory*> _invs;
    std::map<std::string, RuleResearch *> _research;
    std::map<std::string, RuleManufacture *> _manufacture;
    std::map<std::string, UfoTrajectory *> _ufoTrajectories;
    std::map<std::string, RuleAlienMission *> _alienMissions;
    std::map<std::string, MCDPatch *> _MCDPatches;
    std::vector<std::pair<std::string, ExtraSprites *> > _extraSprites;
    std::vector<std::pair<std::string, ExtraSounds *> > _extraSounds;
    std::map<std::string, ExtraStrings *> _extraStrings;
    std::vector<StatString*> _statStrings;
    int _costSoldier, _costEngineer, _costScientist, _timePersonnel, _initialFunding;
    YAML::Node _startingBase;
    GameTime _startingTime;
    std::vector<std::string> _countriesIndex, _regionsIndex, _facilitiesIndex, _craftsIndex, _craftWeaponsIndex, _itemsIndex, _invsIndex, _ufosIndex;
    std::vector<std::string> _aliensIndex, _deploymentsIndex, _armorsIndex, _ufopaediaIndex, _researchIndex, _manufactureIndex, _MCDPatchesIndex;
    std::vector<std::string> _alienMissionsIndex, _terrainIndex, _extraSpritesIndex, _extraSoundsIndex, _extraStringsIndex;
    std::vector<std::vector<int> > _alienItemLevels;
    int _modIndex, _facilityListOrder, _craftListOrder, _itemListOrder, _researchListOrder,  _manufactureListOrder, _ufopaediaListOrder, _invListOrder;
    std::vector<std::string> _psiRequirements; // it's a cache for psiStrengthEval
    void loadFile(const std::string &filename);
    void loadFiles(const std::string &dirname);
    template <typename T>
    T *loadRule(const YAML::Node &node, std::map<std::string, T*> *map, std::vector<std::string> *index = 0, const std::string &key = "type");
public:
    Ruleset();
    ~Ruleset();
    void load(const std::string &source);
    SavedGame *newSave() const;
    const std::vector<SoldierNamePool*> &getPools() const;
    RuleCountry *getCountry(const std::string &id) const;
    const std::vector<std::string> &getCountriesList() const;
    RuleRegion *getRegion(const std::string &id) const;
    const std::vector<std::string> &getRegionsList() const;
    RuleBaseFacility *getBaseFacility(const std::string &id) const;
    const std::vector<std::string> &getBaseFacilitiesList() const;
    RuleCraft *getCraft(const std::string &id) const;
    const std::vector<std::string> &getCraftsList() const;
    RuleCraftWeapon *getCraftWeapon(const std::string &id) const;
    const std::vector<std::string> &getCraftWeaponsList() const;
    RuleItem *getItem(const std::string &id) const;
    const std::vector<std::string> &getItemsList() const;
    RuleUfo *getUfo(const std::string &id) const;
    const std::vector<std::string> &getUfosList() const;
    const std::vector<std::string> &getTerrainList() const;
    RuleTerrain *getTerrain(const std::string &name) const;
    MapDataSet *getMapDataSet(const std::string &name);
    RuleSoldier *getSoldier(const std::string &name) const;
    Unit *getUnit(const std::string &name) const;
    AlienRace *getAlienRace(const std::string &name) const;
    const std::vector<std::string> &getAlienRacesList() const;
    AlienDeployment *getDeployment(const std::string &name) const;
    const std::vector<std::string> &getDeploymentsList() const;
    Armor *getArmor(const std::string &name) const;
    const std::vector<std::string> &getArmorsList() const;
    ArticleDefinition *getUfopaediaArticle(const std::string &name) const;
    const std::vector<std::string> &getUfopaediaList() const;
    std::map<std::string, RuleInventory*> *getInventories();
    RuleInventory *getInventory(const std::string &id) const;
    int getSoldierCost() const;
    int getEngineerCost() const;
    int getScientistCost() const;
    int getPersonnelTime() const;
    RuleResearch *getResearch (const std::string &id) const;
    const std::vector<std::string> &getResearchList () const;
    RuleManufacture *getManufacture (const std::string &id) const;
    const std::vector<std::string> &getManufactureList () const;
    std::vector<OpenXcom::RuleBaseFacility*> getCustomBaseFacilities() const;
    const UfoTrajectory *getUfoTrajectory(const std::string &id) const;
    const RuleAlienMission *getAlienMission(const std::string &id) const;
    const std::vector<std::string> &getAlienMissionList() const;
    const City *locateCity(double lon, double lat) const;
    const std::vector<std::vector<int> > &getAlienItemLevels() const;
    const YAML::Node &getStartingBase();
    MCDPatch *getMCDPatch(const std::string name) const;
    std::vector<std::pair<std::string, ExtraSprites *> > getExtraSprites() const;
    std::vector<std::pair<std::string, ExtraSounds *> > getExtraSounds() const;
    std::map<std::string, ExtraStrings *> getExtraStrings() const;
    std::vector<StatString *> getStatStrings() const;    
    void sortLists();
    std::vector<std::string> getPsiRequirements() const;
    const std::vector<std::string> &getInvsList () const;
    Soldier *genSoldier(SavedGame *save) const;

};

}

#endif