FindNodeVisitor.h

Go to the documentation of this file.

//-----------------------------------------------------------------------------------
//
// Pro-Vocation Light Engine (PVLE)
// Copyright (C) 2007-2009  Sukender, KinoX & Buzib
// For more information, contact us : sukender@free.fr
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 3 of the License, or
// (at your option) any later version.
//
// For any use that is not compatible with the terms of the GNU 
// General Public License, please contact the authors for alternative
// licensing options.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License along
// with this program; if not, write to the Free Software Foundation, Inc.,
// 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
//
//-----------------------------------------------------------------------------------


#ifndef _3D_FIND_NODE_VISITOR_H
#define _3D_FIND_NODE_VISITOR_H

#include <osg/Node>
#include <osg/NodeVisitor>


class FindNamedNodeVisitor : public osg::NodeVisitor {
public:
    FindNamedNodeVisitor(std::string strName, TraversalMode tm=TRAVERSE_ALL_CHILDREN) : NodeVisitor(tm), strName(strName), pResult(NULL) {}
    virtual void reset() { pResult = NULL; }

    virtual void apply(osg::Node & node) {
        if (node.getName() == strName) { pResult = &node; setTraversalMode(TRAVERSE_NONE); }
        else traverse(node);
    }

    osg::Node * pResult;

protected:
    const std::string strName;
};


inline osg::Node * FindNamedNode(std::string strName, osg::Node * pRoot) {
    FindNamedNodeVisitor v(strName);
    pRoot->accept(v);
    return v.pResult;
}

// -----------------------------------


template <class T>
class FindTypedNodeVisitor : public osg::NodeVisitor {
public:
    FindTypedNodeVisitor(TraversalMode tm=TRAVERSE_ALL_CHILDREN) : osg::NodeVisitor(tm), pResult(NULL) {}
    virtual void reset() { pResult = NULL; }

    virtual void apply(T & node) { pResult = &node; setTraversalMode(osg::NodeVisitor::TRAVERSE_NONE); }

    T * pResult;
};


template <class T>
inline T * FindTypedNode(osg::Node * pRoot) {
    FindTypedNodeVisitor<T> v;
    pRoot->accept(v);
    return v.pResult;
}


#endif  // _3D_FIND_NODE_VISITOR_H

---