324 lines
9.6 KiB
C++
324 lines
9.6 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 = file_exists(path + fname);
|
|
return false;
|
|
}
|
|
|
|
bool root = false;
|
|
|
|
string ignore_mtlname = "XXX";
|
|
|
|
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, 3) = root ? sqrt(a) * 255.99 : a * 319.99;
|
|
part(nextcol, 2) = root ? sqrt(b) * 255.99 : b * 319.99;
|
|
part(nextcol, 1) = root ? sqrt(c) * 255.99 : c * 319.99;
|
|
}
|
|
if(s == "newmtl") {
|
|
emit_material();
|
|
nextcol = 0xFFFFFFFF;
|
|
texname = "";
|
|
mtlname = scanline_noblank(fsm);
|
|
}
|
|
if(s == "map_Kd") {
|
|
texname = scanline_noblank(fsm);
|
|
}
|
|
}
|
|
emit_material();
|
|
}
|
|
else if(s == "o" || s == "g") {
|
|
next_object:
|
|
object *co = nullptr;
|
|
bool textured = false;
|
|
string oname = scanline_noblank(fs);
|
|
println(hlog, "reading object: ", oname);
|
|
md.objindex.push_back(isize(md.objs));
|
|
hyperpoint ctr = Hypc;
|
|
int cqty = 0;
|
|
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);
|
|
ctr += h; cqty++;
|
|
}
|
|
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_noblank(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;
|
|
}
|
|
if(mtlname.find(ignore_mtlname) != string::npos) co->color = 0;
|
|
println(hlog, "set textured to ", textured, " color ", co->color, " mtlname = '", mtlname, "'");
|
|
}
|
|
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;
|
|
|
|
if(shift_to_ctr) {
|
|
hyperpoint ctr1 = ctr / cqty;
|
|
ctr1[3] = 0;
|
|
println(hlog, "ctr1 = ", ctr1, "hys = ", hys[0]);
|
|
for(auto& h: hys)
|
|
h -= ctr1;
|
|
}
|
|
|
|
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]; goto next_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");
|
|
|
|
md.objindex.push_back(isize(md.objs));
|
|
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) if(obj->color) {
|
|
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([] { cmode = sm::NOSCR; })
|
|
;
|
|
param_b(shift_to_ctr, "shift_to_ctr");
|
|
});
|
|
|
|
}
|
|
}
|