aboutsummaryrefslogtreecommitdiff
path: root/source/screenshot.c
blob: fc3cccbd201c4900ef59db88ddbc33ee1d013494 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
/*******************************************************************************
  Snes9x - Portable Super Nintendo Entertainment System (TM) emulator.

  (c) Copyright 1996 - 2002 Gary Henderson (gary.henderson@ntlworld.com) and
                            Jerremy Koot (jkoot@snes9x.com)

  (c) Copyright 2001 - 2004 John Weidman (jweidman@slip.net)

  (c) Copyright 2002 - 2004 Brad Jorsch (anomie@users.sourceforge.net),
                            funkyass (funkyass@spam.shaw.ca),
                            Joel Yliluoma (http://iki.fi/bisqwit/)
                            Kris Bleakley (codeviolation@hotmail.com),
                            Matthew Kendora,
                            Nach (n-a-c-h@users.sourceforge.net),
                            Peter Bortas (peter@bortas.org) and
                            zones (kasumitokoduck@yahoo.com)

  C4 x86 assembler and some C emulation code
  (c) Copyright 2000 - 2003 zsKnight (zsknight@zsnes.com),
                            _Demo_ (_demo_@zsnes.com), and Nach

  C4 C++ code
  (c) Copyright 2003 Brad Jorsch

  DSP-1 emulator code
  (c) Copyright 1998 - 2004 Ivar (ivar@snes9x.com), _Demo_, Gary Henderson,
                            John Weidman, neviksti (neviksti@hotmail.com),
                            Kris Bleakley, Andreas Naive

  DSP-2 emulator code
  (c) Copyright 2003 Kris Bleakley, John Weidman, neviksti, Matthew Kendora, and
                     Lord Nightmare (lord_nightmare@users.sourceforge.net

  OBC1 emulator code
  (c) Copyright 2001 - 2004 zsKnight, pagefault (pagefault@zsnes.com) and
                            Kris Bleakley
  Ported from x86 assembler to C by sanmaiwashi

  SPC7110 and RTC C++ emulator code
  (c) Copyright 2002 Matthew Kendora with research by
                     zsKnight, John Weidman, and Dark Force

  S-DD1 C emulator code
  (c) Copyright 2003 Brad Jorsch with research by
                     Andreas Naive and John Weidman

  S-RTC C emulator code
  (c) Copyright 2001 John Weidman

  ST010 C++ emulator code
  (c) Copyright 2003 Feather, Kris Bleakley, John Weidman and Matthew Kendora

  Super FX x86 assembler emulator code
  (c) Copyright 1998 - 2003 zsKnight, _Demo_, and pagefault

  Super FX C emulator code
  (c) Copyright 1997 - 1999 Ivar, Gary Henderson and John Weidman


  SH assembler code partly based on x86 assembler code
  (c) Copyright 2002 - 2004 Marcus Comstedt (marcus@mc.pp.se)


  Specific ports contains the works of other authors. See headers in
  individual files.

  Snes9x homepage: http://www.snes9x.com

  Permission to use, copy, modify and distribute Snes9x in both binary and
  source form, for non-commercial purposes, is hereby granted without fee,
  providing that this license information and copyright notice appear with
  all copies and any derived work.

  This software is provided 'as-is', without any express or implied
  warranty. In no event shall the authors be held liable for any damages
  arising from the use of this software.

  Snes9x is freeware for PERSONAL USE only. Commercial users should
  seek permission of the copyright holders first. Commercial use includes
  charging money for Snes9x or software derived from Snes9x.

  The copyright holders request that bug fixes and improvements to the code
  should be forwarded to them so everyone can benefit from the modifications
  in future versions.

  Super NES and Super Nintendo Entertainment System are trademarks of
  Nintendo Co., Limited and its subsidiary companies.
*******************************************************************************/


#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#include <stdio.h>

#ifndef __WIN32__
#include <unistd.h>
#else
#include <direct.h>
#endif
#include <string.h>
#include <fcntl.h>

#ifdef HAVE_LIBPNG
#include <png.h>
#endif

#include "snes9x.h"
#include "memmap.h"
#include "display.h"
#include "gfx.h"
#include "ppu.h"
#include "screenshot.h"

bool8 S9xDoScreenshot(int width, int height)
{
#ifdef HAVE_LIBPNG
   FILE* fp;
   png_structp png_ptr;
   png_infop info_ptr;
   png_color_8 sig_bit;
   png_color pngpal[256];
   int imgwidth;
   int imgheight;
   const char* fname = S9xGetFilenameInc(".png");

   Settings.TakeScreenshot = FALSE;

   if ((fp = fopen(fname, "wb")) == NULL)
   {
      perror("Screenshot failed");
      return FALSE;
   }

   png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
   if (!png_ptr)
   {
      fclose(fp);
      unlink(fname);
      return FALSE;
   }
   info_ptr = png_create_info_struct(png_ptr);
   if (!info_ptr)
   {
      png_destroy_write_struct(&png_ptr, (png_infopp)NULL);
      fclose(fp);
      unlink(fname);
      return FALSE;
   }

   if (setjmp(png_jmpbuf(png_ptr)))
   {
      perror("Screenshot: setjmp");
      png_destroy_write_struct(&png_ptr, &info_ptr);
      fclose(fp);
      unlink(fname);
      return FALSE;
   }

   imgwidth = width;
   imgheight = height;
   if (Settings.StretchScreenshots == 1)
   {
      if (width <= 256 && height > SNES_HEIGHT_EXTENDED) imgwidth = width << 1;
      if (width > 256 && height <= SNES_HEIGHT_EXTENDED) imgheight = height << 1;
   }
   else if (Settings.StretchScreenshots == 2)
   {
      if (width <= 256) imgwidth = width << 1;
      if (height <= SNES_HEIGHT_EXTENDED) imgheight = height << 1;
   }

   png_init_io(png_ptr, fp);
   if (!Settings.SixteenBit)
   {
      // BJ: credit sanmaiwashi for the idea to do palettized pngs, and to
      //     S9xSetPalette in x11.cpp for how to calculate the RGB values
      int b = IPPU.MaxBrightness * 140;
      for (int i = 0; i < 256; i++)
      {
         pngpal[i].red   = (PPU.CGDATA[i] & 0x1f) * b >> 8;
         pngpal[i].green = ((PPU.CGDATA[i] >> 5) & 0x1f) * b >> 8;
         pngpal[i].blue  = ((PPU.CGDATA[i] >> 10) & 0x1f) * b >> 8;
      }
      png_set_PLTE(png_ptr, info_ptr, pngpal, 256);
   }
   png_set_IHDR(png_ptr, info_ptr, imgwidth, imgheight, 8,
                (Settings.SixteenBit ? PNG_COLOR_TYPE_RGB : PNG_COLOR_TYPE_PALETTE),
                PNG_INTERLACE_NONE, PNG_COMPRESSION_TYPE_DEFAULT,
                PNG_FILTER_TYPE_DEFAULT);

   if (Settings.SixteenBit)
   {
      /* 5 bits per color */
      sig_bit.red = 5;
      sig_bit.green = 5;
      sig_bit.blue = 5;
      png_set_sBIT(png_ptr, info_ptr, &sig_bit);
      png_set_shift(png_ptr, &sig_bit);
   }

   png_write_info(png_ptr, info_ptr);

   png_set_packing(png_ptr);

   png_byte* row_pointer = new png_byte [png_get_rowbytes(png_ptr, info_ptr)];
   uint8* screen = GFX.Screen;
   for (int y = 0; y < height; y++, screen += GFX.Pitch)
   {
      png_byte* rowpix = row_pointer;
      for (int x = 0; x < width; x++)
      {
         if (Settings.SixteenBit)
         {
            uint32 r, g, b;
            DECOMPOSE_PIXEL((*(uint16*)(screen + 2 * x)), r, g, b);
            *(rowpix++) = r;
            *(rowpix++) = g;
            *(rowpix++) = b;
            if (imgwidth != width)
            {
               *(rowpix++) = r;
               *(rowpix++) = g;
               *(rowpix++) = b;
            }
         }
         else
         {
            *(rowpix++) = *(uint8*)(screen + x);
            if (imgwidth != width)
               *(rowpix++) = *(uint8*)(screen + x);
         }
      }
      png_write_row(png_ptr, row_pointer);
      if (imgheight != height)
         png_write_row(png_ptr, row_pointer);
   }

   delete [] row_pointer;

   png_write_end(png_ptr, info_ptr);
   png_destroy_write_struct(&png_ptr, &info_ptr);

   fclose(fp);
   fprintf(stderr, "%s saved.\n", fname);
   return TRUE;
#else
   perror("Screenshot support not available (libpng was not found at build time)");
   return FALSE;
#endif
}