Merge remote-tracking branch 'softmoth/command-prompt-wrap' into develop

develop
lethosor 2022-05-09 22:14:31 -04:00
commit 947098e564
No known key found for this signature in database
GPG Key ID: 76A269552F4F58C1
5 changed files with 107 additions and 67 deletions

@ -1 +1 @@
Subproject commit 87d6ae018cb8d288d854f632e9d8d959d75d7db4 Subproject commit 99d06827848583232dd77afb34cd7ab589567086

@ -168,32 +168,58 @@ std::string to_search_normalized(const std::string &str)
return result; return result;
} }
bool word_wrap(std::vector<std::string> *out, const std::string &str, size_t line_length)
bool word_wrap(std::vector<std::string> *out, const std::string &str,
size_t line_length, bool collapse_whitespace)
{ {
out->clear(); if (line_length == 0)
std::istringstream input(str); line_length = SIZE_MAX;
std::string out_line;
std::string word; std::string line;
if (input >> word) size_t break_pos = 0;
for (auto &c : str)
{ {
out_line += word; if (c == '\n')
// size_t remaining = line_length - std::min(line_length, word.length()); {
while (input >> word) out->push_back(line);
line.clear();
break_pos = 0;
continue;
}
if (isspace(c))
{ {
if (out_line.length() + word.length() + 1 <= line_length) if (break_pos == line.length() && collapse_whitespace)
continue;
line.push_back(collapse_whitespace ? ' ' : c);
break_pos = line.length();
}
else {
line.push_back(c);
}
if (line.length() > line_length)
{
if (break_pos > 0)
{ {
out_line += ' '; // Break before last space, and skip that space
out_line += word; out->push_back(line.substr(0, break_pos - 1));
} }
else else
{ {
out->push_back(out_line); // Single word is too long, just break it
out_line = word; out->push_back(line.substr(0, line_length));
break_pos = line_length;
} }
line = line.substr(break_pos);
break_pos = 0;
} }
if (out_line.length())
out->push_back(out_line);
} }
if (line.length())
out->push_back(line);
return true; return true;
} }

@ -391,7 +391,8 @@ DFHACK_EXPORT std::string to_search_normalized(const std::string &str);
DFHACK_EXPORT bool word_wrap(std::vector<std::string> *out, DFHACK_EXPORT bool word_wrap(std::vector<std::string> *out,
const std::string &str, const std::string &str,
size_t line_length = 80); size_t line_length = 80,
bool collapse_whitespace = false);
inline bool bits_match(unsigned required, unsigned ok, unsigned mask) inline bool bits_match(unsigned required, unsigned ok, unsigned mask)
{ {

@ -1,23 +1,25 @@
//command-prompt a one line command entry at the top of the screen for quick commands // command-prompt: A one-line command entry at the top of the screen for quick commands
#include "Core.h" #include "Core.h"
#include <ColorText.h>
#include <Console.h> #include <Console.h>
#include <Export.h> #include <Export.h>
#include <MiscUtils.h>
#include <PluginManager.h> #include <PluginManager.h>
#include <ColorText.h>
#include <modules/Screen.h>
#include <modules/Gui.h> #include <modules/Gui.h>
#include <modules/Screen.h>
#include <set>
#include <list> #include <list>
#include <set>
#include <string>
#include <utility> #include <utility>
#include <vector> #include <vector>
#include "df/enabler.h"
#include "df/graphic.h"
#include "df/interface_key.h" #include "df/interface_key.h"
#include "df/ui.h" #include "df/ui.h"
#include "df/graphic.h"
#include "df/enabler.h"
using namespace DFHack; using namespace DFHack;
using namespace df::enums; using namespace df::enums;
@ -36,8 +38,10 @@ class prompt_ostream:public buffered_color_ostream
protected: protected:
void flush_proxy(); void flush_proxy();
public: public:
prompt_ostream(viewscreen_commandpromptst* parent):parent_(parent){} prompt_ostream(viewscreen_commandpromptst* parent)
bool empty(){return buffer.empty();} : parent_(parent)
{}
bool empty() { return buffer.empty(); }
}; };
class viewscreen_commandpromptst : public dfhack_viewscreen { class viewscreen_commandpromptst : public dfhack_viewscreen {
public: public:
@ -48,7 +52,7 @@ public:
} }
void render(); void render();
void help() { } void help() {}
int8_t movies_okay() { return 0; } int8_t movies_okay() { return 0; }
df::unit* getSelectedUnit() { return Gui::getAnyUnit(parent); } df::unit* getSelectedUnit() { return Gui::getAnyUnit(parent); }
@ -57,10 +61,11 @@ public:
df::plant* getSelectedPlant() { return Gui::getAnyPlant(parent); } df::plant* getSelectedPlant() { return Gui::getAnyPlant(parent); }
std::string getFocusString() { return "commandprompt"; } std::string getFocusString() { return "commandprompt"; }
viewscreen_commandpromptst(std::string entry):submitted(false), is_response(false) viewscreen_commandpromptst(std::string entry)
: submitted(false), is_response(false)
{ {
show_fps=gps->display_frames; show_fps = gps->display_frames;
gps->display_frames=0; gps->display_frames = 0;
cursor_pos = entry.size(); cursor_pos = entry.size();
frame = 0; frame = 0;
history_idx = command_history.size(); history_idx = command_history.size();
@ -76,7 +81,7 @@ public:
} }
~viewscreen_commandpromptst() ~viewscreen_commandpromptst()
{ {
gps->display_frames=show_fps; gps->display_frames = show_fps;
} }
void add_response(color_value v, std::string s) void add_response(color_value v, std::string s)
@ -125,7 +130,7 @@ public:
} }
protected: protected:
std::list<std::pair<color_value,std::string> > responses; std::list<std::pair<color_value, std::string> > responses;
int cursor_pos; int cursor_pos;
int history_idx; int history_idx;
bool submitted; bool submitted;
@ -138,8 +143,8 @@ void prompt_ostream::flush_proxy()
{ {
if (buffer.empty()) if (buffer.empty())
return; return;
for(auto it=buffer.begin();it!=buffer.end();it++) for(auto it = buffer.begin(); it != buffer.end(); it++)
parent_->add_response(it->first,it->second); parent_->add_response(it->first, it->second);
buffer.clear(); buffer.clear();
} }
void viewscreen_commandpromptst::render() void viewscreen_commandpromptst::render()
@ -154,25 +159,31 @@ void viewscreen_commandpromptst::render()
auto dim = Screen::getWindowSize(); auto dim = Screen::getWindowSize();
parent->render(); parent->render();
if(is_response) if (is_response)
{ {
auto it=responses.begin(); int y = 0;
for(int i=0;i<dim.y && it!=responses.end();i++,it++) for (auto &response : responses)
{ {
Screen::fillRect(Screen::Pen(' ', 7, 0),0,i,dim.x,i); std::vector<std::string> lines;
std::string cur_line=it->second; word_wrap(&lines, response.second, dim.x);
Screen::paintString(Screen::Pen(' ',it->first,0),0,i,cur_line.substr(0,cur_line.size()-1)); for (auto &line : lines)
{
Screen::fillRect(Screen::Pen(' ', 7, 0), 0, y, dim.x, y);
Screen::paintString(Screen::Pen(' ', response.first, 0), 0, y, line);
if (++y >= dim.y)
return;
}
} }
} }
else else
{ {
std::string entry = get_entry(); std::string entry = get_entry();
Screen::fillRect(Screen::Pen(' ', 7, 0),0,0,dim.x,0); Screen::fillRect(Screen::Pen(' ', 7, 0), 0, 0, dim.x, 0);
Screen::paintString(Screen::Pen(' ', 7, 0), 0, 0,"[DFHack]#"); Screen::paintString(Screen::Pen(' ', 7, 0), 0, 0, "[DFHack]#");
std::string cursor = (frame < enabler->gfps / 2) ? "_" : " "; std::string cursor = (frame < enabler->gfps / 2) ? "_" : " ";
if(cursor_pos < (dim.x - 10)) if (cursor_pos < dim.x - 10)
{ {
Screen::paintString(Screen::Pen(' ', 7, 0), 10,0 , entry); Screen::paintString(Screen::Pen(' ', 7, 0), 10, 0, entry);
if (int16_t(entry.size()) > dim.x - 10) if (int16_t(entry.size()) > dim.x - 10)
Screen::paintTile(Screen::Pen('\032', 7, 0), dim.x - 1, 0); Screen::paintTile(Screen::Pen('\032', 7, 0), dim.x - 1, 0);
if (cursor != " ") if (cursor != " ")
@ -191,12 +202,12 @@ void viewscreen_commandpromptst::render()
void viewscreen_commandpromptst::submit() void viewscreen_commandpromptst::submit()
{ {
CoreSuspendClaimer suspend; CoreSuspendClaimer suspend;
if(is_response) if (is_response)
{ {
Screen::dismiss(this); Screen::dismiss(this);
return; return;
} }
if(submitted) if (submitted)
return; return;
submitted = true; submitted = true;
prompt_ostream out(this); prompt_ostream out(this);
@ -204,11 +215,11 @@ void viewscreen_commandpromptst::submit()
Screen::Hide hide_guard(this, Screen::Hide::RESTORE_AT_TOP); Screen::Hide hide_guard(this, Screen::Hide::RESTORE_AT_TOP);
Core::getInstance().runCommand(out, get_entry()); Core::getInstance().runCommand(out, get_entry());
} }
if(out.empty() && responses.empty()) if (out.empty() && responses.empty())
Screen::dismiss(this); Screen::dismiss(this);
else else
{ {
is_response=true; is_response = true;
} }
} }
void viewscreen_commandpromptst::feed(std::set<df::interface_key> *events) void viewscreen_commandpromptst::feed(std::set<df::interface_key> *events)
@ -240,14 +251,14 @@ void viewscreen_commandpromptst::feed(std::set<df::interface_key> *events)
for (auto it = events->begin(); it != events->end(); ++it) for (auto it = events->begin(); it != events->end(); ++it)
{ {
auto key = *it; auto key = *it;
if (key==interface_key::STRING_A000) //delete? if (key == interface_key::STRING_A000) //delete?
{ {
if(entry.size() && cursor_pos > 0) if (entry.size() && cursor_pos > 0)
{ {
entry.erase(cursor_pos - 1, 1); entry.erase(cursor_pos - 1, 1);
cursor_pos--; cursor_pos--;
} }
if(size_t(cursor_pos) > entry.size()) if (size_t(cursor_pos) > entry.size())
cursor_pos = entry.size(); cursor_pos = entry.size();
continue; continue;
} }
@ -261,34 +272,34 @@ void viewscreen_commandpromptst::feed(std::set<df::interface_key> *events)
} }
} }
// Prevent number keys from moving cursor // Prevent number keys from moving cursor
if(events->count(interface_key::CURSOR_RIGHT)) if (events->count(interface_key::CURSOR_RIGHT))
{ {
cursor_pos++; cursor_pos++;
if (size_t(cursor_pos) > entry.size()) if (size_t(cursor_pos) > entry.size())
cursor_pos = entry.size(); cursor_pos = entry.size();
} }
else if(events->count(interface_key::CURSOR_LEFT)) else if (events->count(interface_key::CURSOR_LEFT))
{ {
cursor_pos--; cursor_pos--;
if (cursor_pos < 0) cursor_pos = 0; if (cursor_pos < 0) cursor_pos = 0;
} }
else if(events->count(interface_key::CURSOR_RIGHT_FAST)) else if (events->count(interface_key::CURSOR_RIGHT_FAST))
{ {
forward_word(); forward_word();
} }
else if(events->count(interface_key::CURSOR_LEFT_FAST)) else if (events->count(interface_key::CURSOR_LEFT_FAST))
{ {
back_word(); back_word();
} }
else if(events->count(interface_key::CUSTOM_CTRL_A)) else if (events->count(interface_key::CUSTOM_CTRL_A))
{ {
cursor_pos = 0; cursor_pos = 0;
} }
else if(events->count(interface_key::CUSTOM_CTRL_E)) else if (events->count(interface_key::CUSTOM_CTRL_E))
{ {
cursor_pos = entry.size(); cursor_pos = entry.size();
} }
else if(events->count(interface_key::CURSOR_UP)) else if (events->count(interface_key::CURSOR_UP))
{ {
history_idx--; history_idx--;
if (history_idx < 0) if (history_idx < 0)
@ -296,7 +307,7 @@ void viewscreen_commandpromptst::feed(std::set<df::interface_key> *events)
entry = get_entry(); entry = get_entry();
cursor_pos = entry.size(); cursor_pos = entry.size();
} }
else if(events->count(interface_key::CURSOR_DOWN)) else if (events->count(interface_key::CURSOR_DOWN))
{ {
if (size_t(history_idx) < command_history.size() - 1) if (size_t(history_idx) < command_history.size() - 1)
{ {
@ -321,8 +332,8 @@ command_result show_prompt(color_ostream &out, std::vector <std::string> & param
return CR_OK; return CR_OK;
} }
std::string params; std::string params;
for(size_t i=0;i<parameters.size();i++) for(size_t i = 0; i < parameters.size(); i++)
params+=parameters[i]+" "; params += parameters[i] + " ";
Screen::show(dts::make_unique<viewscreen_commandpromptst>(params), plugin_self); Screen::show(dts::make_unique<viewscreen_commandpromptst>(params), plugin_self);
return CR_OK; return CR_OK;
} }
@ -330,21 +341,23 @@ bool hotkey_allow_all(df::viewscreen *top)
{ {
return true; return true;
} }
DFhackCExport command_result plugin_init ( color_ostream &out, std::vector <PluginCommand> &commands) DFhackCExport command_result plugin_init(color_ostream &out, std::vector <PluginCommand> &commands)
{ {
commands.push_back(PluginCommand( commands.push_back(PluginCommand(
"command-prompt","Shows a command prompt on window.",show_prompt,hotkey_allow_all, "command-prompt", "Shows a command prompt on window.",
"command-prompt [entry] - shows a cmd prompt in df window. Entry is used for default prefix (e.g. ':lua')" show_prompt, hotkey_allow_all,
"command-prompt [entry] - shows a cmd prompt in df window."
" Entry is used for default prefix (e.g. ':lua')"
)); ));
return CR_OK; return CR_OK;
} }
DFhackCExport command_result plugin_onstatechange (color_ostream &out, state_change_event e) DFhackCExport command_result plugin_onstatechange(color_ostream &out, state_change_event e)
{ {
return CR_OK; return CR_OK;
} }
DFhackCExport command_result plugin_shutdown ( color_ostream &out ) DFhackCExport command_result plugin_shutdown(color_ostream &out)
{ {
return CR_OK; return CR_OK;
} }

@ -1 +1 @@
Subproject commit b808050d4a3885aa0e250e726708b2b28fe28260 Subproject commit 64626c81481a2f938f071b7347d07c10312b223d