(*

 * Hedgewars, a free turn based strategy game

 * Copyright (c) 2004-2015 Andrey Korotaev <unC0Rr@gmail.com>

 *

 * This program 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; version 2 of the License

 *

 * 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, write to the Free Software

 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

 *)

{$INCLUDE "options.inc"}

unit uMisc;

interface

uses SDLh, uConsts, GLunit, uTypes;

procedure initModule;

procedure freeModule;

function  doSurfaceConversion(tmpsurf: PSDL_Surface): PSDL_Surface;

function MakeScreenshot(filename: shortstring; k: LongInt; dump: LongWord): boolean;

function  GetTeamStatString(p: PTeam): shortstring;

function  SDL_RectMake(x, y, width, height: LongInt): TSDL_Rect; inline;

implementation

uses uVariables, uUtils

     {$IFDEF PNG_SCREENSHOTS}, PNGh, png {$ENDIF};

type PScreenshot = ^TScreenshot;

     TScreenshot = record

         buffer: PByte;

         filename: shortstring;

         width, height: LongInt;

         size: QWord;

         end;

var conversionFormat : PSDL_PixelFormat;

{$IFDEF PNG_SCREENSHOTS}

// this funtion will be executed in separate thread

function SaveScreenshot(screenshot: pointer): LongInt; cdecl; export;

var i: LongInt;

    png_ptr: ^png_struct;

    info_ptr: ^png_info;

    f: File;

    image: PScreenshot;

begin

image:= PScreenshot(screenshot);

png_ptr := png_create_write_struct(png_get_libpng_ver(nil), nil, nil, nil);

if png_ptr = nil then

begin

    // AddFileLog('Error: Could not create png write struct.');

    SaveScreenshot:= 0;

    exit;

end;

info_ptr := png_create_info_struct(png_ptr);

if info_ptr = nil then

begin

    png_destroy_write_struct(@png_ptr, nil);

    // AddFileLog('Error: Could not create png info struct.');

    SaveScreenshot:= 0;

    exit;

end;

{$IOCHECKS OFF}

Assign(f, image^.filename);

Rewrite(f, 1);

if IOResult = 0 then

    begin

    png_init_pascal_io(png_ptr,@f);

    png_set_IHDR(png_ptr, info_ptr, image^.width, image^.height,

                 8, // bit depth

                 PNG_COLOR_TYPE_RGBA, PNG_INTERLACE_NONE,

                 PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);

    png_write_info(png_ptr, info_ptr);

    // glReadPixels and libpng number rows in different order

    for i:= image^.height-1 downto 0 do

        png_write_row(png_ptr, image^.buffer + i*4*image^.width);

    png_write_end(png_ptr, info_ptr);

    Close(f);

    end;

{$IOCHECKS ON}

// free everything

png_destroy_write_struct(@png_ptr, @info_ptr);

FreeMem(image^.buffer, image^.size);

Dispose(image);

SaveScreenshot:= 0;

end;

{$ELSE} //PNG_SCREENSHOTS

{$IFDEF WINDOWS}

function SaveScreenshot(screenshot: pointer): LongInt; cdecl; export;

var

    surface: PSDL_Surface;

    image: PScreenshot;

    row, stride: LongInt;

begin

    image:= PScreenshot(screenshot);

    stride:= image^.width * 4;

        Move((image^.buffer + row * stride)^,

             stride);

    surface:= SDL_CreateRGBSurfaceFrom(

        $000000FF, $0000FF00, $00FF0000, $FF000000);

    if surface <> nil then

        begin

        IMG_SavePNG(surface, Str2PChar(image^.filename));

        SDL_FreeSurface(surface);

        end;

    FreeMem(image^.buffer, image^.size);

    Dispose(image);

    SaveScreenshot:= 0;

end;

{$ELSE} //WINDOWS

// this funtion will be executed in separate thread

function SaveScreenshot(screenshot: pointer): LongInt; cdecl; export;

var f: file;

    // Windows Bitmap Header

    head: array[0..53] of Byte = (

    $42, $4D,       // identifier ("BM")

    0, 0, 0, 0,     // file size

    0, 0, 0, 0,     // reserved

    54, 0, 0, 0,    // starting offset

    40, 0, 0, 0,    // header size

    0, 0, 0, 0,     // width

    0, 0, 0, 0,     // height

    1, 0,           // color planes

    32, 0,          // bit depth

    0, 0, 0, 0,     // compression method (uncompressed)

    0, 0, 0, 0,     // image size

    96, 0, 0, 0,    // horizontal resolution

    96, 0, 0, 0,    // vertical resolution

    0, 0, 0, 0,     // number of colors (all)

    0, 0, 0, 0      // number of important colors

    );

    image: PScreenshot;

    size: QWord;

    writeResult:LongInt;

begin

image:= PScreenshot(screenshot);

size:= image^.Width*image^.Height*4;

head[$02]:= (size + 54) and $ff;

head[$03]:= ((size + 54) shr 8) and $ff;

head[$04]:= ((size + 54) shr 16) and $ff;

head[$05]:= ((size + 54) shr 24) and $ff;

head[$12]:= image^.Width and $ff;

head[$13]:= (image^.Width shr 8) and $ff;

head[$14]:= (image^.Width shr 16) and $ff;

head[$15]:= (image^.Width shr 24) and $ff;

head[$16]:= image^.Height and $ff;

head[$17]:= (image^.Height shr 8) and $ff;

head[$18]:= (image^.Height shr 16) and $ff;

head[$19]:= (image^.Height shr 24) and $ff;

head[$22]:= size and $ff;

head[$23]:= (size shr 8) and $ff;

head[$24]:= (size shr 16) and $ff;

head[$25]:= (size shr 24) and $ff;

{$IOCHECKS OFF}

Assign(f, image^.filename);

Rewrite(f, 1);

if IOResult = 0 then

    begin

    writeResult:= 0; // suppress fpc hint

    BlockWrite(f, head, sizeof(head), writeResult);

    BlockWrite(f, image^.buffer^, size, writeResult);

    Close(f);

    end

else

    begin

    //AddFileLog('Error: Could not write to ' + filename);

    end;

{$IOCHECKS ON}

// free everything

FreeMem(image^.buffer, image^.size);

Dispose(image);

SaveScreenshot:= 0;

end;

{$ENDIF} // WINDOWS

{$ENDIF} // PNG_SCREENSHOTS

{$IFDEF USE_VIDEO_RECORDING}

// make image k times smaller (useful for saving thumbnails)

procedure ReduceImage(img: PByteArray; width, height, k: LongInt);

var i, j, i0, j0, w, h, r, g, b: LongInt;

begin

    w:= width  div k;

    h:= height div k;

    // rescale inplace

    if k <> 1 then

    begin

        for i:= 0 to h-1 do

            for j:= 0 to w-1 do

            begin

                r:= 0;

                g:= 0;

                b:= 0;

                for i0:= 0 to k-1 do

                    for j0:= 0 to k-1 do

                    begin

                        inc(r, img^[4*(width*(i*k+i0) + j*k + j0)+0]);

                        inc(g, img^[4*(width*(i*k+i0) + j*k + j0)+1]);

                        inc(b, img^[4*(width*(i*k+i0) + j*k + j0)+2]);

                    end;

                img^[4*(w*i + j)+0]:= r div (k*k);

                img^[4*(w*i + j)+1]:= g div (k*k);

                img^[4*(w*i + j)+2]:= b div (k*k);

                img^[4*(w*i + j)+3]:= 255;

            end;

    end;

end;

{$ENDIF}

// captures and saves the screen. returns true on success.

// saved image will be k times smaller than original (useful for saving thumbnails).

function MakeScreenshot(filename: shortstring; k: LongInt; dump: LongWord): boolean;

var p: Pointer;

    size: QWord;

    image: PScreenshot;

    format: GLenum;

    ext: string[4];

    x,y: LongWord;

begin

{$IFDEF PNG_SCREENSHOTS}

format:= GL_RGBA;

ext:= '.png';

{$ELSE}

{$IFDEF WINDOWS}

format:= GL_RGBA;

ext:= '.png';

{$ELSE}

format:= GL_BGRA;

ext:= '.bmp';

{$ENDIF}

{$ENDIF}

if dump > 0 then

     size:= LAND_WIDTH*LAND_HEIGHT*4

else size:= toPowerOf2(cScreenWidth) * toPowerOf2(cScreenHeight) * 4;

p:= GetMem(size); // will be freed in SaveScreenshot()

// memory could not be allocated

if p = nil then

begin

    AddFileLog('Error: Could not allocate memory for screenshot.');

    MakeScreenshot:= false;

    exit;

end;

// read pixels from land array

if dump > 0 then

    begin

    for y:= 0 to LAND_HEIGHT-1 do

        for x:= 0 to LAND_WIDTH-1 do

            if dump = 2 then

                PLongWordArray(p)^[y*LAND_WIDTH+x]:= LandPixels[LAND_HEIGHT-1-y, x]

            else

                begin

                if Land[LAND_HEIGHT-1-y, x] and lfIndestructible = lfIndestructible then

                    PLongWordArray(p)^[y*LAND_WIDTH+x]:= (AMask or RMask)

                else if Land[LAND_HEIGHT-1-y, x] and lfIce = lfIce then

                    PLongWordArray(p)^[y*LAND_WIDTH+x]:= (AMask or BMask)

                else if Land[LAND_HEIGHT-1-y, x] and lfBouncy = lfBouncy then

                    PLongWordArray(p)^[y*LAND_WIDTH+x]:= (AMask or GMask)

                else if Land[LAND_HEIGHT-1-y, x] and lfObject = lfObject then

                    PLongWordArray(p)^[y*LAND_WIDTH+x]:= $FFFFFFFF

                else if Land[LAND_HEIGHT-1-y, x] and lfBasic = lfBasic then

                    PLongWordArray(p)^[y*LAND_WIDTH+x]:= AMask

                else

                    PLongWordArray(p)^[y*LAND_WIDTH+x]:= 0

                end

    end

else

// read pixels from the front buffer

    begin

    glReadPixels(0, 0, cScreenWidth, cScreenHeight, format, GL_UNSIGNED_BYTE, p);

{$IFDEF USE_VIDEO_RECORDING}

    ReduceImage(p, cScreenWidth, cScreenHeight, k)

{$ENDIF}

    end;

// allocate and fill structure that will be passed to new thread

New(image); // will be disposed in SaveScreenshot()

if dump = 2 then

     image^.filename:= shortstring(UserPathPrefix) + filename + '_landpixels' + ext

else if dump = 1 then

     image^.filename:= shortstring(UserPathPrefix) + filename + '_land' + ext

else image^.filename:= shortstring(UserPathPrefix) + filename + ext;

if dump <> 0 then

    begin

    image^.width:= LAND_WIDTH;

    image^.height:= LAND_HEIGHT

    end

else

    begin

    image^.width:= cScreenWidth div k;

    image^.height:= cScreenHeight div k

    end;

image^.size:= size;

image^.buffer:= p;

SDL_CreateThread(@SaveScreenshot, PChar('snapshot'), image);

MakeScreenshot:= true; // possibly it is not true but we will not wait for thread to terminate

end;

// http://www.idevgames.com/forums/thread-5602-post-21860.html#pid21860

function doSurfaceConversion(tmpsurf: PSDL_Surface): PSDL_Surface;

var convertedSurf: PSDL_Surface;

begin

    doSurfaceConversion:= tmpsurf;

    if ((tmpsurf^.format^.bitsperpixel = 32) and (tmpsurf^.format^.rshift > tmpsurf^.format^.bshift)) or

       (tmpsurf^.format^.bitsperpixel = 24) then

    begin

        convertedSurf:= SDL_ConvertSurface(tmpsurf, conversionFormat, SDL_SWSURFACE);

        SDL_FreeSurface(tmpsurf);

        doSurfaceConversion:= convertedSurf;

    end;

end;

function SDL_RectMake(x, y, width, height: LongInt): TSDL_Rect; inline;

begin

    SDL_RectMake.x:= x;

    SDL_RectMake.y:= y;

    SDL_RectMake.w:= width;

    SDL_RectMake.h:= height;

end;

function GetTeamStatString(p: PTeam): shortstring;

var s: shortstring;

begin

    s:= p^.TeamName + ':' + IntToStr(p^.TeamHealth) + ':';

    GetTeamStatString:= s;

end;

procedure initModule;

const SDL_PIXELFORMAT_ABGR8888 = (1 shl 28) or (6 shl 24) or (7 shl 20) or (6 shl 16) or (32 shl 8) or 4;

begin

    conversionFormat:= SDL_AllocFormat(SDL_PIXELFORMAT_ABGR8888);

end;

procedure freeModule;

begin

    SDL_FreeFormat(conversionFormat);

end;

end.