root/OpenSceneGraph/branches/osg-cocoa-dev/examples/osgkeyboardmouse/osgkeyboardmouse.cpp @ 9844

Revision 9844, 18.2 kB (checked in by shuber, 6 years ago)

added an EventDumperEventHandler? dumping all events and its data to the console

  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
Line 
1/* OpenSceneGraph example, osgkeyboardmouse.
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// Simple example of use of osgViewer::GraphicsWindow + Viewer
20// example that provides the user with control over view position with basic picking.
21
22#include <osg/Timer>
23#include <osg/io_utils>
24#include <osg/observer_ptr>
25
26#include <osgUtil/IntersectionVisitor>
27#include <osgUtil/PolytopeIntersector>
28#include <osgUtil/LineSegmentIntersector>
29
30#include <osgDB/ReadFile>
31#include <osgDB/WriteFile>
32
33#include <osgGA/TrackballManipulator>
34#include <osgGA/StateSetManipulator>
35
36#include <osgViewer/Viewer>
37
38#include <osgFX/Scribe>
39
40#include <iostream>
41
42class CreateModelToSaveVisitor : public osg::NodeVisitor
43{
44public:
45
46    CreateModelToSaveVisitor():
47        osg::NodeVisitor(osg::NodeVisitor::TRAVERSE_ALL_CHILDREN)       
48    {
49        _group = new osg::Group;
50        _addToModel = false;
51    }
52   
53    virtual void apply(osg::Node& node)
54    {
55        osgFX::Scribe* scribe = dynamic_cast<osgFX::Scribe*>(&node);
56        if (scribe)
57        {
58            for(unsigned int i=0; i<scribe->getNumChildren(); ++i)
59            {
60                _group->addChild(scribe->getChild(i));
61            }
62        }
63        else
64        {
65            traverse(node);
66        }
67    }
68   
69    osg::ref_ptr<osg::Group> _group;
70    bool _addToModel;
71};
72
73class DeleteSelectedNodesVisitor : public osg::NodeVisitor
74{
75public:
76
77    DeleteSelectedNodesVisitor():
78        osg::NodeVisitor(osg::NodeVisitor::TRAVERSE_ALL_CHILDREN)       
79    {
80    }
81   
82    virtual void apply(osg::Node& node)
83    {
84        osgFX::Scribe* scribe = dynamic_cast<osgFX::Scribe*>(&node);
85        if (scribe)
86        {
87            _selectedNodes.push_back(scribe);
88        }
89        else
90        {
91            traverse(node);
92        }
93    }
94   
95    void pruneSelectedNodes()
96    {
97        for(SelectedNodes::iterator itr = _selectedNodes.begin();
98            itr != _selectedNodes.end();
99            ++itr)
100        {
101            osg::Node* node = itr->get();
102            osg::Node::ParentList parents = node->getParents();
103            for(osg::Node::ParentList::iterator pitr = parents.begin();
104                pitr != parents.end();
105                ++pitr)
106            {
107                osg::Group* parent = *pitr;
108                parent->removeChild(node);
109            }
110        }
111    }
112   
113    typedef std::vector< osg::ref_ptr<osgFX::Scribe> > SelectedNodes;
114    SelectedNodes _selectedNodes;
115   
116};
117
118class EventDumperEventHandler : public osgGA::GUIEventHandler {
119
120public:
121    bool handle(const osgGA::GUIEventAdapter& ea, osgGA::GUIActionAdapter& aa)
122    {
123        if (ea.getEventType() == osgGA::GUIEventAdapter::FRAME)
124            return false;
125       
126        static std::map<osgGA::GUIEventAdapter::EventType, std::string> type_mapping;
127        if (type_mapping.size() == 0)
128        {
129            type_mapping[osgGA::GUIEventAdapter::NONE]                                  = "NONE";
130            type_mapping[osgGA::GUIEventAdapter::PUSH]                                  = "PUSH";
131            type_mapping[osgGA::GUIEventAdapter::RELEASE]                               = "RELEASE";
132            type_mapping[osgGA::GUIEventAdapter::DOUBLECLICK]                   = "DOUBLECLICK";
133            type_mapping[osgGA::GUIEventAdapter::DRAG]                                  = "DRAG";
134            type_mapping[osgGA::GUIEventAdapter::MOVE]                                  = "MOVE";
135            type_mapping[osgGA::GUIEventAdapter::KEYDOWN]                               = "KEYDOWN";
136            type_mapping[osgGA::GUIEventAdapter::KEYUP]                                 = "KEYUP";
137            type_mapping[osgGA::GUIEventAdapter::FRAME]                                 = "FRAME";
138            type_mapping[osgGA::GUIEventAdapter::RESIZE]                                = "RESIZE";
139            type_mapping[osgGA::GUIEventAdapter::SCROLL]                                = "SCROLL";
140            type_mapping[osgGA::GUIEventAdapter::PEN_PRESSURE]                  = "PEN_PRESSURE";
141            type_mapping[osgGA::GUIEventAdapter::PEN_ORIENTATION]               = "PEN_ORIENTATION";
142            type_mapping[osgGA::GUIEventAdapter::PEN_PROXIMITY_ENTER]   = "PEN_PROXIMITY_ENTER";
143            type_mapping[osgGA::GUIEventAdapter::PEN_PROXIMITY_LEAVE]   = "PEN_PROXIMITY_LEAVE";
144            type_mapping[osgGA::GUIEventAdapter::CLOSE_WINDOW]          = "CLOSE_WINDOW";
145            type_mapping[osgGA::GUIEventAdapter::QUIT_APPLICATION]      = "QUIT_APPLICATION";
146            type_mapping[osgGA::GUIEventAdapter::USER]                                  = "USER";
147                }
148               
149                static std::map<osgGA::GUIEventAdapter::TabletPointerType, std::string> pen_mapping;
150                if (pen_mapping.size() == 0) {
151                        pen_mapping[osgGA::GUIEventAdapter::UNKNOWN]    = "UNKNOWN";
152                        pen_mapping[osgGA::GUIEventAdapter::PEN]                = "PEN";
153                        pen_mapping[osgGA::GUIEventAdapter::PUCK]               = "BUCK";
154                        pen_mapping[osgGA::GUIEventAdapter::ERASER]             = "ERASER";     
155                }
156               
157                std::cout << type_mapping[ea.getEventType()] << " with ";
158               
159                switch (ea.getEventType()) {
160                        case osgGA::GUIEventAdapter::PUSH:
161                        case osgGA::GUIEventAdapter::RELEASE:
162                        case osgGA::GUIEventAdapter::DOUBLECLICK:
163                        case osgGA::GUIEventAdapter::DRAG:
164                        case osgGA::GUIEventAdapter::MOVE:
165                                std::cout << ea.getX() << "/" << ea.getY() << " modifiers: " << ea.getModKeyMask() << std::endl;
166                                break;
167                       
168                        case osgGA::GUIEventAdapter::SCROLL:
169                                std::cout << ea.getX() << "/" << ea.getY() << " scroll: ";
170                                switch (ea.getScrollingMotion()) {
171                                        case osgGA::GUIEventAdapter::SCROLL_NONE:
172                                                        std::cout << "NONE";
173                                                        break;
174                                        case osgGA::GUIEventAdapter::SCROLL_LEFT:
175                                                        std::cout << "SCROLL_LEFT";
176                                                        break
177                                        case osgGA::GUIEventAdapter::SCROLL_RIGHT:
178                                                        std::cout << "SCROLL_RIGHT";
179                                                        break;         
180                                        case osgGA::GUIEventAdapter::SCROLL_UP:
181                                                        std::cout << "SCROLL_UP";
182                                                        break;         
183                                        case osgGA::GUIEventAdapter::SCROLL_DOWN:
184                                                        std::cout << "SCROLL_DOWN";
185                                                        break;         
186                                        case osgGA::GUIEventAdapter::SCROLL_2D:
187                                                        std::cout << "SCROLL_2D";
188                                                        break;         
189
190                                }
191                                std::cout << " (" << ea.getScrollingDeltaX() << "/" << ea.getScrollingDeltaY() << ") modifiers: " << ea.getModKeyMask() << std::endl;
192                                break;
193                       
194                        case osgGA::GUIEventAdapter::KEYUP:
195                        case osgGA::GUIEventAdapter::KEYDOWN:
196                                std::cout << ea.getKey() << " modifiers: " << ea.getModKeyMask() << std::endl;
197                                break;
198                               
199                        case osgGA::GUIEventAdapter::RESIZE:
200                                std::cout << pen_mapping[ea.getTabletPointerType()] << ": " << ea.getWindowX() << "/" << ea.getWindowY() << " " << ea.getWindowWidth() << "x" << ea.getWindowHeight() << std::endl;
201                                break;
202                       
203                        case osgGA::GUIEventAdapter::PEN_PRESSURE:
204                                std::cout << pen_mapping[ea.getTabletPointerType()] << ": " << ea.getPenPressure() << " tiltx: " << ea.getPenTiltY() << " tilty: " << ea.getPenTiltY() << std::endl;
205                                break;
206                       
207                        case osgGA::GUIEventAdapter::PEN_ORIENTATION:
208                                std::cout << pen_mapping[ea.getTabletPointerType()] << ": " << ea.getPenOrientation() << std::endl;
209                                break;
210                               
211                        case osgGA::GUIEventAdapter::PEN_PROXIMITY_ENTER:
212                        case osgGA::GUIEventAdapter::PEN_PROXIMITY_LEAVE:
213                                std::cout << pen_mapping[ea.getTabletPointerType()] << std::endl;
214                                break;
215                       
216                        default:
217                                std::cout << std::endl;
218        }
219               
220                return false;
221        }
222};
223
224// class to handle events with a pick
225class PickHandler : public osgGA::GUIEventHandler
226{
227public:
228
229    PickHandler():
230        _mx(0.0),_my(0.0),
231        _usePolytopeIntersector(false),
232        _useWindowCoordinates(false) {}
233
234    ~PickHandler() {}
235
236    bool handle(const osgGA::GUIEventAdapter& ea, osgGA::GUIActionAdapter& aa)
237    {
238        osgViewer::Viewer* viewer = dynamic_cast<osgViewer::Viewer*>(&aa);
239        if (!viewer) return false;
240
241        switch(ea.getEventType())
242        {
243            case(osgGA::GUIEventAdapter::KEYUP):
244            {
245                if (ea.getKey()=='s')
246                {
247                    saveSelectedModel(viewer->getSceneData());
248                }               
249                else if (ea.getKey()=='o')
250                {
251                    osg::notify(osg::NOTICE)<<"Saved model to file 'saved_model.osg'"<<std::endl;
252                    osgDB::writeNodeFile(*(viewer->getSceneData()), "saved_model.osg");
253                }
254                else if (ea.getKey()=='p')
255                {
256                    _usePolytopeIntersector = !_usePolytopeIntersector;
257                    if (_usePolytopeIntersector)
258                    {
259                        osg::notify(osg::NOTICE)<<"Using PolytopeIntersector"<<std::endl;
260                    } else {
261                        osg::notify(osg::NOTICE)<<"Using LineSegmentIntersector"<<std::endl;
262                    }
263                }
264                else if (ea.getKey()=='c')
265                {
266                    _useWindowCoordinates = !_useWindowCoordinates;
267                    if (_useWindowCoordinates)
268                    {
269                        osg::notify(osg::NOTICE)<<"Using window coordinates for picking"<<std::endl;
270                    } else {
271                        osg::notify(osg::NOTICE)<<"Using projection coordiates for picking"<<std::endl;
272                    }
273                }
274                else if (ea.getKey()==osgGA::GUIEventAdapter::KEY_Delete || ea.getKey()==osgGA::GUIEventAdapter::KEY_BackSpace)
275                {
276                    osg::notify(osg::NOTICE)<<"Delete"<<std::endl;
277                    DeleteSelectedNodesVisitor dsnv;
278                    viewer->getSceneData()->accept(dsnv);
279                    dsnv.pruneSelectedNodes();
280                }
281                return false;
282            }
283            case(osgGA::GUIEventAdapter::PUSH):
284            case(osgGA::GUIEventAdapter::MOVE):
285            {
286                _mx = ea.getX();
287                _my = ea.getY();
288                return false;
289            }
290            case(osgGA::GUIEventAdapter::RELEASE):
291            {
292                if (_mx == ea.getX() && _my == ea.getY())
293                {
294                    // only do a pick if the mouse hasn't moved
295                    pick(ea,viewer);
296                }
297                return true;
298            }   
299
300            default:
301                return false;
302        }
303    }
304
305    void pick(const osgGA::GUIEventAdapter& ea, osgViewer::Viewer* viewer)
306    {
307        osg::Node* scene = viewer->getSceneData();
308        if (!scene) return;
309
310        osg::notify(osg::NOTICE)<<std::endl;
311
312        osg::Node* node = 0;
313        osg::Group* parent = 0;
314
315        if (_usePolytopeIntersector)
316        {
317            osgUtil::PolytopeIntersector* picker;
318            if (_useWindowCoordinates)
319            {
320                // use window coordinates
321                // remap the mouse x,y into viewport coordinates.
322                osg::Viewport* viewport = viewer->getCamera()->getViewport();
323                double mx = viewport->x() + (int)((double )viewport->width()*(ea.getXnormalized()*0.5+0.5));
324                double my = viewport->y() + (int)((double )viewport->height()*(ea.getYnormalized()*0.5+0.5));
325
326                // half width, height.
327                double w = 5.0f;
328                double h = 5.0f;
329                picker = new osgUtil::PolytopeIntersector( osgUtil::Intersector::WINDOW, mx-w, my-h, mx+w, my+h );
330            } else {
331                double mx = ea.getXnormalized();
332                double my = ea.getYnormalized();
333                double w = 0.05;
334                double h = 0.05;
335                picker = new osgUtil::PolytopeIntersector( osgUtil::Intersector::PROJECTION, mx-w, my-h, mx+w, my+h );
336            }
337            osgUtil::IntersectionVisitor iv(picker);
338
339            viewer->getCamera()->accept(iv);
340
341            if (picker->containsIntersections())
342            {
343                osgUtil::PolytopeIntersector::Intersection intersection = picker->getFirstIntersection();
344
345                osg::notify(osg::NOTICE)<<"Picked "<<intersection.localIntersectionPoint<<std::endl
346                    <<"  Distance to ref. plane "<<intersection.distance
347                    <<", max. dist "<<intersection.maxDistance
348                    <<", primitive index "<<intersection.primitiveIndex
349                    <<", numIntersectionPoints "
350                    <<intersection.numIntersectionPoints
351                    <<std::endl;
352
353                osg::NodePath& nodePath = intersection.nodePath;
354                node = (nodePath.size()>=1)?nodePath[nodePath.size()-1]:0;
355                parent = (nodePath.size()>=2)?dynamic_cast<osg::Group*>(nodePath[nodePath.size()-2]):0;
356
357                if (node) std::cout<<"  Hits "<<node->className()<<" nodePath size "<<nodePath.size()<<std::endl;   
358                toggleScribe(parent, node);
359            }
360
361        }
362        else
363        {
364            osgUtil::LineSegmentIntersector* picker;
365            if (!_useWindowCoordinates)
366            {
367                // use non dimensional coordinates - in projection/clip space
368                picker = new osgUtil::LineSegmentIntersector( osgUtil::Intersector::PROJECTION, ea.getXnormalized(),ea.getYnormalized() );
369            } else {
370                // use window coordinates
371                // remap the mouse x,y into viewport coordinates.
372                osg::Viewport* viewport = viewer->getCamera()->getViewport();
373                float mx = viewport->x() + (int)((float)viewport->width()*(ea.getXnormalized()*0.5f+0.5f));
374                float my = viewport->y() + (int)((float)viewport->height()*(ea.getYnormalized()*0.5f+0.5f));
375                picker = new osgUtil::LineSegmentIntersector( osgUtil::Intersector::WINDOW, mx, my );
376            }
377            osgUtil::IntersectionVisitor iv(picker);
378
379            viewer->getCamera()->accept(iv);
380
381            if (picker->containsIntersections())
382            {
383                osgUtil::LineSegmentIntersector::Intersection intersection = picker->getFirstIntersection();
384                osg::notify(osg::NOTICE)<<"Picked "<<intersection.localIntersectionPoint<<std::endl;
385
386                osg::NodePath& nodePath = intersection.nodePath;
387                node = (nodePath.size()>=1)?nodePath[nodePath.size()-1]:0;
388                parent = (nodePath.size()>=2)?dynamic_cast<osg::Group*>(nodePath[nodePath.size()-2]):0;
389
390                if (node) std::cout<<"  Hits "<<node->className()<<" nodePath size"<<nodePath.size()<<std::endl;
391                toggleScribe(parent, node);
392            }
393        }       
394
395        // now we try to decorate the hit node by the osgFX::Scribe to show that its been "picked"
396    }
397
398    void toggleScribe(osg::Group* parent, osg::Node* node) {
399        if (!parent || !node) return;
400
401        std::cout<<"  parent "<<parent->className()<<std::endl;
402
403        osgFX::Scribe* parentAsScribe = dynamic_cast<osgFX::Scribe*>(parent);
404        if (!parentAsScribe)
405        {
406            // node not already picked, so highlight it with an osgFX::Scribe
407            osgFX::Scribe* scribe = new osgFX::Scribe();
408            scribe->addChild(node);
409            parent->replaceChild(node,scribe);
410        }
411        else
412        {
413            // node already picked so we want to remove scribe to unpick it.
414            osg::Node::ParentList parentList = parentAsScribe->getParents();
415            for(osg::Node::ParentList::iterator itr=parentList.begin();
416                itr!=parentList.end();
417                ++itr)
418            {
419                (*itr)->replaceChild(parentAsScribe,node);
420            }
421        }
422
423    }
424
425    void saveSelectedModel(osg::Node* scene)
426    {
427        if (!scene) return;
428   
429        CreateModelToSaveVisitor cmtsv;
430        scene->accept(cmtsv);
431       
432        if (cmtsv._group->getNumChildren()>0)
433        {
434            std::cout<<"Writing selected compoents to 'selected_model.osg'"<<std::endl;
435            osgDB::writeNodeFile(*cmtsv._group, "selected_model.osg");
436        }
437    }
438
439protected:
440
441    float _mx,_my;
442    bool _usePolytopeIntersector;
443    bool _useWindowCoordinates;
444};
445
446int main( int argc, char **argv )
447{
448    osg::ref_ptr<osg::Node> loadedModel;
449   
450    // load the scene.
451    if (argc>1) loadedModel = osgDB::readNodeFile(argv[1]);
452   
453    // if not loaded assume no arguments passed in, try use default mode instead.
454    if (!loadedModel) loadedModel = osgDB::readNodeFile("dumptruck.osg");
455   
456    if (!loadedModel)
457    {
458        std::cout << argv[0] <<": No data loaded." << std::endl;
459        return 1;
460    }
461   
462    // create the window to draw to.
463    osg::ref_ptr<osg::GraphicsContext::Traits> traits = new osg::GraphicsContext::Traits;
464    traits->x = 200;
465    traits->y = 200;
466    traits->width = 800;
467    traits->height = 600;
468    traits->windowDecoration = true;
469    traits->doubleBuffer = true;
470    traits->sharedContext = 0;
471
472    osg::ref_ptr<osg::GraphicsContext> gc = osg::GraphicsContext::createGraphicsContext(traits.get());
473    osgViewer::GraphicsWindow* gw = dynamic_cast<osgViewer::GraphicsWindow*>(gc.get());
474    if (!gw)
475    {
476        osg::notify(osg::NOTICE)<<"Error: unable to create graphics window."<<std::endl;
477        return 1;
478    }
479
480    // create the view of the scene.
481    osgViewer::Viewer viewer;
482    viewer.getCamera()->setGraphicsContext(gc.get());
483    viewer.getCamera()->setViewport(0,0,800,600);
484    viewer.setSceneData(loadedModel.get());
485   
486    // create a tracball manipulator to move the camera around in response to keyboard/mouse events
487    viewer.setCameraManipulator( new osgGA::TrackballManipulator );
488   
489    viewer.addEventHandler(new EventDumperEventHandler());
490
491    osg::ref_ptr<osgGA::StateSetManipulator> statesetManipulator = new osgGA::StateSetManipulator(viewer.getCamera()->getStateSet());
492    viewer.addEventHandler(statesetManipulator.get());
493
494    // add the pick handler
495    viewer.addEventHandler(new PickHandler());
496
497    viewer.realize();
498
499    // main loop (note, window toolkits which take control over the main loop will require a window redraw callback containing the code below.)
500    while(!viewer.done())
501    {
502        viewer.frame();
503    }
504
505    return 0;
506}
Note: See TracBrowser for help on using the browser.