root/OpenSceneGraph/trunk/examples/osgmultitexturecontrol/osgmultitexturecontrol.cpp @ 8639

Revision 8639, 13.6 kB (checked in by robert, 6 years ago)

Added --login <url> <username> <password> http authentication.

Line 
1/* OpenSceneGraph example, osgmultitexture.
2*
3*  Permission is hereby granted, free of charge, to any person obtaining a copy
4*  of this software and associated documentation files (the "Software"), to deal
5*  in the Software without restriction, including without limitation the rights
6*  to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
7*  copies of the Software, and to permit persons to whom the Software is
8*  furnished to do so, subject to the following conditions:
9*
10*  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
11*  IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
12*  FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
13*  AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
14*  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
15*  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
16*  THE SOFTWARE.
17*/
18
19
20#include <osg/Notify>
21
22#include <osg/Texture2D>
23#include <osg/io_utils>
24
25#include <osgDB/Registry>
26#include <osgDB/ReadFile>
27
28#include <osgFX/MultiTextureControl>
29
30#include <osgGA/TerrainManipulator>
31#include <osgGA/StateSetManipulator>
32#include <osgGA/AnimationPathManipulator>
33#include <osgGA/TrackballManipulator>
34#include <osgGA/FlightManipulator>
35#include <osgGA/DriveManipulator>
36#include <osgGA/KeySwitchMatrixManipulator>
37#include <osgGA/StateSetManipulator>
38#include <osgGA/AnimationPathManipulator>
39#include <osgGA/TerrainManipulator>
40
41#include <osgTerrain/Terrain>
42
43#include <osgViewer/ViewerEventHandlers>
44#include <osgViewer/Viewer>
45
46
47#include <iostream>
48
49template<class T>
50class FindTopMostNodeOfTypeVisitor : public osg::NodeVisitor
51{
52public:
53    FindTopMostNodeOfTypeVisitor():
54        osg::NodeVisitor(osg::NodeVisitor::TRAVERSE_ALL_CHILDREN),
55        _foundNode(0)
56    {}
57   
58    void apply(osg::Node& node)
59    {
60        T* result = dynamic_cast<T*>(&node);
61        if (result)
62        {
63            _foundNode = result;
64        }
65        else
66        {
67            traverse(node);
68        }
69    }
70   
71    T* _foundNode;
72};
73
74template<class T>
75T* findTopMostNodeOfType(osg::Node* node)
76{
77    if (!node) return 0;
78
79    FindTopMostNodeOfTypeVisitor<T> fnotv;
80    node->accept(fnotv);
81   
82    return fnotv._foundNode;
83}
84
85/** Callback used to track the elevation of the camera and update the texture weights in an MultiTextureControl node.*/
86class ElevationLayerBlendingCallback : public osg::NodeCallback
87{
88    public:
89   
90        typedef std::vector<double> Elevations;
91
92        ElevationLayerBlendingCallback(osgFX::MultiTextureControl* mtc, const Elevations& elevations, float animationTime=4.0f):
93            _previousFrame(-1),
94            _previousTime(0.0),
95            _currentElevation(0.0),
96            _mtc(mtc),
97            _elevations(elevations),
98            _animationTime(animationTime) {}
99   
100        /** Callback method called by the NodeVisitor when visiting a node.*/
101        virtual void operator()(osg::Node* node, osg::NodeVisitor* nv)
102        {
103            if (nv->getVisitorType()==osg::NodeVisitor::UPDATE_VISITOR)
104            {
105
106                float deltaTime = 0.01f;
107                if (_previousFrame!=-1)
108                {
109                    deltaTime = float(nv->getFrameStamp()->getReferenceTime() - _previousTime);
110                }
111
112                _previousTime = nv->getFrameStamp()->getReferenceTime();
113                _previousFrame = nv->getFrameStamp()->getFrameNumber();
114
115                if (_mtc.valid() && !_elevations.empty())
116                {
117                    unsigned int index = _mtc->getNumTextureWeights()-1;
118                    for(unsigned int i=0; i<_elevations.size(); ++i)
119                    {
120                        if (_currentElevation>_elevations[i])
121                        {
122                            index = i;
123                            break;
124                        }
125                    }
126
127                    float delta = std::min(deltaTime/_animationTime, 1.0f);
128
129                    for(unsigned int i=0; i<_mtc->getNumTextureWeights(); ++i)
130                    {
131                        float currentValue = _mtc->getTextureWeight(i);
132                        float desiredValue = (i==index) ? 1.0f : 0.0f;
133                        if (desiredValue != currentValue)
134                        {
135                            if (currentValue<desiredValue)
136                            {
137                                desiredValue = std::min(currentValue + delta, desiredValue);
138                            }
139                            else
140                            {
141                                desiredValue = std::max(currentValue - delta, desiredValue);
142                            }
143
144                            _mtc->setTextureWeight(i, desiredValue);
145                        }
146                    }
147
148                }
149            }
150            else if (nv->getVisitorType()==osg::NodeVisitor::CULL_VISITOR)
151            {
152                _currentElevation = nv->getViewPoint().z();
153
154                osg::CoordinateSystemNode* csn = dynamic_cast<osg::CoordinateSystemNode*>(node);
155                if (csn)
156                {
157                    osg::EllipsoidModel* em = csn->getEllipsoidModel();
158                    if (em)
159                    {
160                        double X = nv->getViewPoint().x();
161                        double Y = nv->getViewPoint().y();
162                        double Z = nv->getViewPoint().z();
163                        double latitude, longitude;
164                        em->convertXYZToLatLongHeight(X,Y,Z,latitude, longitude, _currentElevation);
165                    }
166                }
167
168            }
169
170            traverse(node,nv);
171        }
172
173        int                                             _previousFrame;
174        double                                          _previousTime;
175        float                                           _animationTime;
176        double                                          _currentElevation;
177       
178        osg::observer_ptr<osgFX::MultiTextureControl>   _mtc;
179        Elevations                                      _elevations;
180};
181
182
183// class to handle events with a pick
184class TerrainHandler : public osgGA::GUIEventHandler {
185public:
186
187    TerrainHandler(osgTerrain::Terrain* terrain):
188        _terrain(terrain) {}
189   
190    bool handle(const osgGA::GUIEventAdapter& ea,osgGA::GUIActionAdapter& aa)
191    {
192        switch(ea.getEventType())
193        {
194            case(osgGA::GUIEventAdapter::KEYDOWN):
195            {   
196                if (ea.getKey()=='r')
197                {
198                    _terrain->setSampleRatio(_terrain->getSampleRatio()*0.5);
199                    osg::notify(osg::NOTICE)<<"Sample ratio "<<_terrain->getSampleRatio()<<std::endl;
200                    return true;
201                }
202                else if (ea.getKey()=='R')
203                {
204                    _terrain->setSampleRatio(_terrain->getSampleRatio()/0.5);
205                    osg::notify(osg::NOTICE)<<"Sample ratio "<<_terrain->getSampleRatio()<<std::endl;
206                    return true;
207                }
208                else if (ea.getKey()=='v')
209                {
210                    _terrain->setVerticalScale(_terrain->getVerticalScale()*1.25);
211                    osg::notify(osg::NOTICE)<<"Vertical scale "<<_terrain->getVerticalScale()<<std::endl;
212                    return true;
213                }
214                else if (ea.getKey()=='V')
215                {
216                    _terrain->setVerticalScale(_terrain->getVerticalScale()/1.25);
217                    osg::notify(osg::NOTICE)<<"Vertical scale "<<_terrain->getVerticalScale()<<std::endl;
218                    return true;
219                }
220
221                return false;
222            }   
223            default:
224                return false;
225        }
226    }
227   
228protected:
229
230    ~TerrainHandler() {}
231
232    osg::ref_ptr<osgTerrain::Terrain>  _terrain;
233};
234
235int main( int argc, char **argv )
236{
237    // use an ArgumentParser object to manage the program arguments.
238    osg::ArgumentParser arguments(&argc,argv);
239    arguments.getApplicationUsage()->addCommandLineOption("-v","Set the terrain vertical scale.");
240    arguments.getApplicationUsage()->addCommandLineOption("-r","Set the terrain sample ratio.");
241    arguments.getApplicationUsage()->addCommandLineOption("--login <url> <username> <password>","Provide authentication information for http file access.");
242   
243    // construct the viewer.
244    osgViewer::Viewer viewer(arguments);
245   
246    float verticalScale = 1.0f;
247    while(arguments.read("-v",verticalScale)) {}
248   
249    float sampleRatio = 1.0f;
250    while(arguments.read("-r",sampleRatio)) {}
251
252    std::string url, username, password;
253    while(arguments.read("--login",url, username, password))
254    {
255        if (!osgDB::Registry::instance()->getAuthenticationMap())
256        {
257            osgDB::Registry::instance()->setAuthenticationMap(new osgDB::AuthenticationMap);
258            osgDB::Registry::instance()->getAuthenticationMap()->addAuthenticationDetails(
259                url,
260                new osgDB::AuthenticationDetails(username, password)
261            );
262        }
263    }
264
265    // add all the event handlers to the viewer
266    {
267        // add the state manipulator
268        viewer.addEventHandler( new osgGA::StateSetManipulator(viewer.getCamera()->getOrCreateStateSet()) );
269
270        // add the thread model handler
271        viewer.addEventHandler(new osgViewer::ThreadingHandler);
272
273        // add the window size toggle handler
274        viewer.addEventHandler(new osgViewer::WindowSizeHandler);
275
276        // add the stats handler
277        viewer.addEventHandler(new osgViewer::StatsHandler);
278
279        // add the help handler
280        viewer.addEventHandler(new osgViewer::HelpHandler(arguments.getApplicationUsage()));
281
282        // add the record camera path handler
283        viewer.addEventHandler(new osgViewer::RecordCameraPathHandler);
284
285        // add the LOD Scale handler
286        viewer.addEventHandler(new osgViewer::LODScaleHandler);
287    }
288
289    // add all the camera manipulators
290    {
291        osg::ref_ptr<osgGA::KeySwitchMatrixManipulator> keyswitchManipulator = new osgGA::KeySwitchMatrixManipulator;
292
293        keyswitchManipulator->addMatrixManipulator( '1', "Trackball", new osgGA::TrackballManipulator() );
294        keyswitchManipulator->addMatrixManipulator( '2', "Flight", new osgGA::FlightManipulator() );
295        keyswitchManipulator->addMatrixManipulator( '3', "Drive", new osgGA::DriveManipulator() );
296
297        unsigned int num = keyswitchManipulator->getNumMatrixManipulators();
298        keyswitchManipulator->addMatrixManipulator( '4', "Terrain", new osgGA::TerrainManipulator() );
299
300        std::string pathfile;
301        char keyForAnimationPath = '5';
302        while (arguments.read("-p",pathfile))
303        {
304            osgGA::AnimationPathManipulator* apm = new osgGA::AnimationPathManipulator(pathfile);
305            if (apm || !apm->valid())
306            {
307                num = keyswitchManipulator->getNumMatrixManipulators();
308                keyswitchManipulator->addMatrixManipulator( keyForAnimationPath, "Path", apm );
309                ++keyForAnimationPath;
310            }
311        }
312
313        keyswitchManipulator->selectMatrixManipulator(num);
314
315        viewer.setCameraManipulator( keyswitchManipulator.get() );
316    }
317
318    // set up the scene graph
319    {
320        // load the nodes from the commandline arguments.
321        osg::Node* rootnode = osgDB::readNodeFiles(arguments);
322
323        if (!rootnode)
324        {
325            osg::notify(osg::NOTICE)<<"Warning: no valid data loaded, please specify a database on the command line."<<std::endl;
326            return 1;
327        }
328
329        osgTerrain::Terrain* terrain = findTopMostNodeOfType<osgTerrain::Terrain>(rootnode);
330        if (!terrain)
331        {
332            terrain = new osgTerrain::Terrain;
333            terrain->addChild(rootnode);
334
335            rootnode = terrain;
336        }   
337
338        terrain->setSampleRatio(sampleRatio);
339        terrain->setVerticalScale(verticalScale);
340
341        // register our custom handler for adjust Terrain settings       
342        viewer.addEventHandler(new TerrainHandler(terrain));
343
344        osg::CoordinateSystemNode* csn = findTopMostNodeOfType<osg::CoordinateSystemNode>(rootnode);
345
346        unsigned int numLayers = 1;
347        osgFX::MultiTextureControl* mtc = findTopMostNodeOfType<osgFX::MultiTextureControl>(rootnode);
348        if (mtc)
349        {
350            numLayers = mtc->getNumTextureWeights();
351        }
352
353        if (numLayers<2)
354        {
355            osg::notify(osg::NOTICE)<<"Warning: scene must have MultiTextureControl node with at least 2 texture units defined."<<std::endl;
356            return 1;
357        }
358
359        double maxElevationTransition = 1e6;
360        ElevationLayerBlendingCallback::Elevations elevations;
361        for(unsigned int i=0; i<numLayers; ++i)
362        {
363            elevations.push_back(maxElevationTransition);
364            maxElevationTransition /= 2.0;
365        }
366
367        ElevationLayerBlendingCallback* elbc = new ElevationLayerBlendingCallback(mtc, elevations);
368
369        // assign to the most appropriate node (the CoordinateSystemNode is best as it provides the elevation on the globe.)
370        // note we must assign callback as both an update and cull callback, as update callback to do the update of
371        // the the osgFX::MultiTextureControl node a thread safe way, and as a cull callback to gather the camera
372        // position information.
373        osg::Node* nodeToAssignCallbackTo = csn ? csn : (mtc ? mtc : rootnode);
374        nodeToAssignCallbackTo->setUpdateCallback(elbc);   
375        nodeToAssignCallbackTo->setCullCallback(elbc);   
376
377        // add a viewport to the viewer and attach the scene graph.
378        viewer.setSceneData( rootnode );
379    }
380   
381
382
383    // create the windows and run the threads.
384    viewer.realize();
385
386    return viewer.run();
387}
Note: See TracBrowser for help on using the browser.