621 lines
14 KiB
C++
621 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 {
|
|
#if CAP_THREADS
|
|
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(std::move(sh));
|
|
old_shots = std::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 = sm::SIDE | sm::DIALOG_STRICT_X;
|
|
gamescreen();
|
|
|
|
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 }
|
|
#endif
|
|
EX }
|
|
|