mirror of
https://github.com/zenorogue/hyperrogue.git
synced 2024-12-25 01:20:37 +00:00
rogueviz:: GoBot added
This commit is contained in:
Zeno Rogue
parent
8658d2abde
commit
bbe6843390
621
rogueviz/gobot.cpp
Normal file
621
rogueviz/gobot.cpp
Normal file
@ -0,0 +1,621 @@
|
||||
// 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 "../hyper.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;
|
||||
|
||||
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;
|
||||
}
|
||||
|
||||
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(hooks_drawcell, 100, draw_go) +
|
||||
addHook(shmup::hooks_turn, 100, [] (int t) {
|
||||
if(shot_state == 1) {
|
||||
shot::take("go-temp.png");
|
||||
shot_state = 2;
|
||||
}
|
||||
return false;
|
||||
});
|
||||
|
||||
;
|
||||
|
||||
#endif
|
||||
|
||||
EX }
|
||||
|
||||
EX }
|
||||
|
||||
Loading…
Reference in New Issue
Block a user