root/OpenSceneGraph/trunk/examples/osgimagesequence/osgimagesequence.cpp @ 13085

Revision 13085, 10.9 kB (checked in by robert, 24 hours ago)

Release OpenSceneGraph-3.3.3

  • Property svn:eol-style set to native
Line 
1/* OpenSceneGraph example, osgtexture3D.
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 <osg/Node>
20#include <osg/Geometry>
21#include <osg/Notify>
22#include <osg/Texture1D>
23#include <osg/Texture2D>
24#include <osg/Texture3D>
25#include <osg/TextureRectangle>
26#include <osg/ImageSequence>
27#include <osg/Geode>
28
29#include <osgDB/Registry>
30#include <osgDB/ReadFile>
31#include <osgDB/WriteFile>
32
33#include <osgViewer/Viewer>
34#include <osgViewer/ViewerEventHandlers>
35
36#include <iostream>
37
38
39
40
41//
42// A simple demo demonstrating how to set on an animated texture using an osg::ImageSequence
43//
44
45osg::StateSet* createState(osg::ArgumentParser& arguments)
46{
47    osg::ref_ptr<osg::ImageSequence> imageSequence = new osg::ImageSequence;
48
49    bool preLoad = true;
50       
51    while (arguments.read("--page-and-discard"))
52    {
53        imageSequence->setMode(osg::ImageSequence::PAGE_AND_DISCARD_USED_IMAGES);
54        preLoad = false;
55    }
56   
57    while (arguments.read("--page-and-retain"))
58    {
59        imageSequence->setMode(osg::ImageSequence::PAGE_AND_RETAIN_IMAGES);
60        preLoad = false;
61    }
62   
63    while (arguments.read("--preload"))
64    {
65        imageSequence->setMode(osg::ImageSequence::PRE_LOAD_ALL_IMAGES);
66        preLoad = true;
67    }
68   
69    double length = -1.0;
70    while (arguments.read("--length",length)) {}
71   
72    double fps = 30.0;
73    while (arguments.read("--fps",fps)) {}
74
75    if (arguments.argc()>1)
76    {
77        for(int i=1; i<arguments.argc(); ++i)
78        {
79            if (preLoad)
80            {
81                osg::ref_ptr<osg::Image> image = osgDB::readImageFile(arguments[i]);
82                if (image.valid())
83                {
84                    imageSequence->addImage(image.get());
85                }
86            }
87            else
88            {
89                imageSequence->addImageFile(arguments[i]);
90            }
91        }
92
93       
94        if (length>0.0)
95        {
96            imageSequence->setLength(length);
97        }
98        else
99        {
100            unsigned int maxNum = osg::maximum(imageSequence->getFileNames().size(),
101                                               imageSequence->getImages().size());
102                                               
103            imageSequence->setLength(double(maxNum)*(1.0/fps));
104        }
105    }
106    else
107    {
108        if (length>0.0)
109        {
110            imageSequence->setLength(length);
111        }
112        else
113        {
114            imageSequence->setLength(4.0);
115        }
116        imageSequence->addImage(osgDB::readImageFile("Cubemap_axis/posx.png"));
117        imageSequence->addImage(osgDB::readImageFile("Cubemap_axis/negx.png"));
118        imageSequence->addImage(osgDB::readImageFile("Cubemap_axis/posy.png"));
119        imageSequence->addImage(osgDB::readImageFile("Cubemap_axis/negy.png"));
120        imageSequence->addImage(osgDB::readImageFile("Cubemap_axis/posz.png"));
121        imageSequence->addImage(osgDB::readImageFile("Cubemap_axis/negz.png"));
122    }
123       
124    // start the image sequence playing
125    imageSequence->play();
126
127#if 1
128    osg::Texture2D* texture = new osg::Texture2D;
129    texture->setFilter(osg::Texture::MIN_FILTER,osg::Texture::LINEAR);
130    texture->setFilter(osg::Texture::MAG_FILTER,osg::Texture::LINEAR);
131    texture->setWrap(osg::Texture::WRAP_R,osg::Texture::REPEAT);
132    texture->setResizeNonPowerOfTwoHint(false);
133    texture->setImage(imageSequence.get());
134    //texture->setTextureSize(512,512);
135#else   
136    osg::TextureRectangle* texture = new osg::TextureRectangle;
137    texture->setFilter(osg::Texture::MIN_FILTER,osg::Texture::LINEAR);
138    texture->setFilter(osg::Texture::MAG_FILTER,osg::Texture::LINEAR);
139    texture->setWrap(osg::Texture::WRAP_R,osg::Texture::REPEAT);
140    texture->setImage(imageSequence.get());
141    //texture->setTextureSize(512,512);
142#endif
143
144    // create the StateSet to store the texture data
145    osg::StateSet* stateset = new osg::StateSet;
146
147    stateset->setTextureAttributeAndModes(0,texture,osg::StateAttribute::ON);
148
149    return stateset;
150}
151
152osg::Node* createModel(osg::ArgumentParser& arguments)
153{
154
155    // create the geometry of the model, just a simple 2d quad right now.   
156    osg::Geode* geode = new osg::Geode;
157    geode->addDrawable(osg::createTexturedQuadGeometry(osg::Vec3(0.0f,0.0f,0.0), osg::Vec3(1.0f,0.0f,0.0), osg::Vec3(0.0f,0.0f,1.0f)));
158
159    geode->setStateSet(createState(arguments));
160   
161    return geode;
162
163}
164
165
166osg::ImageStream* s_imageStream = 0;
167class MovieEventHandler : public osgGA::GUIEventHandler
168{
169public:
170
171    MovieEventHandler():_playToggle(true),_trackMouse(false) {}
172   
173    void setMouseTracking(bool track) { _trackMouse = track; }
174    bool getMouseTracking() const { return _trackMouse; }
175   
176    void set(osg::Node* node);
177
178    virtual bool handle(const osgGA::GUIEventAdapter& ea,osgGA::GUIActionAdapter& aa, osg::Object*, osg::NodeVisitor* nv);
179   
180    virtual void getUsage(osg::ApplicationUsage& usage) const;
181
182    typedef std::vector< osg::observer_ptr<osg::ImageStream> > ImageStreamList;
183
184protected:
185
186    virtual ~MovieEventHandler() {}
187
188    class FindImageStreamsVisitor : public osg::NodeVisitor
189    {
190    public:
191        FindImageStreamsVisitor(ImageStreamList& imageStreamList):
192            _imageStreamList(imageStreamList) {}
193           
194        virtual void apply(osg::Geode& geode)
195        {
196            apply(geode.getStateSet());
197
198            for(unsigned int i=0;i<geode.getNumDrawables();++i)
199            {
200                apply(geode.getDrawable(i)->getStateSet());
201            }
202       
203            traverse(geode);
204        }
205
206        virtual void apply(osg::Node& node)
207        {
208            apply(node.getStateSet());
209            traverse(node);
210        }
211       
212        inline void apply(osg::StateSet* stateset)
213        {
214            if (!stateset) return;
215           
216            osg::StateAttribute* attr = stateset->getTextureAttribute(0,osg::StateAttribute::TEXTURE);
217            if (attr)
218            {
219                osg::Texture2D* texture2D = dynamic_cast<osg::Texture2D*>(attr);
220                if (texture2D) apply(dynamic_cast<osg::ImageStream*>(texture2D->getImage()));
221
222                osg::TextureRectangle* textureRec = dynamic_cast<osg::TextureRectangle*>(attr);
223                if (textureRec) apply(dynamic_cast<osg::ImageStream*>(textureRec->getImage()));
224            }
225        }
226       
227        inline void apply(osg::ImageStream* imagestream)
228        {
229            if (imagestream)
230            {
231                _imageStreamList.push_back(imagestream);
232                s_imageStream = imagestream;
233            }
234        }
235       
236        ImageStreamList& _imageStreamList;
237       
238    protected:
239   
240        FindImageStreamsVisitor& operator = (const FindImageStreamsVisitor&) { return *this; }
241    };
242
243
244    bool            _playToggle;
245    bool            _trackMouse;
246    ImageStreamList _imageStreamList;
247   
248};
249
250
251
252void MovieEventHandler::set(osg::Node* node)
253{
254    _imageStreamList.clear();
255    if (node)
256    {
257        FindImageStreamsVisitor fisv(_imageStreamList);
258        node->accept(fisv);
259    }
260}
261
262
263bool MovieEventHandler::handle(const osgGA::GUIEventAdapter& ea,osgGA::GUIActionAdapter& aa, osg::Object*, osg::NodeVisitor* nv)
264{
265    switch(ea.getEventType())
266    {
267        case(osgGA::GUIEventAdapter::KEYDOWN):
268        {
269            if (ea.getKey()=='p')
270            {
271                for(ImageStreamList::iterator itr=_imageStreamList.begin();
272                    itr!=_imageStreamList.end();
273                    ++itr)
274                {
275                    if ((*itr)->getStatus()==osg::ImageStream::PLAYING)
276                    {
277                        // playing, so pause
278                        std::cout<<"Pause"<<std::endl;
279                        (*itr)->pause();
280                    }
281                    else
282                    {
283                        // playing, so pause
284                        std::cout<<"Play"<<std::endl;
285                        (*itr)->play();
286                    }
287                }
288                return true;
289            }
290            else if (ea.getKey()=='r')
291            {
292                for(ImageStreamList::iterator itr=_imageStreamList.begin();
293                    itr!=_imageStreamList.end();
294                    ++itr)
295                {
296                    std::cout<<"Restart"<<std::endl;
297                    (*itr)->rewind();
298                }
299                return true;
300            }
301            else if (ea.getKey()=='L')
302            {
303                for(ImageStreamList::iterator itr=_imageStreamList.begin();
304                    itr!=_imageStreamList.end();
305                    ++itr)
306                {
307                    if ( (*itr)->getLoopingMode() == osg::ImageStream::LOOPING)
308                    {
309                        std::cout<<"Toggle Looping Off"<<std::endl;
310                        (*itr)->setLoopingMode( osg::ImageStream::NO_LOOPING );
311                    }
312                    else
313                    {
314                        std::cout<<"Toggle Looping On"<<std::endl;
315                        (*itr)->setLoopingMode( osg::ImageStream::LOOPING );
316                    }
317                }
318                return true;
319            }
320            return false;
321        }
322
323        default:
324            return false;
325    }
326}
327
328void MovieEventHandler::getUsage(osg::ApplicationUsage& usage) const
329{
330    usage.addKeyboardMouseBinding("p","Play/Pause movie");
331    usage.addKeyboardMouseBinding("r","Restart movie");
332    usage.addKeyboardMouseBinding("l","Toggle looping of movie");
333}
334
335
336
337
338int main(int argc, char **argv)
339{
340    osg::ArgumentParser arguments(&argc,argv);
341
342    // construct the viewer.
343    osgViewer::Viewer viewer(arguments);
344
345    std::string filename;
346    arguments.read("-o",filename);
347
348    // create a model from the images and pass it to the viewer.
349    viewer.setSceneData(createModel(arguments));
350
351    // pass the model to the MovieEventHandler so it can pick out ImageStream's to manipulate.
352    MovieEventHandler* meh = new MovieEventHandler();
353    meh->set( viewer.getSceneData() );
354    viewer.addEventHandler( meh );
355
356    viewer.addEventHandler( new osgViewer::StatsHandler());
357
358    if (!filename.empty())
359    {
360        osgDB::writeNodeFile(*viewer.getSceneData(),filename);
361    }
362
363    return viewer.run();
364}
Note: See TracBrowser for help on using the browser.