root/OpenSceneGraph/trunk/examples/osgmovie/osgmovie.cpp @ 9637

Revision 9637, 18.1 kB (checked in by robert, 5 years ago)

Warning fixes

  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
Line 
1/* OpenSceneGraph example, osgmovie.
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 <osgViewer/Viewer>
20#include <osgViewer/ViewerEventHandlers>
21
22#include <osgDB/ReadFile>
23
24#include <osg/Geode>
25#include <osg/Geometry>
26#include <osg/StateSet>
27#include <osg/Material>
28#include <osg/Texture2D>
29#include <osg/TextureRectangle>
30#include <osg/TextureCubeMap>
31#include <osg/TexMat>
32#include <osg/CullFace>
33#include <osg/ImageStream>
34#include <osg/io_utils>
35
36#include <osgGA/TrackballManipulator>
37#include <osgGA/StateSetManipulator>
38#include <osgGA/EventVisitor>
39
40#include <iostream>
41
42class MovieEventHandler : public osgGA::GUIEventHandler
43{
44public:
45
46    MovieEventHandler():_playToggle(true),_trackMouse(false) {}
47   
48    void setMouseTracking(bool track) { _trackMouse = track; }
49    bool getMouseTracking() const { return _trackMouse; }
50   
51    void set(osg::Node* node);
52
53    virtual bool handle(const osgGA::GUIEventAdapter& ea,osgGA::GUIActionAdapter& aa, osg::Object*, osg::NodeVisitor* nv);
54   
55    virtual void getUsage(osg::ApplicationUsage& usage) const;
56
57    typedef std::vector< osg::observer_ptr<osg::ImageStream> > ImageStreamList;
58
59protected:
60
61    virtual ~MovieEventHandler() {}
62
63    class FindImageStreamsVisitor : public osg::NodeVisitor
64    {
65    public:
66        FindImageStreamsVisitor(ImageStreamList& imageStreamList):
67            _imageStreamList(imageStreamList) {}
68           
69        virtual void apply(osg::Geode& geode)
70        {
71            apply(geode.getStateSet());
72
73            for(unsigned int i=0;i<geode.getNumDrawables();++i)
74            {
75                apply(geode.getDrawable(i)->getStateSet());
76            }
77       
78            traverse(geode);
79        }
80
81        virtual void apply(osg::Node& node)
82        {
83            apply(node.getStateSet());
84            traverse(node);
85        }
86       
87        inline void apply(osg::StateSet* stateset)
88        {
89            if (!stateset) return;
90           
91            osg::StateAttribute* attr = stateset->getTextureAttribute(0,osg::StateAttribute::TEXTURE);
92            if (attr)
93            {
94                osg::Texture2D* texture2D = dynamic_cast<osg::Texture2D*>(attr);
95                if (texture2D) apply(dynamic_cast<osg::ImageStream*>(texture2D->getImage()));
96
97                osg::TextureRectangle* textureRec = dynamic_cast<osg::TextureRectangle*>(attr);
98                if (textureRec) apply(dynamic_cast<osg::ImageStream*>(textureRec->getImage()));
99            }
100        }
101       
102        inline void apply(osg::ImageStream* imagestream)
103        {
104            if (imagestream)
105            {
106                _imageStreamList.push_back(imagestream);
107            }
108        }
109       
110        ImageStreamList& _imageStreamList;
111
112    protected:
113   
114        FindImageStreamsVisitor& operator = (const FindImageStreamsVisitor&) { return *this; }
115
116    };
117
118
119    bool            _playToggle;
120    bool            _trackMouse;
121    ImageStreamList _imageStreamList;
122   
123};
124
125
126
127void MovieEventHandler::set(osg::Node* node)
128{
129    _imageStreamList.clear();
130    if (node)
131    {
132        FindImageStreamsVisitor fisv(_imageStreamList);
133        node->accept(fisv);
134    }
135}
136
137
138bool MovieEventHandler::handle(const osgGA::GUIEventAdapter& ea,osgGA::GUIActionAdapter& aa, osg::Object*, osg::NodeVisitor* nv)
139{
140    switch(ea.getEventType())
141    {
142        case(osgGA::GUIEventAdapter::MOVE):
143        case(osgGA::GUIEventAdapter::PUSH):
144        case(osgGA::GUIEventAdapter::RELEASE):
145        {
146            if (_trackMouse)
147            {
148                osgViewer::View* view = dynamic_cast<osgViewer::View*>(&aa);
149                osgUtil::LineSegmentIntersector::Intersections intersections;
150                bool foundIntersection = view==0 ? false :
151                    (nv==0 ? view->computeIntersections(ea.getX(), ea.getY(), intersections) :
152                             view->computeIntersections(ea.getX(), ea.getY(), nv->getNodePath(), intersections));
153               
154                if (foundIntersection)
155                {
156
157                    // use the nearest intersection                 
158                    const osgUtil::LineSegmentIntersector::Intersection& intersection = *(intersections.begin());
159                    osg::Drawable* drawable = intersection.drawable.get();
160                    osg::Geometry* geometry = drawable ? drawable->asGeometry() : 0;
161                    osg::Vec3Array* vertices = geometry ? dynamic_cast<osg::Vec3Array*>(geometry->getVertexArray()) : 0;
162                    if (vertices)
163                    {
164                        // get the vertex indices.
165                        const osgUtil::LineSegmentIntersector::Intersection::IndexList& indices = intersection.indexList;
166                        const osgUtil::LineSegmentIntersector::Intersection::RatioList& ratios = intersection.ratioList;
167
168                        if (indices.size()==3 && ratios.size()==3)
169                        {
170                            unsigned int i1 = indices[0];
171                            unsigned int i2 = indices[1];
172                            unsigned int i3 = indices[2];
173
174                            float r1 = ratios[0];
175                            float r2 = ratios[1];
176                            float r3 = ratios[2];
177
178                            osg::Array* texcoords = (geometry->getNumTexCoordArrays()>0) ? geometry->getTexCoordArray(0) : 0;
179                            osg::Vec2Array* texcoords_Vec2Array = dynamic_cast<osg::Vec2Array*>(texcoords);
180                            if (texcoords_Vec2Array)
181                            {
182                                // we have tex coord array so now we can compute the final tex coord at the point of intersection.                               
183                                osg::Vec2 tc1 = (*texcoords_Vec2Array)[i1];
184                                osg::Vec2 tc2 = (*texcoords_Vec2Array)[i2];
185                                osg::Vec2 tc3 = (*texcoords_Vec2Array)[i3];
186                                osg::Vec2 tc = tc1*r1 + tc2*r2 + tc3*r3;
187
188                                osg::notify(osg::NOTICE)<<"We hit tex coords "<<tc<<std::endl;
189
190                            }
191                        }
192                        else
193                        {
194                            osg::notify(osg::NOTICE)<<"Intersection has insufficient indices to work with";
195                        }
196
197                    }
198                }
199                else
200                {
201                    osg::notify(osg::NOTICE)<<"No intersection"<<std::endl;
202                }
203            }
204            break;
205        }
206        case(osgGA::GUIEventAdapter::KEYDOWN):
207        {
208            if (ea.getKey()=='p')
209            {
210                for(ImageStreamList::iterator itr=_imageStreamList.begin();
211                    itr!=_imageStreamList.end();
212                    ++itr)
213                {
214                    _playToggle = !_playToggle;
215                    if ( _playToggle )
216                    {
217                        // playing, so pause
218                        std::cout<<"Play"<<std::endl;
219                        (*itr)->play();
220                    }
221                    else
222                    {
223                        // playing, so pause
224                        std::cout<<"Pause"<<std::endl;
225                        (*itr)->pause();
226                    }
227                }
228                return true;
229            }
230            else if (ea.getKey()=='r')
231            {
232                for(ImageStreamList::iterator itr=_imageStreamList.begin();
233                    itr!=_imageStreamList.end();
234                    ++itr)
235                {
236                    std::cout<<"Restart"<<std::endl;
237                    (*itr)->rewind();
238                    (*itr)->play();
239                }
240                return true;
241            }
242            else if (ea.getKey()=='L')
243            {
244                for(ImageStreamList::iterator itr=_imageStreamList.begin();
245                    itr!=_imageStreamList.end();
246                    ++itr)
247                {
248                    if ( (*itr)->getLoopingMode() == osg::ImageStream::LOOPING)
249                    {
250                        std::cout<<"Toggle Looping Off"<<std::endl;
251                        (*itr)->setLoopingMode( osg::ImageStream::NO_LOOPING );
252                    }
253                    else
254                    {
255                        std::cout<<"Toggle Looping On"<<std::endl;
256                        (*itr)->setLoopingMode( osg::ImageStream::LOOPING );
257                    }
258                }
259                return true;
260            }
261            return false;
262        }
263
264        default:
265            return false;
266    }
267    return false;
268}
269
270void MovieEventHandler::getUsage(osg::ApplicationUsage& usage) const
271{
272    usage.addKeyboardMouseBinding("p","Play/Pause movie");
273    usage.addKeyboardMouseBinding("r","Restart movie");
274    usage.addKeyboardMouseBinding("l","Toggle looping of movie");
275}
276
277
278osg::Geometry* myCreateTexturedQuadGeometry(const osg::Vec3& pos,float width,float height, osg::Image* image, bool useTextureRectangle, bool xyPlane, bool option_flip)
279{
280    bool flip = image->getOrigin()==osg::Image::TOP_LEFT;
281    if (option_flip) flip = !flip;
282   
283    if (useTextureRectangle)
284    {
285        osg::Geometry* pictureQuad = osg::createTexturedQuadGeometry(pos,
286                                           osg::Vec3(width,0.0f,0.0f),
287                                           xyPlane ? osg::Vec3(0.0f,height,0.0f) : osg::Vec3(0.0f,0.0f,height),
288                                           0.0f, flip ? image->t() : 0.0, image->s(), flip ? 0.0 : image->t());
289
290        osg::TextureRectangle* texture = new osg::TextureRectangle(image);
291        texture->setWrap(osg::Texture::WRAP_S, osg::Texture::CLAMP_TO_EDGE);
292        texture->setWrap(osg::Texture::WRAP_T, osg::Texture::CLAMP_TO_EDGE);
293       
294       
295        pictureQuad->getOrCreateStateSet()->setTextureAttributeAndModes(0,
296                                                                        texture,
297                                                                        osg::StateAttribute::ON);
298       
299        return pictureQuad;
300    }
301    else
302    {
303        osg::Geometry* pictureQuad = osg::createTexturedQuadGeometry(pos,
304                                           osg::Vec3(width,0.0f,0.0f),
305                                           xyPlane ? osg::Vec3(0.0f,height,0.0f) : osg::Vec3(0.0f,0.0f,height),
306                                           0.0f, flip ? 1.0f : 0.0f , 1.0f, flip ? 0.0f : 1.0f);
307                                   
308        osg::Texture2D* texture = new osg::Texture2D(image);
309        texture->setResizeNonPowerOfTwoHint(false);
310        texture->setFilter(osg::Texture::MIN_FILTER,osg::Texture::LINEAR);
311        texture->setWrap(osg::Texture::WRAP_S, osg::Texture::CLAMP_TO_EDGE);
312        texture->setWrap(osg::Texture::WRAP_T, osg::Texture::CLAMP_TO_EDGE);
313       
314       
315        pictureQuad->getOrCreateStateSet()->setTextureAttributeAndModes(0,
316                    texture,
317                    osg::StateAttribute::ON);
318
319        return pictureQuad;
320    }
321}
322
323int main(int argc, char** argv)
324{
325    // use an ArgumentParser object to manage the program arguments.
326    osg::ArgumentParser arguments(&argc,argv);
327   
328    // set up the usage document, in case we need to print out how to use this program.
329    arguments.getApplicationUsage()->setApplicationName(arguments.getApplicationName());
330    arguments.getApplicationUsage()->setDescription(arguments.getApplicationName()+" example demonstrates the use of ImageStream for rendering movies as textures.");
331    arguments.getApplicationUsage()->setCommandLineUsage(arguments.getApplicationName()+" [options] filename ...");
332    arguments.getApplicationUsage()->addCommandLineOption("-h or --help","Display this information");
333    arguments.getApplicationUsage()->addCommandLineOption("--texture2D","Use Texture2D rather than TextureRectangle.");
334    arguments.getApplicationUsage()->addCommandLineOption("--shader","Use shaders to post process the video.");
335    arguments.getApplicationUsage()->addCommandLineOption("--interactive","Use camera manipulator to allow movement around movie.");
336    arguments.getApplicationUsage()->addCommandLineOption("--flip","Flip the movie so top becomes bottom.");
337    arguments.getApplicationUsage()->addCommandLineOption("--devices","Print the Video input capability via QuickTime and exit.");
338   
339    bool useTextureRectangle = true;
340    bool useShader = false;
341
342    // construct the viewer.
343    osgViewer::Viewer viewer(arguments);
344   
345    if (arguments.argc()<=1)
346    {
347        arguments.getApplicationUsage()->write(std::cout,osg::ApplicationUsage::COMMAND_LINE_OPTION);
348        return 1;
349    }
350
351    // if user requests devices video capability.
352    if (arguments.read("-devices") || arguments.read("--devices"))
353    {
354        // Force load QuickTime plugin, probe video capability, exit
355        osgDB::readImageFile("devices.live");
356        return 1;
357    }
358
359    while (arguments.read("--texture2D")) useTextureRectangle=false;
360    while (arguments.read("--shader")) useShader=true;
361
362    bool mouseTracking = false;
363    while (arguments.read("--mouse")) mouseTracking=true;
364
365
366    // if user request help write it out to cout.
367    if (arguments.read("-h") || arguments.read("--help"))
368    {
369        arguments.getApplicationUsage()->write(std::cout);
370        return 1;
371    }
372
373    bool fullscreen = !arguments.read("--interactive");
374    bool flip = arguments.read("--flip");
375
376    osg::ref_ptr<osg::Geode> geode = new osg::Geode;
377
378    osg::StateSet* stateset = geode->getOrCreateStateSet();
379    stateset->setMode(GL_LIGHTING,osg::StateAttribute::OFF);
380
381    if (useShader)
382    {
383        //useTextureRectangle = false;
384
385        static const char *shaderSourceTextureRec = {
386            "uniform vec4 cutoff_color;\n"
387            "uniform samplerRect movie_texture;\n"
388            "void main(void)\n"
389            "{\n"
390            "    vec4 texture_color = textureRect(movie_texture, gl_TexCoord[0].st); \n"
391            "    if (all(lessThanEqual(texture_color,cutoff_color))) discard; \n"
392            "    gl_FragColor = texture_color;\n"
393            "}\n"
394        };
395
396        static const char *shaderSourceTexture2D = {
397            "uniform vec4 cutoff_color;\n"
398            "uniform sampler2D movie_texture;\n"
399            "void main(void)\n"
400            "{\n"
401            "    vec4 texture_color = texture2D(movie_texture, gl_TexCoord[0].st); \n"
402            "    if (all(lessThanEqual(texture_color,cutoff_color))) discard; \n"
403            "    gl_FragColor = texture_color;\n"
404            "}\n"
405        };
406
407        osg::Program* program = new osg::Program;
408
409        program->addShader(new osg::Shader(osg::Shader::FRAGMENT,
410                                           useTextureRectangle ? shaderSourceTextureRec : shaderSourceTexture2D));
411
412        stateset->addUniform(new osg::Uniform("cutoff_color",osg::Vec4(0.1f,0.1f,0.1f,1.0f)));
413        stateset->addUniform(new osg::Uniform("movie_texture",0));
414
415        stateset->setAttribute(program);
416
417    }
418
419    osg::Vec3 pos(0.0f,0.0f,0.0f);
420    osg::Vec3 topleft = pos;
421    osg::Vec3 bottomright = pos;
422   
423    bool xyPlane = fullscreen;
424   
425    for(int i=1;i<arguments.argc();++i)
426    {
427        if (arguments.isString(i))
428        {
429            osg::Image* image = osgDB::readImageFile(arguments[i]);
430            osg::ImageStream* imagestream = dynamic_cast<osg::ImageStream*>(image);
431            if (imagestream) imagestream->play();
432
433            if (image)
434            {
435                osg::notify(osg::NOTICE)<<"image->s()"<<image->s()<<" image-t()="<<image->t()<<std::endl;
436           
437                geode->addDrawable(myCreateTexturedQuadGeometry(pos,image->s(),image->t(),image, useTextureRectangle, xyPlane, flip));
438       
439                bottomright = pos + osg::Vec3(static_cast<float>(image->s()),static_cast<float>(image->t()),0.0f);
440
441                if (xyPlane) pos.y() += image->t()*1.05f;
442                else pos.z() += image->t()*1.05f;
443            }
444            else
445            {
446                std::cout<<"Unable to read file "<<arguments[i]<<std::endl;
447            }           
448        }
449    }
450   
451    // set the scene to render
452    viewer.setSceneData(geode.get());
453
454    if (viewer.getSceneData()==0)
455    {
456        arguments.getApplicationUsage()->write(std::cout);
457        return 1;
458    }
459
460    // pass the model to the MovieEventHandler so it can pick out ImageStream's to manipulate.
461    MovieEventHandler* meh = new MovieEventHandler();
462    meh->setMouseTracking( mouseTracking );
463    meh->set( viewer.getSceneData() );
464    viewer.addEventHandler( meh );
465
466    viewer.addEventHandler( new osgViewer::StatsHandler );
467    viewer.addEventHandler( new osgGA::StateSetManipulator( viewer.getCamera()->getOrCreateStateSet() ) );
468    viewer.addEventHandler( new osgViewer::WindowSizeHandler );
469
470    // add the record camera path handler
471    viewer.addEventHandler(new osgViewer::RecordCameraPathHandler);
472
473    // report any errors if they have occurred when parsing the program arguments.
474    if (arguments.errors())
475    {
476        arguments.writeErrorMessages(std::cout);
477        return 1;
478    }
479
480    if (fullscreen)
481    {
482        viewer.realize();
483
484        viewer.getCamera()->setViewMatrix(osg::Matrix::identity());
485        viewer.getCamera()->setProjectionMatrixAsOrtho2D(topleft.x(),bottomright.x(),topleft.y(),bottomright.y());
486
487        while(!viewer.done())
488        {
489            viewer.frame();
490        }
491        return 0;
492    }
493    else
494    {
495        // create the windows and run the threads.
496        return viewer.run();
497    }
498}
Note: See TracBrowser for help on using the browser.