1
0
mirror of https://github.com/zenorogue/hyperrogue.git synced 2024-12-25 09:30:35 +00:00
hyperrogue/rogueviz/gobot.cpp

624 lines
14 KiB
C++

// Hyperbolic Rogue -- Go
// Copyright (C) 2011-2019 Zeno Rogue, see 'hyper.cpp' for details
// compile the Discord version:
// ./mymake rogueviz/gobot.cpp -std=c++17 -lssl -lz -lcrypto -I rogueviz/aegis/include/ -I rogueviz/aegis/lib/json/include/ -I rogueviz/aegis/lib/spdlog/include/ -I rogueviz/aegis/lib/websocketpp/ -I rogueviz/aegis/lib/asio/asio/include/
// to run on an interesting board, run RogueViz with parameters:
// -canvas i -fillmodel ff801080 -noscr -geo Bring -gp 5 1 -unrectified -go-local -smart 1
// (add -run before -go-local if you want to select your board manually (press F10 after selecting the board)
/** \file gobot.cpp
* \brief bot to play Go via Discord
*/
#define AEGIS 0
#if AEGIS
#include <aegis.hpp>
#endif
#include "rogueviz.h"
namespace hr {
EX namespace gobot {
eWall empty = waChasm;
bool in = false;
vector<cell*> ac;
map<cell*, int> indices;
const int Free = 2;
struct boarddata {
vector<int> taken, owner;
array<int, 2> captures;
};
boarddata current;
int labels_value = 1;
vector<boarddata> history;
bool draw_go(cell *c, const shiftmatrix& V);
void init_go() {
ac = currentmap->allcells();
current.taken.resize(isize(ac), 2);
current.owner.resize(isize(ac), 2);
current.captures[0] = 0;
current.captures[1] = 0;
for(int i=0; i<isize(ac); i++)
indices[ac[i]] = i;
rogueviz::addHook(hooks_drawcell, 100, draw_go);
}
void hwrite(hstream& hs, const boarddata& b) {
hwrite(hs, b.captures, b.taken, b.owner);
}
vector<int> neigh_indices(int i) {
vector<int> res;
forCellEx(c1, ac[i])
if(indices.count(c1))
res.push_back(indices[c1]);
return res;
}
string chars = "abcdefghijkmnopqrstuvwxyzABCDEFGHJKLMNOPQRSTUVWXYZ123456789";
bool valid_index(int q) {
return q >= 0 && q < isize(ac);
}
string index_to_str(int k) {
if(k < isize(chars))
return s0 + chars[k];
else return chars[k % isize(chars)] + index_to_str(k / isize(chars) - 1);
}
int str_to_index(string s) {
int val = -1;
for(int i=0; i<isize(chars); i++) if(s[0] == chars[i]) val = i;
if(val == -1) return -1;
if(isize(s) == 1) return val;
else return val + isize(chars) * (str_to_index(s.substr(1)) + 1);
}
color_t player_colors[2] = {0x202020FF, 0xFFFFFFFF};
bool draw_go(cell *c, const shiftmatrix& V) {
if(c->wall == waSea) c->wall = waChasm;
if(!indices.count(c)) return false;
int id = indices[c];
int lv_needed;
forCellIdCM(c1, i, c) if(indices.count(c1) && indices[c1] < id) {
vid.linewidth *= 2;
gridline(V, C0, currentmap->adj(c, i) * C0, 0xFF, 4);
vid.linewidth /= 2;
}
if(current.taken[id] != 2) {
vid.linewidth *= 2;
poly_outline = 0xFF;
queuepolyat(V, cgi.shHugeDisk, player_colors[current.taken[id]], PPR::SUPERLINE);
vid.linewidth /= 2;
lv_needed = 3;
}
else if(current.owner[id] == 2) {
lv_needed = 1;
}
else {
queuepoly(V, cgi.shGem[0], player_colors[current.owner[id]]);
lv_needed = 2;
}
if(labels_value >= lv_needed) {
string s = index_to_str(id);
queuestr(V, isize(s) == 1 ? 0.8 : 0.5, s, 0xFFD500);
}
return false;
}
void save_backup() {
history.push_back(current);
}
void undo() {
current = history.back();
history.pop_back();
}
#if AEGIS
aegis::gateway::events::message_create* cur;
#endif
int shot_state;
#if AEGIS
std::vector<aegis::future<aegis::gateway::objects::message> > old_shots;
#endif
void clean_old_shots() {
/*
std::vector<aegis::future<aegis::gateway::objects::message> > remaining;
for(auto& sh: old_shots)
if(sh.available()) {
sh.get().delete_message();
}
else
remaining.emplace_back(move(sh));
old_shots = move(remaining);
*/
}
bool menubased;
void take_shot() {
#if AEGIS
if(cur) {
println(hlog, "taking test screenshot");
shot_state = 1;
while(shot_state == 1) usleep(1000);
shot_state = 0;
aegis::create_message_t msg;
aegis::rest::aegis_file f;
f.name = "go-board.png";
FILE *ff = fopen("go-temp.png", "r");
if(!ff) {
println(hlog, "file missing?!");
return;
}
int c;
while((c = fgetc(ff)) >= 0) f.data.push_back(c);
fclose(ff);
println(hlog, "file size = ", int(f.data.size()));
msg.file(f);
println(hlog, "file attached");
clean_old_shots();
// old_shots.push_back();
cur->msg.get_channel().create_message(msg);
println(hlog, "message sent");
}
#else
if(0) {}
#endif
else if(!menubased) {
println(hlog, "taking test screenshot");
shot::take("go-test.png");
}
}
void go_message(string s) {
println(hlog, s);
addMessage(s);
#if AEGIS
if(cur)
cur->msg.get_channel().create_message(s);
#endif
}
struct dfs {
vector<bool> visited;
vector<int> q;
dfs() {
visited.resize(isize(ac), false);
}
void visit(int i) {
if(visited[i]) return;
visited[i] = true;
q.push_back(i);
};
};
int count_breath(int pos) {
int who = current.taken[pos];
dfs d;
int result = 0;
d.visit(pos);
for(int i=0; i<isize(d.q); i++) {
int at = d.q[i];
if(current.taken[at] == Free)
result++;
else if(current.taken[at] == who)
for(int j: neigh_indices(at)) d.visit(j);
}
return result;
}
void dead_group(int pos) {
int who = current.taken[pos];
dfs d;
d.visit(pos);
for(int i=0; i<isize(d.q); i++) {
int at = d.q[i];
if(current.taken[at] == who) {
current.taken[at] = Free;
current.captures[who]++;
for(int j: neigh_indices(at)) d.visit(j);
}
}
}
bool dead_group(string s) {
int pos = str_to_index(s);
if(!valid_index(pos)) {
go_message("invalid index " + s);
return false;
}
if(current.taken[pos] == Free) {
go_message("this is free: " + s);
return false;
}
dead_group(pos);
return true;
}
bool mark_owned(string s, int who) {
int pos = str_to_index(s);
if(!valid_index(pos)) {
go_message("invalid index " + s);
return false;
}
if(current.taken[pos] != Free) {
go_message("this is not free: " + s);
return false;
}
dfs d;
d.visit(pos);
for(int i=0; i<isize(d.q); i++) {
int at = d.q[i];
if(current.taken[at] == Free) {
current.owner[at] = who;
for(int j: neigh_indices(at)) d.visit(j);
}
}
return true;
}
string mouse_label() {
if(!indices.count(mouseover)) return "?";
return index_to_str(indices[mouseover]);
}
void try_to_play(string s, int who) {
int pos = str_to_index(s);
if(!valid_index(pos)) {
go_message("invalid index: " + s);
return;
}
if(who != Free && current.taken[pos] != Free) {
go_message("this is not free: " + s);
return;
}
save_backup();
current.taken[pos] = who;
for(int i: neigh_indices(pos)) {
println(hlog, "at ", index_to_str(i), " : ", current.taken[i], " vs ", 1-who);
if(current.taken[i] == 1-who) println(hlog, "breath = ", count_breath(i));
if(current.taken[i] == 1-who && count_breath(i) == 0)
dead_group(i);
}
if(count_breath(pos) == 0) {
current.taken[pos] = Free;
go_message("suicidal move at " + s);
undo();
}
else {
take_shot();
}
}
void die_at(vector<string> tokens) {
int t = isize(tokens);
save_backup();
bool ok = false;
for(int i=1; i<t; i++)
if(dead_group(tokens[i]))
ok = true;
if(!ok) undo();
else take_shot();
}
void set_owner(vector<string> tokens, int who) {
int t = isize(tokens);
save_backup();
bool ok = false;
for(int i=1; i<t; i++) {
if(mark_owned(tokens[i], who)) ok = true;
}
if(!ok) undo();
else take_shot();
}
void clear_owner_marks() {
save_backup();
for(int i=0; i<isize(ac); i++) current.owner[i] = Free;
take_shot();
}
void accept_command(string s) {
println(hlog, "accepting command: ", s);
vector<string> tokens;
string ctoken;
for(char c: s + " ")
if(among(c, ' ', '\n', '\r', '\t')) {
if(ctoken != "") tokens.push_back(ctoken), ctoken = "";
}
else ctoken += c;
int t = isize(tokens);
if(t == 0) return;
if(tokens[0] == "b" && t == 2) {
try_to_play(tokens[1], 0);
}
if(tokens[0] == "w" && t == 2) {
try_to_play(tokens[1], 1);
}
if(tokens[0] == "center" && t == 2) {
int pos = str_to_index(tokens[1]);
if(!valid_index(pos)) go_message("invalid cell: " + tokens[1]);
else {
centerover = ac[pos];
// fullcenter();
take_shot();
}
}
if(tokens[0] == "rotate" && t == 2) {
try {
ld angle = parseld(tokens[1]);
View = spin(angle * degree) * View;
take_shot();
}
catch(hr_parse_exception& exc) {
go_message(exc.s);
}
}
if(tokens[0] == "labels" && t == 2) {
try {
labels_value = parseld(tokens[1]);
take_shot();
}
catch(hr_parse_exception& exc) {
go_message(exc.s);
}
}
if(tokens[0] == "hires") {
dynamicval<int> dx(shot::shotx, 1000);
dynamicval<int> dy(shot::shoty, 1000);
take_shot();
}
if(tokens[0] == "help")
go_message(
"b [where] - play as black\n"
"w [where] - play as white\n"
"center [where] - center the screen\n"
"rotate [degrees] - rotate the screen\n"
"die [where] - kill a group\n"
"clear - clear owner marks\n"
"labels 0..3 - show (0) no labels, (1) labels on unowned, (2) labels on empty, (3) all labels\n"
"ob [where] - own area as black\n"
"ow [where] - own area as white\n"
"score - view the score\n"
"hires - take a 1000x1000 screenshot\n"
"restart - restart\n"
"undo - undo last move\n"
);
if(tokens[0] == "save") {
save_backup();
fhstream f("go.saved-game", "wb");
f.write(history);
undo();
}
if(tokens[0] == "die") die_at(tokens);
if(tokens[0] == "clear" && t == 1)
clear_owner_marks();
if(tokens[0] == "ow" || tokens[0] == "ob")
set_owner(tokens, tokens[0][1] == 'w');
if(tokens[0] == "undo") {
if(history.empty())
go_message("no undo history");
else {
undo();
take_shot();
}
}
if(tokens[0] == "score") {
array<int, 2> owned_by, stones;
for(int i=0; i<2; i++)
owned_by[i] = stones[i] = 0;
for(int i=0; i<isize(ac); i++)
if(current.taken[i] != Free)
stones[current.taken[i]]++;
else if(current.owner[i] != Free)
owned_by[current.owner[i]]++;
shstream ss;
println(ss, "black: ", stones[0], " stones, ", owned_by[0], " area, ", current.captures[1], " prisoners");
println(ss, "white: ", stones[1], " stones, ", owned_by[1], " area, ", current.captures[0], " prisoners");
go_message(ss.s);
}
if(tokens[0] == "restart") {
save_backup();
for(int i=0; i<isize(ac); i++) current.owner[i] = current.taken[i] = Free;
current.captures[0] = 0;
current.captures[1] = 0;
take_shot();
}
}
std::thread bot_thread;
void go_discord() {
#if AEGIS
bot_thread = std::thread([] {
aegis::core bot(aegis::create_bot_t().log_level(spdlog::level::trace).token(AEGIS_TOKEN));
std::mutex lock;
bot.set_on_message_create([&](auto obj) {
if(obj.msg.get_channel().get_name() != "go") return;
std::unique_lock<std::mutex> lk(lock);
cur = &obj;
accept_command(obj.msg.get_content());
cur = nullptr;
});
bot.run();
bot.yield();
});
#endif
}
void go_menu() {
getcstat = '-';
cmode = 0;
cmode = sm::SIDE | sm::DIALOG_STRICT_X;
gamescreen(0);
dialog::init("HyperGo", iinf[itPalace].color, 150, 100);
mousing = false; /* always display letters */
array<int, 2> owned_by, stones;
for(int i=0; i<2; i++)
owned_by[i] = stones[i] = 0;
for(int i=0; i<isize(ac); i++)
if(current.taken[i] != Free)
stones[current.taken[i]]++;
else if(current.owner[i] != Free)
owned_by[current.owner[i]]++;
shstream ss;
println(ss, "black: ", stones[0], " stones, ", owned_by[0], " area, ", current.captures[1], " prisoners");
println(ss, "white: ", stones[1], " stones, ", owned_by[1], " area, ", current.captures[0], " prisoners");
dialog::addSelItem("play black", its(stones[0]), 'b');
dialog::add_action([] { try_to_play(mouse_label(), 0); });
dialog::addSelItem("play white", its(stones[1]), 'w');
dialog::add_action([] { try_to_play(mouse_label(), 1); });
dialog::addItem("remove stone", 'r');
dialog::add_action([] { try_to_play(mouse_label(), Free); });
dialog::addItem("kill a group", 'k');
dialog::add_action([] { die_at({"", mouse_label()}); });
dialog::addItem("clear owner marks", 'c');
dialog::add_action(clear_owner_marks);
dialog::addSelItem("own as black", its(owned_by[0]), 'B');
dialog::add_action([] { set_owner({"", mouse_label()}, 0); });
dialog::addSelItem("own as white", its(owned_by[1]), 'W');
dialog::add_action([] { set_owner({"", mouse_label()}, 1); });
dialog::addSelItem("white prisoners", its(current.captures[1]), 0);
dialog::addSelItem("black prisoners", its(current.captures[0]), 0);
dialog::addItem("undo", 'u');
dialog::add_action(undo);
dialog::addItem("quit", 'q');
dialog::add_action(popScreen);
dialog::display();
keyhandler = [] (int sym, int uni) {
handlePanning(sym, uni);
dialog::handleNavigation(sym, uni);
};
}
#if CAP_COMMANDLINE
int rugArgs() {
using namespace arg;
if(0) ;
else if(argis("-gobot")) {
PHASEFROM(3);
start_game();
init_go();
}
else if(argis("-go-local")) {
labels_value = 0;
showstartmenu = false;
mapeditor::drawplayer = false;
menubased = true;
PHASEFROM(3);
start_game();
init_go();
pushScreen(go_menu);
}
else if(argis("-go")) {
shift(); accept_command(args());
}
else if(argis("-go-discord"))
go_discord();
else return 1;
return 0;
}
auto gobot_hook =
addHook(hooks_args, 100, rugArgs) +
addHook(shmup::hooks_turn, 100, [] (int t) {
if(shot_state == 1) {
shot::take("go-temp.png");
shot_state = 2;
}
return false;
});
;
#endif
EX }
EX }