RetroArch/screenshot.c

226 lines
5.9 KiB
C
Raw Normal View History

2012-04-21 21:13:50 +00:00
/* RetroArch - A frontend for libretro.
2014-01-01 00:50:59 +00:00
* Copyright (C) 2010-2014 - Hans-Kristian Arntzen
2011-05-15 14:54:43 +00:00
*
2012-04-21 21:13:50 +00:00
* RetroArch is free software: you can redistribute it and/or modify it under the terms
2011-05-15 14:54:43 +00:00
* of the GNU General Public License as published by the Free Software Found-
* ation, either version 3 of the License, or (at your option) any later version.
*
2012-04-21 21:13:50 +00:00
* RetroArch is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
2011-05-15 14:54:43 +00:00
* without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
* PURPOSE. See the GNU General Public License for more details.
*
2012-04-21 21:31:57 +00:00
* You should have received a copy of the GNU General Public License along with RetroArch.
2011-05-15 14:54:43 +00:00
* If not, see <http://www.gnu.org/licenses/>.
*/
2012-03-16 22:26:57 +00:00
#include "compat/strl.h"
2011-05-15 14:54:43 +00:00
#include <stdio.h>
#include <stddef.h>
2011-05-15 14:54:43 +00:00
#include <time.h>
2011-12-24 12:46:12 +00:00
#include "boolean.h"
2011-05-15 14:54:43 +00:00
#include <stdint.h>
#include <string.h>
#include "general.h"
2012-11-25 15:03:16 +00:00
#include "file.h"
2013-01-19 01:13:02 +00:00
#include "gfx/scaler/scaler.h"
2011-05-15 14:54:43 +00:00
2012-06-17 11:11:12 +00:00
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
2011-05-15 14:54:43 +00:00
2013-01-19 01:13:02 +00:00
#ifdef HAVE_ZLIB_DEFLATE
#include "gfx/rpng/rpng.h"
2012-06-17 11:11:12 +00:00
#else
static bool write_header_bmp(FILE *file, unsigned width, unsigned height)
2011-05-15 14:54:43 +00:00
{
unsigned line_size = (width * 3 + 3) & ~3;
unsigned size = line_size * height + 54;
unsigned size_array = line_size * height;
/* Generic BMP stuff. */
2011-05-15 14:54:43 +00:00
const uint8_t header[] = {
'B', 'M',
2014-09-02 14:13:42 +00:00
(uint8_t)(size >> 0), (uint8_t)(size >> 8),
(uint8_t)(size >> 16), (uint8_t)(size >> 24),
2011-05-15 14:54:43 +00:00
0, 0, 0, 0,
54, 0, 0, 0,
40, 0, 0, 0,
2014-09-02 14:13:42 +00:00
(uint8_t)(width >> 0), (uint8_t)(width >> 8),
(uint8_t)(width >> 16), (uint8_t)(width >> 24),
(uint8_t)(height >> 0), (uint8_t)(height >> 8),
(uint8_t)(height >> 16), (uint8_t)(height >> 24),
2011-05-15 14:54:43 +00:00
1, 0,
24, 0,
0, 0, 0, 0,
2014-09-02 14:13:42 +00:00
(uint8_t)(size_array >> 0), (uint8_t)(size_array >> 8),
(uint8_t)(size_array >> 16), (uint8_t)(size_array >> 24),
2011-05-15 14:54:43 +00:00
19, 11, 0, 0,
19, 11, 0, 0,
0, 0, 0, 0,
0, 0, 0, 0
};
2012-06-17 11:11:12 +00:00
return fwrite(header, 1, sizeof(header), file) == sizeof(header);
2011-05-15 14:54:43 +00:00
}
2014-09-02 14:13:42 +00:00
static void dump_lines_file(FILE *file, uint8_t **lines,
size_t line_size, unsigned height)
2012-06-17 11:11:12 +00:00
{
unsigned i;
for (i = 0; i < height; i++)
2012-06-17 11:11:12 +00:00
fwrite(lines[i], 1, line_size, file);
}
static void dump_line_bgr(uint8_t *line, const uint8_t *src, unsigned width)
2011-05-15 14:54:43 +00:00
{
memcpy(line, src, width * 3);
}
static void dump_line_16(uint8_t *line, const uint16_t *src, unsigned width)
{
unsigned i;
for (i = 0; i < width; i++)
{
uint16_t pixel = *src++;
uint8_t b = (pixel >> 0) & 0x1f;
uint8_t g = (pixel >> 5) & 0x3f;
uint8_t r = (pixel >> 11) & 0x1f;
*line++ = (b << 3) | (b >> 2);
*line++ = (g << 2) | (g >> 4);
*line++ = (r << 3) | (r >> 2);
}
}
static void dump_line_32(uint8_t *line, const uint32_t *src, unsigned width)
{
unsigned i;
for (i = 0; i < width; i++)
{
uint32_t pixel = *src++;
*line++ = (pixel >> 0) & 0xff;
*line++ = (pixel >> 8) & 0xff;
*line++ = (pixel >> 16) & 0xff;
}
}
static void dump_content(FILE *file, const void *frame,
int width, int height, int pitch, bool bgr24)
{
int i, j;
union
{
const uint8_t *u8;
const uint16_t *u16;
const uint32_t *u32;
} u;
u.u8 = (const uint8_t*)frame;
2012-06-17 11:11:12 +00:00
uint8_t **lines = (uint8_t**)calloc(height, sizeof(uint8_t*));
if (!lines)
2011-12-24 12:46:12 +00:00
return;
2011-05-15 14:54:43 +00:00
2012-06-17 11:11:12 +00:00
size_t line_size = (width * 3 + 3) & ~3;
for (i = 0; i < height; i++)
2012-06-17 11:11:12 +00:00
{
lines[i] = (uint8_t*)calloc(1, line_size);
if (!lines[i])
goto end;
}
if (bgr24) /* BGR24 byte order. Can directly copy. */
2011-05-15 14:54:43 +00:00
{
for (j = 0; j < height; j++, u.u8 += pitch)
dump_line_bgr(lines[j], u.u8, width);
}
else if (g_extern.system.pix_fmt == RETRO_PIXEL_FORMAT_XRGB8888)
{
for (j = 0; j < height; j++, u.u8 += pitch)
dump_line_32(lines[j], u.u32, width);
}
else /* RGB565 */
{
for (j = 0; j < height; j++, u.u8 += pitch)
dump_line_16(lines[j], u.u16, width);
2011-05-15 14:54:43 +00:00
}
2011-12-24 12:46:12 +00:00
2012-06-17 11:11:12 +00:00
dump_lines_file(file, lines, line_size, height);
end:
for (i = 0; i < height; i++)
2012-06-17 11:11:12 +00:00
free(lines[i]);
free(lines);
2011-05-15 14:54:43 +00:00
}
2013-01-19 01:13:02 +00:00
#endif
2011-05-15 14:54:43 +00:00
/* Take frame bottom-up. */
bool screenshot_dump(const char *folder, const void *frame,
unsigned width, unsigned height, int pitch, bool bgr24)
{
char filename[PATH_MAX];
char shotname[PATH_MAX];
#ifdef HAVE_ZLIB_DEFLATE
#define IMG_EXT "png"
#else
#define IMG_EXT "bmp"
#endif
fill_dated_filename(shotname, IMG_EXT, sizeof(shotname));
2012-11-25 15:03:16 +00:00
fill_pathname_join(filename, folder, shotname, sizeof(filename));
2011-05-15 14:54:43 +00:00
2013-01-19 01:13:02 +00:00
#ifdef HAVE_ZLIB_DEFLATE
uint8_t *out_buffer = (uint8_t*)malloc(width * height * 3);
if (!out_buffer)
return false;
struct scaler_ctx scaler = {0};
scaler.in_width = width;
scaler.in_height = height;
scaler.out_width = width;
scaler.out_height = height;
scaler.in_stride = -pitch;
scaler.out_stride = width * 3;
scaler.out_fmt = SCALER_FMT_BGR24;
scaler.scaler_type = SCALER_TYPE_POINT;
if (bgr24)
scaler.in_fmt = SCALER_FMT_BGR24;
else if (g_extern.system.pix_fmt == RETRO_PIXEL_FORMAT_XRGB8888)
scaler.in_fmt = SCALER_FMT_ARGB8888;
else
scaler.in_fmt = SCALER_FMT_RGB565;
scaler_ctx_gen_filter(&scaler);
2014-09-02 14:13:42 +00:00
scaler_ctx_scale(&scaler, out_buffer,
(const uint8_t*)frame + ((int)height - 1) * pitch);
2013-01-19 01:13:02 +00:00
scaler_ctx_gen_reset(&scaler);
RARCH_LOG("Using RPNG for PNG screenshots.\n");
2014-09-02 14:13:42 +00:00
bool ret = rpng_save_image_bgr24(filename,
out_buffer, width, height, width * 3);
2013-01-19 01:13:02 +00:00
if (!ret)
RARCH_ERR("Failed to take screenshot.\n");
free(out_buffer);
return ret;
#else
2011-05-15 14:54:43 +00:00
FILE *file = fopen(filename, "wb");
if (!file)
{
2012-04-21 21:25:32 +00:00
RARCH_ERR("Failed to open file \"%s\" for screenshot.\n", filename);
2011-05-15 14:54:43 +00:00
return false;
}
2012-06-17 11:11:12 +00:00
bool ret = write_header_bmp(file, width, height);
2011-05-15 14:54:43 +00:00
2012-06-17 11:11:12 +00:00
if (ret)
dump_content(file, frame, width, height, pitch, bgr24);
else
RARCH_ERR("Failed to write image header.\n");
2011-05-15 14:54:43 +00:00
2012-06-17 11:11:12 +00:00
fclose(file);
return ret;
2013-01-19 01:13:02 +00:00
#endif
2011-05-15 14:54:43 +00:00
}