mirror of https://github.com/OpenRCT2/OpenRCT2.git
269 lines
6.5 KiB
C
269 lines
6.5 KiB
C
/*****************************************************************************
|
|
* Copyright (c) 2014 Ted John
|
|
* OpenRCT2, an open source clone of Roller Coaster Tycoon 2.
|
|
*
|
|
* This file is part of OpenRCT2.
|
|
*
|
|
* OpenRCT2 is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*****************************************************************************/
|
|
#pragma pack(1)
|
|
|
|
#include <lodepng.h>
|
|
#include <stdio.h>
|
|
#include "osinterface.h"
|
|
#include "addresses.h"
|
|
#include "config.h"
|
|
#include "gfx.h"
|
|
#include "game.h"
|
|
#include "rct2.h"
|
|
#include "screenshot.h"
|
|
#include "string_ids.h"
|
|
#include "window_error.h"
|
|
|
|
#include <windows.h> // For MAX_PATH
|
|
|
|
|
|
static int screenshot_dump_bmp();
|
|
static int screenshot_dump_png();
|
|
|
|
/**
|
|
*
|
|
* rct2: 0x006E3AEC
|
|
*/
|
|
void screenshot_check()
|
|
{
|
|
int screenshotIndex;
|
|
|
|
if (RCT2_GLOBAL(RCT2_ADDRESS_SCREENSHOT_COUNTDOWN, uint8) != 0) {
|
|
RCT2_GLOBAL(RCT2_ADDRESS_SCREENSHOT_COUNTDOWN, uint8)--;
|
|
if (RCT2_GLOBAL(RCT2_ADDRESS_SCREENSHOT_COUNTDOWN, uint8) == 0) {
|
|
update_rain_animation();
|
|
screenshotIndex = screenshot_dump();
|
|
RCT2_GLOBAL(0x013CE952, uint16) = STR_SCR_BMP;
|
|
RCT2_GLOBAL(0x013CE952 + 2, uint16) = screenshotIndex;
|
|
RCT2_GLOBAL(0x009A8C29, uint8) |= 1;
|
|
|
|
window_error_open(screenshotIndex == -1 ? STR_SCREENSHOT_FAILED : STR_SCREENSHOT_SAVED_AS, -1);
|
|
RCT2_GLOBAL(0x009A8C29, uint8) &= ~1;
|
|
RCT2_CALLPROC_EBPSAFE(0x006843DC);
|
|
}
|
|
}
|
|
}
|
|
|
|
static int screenshot_get_next_path(char *path, char *extension)
|
|
{
|
|
char *screenshotPath = osinterface_get_orct2_homesubfolder("screenshot");
|
|
if (!osinterface_ensure_directory_exists(screenshotPath)) {
|
|
free(screenshotPath);
|
|
|
|
fprintf(stderr, "Unable to save screenshots in OpenRCT2 screenshot directory.\n");
|
|
return -1;
|
|
}
|
|
|
|
int i;
|
|
for (i = 1; i < 1000; i++) {
|
|
RCT2_GLOBAL(0x013CE952, uint16) = i;
|
|
|
|
// Glue together path and filename
|
|
sprintf(path, "%s%cSCR%d%s", screenshotPath, osinterface_get_path_separator(), i, extension);
|
|
|
|
if (!osinterface_file_exists(path)) {
|
|
return i;
|
|
}
|
|
}
|
|
|
|
free(screenshotPath);
|
|
return -1;
|
|
}
|
|
|
|
int screenshot_dump()
|
|
{
|
|
switch (gGeneral_config.screenshot_format) {
|
|
case SCREENSHOT_FORMAT_BMP:
|
|
return screenshot_dump_bmp();
|
|
case SCREENSHOT_FORMAT_PNG:
|
|
return screenshot_dump_png();
|
|
default:
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
// Bitmap header structs, for cross platform purposes
|
|
typedef struct {
|
|
uint16 bfType;
|
|
uint32 bfSize;
|
|
uint16 bfReserved1;
|
|
uint16 bfReserved2;
|
|
uint32 bfOffBits;
|
|
} BitmapFileHeader;
|
|
|
|
typedef struct {
|
|
uint32 biSize;
|
|
sint32 biWidth;
|
|
sint32 biHeight;
|
|
uint16 biPlanes;
|
|
uint16 biBitCount;
|
|
uint32 biCompression;
|
|
uint32 biSizeImage;
|
|
sint32 biXPelsPerMeter;
|
|
sint32 biYPelsPerMeter;
|
|
uint32 biClrUsed;
|
|
uint32 biClrImportant;
|
|
} BitmapInfoHeader;
|
|
|
|
/**
|
|
*
|
|
* rct2: 0x00683D20
|
|
*/
|
|
int screenshot_dump_bmp()
|
|
{
|
|
BitmapFileHeader header;
|
|
BitmapInfoHeader info;
|
|
|
|
int i, y, index, width, height, stride;
|
|
char *buffer, path[MAX_PATH], *row;
|
|
FILE *fp;
|
|
unsigned int bytesWritten;
|
|
|
|
// Get a free screenshot path
|
|
if ((index = screenshot_get_next_path(path, ".bmp")) == -1)
|
|
return -1;
|
|
|
|
// Open binary file for writing
|
|
if ((fp = fopen(path, "wb")) == NULL){
|
|
return -1;
|
|
}
|
|
|
|
// Allocate buffer
|
|
buffer = malloc(0xFFFF);
|
|
if (buffer == NULL) {
|
|
//CloseHandle(hFile);
|
|
fclose(fp);
|
|
return -1;
|
|
}
|
|
|
|
// Get image size
|
|
width = RCT2_GLOBAL(RCT2_ADDRESS_SCREEN_WIDTH, uint16);
|
|
height = RCT2_GLOBAL(RCT2_ADDRESS_SCREEN_HEIGHT, uint16);
|
|
stride = (width + 3) & 0xFFFFFFFC;
|
|
|
|
// File header
|
|
memset(&header, 0, sizeof(header));
|
|
header.bfType = 0x4D42;
|
|
header.bfSize = height * stride + 1038;
|
|
header.bfOffBits = 1038;
|
|
|
|
bytesWritten = fwrite(&header, sizeof(BitmapFileHeader), 1, fp);
|
|
if (bytesWritten != 1) {
|
|
fclose(fp);
|
|
free(buffer);
|
|
}
|
|
|
|
// Info header
|
|
memset(&info, 0, sizeof(info));
|
|
info.biSize = sizeof(info);
|
|
info.biWidth = width;
|
|
info.biHeight = height;
|
|
info.biPlanes = 1;
|
|
info.biBitCount = 8;
|
|
info.biXPelsPerMeter = 2520;
|
|
info.biYPelsPerMeter = 2520;
|
|
info.biClrUsed = 246;
|
|
|
|
bytesWritten=fwrite(&info, sizeof(BitmapInfoHeader), 1, fp);
|
|
if (bytesWritten != 1) {
|
|
fclose(fp);
|
|
free(buffer);
|
|
}
|
|
|
|
// Palette
|
|
memset(buffer, 0, 246 * 4);
|
|
for (i = 0; i < 246; i++) {
|
|
buffer[i * 4 + 0] = RCT2_ADDRESS(0x01424680, uint8)[i * 4 + 0];
|
|
buffer[i * 4 + 1] = RCT2_ADDRESS(0x01424680, uint8)[i * 4 + 1];
|
|
buffer[i * 4 + 2] = RCT2_ADDRESS(0x01424680, uint8)[i * 4 + 2];
|
|
}
|
|
|
|
bytesWritten = fwrite(buffer, sizeof(char), 246*4, fp);
|
|
if (bytesWritten != 246*4){
|
|
fclose(fp);
|
|
free(buffer);
|
|
}
|
|
|
|
// Image, save upside down
|
|
rct_drawpixelinfo *dpi = RCT2_ADDRESS(RCT2_ADDRESS_SCREEN_DPI, rct_drawpixelinfo);
|
|
for (y = dpi->height - 1; y >= 0; y--) {
|
|
row = dpi->bits + y * (dpi->width + dpi->pitch);
|
|
|
|
memset(buffer, 0, stride);
|
|
memcpy(buffer, row, dpi->width);
|
|
|
|
bytesWritten=fwrite(buffer, sizeof(char), stride, fp);
|
|
if (bytesWritten != stride){
|
|
fclose(fp);
|
|
free(buffer);
|
|
}
|
|
}
|
|
|
|
fclose(fp);
|
|
free(buffer);
|
|
|
|
return index;
|
|
}
|
|
|
|
int screenshot_dump_png()
|
|
{
|
|
int i, index, width, height, stride;
|
|
char path[MAX_PATH] = "";
|
|
|
|
unsigned char r, g, b, a = 255;
|
|
|
|
unsigned char* png;
|
|
size_t pngSize;
|
|
LodePNGState state;
|
|
|
|
// Get a free screenshot path
|
|
if ((index = screenshot_get_next_path(path, ".png")) == -1)
|
|
return -1;
|
|
|
|
|
|
lodepng_state_init(&state);
|
|
state.info_raw.colortype = LCT_PALETTE;
|
|
|
|
// Get image size
|
|
width = RCT2_GLOBAL(RCT2_ADDRESS_SCREEN_WIDTH, uint16);
|
|
height = RCT2_GLOBAL(RCT2_ADDRESS_SCREEN_HEIGHT, uint16);
|
|
stride = (width + 3) & 0xFFFFFFFC;
|
|
|
|
for (i = 0; i < 246; i++) {
|
|
b = RCT2_ADDRESS(0x01424680, uint8)[i * 4 + 0];
|
|
g = RCT2_ADDRESS(0x01424680, uint8)[i * 4 + 1];
|
|
r = RCT2_ADDRESS(0x01424680, uint8)[i * 4 + 2];
|
|
|
|
lodepng_palette_add(&state.info_raw, r, g, b, a);
|
|
}
|
|
|
|
rct_drawpixelinfo *dpi = RCT2_ADDRESS(RCT2_ADDRESS_SCREEN_DPI, rct_drawpixelinfo);
|
|
|
|
unsigned int error = lodepng_encode(&png, &pngSize, dpi->bits, stride, dpi->height, &state);
|
|
if (!error) lodepng_save_file(png, pngSize, path);
|
|
|
|
if (error) {
|
|
fprintf(stderr, "error: %u: %s\n", error, lodepng_error_text(error));
|
|
index = -1;
|
|
}
|
|
|
|
free(png);
|
|
return index;
|
|
} |