dfhack/plugins/remotefortressreader/adventure_control.cpp

#include "adventure_control.h"
#include "DataDefs.h"

#include "df/adventure_movement_attack_creaturest.h"
#include "df/adventure_movement_building_interactst.h"
#include "df/adventure_movement_climbst.h"
#include "df/adventure_movement_hold_itemst.h"
#include "df/adventure_movement_hold_tilest.h"
#include "df/adventure_movement_optionst.h"
#include "df/ui_advmode.h"
#include "df/viewscreen.h"

#include "modules/Gui.h"

#include <queue>

using namespace AdventureControl;
using namespace df::enums;
using namespace DFHack;
using namespace Gui;

std::queue<interface_key::interface_key> keyQueue;

void KeyUpdate()
{
    if (!keyQueue.empty())
    {
        getCurViewscreen()->feed_key(keyQueue.front());
        keyQueue.pop();
    }
}

void SetCoord(df::coord in, RemoteFortressReader::Coord *out)
{
    out->set_x(in.x);
    out->set_y(in.y);
    out->set_z(in.z);
}

command_result MoveCommand(DFHack::color_ostream &stream, const MoveCommandParams *in)
{
    //auto viewScreen = getCurViewscreen();
    //if (!in->has_direction())
    //    return CR_WRONG_USAGE;
    //if (!df::global::ui_advmode->menu == ui_advmode_menu::Default)
    //    return CR_OK;
    //auto dir = in->direction();
    //switch (dir.x())
    //{
    //case -1:
    //    switch (dir.y())
    //    {
    //    case -1:
    //        switch (dir.z())
    //        {
    //        case -1:
    //            viewScreen->feed_key(interface_key::A_MOVE_NW_DOWN);
    //            break;
    //        case 0:
    //            viewScreen->feed_key(interface_key::A_MOVE_NW);
    //            break;
    //        case 1:
    //            viewScreen->feed_key(interface_key::A_MOVE_NW_UP);
    //            break;
    //        }
    //        break;
    //    case 0:
    //        switch (dir.z())
    //        {
    //        case -1:
    //            viewScreen->feed_key(interface_key::A_MOVE_W_DOWN);
    //            break;
    //        case 0:
    //            viewScreen->feed_key(interface_key::A_MOVE_W);
    //            break;
    //        case 1:
    //            viewScreen->feed_key(interface_key::A_MOVE_W_UP);
    //            break;
    //        }
    //        break;
    //    case 1:
    //        switch (dir.z())
    //        {
    //        case -1:
    //            viewScreen->feed_key(interface_key::A_MOVE_SW_DOWN);
    //            break;
    //        case 0:
    //            viewScreen->feed_key(interface_key::A_MOVE_SW);
    //            break;
    //        case 1:
    //            viewScreen->feed_key(interface_key::A_MOVE_SW_UP);
    //            break;
    //        }
    //        break;
    //    }
    //    break;
    //case 0:
    //    switch (dir.y())
    //    {
    //    case -1:
    //        switch (dir.z())
    //        {
    //        case -1:
    //            viewScreen->feed_key(interface_key::A_MOVE_N_DOWN);
    //            break;
    //        case 0:
    //            viewScreen->feed_key(interface_key::A_MOVE_N);
    //            break;
    //        case 1:
    //            viewScreen->feed_key(interface_key::A_MOVE_N_UP);
    //            break;
    //        }
    //        break;
    //    case 0:
    //        switch (dir.z())
    //        {
    //        case -1:
    //            viewScreen->feed_key(interface_key::A_MOVE_DOWN);
    //            break;
    //        case 1:
    //            viewScreen->feed_key(interface_key::A_MOVE_UP);
    //            break;
    //        }
    //        break;
    //    case 1:
    //        switch (dir.z())
    //        {
    //        case -1:
    //            viewScreen->feed_key(interface_key::A_MOVE_S_DOWN);
    //            break;
    //        case 0:
    //            viewScreen->feed_key(interface_key::A_MOVE_S);
    //            break;
    //        case 1:
    //            viewScreen->feed_key(interface_key::A_MOVE_S_UP);
    //            break;
    //        }
    //        break;
    //    }
    //    break;
    //case 1:
    //    switch (dir.y())
    //    {
    //    case -1:
    //        switch (dir.z())
    //        {
    //        case -1:
    //            viewScreen->feed_key(interface_key::A_MOVE_NE_DOWN);
    //            break;
    //        case 0:
    //            viewScreen->feed_key(interface_key::A_MOVE_NE);
    //            break;
    //        case 1:
    //            viewScreen->feed_key(interface_key::A_MOVE_NE_UP);
    //            break;
    //        }
    //        break;
    //    case 0:
    //        switch (dir.z())
    //        {
    //        case -1:
    //            viewScreen->feed_key(interface_key::A_MOVE_E_DOWN);
    //            break;
    //        case 0:
    //            viewScreen->feed_key(interface_key::A_MOVE_E);
    //            break;
    //        case 1:
    //            viewScreen->feed_key(interface_key::A_MOVE_E_UP);
    //            break;
    //        }
    //        break;
    //    case 1:
    //        switch (dir.z())
    //        {
    //        case -1:
    //            viewScreen->feed_key(interface_key::A_MOVE_SE_DOWN);
    //            break;
    //        case 0:
    //            viewScreen->feed_key(interface_key::A_MOVE_SE);
    //            break;
    //        case 1:
    //            viewScreen->feed_key(interface_key::A_MOVE_SE_UP);
    //            break;
    //        }
    //        break;
    //    }
    //    break;
    //}
    return CR_OK;
}

command_result JumpCommand(DFHack::color_ostream &stream, const MoveCommandParams *in)
{
    //if (!in->has_direction())
    //    return CR_WRONG_USAGE;
    //if (!df::global::ui_advmode->menu == ui_advmode_menu::Default)
    //    return CR_OK;
    //auto dir = in->direction();
    //keyQueue.push(interface_key::A_JUMP);
    //int x = dir.x();
    //int y = dir.y();
    //if (x > 0)
    //{
    //    for (int i = 0; i < x; i++)
    //    {
    //        keyQueue.push(interface_key::CURSOR_RIGHT);
    //    }
    //}
    //if (x < 0)
    //{
    //    for (int i = 0; i > x; i--)
    //    {
    //        keyQueue.push(interface_key::CURSOR_LEFT);
    //    }
    //}
    //if (y > 0)
    //{
    //    for (int i = 0; i < y; i++)
    //    {
    //        keyQueue.push(interface_key::CURSOR_DOWN);
    //    }
    //}
    //if (y < 0)
    //{
    //    for (int i = 0; i > y; i--)
    //    {
    //        keyQueue.push(interface_key::CURSOR_UP);
    //    }
    //}
    //keyQueue.push(interface_key::SELECT);
    return CR_OK;
}

command_result MenuQuery(DFHack::color_ostream &stream, const EmptyMessage *in, MenuContents *out)
{
    //auto advUi = df::global::ui_advmode;

    //if (advUi == NULL)
    //    return CR_FAILURE;

    //out->set_current_menu((AdvmodeMenu)advUi->menu);

    ////Fixme: Needs a proper way to control it, but for now, this is the only way to allow Armok Vision to keep going without the user needing to switch to DF.
    //if (advUi->menu == ui_advmode_menu::FallAction)
    //{
    //    getCurViewscreen()->feed_key(interface_key::OPTION1);
    //}

    //switch (advUi->menu)
    //{
    //case ui_advmode_menu::MoveCarefully:
    //    for (size_t i = 0; i < advUi->movements.size(); i++)
    //    {
    //        auto movement = advUi->movements[i];
    //        auto send_movement = out->add_movements();
    //        SetCoord(movement->source, send_movement->mutable_source());
    //        SetCoord(movement->dest, send_movement->mutable_dest());

    //        STRICT_VIRTUAL_CAST_VAR(climbMovement, df::adventure_movement_climbst, movement);
    //        if (climbMovement)
    //        {
    //            SetCoord(climbMovement->grab, send_movement->mutable_grab());
    //            send_movement->set_movement_type(CarefulMovementType::CLIMB);
    //        }
    //        STRICT_VIRTUAL_CAST_VAR(holdTileMovement, df::adventure_movement_hold_tilest, movement);
    //        if (holdTileMovement)
    //        {
    //            SetCoord(holdTileMovement->grab, send_movement->mutable_grab());
    //            send_movement->set_movement_type(CarefulMovementType::HOLD_TILE);
    //        }
    //    }
    //default:
    //    break;
    //}

    return CR_OK;
}

command_result MovementSelectCommand(DFHack::color_ostream &stream, const dfproto::IntMessage *in)
{
    //if (!(df::global::ui_advmode->menu == ui_advmode_menu::MoveCarefully))
    //    return CR_OK;
    //int choice = in->value();
    //int page = choice / 5;
    //int select = choice % 5;
    //for (int i = 0; i < page; i++)
    //{
    //    keyQueue.push(interface_key::SECONDSCROLL_PAGEDOWN);
    //}
    //keyQueue.push((interface_key::interface_key)(interface_key::OPTION1 + select));
    return CR_OK;
}

command_result MiscMoveCommand(DFHack::color_ostream &stream, const MiscMoveParams *in)
{
    //if (!df::global::ui_advmode->menu == ui_advmode_menu::Default)
    //    return CR_OK;

    //auto type = in->type();

    //switch (type)
    //{
    //case AdventureControl::SET_CLIMB:
    //    getCurViewscreen()->feed_key(interface_key::A_HOLD);
    //    break;
    //case AdventureControl::SET_STAND:
    //    getCurViewscreen()->feed_key(interface_key::A_STANCE);
    //    break;
    //case AdventureControl::SET_CANCEL:
    //    getCurViewscreen()->feed_key(interface_key::LEAVESCREEN);
    //    break;
    //default:
    //    break;
    //}

    return CR_OK;
}