Add moth trtrfr

1 files changed, 89 insertions, 0 deletions

diff --git a/src/main/kotlin/trtrfr/TrtrfrProgram.kt b/src/main/kotlin/trtrfr/TrtrfrProgram.kt
new file mode 100644
index 0000000..7d2f959
--- /dev/null
+++ b/src/main/kotlin/trtrfr/TrtrfrProgram.kt
@@ -0,0 +1,89 @@
+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()
+    }
+  }
+}