3 files changed, 67 insertions, 25 deletions

diff --git a/src/main/kotlin/org/openrndr/Preload.kt b/src/main/kotlin/org/openrndr/Preload.kt
index 375294a..1bbd673 100644
--- a/src/main/kotlin/org/openrndr/Preload.kt
+++ b/src/main/kotlin/org/openrndr/Preload.kt
@@ -7,7 +7,7 @@ import phenax.utils.recordingEnabled
 class Preload : ApplicationPreload() {
   override fun onProgramSetup(program: Program) {
     println("BACKEND(${program.name}): ${program.application::class.qualifiedName}")
-    if (recordingEnabled()) {
+    if (recordingEnabled) {
       program.extend(HeadlessRecorder())
     }
   }
diff --git a/src/main/kotlin/phenax/utils/program.kt b/src/main/kotlin/phenax/utils/program.kt
index 2d3e04b..6ff5379 100644
--- a/src/main/kotlin/phenax/utils/program.kt
+++ b/src/main/kotlin/phenax/utils/program.kt
@@ -1,9 +1,66 @@
 package phenax.utils
 
-fun recordDuration(durationSeconds: Int) {
-  System.setProperty("phenax.record.frameCount", (60 * durationSeconds).toString())
+import java.io.File
+import kotlin.properties.ReadOnlyProperty
+import org.openrndr.ApplicationBuilder
+import org.openrndr.Program
+import org.openrndr.draw.*
+import org.openrndr.extra.filewatcher.watchingFile
+import org.openrndr.extra.olive.oliveProgram as oliveProgramOriginal
+
+fun Int.seconds(): Int = this * 1000
+
+fun Int.minutes(): Int = this.seconds() * 60
+
+fun recordDuration(millis: Int) {
+  System.setProperty("phenax.record.frameCount", (60 * millis / 1000).toInt().toString())
 }
 
 fun recordDuration(): Int? = System.getProperty("phenax.record.frameCount")?.toIntOrNull()
 
-fun recordingEnabled(): Boolean = System.getenv("RECORD") != null
+val recordingEnabled: Boolean by lazy { System.getenv("RECORD") != null }
+
+// fun classDir(): String =
+//     "src/main/kotlin/${stackRootClassName(sanitize =
+// true).split(".").dropLast(1).joinToString("/")}";
+
+// fun localComputeShader(name: String): ComputeShader =
+//         ComputeShader.fromCode(File("src/main/resources/shaders/${name}.cs").readText(), name)
+
+// Have to keep the same name (oliveProgram) to get update on save to work
+fun ApplicationBuilder.oliveProgram(init: Program.() -> Unit) {
+  if (recordingEnabled) {
+    program(init)
+  } else {
+    // TODO: Make watching shader files work
+    oliveProgramOriginal(init = init)
+  }
+}
+
+fun Program.liveComputeShader(name: String): ReadOnlyProperty<Any?, ComputeShader> {
+  val shaderCode by
+          watchingFile(File("src/main/resources/shaders/${name}.cs")) { file ->
+            println("change")
+            file.readText()
+          }
+
+  var cs: ComputeShader = ComputeShader.fromCode(shaderCode, name)
+
+  if (!recordingEnabled) {
+    var lastCompiledCode: String = shaderCode
+
+    program.extend {
+      if (shaderCode != lastCompiledCode) {
+        cs.destroy()
+        try {
+          cs = ComputeShader.fromCode(shaderCode, name)
+        } catch (e: Exception) {
+          println("Shader compile failed: ${e.message}")
+        }
+        lastCompiledCode = shaderCode
+      }
+    }
+  }
+
+  return ReadOnlyProperty { _, _ -> cs }
+}
diff --git a/src/main/kotlin/play/PlayShowProgram.kt b/src/main/kotlin/play/PlayProgram.kt
index 4107cd7..0ec0133 100644
--- a/src/main/kotlin/play/PlayShowProgram.kt
+++ b/src/main/kotlin/play/PlayProgram.kt
@@ -1,12 +1,13 @@
 package play
 
+import java.io.File
+import org.openrndr.Program
 import org.openrndr.application
 import org.openrndr.draw.*
-import org.openrndr.extra.olive.oliveProgram
 import phenax.utils.*
 
 fun main() = application {
-  recordDuration(10)
+  recordDuration(10.seconds())
 
   configure {
     width = 640
@@ -15,27 +16,11 @@ fun main() = application {
   }
 
   oliveProgram {
+    val cs by liveComputeShader("play_shader")
     // val image = loadImage("data/images/pm5544.png")
     // val font = loadFont("data/fonts/default.otf", 64.0)
 
-    val cs = computeStyle {
-      computeTransform =
-              """
-          ivec2 coords = ivec2(gl_GlobalInvocationID.xy);
-          ivec2 resolution = imageSize(p_outputImg).xy;
-          float aspect_ratio = resolution.y*1.0 / resolution.x;
-
-          vec2 uv = vec2(1.0, aspect_ratio) * (coords - 0.5*resolution)/resolution;
-          float dist = length(uv) - 0.1;
-          float step = step(0.0, dist);
-
-          vec4 color = vec4(step, 0.0, 0.0, 1.0);
-          imageStore(p_outputImg, coords, color);
-      """.trimIndent()
-    }
-
     // val inputBuffer = colorBuffer(width, height, type = ColorType.UINT8)
-
     val outputBuffer = colorBuffer(width, height, type = ColorType.UINT8)
 
     extend {
@@ -53,8 +38,8 @@ fun main() = application {
       // drawer.fill = ColorRGBa.WHITE
       // drawer.text("OPENRNDR", width / 2.0, height / 2.0)
 
-      cs.parameter("seconds", seconds)
-      cs.image("outputImg", outputBuffer.imageBinding(0, ImageAccess.WRITE))
+      cs.uniform("seconds", seconds)
+      cs.image("outputImg", 1, outputBuffer.imageBinding(0, ImageAccess.WRITE))
       cs.execute(outputBuffer.width, outputBuffer.height, 1)
       drawer.image(outputBuffer)
     }