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

Revision 6034, 12.3 kB (checked in by robert, 8 years ago)

Added delete selected object using delete and back space key, and added save option

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