Source/System/GraphOp/OSGMaterialMergeGraphOp.cpp

   1 /*---------------------------------------------------------------------------*\
   2  *                                OpenSG                                     *
   3  *                                                                           *
   4  *                                                                           *
   5  *             Copyright (C) 2000-2002 by the OpenSG Forum                   *
   6  *                                                                           *
   7  *                            www.opensg.org                                 *
   8  *                                                                           *
   9  *   contact: dirk@opensg.org, gerrit.voss@vossg.org, jbehr@zgdv.de          *
  10  *                                                                           *
  11 \*---------------------------------------------------------------------------*/
  12 /*---------------------------------------------------------------------------*\
  13  *                                License                                    *
  14  *                                                                           *
  15  * This library is free software; you can redistribute it and/or modify it   *
  16  * under the terms of the GNU Library General Public License as published    *
  17  * by the Free Software Foundation, version 2.                               *
  18  *                                                                           *
  19  * This library is distributed in the hope that it will be useful, but       *
  20  * WITHOUT ANY WARRANTY; without even the implied warranty of                *
  21  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU         *
  22  * Library General Public License for more details.                          *
  23  *                                                                           *
  24  * You should have received a copy of the GNU Library General Public         *
  25  * License along with this library; if not, write to the Free Software       *
  26  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.                 *
  27  *                                                                           *
  28 \*---------------------------------------------------------------------------*/
  29 /*---------------------------------------------------------------------------*\
  30  *                                Changes                                    *
  31  *                                                                           *
  32  *                                                                           *
  33  *                                                                           *
  34  *                                                                           *
  35  *                                                                           *
  36  *                                                                           *
  37 \*---------------------------------------------------------------------------*/
  38
  39
  40 /***************************************************************************\
  41 *                             Includes                                    *
  42 \***************************************************************************/
  43
  44 #include <sstream>
  45
  46 #include "OSGMaterialMergeGraphOp.h"
  47 #include "OSGGraphOpFactory.h"
  48 #include "OSGFieldContainerUtils.h"
  49 #include "OSGMaterial.h"
  50
  51 #include <boost/next_prior.hpp>
  52
  53 OSG_BEGIN_NAMESPACE
  54
  55 /***************************************************************************\
  56  *                            Description                                  *
  57 \***************************************************************************/
  58
  59 /*! \class OSG::MaterialMergeGraphOp
  60     \ingroup GrpSystemNodeCoresDrawablesGeometry
  61
  62     Merges equivalent materials in a scene.
  63 */
  64
  65 namespace
  66 {
  67
  68 //! Register the GraphOp with the factory
  69 static bool registerOp(void)
  70 {
  71     GraphOpRefPtr newOp = MaterialMergeGraphOp::create();
  72
  73     GraphOpFactory::the()->registerOp(newOp);
  74     return true;
  75 }
  76
  77 static OSG::StaticInitFuncWrapper registerOpWrapper(registerOp);
  78
  79 } // namespace
  80
  81 MaterialMergeGraphOp::MaterialMergeGraphOp(const char* name) :
  82      GraphOp    (name),
  83     _materialMap(    )
  84 {
  85 }
  86
  87 MaterialMergeGraphOp::~MaterialMergeGraphOp(void)
  88 {
  89 }
  90
  91 MaterialMergeGraphOpTransitPtr
  92 MaterialMergeGraphOp::create(void)
  93 {
  94     return MaterialMergeGraphOpTransitPtr(new MaterialMergeGraphOp);
  95 }
  96
  97 GraphOpTransitPtr MaterialMergeGraphOp::clone(void)
  98 {
  99     return GraphOpTransitPtr(new MaterialMergeGraphOp);
 100 }
 101
 102 bool MaterialMergeGraphOp::traverse(Node *node)
 103 {
 104     // Find the materials.
 105     if(!GraphOp::traverse(node))
 106     {
 107         return false;
 108     }
 109
 110     SINFO << "Number of materials before merge: "
 111           << _materialMap.size() << std::endl;
 112
 113     // Now do the merge.
 114     MaterialObjectMap::iterator mmIt = _materialMap.begin();
 115
 116     for (; mmIt != _materialMap.end(); ++mmIt)
 117     {
 118         Material                    *currentMat  = mmIt->first;
 119         MaterialObjectList          &currentList = mmIt->second;
 120         MaterialObjectMap::iterator  mmWalker    = boost::next(mmIt);
 121
 122         while(mmWalker != _materialMap.end())
 123         {
 124             // Store the next iterator in case we have to delete
 125             // 'walker' from the map.
 126             MaterialObjectMap::iterator nextStep = boost::next(mmWalker);
 127
 128             if(compareContainerEqual(currentMat, mmWalker->first))
 129             {
 130                 // Set the new objects to have the current material,
 131                 // and move the objects to the current list.
 132                 MaterialObjectList::iterator mlIt  = mmWalker->second.begin();
 133                 MaterialObjectList::iterator mlEnd = mmWalker->second.end  ();
 134
 135                 for (; mlIt != mlEnd; ++mlIt)
 136                 {
 137                     mlIt->setMaterial(currentMat);
 138                     currentList.push_back(*mlIt);
 139                 }
 140
 141                 _materialMap.erase(mmWalker);
 142             }
 143
 144             mmWalker = nextStep;
 145         }
 146     }
 147
 148     SINFO << "Number of materials after merge: " << _materialMap.size() << std::endl;
 149     return true;
 150 }
 151
 152
 153 void MaterialMergeGraphOp::setParams(const std::string params)
 154 {
 155     ParamSet ps(params);
 156
 157     std::string out = ps.getUnusedParams();
 158     if(out.length())
 159     {
 160         FWARNING(("MaterialMergeGraphOp doesn't have parameters '%s'.\n",
 161                 out.c_str()));
 162     }
 163 }
 164
 165 std::string MaterialMergeGraphOp::usage(void)
 166 {
 167     return
 168     "MaterialMerge: merge Materials in given subtree\n"
 169     "  Tries to find and merge equiavlent Materials to reduce the number\n"
 170     "  of Materials used.\n"
 171     ;
 172 }
 173
 174 Action::ResultE MaterialMergeGraphOp::traverseEnter(Node * const node)
 175 {
 176     if(isInExcludeList(node))
 177         return Action::Skip;
 178
 179     if(isInPreserveList(node))
 180         return Action::Continue;
 181
 182     MaterialDrawable *md = dynamic_cast<MaterialDrawable *>(node->getCore());
 183     if(md != NULL)
 184     {
 185         addObject(MaterialObject(md));
 186         return Action::Continue;
 187     }
 188
 189     MaterialGroup *mg = dynamic_cast<MaterialGroup *>(node->getCore());
 190     if(mg != NULL)
 191     {
 192         addObject(MaterialObject(mg));
 193         return Action::Continue;
 194     }
 195
 196     // Otherwise, keep looking.
 197     return Action::Continue;
 198 }
 199
 200 Action::ResultE MaterialMergeGraphOp::traverseLeave(Node * const node, Action::ResultE res)
 201 {
 202     return res;
 203 }
 204
 205 void MaterialMergeGraphOp::addObject(MaterialObject m)
 206 {
 207     Material *mat = m.getMaterial();
 208     if (mat == NULL)
 209         return;
 210
 211     _materialMap[mat].push_back(m);
 212 }
 213
 214 OSG_END_NAMESPACE