root/OpenSceneGraph/trunk/examples/osgkeyboardmouse/osgkeyboardmouse.cpp @ 6817

Revision 6817, 12.6 kB (checked in by robert, 8 years ago)

Ported across from SimpleViewer? to using Viewer

  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
Line 
1// C++ source file - (C) 2003 Robert Osfield, released under the OSGPL.
2//
3// Simple example of use of osgViewer::GraphicsWindow + Viewer
4// example that provides the user with control over view position with basic picking.
5
6#include <osg/Timer>
7#include <osg/io_utils>
8#include <osg/observer_ptr>
9
10#include <osgUtil/IntersectionVisitor>
11#include <osgUtil/PolytopeIntersector>
12#include <osgUtil/LineSegmentIntersector>
13
14#include <osgDB/ReadFile>
15#include <osgDB/WriteFile>
16
17#include <osgGA/TrackballManipulator>
18#include <osgGA/StateSetManipulator>
19
20#include <osgViewer/Viewer>
21
22#include <osgFX/Scribe>
23
24#include <iostream>
25
26class CreateModelToSaveVisitor : public osg::NodeVisitor
27{
28public:
29
30    CreateModelToSaveVisitor():
31        osg::NodeVisitor(osg::NodeVisitor::TRAVERSE_ALL_CHILDREN)       
32    {
33        _group = new osg::Group;
34        _addToModel = false;
35    }
36   
37    virtual void apply(osg::Node& node)
38    {
39        osgFX::Scribe* scribe = dynamic_cast<osgFX::Scribe*>(&node);
40        if (scribe)
41        {
42            for(unsigned int i=0; i<scribe->getNumChildren(); ++i)
43            {
44                _group->addChild(scribe->getChild(i));
45            }
46        }
47        else
48        {
49            traverse(node);
50        }
51    }
52   
53    osg::ref_ptr<osg::Group> _group;
54    bool _addToModel;
55};
56
57class DeleteSelectedNodesVisitor : public osg::NodeVisitor
58{
59public:
60
61    DeleteSelectedNodesVisitor():
62        osg::NodeVisitor(osg::NodeVisitor::TRAVERSE_ALL_CHILDREN)       
63    {
64    }
65   
66    virtual void apply(osg::Node& node)
67    {
68        osgFX::Scribe* scribe = dynamic_cast<osgFX::Scribe*>(&node);
69        if (scribe)
70        {
71            _selectedNodes.push_back(scribe);
72        }
73        else
74        {
75            traverse(node);
76        }
77    }
78   
79    void pruneSelectedNodes()
80    {
81        for(SelectedNodes::iterator itr = _selectedNodes.begin();
82            itr != _selectedNodes.end();
83            ++itr)
84        {
85            osg::Node* node = itr->get();
86            osg::Node::ParentList parents = node->getParents();
87            for(osg::Node::ParentList::iterator pitr = parents.begin();
88                pitr != parents.end();
89                ++pitr)
90            {
91                osg::Group* parent = *pitr;
92                parent->removeChild(node);
93            }
94        }
95    }
96   
97    typedef std::vector< osg::ref_ptr<osgFX::Scribe> > SelectedNodes;
98    SelectedNodes _selectedNodes;
99   
100};
101
102// class to handle events with a pick
103class PickHandler : public osgGA::GUIEventHandler
104{
105public:
106
107    PickHandler():
108        _mx(0.0),_my(0.0),
109        _usePolytopeIntersector(false),
110        _useWindowCoordinates(false) {}
111
112    ~PickHandler() {}
113
114    bool handle(const osgGA::GUIEventAdapter& ea, osgGA::GUIActionAdapter& aa)
115    {
116        osgViewer::Viewer* viewer = dynamic_cast<osgViewer::Viewer*>(&aa);
117        if (!viewer) return false;
118
119        switch(ea.getEventType())
120        {
121            case(osgGA::GUIEventAdapter::KEYUP):
122            {
123                if (ea.getKey()=='s')
124                {
125                    saveSelectedModel(viewer->getSceneData());
126                }               
127                else if (ea.getKey()=='o')
128                {
129                    osg::notify(osg::NOTICE)<<"Saved model to file 'saved_model.osg'"<<std::endl;
130                    osgDB::writeNodeFile(*(viewer->getSceneData()), "saved_model.osg");
131                }
132                else if (ea.getKey()=='p')
133                {
134                    _usePolytopeIntersector = !_usePolytopeIntersector;
135                    if (_usePolytopeIntersector)
136                    {
137                        osg::notify(osg::NOTICE)<<"Using PolytopeIntersector"<<std::endl;
138                    } else {
139                        osg::notify(osg::NOTICE)<<"Using LineSegmentIntersector"<<std::endl;
140                    }
141                }
142                else if (ea.getKey()=='c')
143                {
144                    _useWindowCoordinates = !_useWindowCoordinates;
145                    if (_useWindowCoordinates)
146                    {
147                        osg::notify(osg::NOTICE)<<"Using window coordinates for picking"<<std::endl;
148                    } else {
149                        osg::notify(osg::NOTICE)<<"Using projection coordiates for picking"<<std::endl;
150                    }
151                }
152                else if (ea.getKey()==osgGA::GUIEventAdapter::KEY_Delete || ea.getKey()==osgGA::GUIEventAdapter::KEY_BackSpace)
153                {
154                    osg::notify(osg::NOTICE)<<"Delete"<<std::endl;
155                    DeleteSelectedNodesVisitor dsnv;
156                    viewer->getSceneData()->accept(dsnv);
157                    dsnv.pruneSelectedNodes();
158                }
159                return false;
160            }
161            case(osgGA::GUIEventAdapter::PUSH):
162            case(osgGA::GUIEventAdapter::MOVE):
163            {
164                _mx = ea.getX();
165                _my = ea.getY();
166                return false;
167            }
168            case(osgGA::GUIEventAdapter::RELEASE):
169            {
170                if (_mx == ea.getX() && _my == ea.getY())
171                {
172                    // only do a pick if the mouse hasn't moved
173                    pick(ea,viewer);
174                }
175                return true;
176            }   
177
178            default:
179                return false;
180        }
181    }
182
183    void pick(const osgGA::GUIEventAdapter& ea, osgViewer::Viewer* viewer)
184    {
185        osg::Node* scene = viewer->getSceneData();
186        if (!scene) return;
187
188        osg::notify(osg::NOTICE)<<std::endl;
189
190        osg::Node* node = 0;
191        osg::Group* parent = 0;
192
193        if (_usePolytopeIntersector)
194        {
195            osgUtil::PolytopeIntersector* picker;
196            if (_useWindowCoordinates)
197            {
198                // use window coordinates
199                // remap the mouse x,y into viewport coordinates.
200                osg::Viewport* viewport = viewer->getCamera()->getViewport();
201                double mx = viewport->x() + (int)((double )viewport->width()*(ea.getXnormalized()*0.5+0.5));
202                double my = viewport->y() + (int)((double )viewport->height()*(ea.getYnormalized()*0.5+0.5));
203
204                // half width, height.
205                double w = 5.0f;
206                double h = 5.0f;
207                picker = new osgUtil::PolytopeIntersector( osgUtil::Intersector::WINDOW, mx-w, my-h, mx+w, my+h );
208            } else {
209                double mx = ea.getXnormalized();
210                double my = ea.getYnormalized();
211                double w = 0.05;
212                double h = 0.05;
213                picker = new osgUtil::PolytopeIntersector( osgUtil::Intersector::PROJECTION, mx-w, my-h, mx+w, my+h );
214            }
215            osgUtil::IntersectionVisitor iv(picker);
216
217            viewer->getCamera()->accept(iv);
218
219            if (picker->containsIntersections())
220            {
221                osgUtil::PolytopeIntersector::Intersections& intersections = picker->getIntersections();
222
223                for (osgUtil::PolytopeIntersector::Intersections::iterator it=intersections.begin();
224                     it!=intersections.end(); ++it) {
225                    osgUtil::PolytopeIntersector::Intersection intersection=*it;
226
227                    osg::NodePath& nodePath = intersection.nodePath;
228                    node = (nodePath.size()>=1)?nodePath[nodePath.size()-1]:0;
229                    parent = (nodePath.size()>=2)?dynamic_cast<osg::Group*>(nodePath[nodePath.size()-2]):0;
230
231                    if (node) std::cout<<"  Hits "<<node->className()<<" nodePath size"<<nodePath.size()<<std::endl;
232                    toggleScribe(parent, node);
233                }
234
235            }
236
237        }
238        else
239        {
240            osgUtil::LineSegmentIntersector* picker;
241            if (!_useWindowCoordinates)
242            {
243                // use non dimensional coordinates - in projection/clip space
244                picker = new osgUtil::LineSegmentIntersector( osgUtil::Intersector::PROJECTION, ea.getXnormalized(),ea.getYnormalized() );
245            } else {
246                // use window coordinates
247                // remap the mouse x,y into viewport coordinates.
248                osg::Viewport* viewport = viewer->getCamera()->getViewport();
249                float mx = viewport->x() + (int)((float)viewport->width()*(ea.getXnormalized()*0.5f+0.5f));
250                float my = viewport->y() + (int)((float)viewport->height()*(ea.getYnormalized()*0.5f+0.5f));
251                picker = new osgUtil::LineSegmentIntersector( osgUtil::Intersector::WINDOW, mx, my );
252            }
253            osgUtil::IntersectionVisitor iv(picker);
254
255            viewer->getCamera()->accept(iv);
256
257            if (picker->containsIntersections())
258            {
259                osgUtil::LineSegmentIntersector::Intersection intersection = picker->getFirstIntersection();
260                osg::notify(osg::NOTICE)<<"Picked "<<intersection.localIntersectionPoint<<std::endl;
261
262                osg::NodePath& nodePath = intersection.nodePath;
263                node = (nodePath.size()>=1)?nodePath[nodePath.size()-1]:0;
264                parent = (nodePath.size()>=2)?dynamic_cast<osg::Group*>(nodePath[nodePath.size()-2]):0;
265
266                if (node) std::cout<<"  Hits "<<node->className()<<" nodePath size"<<nodePath.size()<<std::endl;
267                toggleScribe(parent, node);
268            }
269        }       
270
271        // now we try to decorate the hit node by the osgFX::Scribe to show that its been "picked"
272    }
273
274    void toggleScribe(osg::Group* parent, osg::Node* node) {
275        if (!parent || !node) return;
276
277        std::cout<<"  parent "<<parent->className()<<std::endl;
278
279        osgFX::Scribe* parentAsScribe = dynamic_cast<osgFX::Scribe*>(parent);
280        if (!parentAsScribe)
281        {
282            // node not already picked, so highlight it with an osgFX::Scribe
283            osgFX::Scribe* scribe = new osgFX::Scribe();
284            scribe->addChild(node);
285            parent->replaceChild(node,scribe);
286        }
287        else
288        {
289            // node already picked so we want to remove scribe to unpick it.
290            osg::Node::ParentList parentList = parentAsScribe->getParents();
291            for(osg::Node::ParentList::iterator itr=parentList.begin();
292                itr!=parentList.end();
293                ++itr)
294            {
295                (*itr)->replaceChild(parentAsScribe,node);
296            }
297        }
298
299    }
300
301    void saveSelectedModel(osg::Node* scene)
302    {
303        if (!scene) return;
304   
305        CreateModelToSaveVisitor cmtsv;
306        scene->accept(cmtsv);
307       
308        if (cmtsv._group->getNumChildren()>0)
309        {
310            std::cout<<"Writing selected compoents to 'selected_model.osg'"<<std::endl;
311            osgDB::writeNodeFile(*cmtsv._group, "selected_model.osg");
312        }
313    }
314
315protected:
316
317    float _mx,_my;
318    bool _usePolytopeIntersector;
319    bool _useWindowCoordinates;
320};
321
322int main( int argc, char **argv )
323{
324    if (argc<2)
325    {
326        std::cout << argv[0] <<": requires filename argument." << std::endl;
327        return 1;
328    }
329
330    // load the scene.
331    osg::ref_ptr<osg::Node> loadedModel = osgDB::readNodeFile(argv[1]);
332    if (!loadedModel)
333    {
334        std::cout << argv[0] <<": No data loaded." << std::endl;
335        return 1;
336    }
337   
338    // create the window to draw to.
339    osg::ref_ptr<osg::GraphicsContext::Traits> traits = new osg::GraphicsContext::Traits;
340    traits->x = 200;
341    traits->y = 200;
342    traits->width = 800;
343    traits->height = 600;
344    traits->windowDecoration = true;
345    traits->doubleBuffer = true;
346    traits->sharedContext = 0;
347
348    osg::ref_ptr<osg::GraphicsContext> gc = osg::GraphicsContext::createGraphicsContext(traits.get());
349    osgViewer::GraphicsWindow* gw = dynamic_cast<osgViewer::GraphicsWindow*>(gc.get());
350    if (!gw)
351    {
352        osg::notify(osg::NOTICE)<<"Error: unable to create graphics window."<<std::endl;
353        return 1;
354    }
355
356    // create the view of the scene.
357    osgViewer::Viewer viewer;
358    viewer.getCamera()->setGraphicsContext(gc.get());
359    viewer.getCamera()->setViewport(0,0,800,600);
360    viewer.setSceneData(loadedModel.get());
361   
362    // create a tracball manipulator to move the camera around in response to keyboard/mouse events
363    viewer.setCameraManipulator( new osgGA::TrackballManipulator );
364
365    osg::ref_ptr<osgGA::StateSetManipulator> statesetManipulator = new osgGA::StateSetManipulator(viewer.getCamera()->getStateSet());
366    viewer.addEventHandler(statesetManipulator.get());
367
368    // add the pick handler
369    viewer.addEventHandler(new PickHandler());
370
371    viewer.realize();
372
373    // main loop (note, window toolkits which take control over the main loop will require a window redraw callback containing the code below.)
374    while(!viewer.done())
375    {
376        viewer.frame();
377    }
378
379    return 0;
380}
Note: See TracBrowser for help on using the browser.