0
|
1 |
#include "stdafx.h"
|
|
2 |
#include "ttd.h"
|
|
3 |
#include "gfx.h"
|
|
4 |
#include "viewport.h"
|
|
5 |
#include "player.h"
|
|
6 |
#include "gui.h"
|
|
7 |
|
|
8 |
// called by the ScreenShot proc to generate screenshot lines.
|
|
9 |
typedef void ScreenshotCallback(void *userdata, byte *buf, uint y, uint pitch, uint n);
|
|
10 |
typedef bool ScreenshotHandlerProc(const char *name, ScreenshotCallback *callb, void *userdata, uint w, uint h, int pixelformat, const byte *palette);
|
|
11 |
|
|
12 |
typedef struct {
|
|
13 |
const char *name;
|
|
14 |
const char *extension;
|
|
15 |
ScreenshotHandlerProc *proc;
|
|
16 |
byte id;
|
|
17 |
} ScreenshotFormat;
|
|
18 |
|
|
19 |
//************************************************
|
|
20 |
//*** SCREENSHOT CODE FOR WINDOWS BITMAP (.BMP)
|
|
21 |
//************************************************
|
|
22 |
#if defined(_MSC_VER)
|
|
23 |
#pragma pack(push, 1)
|
|
24 |
#endif
|
|
25 |
|
|
26 |
typedef struct BitmapFileHeader {
|
|
27 |
uint16 type;
|
|
28 |
uint32 size;
|
|
29 |
uint32 reserved;
|
|
30 |
uint32 off_bits;
|
|
31 |
} GCC_PACK BitmapFileHeader;
|
|
32 |
assert_compile(sizeof(BitmapFileHeader) == 14);
|
|
33 |
|
|
34 |
#if defined(_MSC_VER)
|
|
35 |
#pragma pack(pop)
|
|
36 |
#endif
|
|
37 |
|
|
38 |
typedef struct BitmapInfoHeader {
|
|
39 |
uint32 size;
|
|
40 |
int32 width, height;
|
|
41 |
uint16 planes, bitcount;
|
|
42 |
uint32 compression, sizeimage, xpels, ypels, clrused, clrimp;
|
|
43 |
} BitmapInfoHeader;
|
|
44 |
assert_compile(sizeof(BitmapInfoHeader) == 40);
|
|
45 |
|
|
46 |
typedef struct RgbQuad {
|
|
47 |
byte blue, green, red, reserved;
|
|
48 |
} RgbQuad;
|
|
49 |
assert_compile(sizeof(RgbQuad) == 4);
|
|
50 |
|
|
51 |
// generic .BMP writer
|
|
52 |
static bool MakeBmpImage(const char *name, ScreenshotCallback *callb, void *userdata, uint w, uint h, int pixelformat, const byte *palette)
|
|
53 |
{
|
|
54 |
BitmapFileHeader bfh;
|
|
55 |
BitmapInfoHeader bih;
|
|
56 |
RgbQuad *rq = alloca(sizeof(RgbQuad) * 256);
|
|
57 |
byte *buff;
|
|
58 |
FILE *f;
|
|
59 |
uint i, padw;
|
|
60 |
uint n, maxlines;
|
|
61 |
|
|
62 |
// only implemented for 8bit images so far.
|
|
63 |
if (pixelformat != 8)
|
|
64 |
return false;
|
|
65 |
|
|
66 |
f = fopen(name, "wb");
|
|
67 |
if (f == NULL) return false;
|
|
68 |
|
|
69 |
// each scanline must be aligned on a 32bit boundary
|
|
70 |
padw = (w + 3) & ~3;
|
|
71 |
|
|
72 |
// setup the file header
|
|
73 |
bfh.type = TO_LE16('MB');
|
|
74 |
bfh.size = TO_LE32(sizeof(bfh) + sizeof(bih) + sizeof(RgbQuad) * 256 + padw * h);
|
|
75 |
bfh.reserved = 0;
|
|
76 |
bfh.off_bits = TO_LE32(sizeof(bfh) + sizeof(bih) + sizeof(RgbQuad) * 256);
|
|
77 |
|
|
78 |
// setup the info header
|
|
79 |
bih.size = TO_LE32(sizeof(BitmapInfoHeader));
|
|
80 |
bih.width = TO_LE32(w);
|
|
81 |
bih.height = TO_LE32(h);
|
|
82 |
bih.planes = TO_LE16(1);
|
|
83 |
bih.bitcount = TO_LE16(8);
|
|
84 |
bih.compression = 0;
|
|
85 |
bih.sizeimage = 0;
|
|
86 |
bih.xpels = 0;
|
|
87 |
bih.ypels = 0;
|
|
88 |
bih.clrused = 0;
|
|
89 |
bih.clrimp = 0;
|
|
90 |
|
|
91 |
// convert the palette to the windows format
|
|
92 |
for(i=0; i!=256; i++) {
|
|
93 |
rq[i].red = *palette++;
|
|
94 |
rq[i].green = *palette++;
|
|
95 |
rq[i].blue = *palette++;
|
|
96 |
rq[i].reserved = 0;
|
|
97 |
}
|
|
98 |
|
|
99 |
// write file header and info header and palette
|
|
100 |
fwrite(&bfh, 1, sizeof(bfh), f);
|
|
101 |
fwrite(&bih, 1, sizeof(bih), f);
|
|
102 |
fwrite(rq, 1, sizeof(RgbQuad) * 256, f);
|
|
103 |
|
|
104 |
// use by default 64k temp memory
|
|
105 |
maxlines = clamp(65536 / padw, 16, 128);
|
|
106 |
|
|
107 |
// now generate the bitmap bits
|
|
108 |
buff = (byte*)alloca(padw * maxlines); // by default generate 128 lines at a time.
|
|
109 |
memset(buff, 0, padw * maxlines); // zero the buffer to have the padding bytes set to 0
|
|
110 |
|
|
111 |
// start at the bottom, since bitmaps are stored bottom up.
|
|
112 |
do {
|
|
113 |
// determine # lines
|
|
114 |
n = min(h, maxlines);
|
|
115 |
h -= n;
|
|
116 |
|
|
117 |
// render the pixels
|
|
118 |
callb(userdata, buff, h, padw, n);
|
|
119 |
|
|
120 |
// write each line
|
|
121 |
while (n)
|
|
122 |
fwrite(buff + (--n) * padw, 1, padw, f);
|
|
123 |
} while (h);
|
|
124 |
|
|
125 |
fclose(f);
|
|
126 |
|
|
127 |
return true;
|
|
128 |
}
|
|
129 |
|
|
130 |
//********************************************************
|
|
131 |
//*** SCREENSHOT CODE FOR PORTABLE NETWORK GRAPHICS (.PNG)
|
|
132 |
//********************************************************
|
|
133 |
#if defined(WITH_PNG)
|
|
134 |
#include <png.h>
|
|
135 |
|
|
136 |
static void PNGAPI png_my_error(png_structp png_ptr, png_const_charp message)
|
|
137 |
{
|
|
138 |
DEBUG(misc, 0) ("ERROR(libpng): %s - %s\n", message,(char *)png_get_error_ptr(png_ptr));
|
|
139 |
longjmp(png_ptr->jmpbuf, 1);
|
|
140 |
}
|
|
141 |
|
|
142 |
static void PNGAPI png_my_warning(png_structp png_ptr, png_const_charp message)
|
|
143 |
{
|
|
144 |
DEBUG(misc, 0) ("WARNING(libpng): %s - %s\n", message, (char *)png_get_error_ptr(png_ptr));
|
|
145 |
}
|
|
146 |
|
|
147 |
static bool MakePNGImage(const char *name, ScreenshotCallback *callb, void *userdata, uint w, uint h, int pixelformat, const byte *palette)
|
|
148 |
{
|
|
149 |
png_colorp rq = alloca(sizeof(png_color) * 256);
|
|
150 |
byte *buff;
|
|
151 |
FILE *f;
|
|
152 |
uint i, y, n;
|
|
153 |
uint maxlines;
|
|
154 |
png_structp png_ptr;
|
|
155 |
png_infop info_ptr;
|
|
156 |
|
|
157 |
// only implemented for 8bit images so far.
|
|
158 |
if (pixelformat != 8)
|
|
159 |
return false;
|
|
160 |
|
|
161 |
f = fopen(name, "wb");
|
|
162 |
if (f == NULL) return false;
|
|
163 |
|
|
164 |
png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, (char *) name, png_my_error, png_my_warning);
|
|
165 |
|
|
166 |
if (!png_ptr) {
|
|
167 |
fclose(f);
|
|
168 |
return false;
|
|
169 |
}
|
|
170 |
|
|
171 |
info_ptr = png_create_info_struct(png_ptr);
|
|
172 |
if (!info_ptr) {
|
|
173 |
png_destroy_write_struct(&png_ptr, (png_infopp)NULL);
|
|
174 |
fclose(f);
|
|
175 |
return false;
|
|
176 |
}
|
|
177 |
|
|
178 |
if (setjmp(png_jmpbuf(png_ptr))) {
|
|
179 |
png_destroy_write_struct(&png_ptr, &info_ptr);
|
|
180 |
fclose(f);
|
|
181 |
return false;
|
|
182 |
}
|
|
183 |
|
|
184 |
png_init_io(png_ptr, f);
|
|
185 |
|
|
186 |
png_set_filter(png_ptr, 0, PNG_FILTER_NONE);
|
|
187 |
|
|
188 |
png_set_IHDR(png_ptr, info_ptr, w, h, pixelformat, PNG_COLOR_TYPE_PALETTE, PNG_INTERLACE_NONE,
|
|
189 |
PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);
|
|
190 |
|
|
191 |
// convert the palette to the .PNG format.
|
|
192 |
{
|
|
193 |
// avoids "might be clobbered" warning of argument "palette"
|
|
194 |
const byte *pal = palette;
|
|
195 |
|
|
196 |
for(i=0; i!=256; i++) {
|
|
197 |
rq[i].red = *pal++;
|
|
198 |
rq[i].green = *pal++;
|
|
199 |
rq[i].blue = *pal++;
|
|
200 |
}
|
|
201 |
}
|
|
202 |
|
|
203 |
png_set_PLTE(png_ptr, info_ptr, rq, 256);
|
|
204 |
png_write_info(png_ptr, info_ptr);
|
|
205 |
png_set_flush(png_ptr, 512);
|
|
206 |
|
|
207 |
// use by default 64k temp memory
|
|
208 |
maxlines = clamp(65536 / w, 16, 128);
|
|
209 |
|
|
210 |
// now generate the bitmap bits
|
|
211 |
buff = (byte*)alloca(w * maxlines); // by default generate 128 lines at a time.
|
|
212 |
memset(buff, 0, w * maxlines); // zero the buffer to have the padding bytes set to 0
|
|
213 |
|
|
214 |
y = 0;
|
|
215 |
do {
|
|
216 |
// determine # lines to write
|
|
217 |
n = min(h - y, maxlines);
|
|
218 |
|
|
219 |
// render the pixels into the buffer
|
|
220 |
callb(userdata, buff, y, w, n);
|
|
221 |
y += n;
|
|
222 |
|
|
223 |
// write them to png
|
|
224 |
for(i=0; i!=n; i++)
|
|
225 |
png_write_row(png_ptr, buff + i * w);
|
|
226 |
} while (y != h);
|
|
227 |
|
|
228 |
png_write_end(png_ptr, info_ptr);
|
|
229 |
png_destroy_write_struct(&png_ptr, &info_ptr);
|
|
230 |
|
|
231 |
fclose(f);
|
|
232 |
return true;
|
|
233 |
}
|
|
234 |
#endif // WITH_PNG
|
|
235 |
|
|
236 |
|
|
237 |
//************************************************
|
|
238 |
//*** SCREENSHOT CODE FOR ZSOFT PAINTBRUSH (.PCX)
|
|
239 |
//************************************************
|
|
240 |
|
|
241 |
typedef struct {
|
|
242 |
byte manufacturer;
|
|
243 |
byte version;
|
|
244 |
byte rle;
|
|
245 |
byte bpp;
|
|
246 |
uint32 unused;
|
|
247 |
uint16 xmax, ymax;
|
|
248 |
uint16 hdpi, vdpi;
|
|
249 |
byte pal_small[16*3];
|
|
250 |
byte reserved;
|
|
251 |
byte planes;
|
|
252 |
uint16 pitch;
|
|
253 |
uint16 cpal;
|
|
254 |
uint16 width;
|
|
255 |
uint16 height;
|
|
256 |
byte filler[54];
|
|
257 |
} PcxHeader;
|
|
258 |
assert_compile(sizeof(PcxHeader) == 128);
|
|
259 |
|
|
260 |
static bool MakePCXImage(const char *name, ScreenshotCallback *callb, void *userdata, uint w, uint h, int pixelformat, const byte *palette)
|
|
261 |
{
|
|
262 |
byte *buff;
|
|
263 |
FILE *f;
|
|
264 |
uint maxlines;
|
|
265 |
uint y;
|
|
266 |
PcxHeader pcx;
|
|
267 |
|
|
268 |
if (pixelformat != 8 || w == 0)
|
|
269 |
return false;
|
|
270 |
|
|
271 |
f = fopen(name, "wb");
|
|
272 |
if (f == NULL) return false;
|
|
273 |
|
|
274 |
memset(&pcx, 0, sizeof(pcx));
|
|
275 |
|
|
276 |
// setup pcx header
|
|
277 |
pcx.manufacturer = 10;
|
|
278 |
pcx.version = 5;
|
|
279 |
pcx.rle = 1;
|
|
280 |
pcx.bpp = 8;
|
|
281 |
pcx.xmax = TO_LE16(w-1);
|
|
282 |
pcx.ymax = TO_LE16(h-1);
|
|
283 |
pcx.hdpi = TO_LE16(320);
|
|
284 |
pcx.vdpi = TO_LE16(320);
|
|
285 |
|
|
286 |
pcx.planes = 1;
|
|
287 |
pcx.cpal = TO_LE16(1);
|
|
288 |
pcx.width = pcx.pitch = TO_LE16(w);
|
|
289 |
pcx.height = TO_LE16(h);
|
|
290 |
|
|
291 |
// write pcx header
|
|
292 |
fwrite(&pcx, sizeof(pcx), 1, f);
|
|
293 |
|
|
294 |
// use by default 64k temp memory
|
|
295 |
maxlines = clamp(65536 / w, 16, 128);
|
|
296 |
|
|
297 |
// now generate the bitmap bits
|
|
298 |
buff = (byte*)alloca(w * maxlines); // by default generate 128 lines at a time.
|
|
299 |
memset(buff, 0, w * maxlines); // zero the buffer to have the padding bytes set to 0
|
|
300 |
|
|
301 |
y = 0;
|
|
302 |
do {
|
|
303 |
// determine # lines to write
|
|
304 |
uint n = min(h - y, maxlines), i;
|
|
305 |
|
|
306 |
// render the pixels into the buffer
|
|
307 |
callb(userdata, buff, y, w, n);
|
|
308 |
y += n;
|
|
309 |
|
|
310 |
// write them to pcx
|
|
311 |
for(i=0; i!=n; i++) {
|
|
312 |
int runcount = 1;
|
|
313 |
byte *bufp = buff + i * w;
|
|
314 |
byte runchar = buff[0];
|
|
315 |
uint left = w - 1;
|
|
316 |
|
|
317 |
// for each pixel...
|
|
318 |
while (left) {
|
|
319 |
byte ch = *bufp++;
|
|
320 |
if (ch != runchar || runcount >= 0x3f) {
|
|
321 |
if (runcount > 1 || (runchar & 0xC0) == 0xC0) fputc(0xC0 | runcount, f);
|
|
322 |
fputc(runchar,f);
|
|
323 |
runcount = 0;
|
|
324 |
runchar = ch;
|
|
325 |
}
|
|
326 |
runcount++;
|
|
327 |
left--;
|
|
328 |
}
|
|
329 |
|
|
330 |
// write remaining bytes..
|
|
331 |
if (runcount > 1 || (runchar & 0xC0) == 0xC0) fputc(0xC0 | runcount, f);
|
|
332 |
fputc(runchar,f);
|
|
333 |
}
|
|
334 |
} while (y != h);
|
|
335 |
|
|
336 |
// write 8-bit color palette
|
|
337 |
fputc(12, f);
|
|
338 |
fwrite(palette, 256*3, 1, f);
|
|
339 |
fclose(f);
|
|
340 |
|
|
341 |
return true;
|
|
342 |
}
|
|
343 |
|
|
344 |
//************************************************
|
|
345 |
//*** GENERIC SCREENSHOT CODE
|
|
346 |
//************************************************
|
|
347 |
|
|
348 |
static const ScreenshotFormat _screenshot_formats[] = {
|
|
349 |
{"BMP", "bmp", &MakeBmpImage},
|
|
350 |
#if defined(WITH_PNG)
|
|
351 |
{"PNG", "png", &MakePNGImage},
|
|
352 |
#endif
|
|
353 |
{"PCX", "pcx", &MakePCXImage},
|
|
354 |
};
|
|
355 |
|
|
356 |
void InitializeScreenshotFormats()
|
|
357 |
{
|
|
358 |
int i,j;
|
|
359 |
for (i=0,j=0; i!=lengthof(_screenshot_formats); i++)
|
|
360 |
if (!strcmp(_screenshot_format_name, _screenshot_formats[i].extension)) { j=i; break; }
|
|
361 |
_cur_screenshot_format = j;
|
|
362 |
_num_screenshot_formats = lengthof(_screenshot_formats);
|
|
363 |
}
|
|
364 |
|
|
365 |
const char *GetScreenshotFormatDesc(int i)
|
|
366 |
{
|
|
367 |
return _screenshot_formats[i].name;
|
|
368 |
}
|
|
369 |
|
|
370 |
void SetScreenshotFormat(int i)
|
|
371 |
{
|
|
372 |
_cur_screenshot_format = i;
|
|
373 |
strcpy(_screenshot_format_name, _screenshot_formats[i].extension);
|
|
374 |
}
|
|
375 |
|
|
376 |
// screenshot generator that dumps the current video buffer
|
|
377 |
void CurrentScreenCallback(void *userdata, byte *buf, uint y, uint pitch, uint n)
|
|
378 |
{
|
|
379 |
assert(_screen.pitch == (int)pitch);
|
|
380 |
memcpy(buf, _screen.dst_ptr + y * _screen.pitch, n * _screen.pitch);
|
|
381 |
}
|
|
382 |
|
|
383 |
extern void ViewportDoDraw(ViewPort *vp, int left, int top, int right, int bottom);
|
|
384 |
|
|
385 |
// generate a large piece of the world
|
|
386 |
void LargeWorldCallback(void *userdata, byte *buf, uint y, uint pitch, uint n)
|
|
387 |
{
|
|
388 |
ViewPort *vp = (ViewPort *)userdata;
|
|
389 |
DrawPixelInfo dpi, *old_dpi;
|
|
390 |
int wx, left;
|
|
391 |
|
|
392 |
old_dpi = _cur_dpi;
|
|
393 |
_cur_dpi = &dpi;
|
|
394 |
|
|
395 |
dpi.dst_ptr = buf;
|
|
396 |
dpi.height = n;
|
|
397 |
dpi.width = vp->width;
|
|
398 |
dpi.pitch = pitch;
|
|
399 |
dpi.zoom = 0;
|
|
400 |
dpi.left = 0;
|
|
401 |
dpi.top = y;
|
|
402 |
|
|
403 |
left = 0;
|
|
404 |
while (vp->width - left != 0) {
|
|
405 |
wx = min(vp->width - left, 1600);
|
|
406 |
left += wx;
|
|
407 |
|
|
408 |
ViewportDoDraw(vp,
|
|
409 |
((left - wx - vp->left) << vp->zoom) + vp->virtual_left,
|
|
410 |
((y - vp->top) << vp->zoom) + vp->virtual_top,
|
|
411 |
((left - vp->left) << vp->zoom) + vp->virtual_left,
|
|
412 |
(((y+n) - vp->top) << vp->zoom) + vp->virtual_top
|
|
413 |
);
|
|
414 |
}
|
|
415 |
|
|
416 |
_cur_dpi = old_dpi;
|
|
417 |
}
|
|
418 |
|
|
419 |
static char *MakeScreenshotName(const char *ext)
|
|
420 |
{
|
|
421 |
static char filename[256];
|
|
422 |
char *base;
|
|
423 |
int serial;
|
|
424 |
|
1
|
425 |
if (_game_mode == GM_EDITOR || _local_player == OWNER_SPECTATOR) {
|
0
|
426 |
sprintf(_screenshot_name, "screenshot");
|
|
427 |
} else {
|
|
428 |
Player *p = &_players[_local_player];
|
|
429 |
SET_DPARAM16(0, p->name_1);
|
|
430 |
SET_DPARAM32(1, p->name_2);
|
|
431 |
SET_DPARAM16(2, _date);
|
|
432 |
GetString(_screenshot_name, STR_4004);
|
|
433 |
}
|
|
434 |
|
|
435 |
base = strchr(_screenshot_name, 0);
|
|
436 |
base[0] = '.'; strcpy(base + 1, ext);
|
|
437 |
|
|
438 |
serial = 0;
|
|
439 |
for(;;) {
|
|
440 |
snprintf(filename, sizeof(filename), "%s%s", _path.personal_dir, _screenshot_name);
|
|
441 |
if (!FileExists(filename))
|
|
442 |
break;
|
|
443 |
sprintf(base, " #%d.%s", ++serial, ext);
|
|
444 |
}
|
|
445 |
|
|
446 |
return filename;
|
|
447 |
}
|
|
448 |
|
|
449 |
extern byte _cur_palette[768];
|
|
450 |
|
|
451 |
bool MakeScreenshot()
|
|
452 |
{
|
|
453 |
const ScreenshotFormat *sf = _screenshot_formats + _cur_screenshot_format;
|
|
454 |
return sf->proc(MakeScreenshotName(sf->extension), CurrentScreenCallback, NULL, _screen.width, _screen.height, 8, _cur_palette);
|
|
455 |
}
|
|
456 |
|
|
457 |
bool MakeWorldScreenshot(int left, int top, int width, int height, int zoom)
|
|
458 |
{
|
|
459 |
ViewPort vp;
|
|
460 |
const ScreenshotFormat *sf;
|
|
461 |
|
|
462 |
vp.zoom = zoom;
|
|
463 |
vp.left = 0;
|
|
464 |
vp.top = 0;
|
|
465 |
vp.virtual_width = width;
|
|
466 |
vp.width = width >> zoom;
|
|
467 |
vp.virtual_height = height;
|
|
468 |
vp.height = height >> zoom;
|
|
469 |
vp.virtual_left = left;
|
|
470 |
vp.virtual_top = top;
|
|
471 |
|
|
472 |
sf = _screenshot_formats + _cur_screenshot_format;
|
|
473 |
return sf->proc(MakeScreenshotName(sf->extension), LargeWorldCallback, &vp, vp.width, vp.height, 8, _cur_palette);
|
|
474 |
}
|