simgear/scene/material/EffectCullVisitor.cxx

   1 // Copyright (C) 2008  Timothy Moore timoore@redhat.com
   2 //
   3 // This program is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU General Public License as
   5 // published by the Free Software Foundation; either version 2 of the
   6 // License, or (at your option) any later version.
   7 //
   8 // This program is distributed in the hope that it will be useful, but
   9 // WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU General Public License
  14 // along with this program; if not, write to the Free Software
  15 // Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  16
  17 #ifdef HAVE_CONFIG_H
  18 #  include <simgear_config.h>
  19 #endif
  20
  21 #include <osg/StateSet>
  22 #include <osg/Texture2D>
  23
  24 #include "EffectCullVisitor.hxx"
  25
  26 #include "EffectGeode.hxx"
  27 #include "Effect.hxx"
  28 #include "Technique.hxx"
  29
  30 #include <simgear/scene/util/RenderConstants.hxx>
  31
  32 namespace simgear
  33 {
  34
  35 using osgUtil::CullVisitor;
  36
  37 EffectCullVisitor::EffectCullVisitor(bool collectLights) :
  38     _collectLights(collectLights)
  39 {
  40 }
  41
  42 EffectCullVisitor::EffectCullVisitor(const EffectCullVisitor& rhs) :
  43     CullVisitor(rhs)
  44 {
  45 }
  46
  47 CullVisitor* EffectCullVisitor::clone() const
  48 {
  49     return new EffectCullVisitor(*this);
  50 }
  51
  52 void EffectCullVisitor::apply(osg::Geode& node)
  53 {
  54     if (isCulled(node))
  55         return;
  56     EffectGeode *eg = dynamic_cast<EffectGeode*>(&node);
  57     if (!eg) {
  58         CullVisitor::apply(node);
  59         return;
  60     }
  61     if (_collectLights && ( eg->getNodeMask() & MODELLIGHT_BIT ) ) {
  62         _lightList.push_back( eg );
  63     }
  64     Effect* effect = eg->getEffect();
  65     Technique* technique = 0;
  66     if (!effect) {
  67         CullVisitor::apply(node);
  68         return;
  69     } else if (!(technique = effect->chooseTechnique(&getRenderInfo()))) {
  70         return;
  71     }
  72     // push the node's state.
  73     osg::StateSet* node_state = node.getStateSet();
  74     if (node_state)
  75         pushStateSet(node_state);
  76     for (EffectGeode::DrawablesIterator beginItr = eg->drawablesBegin(),
  77              e = eg->drawablesEnd();
  78          beginItr != e;
  79          beginItr = technique->processDrawables(beginItr, e, this,
  80                                                 eg->isCullingActive()))
  81         ;
  82     // pop the node's state off the geostate stack.
  83     if (node_state)
  84         popStateSet();
  85
  86 }
  87
  88 void EffectCullVisitor::reset()
  89 {
  90     _lightList.clear();
  91
  92     osgUtil::CullVisitor::reset();
  93 }
  94
  95 void EffectCullVisitor::clearBufferList()
  96 {
  97     _bufferList.clear();
  98 }
  99
 100 void EffectCullVisitor::addBuffer(std::string b, osg::Texture2D* tex)
 101 {
 102     _bufferList.insert(std::make_pair(b,tex));
 103 }
 104
 105 osg::Texture2D* EffectCullVisitor::getBuffer(std::string b)
 106 {
 107     return _bufferList[b];
 108 }
 109
 110 }