#include "img.ldr.tga.inc.c"

#ifdef PNG_LIBPNG_VER_STRING

#include "img.ldr.png.inc.c"

#endif

typedef int (*ImageLoadFunc)(const char *path, int *outWidth, int *outHeight, unsigned char **pixels);

typedef int (*ImageSaveFunc)(const char *path, int width, int height, const unsigned char *pixels);

int checkExt(const char *path, const char *ext) {

  int lp = strlen(path);

  int le = strlen(ext);

  if (lp <= le) return FALSE;

  if (path[lp-le-1] != '.') return FALSE;

  for(const char *a = path+lp-le, *b = ext; *b; ++a, ++b)

    if (tolower(*a) != tolower(*b)) return FALSE;

  return TRUE;

}

int imageLoadAuto(const char *path, int *outWidth, int *outHeight, unsigned char **pixels) {

  struct {

    const char *ext;

    ImageLoadFunc func;

  } formats[] = {

    { "tga", &imageLoadTga },

    #if defined(HELI_HELIANTUS_H)

    { "png", &imageLoad },

    #elif defined(PNG_LIBPNG_VER_STRING)

    { "png", &imageLoadPng },

    #endif

  };

  int cnt = sizeof(formats)/sizeof(*formats);

  for(int i = 0; i < cnt; ++i)

    if (checkExt(path, formats[i].ext))

      return formats[i].func(path, outWidth, outHeight, pixels);

  fprintf(stderr, "Cannot load this image format (supported formats:");

  for(int i = 0; i < cnt; ++i)

    fprintf(stderr, " %s", formats[i].ext);

  fprintf(stderr, "): %s\n", path);

  fflush(stderr);

  return FALSE;

}

int imageSaveAuto(const char *path, int width, int height, const unsigned char *pixels) {

  struct {

    const char *ext;

    ImageSaveFunc func;

  } formats[] = {

    { "tga", &imageSaveTga },

    #if defined(HELI_HELIANTUS_H)

    { "png", (ImageSaveFunc)&imageSave },

    #elif defined(PNG_LIBPNG_VER_STRING)

    { "png", &imageSavePng },

    #endif

  };

  int cnt = sizeof(formats)/sizeof(*formats);

  for(int i = 0; i < cnt; ++i)

    if (checkExt(path, formats[i].ext))

      return formats[i].func(path, width, height, pixels);

  fprintf(stderr, "Cannot save this image format (supported formats:");

  for(int i = 0; i < cnt; ++i)

    fprintf(stderr, " %s", formats[i].ext);

  fprintf(stderr, "): %s\n", path);

  fflush(stderr);

  return FALSE;

}

int imgLoad(Img *img, const char *path) {

  imgDestroy(img);

  int w, h;

  unsigned char *data;

  if (!imageLoadAuto(path, &w, &h, &data))

    { free(data); return FALSE; }

  imgFromInt(img, w, h, data);

  free(data);

  return TRUE;

}

int imgSave(Img *img, const char *path) {

  unsigned char *data = imgToInt(img);

  if (!imageSaveAuto(path, img->w, img->h, data))

    { free(data); return FALSE; }

  free(data);

  return TRUE;

}