1
0
mirror of https://github.com/zenorogue/hyperrogue.git synced 2024-11-30 15:39:54 +00:00
hyperrogue/rogueviz/objmodels.cpp

306 lines
8.9 KiB
C++

/* a library for loading 3D models */
#include "rogueviz.h"
namespace rogueviz {
namespace objmodels {
bool scan(fhstream& hs, char& c) { return fscanf(hs.f, "%c", &c) == 1; }
char peek(fhstream& fs) {
char g = fgetc(fs.f);
ungetc(g, fs.f);
return g;
}
bool model::available() {
if(av_checked) return is_available;
av_checked = true;
is_available = false;
return false;
}
void model::load_obj(model_data& md) {
md.prec_used = prec;
auto& objects = md.objs;
fhstream fs(path+fname, "rt");
if(!fs.f)
throw hr_exception("failed to open model file: " + path + fname);
preparer();
vector<hyperpoint> vertices;
vector<hyperpoint> normals;
vector<hyperpoint> tvertices;
while(!feof(fs.f)) {
string s;
scan(fs, s);
if(s == "#") { scanline(fs); }
else if(s == "mtllib") {
string mtllib;
scan(fs, mtllib);
fhstream fsm(path+mtllib, "rt");
if(!fsm.f)
throw hr_exception("failed to open mtllib: " + mtllib);
color_t nextcol = 0xFFFFFFFF;
string mtlname, texname = "";
auto emit_material = [&] {
if(texname != "") {
texture::texture_data tdata;
materials[mtlname] = tdata;
auto& mat = materials[mtlname];
mat.twidth = mat.theight = 0; mat.stretched = true;
println(hlog, "texname: ", texname);
mat.readtexture(path+texname);
mat.loadTextureGL();
println(hlog, "texture ID: ", mat.textureid);
}
colors[mtlname] = nextcol;
println(hlog, "color of ", mtlname, " is ", nextcol);
};
while(!feof(fsm.f)) {
string s;
scan(fsm, s);
if(s == "#") { scanline(fsm); }
if(s == "Kd") {
ld a, b, c;
scan(fsm, a, b, c);
part(nextcol, 1) = a * 319.99;
part(nextcol, 2) = b * 319.99;
part(nextcol, 3) = c * 319.99;
}
if(s == "newmtl") {
emit_material();
nextcol = 0xFFFFFFFF;
texname = "";
mtlname = scanline(fsm);
}
if(s == "map_Kd") {
scan(fsm, texname);
}
}
emit_material();
}
else if(s == "o" || s == "g") {
next_object:
object *co = nullptr;
bool textured = false;
string oname = scanline(fs);
println(hlog, "reading object: ", oname);
while(true) {
if(feof(fs.f)) {
if(co) cgi.finishshape();
if(co) println(hlog, "vertices = ", co->sh.e-co->sh.s, " tvertices = ", isize(co->tv.tvertices));
break;
}
scan(fs, s);
if(s == "#") {
scanline(fs);
}
else if(s == "o" || s == "g") {
if(co) cgi.finishshape();
if(co) println(hlog, "vertices = ", co->sh.e-co->sh.s, " tvertices = ", isize(co->tv.tvertices));
goto next_object;
}
else if(s == "v") {
hyperpoint h = C0;
scan(fs, h[0], h[1], h[2]); // assume all
h[0] /= 100;
h[1] /= 100;
h[2] /= 100;
vertices.push_back(h);
}
else if(s == "vt") {
ld u, v;
scan(fs, u, v);
tvertices.push_back(point3(u, 1-v, 0));
}
else if(s == "vn") {
hyperpoint hn = C0;
scan(fs, hn[0], hn[1], hn[2]);
normals.push_back(hn);
}
else if(s == "s") {
scan(fs, s);
}
else if(s == "usemtl") {
if(co) cgi.finishshape();
if(co) println(hlog, "vertices = ", co->sh.e-co->sh.s, " tvertices = ", isize(co->tv.tvertices));
string mtlname = scanline(fs);
co = nullptr;
if(mtlname.find("Layer_Layer0") != string::npos) continue;
objects.push_back(make_shared<object>());
co = &*objects.back();
cgi.bshape(co->sh, PPR::WALL);
cgi.last->flags |= POLY_TRIANGLES;
cgi.last->texture_offset = 0;
if(materials.count(mtlname)) {
textured = true;
cgi.last->tinf = &co->tv;
co->tv.texture_id = materials[mtlname].textureid;
println(hlog, "using texture_id : ", co->tv.texture_id);
co->color = 0xFFFFFFFF;
}
else {
textured = false;
cgi.last->tinf = &co->tv;
co->tv.texture_id = floor_textures->renderedTexture;
if(colors.count(mtlname))
co->color = colors[mtlname];
else
co->color = 0xFFFFFFFF;
}
println(hlog, "set textured to ", textured);
}
else if(s == "f") {
struct vertexinfo { int f, t, n; };
array<vertexinfo, 3> vis;
char bar;
auto get_vi = [&] (vertexinfo& vi) {
vi.f = vi.t = vi.n = 1;
scan(fs, vi.f);
if(peek(fs) == '/') {
scan(fs, bar);
if(peek(fs) != '/') scan(fs, vi.t);
}
if(peek(fs) == '/') {
scan(fs, bar);
scan(fs, vi.n);
}
vi.f--; vi.t--; vi.n--;
if(vi.f < 0 || vi.f >= isize(vertices))
throw hr_exception("illegal ID");
};
get_vi(vis[0]);
get_vi(vis[1]);
next_triangle:
get_vi(vis[2]);
vector<hyperpoint> hys;
vector<hyperpoint> tot;
for(int i=0; i<3; i++) {
hys.push_back(vertices[vis[i].f]);
tot.push_back(textured ? tvertices[vis[i].t] : point3(0,0,0));
}
if(!co) continue;
hyperpoint norm = (hys[1] - hys[0]) ^ (hys[2] - hys[0]);
norm /= hypot_d(3, norm);
ld y = .5 + (.2 * norm[0] + .16 * norm[1] + .14 * norm[2]);
glvertex shade = glhr::makevertex(0, y, 0);
glvertex shadecol = glhr::makevertex(y, y, y);
auto n0 = tf(hys[0]);
auto n1 = tf(hys[1]);
auto n2 = tf(hys[2]);
auto mi = min(n0.first, min(n1.first, n2.first));
auto ma = max(n0.first, max(n1.first, n2.first));
if(ma - mi > 1) continue;
int parts = sd(hys);
auto tri = [&] (int a, int b) {
cgi.hpcpush(tf(hys[0] + (hys[1] - hys[0]) * a / parts + (hys[2] - hys[0]) * b / parts).second);
// cgi.hpcpush(tf(tot[0] + (tot[1] - tot[0]) * a / parts + (tot[2] - tot[0]) * b / parts).second);
if(textured) {
co->tv.tvertices.push_back(glhr::pointtogl(tot[0] + (tot[1] - tot[0]) * a / parts + (tot[2] - tot[0]) * b / parts));
co->tv.colors.push_back(shadecol);
}
else {
co->tv.tvertices.push_back(shade);
}
};
for(int a=0; a<parts; a++)
for(int b=0; b<parts-a; b++) {
tri(a, b);
tri(a+1, b);
tri(a, b+1);
if(a+b < parts-1) {
tri(a, b+1);
tri(a+1, b);
tri(a+1, b+1);
}
}
while(among(peek(fs), ' ', '\r', '\n')) scan(fs, bar);
if(isdigit(peek(fs))) { vis[1] = vis[2]; println(hlog, "next triangle"); goto next_triangle; }
println(hlog, "last triangle");
}
else if(s == "l") {
int a, b;
scan(fs, a, b);
/* ignore */
}
else if(s == "") { }
else
throw hr_exception("unknown command: " + s);
}
}
else
throw("unknown command: " + s);
}
println(hlog, "reading finished");
cgi.extra_vertices();
}
model_data& model::get() {
auto& md = (unique_ptr<model_data>&) cgi.ext[fname];
if(!md) {
md = std::make_unique<model_data>();
load_obj(*md);
}
if(md && md->prec_used < prec) {
println(hlog, "need prec=", prec, " used = ", md->prec_used);
md->objs.clear();
load_obj(*md);
}
return *md;
}
void model_data::render(const shiftmatrix& V) {
for(auto& obj: objs) {
queuepoly(V, obj->sh, obj->color);
}
}
void model_settings() {
emptyscreen();
dialog::init();
add_edit(prec);
dialog::addBack();
dialog::display();
}
void o_key(o_funcs& v) {
v.push_back(named_dialog("set model settings", model_settings));
}
void add_model_settings() {
rogueviz::rv_hook(hooks_o_key, 200, o_key);
}
auto cf = addHook(hooks_configfile, 100, [] {
param_f(prec, "obj_prec")
->editable(1, 100, 1, "3D model precision", "higher-precision models take more time to load and to render.", 'p')
->set_sets([] { dialog::numberdark = dialog::DONT_SHOW; })
;
});
}
}