import std/os
import std/algorithm
import std/strutils
import std/sequtils
import std/monotimes
import std/times
import std/tables
import std/options
import std/random
import std/unicode

import ../semicongine
import ../semicongine/rendering
import ../semicongine/text
import ../semicongine/input
import ../semicongine/fonts

const MAX_CODEPOINTS = 200
const FONTNAME = "Overhaul.ttf"
# const FONTNAME = "DejaVuSans.ttf"

proc test_01_static_label(time: float32, renderPass: RenderPass) =
  var font = loadFont[MAX_CODEPOINTS](FONTNAME, lineHeightPixels = 200)
  var renderdata = initRenderData()
  var pipeline = createPipeline(GlyphShader[MAX_CODEPOINTS](), renderPass = renderPass)
  var textbuffer = font.initTextBuffer(1000, renderdata, baseScale = 0.1)

  font.upload(renderdata)
  font.addToPipeline(renderdata, pipeline)

  # small drop-shadow
  discard textbuffer.add(
    "Hello semicongine!",
    vec3(0.009, -0.009 * getAspectRatio(), 0.2),
    color = vec4(0.02, 0.02, 0.02, 1),
    scale = 1.01,
  )
  discard textbuffer.add("Hello semicongine!", vec3(0, 0, 0.1))

  var start = getMonoTime()
  while ((getMonoTime() - start).inMilliseconds().int / 1000) < time:
    let t = getMonoTime()
    if windowWasResized():
      textbuffer.refresh()

    withNextFrame(framebuffer, commandbuffer):
      font.bindTo(pipeline, commandbuffer)
      withRenderPass(
        renderPass,
        framebuffer,
        commandbuffer,
        frameWidth(),
        frameHeight(),
        vec4(0, 0, 0, 0),
      ):
        withPipeline(commandbuffer, pipeline):
          renderTextBuffer(commandbuffer, pipeline, textbuffer)

  # cleanup
  checkVkResult vkDeviceWaitIdle(engine().vulkan.device)
  destroyPipeline(pipeline)
  destroyRenderData(renderdata)

proc test_02_multi_counter(time: float32, renderPass: RenderPass) =
  var font1 = loadFont[MAX_CODEPOINTS]("Overhaul.ttf", lineHeightPixels = 40)
  var font2 = loadFont[MAX_CODEPOINTS]("Overhaul.ttf", lineHeightPixels = 160)
  var font3 = loadFont[MAX_CODEPOINTS]("DejaVuSans.ttf", lineHeightPixels = 160)
  var renderdata = initRenderData()

  var pipeline = createPipeline(GlyphShader[MAX_CODEPOINTS](), renderPass = renderPass)

  font1.upload(renderdata)
  font2.upload(renderdata)
  font3.upload(renderdata)
  font1.addToPipeline(renderdata, pipeline)
  font2.addToPipeline(renderdata, pipeline)
  font3.addToPipeline(renderdata, pipeline)

  var textbuffer1 = font1.initTextBuffer(10, renderdata, baseScale = 0.1)
  var textbuffer2 = font2.initTextBuffer(10, renderdata, baseScale = 0.1)
  var textbuffer3 = font3.initTextBuffer(10, renderdata, baseScale = 0.1)

  var p = 0
  let l1 = textbuffer1.add($(p + 0), vec3(0.3, 0.5), capacity = 5)
  let l2 = textbuffer2.add($(p + 1), vec3(0.5, 0.5), capacity = 5)
  let l3 = textbuffer3.add($(p + 2), vec3(0.7, 0.5), capacity = 5)

  var start = getMonoTime()
  while ((getMonoTime() - start).inMilliseconds().int / 1000) < time:
    let progress = ((getMonoTime() - start).inMilliseconds().int / 1000) / time

    inc p

    textbuffer1.text(l1, $(p + 0))
    textbuffer2.text(l2, $(p + 1))
    textbuffer3.text(l3, $(p + 2))

    textbuffer1.refresh()
    textbuffer2.refresh()
    textbuffer3.refresh()

    withNextFrame(framebuffer, commandbuffer):
      withRenderPass(
        renderPass,
        framebuffer,
        commandbuffer,
        frameWidth(),
        frameHeight(),
        vec4(0, 0, 0, 0),
      ):
        withPipeline(commandbuffer, pipeline):
          bindDescriptorSet(commandbuffer, font1.descriptorSet, 3, pipeline)
          renderTextBuffer(commandbuffer, pipeline, textbuffer1)
          bindDescriptorSet(commandbuffer, font2.descriptorSet, 3, pipeline)
          renderTextBuffer(commandbuffer, pipeline, textbuffer2)
          bindDescriptorSet(commandbuffer, font3.descriptorSet, 3, pipeline)
          renderTextBuffer(commandbuffer, pipeline, textbuffer3)

      # cleanup
  checkVkResult vkDeviceWaitIdle(engine().vulkan.device)
  destroyPipeline(pipeline)
  destroyRenderData(renderdata)

proc test_03_layouting(time: float32, renderPass: RenderPass) =
  var font = loadFont[MAX_CODEPOINTS]("DejaVuSans.ttf", lineHeightPixels = 160)
  var renderdata = initRenderData()

  var pipeline = createPipeline(GlyphShader[MAX_CODEPOINTS](), renderPass = renderPass)

  font.upload(renderdata)
  font.addToPipeline(renderdata, pipeline)

  var textbuffer = font.initTextBuffer(1000, renderdata, baseScale = 0.1)

  discard textbuffer.add("Anchor at center", vec3(0, 0), anchor = vec2(0, 0))
  discard textbuffer.add("Anchor at top left`", vec3(-1, 1), anchor = vec2(-1, 1))
  discard textbuffer.add("Anchor at top right", vec3(1, 1), anchor = vec2(1, 1))
  discard textbuffer.add("Anchor at bottom left", vec3(-1, -1), anchor = vec2(-1, -1))
  discard textbuffer.add("Anchor at bottom right", vec3(1, -1), anchor = vec2(1, -1))

  discard textbuffer.add(
    "Mutiline text\nLeft aligned\nCool!", vec3(-0.5, -0.5), alignment = Left
  )
  discard textbuffer.add(
    "Mutiline text\nCenter aligned\nCool!!", vec3(0, -0.5), alignment = Center
  )
  discard textbuffer.add(
    "Mutiline text\nRight aligned\nCool!!!", vec3(0.5, -0.5), alignment = Right
  )

  var start = getMonoTime()
  while ((getMonoTime() - start).inMilliseconds().int / 1000) < time:
    let progress = ((getMonoTime() - start).inMilliseconds().int / 1000) / time
    if windowWasResized():
      textbuffer.refresh()

    withNextFrame(framebuffer, commandbuffer):
      bindDescriptorSet(commandbuffer, font.descriptorSet, 3, pipeline)
      withRenderPass(
        renderPass,
        framebuffer,
        commandbuffer,
        frameWidth(),
        frameHeight(),
        vec4(0, 0, 0, 0),
      ):
        withPipeline(commandbuffer, pipeline):
          renderTextBuffer(commandbuffer, pipeline, textbuffer)

      # cleanup
  checkVkResult vkDeviceWaitIdle(engine().vulkan.device)
  destroyPipeline(pipeline)
  destroyRenderData(renderdata)

proc test_04_lots_of_texts(time: float32, renderPass: RenderPass) =
  var font = loadFont[MAX_CODEPOINTS]("DejaVuSans.ttf", lineHeightPixels = 160)
  var renderdata = initRenderData()

  var pipeline = createPipeline(GlyphShader[MAX_CODEPOINTS](), renderPass = renderPass)

  font.upload(renderdata)
  font.addToPipeline(renderdata, pipeline)

  var textbuffer = font.initTextBuffer(3000, renderdata, baseScale = 0.1)

  for i in 0 ..< 1000:
    discard textbuffer.add(
      $i,
      vec3(rand(-0.8 .. 0.8), rand(-0.8 .. 0.8), rand(0.1 .. 0.2)),
      color =
        vec4(rand(0.5 .. 1.0), rand(0.5 .. 1.0), rand(0.5 .. 1.0), rand(0.5 .. 1.0)),
      scale = rand(0.5'f32 .. 1.5'f32),
    )

  var start = getMonoTime()
  var last = start
  while ((getMonoTime() - start).inMilliseconds().int / 1000) < time:
    let n = getMonoTime()
    echo (n - last).inMicroseconds() / 1000
    last = n
    withNextFrame(framebuffer, commandbuffer):
      if windowWasResized():
        textbuffer.refresh()
      bindDescriptorSet(commandbuffer, font.descriptorSet, 3, pipeline)
      withRenderPass(
        renderPass,
        framebuffer,
        commandbuffer,
        frameWidth(),
        frameHeight(),
        vec4(0, 0, 0, 0),
      ):
        withPipeline(commandbuffer, pipeline):
          renderTextBuffer(commandbuffer, pipeline, textbuffer)

        # cleanup
  checkVkResult vkDeviceWaitIdle(engine().vulkan.device)
  destroyPipeline(pipeline)
  destroyRenderData(renderdata)

when isMainModule:
  var time = 3'f32
  initEngine("Test text")

  # for depthBuffer in [true, false]:
  var renderpass = createDirectPresentationRenderPass(depthBuffer = true)
  setupSwapchain(renderpass = renderpass)

  # tests a simple triangle with minimalistic shader and vertex format
  test_01_static_label(time, renderpass)
  test_02_multi_counter(time, renderpass)
  test_03_layouting(time, renderpass)
  test_04_lots_of_texts(time, renderpass)

  checkVkResult vkDeviceWaitIdle(engine().vulkan.device)
  destroyRenderPass(renderpass)
  clearSwapchain()

  destroyVulkan()