package trtrfr

import org.openrndr.application
import org.openrndr.color.*
import org.openrndr.draw.*
import org.openrndr.math.IntVector2
import org.openrndr.math.Vector2
import phenax.utils.*

fun main() = application {
  if (captureEnabled) {
    val hiRes = System.getenv("PHENAX_HIRES")?.isNotEmpty() ?: false
    configure {
      width = if (hiRes) 3840 else 1920
      height = if (hiRes) 2160 else 1080
      windowResizable = false
    }
  } else {
    configure {
      width = 640
      height = 360
      windowResizable = false
    }
  }

  oliveProgram {
    val cs by liveComputeShader("trtrfr")
    val enableAntialiasing = captureEnabled
    val shaderLocalSize = IntVector2(8, 8)

    val outputSize = IntVector2(width, height)
    val outputBuffer = colorBuffer(outputSize.x, outputSize.y, type = ColorType.UINT8)

    var scale: Double = 1.0
    var offset: Vector2 = Vector2.ZERO
    var moving = false

    program.keyboard.keyDown.listen {
      if (it.name == "r") {
        scale = 1.0
        offset = Vector2.ZERO
      }
    }
    program.mouse.buttonDown.listen { moving = true }
    program.mouse.buttonUp.listen { moving = false }
    var lastPos = Vector2.ZERO
    program.mouse.moved.listen {
      val dimens = Vector2(program.width.toDouble(), program.height.toDouble())
      var pos = (Vector2(it.position.x, -it.position.y) - dimens / 2.0).div(dimens)
      val del = lastPos - pos
      lastPos = pos
      if (moving) offset = offset + del * 5.0 / scale
    }

    program.mouse.scrolled.listen { scale = scale * (1 + it.rotation.y / 3) }

    extend {
      cs.uniform("enable_antialiasing", enableAntialiasing)
      cs.uniform("offset", offset)
      cs.uniform("scale", scale)
      cs.uniform("seconds", seconds)
      cs.image("output_image", 0, outputBuffer.imageBinding(0, ImageAccess.WRITE))
      cs.execute(
              (outputBuffer.width + shaderLocalSize.x - 1) / shaderLocalSize.x,
              (outputBuffer.height + shaderLocalSize.y - 1) / shaderLocalSize.y,
              1
      )
      drawer.image(outputBuffer, 0.0, 0.0, width.toDouble(), height.toDouble())

      // val shadow = outputBuffer.shadow
      // shadow.download()
      // for (y in 0 until outputBuffer.height) {
      //   for (x in 0 until outputBuffer.width) {
      //     val c = shadow[x, y]
      //     shadow[x, y] =
      //             ColorRGBa(
      //                     Math.pow(c.r, (1.0 / 2.2)),
      //                     Math.pow(c.g, (1.0 / 2.2)),
      //                     Math.pow(c.b, (1.0 / 2.2)),
      //                     c.alpha
      //             )
      //   }
      // }
      // shadow.upload()
      // outputBuffer.saveToFile(java.io.File("screenshot/${program.name}.png"), async = false)
      // program.application.exit()
    }
  }
}