map_context.cpp

Go to the documentation of this file.

/*
    Copyright (C) 2008 - 2025
    by Tomasz Sniatowski <kailoran@gmail.com>
    Part of the Battle for Wesnoth Project https://www.wesnoth.org/
 
    This program 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 2 of the License, or
    (at your option) any later version.
    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY.
 
    See the COPYING file for more details.
*/
 
#define GETTEXT_DOMAIN "wesnoth-editor"
 
#include "editor/map/map_context.hpp"
 
#include "display.hpp"
#include "editor/action/action.hpp"
#include "filesystem.hpp"
#include "formula/string_utils.hpp"
#include "gettext.hpp"
#include "gui/dialogs/message.hpp"
#include "gui/dialogs/transient_message.hpp"
#include "map/label.hpp"
#include "preferences/preferences.hpp"
#include "serialization/binary_or_text.hpp"
#include "serialization/parser.hpp"
#include "serialization/preprocessor.hpp"
#include "team.hpp"
#include "units/unit.hpp"
 
#include <boost/algorithm/string/predicate.hpp>
#include <boost/algorithm/string/trim.hpp>
#include <boost/regex.hpp>
 
namespace editor
{
editor_team_info::editor_team_info(const team& t)
    : side(t.side())
    , id(t.team_name())
    , name(t.user_team_name())
    , recruit_list(utils::join(t.recruits(), ","))
    , gold(t.gold())
    , income(t.base_income())
    , village_income(t.village_gold())
    , village_support(t.village_support())
    , fog(t.uses_fog())
    , shroud(t.uses_shroud())
    , share_vision(t.share_vision())
    , controller(t.controller())
    , no_leader(t.no_leader())
    , hidden(t.hidden())
{
}
 
const std::size_t map_context::max_action_stack_size_ = 100;
 
namespace {
    static const int editor_team_default_gold = 100;
}
 
map_context::map_context(const editor_map& map, bool pure_map, const std::string& addon_id)
    : filename_()
    , map_data_key_()
    , embedded_(false)
    , pure_map_(pure_map)
    , map_(map)
    , undo_stack_()
    , redo_stack_()
    , actions_since_save_(0)
    , needs_reload_(false)
    , needs_terrain_rebuild_(false)
    , needs_labels_reset_(false)
    , everything_changed_(false)
    , changed_locations_()
    , starting_position_label_locs_()
    , addon_id_(addon_id)
    , previous_cfg_()
    , scenario_id_()
    , scenario_name_()
    , scenario_description_()
    , xp_mod_()
    , victory_defeated_(true)
    , random_time_(false)
    , active_area_(-1)
    , labels_(nullptr)
    , units_()
    , teams_()
    , tod_manager_(new tod_manager)
    , mp_settings_()
    , game_classification_()
    , music_tracks_()
    , last_map_generator_(nullptr)
{
}
 
// Look for attr in file_contents, if there is an equal sign on the same line assume that what follows to the line end is the attribute value
static std::string get_map_location(const std::string& file_contents, const std::string& attr)
{
    std::size_t attr_name_start = file_contents.find(attr);
    if(attr_name_start == std::string::npos) return "";
 
    std::size_t attr_value_start = file_contents.find("=", attr_name_start);
    std::size_t line_end = file_contents.find("\n", attr_name_start);
    if(line_end < attr_value_start) return ""; // assume not both values are string::npos
 
    attr_value_start++;
    std::string attr_value = file_contents.substr(attr_value_start, line_end - attr_value_start);
 
    return boost::trim_copy(attr_value);
}
 
map_context::map_context(const std::string& filename, const std::string& addon_id)
    : filename_(filename)
    , map_data_key_()
    , embedded_(false)
    , pure_map_(false)
    , map_()
    , undo_stack_()
    , redo_stack_()
    , actions_since_save_(0)
    , needs_reload_(false)
    , needs_terrain_rebuild_(false)
    , needs_labels_reset_(false)
    , everything_changed_(false)
    , changed_locations_()
    , starting_position_label_locs_()
    , addon_id_(addon_id)
    , previous_cfg_()
    , scenario_id_()
    , scenario_name_()
    , scenario_description_()
    , xp_mod_()
    , victory_defeated_(true)
    , random_time_(false)
    , active_area_(-1)
    , labels_(nullptr)
    , units_()
    , teams_()
    , tod_manager_(new tod_manager)
    , mp_settings_()
    , game_classification_()
    , music_tracks_()
    , last_map_generator_(nullptr)
{
    /*
     * Overview of situations possibly found in the file:
     *
     * embedded_ - the map data is directly in the scenario file
     * pure_map_ - the map data is in its own separate file (map_file, map_data+macro inclusion) or this is a .map file
     *
     * an editor-generated file uses neither of these and is its own thing - it's not embedded (since the editor now saves using map_file) and it's not a pure map since there's also scenario data involved
     *
     * 0. Not a scenario or map file.
     *    0.1 File not found
     *    0.2 Map file empty
     *    0.3 Not a .map or .cfg file
     * 1. It's a .map file.
     *    * embedded_ = false
     *    * pure_map_ = true
     * 2. A scenario embedding the map
     *    * embedded_ = true
     *    * pure_map_ = true
     *    The scenario-test.cfg for example.
     *    The map is written back to the file.
     * 3. The map file is referenced by map_data={MACRO_ARGUEMENT}.
     *    * embedded_ = false
     *    * pure_map_ = true
     * 4. The file contains an editor generated scenario file.
     *    * embedded_ = false
     *    * pure_map_ = false
     * 5. The file is using map_file.
     *    5.1 The file doesn't contain a macro and so can be loaded by the editor as a scenario
     *       * embedded_ = false
     *       * pure_map_ = false
     *    5.2 The file contains a macro and so can't be loaded by the editor as a scenario
     *       * embedded_ = false
     *       * pure_map_ = true
     */
 
    log_scope2(log_editor, "Loading file " + filename);
 
    // 0.1 File not found
    if(!filesystem::file_exists(filename) || filesystem::is_directory(filename)) {
        throw editor_map_load_exception(filename, _("File not found"));
    }
 
    std::string file_string = filesystem::read_file(filename);
 
    // 0.2 Map file empty
    if(file_string.empty()) {
        std::string message = _("Empty file");
        throw editor_map_load_exception(filename, message);
    }
 
    // 0.3 Not a .map or .cfg file
    if(!filesystem::is_map(filename)
        && !filesystem::is_mask(filename)
        && !filesystem::is_cfg(filename))
    {
        std::string message = _("File does not have .map, .cfg, or .mask extension");
        throw editor_map_load_exception(filename, message);
    }
 
    // 1.0 Pure map data
    if(filesystem::is_map(filename)
        || filesystem::is_mask(filename)) {
        LOG_ED << "Loading map or mask file";
        map_ = editor_map::from_string(file_string); // throws on error
        pure_map_ = true;
 
        add_to_recent_files();
        return;
    }
 
    // 2.0 try loading cfg file
    LOG_ED << "Try loading scenario file";
    bool loaded = false;
    try {
        loaded = load_scenario();
    } catch(const std::exception& e) {
        utils::string_map symbols;
        symbols["error"] = e.what();
        const std::string msg = VGETTEXT("Failed to load the scenario: $error\n\nAttempting to load only the map.", symbols);
        gui2::show_message(_("Error"), msg, gui2::dialogs::message::auto_close);
 
        std::string map_data_loc = get_map_location(file_string, "map_data");
        std::string map_file_loc = get_map_location(file_string, "map_file");
 
        if(!map_data_loc.empty()) {
            if(map_data_loc.find("\"{") == std::string::npos) {
                // 2.1 Embedded pure map
                LOG_ED << "Loading embedded map_data";
                embedded_ = true;
                pure_map_ = true;
                // build on sloppy parsing from get_map_location
                std::size_t start = file_string.find(map_data_loc)+1; // assume that the "parsed" attribute value is not found earlier in the scenario file and assume it starts with a double quote
                std::size_t length = file_string.find("\"", start)-start; // assume the attribute value ends with a double quote and does not contain one
                std::string map_data = file_string.substr(start, length);
                map_ = editor_map::from_string(map_data);
            } else {
                // 2.2 Macro referenced pure map
                // assume `map_data = "{...}"` so skip the quotes and curly braces
                const std::string& macro_argument = map_data_loc.substr(2, map_data_loc.size()-4);
                LOG_ED << "Map looks like a scenario, trying {" << macro_argument << "}";
 
                auto new_filename = filesystem::get_wml_location(macro_argument, filesystem::directory_name(filesystem::get_short_wml_path(filename_)));
 
                if(!new_filename) {
                    std::string message = _("The file looks like a scenario, but the map_data attribute does not point to an existing file") + std::string("\n") + macro_argument;
                    throw editor_map_load_exception(filename, message);
                }
 
                LOG_ED << "New filename is: " << new_filename.value();
 
                pure_map_ = true;
                filename_ = new_filename.value();
                map_ = editor_map::from_string(filesystem::read_file(filename_));
            }
        } else if(!map_file_loc.empty()) {
            // 2.3 map_file referenced pure map
 
            const std::string& new_filename = filesystem::get_map_file(map_file_loc);
            if(!filesystem::file_exists(new_filename)) {
                std::string message = _("The file looks like a scenario, but the map_file attribute does not point to an existing file") + std::string("\n") + map_file_loc;
                throw editor_map_load_exception(filename, message);
            }
 
            LOG_ED << "New filename is: " << new_filename;
 
            pure_map_ = true;
            filename_ = new_filename;
            map_ = editor_map::from_string(filesystem::read_file(filename_));
        } else {
            // 2.4 could not load the scenario or find the map data in it; throw error from load_scenario
            throw editor_map_load_exception(filename, e.what());
        }
    }
    if(!loaded) {
        throw editor_map_load_exception(filename, "Aborted");
    }
    add_to_recent_files();
}
 
map_context::~map_context()
{
    undo_stack_.clear();
    redo_stack_.clear();
}
 
void map_context::new_side()
{
    teams_.emplace_back();
 
    config cfg;
    cfg["side"] = teams_.size(); // side is 1-indexed, so we can just use size()
    cfg["hidden"] = false;
    cfg["gold"] = editor_team_default_gold;
 
    teams_.back().build(cfg, map());
 
    ++actions_since_save_;
}
 
void map_context::set_side_setup(editor_team_info& info)
{
    assert(teams_.size() >= static_cast<unsigned int>(info.side));
 
    team& t = teams_[info.side - 1];
    t.change_team(info.id, info.name);
    t.set_recruits(utils::split_set(info.recruit_list, ','));
    t.have_leader(!info.no_leader);
    t.change_controller(info.controller);
    t.set_gold(info.gold);
    t.set_base_income(info.income);
    t.set_hidden(info.hidden);
    t.set_fog(info.fog);
    t.set_shroud(info.shroud);
    t.set_share_vision(info.share_vision);
    t.set_village_gold(info.village_income);
    t.set_village_support(info.village_support);
 
    ++actions_since_save_;
}
 
void map_context::set_scenario_setup(const std::string& id,
        const std::string& name,
        const std::string& description,
        int turns,
        int xp_mod,
        bool victory_defeated,
        bool random_time)
{
    scenario_id_ = id;
    scenario_name_ = name;
    scenario_description_ = description;
    random_time_ = random_time;
    victory_defeated_ = victory_defeated;
    tod_manager_->set_number_of_turns(turns);
    xp_mod_ = xp_mod;
    ++actions_since_save_;
}
 
void map_context::set_starting_time(int time)
{
    tod_manager_->set_current_time(time);
    if(!pure_map_) {
        ++actions_since_save_;
    }
}
 
void map_context::remove_area(int index)
{
    tod_manager_->remove_time_area(index);
    active_area_--;
    ++actions_since_save_;
}
 
void map_context::replace_schedule(const std::vector<time_of_day>& schedule)
{
    tod_manager_->replace_schedule(schedule);
    if(!pure_map_) {
        ++actions_since_save_;
    }
}
 
void map_context::replace_local_schedule(const std::vector<time_of_day>& schedule)
{
    tod_manager_->replace_local_schedule(schedule, active_area_);
    if(!pure_map_) {
        ++actions_since_save_;
    }
}
 
config map_context::convert_scenario(const config& old_scenario)
{
    config cfg;
    config& multiplayer = cfg.add_child("multiplayer");
    multiplayer.append_attributes(old_scenario);
    std::string map_data = multiplayer["map_data"];
    std::string separate_map_file = filesystem::get_current_editor_dir(addon_id_) + "/maps/" + filesystem::base_name(filename_, true) + filesystem::map_extension;
 
    // check that there's embedded map data, since that's how the editor used to save scenarios
    if(!map_data.empty()) {
        // check if a .map file already exists as a separate standalone .map in the editor folders or if a .map file already exists in the add-on
        if(filesystem::file_exists(separate_map_file)) {
            separate_map_file = filesystem::get_current_editor_dir(addon_id_) + "/maps/" + filesystem::get_next_filename(filesystem::base_name(filename_, true), filesystem::map_extension);
        }
        multiplayer["id"] = filesystem::base_name(separate_map_file, true);
 
        filesystem::write_file(separate_map_file, map_data);
        multiplayer.remove_attribute("map_data");
        multiplayer["map_file"] = filesystem::base_name(separate_map_file);
    } else {
        ERR_ED << "Cannot convert " << filename_ << " due to missing map_data attribute.";
        throw editor_map_load_exception("load_scenario: no embedded map_data attribute found in old-style scenario", filename_);
    }
 
    config& event = multiplayer.add_child("event");
    event["name"] = "prestart";
    event["id"] = "editor_event-prestart";
 
    // for all children that aren't [side] or [time], move them to an event
    // for [side]:
    //   keep all attributes in [side]
    //   also keep any [village]s in [side]
    //   move all other children to the start [event]
    //   if [unit], set the unit's side
    // for [time]:
    //   keep under [multiplayer]
    for(const auto [child_key, child_cfg]: old_scenario.all_children_view()) {
        if(child_key != "side" && child_key != "time") {
            config& c = event.add_child(child_key);
            c.append_attributes(child_cfg);
            c.append_children(child_cfg);
        } else if(child_key == "side") {
            config& c = multiplayer.add_child("side");
            c.append_attributes(child_cfg);
            for(const auto [side_key, side_cfg] : child_cfg.all_children_view()) {
                if(side_key == "village") {
                    config& c1 = c.add_child("village");
                    c1.append_attributes(side_cfg);
                } else {
                    config& c1 = event.add_child(side_key);
                    c1.append_attributes(side_cfg);
                    if(side_key == "unit") {
                        c1["side"] = child_cfg["side"];
                    }
                }
            }
        } else if(child_key == "time") {
            config& c = multiplayer.add_child("time");
            c.append_attributes(child_cfg);
        }
    }
 
    return cfg;
}
 
bool map_context::load_scenario()
{
    config scen = io::read(*preprocess_file(filename_)); // this call might throw anyway so ask only after trying
    std::string scenario_text = filesystem::read_file(filename_);
    if(!boost::algorithm::starts_with(scenario_text, "# This file was generated using the scenario editor.")) {
        int res = gui2::show_message(_("Confirm"), "This file was not generated by the scenario editor. Saving it may have unwanted consequences including removal of all preprocessor constructs like comments and macro use. Do you want to continue?", gui2::dialogs::message::yes_no_buttons);
        if(res != gui2::retval::OK) {
            return false;
        }
    }
 
    config scenario;
    if(scen.has_child("scenario")) {
        scenario = scen.mandatory_child("scenario");
    } else if(scen.has_child("multiplayer")) {
        scenario = scen.mandatory_child("multiplayer");
    } else if(scen.has_child("test")) {
        scenario = scen.mandatory_child("test");
    } else {
        ERR_ED << "Found no [scenario], [multiplayer], or [test] tag in " << filename_ << ", assuming old-style editor scenario and defaulting to [multiplayer]";
        scen = convert_scenario(scen);
        scenario = scen.mandatory_child("multiplayer");
    }
 
    scenario_id_ = scenario["id"].str();
    scenario_name_ = scenario["name"].str();
    scenario_description_ = scenario["description"].str();
 
    if(const config::attribute_value* experience_modifier = scenario.get("experience_modifier")) {
        xp_mod_ = experience_modifier->to_int();
    }
    victory_defeated_ = scenario["victory_when_enemies_defeated"].to_bool(true);
    random_time_ = scenario["random_start_time"].to_bool(false);
 
    if(!scenario["map_data"].str().empty()) {
        map_ = editor_map::from_string(scenario["map_data"].str()); // throws on error
    } else if(!scenario["map_file"].str().empty()) {
        map_ = editor_map::from_string(filesystem::read_file(filesystem::get_current_editor_dir(addon_id_) + "/maps/" + filesystem::base_name(scenario["map_file"]))); // throws on error
    } else {
        throw editor_map_load_exception("load_scenario: no map_file or map_data attribute found", filename_);
    }
 
    for(config& side : scenario.child_range("side")) {
        teams_.emplace_back();
        teams_.back().build(side, map_);
        if(!side["recruit"].str().empty()) {
            teams_.back().set_recruits(utils::split_set(side["recruit"].str(), ','));
        }
    }
 
    tod_manager_.reset(new tod_manager(scenario));
 
    auto event = scenario.find_child("event", "id", "editor_event-start");
    if(!event) {
        event = scenario.find_child("event", "id", "editor_event-prestart");
    }
    if(event) {
        config& evt = event.value();
 
        labels_.read(evt);
 
        for(const config& time_area : evt.child_range("time_area")) {
            tod_manager_->add_time_area(map_, time_area);
        }
 
        for(const config& item : evt.child_range("item")) {
            const map_location loc(item);
            overlays_[loc].push_back(overlay(item));
        }
 
        for(const config& music : evt.child_range("music")) {
            music_tracks_.emplace_back(sound::music_track::create(music));
        }
 
        for(config& a_unit : evt.child_range("unit")) {
            units_.insert(unit::create(a_unit, true));
        }
    }
 
    previous_cfg_ = scen;
    return true;
}
 
void map_context::select_area(int index)
{
    map_.set_selection(tod_manager_->get_area_by_index(index));
}
 
bool map_context::playlist_contains(const std::shared_ptr<sound::music_track>& track) const
{
    return utils::contains(music_tracks_, track);
}
 
void map_context::toggle_track(const std::shared_ptr<sound::music_track>& track)
{
    if(playlist_contains(track)) {
        music_tracks_.remove(track);
    } else {
        music_tracks_.push_back(track);
    }
}
 
void map_context::draw_terrain(const t_translation::terrain_code& terrain, const map_location& loc, bool one_layer_only)
{
    t_translation::terrain_code full_terrain = one_layer_only
        ? terrain
        : map_.get_terrain_info(terrain).terrain_with_default_base();
 
    draw_terrain_actual(full_terrain, loc, one_layer_only);
}
 
void map_context::draw_terrain_actual(
        const t_translation::terrain_code& terrain, const map_location& loc, bool one_layer_only)
{
    if(!map_.on_board_with_border(loc)) {
        // requests for painting off the map are ignored in set_terrain anyway,
        // but ideally we should not have any
        LOG_ED << "Attempted to draw terrain off the map (" << loc << ")";
        return;
    }
 
    t_translation::terrain_code old_terrain = map_.get_terrain(loc);
 
    if(terrain != old_terrain) {
        if(terrain.base == t_translation::NO_LAYER) {
            map_.set_terrain(loc, terrain, terrain_type_data::OVERLAY);
        } else if(one_layer_only) {
            map_.set_terrain(loc, terrain, terrain_type_data::BASE);
        } else {
            map_.set_terrain(loc, terrain);
        }
 
        add_changed_location(loc);
    }
}
 
void map_context::draw_terrain(
        const t_translation::terrain_code& terrain, const std::set<map_location>& locs, bool one_layer_only)
{
    t_translation::terrain_code full_terrain = one_layer_only
        ? terrain
        : map_.get_terrain_info(terrain).terrain_with_default_base();
 
    for(const map_location& loc : locs) {
        draw_terrain_actual(full_terrain, loc, one_layer_only);
    }
}
 
void map_context::clear_changed_locations()
{
    everything_changed_ = false;
    changed_locations_.clear();
}
 
void map_context::add_changed_location(const map_location& loc)
{
    if(!everything_changed()) {
        changed_locations_.insert(loc);
    }
}
 
void map_context::add_changed_location(const std::set<map_location>& locs)
{
    if(!everything_changed()) {
        changed_locations_.insert(locs.begin(), locs.end());
    }
}
 
void map_context::set_everything_changed()
{
    everything_changed_ = true;
}
 
bool map_context::everything_changed() const
{
    return everything_changed_;
}
 
void map_context::clear_starting_position_labels(display& disp)
{
    disp.labels().clear_all();
    starting_position_label_locs_.clear();
}
 
void map_context::set_starting_position_labels(display& disp)
{
    std::set<map_location> new_label_locs = map_.set_starting_position_labels(disp);
    starting_position_label_locs_.insert(new_label_locs.begin(), new_label_locs.end());
}
 
void map_context::reset_starting_position_labels(display& disp)
{
    clear_starting_position_labels(disp);
    set_starting_position_labels(disp);
    set_needs_labels_reset(false);
}
 
config map_context::to_config()
{
    config scen;
 
    // Textdomain
    std::string current_textdomain = "wesnoth-"+addon_id_;
 
    // the state of the previous scenario cfg
    // if it exists, alter specific parts of it (sides, times, and editor events) rather than replacing it entirely
    if(previous_cfg_) {
        scen = *previous_cfg_;
    }
 
    // if this has [multiplayer], use [multiplayer]
    // else if this has [scenario], use [scenario]
    // else if this has [test], use [test]
    // else if none, add a [multiplayer]
    config& scenario = scen.has_child("multiplayer")
        ? scen.mandatory_child("multiplayer")
        : scen.has_child("scenario")
            ? scen.mandatory_child("scenario")
            : scen.has_child("test")
                ? scen.mandatory_child("test")
                : scen.add_child("multiplayer");
 
    scenario.remove_children("side");
    scenario.remove_children("event", [](const config& cfg) {
        return cfg["id"].str() == "editor_event-start" || cfg["id"].str() == "editor_event-prestart";
    });
    scenario.remove_children("time");
 
    scenario["id"] = scenario_id_;
    scenario["name"] = t_string(scenario_name_, current_textdomain);
    scenario["description"] = t_string(scenario_description_, current_textdomain);
 
    if(xp_mod_) {
        scenario["experience_modifier"] = *xp_mod_;
    }
    if(victory_defeated_) {
        scenario["victory_when_enemies_defeated"] = *victory_defeated_;
    }
    scenario["random_start_time"] = random_time_;
 
    // write out the map data
    scenario["map_file"] = scenario_id_ + filesystem::map_extension;
    filesystem::write_file(filesystem::get_current_editor_dir(addon_id_) + "/maps/" + scenario_id_ + filesystem::map_extension, map_.write());
 
    // find or add the editor's start event
    config& event = scenario.add_child("event");
    event["name"] = "prestart";
    event["id"] = "editor_event-prestart";
    event["priority"] = 1000;
 
    // write out all the scenario data below
 
    // [time]s and [time_area]s
    config times = tod_manager_->to_config(current_textdomain);
 
    // TODO: random_start_time is written separately above. Should we use the value from the ToD manager?
    times.remove_attributes("turn_at", "it_is_a_new_turn", "random_start_time");
 
    if(times["turns"].to_int() == -1) {
        times.remove_attribute("turns");
    }
 
    if(times["current_time"].to_int() == 0) {
        times.remove_attribute("current_time");
    }
 
    scenario.merge_attributes(times);
    scenario.append_children_by_move(times, "time");
 
    // put the [time_area]s into the event to keep as much editor-specific stuff separated in its own event as possible
    event.append_children_by_move(times, "time_area");
 
    // [label]s
    labels_.write(event);
 
    // [item]s
    for(const auto& overlay_pair : overlays_) {
        for(const overlay& o : overlay_pair.second) {
            config& item = event.add_child("item");
 
            // Write x,y location
            overlay_pair.first.write(item);
 
            // These should always have a value
            item["image"] = o.image;
            item["visible_in_fog"] = o.visible_in_fog;
 
            // Optional keys
            item["id"].write_if_not_empty(o.id);
            item["name"].write_if_not_empty(t_string(o.name, current_textdomain));
            item["team_name"].write_if_not_empty(o.team_name);
            item["halo"].write_if_not_empty(o.halo);
            if(o.submerge) {
                item["submerge"] = o.submerge;
            }
        }
    }
 
    // [music]s
    for(const auto& track : music_tracks_) {
        track->write(event, true);
    }
 
    // [unit]s
    preproc_map traits_map;
    preprocess_file(game_config::path + "/data/core/macros/traits.cfg", traits_map);
 
    for(const auto& unit : units_) {
        config& u = event.add_child("unit");
 
        unit.get_location().write(u);
 
        u["side"] = unit.side();
        u["type"] = unit.type_id();
        u["name"].write_if_not_empty(t_string(unit.name(), current_textdomain));
        u["facing"] = map_location::write_direction(unit.facing());
 
        if(!boost::regex_match(unit.id(), boost::regex(".*-[0-9]+"))) {
            u["id"] = unit.id();
        }
 
        if(unit.can_recruit()) {
            u["canrecruit"] = unit.can_recruit();
        }
 
        if(unit.unrenamable()) {
            u["unrenamable"] = unit.unrenamable();
        }
 
        config& mods = u.add_child("modifications");
        if(unit.loyal()) {
            mods.append(io::read(*preprocess_string("{TRAIT_LOYAL}", "wesnoth-help", traits_map)));
        }
        //TODO this entire block could also be replaced by unit.write(u, true)
        //however, the resultant config is massive and contains many attributes we don't need.
        //need to find a middle ground here.
    }
 
    // [side]s
    for(const auto& team : teams_) {
        config& side = scenario.add_child("side");
 
        side["side"] = scenario.child_count("side");
        side["hidden"] = team.hidden();
 
        side["controller"] = side_controller::get_string(team.controller());
        side["no_leader"] = team.no_leader();
 
        side["team_name"] = team.team_name();
        side["user_team_name"].write_if_not_empty(t_string(team.user_team_name(), current_textdomain));
        if(team.recruits().size() > 0) {
            side["recruit"] = utils::join(team.recruits(), ",");
            side["faction"] = "Custom";
        }
 
        side["fog"] = team.uses_fog();
        side["shroud"] = team.uses_shroud();
        side["share_vision"] = team_shared_vision::get_string(team.share_vision());
 
        side["gold"] = team.gold();
        side["income"] = team.raw_income();
 
        for(const map_location& village : team.villages()) {
            village.write(side.add_child("village"));
        }
    }
 
    previous_cfg_ = scen;
    return scen;
}
 
void map_context::save_schedule(const std::string& schedule_id, const std::string& schedule_name)
{
    // Textdomain
    std::string current_textdomain = "wesnoth-"+addon_id_;
 
    // Path to schedule.cfg
    std::string schedule_path = filesystem::get_current_editor_dir(addon_id_) + "/utils/schedule.cfg";
 
    // Create schedule config
    config schedule;
    try {
        if (filesystem::file_exists(schedule_path)) {
            /* If exists, read the schedule.cfg
             * and insert [editor_times] block at correct place */
            preproc_map editor_map;
            editor_map.try_emplace("EDITOR");
            schedule = io::read(*preprocess_file(schedule_path, editor_map));
        }
    } catch(const filesystem::io_exception& e) {
        utils::string_map symbols;
        symbols["msg"] = e.what();
        const std::string msg = VGETTEXT("Could not save time schedule: $msg", symbols);
        throw editor_map_save_exception(msg);
    }
 
    config& editor_times = schedule.add_child("editor_times");
 
    editor_times["id"] = schedule_id;
    editor_times["name"] = t_string(schedule_name, current_textdomain);
    config times = tod_manager_->to_config(current_textdomain);
    for(const config& time : times.child_range("time")) {
        config& t = editor_times.add_child("time");
        t.append(time);
    }
 
    // Write to file
    try {
        std::stringstream wml_stream;
 
        wml_stream
            << "#textdomain " << current_textdomain << "\n"
            << "#\n"
            << "# This file was generated using the scenario editor.\n"
            << "#\n"
            << "#ifdef EDITOR\n";
 
        {
            config_writer out(wml_stream, false);
            out.write(schedule);
        }
 
        wml_stream << "#endif";
 
        if(!wml_stream.str().empty()) {
            filesystem::write_file(schedule_path, wml_stream.str());
            gui2::show_transient_message("", _("Time schedule saved."));
        }
 
    } catch(const filesystem::io_exception& e) {
        utils::string_map symbols;
        symbols["msg"] = e.what();
        const std::string msg = VGETTEXT("Could not save time schedule: $msg", symbols);
        throw editor_map_save_exception(msg);
    }
}
 
void map_context::save_scenario()
{
    assert(!is_embedded());
 
    if(scenario_id_.empty()) {
        scenario_id_ = filesystem::base_name(filename_, true);
    }
 
    if(scenario_name_.empty()) {
        scenario_name_ = scenario_id_;
    }
 
    try {
        std::stringstream wml_stream;
        wml_stream
            << "# This file was generated using the scenario editor.\n"
            << "#\n"
            << "# If you edit this file by hand, then do not expect common macros to be available when the editor loads the file.\n"
            << "# Also expect any preprocessor constructs (which includes macros and comments) to be expanded/vanish on saving.\n"
            << "# Additionally, the contents of all [side] and [time] tags as well as any events that have an id starting with 'editor_event-' are replaced entirely.\n"
            << "# Any manual changes made to those will be lost.\n"
            << "\n";
        {
            config_writer out(wml_stream, false);
            out.write(to_config(), true);
        }
 
        if(!wml_stream.str().empty()) {
            filesystem::write_file(get_filename(), wml_stream.str());
        }
 
        clear_modified();
    } catch(const filesystem::io_exception& e) {
        utils::string_map symbols;
        symbols["msg"] = e.what();
        const std::string msg = VGETTEXT("Could not save the scenario: $msg", symbols);
 
        throw editor_map_save_exception(msg);
    }
 
    // After saving the map as a scenario, it's no longer a pure map.
    pure_map_ = false;
}
 
void map_context::save_map()
{
    std::string map_data = map_.write();
 
    try {
        if(!is_embedded()) {
            filesystem::write_file(get_filename(), map_data);
        } else {
            std::string map_string = filesystem::read_file(get_filename());
 
            boost::regex rexpression_map_data(R"((.*map_data\s*=\s*")(.+?)(".*))");
            boost::smatch matched_map_data;
 
            if(boost::regex_search(map_string, matched_map_data, rexpression_map_data,
                       boost::regex_constants::match_not_dot_null)) {
                std::stringstream ss;
                ss << matched_map_data[1];
                ss << map_data;
                ss << matched_map_data[3];
 
                filesystem::write_file(get_filename(), ss.str());
            } else {
                throw editor_map_save_exception(_("Could not save into scenario"));
            }
        }
 
        add_to_recent_files();
 
        clear_modified();
    } catch(const filesystem::io_exception& e) {
        utils::string_map symbols;
        symbols["msg"] = e.what();
        const std::string msg = VGETTEXT("Could not save the map: $msg", symbols);
 
        throw editor_map_save_exception(msg);
    }
}
 
void map_context::set_map(const editor_map& map)
{
    if(map_.h() != map.h() || map_.w() != map.w()) {
        set_needs_reload();
    } else {
        set_needs_terrain_rebuild();
    }
 
    map_ = map;
}
 
void map_context::perform_action(const editor_action& action)
{
    LOG_ED << "Performing action " << action.get_id() << ": " << action.get_name() << ", actions count is "
           << action.get_instance_count();
    auto undo = action.perform(*this);
    if(actions_since_save_ < 0) {
        // set to a value that will make it impossible to get to zero, as at this point
        // it is no longer possible to get back the original map state using undo/redo
        actions_since_save_ = 1 + undo_stack_.size();
    }
 
    ++actions_since_save_;
 
    undo_stack_.emplace_back(std::move(undo));
 
    trim_stack(undo_stack_);
 
    redo_stack_.clear();
}
 
void map_context::perform_partial_action(const editor_action& action)
{
    LOG_ED << "Performing (partial) action " << action.get_id() << ": " << action.get_name() << ", actions count is "
           << action.get_instance_count();
    if(!can_undo()) {
        throw editor_logic_exception("Empty undo stack in perform_partial_action()");
    }
 
    editor_action_chain* undo_chain = dynamic_cast<editor_action_chain*>(last_undo_action());
    if(undo_chain == nullptr) {
        throw editor_logic_exception("Last undo action not a chain in perform_partial_action()");
    }
 
    auto undo = action.perform(*this);
 
    // actions_since_save_ += action.action_count();
    undo_chain->prepend_action(std::move(undo));
 
    redo_stack_.clear();
}
 
bool map_context::modified() const
{
    return actions_since_save_ != 0;
}
 
void map_context::clear_modified()
{
    actions_since_save_ = 0;
}
 
void map_context::add_to_recent_files()
{
    prefs::get().add_recent_files_entry(get_filename());
}
 
bool map_context::can_undo() const
{
    return !undo_stack_.empty();
}
 
bool map_context::can_redo() const
{
    return !redo_stack_.empty();
}
 
editor_action* map_context::last_undo_action()
{
    return undo_stack_.empty() ? nullptr : undo_stack_.back().get();
}
 
editor_action* map_context::last_redo_action()
{
    return redo_stack_.empty() ? nullptr : redo_stack_.back().get();
}
 
const editor_action* map_context::last_undo_action() const
{
    return undo_stack_.empty() ? nullptr : undo_stack_.back().get();
}
 
const editor_action* map_context::last_redo_action() const
{
    return redo_stack_.empty() ? nullptr : redo_stack_.back().get();
}
 
void map_context::undo()
{
    LOG_ED << "undo() beg, undo stack is " << undo_stack_.size() << ", redo stack " << redo_stack_.size();
 
    if(can_undo()) {
        perform_action_between_stacks(undo_stack_, redo_stack_);
        actions_since_save_--;
    } else {
        WRN_ED << "undo() called with an empty undo stack";
    }
 
    LOG_ED << "undo() end, undo stack is " << undo_stack_.size() << ", redo stack " << redo_stack_.size();
}
 
void map_context::redo()
{
    LOG_ED << "redo() beg, undo stack is " << undo_stack_.size() << ", redo stack " << redo_stack_.size();
 
    if(can_redo()) {
        perform_action_between_stacks(redo_stack_, undo_stack_);
        ++actions_since_save_;
    } else {
        WRN_ED << "redo() called with an empty redo stack";
    }
 
    LOG_ED << "redo() end, undo stack is " << undo_stack_.size() << ", redo stack " << redo_stack_.size();
}
 
void map_context::partial_undo()
{
    // callers should check for these conditions
    if(!can_undo()) {
        throw editor_logic_exception("Empty undo stack in partial_undo()");
    }
 
    editor_action_chain* undo_chain = dynamic_cast<editor_action_chain*>(last_undo_action());
    if(undo_chain == nullptr) {
        throw editor_logic_exception("Last undo action not a chain in partial undo");
    }
 
    // a partial undo performs the first action form the current action's action_chain that would be normally performed
    // i.e. the *first* one.
    const auto first_action_in_chain = undo_chain->pop_first_action();
    if(undo_chain->empty()) {
        actions_since_save_--;
        undo_stack_.pop_back();
    }
 
    redo_stack_.emplace_back(first_action_in_chain->perform(*this));
    // actions_since_save_ -= last_redo_action()->action_count();
}
 
void map_context::clear_undo_redo()
{
    undo_stack_.clear();
    redo_stack_.clear();
}
 
void map_context::trim_stack(action_stack& stack)
{
    if(stack.size() > max_action_stack_size_) {
        stack.pop_front();
    }
}
 
void map_context::perform_action_between_stacks(action_stack& from, action_stack& to)
{
    assert(!from.empty());
 
    std::unique_ptr<editor_action> action;
    action.swap(from.back());
 
    from.pop_back();
 
    auto reverse_action = action->perform(*this);
    to.emplace_back(std::move(reverse_action));
 
    trim_stack(to);
}
 
const t_string map_context::get_default_context_name() const
{
    return is_pure_map() ? _("New Map") : _("New Scenario");
}
 
} // end namespace editor