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

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

Warning fixes

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