1
0
mirror of https://github.com/zenorogue/hyperrogue.git synced 2024-12-26 18:10:35 +00:00
hyperrogue/rogueviz/dhrg/readgraph.cpp
2022-07-23 15:45:42 +02:00

93 lines
2.7 KiB
C++

namespace dhrg {
double graph_R, graph_alpha, graph_T;
vector<pair<double, double> > coords;
rogueviz::edgetype *any;
int N;
void fixedges() {
using namespace rogueviz;
for(int i=N; i<isize(vdata); i++) if(vdata[i].m) vdata[i].m->dead = true;
for(int i=0; i<isize(vdata); i++) vdata[i].edges.clear();
vdata.resize(N);
for(auto e: edgeinfos) {
e->orig = NULL;
addedge(e->i, e->j, e);
}
storeall(N);
}
void tst() {}
void read_graph(string fn, bool subdiv, bool doRebase, bool doStore) {
any = rogueviz::add_edgetype("embedded edges");
rogueviz::fname = fn;
fhstream f(fn + "-coordinates.txt", "rt");
if(!f.f) {
printf("Missing file: %s-coordinates.txt\n", rogueviz::fname.c_str());
exit(1);
}
printf("Reading coordinates...\n");
string ignore;
if(!scan(f, ignore, ignore, ignore, ignore, N, graph_R, graph_alpha, graph_T)) {
printf("Error: incorrect format of the first line\n"); exit(1);
}
rogueviz::vdata.reserve(N);
while(true) {
string s = scan<string>(f);
if(s == "D11.11") tst();
if(s == "" || s == "#ROGUEVIZ_ENDOFDATA") break;
int id = rogueviz::getid(s);
rogueviz::vertexdata& vd(rogueviz::vdata[id]);
vd.name = s;
vd.cp = rogueviz::colorpair(rogueviz::dftcolor);
double r, alpha;
if(!scan(f, r, alpha)) { printf("Error: incorrect format of r/alpha\n"); exit(1); }
coords.push_back(make_pair(r, alpha));
transmatrix h = spin(alpha * degree) * xpush(r);
rogueviz::createViz(id, currentmap->gamestart(), h);
}
fhstream g(fn + "-links.txt", "rt");
if(!g.f) {
println(hlog, "Missing file: ", rogueviz::fname, "-links.txt");
exit(1);
}
println(hlog, "Reading links...");
int qlink = 0;
while(true) {
int i = rogueviz::readLabel(g), j = rogueviz::readLabel(g);
if(i == -1 || j == -1) break;
addedge(i, j, 1, subdiv, any);
qlink++;
}
if(doRebase) {
printf("Rebasing...\n");
for(int i=0; i<isize(rogueviz::vdata); i++) {
if(i % 10000 == 0) printf("%d/%d\n", i, isize(rogueviz::vdata));
if(rogueviz::vdata[i].m) virtualRebase(rogueviz::vdata[i].m);
}
printf("Done.\n");
}
if(doStore) rogueviz::storeall();
}
void unsnap() {
for(int i=0; i<N; i++) {
using rogueviz::vdata;
transmatrix h = spin(coords[i].second * degree) * xpush(coords[i].first);
vdata[i].m->base = currentmap->gamestart();
vdata[i].m->at = h;
virtualRebase(vdata[i].m);
}
fixedges();
}
}