1
0
mirror of https://github.com/zenorogue/hyperrogue.git synced 2024-12-14 21:30:26 +00:00
hyperrogue/rogueviz/functions.cpp

162 lines
3.8 KiB
C++
Raw Normal View History

#include "rogueviz.h"
2018-10-23 15:05:46 +00:00
namespace rogueviz {
namespace graph {
vector<string> formula;
2018-10-23 15:05:46 +00:00
bool graph_on;
color_t graphcolor;
2019-09-22 23:05:36 +00:00
hyperpoint err = point3(500,0,0);
2018-10-23 15:05:46 +00:00
bool iserror(hyperpoint h) { return sqhypot_d(2, h) > 10000 || std::isnan(h[0]) || std::isnan(h[1]) || std::isnan(h[2]) || std::isinf(h[0]) || std::isinf(h[1]) || std::isinf(h[2]); }
2018-10-23 15:05:46 +00:00
hyperpoint xy_to_point(ld x, ld y) {
if(sphere && hypot(x, y) > 1)
return err;
return hpxy(x, y);
}
2018-10-23 15:05:46 +00:00
hyperpoint find_point(ld t) {
exp_parser ep;
auto &dict = ep.extra_params;
dict["t"] = t;
dict["phi"] = t * 2 * M_PI;
dict["x"] = tan(t * M_PI - M_PI/2);
for(auto& ff: formula) {
ep.s = ff;
string varname = "";
ep.at = 0;
while(!among(ep.next(), '=', -1)) varname += ep.next(), ep.at++;
ep.at++;
cld x = ep.parse();
2018-10-23 15:05:46 +00:00
if(!ep.ok()) return err;
dict[varname] = x;
}
if(!dict.count("y") && dict.count("r"))
return xspinpush0(real(dict["phi"]), real(dict["r"]));
if(dict.count("z") && dict.count("x"))
return hpxyz(real(dict["x"]), real(dict["y"]), real(dict["z"]));
if(dict.count("z")) {
return xy_to_point(real(dict["z"]), imag(dict["z"]));
}
return xy_to_point(real(dict["x"]), real(dict["y"]));
2018-10-23 15:05:46 +00:00
}
hyperpoint gcurvestart = err;
void xcurvepoint(hyperpoint h) {
2020-07-29 21:34:00 +00:00
curvepoint(h);
2018-10-23 15:05:46 +00:00
if(iserror(gcurvestart))
gcurvestart = h;
else if(sphere && intval(gcurvestart, h) > .1) {
2020-07-29 21:34:00 +00:00
queuecurve(cwtV, graphcolor, 0, PPR::LINE);
curvepoint(h);
2018-10-23 15:05:46 +00:00
gcurvestart = h;
}
}
void finish() {
if(!iserror(gcurvestart)) {
2020-07-29 21:34:00 +00:00
queuecurve(cwtV, graphcolor, 0, PPR::LINE);
2018-10-23 15:05:46 +00:00
gcurvestart = err;
}
}
int small_limit = 6, big_limit = 20;
void draw_to(ld t0, hyperpoint h0, ld t1, hyperpoint h1, int small = 0, int big = 0) {
if(iserror(h0) || iserror(h1) || intval(h0, h1) < .01) small++;
else small = 0;
if(small >= small_limit || big >= big_limit) {
xcurvepoint(h1);
return;
}
if(t1-t0 < 1e-6) {
finish();
return;
}
ld t2 = (t0 + t1) / 2;
hyperpoint h2 = find_point(t2);
draw_to(t0, h0, t2, h2, small, big+1);
draw_to(t2, h2, t1, h1, small, big+1);
}
2018-11-06 23:50:03 +00:00
int editwhich = -1;
2018-10-23 15:05:46 +00:00
void show_graph() {
cmode = sm::SIDE | sm::MAYDARK;
gamescreen(0);
dialog::init(XLAT("graph"));
for(int i=0; i<isize(formula); i++) {
if(editwhich == i) {
2018-11-06 23:50:03 +00:00
dialog::addItem(dialog::view_edited_string(), '1'+i);
2018-10-23 15:05:46 +00:00
}
else {
dialog::addItem(formula[i], editwhich == -1 ? '1'+i : 0);
2018-11-06 23:50:03 +00:00
dialog::add_action([i] () { editwhich = i; dialog::start_editing(formula[i]); });
2018-10-23 15:05:46 +00:00
}
}
dialog::addBack();
dialog::display();
keyhandler = [] (int sym, int uni) {
if(editwhich >= 0) {
2018-11-06 23:50:03 +00:00
if(dialog::handle_edit_string(sym, uni)) ;
2018-10-23 15:05:46 +00:00
else if(doexiton(sym, uni))
editwhich = -1;
}
else {
handlePanning(sym, uni);
dialog::handleNavigation(sym, uni);
// if(doexiton(sym, uni)) popScreen();
}
};
}
void frame() {
2018-10-23 15:05:46 +00:00
if(graphcolor) {
hyperpoint h0 = find_point(0);
hyperpoint h1 = find_point(1);
if(!iserror(h0)) xcurvepoint(h0);
draw_to(0, h0, 1, h1);
finish();
}
}
#if CAP_COMMANDLINE
int readArgs() {
using namespace arg;
if(0) ;
else if(argis("-dgraph")) {
PHASE(3);
showstartmenu = false;
pushScreen(show_graph);
shift();
while(args().find("=") != string::npos) {
formula.emplace_back(args());
shift();
}
graphcolor = arghex();
}
else if(argis("-dgs")) {
small_limit = argi();
}
else if(argis("-dgl")) {
big_limit = argi();
}
else return 1;
return 0;
}
#endif
auto xhook = addHook(hooks_args, 100, readArgs)
+ addHook(hooks_frame, 0, frame);
}
}