#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdint.h>
#include "sdkconfig.h"
#include "unity.h"

// Testing ROM code is only useful during FPGA testing and chip bringup
#if CONFIG_IDF_ENV_FPGA

// compression/decompression will take off a bunch of memory
// test it only with PSRAM enabled
#ifdef CONFIG_SPIRAM

#if !TEMPORARY_DISABLED_FOR_TARGETS(ESP32)
// miniz unit test can't pass on ESP32 non-ECO3 version IDF-1807

#if CONFIG_IDF_TARGET_ESP32
#include "esp32/rom/miniz.h"
#elif CONFIG_IDF_TARGET_ESP32S2
#include "esp32s2/rom/miniz.h"
#else
#error "unsupported target"
#endif


#define DATASIZE (1024 * 64)

TEST_CASE("Test miniz compression/decompression", "[rom][miniz]")
{
    size_t inbytes = 0, outbytes = 0, inpos = 0, outpos = 0, compsz;
    printf("Allocating data buffer and filling it with semi-random data\r\n");
    uint8_t *inbuf = calloc(1, DATASIZE);
    TEST_ASSERT_NOT_NULL_MESSAGE(inbuf, "allocate input buffer failed");

    srand(0); // semi random data
    for (int i = 0; i < DATASIZE; i++) {
        inbuf[i] = (i & 0x01) ? rand() & 0xff : 0;
    }

    printf("Allocating compressor\r\n");
    tdefl_compressor *comp = calloc(1, sizeof(tdefl_compressor));
    TEST_ASSERT_NOT_NULL_MESSAGE(comp, "allocate tdefl_compressor failed");

    uint8_t *outbuf = calloc(1, DATASIZE);
    TEST_ASSERT_NOT_NULL_MESSAGE(outbuf, "allocate output buffer failed");

    printf("Compressing...\r\n");
    tdefl_status comp_status = tdefl_init(comp, NULL, NULL, TDEFL_WRITE_ZLIB_HEADER | 1500);
    TEST_ASSERT_EQUAL_MESSAGE(TDEFL_STATUS_OKAY, comp_status, "tdefl_init failed");

    while (1) {
        outbytes = DATASIZE - outpos;
        inbytes = DATASIZE - inpos;
        comp_status = tdefl_compress(comp, &inbuf[inpos], &inbytes, &outbuf[outpos], &outbytes, TDEFL_FINISH);
        inpos += inbytes;
        outpos += outbytes;
        printf("...Compressed %d into %d bytes\r\n", inpos, outpos);
        if (comp_status == TDEFL_STATUS_DONE) {
            break;
        } else if (comp_status != TDEFL_STATUS_OKAY) {
            TEST_ASSERT_MESSAGE(0, "tdefl_compress failed");
        }
    }

    compsz = outpos;
    free(comp);
    free(inbuf);

    inbuf = outbuf;
    outbuf = calloc(1, DATASIZE);
    TEST_ASSERT_NOT_NULL_MESSAGE(outbuf, "allocate output buffer failed");

    printf("Decompressing...\r\n");
    tinfl_decompressor *decomp = calloc(1, sizeof(tinfl_decompressor));
    TEST_ASSERT_NOT_NULL_MESSAGE(decomp, "allocate tinfl_decompressor failed");
    tinfl_init(decomp);

    inpos = 0;
    outpos = 0;
    while (1) {
        outbytes = DATASIZE - outpos;
        inbytes = compsz - inpos;
        tinfl_status decomp_status = tinfl_decompress(decomp, &inbuf[inpos], &inbytes, outbuf, &outbuf[outpos], &outbytes, TINFL_FLAG_PARSE_ZLIB_HEADER);
        inpos += inbytes;
        outpos += outbytes;
        printf("...Decompressed %d into %d bytes\r\n", inpos, outpos);
        if (decomp_status == TINFL_STATUS_DONE) {
            break;
        } else if (decomp_status < TINFL_STATUS_DONE) {
            printf("decomp status=%d\r\n", decomp_status);
            TEST_ASSERT_MESSAGE(0, "tinfl_decompress failed");
        }
    }

    printf("Verifying data between compression and decompression...\r\n");
    srand(0); // semi random data
    for (int i = 0; i < DATASIZE; i++) {
        uint8_t original = (i & 1) ? rand() & 0xff : 0;
        TEST_ASSERT_EQUAL_MESSAGE(original, outbuf[i], "data after decompression doesn't match the original one");
    }
    printf("Great Success!\n");
    free(inbuf);
    free(outbuf);
    free(decomp);
}

#endif //#if !TEMPORARY_DISABLED_FOR_TARGETS(ESP32)
#endif // CONFIG_SPIRAM
#endif // CONFIG_IDF_ENV_FPGA