root/OpenSceneGraph/trunk/examples/osgterrain/osgterrain.cpp @ 6941

Revision 6941, 10.5 kB (checked in by robert, 7 years ago)

From Martin Lavery and Robert Osfield, Updated examples to use a variation of the MIT License

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 <osg/Group>
20#include <osg/Geode>
21#include <osg/ShapeDrawable>
22#include <osg/Texture2D>
23#include <osg/PositionAttitudeTransform>
24#include <osg/MatrixTransform>
25#include <osg/CoordinateSystemNode>
26#include <osg/ClusterCullingCallback>
27#include <osg/ArgumentParser>
28
29#include <osgDB/FileUtils>
30#include <osgDB/ReadFile>
31
32#include <osgText/FadeText>
33
34#include <osgViewer/Viewer>
35#include <osgViewer/ViewerEventHandlers>
36
37#include <osgGA/TrackballManipulator>
38#include <osgGA/FlightManipulator>
39#include <osgGA/DriveManipulator>
40#include <osgGA/KeySwitchMatrixManipulator>
41#include <osgGA/StateSetManipulator>
42#include <osgGA/AnimationPathManipulator>
43#include <osgGA/TerrainManipulator>
44
45
46#include <osgTerrain/TerrainNode>
47#include <osgTerrain/GeometryTechnique>
48#include <osgTerrain/Layer>
49
50#include <iostream>
51
52
53
54
55int main(int argc, char** argv)
56{
57    osg::ArgumentParser arguments(&argc, argv);
58
59    // construct the viewer.
60    osgViewer::Viewer viewer;
61
62    // set up the camera manipulators.
63    {
64        osg::ref_ptr<osgGA::KeySwitchMatrixManipulator> keyswitchManipulator = new osgGA::KeySwitchMatrixManipulator;
65
66        keyswitchManipulator->addMatrixManipulator( '1', "Trackball", new osgGA::TrackballManipulator() );
67        keyswitchManipulator->addMatrixManipulator( '2', "Flight", new osgGA::FlightManipulator() );
68        keyswitchManipulator->addMatrixManipulator( '3', "Drive", new osgGA::DriveManipulator() );
69        keyswitchManipulator->addMatrixManipulator( '4', "Terrain", new osgGA::TerrainManipulator() );
70
71        std::string pathfile;
72        char keyForAnimationPath = '5';
73        while (arguments.read("-p",pathfile))
74        {
75            osgGA::AnimationPathManipulator* apm = new osgGA::AnimationPathManipulator(pathfile);
76            if (apm || !apm->valid())
77            {
78                unsigned int num = keyswitchManipulator->getNumMatrixManipulators();
79                keyswitchManipulator->addMatrixManipulator( keyForAnimationPath, "Path", apm );
80                keyswitchManipulator->selectMatrixManipulator(num);
81                ++keyForAnimationPath;
82            }
83        }
84
85        viewer.setCameraManipulator( keyswitchManipulator.get() );
86    }
87
88
89    // add the state manipulator
90    viewer.addEventHandler( new osgGA::StateSetManipulator(viewer.getCamera()->getOrCreateStateSet()) );
91
92    // add the stats handler
93    viewer.addEventHandler(new osgViewer::StatsHandler);
94
95    double x = 0.0;
96    double y = 0.0;
97    double w = 1.0;
98    double h = 1.0;
99
100    osg::ref_ptr<osgTerrain::TerrainNode> terrain = new osgTerrain::TerrainNode;
101    osg::ref_ptr<osgTerrain::Locator> locator = new osgTerrain::EllipsoidLocator(-osg::PI, -osg::PI*0.5, 2.0*osg::PI, osg::PI, 0.0);
102    osg::ref_ptr<osgTerrain::ValidDataOperator> validDataOperator = new osgTerrain::NoDataValue(0.0);
103
104    unsigned int layerNum = 0;
105
106    std::string filterName;
107
108    bool readParameter = false;
109    float minValue, maxValue;
110    float scale = 1.0f;
111    float offset = 0.0f;
112
113    int pos = 1;
114    while(pos<arguments.argc())
115    {
116        readParameter = false;
117        std::string filename;
118       
119        if (arguments.read(pos, "--layer",layerNum))
120        {
121            osg::notify(osg::NOTICE)<<"Set layer number to "<<layerNum<<std::endl;
122            readParameter = true;
123        }
124
125        else if (arguments.read(pos, "-b"))
126        {
127            terrain->setTreatBoundariesToValidDataAsDefaultValue(true);
128        }
129       
130        else if (arguments.read(pos, "-e",x,y,w,h))
131        {
132            // define the extents.
133            locator = new osgTerrain::EllipsoidLocator(x,y,w,h,0);
134            readParameter = true;
135        }
136
137        else if (arguments.read(pos, "--transform",offset, scale) || arguments.read(pos, "-t",offset, scale))
138        {
139            // define the extents.
140            readParameter = true;
141        }
142
143        else if (arguments.read(pos, "--cartizian",x,y,w,h))
144        {
145            // define the extents.
146            locator = new osgTerrain::CartizianLocator(x,y,w,h,0);
147            readParameter = true;
148        }
149
150        else if (arguments.read(pos, "--hf",filename))
151        {
152            readParameter = true;
153           
154            osg::notify(osg::NOTICE)<<"--hf "<<filename<<std::endl;
155
156            osg::ref_ptr<osg::HeightField> hf = osgDB::readHeightFieldFile(filename);
157            if (hf.valid())
158            {
159                osg::ref_ptr<osgTerrain::HeightFieldLayer> hfl = new osgTerrain::HeightFieldLayer;
160                hfl->setHeightField(hf.get());
161               
162                hfl->setLocator(locator.get());
163                hfl->setValidDataOperator(validDataOperator.get());
164               
165                if (offset!=0.0f || scale!=1.0f)
166                {
167                    hfl->transform(offset,scale);
168                }
169               
170                terrain->setElevationLayer(hfl.get());
171               
172                osg::notify(osg::NOTICE)<<"created osgTerrain::HeightFieldLayer"<<std::endl;
173            }
174            else
175            {
176                osg::notify(osg::NOTICE)<<"failed to create osgTerrain::HeightFieldLayer"<<std::endl;
177            }
178           
179            scale = 1.0f;
180            offset = 0.0f;
181           
182        }
183
184        else if (arguments.read(pos, "-d",filename) || arguments.read(pos, "--elevation-image",filename))
185        {
186            readParameter = true;
187            osg::notify(osg::NOTICE)<<"--elevation-image "<<filename<<std::endl;
188
189            osg::ref_ptr<osg::Image> image = osgDB::readImageFile(filename);
190            if (image.valid())
191            {
192                osg::ref_ptr<osgTerrain::ImageLayer> imageLayer = new osgTerrain::ImageLayer;
193                imageLayer->setImage(image.get());
194                imageLayer->setLocator(locator.get());
195                imageLayer->setValidDataOperator(validDataOperator.get());
196               
197                if (offset!=0.0f || scale!=1.0f)
198                {
199                    imageLayer->transform(offset,scale);
200                }
201               
202                terrain->setElevationLayer(imageLayer.get());
203               
204                osg::notify(osg::NOTICE)<<"created Elevation osgTerrain::ImageLayer"<<std::endl;
205            }
206            else
207            {
208                osg::notify(osg::NOTICE)<<"failed to create osgTerrain::ImageLayer"<<std::endl;
209            }
210
211            scale = 1.0f;
212            offset = 0.0f;
213           
214        }
215       
216        else if (arguments.read(pos, "-c",filename) || arguments.read(pos, "--image",filename))
217        {
218            readParameter = true;
219            osg::notify(osg::NOTICE)<<"--image "<<filename<<" x="<<x<<" y="<<y<<" w="<<w<<" h="<<h<<std::endl;
220
221            osg::ref_ptr<osg::Image> image = osgDB::readImageFile(filename);
222            if (image.valid())
223            {
224                osg::ref_ptr<osgTerrain::ImageLayer> imageLayer = new osgTerrain::ImageLayer;
225                imageLayer->setImage(image.get());
226                imageLayer->setLocator(locator.get());
227                imageLayer->setValidDataOperator(validDataOperator.get());
228               
229                if (offset!=0.0f || scale!=1.0f)
230                {
231                    imageLayer->transform(offset,scale);
232                }
233
234                terrain->setColorLayer(layerNum, imageLayer.get());
235               
236                osg::notify(osg::NOTICE)<<"created Color osgTerrain::ImageLayer"<<std::endl;
237            }
238            else
239            {
240                osg::notify(osg::NOTICE)<<"failed to create osgTerrain::ImageLayer"<<std::endl;
241            }
242
243            scale = 1.0f;
244            offset = 0.0f;
245           
246        }
247
248        else if (arguments.read(pos, "--filter",filterName))
249        {
250            readParameter = true;
251           
252            if (filterName=="NEAREST")
253            {
254                osg::notify(osg::NOTICE)<<"--filter "<<filterName<<std::endl;
255                terrain->setColorFilter(layerNum, osgTerrain::TerrainNode::NEAREST);
256            }
257            else if (filterName=="LINEAER")
258            {
259                osg::notify(osg::NOTICE)<<"--filter "<<filterName<<std::endl;
260                terrain->setColorFilter(layerNum, osgTerrain::TerrainNode::LINEAR);
261            }
262            else
263            {
264                osg::notify(osg::NOTICE)<<"--filter "<<filterName<<" unrecognized filter name, please use LINEAER or NEAREST."<<std::endl;
265            }           
266           
267        }
268
269        else if (arguments.read(pos, "--tf",minValue, maxValue))
270        {
271            readParameter = true;
272
273            osg::ref_ptr<osg::TransferFunction1D> tf = new osg::TransferFunction1D;
274           
275            tf->setInputRange(minValue, maxValue);
276           
277            tf->allocate(6);
278            tf->setValue(0, osg::Vec4(1.0,1.0,1.0,1.0));
279            tf->setValue(1, osg::Vec4(1.0,0.0,1.0,1.0));
280            tf->setValue(2, osg::Vec4(1.0,0.0,0.0,1.0));
281            tf->setValue(3, osg::Vec4(1.0,1.0,0.0,1.0));
282            tf->setValue(4, osg::Vec4(0.0,1.0,1.0,1.0));
283            tf->setValue(5, osg::Vec4(0.0,1.0,0.0,1.0));
284           
285            osg::notify(osg::NOTICE)<<"--tf "<<minValue<<" "<<maxValue<<std::endl;
286
287            terrain->setColorTransferFunction(layerNum, tf.get());
288        }
289        else
290        {
291            ++pos;
292        }
293
294    }
295   
296
297    osg::ref_ptr<osgTerrain::GeometryTechnique> geometryTechnique = new osgTerrain::GeometryTechnique;
298   
299    terrain->setTerrainTechnique(geometryTechnique.get());
300   
301    if (!terrain) return 0;
302   
303    // return 0;
304
305    // add a viewport to the viewer and attach the scene graph.
306    viewer.setSceneData(terrain.get());
307
308    return viewer.run();
309}
Note: See TracBrowser for help on using the browser.