#include <gint/gint.h>
#include <gint/display.h>
#include <gint/keyboard.h>
#include <gint/hardware.h>
#include <gint/kmalloc.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>  // for strlen
#include <assert.h>
#include <sys/stat.h>
#include "zlib.h"
#include <libprof.h>


char *filesin[10] = { "photo0.png", "photo1.png", "photo2.png", "Conv.g3a", "PictPlot.g3a", "Afterbur.g3a", "texte0.txt", "image0.bmp", "image1.bmp", "image2.bmp" };
char *filesout[10] = { "Z_photo0.png", "Z_photo1.png", "Z_photo2.png", "Z_Conv.g3a", "Z_PictPlot.g3a", "Z_Afterbur.g3a", "Z_texte0.txt", "Z_image0.bmp", "Z_image1.bmp", "Z_image2.bmp" };
struct stat sb;

uint32_t time_compress=0, time_decompress=0;

uint32_t sizeStream;
char *inputStream = NULL;
char *outputStream = NULL;
char *secondOutputStream = NULL;

int OpenFileAndLoadStream( int which )
{
    if (inputStream!=NULL) {
        free( inputStream );
        inputStream = NULL;
    }
    if (outputStream!=NULL) {
        free(outputStream );
        outputStream = NULL;
    }
    if (secondOutputStream!=NULL) {
        free( secondOutputStream );
        secondOutputStream = NULL;
    }

    if (stat(filesin[which], &sb) == -1) {
        return -1;
    }

    sizeStream = (uint32_t) sb.st_size;

    inputStream = (char*) malloc( sb.st_size );
    if (inputStream == NULL) {
        return -2;
    }

    outputStream = (char*) malloc( sb.st_size+10000 );
    if (outputStream == NULL)
    {
        free(inputStream);
        inputStream = NULL;
        return -3;
    }

    secondOutputStream = (char*) malloc( sb.st_size );
    if (secondOutputStream == NULL)
    {
        free(outputStream);
        inputStream = NULL;
        free(outputStream);
        inputStream = NULL;
        return -4;
    }

    FILE* fp = fopen(filesin[which], "rb" );
    if (fp==NULL) {
        return -5;
    }

    fread( inputStream, sb.st_size, 1, fp );

    fclose ( fp );

    return 1;
}

int SaveOutputFile( int which )
{
    FILE* fp = fopen(filesout[which], "wb" );
    if (fp==NULL) {
        return -5;
    }

    fwrite( secondOutputStream, sizeStream, 1, fp );

    fclose ( fp );

    return 1;
}


void CloseAndFree( void )
{
    if (inputStream!=NULL) {
        free( inputStream );
        inputStream = NULL;
    }
    if (outputStream!=NULL) {
        free( outputStream );
        outputStream = NULL;
    }
    if (secondOutputStream!=NULL) {
        free( secondOutputStream );
        secondOutputStream = NULL;
    }
    exit(0);
}


bool canWeAllocate3Mb = false;
static kmalloc_arena_t extended_ram = { 0 };

void increaseRAM( void )
{
    char const *osv = (char*) 0x80020020;

    //kmalloc_gint_stats_t *extram_stats;

    if((!strncmp(osv, "03.", 3) && osv[3] <= '6') && gint[HWCALC] == HWCALC_FXCG50) // CG-50
    {
        extended_ram.name = "extram";
        extended_ram.is_default = true;
        extended_ram.start =   (void *)0x8c200000;
        extended_ram.end = (void *)0x8c500000 ;

        kmalloc_init_arena(&extended_ram, true);
        kmalloc_add_arena(&extended_ram );
        canWeAllocate3Mb = true;
    }
    else if (gint[HWCALC] == HWCALC_PRIZM)  // CG-10/20
    {

        extended_ram.name = "extram";
        extended_ram.is_default = true;

        uint16_t *vram1, *vram2;
        dgetvram(&vram1, &vram2);
        dsetvram(vram1, vram1);

        extended_ram.start = vram2;
        extended_ram.end = (char *)vram2 + 396*224*2;

        kmalloc_init_arena(&extended_ram, true);
        kmalloc_add_arena(&extended_ram );
        canWeAllocate3Mb = false;

    }
    else if (gint[HWCALC] == HWCALC_FXCG_MANAGER) // CG-50 EMULATOR
    {

        extended_ram.name = "extram";
        extended_ram.is_default = true;
        extended_ram.start =   (void *)0x88200000;
        extended_ram.end = (void *)0x88500000 ;

        kmalloc_init_arena(&extended_ram, true);
        kmalloc_add_arena(&extended_ram );
        canWeAllocate3Mb = true;

    }
    else abort();
}



int main()
{
    uint16_t *vram1, *vram2;
    dgetvram(&vram1, &vram2);
    dsetvram(vram1, vram1);

    increaseRAM();

    prof_init();
    prof_t perf_compress, perf_decompress;

    int file=9;

    int retour = (int) gint_world_switch( GINT_CALL(OpenFileAndLoadStream, file) );

    if (retour<=0) CloseAndFree();

    uint32_t ucompSize = sizeStream; // "Hello, world!" + NULL delimiter.
    uint32_t compSize = compressBound(ucompSize);
    uint32_t reucompSize = sizeStream; // "Hello, world!" + NULL delimiter.

    perf_compress = prof_make();
    prof_enter(perf_compress);
    // Deflate
    compress((Bytef *)outputStream, &compSize, (Bytef *)inputStream, ucompSize);

    prof_leave(perf_compress);
    time_compress = prof_time(perf_compress);


    perf_decompress = prof_make();
    prof_enter(perf_decompress);
    // Inflate
    uncompress((Bytef *)secondOutputStream, &reucompSize, (Bytef *)outputStream, compSize);

    prof_leave(perf_decompress);
    time_decompress = prof_time(perf_decompress);


    dclear( 0xFFFF );
    dprint(1,10,0x0000, "Uncompressed size is: %lu", ucompSize );
    dprint(1,20,0x0000, "Compressed size is: %lu", compSize );
    dprint(1,30,0x0000, "Uncompressed size is: %lu", reucompSize );

    dprint(1,40, C_GREEN, "time to compress: %lu", time_compress/1000 );
    dprint(1,50, C_GREEN, "time to decompress: %lu", time_decompress/1000 );

    dupdate();

    for( uint32_t k=0; k<sizeStream; k++ )
    {
        if (inputStream[k]!=secondOutputStream[k])
        {
                dprint(1,70,C_RED, "Error at byte %d", k );
                break;
        }
    }

    dprint(1,90,C_BLUE, "End of comparison" );
    dprint(1,100, 0x0000, "Now Saving Ouput ... " );
    dupdate();


    retour = (int) gint_world_switch( GINT_CALL(SaveOutputFile, file ) );

    dprint(1,110, 0x0000, " ... Done :-) " );
    dprint(1,130, 0x0000, "Press a key to exit" );

    dupdate();

    getkey();

    // We set back zeros at the end of the program
    memset(extended_ram.start, 0, (char *)extended_ram.end - (char *)extended_ram.start);

    return 0;
}