SceneNode bounds calculation

[laserbrain_demo] / src / snode.cc

diff --git a/src/snode.cc b/src/snode.cc
index 101fe62..eee5fce 100644 (file)
--- a/src/snode.cc
+++ b/src/snode.cc
@@ -2,19 +2,25 @@
 #include <assert.h>
 #include <algorithm>
 #include "snode.h"
+#include "objmesh.h"
+#include "dbg_gui.h"
 
 SceneNode::SceneNode()
        : scale(1, 1, 1)
 {
+       scene = 0;
        parent = 0;
        name = 0;
+       local_bvol_valid = false;
 }
 
 SceneNode::SceneNode(Object *obj)
        : scale(1, 1, 1)
 {
+       scene = 0;
        parent = 0;
        name = 0;
+       local_bvol_valid = false;
        add_object(obj);
 }
 
@@ -48,6 +54,7 @@ void SceneNode::add_child(SceneNode *node)
 
        children.push_back(node);
        node->parent = this;
+       node->scene = scene;
 }
 
 bool SceneNode::remove_child(SceneNode *node)
@@ -58,6 +65,8 @@ bool SceneNode::remove_child(SceneNode *node)
        if(it != children.end()) {
                assert(node->parent == this);
                node->parent = 0;
+               node->scene = 0;
+               children.erase(it);
                return true;
        }
        return false;
@@ -88,6 +97,8 @@ void SceneNode::add_object(Object *obj)
 
        this->obj.push_back(obj);
        obj->node = this;
+
+       local_bvol_valid = false;
 }
 
 bool SceneNode::remove_object(Object *o)
@@ -102,6 +113,8 @@ bool SceneNode::remove_object(Object *o)
                return false;
        }
        obj.erase(it);
+
+       local_bvol_valid = false;
        return true;
 }
 
@@ -188,13 +201,53 @@ void SceneNode::update_node(float dt)
 
 void SceneNode::update(float dt)
 {
+       bool expanded = false;
+
+       if(debug_gui) {
+               if(parent_expanded) {
+                       int flags = children.empty() ? ImGuiTreeNodeFlags_Leaf : 0;
+                       expanded = ImGui::TreeNodeEx(name ? name : "<nameless node>", flags);
+               }
+       }
+
        update_node(dt);
 
-       for(size_t i=0; i<children.size(); i++) {
+       int num = children.size();
+       for(int i=0; i<num; i++) {
+               parent_expanded = expanded;
                children[i]->update(dt);
        }
+
+       if(debug_gui && expanded) {
+               ImGui::TreePop();
+       }
 }
 
+void SceneNode::apply_xform()
+{
+       update_node();
+
+       // apply post-order to make sure we don't affect the children xform by our reset
+
+       int nchild = children.size();
+       for(int i=0; i<nchild; i++) {
+               children[i]->apply_xform();
+       }
+
+       int nobj = obj.size();
+       for(int i=0; i<nobj; i++) {
+               if(obj[i]->get_type() == OBJ_MESH) {
+                       ObjMesh *om = (ObjMesh*)obj[i];
+                       if(om->mesh) {
+                               om->mesh->apply_xform(xform);
+                       }
+               }
+       }
+
+       pos = Vec3(0, 0, 0);
+       rot = Quat::identity;
+       scale = Vec3(1, 1, 1);
+}
 
 bool SceneNode::intersect(const Ray &ray, HitPoint *hit) const
 {
@@ -229,3 +282,55 @@ bool SceneNode::intersect(const Ray &ray, HitPoint *hit) const
        }
        return false;
 }
+
+const AABox &SceneNode::calc_local_bounds()
+{
+       local_bvol = AABox(Vec3(FLT_MAX, FLT_MAX, FLT_MAX), Vec3(-FLT_MAX, -FLT_MAX, -FLT_MAX));
+
+       // calculate the axis-aligned bounding box of all objects in this node
+       int nobj = obj.size();
+       for(int i=0; i<nobj; i++) {
+               AABox tmp = obj[i]->get_aabox();
+               calc_bounding_aabox(&local_bvol, &local_bvol, &tmp);
+       }
+
+       local_bvol_valid = true;
+       return local_bvol;
+}
+
+const AABox &SceneNode::get_local_bounds() const
+{
+       if(!local_bvol_valid) {
+               ((SceneNode*)this)->calc_local_bounds();
+       }
+       return local_bvol;
+}
+
+AABox SceneNode::get_node_bounds() const
+{
+       get_local_bounds();     // validate local_bvol
+
+       // calculate the transformed local_bvol
+       Box node_bbox = Box(local_bvol, xform);
+
+       // then calculate the axis-aligned bounding box
+       AABox aabox;
+       calc_bounding_aabox(&aabox, &node_bbox);
+       return aabox;
+}
+
+AABox SceneNode::get_bounds() const
+{
+       AABox sub_aabb = AABox(Vec3(FLT_MAX, FLT_MAX, FLT_MAX), Vec3(-FLT_MAX, -FLT_MAX, -FLT_MAX));
+
+       // calculate the bounding box of all children
+       int nchild = children.size();
+       for(int i=0; i<nchild; i++) {
+               AABox tmp = children[i]->get_bounds();
+               calc_bounding_aabox(&sub_aabb, &sub_aabb, &tmp);
+       }
+
+       AABox aabb;
+       calc_bounding_aabox(&aabb, &local_bvol, &sub_aabb);
+       return aabb;
+}