simgear/scene/util/SGTextureStateAttributeVisitor.hxx

   1 /* -*-c++-*-
   2  *
   3  * Copyright (C) 2006 Mathias Froehlich
   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 #ifndef SG_SCENE_TEXTURESTATEATTRIBUTEVISITOR_HXX
  23 #define SG_SCENE_TEXTURESTATEATTRIBUTEVISITOR_HXX
  24
  25 class SGTextureStateAttributeVisitor : public osg::NodeVisitor {
  26 public:
  27   SGTextureStateAttributeVisitor() :
  28     osg::NodeVisitor(osg::NodeVisitor::NODE_VISITOR,
  29                      osg::NodeVisitor::TRAVERSE_ALL_CHILDREN)
  30   { }
  31
  32   virtual void apply(int textureUnit, osg::StateSet::RefAttributePair& refAttr)
  33   { }
  34   virtual void apply(int textureUnit, osg::StateSet::AttributeList& attrList)
  35   {
  36     osg::StateSet::AttributeList::iterator i;
  37     i = attrList.begin();
  38     while (i != attrList.end()) {
  39       apply(textureUnit, i->second);
  40       ++i;
  41     }
  42   }
  43   virtual void apply(osg::StateSet::TextureAttributeList& attrList)
  44   {
  45     for (unsigned i = 0; i < attrList.size(); ++i)
  46       apply(i, attrList[i]);
  47   }
  48   virtual void apply(osg::StateSet* stateSet)
  49   {
  50     if (!stateSet)
  51       return;
  52     apply(stateSet->getTextureAttributeList());
  53   }
  54
  55   virtual void apply(osg::Node& node)
  56   {
  57     apply(node.getStateSet());
  58     traverse(node);
  59   }
  60   virtual void apply(osg::Geode& node)
  61   {
  62     unsigned nDrawables = node.getNumDrawables();
  63     for (unsigned i = 0; i < nDrawables; ++i)
  64       apply(node.getDrawable(i)->getStateSet());
  65     apply(node.getStateSet());
  66     traverse(node);
  67   }
  68 };
  69
  70 #endif