Annotation of hatari/src/screenSnapShot.c, revision 1.1.1.17

1.1       root        1: /*
1.1.1.5   root        2:   Hatari - screenSnapShot.c
1.1       root        3: 
1.1.1.14  root        4:   This file is distributed under the GNU General Public License, version 2
                      5:   or at your option any later version. Read the file gpl.txt for details.
1.1.1.5   root        6: 
                      7:   Screen Snapshots.
1.1       root        8: */
1.1.1.11  root        9: const char ScreenSnapShot_fileid[] = "Hatari screenSnapShot.c : " __DATE__ " " __TIME__;
1.1       root       10: 
1.1.1.2   root       11: #include <SDL.h>
                     12: #include <dirent.h>
                     13: #include <string.h>
1.1       root       14: #include "main.h"
1.1.1.13  root       15: #include "configuration.h"
1.1.1.6   root       16: #include "log.h"
1.1.1.8   root       17: #include "paths.h"
1.1       root       18: #include "screen.h"
                     19: #include "screenSnapShot.h"
1.1.1.13  root       20: #include "statusbar.h"
1.1       root       21: #include "video.h"
1.1.1.9   root       22: /* after above that bring in config.h */
                     23: #if HAVE_LIBPNG
                     24: # include <png.h>
                     25: # include <assert.h>
1.1.1.12  root       26: # include "pixel_convert.h"                            /* inline functions */
1.1.1.9   root       27: #endif
1.1       root       28: 
1.1.1.2   root       29: 
1.1.1.6   root       30: static int nScreenShots = 0;                /* Number of screen shots saved */
                     31: 
1.1       root       32: 
1.1.1.2   root       33: /*-----------------------------------------------------------------------*/
1.1.1.8   root       34: /**
                     35:  * Scan working directory to get the screenshot number
                     36:  */
1.1.1.5   root       37: static void ScreenSnapShot_GetNum(void)
                     38: {
1.1.1.8   root       39:        char dummy[5];
                     40:        int i, num;
                     41:        DIR *workingdir = opendir(Paths_GetWorkingDir());
                     42:        struct dirent *file;
                     43: 
                     44:        nScreenShots = 0;
                     45:        if (workingdir == NULL)  return;
                     46: 
                     47:        file = readdir(workingdir);
                     48:        while (file != NULL)
                     49:        {
                     50:                if ( strncmp("grab", file->d_name, 4) == 0 )
                     51:                {
                     52:                        /* copy next 4 numbers */
                     53:                        for (i = 0; i < 4; i++)
                     54:                        {
                     55:                                if (file->d_name[4+i] >= '0' && file->d_name[4+i] <= '9')
                     56:                                        dummy[i] = file->d_name[4+i];
                     57:                                else
                     58:                                        break;
                     59:                        }
                     60: 
                     61:                        dummy[i] = '\0'; /* null terminate */
                     62:                        num = atoi(dummy);
                     63:                        if (num > nScreenShots)  nScreenShots = num;
                     64:                }
                     65:                /* next file.. */
                     66:                file = readdir(workingdir);
                     67:        }
                     68: 
                     69:        closedir(workingdir);
1.1.1.2   root       70: }
                     71: 
1.1.1.8   root       72: 
1.1.1.9   root       73: #if HAVE_LIBPNG
1.1.1.8   root       74: /**
1.1.1.12  root       75:  * Save given SDL surface as PNG. Return png file size > 0 for success.
1.1.1.9   root       76:  */
1.1.1.12  root       77: static int ScreenSnapShot_SavePNG(SDL_Surface *surface, const char *filename)
1.1.1.9   root       78: {
1.1.1.12  root       79:        FILE *fp = NULL;
1.1.1.13  root       80:        int ret, bottom;
1.1.1.12  root       81:   
                     82:        fp = fopen(filename, "wb");
                     83:        if (!fp)
                     84:                return -1;
1.1.1.9   root       85: 
1.1.1.13  root       86:        if (ConfigureParams.Screen.bCrop)
                     87:                bottom = Statusbar_GetHeight();
                     88:        else
                     89:                bottom = 0;
                     90: 
                     91:        /* default compression/filter and configured cropping */
1.1.1.17! root       92:        ret = ScreenSnapShot_SavePNG_ToFile(surface, 0, 0, fp, -1, -1, 0, 0, 0, bottom);
1.1.1.9   root       93: 
1.1.1.12  root       94:        fclose (fp);
                     95:        return ret;                                     /* >0 if OK, -1 if error */
1.1.1.9   root       96: }
                     97: 
1.1.1.12  root       98: 
1.1.1.9   root       99: /**
1.1.1.12  root      100:  * Save given SDL surface as PNG in an already opened FILE, eventually cropping some borders.
                    101:  * Return png file size > 0 for success.
                    102:  * This function is also used by avi_record.c to save individual frames as png images.
1.1.1.9   root      103:  */
1.1.1.17! root      104: int ScreenSnapShot_SavePNG_ToFile(SDL_Surface *surface, int dw, int dh,
        !           105:                FILE *fp, int png_compression_level, int png_filter,
1.1.1.12  root      106:                int CropLeft , int CropRight , int CropTop , int CropBottom )
1.1.1.9   root      107: {
1.1.1.12  root      108:        bool do_lock;
1.1.1.16  root      109:        int y, ret;
1.1.1.17! root      110:        int sw = surface->w - CropLeft - CropRight;
        !           111:        int sh = surface->h - CropTop - CropBottom;
1.1.1.13  root      112:        Uint8 *src_ptr;
1.1.1.17! root      113:        Uint8 *rowbuf;
1.1.1.9   root      114:        SDL_PixelFormat *fmt = surface->format;
                    115:        png_infop info_ptr = NULL;
                    116:        png_structp png_ptr;
                    117:        png_text pngtext;
                    118:        char key[] = "Title";
                    119:        char text[] = "Hatari screenshot";
1.1.1.12  root      120:        long start;
1.1.1.17! root      121: 
        !           122:        if (!dw)
        !           123:                dw = sw;
        !           124:        if (!dh)
        !           125:                dh = sh;
        !           126: 
        !           127:        rowbuf = alloca(3 * dw);
        !           128: 
1.1.1.9   root      129:        /* Create and initialize the png_struct with error handler functions. */
                    130:        png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
                    131:        if (!png_ptr) 
                    132:        {
1.1.1.16  root      133:                return -1;
1.1.1.9   root      134:        }
                    135:        
                    136:        /* Allocate/initialize the image information data. */
                    137:        info_ptr = png_create_info_struct(png_ptr);
1.1.1.16  root      138:        if (!info_ptr) {
                    139:                ret = -1;
1.1.1.9   root      140:                goto png_cleanup;
1.1.1.16  root      141:        }
1.1.1.9   root      142: 
                    143:        /* libpng ugliness: Set error handling when not supplying own
                    144:         * error handling functions in the png_create_write_struct() call.
                    145:         */
1.1.1.16  root      146:        if (setjmp(png_jmpbuf(png_ptr))) {
                    147:                ret = -1;
1.1.1.9   root      148:                goto png_cleanup;
1.1.1.16  root      149:        }
1.1.1.9   root      150: 
1.1.1.12  root      151:        /* store current pos in fp (could be != 0 for avi recording) */
                    152:        start = ftell ( fp );
1.1.1.9   root      153: 
                    154:        /* initialize the png structure */
                    155:        png_init_io(png_ptr, fp);
                    156: 
                    157:        /* image data properties */
1.1.1.17! root      158:        png_set_IHDR(png_ptr, info_ptr, dw, dh, 8, PNG_COLOR_TYPE_RGB,
1.1.1.9   root      159:                     PNG_INTERLACE_NONE, PNG_COMPRESSION_TYPE_DEFAULT,
                    160:                     PNG_FILTER_TYPE_DEFAULT);
1.1.1.12  root      161: 
                    162:        if ( png_compression_level >= 0 )
                    163:                png_set_compression_level ( png_ptr , png_compression_level );
                    164:        if ( png_filter >= 0 )
                    165:                png_set_filter ( png_ptr , 0 , png_filter );
                    166:                     
1.1.1.9   root      167:        /* image info */
                    168:        pngtext.key = key;
                    169:        pngtext.text = text;
                    170:        pngtext.compression = PNG_TEXT_COMPRESSION_NONE;
                    171: #ifdef PNG_iTXt_SUPPORTED
                    172:        pngtext.lang = NULL;
                    173: #endif
                    174:        png_set_text(png_ptr, info_ptr, &pngtext, 1);
                    175: 
                    176:        /* write the file header information */
                    177:        png_write_info(png_ptr, info_ptr);
                    178: 
1.1.1.12  root      179:        /* write surface data rows one at a time (after cropping if necessary) */
                    180:        do_lock = SDL_MUSTLOCK(surface);
1.1.1.17! root      181:        for (y = 0; y < dh; y++)
        !           182:        {
1.1.1.12  root      183:                /* need to lock the surface while accessing it directly */
                    184:                if (do_lock)
                    185:                        SDL_LockSurface(surface);
1.1.1.17! root      186: 
        !           187: 
        !           188:                src_ptr = (Uint8 *)surface->pixels
        !           189:                          + (CropTop + (y * sh + dh/2) / dh) * surface->pitch
        !           190:                          + CropLeft * surface->format->BytesPerPixel;
        !           191: 
        !           192:                switch (fmt->BytesPerPixel)
        !           193:                {
        !           194:                 case 2:
1.1.1.9   root      195:                        /* unpack 16-bit RGB pixels */
1.1.1.17! root      196:                        PixelConvert_16to24Bits(rowbuf, (Uint16*)src_ptr, dw, surface);
1.1.1.9   root      197:                        break;
1.1.1.17! root      198:                 case 4:
1.1.1.9   root      199:                        /* unpack 32-bit RGBA pixels */
1.1.1.17! root      200:                        PixelConvert_32to24Bits(rowbuf, (Uint32*)src_ptr, dw, surface);
1.1.1.9   root      201:                        break;
1.1.1.17! root      202:                 default:
        !           203:                        abort();
1.1.1.9   root      204:                }
1.1.1.12  root      205:                /* and unlock surface before syscalls */
                    206:                if (do_lock)
                    207:                        SDL_UnlockSurface(surface);
1.1.1.9   root      208:                png_write_row(png_ptr, rowbuf);
                    209:        }
1.1.1.12  root      210: 
1.1.1.9   root      211:        /* write the additional chuncks to the PNG file */
                    212:        png_write_end(png_ptr, info_ptr);
                    213: 
1.1.1.12  root      214:        ret = ftell ( fp ) - start;                             /* size of the png image */
1.1.1.9   root      215: png_cleanup:
                    216:        if (png_ptr)
                    217:                png_destroy_write_struct(&png_ptr, NULL);
                    218:        return ret;
                    219: }
                    220: #endif
                    221: 
                    222: 
                    223: /*-----------------------------------------------------------------------*/
                    224: /**
1.1.1.11  root      225:  * Save screen shot file with filename like 'grab0000.[png|bmp]',
                    226:  * 'grab0001.[png|bmp]', etc... Whether screen shots are saved as BMP
                    227:  * or PNG depends on Hatari configuration.
1.1.1.8   root      228:  */
1.1       root      229: void ScreenSnapShot_SaveScreen(void)
                    230: {
1.1.1.8   root      231:        char *szFileName = malloc(FILENAME_MAX);
1.1.1.6   root      232: 
1.1.1.8   root      233:        if (!szFileName)  return;
1.1.1.5   root      234: 
1.1.1.8   root      235:        ScreenSnapShot_GetNum();
                    236:        /* Create our filename */
                    237:        nScreenShots++;
1.1.1.9   root      238: #if HAVE_LIBPNG
                    239:        /* try first PNG */
                    240:        sprintf(szFileName,"%s/grab%4.4d.png", Paths_GetWorkingDir(), nScreenShots);
1.1.1.12  root      241:        if (ScreenSnapShot_SavePNG(sdlscrn, szFileName) > 0)
1.1.1.9   root      242:        {
                    243:                fprintf(stderr, "Screen dump saved to: %s\n", szFileName);
                    244:                free(szFileName);
                    245:                return;
                    246:        }
                    247: #endif
1.1.1.8   root      248:        sprintf(szFileName,"%s/grab%4.4d.bmp", Paths_GetWorkingDir(), nScreenShots);
                    249:        if (SDL_SaveBMP(sdlscrn, szFileName))
                    250:                fprintf(stderr, "Screen dump failed!\n");
                    251:        else
                    252:                fprintf(stderr, "Screen dump saved to: %s\n", szFileName);
1.1.1.5   root      253: 
1.1.1.8   root      254:        free(szFileName);
1.1       root      255: }
                    256: 

unix.superglobalmegacorp.com

This archive runs on limited infrastructure. Preserving old code on modern bandwidth. Automated agents are requested to crawl responsibly.