Move Controller base classes to SceneUtil, add visitor to assign ControllerSources

components/sceneutil/controller.cpp

@@ -0,0 +1,86 @@
+#include "controller.hpp"
+
+#include "statesetupdater.hpp"
+
+#include <osg/Drawable>
+#include <osg/Geode>
+
+namespace SceneUtil
+{
+
+
+    Controller::Controller()
+    {
+    }
+
+    bool Controller::hasInput() const
+    {
+        return mSource.get() != NULL;
+    }
+
+    float Controller::getInputValue(osg::NodeVisitor* nv)
+    {
+        return mFunction->calculate(mSource->getValue(nv));
+    }
+
+    FrameTimeSource::FrameTimeSource()
+    {
+    }
+
+    float FrameTimeSource::getValue(osg::NodeVisitor *nv)
+    {
+        return nv->getFrameStamp()->getSimulationTime();
+    }
+
+    AssignControllerSourcesVisitor::AssignControllerSourcesVisitor()
+        : osg::NodeVisitor(TRAVERSE_ALL_CHILDREN)
+    {
+    }
+
+    AssignControllerSourcesVisitor::AssignControllerSourcesVisitor(boost::shared_ptr<ControllerSource> toAssign)
+        : mToAssign(toAssign)
+        , osg::NodeVisitor(TRAVERSE_ALL_CHILDREN)
+    {
+    }
+
+    void AssignControllerSourcesVisitor::apply(osg::Node &node)
+    {
+        osg::NodeCallback* callback = node.getUpdateCallback();
+        while (callback)
+        {
+            if (Controller* ctrl = dynamic_cast<Controller*>(callback))
+                assign(node, *ctrl);
+            if (CompositeStateSetUpdater* composite = dynamic_cast<CompositeStateSetUpdater*>(callback))
+            {
+                for (unsigned int i=0; i<composite->getNumControllers(); ++i)
+                {
+                    StateSetUpdater* statesetcontroller = composite->getController(i);
+                    if (Controller* ctrl = dynamic_cast<Controller*>(statesetcontroller))
+                        assign(node, *ctrl);
+                }
+            }
+
+            callback = callback->getNestedCallback();
+        }
+
+        traverse(node);
+    }
+
+    void AssignControllerSourcesVisitor::apply(osg::Geode &geode)
+    {
+        for (unsigned int i=0; i<geode.getNumDrawables(); ++i)
+        {
+            osg::Drawable* drw = geode.getDrawable(i);
+            osg::Drawable::UpdateCallback* callback = drw->getUpdateCallback();
+            if (Controller* ctrl = dynamic_cast<Controller*>(callback))
+                assign(geode, *ctrl);
+        }
+    }
+
+    void AssignControllerSourcesVisitor::assign(osg::Node&, Controller &ctrl)
+    {
+        if (!ctrl.mSource.get())
+            ctrl.mSource = mToAssign;
+    }
+
+}