#include "api.h"
#include "data/data.h"
#include "data/audioStream.h"
#include "data/modelData.h"
#include "data/rasterizer.h"
#include "data/textureData.h"

int l_lovrDataInit(lua_State* L) {
  lua_newtable(L);
  luaL_register(L, NULL, lovrData);
  luax_registertype(L, "AudioStream", lovrAudioStream);
  luax_registertype(L, "ModelData", lovrModelData);
  luax_registertype(L, "Rasterizer", lovrRasterizer);
  luax_registertype(L, "TextureData", lovrTextureData);
  luax_registertype(L, "VertexData", lovrVertexData);
  return 1;
}

int l_lovrDataNewAudioStream(lua_State* L) {
  Blob* blob = luax_readblob(L, 1, "Sound");
  size_t bufferSize = luaL_optinteger(L, 2, 4096);
  AudioStream* stream = lovrAudioStreamCreate(blob, bufferSize);
  luax_pushtype(L, AudioStream, stream);
  lovrRelease(&blob->ref);
  lovrRelease(&stream->ref);
  return 1;
}

int l_lovrDataNewModelData(lua_State* L) {
  Blob* blob = luax_readblob(L, 1, "Model");
  ModelData* modelData = lovrModelDataCreate(blob);
  luax_pushtype(L, ModelData, modelData);
  lovrRelease(&blob->ref);
  lovrRelease(&modelData->ref);
  return 1;
}

int l_lovrDataNewRasterizer(lua_State* L) {
  Blob* blob = NULL;
  float size;

  if (lua_type(L, 1) == LUA_TNUMBER || lua_isnoneornil(L, 1)) {
    size = luaL_optnumber(L, 1, 32);
  } else {
    blob = luax_readblob(L, 1, "Font");
    size = luaL_optnumber(L, 2, 32);
  }

  Rasterizer* rasterizer = lovrRasterizerCreate(blob, size);
  luax_pushtype(L, Rasterizer, rasterizer);

  if (blob) {
    lovrRelease(&blob->ref);
  }

  lovrRelease(&rasterizer->ref);
  return 1;
}

int l_lovrDataNewTextureData(lua_State* L) {
  TextureData* textureData = NULL;
  if (lua_type(L, 1) == LUA_TNUMBER) {
    int width = luaL_checknumber(L, 1);
    int height = luaL_checknumber(L, 2);
    textureData = lovrTextureDataGetBlank(width, height, 0x0, FORMAT_RGBA);
  } else {
    Blob* blob = luax_readblob(L, 1, "Texture");
    textureData = lovrTextureDataFromBlob(blob);
    lovrRelease(&blob->ref);
  }

  luax_pushtype(L, TextureData, textureData);
  lovrRelease(&textureData->ref);
  return 1;
}

int l_lovrDataNewVertexData(lua_State* L) {
  uint32_t count = luaL_checkinteger(L, 1);
  VertexFormat format;
  vertexFormatInit(&format);
  luax_checkvertexformat(L, 2, &format);
  VertexData* vertexData = lovrVertexDataCreate(count, format.count > 0 ? &format : NULL, true);
  luax_pushtype(L, VertexData, vertexData);
  lovrRelease(&vertexData->ref);
  return 1;
}

const luaL_Reg lovrData[] = {
  { "newAudioStream", l_lovrDataNewAudioStream },
  { "newModelData", l_lovrDataNewModelData },
  { "newRasterizer", l_lovrDataNewRasterizer },
  { "newTextureData", l_lovrDataNewTextureData },
  { "newVertexData", l_lovrDataNewVertexData },
  { NULL, NULL }
};