package phenax.extensions

import org.openrndr.*
import org.openrndr.draw.*
import org.openrndr.extensions.SingleScreenshot
import org.openrndr.ffmpeg.VideoWriter

class HeadlessScreenshot : Extension {
  override var enabled: Boolean = true

  var screenshots: SingleScreenshot? = null
  var videoWriter: VideoWriter? = null
  var maxFrameCount: Int? = null

  override fun setup(program: Program) {
    val scr = SingleScreenshot()
    screenshots = scr
    program.extend(scr) {
      outputFile = "screenshot/${program.name}.png"
    }
  }

  override fun beforeDraw(drawer: Drawer, program: Program) {}

  override fun afterDraw(drawer: Drawer, program: Program) {
    screenshots?.trigger()

    // program.application.exit()
  }
}