simgear/scene/util/NodeAndDrawableVisitor.cxx

   1 /* -*-c++-*-
   2  *
   3  * Copyright (C) 2008 Tim Moore
   4  *
   5  * This program is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU General Public License as
   7  * published by the Free Software Foundation; either version 2 of the
   8  * License, or (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful, but
  11  * WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
  18  * MA 02110-1301, USA.
  19  *
  20  */
  21
  22 #include <osg/Drawable>
  23 #include <osg/Geode>
  24
  25 #include "NodeAndDrawableVisitor.hxx"
  26
  27 namespace simgear
  28 {
  29 using namespace osg;
  30
  31 NodeAndDrawableVisitor::NodeAndDrawableVisitor(
  32   osg::NodeVisitor::TraversalMode tm
  33 ):
  34   NodeVisitor(tm)
  35 {
  36 }
  37
  38 NodeAndDrawableVisitor::NodeAndDrawableVisitor(
  39   osg::NodeVisitor::VisitorType type,
  40   osg::NodeVisitor::TraversalMode tm
  41 ):
  42   NodeVisitor(type, tm)
  43 {
  44 }
  45
  46 NodeAndDrawableVisitor::~NodeAndDrawableVisitor()
  47 {
  48 }
  49
  50 void NodeAndDrawableVisitor::apply(osg::Node& node)
  51 {
  52     traverse(node);
  53 }
  54
  55 void NodeAndDrawableVisitor::apply(osg::Drawable& Drawable)
  56 {
  57 }
  58
  59 void NodeAndDrawableVisitor::traverse(osg::Node& node)
  60 {
  61     TraversalMode tm = getTraversalMode();
  62     if (tm == TRAVERSE_NONE) {
  63         return;
  64     } else if (tm == TRAVERSE_PARENTS) {
  65         NodeVisitor::traverse(node);
  66         return;
  67     }
  68     Geode* geode = dynamic_cast<Geode*>(&node);
  69     if (geode) {
  70         unsigned numDrawables = geode->getNumDrawables();
  71         for (unsigned i = 0; i < numDrawables; ++i)
  72             apply(*geode->getDrawable(i));
  73     } else {
  74         NodeVisitor::traverse(node);
  75     }
  76 }
  77 }