1 // screen-dump.cxx -- dump a copy of the opengl screen buffer to a file
3 // Contributed by Richard Kaszeta <bofh@me.umn.edu>, started October 1999.
5 // This library is free software; you can redistribute it and/or
6 // modify it under the terms of the GNU Library General Public
7 // License as published by the Free Software Foundation; either
8 // version 2 of the License, or (at your option) any later version.
10 // This library is distributed in the hope that it will be useful,
11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 // Library General Public License for more details.
15 // You should have received a copy of the GNU General Public License
16 // along with this program; if not, write to the Free Software
17 // Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
23 # include <simgear_config.h>
34 #include <simgear/compiler.h>
37 #include <osgDB/WriteFile>
39 #include "screen-dump.hxx"
42 #define RGB3 3 // 3 bytes of color info per pixel
43 #define RGBA 4 // 4 bytes of color+alpha info
45 bool sg_glWritePPMFile(const char *filename, GLubyte *buffer, int win_width, int win_height, int mode)
48 unsigned char *ibuffer;
50 int pixelSize = mode==GL_RGBA?4:3;
52 ibuffer = (unsigned char *) malloc(win_width*win_height*RGB3);
54 if ( (fp = fopen(filename, "wb")) == NULL ) {
56 printf("Warning: cannot open %s\n", filename);
60 fprintf(fp, "P6\n# CREATOR: glReadPixel()\n%d %d\n%d\n",
61 win_width, win_height, UCHAR_MAX);
63 for (i = 0; i < win_height; i++)
64 for (j = 0; j < win_width; j++)
65 for (k = 0; k < RGB3; k++)
66 ibuffer[q++] = (unsigned char)
67 *(buffer + (pixelSize*((win_height-1-i)*win_width+j)+k));
68 int written = fwrite(ibuffer, sizeof(unsigned char), RGB3*win_width*win_height, fp);
72 if ( written != RGB3*win_width*win_height )
74 printf("Warning: failed to write %s. File truncated.\n", filename);
78 printf("wrote file '%s' (%d x %d pixels, %d bytes)\n",
79 filename, win_width, win_height, RGB3*win_width*win_height);
84 // dump the screen buffer to a png file, returns true on success
85 bool sg_glDumpWindow(const char *filename, int win_width, int win_height) {
86 osg::ref_ptr<osg::Image> img(new osg::Image);
87 img->readPixels(0,0, win_width, win_height, GL_RGB, GL_UNSIGNED_BYTE);
88 return osgDB::writeImageFile(*img, filename);