#include "Contents/Configuration/ServiceConfiguration.h"
#include "Render/DeviceContext/DCSurface.h"
#include "Render/Primitives/RenderPrimitivePool.h"
#include "Render/Renderer.h"
#include "Render/Textures/SpriteAtlas.h"
#include "Render/Textures/SpritePixels.h"
#include "Render/Textures/Texture.h"
#include "Resource/ResourceMgr.h"
#include "UI/Window/UIWindow.h"
#include "Utility/Math/AffineMatrix.h"

// OFFSET: 0x00A7A0C0
// FUNCTION: fn_16922
void UIWindow::AddSpriteCellOverlay(
  int x, int y, int, int, const SpriteRleImage* image, const SpriteFrameCell* cell, float scale, float angle, std::uint32_t color,
  const std::uint32_t* palette, float extentScale
) {
  x += this->geometry.x;
  y += this->geometry.y;

  SpriteAtlasRecord record{};
  RenderPrimitive* primitive = g_renderer->primitives.Acquire();

  float left = static_cast<float>(cell->x) * scale * cell->scaleX;
  float right = static_cast<float>((image->unknown_0004 + 1) * image->width + cell->x - 1) * scale * cell->scaleX * extentScale;
  float top = static_cast<float>(cell->y) * scale * cell->scaleY;
  float bottom = static_cast<float>((image->unknown_0006 + 1) * image->height - 1 + cell->y) * scale * cell->scaleY * extentScale;

  CTexture* texture;
  if (cell->imageType == 0) {
    texture = g_renderer->atlas.Find(image, palette, record);
    if (!texture) {
      texture = g_renderer->atlas.Insert(*GetRenderSystemHolder()->system, image, palette, record, g_serviceType, g_resourceReuseDelay);
    }
  } else {
    texture = static_cast<CTexture*>(const_cast<void*>(image->unknown_0008));
    record.u1 = static_cast<float>(image->width) / static_cast<float>(texture->TextureWidth());
    record.v1 = static_cast<float>(image->height) / static_cast<float>(texture->TextureHeight());
  }
  primitive->texture = texture;

  RenderVertex* vertices = primitive->quad;
  if (angle != 0.0f) {
    const float cosine = ClientCosine(static_cast<int>(angle));
    const float sine = ClientSine(static_cast<int>(angle));
    const int farX = (image->unknown_0004 + 1) * image->width + cell->x;
    const float startX = static_cast<float>((cell->mirrored & 1) ? farX : cell->x) * scale + static_cast<float>(x);
    const float endX = static_cast<float>((cell->mirrored & 1) ? cell->x : farX) * scale + static_cast<float>(x) + 1.0f;
    const float upper = static_cast<float>(cell->y) * scale + static_cast<float>(y);
    const float lower = static_cast<float>((image->unknown_0006 + 1) * image->height + cell->y) * scale + static_cast<float>(y) + 1.0f;
    const float halfWidth = (endX - startX) * 0.5f;
    const float halfHeight = (lower - upper) * 0.5f;
    const float centerX = startX + halfWidth;
    const float centerY = upper + halfHeight;
    const float minX = -halfWidth;
    const float minY = -halfHeight;

    vertices[0].x = minX * cosine - minY * sine + centerX + 0.5f;
    vertices[0].y = minY * cosine + minX * sine + centerY + 0.5f;
    vertices[1].x = halfWidth * cosine - minY * sine + centerX + 0.5f;
    vertices[1].y = minY * cosine + halfWidth * sine + centerY + 0.5f;
    vertices[2].x = minX * cosine - halfHeight * sine + centerX + 0.5f;
    vertices[2].y = halfHeight * cosine + minX * sine + centerY + 0.5f;
    vertices[3].x = halfWidth * cosine - halfHeight * sine + centerX + 0.5f;
    vertices[3].y = halfHeight * cosine + halfWidth * sine + centerY + 0.5f;
  } else {
    if (cell->mirrored & 1) {
      const float swap = left;
      left = right;
      right = swap;
    }
    left += static_cast<float>(x);
    right += static_cast<float>(x);
    top += static_cast<float>(y);
    bottom += static_cast<float>(y);

    vertices[0].x = left + 0.5f;
    vertices[0].y = top + 0.5f;
    vertices[1].x = right + 0.5f;
    vertices[1].y = top + 0.5f;
    vertices[2].x = left + 0.5f;
    vertices[2].y = bottom + 0.5f;
    vertices[3].x = right + 0.5f;
    vertices[3].y = bottom + 0.5f;
  }

  for (RenderVertex& vertex : primitive->quad) {
    vertex.z = 0.000001f;
    vertex.rhw = 0.999999f;
    vertex.diffuse = color;
  }
  vertices[0].u = record.u0;
  vertices[0].v = record.v0;
  vertices[1].u = record.u1;
  vertices[1].v = record.v0;
  vertices[2].u = record.u0;
  vertices[2].v = record.v1;
  vertices[3].u = record.u1;
  vertices[3].v = record.v1;

  this->m_overlayPrimitives.push_back(primitive);
}