rhubarb-lip-sync/extras/EsotericSoftwareSpine/src/main/kotlin/AudioFileModel.kt

package com.rhubarb_lip_sync.rhubarb_for_spine

import javafx.application.Platform
import javafx.beans.binding.BooleanBinding
import javafx.beans.binding.ObjectBinding
import javafx.beans.binding.StringBinding
import javafx.beans.property.SimpleBooleanProperty
import javafx.beans.property.SimpleObjectProperty
import javafx.beans.property.SimpleStringProperty
import javafx.scene.control.Alert
import javafx.scene.control.ButtonType
import tornadofx.getValue
import tornadofx.setValue
import java.nio.file.Path
import java.util.concurrent.ExecutorService
import java.util.concurrent.Future

class AudioFileModel(
	audioEvent: SpineJson.AudioEvent,
	private val parentModel: AnimationFileModel,
	private val executor: ExecutorService,
	private val reportResult: (List<MouthCue>) -> Unit
) {
	private val spineJson = parentModel.spineJson

	private val audioFilePath: Path = spineJson.audioDirectoryPath.resolve(audioEvent.relativeAudioFilePath)

	val eventNameProperty = SimpleStringProperty(audioEvent.name)
	val eventName: String by eventNameProperty

	val displayFilePathProperty = SimpleStringProperty(audioEvent.relativeAudioFilePath)

	val animationNameProperty = SimpleStringProperty().apply {
		val mainModel = parentModel.parentModel
		bind(object : ObjectBinding<String>() {
			init {
				super.bind(
					mainModel.animationPrefixProperty,
					eventNameProperty,
					mainModel.animationSuffixProperty
				)
			}
			override fun computeValue(): String {
				return mainModel.animationPrefix + eventName + mainModel.animationSuffix
			}
		})
	}
	val animationName: String by animationNameProperty

	val dialogProperty = SimpleStringProperty(audioEvent.dialog)
	private val dialog: String? by dialogProperty

	val animationProgressProperty = SimpleObjectProperty<Double?>(null)
	var animationProgress: Double? by animationProgressProperty
		private set

	private val animatedProperty = SimpleBooleanProperty().apply {
		bind(object : ObjectBinding<Boolean>() {
			init {
				super.bind(animationNameProperty, parentModel.spineJson.animationNames)
			}
			override fun computeValue(): Boolean {
				return parentModel.spineJson.animationNames.contains(animationName)
			}
		})
	}
	private var animated by animatedProperty

	private val futureProperty = SimpleObjectProperty<Future<*>?>()
	private var future by futureProperty

	val audioFileStateProperty = SimpleObjectProperty<AudioFileState>().apply {
		bind(object : ObjectBinding<AudioFileState>() {
			init {
				super.bind(animatedProperty, futureProperty, animationProgressProperty)
			}
			override fun computeValue(): AudioFileState {
				return if (future != null) {
					if (animationProgress != null)
						if (future!!.isCancelled)
							AudioFileState(AudioFileStatus.Canceling)
						else
							AudioFileState(AudioFileStatus.Animating, animationProgress)
					else
						AudioFileState(AudioFileStatus.Pending)
				} else {
					if (animated)
						AudioFileState(AudioFileStatus.Done)
					else
						AudioFileState(AudioFileStatus.NotAnimated)
				}
			}
		})
	}

	val busyProperty = SimpleBooleanProperty().apply {
		bind(object : BooleanBinding() {
			init {
				super.bind(futureProperty)
			}
			override fun computeValue(): Boolean {
				return future != null
			}

		})
	}
	val busy by busyProperty

	val actionLabelProperty = SimpleStringProperty().apply {
		bind(object : StringBinding() {
			init {
				super.bind(futureProperty)
			}
			override fun computeValue(): String {
				return if (future != null)
					"Cancel"
				else
					"Animate"
			}
		})
	}

	fun performAction() {
		if (future == null) {
			if (animated) {
				Alert(Alert.AlertType.CONFIRMATION).apply {
					headerText = "Animation '$animationName' already exists."
					contentText = "Do you want to replace the existing animation?"
					val result = showAndWait()
					if (result.get() != ButtonType.OK) {
						return
					}
				}
			}

			startAnimation()
		} else {
			cancelAnimation()
		}
	}

	private fun startAnimation() {
		val wrapperTask = Runnable {
			val recognizer = parentModel.parentModel.recognizer.value
			val extendedMouthShapes = parentModel.mouthShapes.filter { it.isExtended }.toSet()
			val reportProgress: (Double?) -> Unit = {
				progress -> runAndWait { this@AudioFileModel.animationProgress = progress }
			}
			val rhubarbTask = RhubarbTask(audioFilePath, recognizer, dialog, extendedMouthShapes, reportProgress)
			try {
				try {
					val result = rhubarbTask.call()
					runAndWait {
						reportResult(result)
					}
				} finally {
					runAndWait {
						animationProgress = null
						future = null
					}
				}
			} catch (e: InterruptedException) {
			} catch (e: Exception) {
				e.printStackTrace(System.err)

				Platform.runLater {
					Alert(Alert.AlertType.ERROR).apply {
						headerText = "Error performing lip sync for event '$eventName'."
						contentText = if (e is EndUserException)
							e.message
						else
							("An internal error occurred.\n"
								+ "Please report an issue, including the following information.\n"
								+ getStackTrace(e))
						show()
					}
				}
			}
		}
		future = executor.submit(wrapperTask)
	}

	private fun cancelAnimation() {
		future?.cancel(true)
	}
}

enum class AudioFileStatus {
	NotAnimated,
	Pending,
	Animating,
	Canceling,
	Done
}

data class AudioFileState(val status: AudioFileStatus, val progress: Double? = null)