Changeset 8333

Timestamp:

05/23/08 18:26:03

Author:

robert

Message:

Added single buffered and double buffered PBO support, and --no-pbo,
--single-pbo and --double-pbo command line parameters

Files:

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

OpenSceneGraph/trunk/examples/osgscreencapture/osgscreencapture.cpp

@@ -37 +37 @@
-    public:
-    
+        enum Mode
+        {
+            READ_PIXELS,
+            SINGLE_PBO,
+            DOUBLE_PBO
+        };
+    
-        struct ContextData : public osg::Referenced
-        {
-        
-
+Mode mode, 
-                _gc(gc),
+                _mode(mode),
-                _fileName(name),
-                _pixelFormat(GL_RGB),
@@ -58 +66 @@
-                
-                // double buffer PBO.
-
-
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
-            }
-            
@@ -73 +96 @@
-            void read()
-            {
-
-
-
-
-
-
-
-
+
+
+
-                {
-
-
-
+
+
+
+
+
+
+
+
-                }
-
-
-
-
-
-
-
+
-                {
-osgDB::writeImageFile(*image, _fileName
+readPixels(
-                }
-                
-                _currentImageIndex = nextImageIndex;
-                _currentPboIndex = nextPboIndex;
-
-            }
+            
+            void readPixels();
+
+            void singlePBO(osg::BufferObject::Extensions* ext);
+
+            void multiPBO(osg::BufferObject::Extensions* ext);
-        
-            typedef std::vector< osg::ref_ptr<osg::Image> >             ImageBuffer;
-osg::ref_ptr<osg::PixelBufferObject>
+GLuint
-        
-            osg::GraphicsContext*   _gc;
+            Mode                    _mode;
-            std::string             _fileName;
-            
@@ -120 +140 @@
-        };
-    
-
+
+
-        {
-        }
@@ -128 +149 @@
-            std::stringstream filename;
-            filename << "test_"<<_contextDataMap.size()<<".jpg";
-
+ _mode, 
-        }
-        
@@ -143 +164 @@
-        {
-            osg::GraphicsContext* gc = renderInfo.getState()->getGraphicsContext();
-            osg::notify(osg::NOTICE)<<"Capture screen image "<<gc<<std::endl;
-            
-            osg::ref_ptr<ContextData> cd = getContextData(gc);
-            cd->read();
@@ -150 +169 @@
-        
-        typedef std::map<osg::GraphicsContext*, osg::ref_ptr<ContextData> > ContextDataMap;
-
+
+
-        mutable OpenThreads::Mutex  _mutex;
-        mutable ContextDataMap      _contextDataMap;
-        
-};
+
+void WindowCaptureCallback::ContextData::readPixels()
+{
+    // std::cout<<"readPixels("<<_fileName<<" image "<<_currentImageIndex<<" "<<_currentPboIndex<<std::endl;
+
+    unsigned int nextImageIndex = (_currentImageIndex+1)%_imageBuffer.size();
+    unsigned int nextPboIndex = _pboBuffer.empty() ? 0 : (_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;
+}
+
+void WindowCaptureCallback::ContextData::singlePBO(osg::BufferObject::Extensions* ext)
+{
+    // std::cout<<"singelPBO(  "<<_fileName<<" image "<<_currentImageIndex<<" "<<_currentPboIndex<<std::endl;
+
+    unsigned int nextImageIndex = (_currentImageIndex+1)%_imageBuffer.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;
+    }
+
+    GLuint& pbo = _pboBuffer[0];
+    
+    osg::Image* image = _imageBuffer[_currentImageIndex].get();
+    if (image->s() != _width || 
+        image->t() != _height)
+    {
+        osg::notify(osg::NOTICE)<<"Allocating image "<<std::endl;
+        image->allocateImage(_width, _height, 1, _pixelFormat, _type);
+        
+        if (pbo!=0)
+        {
+            osg::notify(osg::NOTICE)<<"deleting pbo "<<pbo<<std::endl;
+            ext->glDeleteBuffers (1, &pbo);
+            pbo = 0;
+        }
+    }
+    
+    
+    if (pbo==0)
+    {
+        ext->glGenBuffers(1, &pbo);
+        ext->glBindBuffer(GL_PIXEL_PACK_BUFFER_ARB, pbo);
+        ext->glBufferData(GL_PIXEL_PACK_BUFFER_ARB, image->getTotalSizeInBytes(), 0, GL_STREAM_READ);
+
+        osg::notify(osg::NOTICE)<<"Generating pbo "<<pbo<<std::endl;
+    }
+    else
+    {
+        ext->glBindBuffer(GL_PIXEL_PACK_BUFFER_ARB, pbo);
+    }
+
+    glReadPixels(0, 0, _width, _height, _pixelFormat, _type, 0);
+
+    GLubyte* src = (GLubyte*)ext->glMapBuffer(GL_PIXEL_PACK_BUFFER_ARB,
+                                              GL_READ_ONLY_ARB);
+
+    if(src)
+    {
+        memcpy(image->data(), src, image->getTotalSizeInBytes());
+
+        ext->glUnmapBuffer(GL_PIXEL_PACK_BUFFER_ARB);
+    }
+
+    if (!_fileName.empty())
+    {
+        // osgDB::writeImageFile(*image, _fileName);
+    }
+
+    ext->glBindBuffer(GL_PIXEL_PACK_BUFFER_ARB, 0);
+
+    _currentImageIndex = nextImageIndex;
+}
+
+void WindowCaptureCallback::ContextData::multiPBO(osg::BufferObject::Extensions* ext)
+{
+    // std::cout<<"multiPBO(  "<<_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;
+    }
+
+    GLuint& copy_pbo = _pboBuffer[_currentPboIndex];
+    GLuint& read_pbo = _pboBuffer[nextPboIndex];
+    
+    osg::Image* image = _imageBuffer[_currentImageIndex].get();
+    if (image->s() != _width || 
+        image->t() != _height)
+    {
+        osg::notify(osg::NOTICE)<<"Allocating image "<<std::endl;
+        image->allocateImage(_width, _height, 1, _pixelFormat, _type);
+        
+        if (read_pbo!=0)
+        {
+            osg::notify(osg::NOTICE)<<"deleting pbo "<<read_pbo<<std::endl;
+            ext->glDeleteBuffers (1, &read_pbo);
+            read_pbo = 0;
+        }
+
+        if (copy_pbo!=0)
+        {
+            osg::notify(osg::NOTICE)<<"deleting pbo "<<copy_pbo<<std::endl;
+            ext->glDeleteBuffers (1, &copy_pbo);
+            copy_pbo = 0;
+        }
+    }
+    
+    
+    if (read_pbo==0)
+    {
+        ext->glGenBuffers(1, &read_pbo);
+        ext->glBindBuffer(GL_PIXEL_PACK_BUFFER_ARB, read_pbo);
+        ext->glBufferData(GL_PIXEL_PACK_BUFFER_ARB, image->getTotalSizeInBytes(), 0, GL_STREAM_READ);
+
+        osg::notify(osg::NOTICE)<<"Generating pbo "<<read_pbo<<std::endl;
+    }
+    else
+    {
+        ext->glBindBuffer(GL_PIXEL_PACK_BUFFER_ARB, read_pbo);
+    }
+
+    glReadPixels(0, 0, _width, _height, _pixelFormat, _type, 0);
+
+
+    if (copy_pbo!=0)
+    {
+
+        ext->glBindBuffer(GL_PIXEL_PACK_BUFFER_ARB, copy_pbo);
+
+        GLubyte* src = (GLubyte*)ext->glMapBuffer(GL_PIXEL_PACK_BUFFER_ARB,
+                                                  GL_READ_ONLY_ARB);
+
+        if(src)
+        {
+            memcpy(image->data(), src, image->getTotalSizeInBytes());
+
+            ext->glUnmapBuffer(GL_PIXEL_PACK_BUFFER_ARB);
+        }
+
+        if (!_fileName.empty())
+        {
+            // osgDB::writeImageFile(*image, _fileName);
+        }
+    }
+    
+    ext->glBindBuffer(GL_PIXEL_PACK_BUFFER_ARB, 0);
+
+    _currentImageIndex = nextImageIndex;
+    _currentPboIndex = nextPboIndex;
+}
-
-void addCallbackToViewer(osgViewer::ViewerBase& viewer, WindowCaptureCallback* callback)
@@ -282 +484 @@
-    viewer.addEventHandler(new osgViewer::LODScaleHandler);
-
+    WindowCaptureCallback::Mode mode = WindowCaptureCallback::DOUBLE_PBO;
+    while (arguments.read("--no-pbo")) mode = WindowCaptureCallback::READ_PIXELS;
+    while (arguments.read("--single-pbo")) mode = WindowCaptureCallback::SINGLE_PBO;
+    while (arguments.read("--double-pbo")) mode = WindowCaptureCallback::DOUBLE_PBO;
+        
-    // load the data
-    osg::ref_ptr<osg::Node> loadedModel = osgDB::readNodeFiles(arguments);
@@ -309 +516 @@
-    viewer.realize();
-    
-
+(mode)
-
-    return viewer.run();