#include "base/mem.h"
#include "gtest/gtest.h"
#include "tools/common/utils.h"
#include "base/image_manager.h"
#include "base/assets_manager.h"
#include "tools/image_gen/image_gen.h"
#include "image_loader/image_loader_stb.h"

#define PNG_NAME TK_ROOT "/tests/testdata/test.png"
#define JPG_NAME TK_ROOT "/tests/testdata/test.jpg"
#define PNG_OPAQUE_NAME TK_ROOT "/tests/testdata/test_opaque.png"

static ret_t load_image(const char* filename, bitmap_t* image) {
  uint32_t size = 0;
  ret_t ret = RET_OK;
  image_loader_t* loader = image_loader_stb();
  printf("%s\n", filename);
  uint8_t* buff = (uint8_t*)read_file(filename, &size);
  ret = image_loader_load(loader, buff, size, image);
  TKMEM_FREE(buff);

  return ret;
}

TEST(ImageLoaderStb, basic) {
  bitmap_t image;
  memset(&image, 0x00, sizeof(image));

  ret_t ret = load_image(PNG_NAME, &image);

  ASSERT_EQ(ret, RET_OK);
  ASSERT_EQ(32, image.w);
  ASSERT_EQ(32, image.h);
  ASSERT_EQ(!!(image.flags & BITMAP_FLAG_IMMUTABLE), true);
  ASSERT_EQ(!!(image.flags & BITMAP_FLAG_OPAQUE), false);
  bitmap_destroy(&image);

  ret = load_image(JPG_NAME, &image);
  ASSERT_EQ(ret, RET_OK);
  ASSERT_EQ(32, image.w);
  ASSERT_EQ(32, image.h);
  ASSERT_EQ(!!(image.flags & BITMAP_FLAG_IMMUTABLE), true);
  ASSERT_EQ(!!(image.flags & BITMAP_FLAG_OPAQUE), true);
  bitmap_destroy(&image);

  ret = load_image(PNG_OPAQUE_NAME, &image);
  ASSERT_EQ(ret, RET_OK);
  ASSERT_EQ(32, image.w);
  ASSERT_EQ(32, image.h);
  ASSERT_EQ(!!(image.flags & BITMAP_FLAG_IMMUTABLE), true);
  ASSERT_EQ(!!(image.flags & BITMAP_FLAG_OPAQUE), true);
  bitmap_destroy(&image);
}

static ret_t add_image_res(const char* filename, const char* name) {
  bitmap_t image;
  static uint8_t buff[8092];
  memset(&image, 0x00, sizeof(image));
  ret_t ret = load_image(filename, &image);
  asset_info_t* r = (asset_info_t*)buff;
  return_value_if_fail(ret == RET_OK, RET_FAIL);

  strcpy(r->name, name);
  r->is_in_rom = TRUE;
  r->type = ASSET_TYPE_IMAGE;
  r->subtype = ASSET_TYPE_IMAGE_RAW;
  r->size = image_gen_buff(&image, r->data, sizeof(buff) - sizeof(asset_info_t));
  bitmap_destroy(&image);

  return assets_manager_add(assets_manager(), buff);
}

TEST(ImageLoaderStb, gen) {
  bitmap_t image;
  ASSERT_EQ(add_image_res(PNG_OPAQUE_NAME, "test.png"), RET_OK);
  ASSERT_EQ(image_manager_load(image_manager(), "test.png", &image), RET_OK);

  ASSERT_EQ(32, image.w);
  ASSERT_EQ(32, image.h);
  ASSERT_EQ(!!(image.flags & BITMAP_FLAG_IMMUTABLE), true);
  ASSERT_EQ(!!(image.flags & BITMAP_FLAG_OPAQUE), true);
  image_manager_unload_unused(image_manager(), 0);
}