adding mirror plane options other than auto
[laserbrain_demo] / src / metascene.cc
index 399295f..09527e9 100644 (file)
@@ -1,26 +1,56 @@
+/*! \file
+ * \brief Metascene implementation
+ *
+ * Loading starts at `MetaScene::load`, which calls `ts_load` (libtreestore)
+ * to load the metascene description tree into memory. Then `proc_node` is
+ * called at the root to recursively process the tree. `scenefile` nodes are
+ * handed over to `proc_scenefile`, which will trigger scene loading for any
+ * nodes with a `file` attribute. Scene loading is handled by requesting the
+ * filename from the scene resource manager, which is of type [SceneSet](\ref SceneSet).
+ */
+#include <assert.h>
 #include <string>
 #include <regex>
 #include "metascene.h"
 #include "scene.h"
+#include "objmesh.h"
 #include "treestore.h"
 #include "logger.h"
 #include "app.h"
+#include "dbg_gui.h"
 
-#ifdef WIN32
+#if defined(WIN32) || defined(__WIN32__)
 #include <malloc.h>
 #else
 #include <alloca.h>
 #endif
 
-struct MaterialEdit {
-       std::regex name_re;
+enum MaterialEditType {
+       MTL_EDIT_TEXTURE,
+       MTL_EDIT_MIRROR
+};
+
+struct MaterialEditTexture {
        int attr;
        Texture *tex;
 };
 
+struct MaterialEditMirror {
+       float reflectivity;
+};
+
+struct MaterialEdit {
+       std::regex name_re;
+       MaterialEditType type;
+
+       MaterialEditTexture tex;
+       MaterialEditMirror mirror;
+};
+
 static bool proc_node(MetaScene *mscn, struct ts_node *node);
 static bool proc_scenefile(MetaScene *mscn, struct ts_node *node);
 static bool proc_mtledit(MetaScene *mscn, MaterialEdit *med, struct ts_node *node);
+static bool proc_music(MetaScene *mscn, struct ts_node *node);
 static void apply_mtledit(Scene *scn, const MaterialEdit &med);
 static void apply_mtledit(Material *mtl, const MaterialEdit &med);
 static struct ts_attr *attr_inscope(struct ts_node *node, const char *name);
@@ -31,13 +61,21 @@ static void print_scene_graph(SceneNode *n, int level);
 MetaScene::MetaScene()
 {
        walk_mesh = 0;
+       music = 0;
+       mirrors = 0;
 }
 
 MetaScene::~MetaScene()
 {
        delete walk_mesh;
-}
+       delete music;
 
+       while(mirrors) {
+               FlatMirror *m = mirrors;
+               mirrors = mirrors->next;
+               delete m;
+       }
+}
 
 bool MetaScene::load(const char *fname)
 {
@@ -60,20 +98,173 @@ bool MetaScene::load(const char *fname)
 
 void MetaScene::update(float dt)
 {
+       bool expanded;
+       static char text[256];
+       if(debug_gui) {
+               ImGui::Begin("MetaScene nodes", 0, 0);
+               ImGui::Columns(2);
+
+               static bool once;
+               if(!once) {
+                       float x = ImGui::GetColumnOffset(1);
+                       ImGui::SetColumnOffset(1, x * 1.7);
+                       once = true;
+               }
+       }
+
        int nscn = scenes.size();
        for(int i=0; i<nscn; i++) {
+
+               if(debug_gui) {
+                       if(scenes[i]->name.empty()) {
+                               sprintf(text, "scene %3d", i);
+                       } else {
+                               sprintf(text, "scene %3d: %s", i, scenes[i]->name.c_str());
+                       }
+                       expanded = parent_expanded = ImGui::TreeNode(text);
+                       ImGui::NextColumn();
+                       ImGui::NextColumn();
+               }
+
                scenes[i]->update(dt);
+
+               if(debug_gui && expanded) {
+                       ImGui::TreePop();
+               }
+       }
+
+       if(debug_gui) {
+               ImGui::Columns(1);
+               ImGui::End();
        }
 }
 
+// XXX not used, renderer draws
 void MetaScene::draw() const
 {
+       error_log("Don't call MetaScene::draw, use the Renderer\n");
        int nscn = scenes.size();
        for(int i=0; i<nscn; i++) {
                scenes[i]->draw();
        }
 }
 
+SceneNode *MetaScene::find_node(const char *name) const
+{
+       int num = scenes.size();
+       for(int i=0; i<num; i++) {
+               SceneNode *n = scenes[i]->find_node(name);
+               if(n) return n;
+       }
+       return 0;
+}
+
+SceneNode *MetaScene::match_node(const char *qstr) const
+{
+       int num = scenes.size();
+       for(int i=0; i<num; i++) {
+               SceneNode *n = scenes[i]->match_node(qstr);
+               if(n) return n;
+       }
+       return 0;
+}
+
+std::list<SceneNode*> MetaScene::match_nodes(const char *qstr) const
+{
+       std::list<SceneNode*> res;
+       int num = scenes.size();
+       for(int i=0; i<num; i++) {
+               std::list<SceneNode*> tmp = scenes[i]->match_nodes(qstr);
+               if(!tmp.empty()) {
+                       res.splice(res.end(), tmp);
+               }
+       }
+       return std::move(res);
+}
+
+Scene *MetaScene::extract_nodes(const char *qstr)
+{
+       Scene *scn = 0;
+       int nscn = scenes.size();
+       for(int i=0; i<nscn; i++) {
+               Scene *tmp = scenes[i]->extract_nodes(qstr);
+               if(tmp) {
+                       if(!scn) {
+                               scn = tmp;
+                       } else {
+                               scn->merge(tmp);
+                               delete tmp;
+                       }
+               }
+       }
+       return scn;
+}
+
+int MetaScene::calc_mirror_planes()
+{
+       int num_mirrors = 0;
+       while(mirrors) {
+               FlatMirror *m = mirrors;
+               mirrors = mirrors->next;
+               delete m;
+       }
+       mirrors = 0;
+       objmirror.clear();
+
+       int numscn = scenes.size();
+       for(int i=0; i<numscn; i++) {
+               Scene *scn = scenes[i];
+
+               int numobj = scn->objects.size();
+               for(int j=0; j<numobj; j++) {
+                       Object *obj = scn->objects[j];
+
+                       if(obj->mtl.flat_mirror && obj->get_type() == OBJ_MESH) {
+                               const Mesh *mesh = ((ObjMesh*)obj)->mesh;
+                               if(!mesh) continue;
+
+                               FlatMirror *mir = new FlatMirror;
+                               mir->reflect = obj->mtl.reflect;
+
+                               if(obj->mtl.flat_mirror == MTL_MIRROR_AUTO) {
+                                       // assume the object is actually flat, so grab the first triangle and make a plane
+                                       Triangle face = Triangle(0, (const Vec3*)mesh->get_attrib_data(MESH_ATTR_VERTEX),
+                                                       mesh->get_index_data());
+                                       face.calc_normal();
+
+                                       mir->plane.pt = face.v[0];
+                                       mir->plane.normal = face.normal;
+                               } else {
+                               }
+
+                               // check to see if we have found this mirror plane already
+                               bool found = false;
+                               FlatMirror *node = mirrors;
+                               while(node) {
+                                       if(1.0f - dot(mir->plane.normal, node->plane.normal) < 1e-4f &&
+                                                       fabs(dot(mir->plane.normal, normalize(mir->plane.pt - node->plane.pt))) < 1e-4) {
+                                               found = true;
+                                               break;
+                                       }
+                                       node = node->next;
+                               }
+
+                               if(!found) {
+                                       mir->next = mirrors;
+                                       mirrors = mir;
+
+                                       objmirror[obj] = mir;   // associate with object
+                                       ++num_mirrors;
+                               } else {
+                                       delete mir;
+                               }
+                       }
+               }
+       }
+
+       return num_mirrors;
+}
+
 static bool proc_node(MetaScene *mscn, struct ts_node *node)
 {
        struct ts_node *c = node->child_list;
@@ -94,6 +285,9 @@ static bool proc_node(MetaScene *mscn, struct ts_node *node)
                if(match && replace) {
                        mscn->datamap.map(match, replace);
                }
+
+       } else if(strcmp(node->name, "music") == 0) {
+               return proc_music(mscn, node);
        }
 
        return true;
@@ -102,36 +296,68 @@ static bool proc_node(MetaScene *mscn, struct ts_node *node)
 
 
 struct SceneData {
-       std::string walkmesh_regexp;
+       MetaScene *meta;
+       std::string walkmesh_regexp, spawn_regexp;
        std::vector<MaterialEdit> mtledit;
 };
 
+/*! Processes a `scenefile` node. And kicks off scene loading (if necessary) by
+ * calling `SceneSet::get`.
+ */
 static bool proc_scenefile(MetaScene *mscn, struct ts_node *node)
 {
        const char *fname = ts_get_attr_str(node, "file");
        if(fname) {
                SceneData *sdat = new SceneData;
+               sdat->meta = mscn;
 
                // datapath
                struct ts_attr *adpath = attr_inscope(node, "datapath");
                if(adpath && adpath->val.type == TS_STRING) {
-                       info_log("adding data path: %s\n", adpath->val.str);
                        mscn->datamap.set_path(adpath->val.str);
                }
 
+               // strip path
+               struct ts_attr *aspath = attr_inscope(node, "strip_path");
+               if(aspath && aspath->val.type == TS_NUMBER) {
+                       mscn->datamap.set_strip(aspath->val.inum);
+               }
+
                // walkmesh
                struct ts_attr *awmesh = attr_inscope(node, "walkmesh");
                if(awmesh && awmesh->val.type == TS_STRING) {
                        sdat->walkmesh_regexp = std::string(awmesh->val.str);
                }
 
+               // spawn node
+               struct ts_attr *awspawn = attr_inscope(node, "spawn");
+               if(awspawn) {
+                       switch(awspawn->val.type) {
+                       case TS_VECTOR:
+                               mscn->start_pos = Vec3(awspawn->val.vec[0], awspawn->val.vec[1],
+                                               awspawn->val.vec[2]);
+                               break;
+
+                       case TS_STRING:
+                       default:
+                               sdat->spawn_regexp = std::string(awspawn->val.str);
+                       }
+               }
+               if((awspawn = attr_inscope(node, "spawn_rot")) && awspawn->val.type == TS_VECTOR) {
+                       Quat rot;
+                       rot.rotate(Vec3(1, 0, 0), deg_to_rad(awspawn->val.vec[0]));
+                       rot.rotate(Vec3(0, 1, 0), deg_to_rad(awspawn->val.vec[1]));
+                       rot.rotate(Vec3(0, 0, 1), deg_to_rad(awspawn->val.vec[2]));
+                       mscn->start_rot = rot;
+               }
+
                int namesz = mscn->datamap.lookup(fname, 0, 0);
                char *namebuf = (char*)alloca(namesz + 1);
                if(mscn->datamap.lookup(fname, namebuf, namesz + 1)) {
                        fname = namebuf;
                }
 
-               // material edits
+               // material edits are kept in a list to be applied when the scene has been loaded
                struct ts_node *child = node->child_list;
                while(child) {
                        MaterialEdit medit;
@@ -184,6 +410,30 @@ bool MetaScene::scene_loaded(Scene *newscn)
                delete wscn;
        }
 
+       // extract the spawn node
+       if(!sdat->spawn_regexp.empty() && (wscn = newscn->extract_nodes(sdat->spawn_regexp.c_str()))) {
+
+               int nmeshes = wscn->meshes.size();
+               int nnodes = wscn->nodes ? wscn->nodes->get_num_children() : 0;
+
+               if(nmeshes) {
+                       Vec3 pos;
+                       for(int i=0; i<nmeshes; i++) {
+                               const Sphere &bsph = wscn->meshes[i]->get_bsphere();
+                               pos += bsph.center;
+                       }
+                       pos /= (float)nmeshes;
+                       sdat->meta->start_pos = pos;
+
+               } else if(nnodes) {
+                       // just use the first one
+                       SceneNode *first = wscn->nodes->get_child(0);
+                       sdat->meta->start_pos = first->get_position();
+                       sdat->meta->start_rot = first->get_rotation();
+               }
+               delete wscn;
+       }
+
        int num_medits = sdat->mtledit.size();
        for(int i=0; i<num_medits; i++) {
                // perform material edits
@@ -205,7 +455,6 @@ static bool proc_mtledit(MetaScene *mscn, MaterialEdit *med, struct ts_node *nod
        if(amtl && amtl->val.type == TS_STRING) {
                restr = amtl->val.str;
        }
-
        med->name_re = std::regex(restr);
 
        node = node->child_list;
@@ -234,16 +483,37 @@ static bool proc_mtledit(MetaScene *mscn, MaterialEdit *med, struct ts_node *nod
                                }
                        }
 
-                       med->attr = textype;
+                       med->tex.attr = textype;
 
                        if(!afile || !afile->val.str || !*afile->val.str) {
                                // remove
-                               med->tex = 0;
+                               med->tex.tex = 0;
                        } else {
-                               med->tex = texman.get_texture(afile->val.str, TEX_2D, &mscn->datamap);
+                               med->tex.tex = texman.get_texture(afile->val.str, TEX_2D, &mscn->datamap);
                        }
+
+                       med->type = MTL_EDIT_TEXTURE;
+                       break;
+               }
+
+               if(strcmp(cn->name, "mirror") == 0) {
+                       // make this object a flat mirror (hopefully the object is flat otherwise this won't work)
+                       float refl = 1.0f;
+
+                       struct ts_attr *arefl = ts_get_attr(cn, "reflect");
+                       if(arefl) {
+                               if(arefl->val.type == TS_NUMBER) {
+                                       refl = arefl->val.fnum;
+                               } else {
+                                       error_log("invalid reflect attribute in mirror mtledit: %s\n", arefl->val.str);
+                                       continue;
+                               }
+                       }
+
+                       med->type = MTL_EDIT_MIRROR;
+                       med->mirror.reflectivity = refl;
+                       break;
                }
-               // TODO add more edit modes
        }
 
        return true;
@@ -255,22 +525,62 @@ static void apply_mtledit(Scene *scn, const MaterialEdit &med)
        int nobj = scn->objects.size();
        for(int i=0; i<nobj; i++) {
                Object *obj = scn->objects[i];
-               if(std::regex_match(obj->get_name(), med.name_re)) {
+               if(std::regex_match(obj->mtl.name, med.name_re)) {
                        apply_mtledit(&obj->mtl, med);
                }
        }
 }
 
+static bool proc_music(MetaScene *mscn, struct ts_node *node)
+{
+       const char *fname = ts_get_attr_str(node, "file");
+       if(fname) {
+               SceneData *sdat = new SceneData;
+               sdat->meta = mscn;
+
+               // datapath
+               struct ts_attr *adpath = attr_inscope(node, "datapath");
+               if(adpath && adpath->val.type == TS_STRING) {
+                       mscn->datamap.set_path(adpath->val.str);
+               }
+
+               int namesz = mscn->datamap.lookup(fname, 0, 0);
+               char *namebuf = (char*)alloca(namesz + 1);
+               if(mscn->datamap.lookup(fname, namebuf, namesz + 1)) {
+                       fname = namebuf;
+               }
+
+               OggVorbisStream *ovstream = new OggVorbisStream;
+               if(!ovstream->open(fname)) {
+                       delete ovstream;
+                       return false;
+               }
+
+               delete mscn->music;
+               mscn->music = ovstream;
+       }
+       return true;
+}
+
 static void apply_mtledit(Material *mtl, const MaterialEdit &med)
 {
        // TODO more edit modes...
-       if(med.tex) {
-               mtl->add_texture(med.tex, med.attr);
-       } else {
-               Texture *tex = mtl->stdtex[med.attr];
-               if(tex) {
-                       mtl->remove_texture(tex);
+       switch(med.type) {
+       case MTL_EDIT_TEXTURE:
+               if(med.tex.tex) {
+                       mtl->add_texture(med.tex.tex, med.tex.attr);
+               } else {
+                       Texture *tex = mtl->stdtex[med.tex.attr];
+                       if(tex) {
+                               mtl->remove_texture(tex);
+                       }
                }
+               break;
+
+       case MTL_EDIT_MIRROR:
+               mtl->flat_mirror = med.mirror.reflectivity > 1e-6;
+               mtl->reflect = med.mirror.reflectivity;
+               break;
        }
 }