simgear/scene/material/matmodel.cxx

   1 // matmodel.cxx -- class to handle models tied to a material property
   2 //
   3 // Written by David Megginson, started May 1998.
   4 //
   5 // Copyright (C) 1998 - 2003  Curtis L. Olson  - http://www.flightgear.org/~curt
   6 //
   7 // This program is free software; you can redistribute it and/or
   8 // modify it under the terms of the GNU General Public License as
   9 // published by the Free Software Foundation; either version 2 of the
  10 // License, or (at your option) any later version.
  11 //
  12 // This program is distributed in the hope that it will be useful, but
  13 // WITHOUT ANY WARRANTY; without even the implied warranty of
  14 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15 // General Public License for more details.
  16 //
  17 // You should have received a copy of the GNU General Public License
  18 // along with this program; if not, write to the Free Software
  19 // Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  20 //
  21 // $Id$
  22
  23
  24 #ifdef HAVE_CONFIG_H
  25 #  include <simgear_config.h>
  26 #endif
  27
  28 #include <simgear/compiler.h>
  29
  30 #include <map>
  31 SG_USING_STD(map);
  32
  33 #include <osg/AlphaFunc>
  34 #include <osg/Group>
  35 #include <osg/LOD>
  36 #include <osg/StateSet>
  37 #include <osg/Transform>
  38
  39 #ifdef SG_MATH_EXCEPTION_CLASH
  40 #  include <math.h>
  41 #endif
  42
  43 #include <simgear/debug/logstream.hxx>
  44 #include <simgear/math/sg_random.h>
  45 #include <simgear/misc/sg_path.hxx>
  46 #include <simgear/misc/sgstream.hxx>
  47 #include <simgear/scene/model/modellib.hxx>
  48
  49 #include "matmodel.hxx"
  50
  51 \f
  52 ////////////////////////////////////////////////////////////////////////
  53 // Local static functions.
  54 ////////////////////////////////////////////////////////////////////////
  55
  56 /**
  57  * Internal method to test whether a file exists.
  58  *
  59  * TODO: this should be moved to a SimGear library of local file
  60  * functions.
  61  */
  62 static inline bool
  63 local_file_exists( const string& path ) {
  64     sg_gzifstream in( path );
  65     if ( ! in.is_open() ) {
  66         return false;
  67     } else {
  68         return true;
  69     }
  70 }
  71
  72
  73 \f
  74 ////////////////////////////////////////////////////////////////////////
  75 // Implementation of SGMatModel.
  76 ////////////////////////////////////////////////////////////////////////
  77
  78 SGMatModel::SGMatModel (const SGPropertyNode * node, double range_m)
  79   : _models_loaded(false),
  80     _coverage_m2(node->getDoubleValue("coverage-m2", 1000000)),
  81     _range_m(range_m)
  82 {
  83                                 // Sanity check
  84   if (_coverage_m2 < 1000) {
  85     SG_LOG(SG_INPUT, SG_ALERT, "Random object coverage " << _coverage_m2
  86            << " is too small, forcing, to 1000");
  87     _coverage_m2 = 1000;
  88   }
  89
  90                                 // Note all the model paths
  91   vector <SGPropertyNode_ptr> path_nodes = node->getChildren("path");
  92   for (unsigned int i = 0; i < path_nodes.size(); i++)
  93     _paths.push_back(path_nodes[i]->getStringValue());
  94
  95                                 // Note the heading type
  96   string hdg = node->getStringValue("heading-type", "fixed");
  97   if (hdg == "fixed") {
  98     _heading_type = HEADING_FIXED;
  99   } else if (hdg == "billboard") {
 100     _heading_type = HEADING_BILLBOARD;
 101   } else if (hdg == "random") {
 102     _heading_type = HEADING_RANDOM;
 103   } else {
 104     _heading_type = HEADING_FIXED;
 105     SG_LOG(SG_INPUT, SG_ALERT, "Unknown heading type: " << hdg
 106            << "; using 'fixed' instead.");
 107   }
 108
 109   // uncomment to preload models
 110   // load_models();
 111 }
 112
 113 SGMatModel::~SGMatModel ()
 114 {
 115 }
 116
 117 int
 118 SGMatModel::get_model_count( SGModelLib *modellib,
 119                              const string &fg_root,
 120                              SGPropertyNode *prop_root,
 121                              double sim_time_sec )
 122 {
 123   load_models( modellib, fg_root, prop_root, sim_time_sec );
 124   return _models.size();
 125 }
 126
 127 inline void
 128 SGMatModel::load_models ( SGModelLib *modellib,
 129                           const string &fg_root,
 130                           SGPropertyNode *prop_root,
 131                           double sim_time_sec )
 132 {
 133                                 // Load model only on demand
 134   if (!_models_loaded) {
 135     for (unsigned int i = 0; i < _paths.size(); i++) {
 136       osg::Node *entity = modellib->load_model( fg_root, _paths[i],
 137                                                 prop_root, sim_time_sec,
 138                                                 /*cache_object*/ true );
 139       if (entity != 0) {
 140                                 // FIXME: this stuff can be handled
 141                                 // in the XML wrapper as well (at least,
 142                                 // the billboarding should be handled
 143                                 // there).
 144         osg::LOD * lod = new osg::LOD;
 145         lod->setName("Model LOD");
 146         lod->setRangeMode(osg::LOD::DISTANCE_FROM_EYE_POINT);
 147         lod->setRange(0, 0, _range_m);
 148         if (_heading_type == HEADING_BILLBOARD) {
 149           // if the model is a billboard, it is likely :
 150           // 1. a branch with only leaves,
 151           // 2. a tree or a non rectangular shape faked by transparency
 152           // We add alpha clamp then
 153           osg::StateSet* stateSet = entity->getOrCreateStateSet();
 154           osg::AlphaFunc* alphaFunc =
 155             new osg::AlphaFunc(osg::AlphaFunc::GREATER, 0.01f);
 156           stateSet->setAttributeAndModes(alphaFunc,
 157                                          osg::StateAttribute::OVERRIDE);
 158           stateSet->setRenderingHint(osg::StateSet::TRANSPARENT_BIN);
 159
 160           osg::Transform* transform = new osg::Transform;
 161           transform->setName("Model Billboard Transform");
 162           transform->setReferenceFrame(osg::Transform::ABSOLUTE_RF);
 163           transform->addChild(entity);
 164           lod->addChild(transform);
 165         } else {
 166           lod->addChild(entity);
 167         }
 168         _models.push_back(lod);
 169       } else {
 170         SG_LOG(SG_INPUT, SG_ALERT, "Failed to load object " << _paths[i]);
 171       }
 172     }
 173   }
 174   _models_loaded = true;
 175 }
 176
 177 osg::Node*
 178 SGMatModel::get_model( int index,
 179                        SGModelLib *modellib,
 180                        const string &fg_root,
 181                        SGPropertyNode *prop_root,
 182                        double sim_time_sec )
 183 {
 184   load_models( modellib, fg_root, prop_root, sim_time_sec ); // comment this out if preloading models
 185   return _models[index].get();
 186 }
 187
 188 osg::Node*
 189 SGMatModel::get_random_model( SGModelLib *modellib,
 190                               const string &fg_root,
 191                               SGPropertyNode *prop_root,
 192                               double sim_time_sec )
 193 {
 194   load_models( modellib, fg_root, prop_root, sim_time_sec ); // comment this out if preloading models
 195   int nModels = _models.size();
 196   int index = int(sg_random() * nModels);
 197   if (index >= nModels)
 198     index = 0;
 199   return _models[index].get();
 200 }
 201
 202 double
 203 SGMatModel::get_coverage_m2 () const
 204 {
 205   return _coverage_m2;
 206 }
 207
 208 SGMatModel::HeadingType
 209 SGMatModel::get_heading_type () const
 210 {
 211   return _heading_type;
 212 }
 213
 214
 215 \f
 216 ////////////////////////////////////////////////////////////////////////
 217 // Implementation of SGMatModelGroup.
 218 ////////////////////////////////////////////////////////////////////////
 219
 220 SGMatModelGroup::SGMatModelGroup (SGPropertyNode * node)
 221   : _range_m(node->getDoubleValue("range-m", 2000))
 222 {
 223                                 // Load the object subnodes
 224   vector<SGPropertyNode_ptr> object_nodes =
 225     ((SGPropertyNode *)node)->getChildren("object");
 226   for (unsigned int i = 0; i < object_nodes.size(); i++) {
 227     const SGPropertyNode * object_node = object_nodes[i];
 228     if (object_node->hasChild("path"))
 229       _objects.push_back(new SGMatModel(object_node, _range_m));
 230     else
 231       SG_LOG(SG_INPUT, SG_ALERT, "No path supplied for object");
 232   }
 233 }
 234
 235 SGMatModelGroup::~SGMatModelGroup ()
 236 {
 237 }
 238
 239 double
 240 SGMatModelGroup::get_range_m () const
 241 {
 242   return _range_m;
 243 }
 244
 245 int
 246 SGMatModelGroup::get_object_count () const
 247 {
 248   return _objects.size();
 249 }
 250
 251 SGMatModel *
 252 SGMatModelGroup::get_object (int index) const
 253 {
 254   return _objects[index];
 255 }
 256
 257
 258 // end of matmodel.cxx