Changeset 8332

Timestamp:

05/22/08 19:50:22

Author:

robert

Message:

Added basic glReadPixels code

Files:

• OpenSceneGraph/trunk/examples/osgscreencapture/osgscreencapture.cpp (modified) (4 diffs)

OpenSceneGraph/trunk/examples/osgscreencapture/osgscreencapture.cpp

@@ -11 +11 @@
-
-#include <osgDB/ReadFile>
+#include <osgDB/WriteFile>
+
-#include <osgUtil/Optimizer>
-#include <osg/CoordinateSystemNode>
@@ -29 +31 @@
-
-#include <iostream>
+#include <sstream>
-
-class WindowCaptureCallback : public osg::Camera::DrawCallback
@@ -34 +37 @@
-    public:
-    
+        struct ContextData : public osg::Referenced
+        {
+        
+            ContextData(osg::GraphicsContext* gc, const std::string& name):
+                _gc(gc),
+                _fileName(name),
+                _pixelFormat(GL_RGB),
+                _type(GL_UNSIGNED_BYTE),
+                _width(0),
+                _height(0),
+                _currentImageIndex(0),
+                _currentPboIndex(0)
+            {
+                getSize(gc, _width, _height);
+                
+                std::cout<<"Window size "<<_width<<", "<<_height<<std::endl;
+            
+                // single buffered image
+                _imageBuffer.push_back(new osg::Image);
+                
+                // double buffer PBO.
+                _pboBuffer.push_back(new osg::PixelBufferObject);
+                _pboBuffer.push_back(new osg::PixelBufferObject);
+            }
+            
+            void getSize(osg::GraphicsContext* gc, int& width, int& height)
+            {
+                if (gc->getTraits())
+                {
+                    width = gc->getTraits()->width;
+                    height = gc->getTraits()->height;
+                }
+            }
+            
+            void read()
+            {
+                std::cout<<"Read to "<<_fileName<<" image "<<_currentImageIndex<<" "<<_currentPboIndex<<std::endl;
+                
+                unsigned int nextImageIndex = (_currentImageIndex+1)%_imageBuffer.size();
+                unsigned int nextPboIndex = (_currentPboIndex+1)%_pboBuffer.size();
+
+                int width=0, height=0;
+                getSize(_gc, width, height);
+                if (width!=_width || _height!=height)
+                {
+                    std::cout<<"   Window resized "<<width<<", "<<height<<std::endl;
+                    _width = width;
+                    _height = height;
+                }
+                
+                osg::Image* image = _imageBuffer[_currentImageIndex].get();
+                
+                image->readPixels(0,0,_width,_height,
+                                  _pixelFormat,_type);
+                                   
+                if (!_fileName.empty())
+                {
+                    osgDB::writeImageFile(*image, _fileName);
+                }
+                
+                _currentImageIndex = nextImageIndex;
+                _currentPboIndex = nextPboIndex;
+
+            }
+        
+            typedef std::vector< osg::ref_ptr<osg::Image> >             ImageBuffer;
+            typedef std::vector< osg::ref_ptr<osg::PixelBufferObject> > PBOBuffer;
+        
+            osg::GraphicsContext*   _gc;
+            std::string             _fileName;
+            
+            GLenum                  _pixelFormat;
+            GLenum                  _type;
+            int                     _width;
+            int                     _height;
+            
+            unsigned int            _currentImageIndex;
+            ImageBuffer             _imageBuffer;
+            
+            unsigned int            _currentPboIndex;
+            PBOBuffer               _pboBuffer;
+        };
+    
-        WindowCaptureCallback()
-        {
+        }
+
+        ContextData* createContextData(osg::GraphicsContext* gc) const
+        {
+            std::stringstream filename;
+            filename << "test_"<<_contextDataMap.size()<<".jpg";
+            return new ContextData(gc,filename.str());
+        }
+        
+        ContextData* getContextData(osg::GraphicsContext* gc) const
+        {
+            OpenThreads::ScopedLock<OpenThreads::Mutex> lock(_mutex);
+            osg::ref_ptr<ContextData>& data = _contextDataMap[gc];
+            if (!data) data = createContextData(gc);
+            
+            return data.get();
-        }
-
@@ -43 +145 @@
-            osg::notify(osg::NOTICE)<<"Capture screen image "<<gc<<std::endl;
-            
-
+
+
+
+
+
+
+
+
+
-};
-