#include "Render/Primitives/RenderPrimitivePool.h"
#include "Render/Primitives/UIRPData.h"
#include "Render/Textures/Texture.h"
#include "Render/Textures/TextureCache.h"

// OFFSET: 0x008130A0
// FUNCTION: fn_10121
void UIRPData::DrawStretched(RenderPrimitivePool& pool, int x, int y, CTexture* texture, std::uint32_t color, int width, int height) {
  if (!texture) {
    return;
  }

  RenderPrimitive* primitive = pool.Acquire();
  primitive->texture = texture;
  const int right = x + width;
  const int bottom = y + height;
  primitive->quad[0].x = static_cast<float>(x);
  primitive->quad[0].y = static_cast<float>(y);
  primitive->quad[0].u = g_textureOffsetX / static_cast<float>(texture->TextureWidth());
  primitive->quad[0].v = g_textureOffsetY / static_cast<float>(texture->TextureHeight());
  primitive->quad[1].x = static_cast<float>(right);
  primitive->quad[1].y = static_cast<float>(y);
  primitive->quad[1].u = (static_cast<float>(texture->Width()) + g_textureOffsetX) / static_cast<float>(texture->TextureWidth());
  primitive->quad[1].v = g_textureOffsetY / static_cast<float>(texture->TextureHeight());
  primitive->quad[2].x = static_cast<float>(x);
  primitive->quad[2].y = static_cast<float>(bottom);
  primitive->quad[2].u = g_textureOffsetX / static_cast<float>(texture->TextureWidth());
  primitive->quad[2].v = (static_cast<float>(texture->Height()) + g_textureOffsetX) / static_cast<float>(texture->TextureHeight());
  primitive->quad[3].x = static_cast<float>(right);
  primitive->quad[3].y = static_cast<float>(bottom);
  primitive->quad[3].u = (static_cast<float>(texture->Width()) + g_textureOffsetX) / static_cast<float>(texture->TextureWidth());
  primitive->quad[3].v = (static_cast<float>(texture->Height()) + g_textureOffsetX) / static_cast<float>(texture->TextureHeight());
  for (RenderVertex& vertex : primitive->quad) {
    vertex.diffuse = color;
  }

  for (RenderVertex& vertex : primitive->quad) {
    vertex.z = 0.000001f;
    vertex.rhw = 0.999999f;
  }

  this->m_primitives.push_back(primitive);
}

// OFFSET: 0x008134B0
// FUNCTION: fn_10122
void UIRPData::DrawNamed(RenderPrimitivePool& pool, int x, int y, std::string name, std::uint32_t color, int width, int height) {
  CTexture* texture = g_texMgr.Load(name.c_str(), 0, true, true, false);
  if (!texture) {
    return;
  }

  RenderPrimitive* primitive = pool.Acquire();
  primitive->texture = texture;
  const int imageWidth = texture->Width();
  const int imageHeight = texture->Height();
  if (!width) {
    width = texture->Width();
  }

  if (!height) {
    height = texture->Height();
  }

  const int textureWidth = texture->TextureWidth();
  const int textureHeight = texture->TextureHeight();
  primitive->quad[0].x = static_cast<float>(x);
  primitive->quad[0].y = static_cast<float>(y);
  primitive->quad[0].u = g_textureOffsetX / static_cast<float>(textureWidth);
  primitive->quad[0].v = g_textureOffsetY / static_cast<float>(textureHeight);
  primitive->quad[1].x = static_cast<float>(x + width);
  primitive->quad[1].y = static_cast<float>(y);
  if (width > imageWidth) {
    primitive->quad[1].u = (static_cast<float>(imageWidth) + g_textureOffsetX) / static_cast<float>(textureWidth);
  } else {
    primitive->quad[1].u = (static_cast<float>(width) + g_textureOffsetX) / static_cast<float>(textureWidth);
  }

  primitive->quad[1].v = g_textureOffsetY / static_cast<float>(textureHeight);
  primitive->quad[2].x = static_cast<float>(x);
  primitive->quad[2].y = static_cast<float>(y + height);
  primitive->quad[2].u = g_textureOffsetX / static_cast<float>(textureWidth);
  if (height > imageHeight) {
    primitive->quad[2].v = (static_cast<float>(imageHeight) + g_textureOffsetY) / static_cast<float>(textureHeight);
  } else {
    primitive->quad[2].v = (static_cast<float>(height) + g_textureOffsetX) / static_cast<float>(textureHeight);
  }

  primitive->quad[3].x = static_cast<float>(x + width);
  primitive->quad[3].y = static_cast<float>(y + height);
  if (width > imageWidth) {
    primitive->quad[3].u = (static_cast<float>(imageWidth) + g_textureOffsetX) / static_cast<float>(textureWidth);
  } else {
    primitive->quad[3].u = (static_cast<float>(width) + g_textureOffsetX) / static_cast<float>(textureWidth);
  }

  if (height > imageHeight) {
    primitive->quad[3].v = (static_cast<float>(imageHeight) + g_textureOffsetY) / static_cast<float>(textureHeight);
  } else {
    primitive->quad[3].v = (static_cast<float>(height) + g_textureOffsetX) / static_cast<float>(textureHeight);
  }

  for (int i = 0; i < 4; ++i) {
    primitive->quad[i].diffuse = color;
  }

  for (int i = 0; i < 4; ++i) {
    primitive->quad[i].z = 0.000001f;
    primitive->quad[i].rhw = 0.999999f;
  }

  this->m_primitives.push_back(primitive);
}

// OFFSET: 0x00813A00
// FUNCTION: fn_10123
void UIRPData::SetSolid(RenderPrimitivePool& pool, int width, int height, std::uint32_t color) {
  this->m_primitives.clear();

  RenderPrimitive* primitive = pool.Acquire();
  primitive->texture = nullptr;
  primitive->quad[0].x = 0.0f;
  primitive->quad[0].y = 0.0f;
  primitive->quad[1].x = static_cast<float>(width);
  primitive->quad[1].y = 0.0f;
  primitive->quad[2].x = 0.0f;
  primitive->quad[2].y = static_cast<float>(height);
  primitive->quad[3].x = static_cast<float>(width);
  primitive->quad[3].y = static_cast<float>(height);
  for (RenderVertex& vertex : primitive->quad) {
    vertex.u = 0.0f;
    vertex.v = 0.0f;
    vertex.z = 0.000001f;
    vertex.rhw = 0.999999f;
    vertex.diffuse = color;
  }

  this->m_primitives.push_back(primitive);
}