root/OpenSceneGraph/trunk/examples/osgsimulation/osgsimulation.cpp @ 6933

Revision 6933, 15.9 kB (checked in by robert, 7 years ago)

Added support for specifying external overlay file.

  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
Line 
1#ifdef WIN32
2/////////////////////////////////////////////////////////////////////////////
3// Disable unavoidable warning messages:
4
5//  4103: used #pragma pack to change alignment
6//  4114: same type qualifier used more than once
7//  4201: nonstandard extension used : nameless struct/union
8//  4237: "keyword" reserved for future use
9//  4251: class needs to have dll-interface to export class
10//  4275: non DLL-interface class used as base for DLL-interface class
11//  4290: C++ Exception Specification ignored
12//  4503: decorated name length exceeded, name was truncated
13//  4786: string too long - truncated to 255 characters
14
15#pragma warning(disable : 4103 4114 4201 4237 4251 4275 4290 4503 4335 4786)
16
17#endif // WIN32
18
19#include <osgViewer/Viewer>
20#include <osgViewer/ViewerEventHandlers>
21
22#include <osg/Group>
23#include <osg/Geode>
24#include <osg/ShapeDrawable>
25#include <osg/Texture2D>
26#include <osg/PositionAttitudeTransform>
27#include <osg/MatrixTransform>
28#include <osg/CoordinateSystemNode>
29
30#include <osgDB/FileUtils>
31#include <osgDB/ReadFile>
32
33#include <osgText/Text>
34
35#include <osg/CoordinateSystemNode>
36
37#include <osgSim/OverlayNode>
38#include <osgSim/SphereSegment>
39
40#include <osgGA/NodeTrackerManipulator>
41#include <osgGA/StateSetManipulator>
42#include <osgGA/TrackballManipulator>
43#include <osgGA/FlightManipulator>
44#include <osgGA/DriveManipulator>
45#include <osgGA/KeySwitchMatrixManipulator>
46#include <osgGA/AnimationPathManipulator>
47#include <osgGA/TerrainManipulator>
48
49#include <iostream>
50
51osg::Node* createEarth()
52{
53    osg::TessellationHints* hints = new osg::TessellationHints;
54    hints->setDetailRatio(5.0f);
55
56   
57    osg::ShapeDrawable* sd = new osg::ShapeDrawable(new osg::Sphere(osg::Vec3(0.0,0.0,0.0), osg::WGS_84_RADIUS_POLAR), hints);
58   
59    osg::Geode* geode = new osg::Geode;
60    geode->addDrawable(sd);
61   
62    std::string filename = osgDB::findDataFile("Images/land_shallow_topo_2048.jpg");
63    geode->getOrCreateStateSet()->setTextureAttributeAndModes(0, new osg::Texture2D(osgDB::readImageFile(filename)));
64   
65    osg::CoordinateSystemNode* csn = new osg::CoordinateSystemNode;
66    csn->setEllipsoidModel(new osg::EllipsoidModel());
67    csn->addChild(geode);
68   
69    return csn;
70   
71}
72
73
74class ModelPositionCallback : public osg::NodeCallback
75{
76public:
77
78    ModelPositionCallback(double speed):
79        _latitude(0.0),
80        _longitude(0.0),
81        _height(100000.0),
82        _speed(speed)
83    {
84        _rotation.makeRotate(osg::DegreesToRadians(90.0),0.0,0.0,1.0);
85    }
86   
87    void updateParameters()
88    {
89        _longitude += _speed * ((2.0*osg::PI)/360.0)/20.0;
90    }
91
92
93    virtual void operator()(osg::Node* node, osg::NodeVisitor* nv)
94    {
95        updateParameters();
96       
97        osg::NodePath nodePath = nv->getNodePath();
98
99        osg::MatrixTransform* mt = nodePath.empty() ? 0 : dynamic_cast<osg::MatrixTransform*>(nodePath.back());
100        if (mt)
101        {
102            osg::CoordinateSystemNode* csn = 0;
103
104            // find coordinate system node from our parental chain
105            unsigned int i;
106            for(i=0; i<nodePath.size() && csn==0; ++i)
107            {
108                csn = dynamic_cast<osg::CoordinateSystemNode*>(nodePath[i]);
109            }
110           
111            if (csn)
112            {
113
114
115                osg::EllipsoidModel* ellipsoid = csn->getEllipsoidModel();
116                if (ellipsoid)
117                {
118                    osg::Matrix inheritedMatrix;
119                    for(i+=1; i<nodePath.size()-1; ++i)
120                    {
121                        osg::Transform* transform = nodePath[i]->asTransform();
122                        if (transform) transform->computeLocalToWorldMatrix(inheritedMatrix, nv);
123                    }
124                   
125                    osg::Matrixd matrix(inheritedMatrix);
126
127                    //osg::Matrixd matrix;
128                    ellipsoid->computeLocalToWorldTransformFromLatLongHeight(_latitude,_longitude,_height,matrix);
129                    matrix.preMult(osg::Matrix::rotate(_rotation));
130                   
131                    mt->setMatrix(matrix);
132                }
133
134            }       
135        }
136           
137        traverse(node,nv);
138    }   
139   
140    double                  _latitude;
141    double                  _longitude;
142    double                  _height;
143    osg::Quat               _rotation;
144    double                  _speed;
145};
146
147
148class FindNamedNodeVisitor : public osg::NodeVisitor
149{
150public:
151    FindNamedNodeVisitor(const std::string& name):
152        osg::NodeVisitor(osg::NodeVisitor::TRAVERSE_ALL_CHILDREN),
153        _name(name) {}
154   
155    virtual void apply(osg::Node& node)
156    {
157        if (node.getName()==_name)
158        {
159            _foundNodes.push_back(&node);
160        }
161        traverse(node);
162    }
163   
164    typedef std::vector< osg::ref_ptr<osg::Node> > NodeList;
165
166    std::string _name;
167    NodeList _foundNodes;
168};
169
170
171int main(int argc, char **argv)
172{
173    // use an ArgumentParser object to manage the program arguments.
174    osg::ArgumentParser arguments(&argc,argv);
175   
176    // set up the usage document, in case we need to print out how to use this program.
177    arguments.getApplicationUsage()->setDescription(arguments.getApplicationName()+" is the example which demonstrates use of node tracker.");
178    arguments.getApplicationUsage()->setCommandLineUsage(arguments.getApplicationName());
179    arguments.getApplicationUsage()->addCommandLineOption("-h or --help","Display this information");
180   
181
182    // construct the viewer.
183    osgViewer::Viewer viewer;
184
185    // add the state manipulator
186    viewer.addEventHandler( new osgGA::StateSetManipulator(viewer.getCamera()->getOrCreateStateSet()) );
187   
188    // add the thread model handler
189    viewer.addEventHandler(new osgViewer::ThreadingHandler);
190
191    // add the window size toggle handler
192    viewer.addEventHandler(new osgViewer::WindowSizeHandler);
193       
194    // add the stats handler
195    viewer.addEventHandler(new osgViewer::StatsHandler);
196       
197    // add the record camera path  handler
198    viewer.addEventHandler(new osgViewer::RecordCameraPathHandler);
199
200    // add the help handler
201    viewer.addEventHandler(new osgViewer::HelpHandler(arguments.getApplicationUsage()));
202
203    // set the near far ration computation up.
204    viewer.getCamera()->setComputeNearFarMode(osg::CullSettings::COMPUTE_NEAR_FAR_USING_PRIMITIVES);
205    viewer.getCamera()->setNearFarRatio(0.000003f);
206
207
208    double speed = 1.0;
209    while (arguments.read("-f") || arguments.read("--fixed")) speed = 0.0;
210
211
212    osg::Quat rotation;
213    osg::Vec4 vec4;
214    while (arguments.read("--rotate-model",vec4[0],vec4[1],vec4[2],vec4[3]))
215    {
216        osg::Quat local_rotate;
217        local_rotate.makeRotate(osg::DegreesToRadians(vec4[0]),vec4[1],vec4[2],vec4[3]);
218       
219        rotation = rotation * local_rotate;
220    }
221
222    osg::NodeCallback* nc = 0;
223    std::string flightpath_filename;
224    while (arguments.read("--flight-path",flightpath_filename))
225    {
226        std::ifstream fin(flightpath_filename.c_str());
227        if (fin)
228        {
229            osg::AnimationPath* path = new osg::AnimationPath;
230            path->read(fin);
231            nc = new osg::AnimationPathCallback(path);
232        }
233    }
234   
235    osgGA::NodeTrackerManipulator::TrackerMode trackerMode = osgGA::NodeTrackerManipulator::NODE_CENTER_AND_ROTATION;
236    std::string mode;
237    while (arguments.read("--tracker-mode",mode))
238    {
239        if (mode=="NODE_CENTER_AND_ROTATION") trackerMode = osgGA::NodeTrackerManipulator::NODE_CENTER_AND_ROTATION;
240        else if (mode=="NODE_CENTER_AND_AZIM") trackerMode = osgGA::NodeTrackerManipulator::NODE_CENTER_AND_AZIM;
241        else if (mode=="NODE_CENTER") trackerMode = osgGA::NodeTrackerManipulator::NODE_CENTER;
242        else
243        {
244            std::cout<<"Unrecognized --tracker-mode option "<<mode<<", valid options are:"<<std::endl;
245            std::cout<<"    NODE_CENTER_AND_ROTATION"<<std::endl;
246            std::cout<<"    NODE_CENTER_AND_AZIM"<<std::endl;
247            std::cout<<"    NODE_CENTER"<<std::endl;
248            return 1;
249        }
250    }
251   
252   
253    osgGA::NodeTrackerManipulator::RotationMode rotationMode = osgGA::NodeTrackerManipulator::TRACKBALL;
254    while (arguments.read("--rotation-mode",mode))
255    {
256        if (mode=="TRACKBALL") rotationMode = osgGA::NodeTrackerManipulator::TRACKBALL;
257        else if (mode=="ELEVATION_AZIM") rotationMode = osgGA::NodeTrackerManipulator::ELEVATION_AZIM;
258        else
259        {
260            std::cout<<"Unrecognized --rotation-mode option "<<mode<<", valid options are:"<<std::endl;
261            std::cout<<"    TRACKBALL"<<std::endl;
262            std::cout<<"    ELEVATION_AZIM"<<std::endl;
263            return 1;
264        }
265    }
266
267    bool useOverlay = true;
268    while (arguments.read("--no-overlay") || arguments.read("-n")) useOverlay = false;
269   
270    osgSim::OverlayNode::OverlayTechnique technique = osgSim::OverlayNode::OBJECT_DEPENDENT_WITH_ORTHOGRAPHIC_OVERLAY;
271    while (arguments.read("--object")) technique = osgSim::OverlayNode::OBJECT_DEPENDENT_WITH_ORTHOGRAPHIC_OVERLAY;
272    while (arguments.read("--ortho") || arguments.read("--orthographic")) technique = osgSim::OverlayNode::VIEW_DEPENDENT_WITH_ORTHOGRAPHIC_OVERLAY;
273    while (arguments.read("--persp") || arguments.read("--perspective")) technique = osgSim::OverlayNode::VIEW_DEPENDENT_WITH_PERSPECTIVE_OVERLAY;
274
275    unsigned int screenNum = 0;   
276    while (arguments.read("--screen", screenNum) || arguments.read("-s", screenNum)) viewer.setUpViewOnSingleScreen(screenNum);
277
278    std::string pathfile;
279    while (arguments.read("-p",pathfile)) {}
280
281
282    // if user request help write it out to cout.
283    if (arguments.read("-h") || arguments.read("--help"))
284    {
285        arguments.getApplicationUsage()->write(std::cout);
286        return 1;
287    }
288   
289   
290    osg::ref_ptr<osgGA::NodeTrackerManipulator> tm;
291   
292    std::string overlayFilename;
293    while(arguments.read("--overlay", overlayFilename)) {}
294
295    // read the scene from the list of file specified commandline args.
296    osg::ref_ptr<osg::Node> root = osgDB::readNodeFiles(arguments);
297
298    if (!root) root = createEarth();
299
300    if (!root) return 0;
301
302
303    if (!overlayFilename.empty())
304    {
305        //osg::Object *pObj = osgDB::readObjectFile("alaska_clean.shp");
306        //osg::ref_ptr<osg::Geode> shapefile = dynamic_cast<osg::Geode*> (pObj);
307        //
308        //ConvertLatLon2EllipsoidCoordinates latlon2em;
309        //shapefile->accept(latlon2em);
310
311        osg::ref_ptr<osg::Node> shapefile = osgDB::readNodeFile(overlayFilename);
312       
313        if (!shapefile)
314        {
315            osg::notify(osg::NOTICE)<<"File `"<<overlayFilename<<"` not found"<<std::endl;
316            return 1;
317        }
318
319        osg::CoordinateSystemNode* csn = dynamic_cast<osg::CoordinateSystemNode*>(root.get());
320        if (csn)
321        {
322
323            osgSim::OverlayNode* overlayNode = new osgSim::OverlayNode(technique);
324            overlayNode->getOrCreateStateSet()->setTextureAttribute(1, new osg::TexEnv(osg::TexEnv::DECAL));
325            overlayNode->setOverlaySubgraph(shapefile.get());
326            overlayNode->setOverlayTextureSizeHint(1024);
327
328            // insert the OverlayNode between the coordinate system node and its children.
329            for(unsigned int i=0; i<csn->getNumChildren(); ++i)
330            {
331                overlayNode->addChild( csn->getChild(i) );
332            }
333
334            csn->removeChildren(0, csn->getNumChildren());
335            csn->addChild(overlayNode);
336
337            viewer.setSceneData(csn);
338        }
339        else
340        {
341            osgSim::OverlayNode* overlayNode = new osgSim::OverlayNode(technique);
342            overlayNode->getOrCreateStateSet()->setTextureAttribute(1, new osg::TexEnv(osg::TexEnv::DECAL));
343            overlayNode->setOverlaySubgraph(shapefile.get());
344            overlayNode->setOverlayTextureSizeHint(1024);
345            overlayNode->addChild(root.get());
346
347            viewer.setSceneData(overlayNode);
348        }
349    }
350    else
351    {
352   
353
354        // add a viewport to the viewer and attach the scene graph.
355        viewer.setSceneData(root.get());
356
357        osg::CoordinateSystemNode* csn = dynamic_cast<osg::CoordinateSystemNode*>(root.get());
358        if (csn)
359        {
360
361            osg::ref_ptr<osgSim::OverlayNode> overlayNode;
362            if (useOverlay)
363            {
364                overlayNode = new osgSim::OverlayNode(technique);
365
366                // insert the OverlayNode between the coordinate system node and its children.
367                for(unsigned int i=0; i<csn->getNumChildren(); ++i)
368                {
369                    overlayNode->addChild( csn->getChild(i) );
370                }
371
372                csn->removeChildren(0, csn->getNumChildren());
373                csn->addChild(overlayNode.get());
374
375                // tell the overlay node to continously update its overlay texture
376                // as we know we'll be tracking a moving target.
377                overlayNode->setContinuousUpdate(true);
378            }
379
380
381            osg::Node* cessna = osgDB::readNodeFile("cessna.osg");
382            if (cessna)
383            {
384                double s = 200000.0 / cessna->getBound().radius();
385
386                osg::MatrixTransform* scaler = new osg::MatrixTransform;
387                scaler->addChild(cessna);
388                scaler->setMatrix(osg::Matrixd::scale(s,s,s)*osg::Matrixd::rotate(rotation));
389                scaler->getOrCreateStateSet()->setMode(GL_RESCALE_NORMAL,osg::StateAttribute::ON);       
390
391                if (false)
392                {
393                    osgSim::SphereSegment* ss = new osgSim::SphereSegment(
394                                        osg::Vec3(0.0f,0.0f,0.0f), // center
395                                        19.9f, // radius
396                                        osg::DegreesToRadians(135.0f),
397                                        osg::DegreesToRadians(240.0f),
398                                        osg::DegreesToRadians(-10.0f),
399                                        osg::DegreesToRadians(30.0f),
400                                        60);
401
402                    scaler->addChild(ss);
403                }
404
405                osg::MatrixTransform* mt = new osg::MatrixTransform;
406                mt->addChild(scaler);
407
408
409                if (!nc) nc = new ModelPositionCallback(speed);
410
411                mt->setUpdateCallback(nc);
412
413                csn->addChild(mt);
414
415                // if we are using an overaly node, use the cessna subgraph as the overlay subgraph
416                if (overlayNode.valid())
417                {
418                    overlayNode->setOverlaySubgraph(mt);
419                }
420
421                tm = new osgGA::NodeTrackerManipulator;
422                tm->setTrackerMode(trackerMode);
423                tm->setRotationMode(rotationMode);
424                tm->setTrackNode(scaler);
425            }
426            else
427            {
428                 std::cout<<"Failed to read cessna.osg"<<std::endl;
429            }
430
431        }   
432    }
433
434    // set up camera manipulators.
435    {
436        osg::ref_ptr<osgGA::KeySwitchMatrixManipulator> keyswitchManipulator = new osgGA::KeySwitchMatrixManipulator;
437
438        if (tm.valid()) keyswitchManipulator->addMatrixManipulator( '0', "NodeTracker", tm.get() );
439
440        keyswitchManipulator->addMatrixManipulator( '1', "Trackball", new osgGA::TrackballManipulator() );
441        keyswitchManipulator->addMatrixManipulator( '2', "Flight", new osgGA::FlightManipulator() );
442        keyswitchManipulator->addMatrixManipulator( '3', "Drive", new osgGA::DriveManipulator() );
443        keyswitchManipulator->addMatrixManipulator( '4', "Terrain", new osgGA::TerrainManipulator() );
444
445        if (!pathfile.empty())
446        {
447            osgGA::AnimationPathManipulator* apm = new osgGA::AnimationPathManipulator(pathfile);
448            if (apm || !apm->valid())
449            {
450                unsigned int num = keyswitchManipulator->getNumMatrixManipulators();
451                keyswitchManipulator->addMatrixManipulator( '5', "Path", apm );
452                keyswitchManipulator->selectMatrixManipulator(num);
453            }
454        }
455
456        viewer.setCameraManipulator( keyswitchManipulator.get() );
457    }
458
459    // viewer.setThreadingModel(osgViewer::Viewer::SingleThreaded);
460
461    return viewer.run();
462}
Note: See TracBrowser for help on using the browser.