#include "Internal.h"

#include <string>

#include <vector>

#include <map>

using namespace std;

#include "modules/Gui.h"

#include "MemAccess.h"

#include "VersionInfo.h"

#include "Types.h"

#include "Error.h"

#include "ModuleFactory.h"

#include "Core.h"

#include "Debug.h"

#include "PluginManager.h"

#include "MiscUtils.h"

using namespace DFHack;

#include "modules/Job.h"

#include "modules/Screen.h"

#include "modules/Maps.h"

#include "modules/Units.h"

#include "modules/World.h"

#include "DataDefs.h"

#include "df/announcement_flags.h"

#include "df/building_cagest.h"

#include "df/building_civzonest.h"

#include "df/building_furnacest.h"

#include "df/building_trapst.h"

#include "df/building_type.h"

#include "df/building_workshopst.h"

#include "df/cri_unitst.h"

#include "df/d_init.h"

#include "df/game_mode.h"

#include "df/general_ref.h"

#include "df/global_objects.h"

#include "df/graphic.h"

#include "df/graphic_viewportst.h"

#include "df/historical_figure.h"

#include "df/interfacest.h"

#include "df/item_corpsepiecest.h"

#include "df/item_corpsest.h"

#include "df/job.h"

#include "df/legend_pagest.h"

#include "df/occupation.h"

#include "df/plant.h"

#include "df/popup_message.h"

#include "df/report.h"

#include "df/report_zoom_type.h"

#include "df/route_stockpile_link.h"

#include "df/stop_depart_condition.h"

#include "df/adventurest.h"

#include "df/buildreq.h"

#include "df/ui_look_list.h"

#include "df/gamest.h"

#include "df/ui_unit_view_mode.h"

#include "df/unit.h"

#include "df/unit_inventory_item.h"

#include "df/viewscreen_choose_start_sitest.h"

#include "df/viewscreen_dwarfmodest.h"

#include "df/viewscreen_legendsst.h"

#include "df/viewscreen_new_regionst.h"

#include "df/viewscreen_setupdwarfgamest.h"

#include "df/viewscreen_titlest.h"

#include "df/viewscreen_worldst.h"

#include "df/world.h"

const size_t MAX_REPORTS_SIZE = 3000; // DF clears old reports to maintain this vector size

const int32_t RECENT_REPORT_TICKS = 500; // used by UNIT_COMBAT_REPORT_ALL_ACTIVE

const int32_t ANNOUNCE_LINE_DURATION = 100; // time to display each line in announcement bar; 2 sec at 50 GFPS

const int16_t ANNOUNCE_DISPLAY_TIME = 2000; // DF uses this value for most announcements; 40 sec at 50 GFPS

namespace DFHack

{

DBG_DECLARE(core, gui, DebugCategory::LINFO);

}

using namespace df::enums;

using df::building_civzonest;

using df::global::game;

using df::global::gamemode;

using df::global::gps;

using df::global::gview;

using df::global::init;

using df::global::plotinfo;

using df::global::selection_rect;

using df::global::ui_menu_width;

using df::global::world;

static std::string getNameChunk(virtual_identity *id, int start, int end)

}

typedef void (*getFocusStringsHandler)(std::string &str, std::vector<std::string> &strList, df::viewscreen *screen);

static std::map<virtual_identity*, getFocusStringsHandler> getFocusStringsHandlers;

#define VIEWSCREEN(name) df::viewscreen_##name##st

#define DEFINE_GET_FOCUS_STRING_HANDLER(screen_type) \

static void getFocusStrings_##screen_type(std::string &baseFocus, std::vector<std::string> &focusStrings, VIEWSCREEN(screen_type) *screen);\

DFHACK_STATIC_ADD_TO_MAP(\

&getFocusStringsHandlers, &VIEWSCREEN(screen_type)::_identity, \

(getFocusStringsHandler)getFocusStrings_##screen_type \

); \

static void getFocusStrings_##screen_type(std::string &baseFocus, std::vector<std::string> &focusStrings, VIEWSCREEN(screen_type) *screen)

DEFINE_GET_FOCUS_STRING_HANDLER(title)

}

DEFINE_GET_FOCUS_STRING_HANDLER(new_region)

}

DEFINE_GET_FOCUS_STRING_HANDLER(choose_start_site)

}

DEFINE_GET_FOCUS_STRING_HANDLER(setupdwarfgame)

}

DEFINE_GET_FOCUS_STRING_HANDLER(legends)

}

DEFINE_GET_FOCUS_STRING_HANDLER(world)

{

focusStrings.push_back(baseFocus + '/' + enum_item_key(screen->view_mode));

}

DEFINE_GET_FOCUS_STRING_HANDLER(dwarfmode)

}

bool Gui::matchFocusString(std::string focus_string, df::viewscreen *top) {

}

static void push_dfhack_focus_string(dfhack_viewscreen *vs, std::vector<std::string> &focusStrings)

}

std::vector<std::string> Gui::getFocusStrings(df::viewscreen* top)

}

bool Gui::default_hotkey(df::viewscreen *top)

{

return World::isFortressMode() || World::isAdventureMode();

}

bool Gui::anywhere_hotkey(df::viewscreen *) {

return true;

}

bool Gui::dwarfmode_hotkey(df::viewscreen *top) {

return matchFocusString("dwarfmode", top);

}

static bool has_cursor()

{

return Gui::getCursorPos().isValid();

}

bool Gui::cursor_hotkey(df::viewscreen *top)

}

bool Gui::workshop_job_hotkey(df::viewscreen *top)

}

bool Gui::build_selector_hotkey(df::viewscreen *top)

}

bool Gui::view_unit_hotkey(df::viewscreen *top)

}

bool Gui::any_job_hotkey(df::viewscreen *top)

}

df::job *Gui::getAnyWorkshopJob(df::viewscreen *top) {

auto bld = getAnyBuilding(top);

if (!bld)

return NULL;

// no way to select a specific job; just get the first one

return bld->jobs.size() ? bld->jobs[0] : NULL;

}

df::job *Gui::getSelectedWorkshopJob(color_ostream &out, bool quiet) {

df::job *job = getAnyWorkshopJob(Core::getTopViewscreen());

if (!job && !quiet)

out.printerr("No workshop selected or job cannot be found in the selected workshop.\n");

return job;

}

df::job *Gui::getAnyJob(df::viewscreen *top) {

if (auto dfscreen = dfhack_viewscreen::try_cast(top))

return dfscreen->getSelectedJob();

if (matchFocusString("dwarfmode/Info/JOBS")) {

auto &cri_job = game->main_interface.info.jobs.cri_job;

// no way to select specific jobs; just get the first one

return cri_job.size() ? cri_job[0]->jb : NULL;

}

if (auto unit = getAnyUnit(top)) {

return unit->job.current_job;

}

return getAnyWorkshopJob(top);

}

df::job *Gui::getSelectedJob(color_ostream &out, bool quiet)

{

df::job *job = getAnyJob(Core::getTopViewscreen());

if (!job && !quiet)

out.printerr("No job can be found from what is selected in the UI.\n");

return job;

}

df::unit *Gui::getAnyUnit(df::viewscreen *top)

{

using df::global::game;

if (auto dfscreen = dfhack_viewscreen::try_cast(top))

return dfscreen->getSelectedUnit();

if (game->main_interface.view_sheets.open

&& game->main_interface.view_sheets.active_sheet == view_sheet_type::UNIT)

return df::unit::find(game->main_interface.view_sheets.active_id);