package org.openrndr.extras.imageFit

import org.openrndr.draw.ColorBuffer
import org.openrndr.draw.Drawer
import org.openrndr.math.map
import org.openrndr.shape.Rectangle


enum class FitMethod {
    Cover,
    Contain
}

fun Drawer.imageFit(
        img: ColorBuffer,
        x: Double = 0.0,
        y: Double = 0.0,
        width: Double = img.width.toDouble(),
        height: Double = img.height.toDouble(),
        fitMethod: FitMethod = FitMethod.Cover,
        horizontalPosition: Double = 0.0,
        verticalPosition: Double = 0.0
) {
    val sourceWidth = img.width.toDouble()
    val sourceHeight = img.height.toDouble()

    var targetX = x
    var targetY = y

    var targetWidth: Double
    var targetHeight: Double

    val source: Rectangle
    val target: Rectangle

    when (fitMethod) {
        FitMethod.Contain -> {
            targetWidth = width
            targetHeight = height

            if (width <= targetWidth) {
                targetWidth = width
                targetHeight = (sourceHeight / sourceWidth) * width
            }

            if (height <= targetHeight) {
                targetHeight = height
                targetWidth = (sourceWidth / sourceHeight) * height
            }

            val left = x
            val right = x + width - targetWidth
            val top = y
            val bottom = y + height - targetHeight

            targetX = map(-1.0, 1.0, left, right, horizontalPosition)
            targetY = map(-1.0, 1.0, top, bottom, verticalPosition)

            source = Rectangle(0.0, 0.0, sourceWidth, sourceHeight)
            target = Rectangle(targetX, targetY, targetWidth, targetHeight)
        }

        FitMethod.Cover -> {
            targetWidth = sourceWidth
            targetHeight = sourceHeight

            if (sourceWidth <= targetWidth) {
                targetWidth = sourceWidth
                targetHeight = (height / width) * sourceWidth
            }

            if (sourceHeight <= targetHeight) {
                targetHeight = sourceHeight
                targetWidth = (width / height) * sourceHeight
            }

            val left = 0.0
            val right = sourceWidth - targetWidth
            val top = 0.0
            val bottom = sourceHeight - targetHeight

            targetX = map(-1.0, 1.0, left, right, horizontalPosition)
            targetY = map(-1.0, 1.0, top, bottom, verticalPosition)

            source = Rectangle(targetX, targetY, targetWidth, targetHeight)
            target = Rectangle(x, y, width, height)
        }
    }

    image(img, source, target)
}