RuleRegion.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_RULEREGION_H
#define OPENXCOM_RULEREGION_H

#include <string>
#include <vector>
#include <yaml-cpp/yaml.h>
#include "../fmath.h"
#include "../Savegame/WeightedOptions.h"

namespace OpenXcom
{

struct MissionArea
{
    double lonMin, lonMax, latMin, latMax;

    bool operator== (const MissionArea& ma) const
    {
        return AreSame(lonMax, ma.lonMax) && AreSame(lonMin, ma.lonMin) && AreSame(latMax, ma.latMax) && AreSame(latMin, ma.latMin);
    }
};

struct MissionZone
{
    std::vector<MissionArea> areas;

    void swap(MissionZone &other)
    {
        areas.swap(other.areas);
    }
};

class City;

class RuleRegion
{
private:
    std::string _type;
    int _cost;
    std::vector<double> _lonMin, _lonMax, _latMin, _latMax;
    std::vector<City*> _cities;
    WeightedOptions _missionWeights;
    size_t _regionWeight;
    std::vector<MissionZone> _missionZones;
    std::string _missionRegion;
public:
    static const int CITY_MISSION_ZONE = 3;
    static const int ALIEN_BASE_ZONE = 4;
    RuleRegion(const std::string &type);
    ~RuleRegion();
    void load(const YAML::Node& node);
    std::string getType() const;
    int getBaseCost() const;
    bool insideRegion(double lon, double lat) const;
    std::vector<City*> *getCities();
    size_t getWeight() const;
    const WeightedOptions &getAvailableMissions() const { return _missionWeights; }
    const std::string &getMissionRegion() const { return _missionRegion; }
    std::pair<double, double> getRandomPoint(size_t site) const;
    const std::vector<double> &getLonMax() const { return _lonMax; }
    const std::vector<double> &getLonMin() const { return _lonMin; }
    const std::vector<double> &getLatMax() const { return _latMax; }
    const std::vector<double> &getLatMin() const { return _latMin; }
    const std::vector<MissionZone> &getMissionZones() const;
};

}

namespace YAML
{
    template<>
    struct convert<OpenXcom::MissionArea>
    {
        static Node encode(const OpenXcom::MissionArea& rhs)
        {
            Node node;
            node.push_back(rhs.lonMin);
            node.push_back(rhs.lonMax);
            node.push_back(rhs.latMin);
            node.push_back(rhs.latMax);
            return node;
        }

        static bool decode(const Node& node, OpenXcom::MissionArea& rhs)
        {
            if (!node.IsSequence() || node.size() != 4)
                return false;

            rhs.lonMin = node[0].as<double>();
            rhs.lonMax = node[1].as<double>();
            rhs.latMin = node[2].as<double>();
            rhs.latMax = node[3].as<double>();
            return true;
        }
    };

    template<>
    struct convert<OpenXcom::MissionZone>
    {
        static Node encode(const OpenXcom::MissionZone& rhs)
        {
            Node node;
            node = rhs.areas;
            return node;
        }

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

            rhs.areas = node.as< std::vector<OpenXcom::MissionArea> >(rhs.areas);
            return true;
        }
    };
}

#endif