root/OpenSceneGraph/trunk/examples/osgthreadedterrain/osgthreadedterrain.cpp @ 11453

Revision 11453, 31.3 kB (checked in by robert, 5 years ago)

From Wojciech Lewandowski, "Support for compressed texture arrays + mipmaps + auto mipmap generation. Changes vs OSG trunk. Thanks to Ricardo Corsi I was able to add last time tweaks and remove few issues. This version was so well tested that it has to be posted for inclusion into OSG ;-)

"

Line 
1/* OpenSceneGraph example, osgterrain.
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#include <OpenThreads/Block>
20
21#include <osg/Group>
22#include <osg/Geode>
23#include <osg/ShapeDrawable>
24#include <osg/Texture2D>
25#include <osg/PositionAttitudeTransform>
26#include <osg/MatrixTransform>
27#include <osg/CoordinateSystemNode>
28#include <osg/ClusterCullingCallback>
29#include <osg/ArgumentParser>
30
31
32#include <osgDB/FileUtils>
33#include <osgDB/fstream>
34#include <osgDB/ReadFile>
35
36#include <osgUtil/IncrementalCompileOperation>
37
38#include <osgText/FadeText>
39
40#include <osgViewer/Viewer>
41#include <osgViewer/ViewerEventHandlers>
42
43#include <osgGA/TrackballManipulator>
44#include <osgGA/FlightManipulator>
45#include <osgGA/DriveManipulator>
46#include <osgGA/KeySwitchMatrixManipulator>
47#include <osgGA/StateSetManipulator>
48#include <osgGA/AnimationPathManipulator>
49#include <osgGA/TerrainManipulator>
50
51#include <osgTerrain/TerrainTile>
52#include <osgTerrain/GeometryTechnique>
53#include <osgTerrain/Layer>
54
55#include <iostream>
56
57
58typedef std::vector< osg::ref_ptr<osg::GraphicsThread> > GraphicsThreads;
59
60struct ReleaseBlockOnCompileCompleted : public osgUtil::IncrementalCompileOperation::CompileCompletedCallback
61{
62
63    ReleaseBlockOnCompileCompleted(osg::RefBlockCount* block):
64        _block(block) {}
65
66    virtual bool compileCompleted(osgUtil::IncrementalCompileOperation::CompileSet* compileSet)
67    {
68        if (_block.valid()) _block->completed();
69       
70        // tell IncrementalCompileOperation that it's now safe to remove the compileSet
71       
72        osg::notify(osg::NOTICE)<<"compileCompleted("<<compileSet<<")"<<std::endl;
73       
74        return true;
75    }
76
77    osg::ref_ptr<osg::RefBlockCount> _block;
78};
79
80
81class LoadAndCompileOperation : public osg::Operation
82{
83public:
84
85    LoadAndCompileOperation(const std::string& filename, osgUtil::IncrementalCompileOperation* ico , osg::RefBlockCount* block):
86        Operation("Load and compile Operation", false),
87        _filename(filename),
88        _incrementalCompileOperation(ico),
89        _block(block) {}
90
91    virtual void operator () (osg::Object* object)
92    {
93        // osg::notify(osg::NOTICE)<<"LoadAndCompileOperation "<<_filename<<std::endl;
94
95        _loadedModel = osgDB::readNodeFile(_filename);
96
97        if (_loadedModel.valid() && _incrementalCompileOperation.valid())
98        {
99            osg::ref_ptr<osgUtil::IncrementalCompileOperation::CompileSet> compileSet =
100                new osgUtil::IncrementalCompileOperation::CompileSet(_loadedModel.get());
101
102            compileSet->_compileCompletedCallback = new ReleaseBlockOnCompileCompleted(_block.get());
103
104            _incrementalCompileOperation->add(compileSet.get());
105        }
106        else 
107        {
108            if (_block.valid()) _block->completed();
109        }
110
111        // osg::notify(osg::NOTICE)<<"done LoadAndCompileOperation "<<_filename<<std::endl;
112    }
113   
114    std::string                                         _filename;
115    osg::ref_ptr<osg::Node>                             _loadedModel;
116    osg::ref_ptr<osgUtil::IncrementalCompileOperation>  _incrementalCompileOperation;
117    osg::ref_ptr<osg::RefBlockCount>                    _block;
118
119};
120
121class MasterOperation : public osg::Operation
122{
123public:
124
125    typedef std::set<std::string> Files;
126    typedef std::map<std::string, osg::ref_ptr<osg::Node> >  FilenameNodeMap;
127    typedef std::vector< osg::ref_ptr<osg::Node> >  Nodes;
128
129
130    MasterOperation(const std::string& filename, osgUtil::IncrementalCompileOperation* ico):
131        Operation("Master reading operation",true),
132        _filename(filename),
133        _incrementalCompileOperation(ico)
134    {
135    }
136   
137    /** Set the OperationQueue that the MasterOperation can use to place tasks like file loading on for other processes to handle.*/ 
138    void setOperationQueue(osg::OperationQueue* oq) { _operationQueue = oq; }
139   
140    osg::OperationQueue* getOperationQueue() { return _operationQueue.get(); }
141   
142    bool readMasterFile(Files& files) const
143    {
144        osgDB::ifstream fin(_filename.c_str());
145        if (fin)
146        {
147            osgDB::Input fr;
148            fr.attach(&fin);
149
150            bool readFilename = false;
151
152            while(!fr.eof())
153            {
154                bool itrAdvanced = false;
155                if (fr.matchSequence("file %s") || fr.matchSequence("file %w") )
156                {
157                    files.insert(fr[1].getStr());
158                    fr += 2;
159                    itrAdvanced = true;
160                    readFilename = true;
161                }
162
163                if (!itrAdvanced)
164                {
165                    ++fr;
166                }
167            }
168           
169            return readFilename;
170        }
171        return false;
172    }
173   
174    bool open(osg::Group* group)
175    {
176        Files files;
177        readMasterFile(files);
178        for(Files::iterator itr = files.begin();
179            itr != files.end();
180            ++itr)
181        {
182            osg::Node* model = osgDB::readNodeFile(*itr);
183            if (model)
184            {
185                osg::notify(osg::NOTICE)<<"open: Loaded file "<<*itr<<std::endl;
186                group->addChild(model);
187                _existingFilenameNodeMap[*itr] = model;
188            }
189        }
190       
191        return true;
192    }
193   
194
195    virtual void operator () (osg::Object* callingObject)
196    {
197        // decided which method to call according to whole has called me.
198        osgViewer::Viewer* viewer = dynamic_cast<osgViewer::Viewer*>(callingObject);
199
200        if (viewer) update(viewer->getSceneData());
201        else load();
202    }
203
204    void load()
205    {
206        //osg::notify(osg::NOTICE)<<"void load(Object)"<<std::endl;
207        Files filesA;
208        Files filesB;
209       
210        readMasterFile(filesB);
211        // osg::notify(osg::NOTICE)<<"First read "<<filesA.size()<<std::endl;
212
213        // itererate until the master file is stable
214        do
215        {
216            OpenThreads::Thread::microSleep(100000);
217
218            filesB.swap(filesA);
219            filesB.clear();
220            readMasterFile(filesB);
221
222            // osg::notify(osg::NOTICE)<<"second read "<<filesB.size()<<std::endl;
223           
224        } while (filesA!=filesB);
225
226        Files files;
227        files.swap(filesB);
228
229        // osg::notify(osg::NOTICE)<<"Now equal "<<files.size()<<std::endl;
230
231
232        Files newFiles;
233        Files removedFiles;
234
235        // find out which files are new, and which ones have been removed.
236        {           
237            OpenThreads::ScopedLock<OpenThreads::Mutex> lock(_mutex);
238
239            for(Files::iterator fitr = files.begin();
240                fitr != files.end();
241                ++fitr)
242            {
243                if (_existingFilenameNodeMap.count(*fitr)==0) newFiles.insert(*fitr);
244            }
245
246            for(FilenameNodeMap::iterator litr = _existingFilenameNodeMap.begin();
247                litr != _existingFilenameNodeMap.end();
248                ++litr)
249            {
250                if (files.count(litr->first)==0)
251                {
252                    removedFiles.insert(litr->first);
253                }
254            }
255        }
256       
257#if 0
258        if (!newFiles.empty() || !removedFiles.empty())
259        {
260            osg::notify(osg::NOTICE)<<std::endl<<std::endl<<"void operator () files.size()="<<files.size()<<std::endl;
261        }
262#endif
263
264        // first load the new files.
265        FilenameNodeMap nodesToAdd;
266        if (!newFiles.empty())
267        {
268
269            typedef std::vector< osg::ref_ptr<osg::GraphicsThread> > GraphicsThreads;
270            GraphicsThreads threads;
271       
272            for(unsigned int i=0; i<= osg::GraphicsContext::getMaxContextID(); ++i)
273            {
274                osg::GraphicsContext* gc = osg::GraphicsContext::getCompileContext(i);
275                osg::GraphicsThread* gt = gc ? gc->getGraphicsThread() : 0;
276                if (gt) threads.push_back(gt);
277            }
278
279            if (_operationQueue.valid())
280            {
281                // osg::notify(osg::NOTICE)<<"Using OperationQueue"<<std::endl;
282
283                _endOfLoadBlock = new osg::RefBlockCount(newFiles.size());
284               
285                _endOfLoadBlock->reset();
286     
287                typedef std::list< osg::ref_ptr<LoadAndCompileOperation> > LoadAndCompileList;
288                LoadAndCompileList loadAndCompileList;
289     
290                for(Files::iterator nitr = newFiles.begin();
291                    nitr != newFiles.end();
292                    ++nitr)
293                {
294                    // osg::notify(osg::NOTICE)<<"Adding LoadAndCompileOperation "<<*nitr<<std::endl;
295
296                    osg::ref_ptr<LoadAndCompileOperation> loadAndCompile = new LoadAndCompileOperation( *nitr, _incrementalCompileOperation.get(), _endOfLoadBlock.get() );
297                    loadAndCompileList.push_back(loadAndCompile);
298                    _operationQueue->add( loadAndCompile.get() );
299                }
300
301#if 1
302                osg::ref_ptr<osg::Operation> operation;
303                while ((operation=_operationQueue->getNextOperation()).valid())
304                {
305                    // osg::notify(osg::NOTICE)<<"Local running of operation"<<std::endl;
306                    (*operation)(0);
307                }
308#endif                               
309                // osg::notify(osg::NOTICE)<<"Waiting for completion of LoadAndCompile operations"<<std::endl;
310                _endOfLoadBlock->block();
311                // osg::notify(osg::NOTICE)<<"done ... Waiting for completion of LoadAndCompile operations"<<std::endl;
312               
313                for(LoadAndCompileList::iterator litr = loadAndCompileList.begin();
314                    litr != loadAndCompileList.end();
315                    ++litr)
316                {
317                    if ((*litr)->_loadedModel.valid())
318                    {
319                        nodesToAdd[(*litr)->_filename] = (*litr)->_loadedModel;
320                    }
321                }
322
323            }
324           
325            else
326            {
327
328                _endOfLoadBlock = new osg::RefBlockCount(newFiles.size());
329               
330                _endOfLoadBlock->reset();
331
332                for(Files::iterator nitr = newFiles.begin();
333                    nitr != newFiles.end();
334                    ++nitr)
335                {
336                    osg::ref_ptr<osg::Node> loadedModel = osgDB::readNodeFile(*nitr);
337
338                    if (loadedModel.get())
339                    {
340                        nodesToAdd[*nitr] = loadedModel;
341
342                        if (_incrementalCompileOperation.valid())
343                        {
344                            osg::ref_ptr<osgUtil::IncrementalCompileOperation::CompileSet> compileSet =
345                                new osgUtil::IncrementalCompileOperation::CompileSet(loadedModel.get());
346
347                            compileSet->_compileCompletedCallback = new ReleaseBlockOnCompileCompleted(_endOfLoadBlock.get());
348
349                            _incrementalCompileOperation->add(compileSet.get());
350                        }
351                        else
352                        {
353                            _endOfLoadBlock->completed();
354                        }
355                    }
356                    else
357                    {
358                        _endOfLoadBlock->completed();
359                    }
360                }
361
362                _endOfLoadBlock->block();
363
364            }
365                       
366        }
367       
368        bool requiresBlock = false;
369       
370        // pass the locally peppared data to MasterOperations shared data
371        // so that updated thread can merge these changes with the main scene
372        // graph.  This merge is carried out via the update(..) method.
373        if (!removedFiles.empty() || !nodesToAdd.empty())       
374        {
375            OpenThreads::ScopedLock<OpenThreads::Mutex> lock(_mutex);
376            _nodesToRemove.swap(removedFiles);
377            _nodesToAdd.swap(nodesToAdd);
378            requiresBlock = true;
379        }
380
381        // now block so we don't try to load anything till the new data has been merged
382        // otherwise _existingFilenameNodeMap will get out of sync.
383        if (requiresBlock)
384        {
385            _updatesMergedBlock.block();
386        }
387        else
388        {
389            OpenThreads::Thread::YieldCurrentThread();
390        }
391
392    }
393   
394    // merge the changes with the main scene graph.
395    void update(osg::Node* scene)
396    {
397        // osg::notify(osg::NOTICE)<<"void update(Node*)"<<std::endl;
398
399        osg::Group* group = dynamic_cast<osg::Group*>(scene);
400        if (!group)
401        {
402            osg::notify(osg::NOTICE)<<"Error, MasterOperation::update(Node*) can only work with a Group as Viewer::getSceneData()."<<std::endl;
403            return;
404        }
405   
406        OpenThreads::ScopedLock<OpenThreads::Mutex> lock(_mutex);
407       
408        if (!_nodesToRemove.empty() || !_nodesToAdd.empty())
409        {
410            osg::notify(osg::NOTICE)<<"update().................. "<<std::endl;
411        }
412
413        if (!_nodesToRemove.empty())
414        {
415            for(Files::iterator itr = _nodesToRemove.begin();
416                itr != _nodesToRemove.end();
417                ++itr)
418            {
419                FilenameNodeMap::iterator fnmItr = _existingFilenameNodeMap.find(*itr);
420                if (fnmItr != _existingFilenameNodeMap.end())
421                {
422                    osg::notify(osg::NOTICE)<<"  update():removing "<<*itr<<std::endl;
423               
424                    group->removeChild(fnmItr->second.get());
425                    _existingFilenameNodeMap.erase(fnmItr);
426                }
427            }
428
429            _nodesToRemove.clear();
430        }
431       
432        if (!_nodesToAdd.empty())
433        {
434            for(FilenameNodeMap::iterator itr = _nodesToAdd.begin();
435                itr != _nodesToAdd.end();
436                ++itr)
437            {
438                osg::notify(osg::NOTICE)<<"  update():inserting "<<itr->first<<std::endl;
439                group->addChild(itr->second.get());
440                _existingFilenameNodeMap[itr->first] = itr->second;
441            }
442           
443            _nodesToAdd.clear();
444        }
445
446        _updatesMergedBlock.release();
447
448    }
449   
450    // add release implementation so that any thread cancellation can
451    // work even when blocks and barriers are used.
452    virtual void release()
453    {
454        if (_operationQueue.valid()) _operationQueue->removeAllOperations();
455
456        _updatesMergedBlock.release();
457        if (_endOfCompilebarrier.valid()) _endOfCompilebarrier.release();
458        if (_endOfLoadBlock.valid()) _endOfLoadBlock.release();
459    }
460
461   
462    std::string                         _filename;
463   
464    OpenThreads::Mutex                  _mutex;
465    FilenameNodeMap                     _existingFilenameNodeMap;
466    Files                               _nodesToRemove;
467    FilenameNodeMap                     _nodesToAdd;
468    OpenThreads::Block                  _updatesMergedBlock;
469
470    osg::ref_ptr<osgUtil::IncrementalCompileOperation>  _incrementalCompileOperation;
471    osg::ref_ptr<osg::BarrierOperation> _endOfCompilebarrier;
472    osg::ref_ptr<osg::RefBlockCount>    _endOfLoadBlock;
473   
474    osg::ref_ptr<osg::OperationQueue>   _operationQueue;
475};
476
477class FilterHandler : public osgGA::GUIEventHandler
478{
479public:
480
481    FilterHandler(osgTerrain::GeometryTechnique* gt):
482        _gt(gt) {}
483
484    bool handle(const osgGA::GUIEventAdapter &ea, osgGA::GUIActionAdapter &aa)
485    {
486        if (!_gt) return false;
487
488        switch(ea.getEventType())
489        {
490        case(osgGA::GUIEventAdapter::KEYDOWN):
491            {
492                if (ea.getKey() == 'g')
493                {
494                    osg::notify(osg::NOTICE)<<"Gaussian"<<std::endl;
495                    _gt->setFilterMatrixAs(osgTerrain::GeometryTechnique::GAUSSIAN);
496                    return true;
497                }
498                else if (ea.getKey() == 's')
499                {
500                    osg::notify(osg::NOTICE)<<"Smooth"<<std::endl;
501                    _gt->setFilterMatrixAs(osgTerrain::GeometryTechnique::SMOOTH);
502                    return true;
503                }
504                else if (ea.getKey() == 'S')
505                {
506                    osg::notify(osg::NOTICE)<<"Sharpen"<<std::endl;
507                    _gt->setFilterMatrixAs(osgTerrain::GeometryTechnique::SHARPEN);
508                    return true;
509                }
510                else if (ea.getKey() == '+')
511                {
512                    _gt->setFilterWidth(_gt->getFilterWidth()*1.1);
513                    osg::notify(osg::NOTICE)<<"Filter width = "<<_gt->getFilterWidth()<<std::endl;
514                    return true;
515                }
516                else if (ea.getKey() == '-')
517                {
518                    _gt->setFilterWidth(_gt->getFilterWidth()/1.1);
519                    osg::notify(osg::NOTICE)<<"Filter width = "<<_gt->getFilterWidth()<<std::endl;
520                    return true;
521                }
522                else if (ea.getKey() == '>')
523                {
524                    _gt->setFilterBias(_gt->getFilterBias()+0.1);
525                    osg::notify(osg::NOTICE)<<"Filter bias = "<<_gt->getFilterBias()<<std::endl;
526                    return true;
527                }
528                else if (ea.getKey() == '<')
529                {
530                    _gt->setFilterBias(_gt->getFilterBias()-0.1);
531                    osg::notify(osg::NOTICE)<<"Filter bias = "<<_gt->getFilterBias()<<std::endl;
532                    return true;
533                }
534                break;
535            }
536        default:
537            break;
538        }
539        return false;
540
541    }
542
543protected:
544
545    osg::observer_ptr<osgTerrain::GeometryTechnique> _gt;
546
547};
548
549
550
551class LayerHandler : public osgGA::GUIEventHandler
552{
553public:
554
555    LayerHandler(osgTerrain::Layer* layer):
556        _layer(layer) {}
557
558    bool handle(const osgGA::GUIEventAdapter &ea, osgGA::GUIActionAdapter &aa)
559    {
560        if (!_layer) return false;
561
562        float scale = 1.2;
563
564        switch(ea.getEventType())
565        {
566        case(osgGA::GUIEventAdapter::KEYDOWN):
567            {
568                if (ea.getKey() == 'q')
569                {
570                    _layer->transform(0.0, scale);
571                    return true;
572                }
573                else if (ea.getKey() == 'a')
574                {
575                    _layer->transform(0.0, 1.0f/scale);
576                    return true;
577                }
578                break;
579            }
580        default:
581            break;
582        }
583        return false;
584
585    }
586
587protected:
588
589    osg::observer_ptr<osgTerrain::Layer> _layer;
590};
591
592int main(int argc, char** argv)
593{
594    osg::ArgumentParser arguments(&argc, argv);
595
596    // construct the viewer.
597    osgViewer::Viewer viewer(arguments);
598
599    // set up the camera manipulators.
600    {
601        osg::ref_ptr<osgGA::KeySwitchMatrixManipulator> keyswitchManipulator = new osgGA::KeySwitchMatrixManipulator;
602
603        keyswitchManipulator->addMatrixManipulator( '1', "Trackball", new osgGA::TrackballManipulator() );
604        keyswitchManipulator->addMatrixManipulator( '2', "Flight", new osgGA::FlightManipulator() );
605        keyswitchManipulator->addMatrixManipulator( '3', "Drive", new osgGA::DriveManipulator() );
606        keyswitchManipulator->addMatrixManipulator( '4', "Terrain", new osgGA::TerrainManipulator() );
607
608        std::string pathfile;
609        char keyForAnimationPath = '5';
610        while (arguments.read("-p",pathfile))
611        {
612            osgGA::AnimationPathManipulator* apm = new osgGA::AnimationPathManipulator(pathfile);
613            if (apm || !apm->valid())
614            {
615                unsigned int num = keyswitchManipulator->getNumMatrixManipulators();
616                keyswitchManipulator->addMatrixManipulator( keyForAnimationPath, "Path", apm );
617                keyswitchManipulator->selectMatrixManipulator(num);
618                ++keyForAnimationPath;
619            }
620        }
621
622        viewer.setCameraManipulator( keyswitchManipulator.get() );
623    }
624
625
626    // add the state manipulator
627    viewer.addEventHandler( new osgGA::StateSetManipulator(viewer.getCamera()->getOrCreateStateSet()) );
628
629    // add the stats handler
630    viewer.addEventHandler(new osgViewer::StatsHandler);
631
632    // add the record camera path handler
633    viewer.addEventHandler(new osgViewer::RecordCameraPathHandler);
634
635    // attach an IncrementaCompileOperation to allow the master loading
636    // to be handled with an incremental compile to avoid frame drops when large objects are added.
637    viewer.setIncrementalCompileOperation(new osgUtil::IncrementalCompileOperation());
638
639    double x = 0.0;
640    double y = 0.0;
641    double w = 1.0;
642    double h = 1.0;
643
644    unsigned int numLoadThreads = 1;
645    while (arguments.read("--load-threads",numLoadThreads)) { }
646
647    osg::ref_ptr<MasterOperation> masterOperation;
648    std::string masterFilename;
649    while(arguments.read("-m",masterFilename))
650    {
651        masterOperation = new MasterOperation(masterFilename, viewer.getIncrementalCompileOperation());
652    }
653   
654
655    osg::ref_ptr<osgTerrain::TerrainTile> terrainTile = new osgTerrain::TerrainTile;
656    osg::ref_ptr<osgTerrain::Locator> locator = new osgTerrain::Locator;
657    osg::ref_ptr<osgTerrain::ValidDataOperator> validDataOperator = new osgTerrain::NoDataValue(0.0);
658    osg::ref_ptr<osgTerrain::Layer> lastAppliedLayer;
659
660    locator->setCoordinateSystemType(osgTerrain::Locator::GEOCENTRIC);
661    locator->setTransformAsExtents(-osg::PI, -osg::PI*0.5, osg::PI, osg::PI*0.5);
662
663    unsigned int layerNum = 0;
664
665    std::string filterName;
666
667    osg::Texture::FilterMode filter = osg::Texture::LINEAR;
668
669    bool readParameter = false;
670    float minValue, maxValue;
671    float scale = 1.0f;
672    float offset = 0.0f;
673
674    int pos = 1;
675    while(pos<arguments.argc())
676    {
677        readParameter = false;
678        std::string filename;
679       
680        if (arguments.read(pos, "--layer",layerNum))
681        {
682            osg::notify(osg::NOTICE)<<"Set layer number to "<<layerNum<<std::endl;
683            readParameter = true;
684        }
685
686        else if (arguments.read(pos, "-b"))
687        {
688            terrainTile->setTreatBoundariesToValidDataAsDefaultValue(true);
689        }
690       
691        else if (arguments.read(pos, "-e",x,y,w,h))
692        {
693            // define the extents.
694            locator->setCoordinateSystemType(osgTerrain::Locator::GEOCENTRIC);
695            locator->setTransformAsExtents(x,y,x+w,y+h);
696            readParameter = true;
697        }
698
699        else if (arguments.read(pos, "--transform",offset, scale) || arguments.read(pos, "-t",offset, scale))
700        {
701            // define the extents.
702            readParameter = true;
703        }
704
705        else if (arguments.read(pos, "--cartesian",x,y,w,h))
706        {
707            // define the extents.
708            locator->setCoordinateSystemType(osgTerrain::Locator::PROJECTED);
709            locator->setTransformAsExtents(x,y,x+w,y+h);
710        }
711
712        else if (arguments.read(pos, "--hf",filename))
713        {
714            readParameter = true;
715           
716            osg::notify(osg::NOTICE)<<"--hf "<<filename<<std::endl;
717
718            osg::ref_ptr<osg::HeightField> hf = osgDB::readHeightFieldFile(filename);
719            if (hf.valid())
720            {
721                osg::ref_ptr<osgTerrain::HeightFieldLayer> hfl = new osgTerrain::HeightFieldLayer;
722                hfl->setHeightField(hf.get());
723               
724                hfl->setLocator(locator.get());
725                hfl->setValidDataOperator(validDataOperator.get());
726                hfl->setMagFilter(filter);
727               
728                if (offset!=0.0f || scale!=1.0f)
729                {
730                    hfl->transform(offset,scale);
731                }
732               
733                terrainTile->setElevationLayer(hfl.get());
734               
735                lastAppliedLayer = hfl.get();
736               
737                osg::notify(osg::NOTICE)<<"created osgTerrain::HeightFieldLayer"<<std::endl;
738            }
739            else
740            {
741                osg::notify(osg::NOTICE)<<"failed to create osgTerrain::HeightFieldLayer"<<std::endl;
742            }
743           
744            scale = 1.0f;
745            offset = 0.0f;
746           
747        }
748
749        else if (arguments.read(pos, "-d",filename) || arguments.read(pos, "--elevation-image",filename))
750        {
751            readParameter = true;
752            osg::notify(osg::NOTICE)<<"--elevation-image "<<filename<<std::endl;
753
754            osg::ref_ptr<osg::Image> image = osgDB::readImageFile(filename);
755            if (image.valid())
756            {
757                osg::ref_ptr<osgTerrain::ImageLayer> imageLayer = new osgTerrain::ImageLayer;
758                imageLayer->setImage(image.get());
759                imageLayer->setLocator(locator.get());
760                imageLayer->setValidDataOperator(validDataOperator.get());
761                imageLayer->setMagFilter(filter);
762               
763                if (offset!=0.0f || scale!=1.0f)
764                {
765                    imageLayer->transform(offset,scale);
766                }
767               
768                terrainTile->setElevationLayer(imageLayer.get());
769               
770                lastAppliedLayer = imageLayer.get();
771
772                osg::notify(osg::NOTICE)<<"created Elevation osgTerrain::ImageLayer"<<std::endl;
773            }
774            else
775            {
776                osg::notify(osg::NOTICE)<<"failed to create osgTerrain::ImageLayer"<<std::endl;
777            }
778
779            scale = 1.0f;
780            offset = 0.0f;
781           
782        }
783       
784        else if (arguments.read(pos, "-c",filename) || arguments.read(pos, "--image",filename))
785        {
786            readParameter = true;
787            osg::notify(osg::NOTICE)<<"--image "<<filename<<" x="<<x<<" y="<<y<<" w="<<w<<" h="<<h<<std::endl;
788
789            osg::ref_ptr<osg::Image> image = osgDB::readImageFile(filename);
790            if (image.valid())
791            {
792                osg::ref_ptr<osgTerrain::ImageLayer> imageLayer = new osgTerrain::ImageLayer;
793                imageLayer->setImage(image.get());
794                imageLayer->setLocator(locator.get());
795                imageLayer->setValidDataOperator(validDataOperator.get());
796                imageLayer->setMagFilter(filter);
797               
798                if (offset!=0.0f || scale!=1.0f)
799                {
800                    imageLayer->transform(offset,scale);
801                }
802
803                terrainTile->setColorLayer(layerNum, imageLayer.get());
804
805                lastAppliedLayer = imageLayer.get();
806
807                osg::notify(osg::NOTICE)<<"created Color osgTerrain::ImageLayer"<<std::endl;
808            }
809            else
810            {
811                osg::notify(osg::NOTICE)<<"failed to create osgTerrain::ImageLayer"<<std::endl;
812            }
813
814            scale = 1.0f;
815            offset = 0.0f;
816           
817        }
818
819        else if (arguments.read(pos, "--filter",filterName))
820        {
821            readParameter = true;
822           
823            if (filterName=="NEAREST")
824            {
825                osg::notify(osg::NOTICE)<<"--filter "<<filterName<<std::endl;
826                filter = osg::Texture::NEAREST;
827            }
828            else if (filterName=="LINEAR")
829            {
830                filter = osg::Texture::LINEAR;
831                osg::notify(osg::NOTICE)<<"--filter "<<filterName<<std::endl;
832            }
833            else
834            {
835                osg::notify(osg::NOTICE)<<"--filter "<<filterName<<" unrecognized filter name, please use LINEAER or NEAREST."<<std::endl;
836            }           
837
838            if (terrainTile->getColorLayer(layerNum))
839            {
840                terrainTile->getColorLayer(layerNum)->setMagFilter(filter);
841            }
842           
843        }
844
845        else if (arguments.read(pos, "--tf",minValue, maxValue))
846        {
847            readParameter = true;
848
849            osg::ref_ptr<osg::TransferFunction1D> tf = new osg::TransferFunction1D;
850           
851            unsigned int numCells = 6;
852            float delta = (maxValue-minValue)/float(numCells-1);
853            float v = minValue;
854           
855            tf->allocate(6);
856            tf->setColor(v, osg::Vec4(1.0,1.0,1.0,1.0)); v += delta;
857            tf->setColor(v, osg::Vec4(1.0,0.0,1.0,1.0)); v += delta;
858            tf->setColor(v, osg::Vec4(1.0,0.0,0.0,1.0)); v += delta;
859            tf->setColor(v, osg::Vec4(1.0,1.0,0.0,1.0)); v += delta;
860            tf->setColor(v, osg::Vec4(0.0,1.0,1.0,1.0)); v += delta;
861            tf->setColor(v, osg::Vec4(0.0,1.0,0.0,1.0));
862           
863            osg::notify(osg::NOTICE)<<"--tf "<<minValue<<" "<<maxValue<<std::endl;
864
865            terrainTile->setColorLayer(layerNum, new osgTerrain::ContourLayer(tf.get()));
866        }
867        else
868        {
869            ++pos;
870        }
871
872    }
873   
874
875    osg::ref_ptr<osg::Group> scene = new osg::Group;
876
877    if (terrainTile.valid() && (terrainTile->getElevationLayer() || terrainTile->getColorLayer(0)))
878    {
879        osg::notify(osg::NOTICE)<<"Terrain created"<<std::endl;
880   
881        scene->addChild(terrainTile.get());
882
883        osg::ref_ptr<osgTerrain::GeometryTechnique> geometryTechnique = new osgTerrain::GeometryTechnique;
884        terrainTile->setTerrainTechnique(geometryTechnique.get());
885        viewer.addEventHandler(new FilterHandler(geometryTechnique.get()));
886        viewer.addEventHandler(new LayerHandler(lastAppliedLayer.get()));
887    }
888
889    if (masterOperation.valid())
890    {
891        osg::notify(osg::NOTICE)<<"Master operation created"<<std::endl;
892
893        masterOperation->open(scene.get());
894    }
895   
896    if (scene->getNumChildren()==0)
897    {
898        osg::notify(osg::NOTICE)<<"No model created, please specify terrain or master file on command line."<<std::endl;
899        return 0;
900    }
901   
902    viewer.setSceneData(scene.get());
903
904
905    // start operation thread if a master file has been used.
906    osg::ref_ptr<osg::OperationThread> masterOperationThread;
907   
908    typedef std::list< osg::ref_ptr<osg::OperationThread> > OperationThreadList;
909    OperationThreadList generalThreadList;
910   
911    if (masterOperation.valid())
912    {
913        masterOperationThread = new osg::OperationThread;
914        masterOperationThread->startThread();
915       
916        masterOperationThread->add(masterOperation.get());
917
918//        if (numLoadThreads>0)
919        {
920            osg::ref_ptr<osg::OperationQueue> operationQueue = new osg::OperationQueue;
921            masterOperation->setOperationQueue(operationQueue.get());
922
923            for(unsigned int i=0; i<numLoadThreads; ++i)
924            {       
925                osg::ref_ptr<osg::OperationThread> thread = new osg::OperationThread;
926                thread->setOperationQueue(operationQueue.get());
927                thread->startThread();
928                generalThreadList.push_back(thread);
929            }
930        }
931       
932        viewer.addUpdateOperation(masterOperation.get());
933    }
934   
935    viewer.setThreadingModel(osgViewer::Viewer::SingleThreaded);
936   
937    // enable the use of compile contexts and associated threads.
938    // osg::DisplaySettings::instance()->setCompileContextsHint(true);
939
940    // realize the graphics windows.
941    viewer.realize();
942
943    // run the viewers main loop
944    return viewer.run();
945
946}
Note: See TracBrowser for help on using the browser.