Terrarum/src/net/torvald/terrarum/modulebasegame/ui/UIBuildingMakerBlockChooser.kt

package net.torvald.terrarum.modulebasegame.ui

import com.badlogic.gdx.graphics.Camera
import com.badlogic.gdx.graphics.Color
import com.badlogic.gdx.graphics.OrthographicCamera
import com.badlogic.gdx.graphics.g2d.SpriteBatch
import net.torvald.terrarum.*
import net.torvald.terrarum.modulebasegame.BuildingMaker
import net.torvald.terrarum.modulebasegame.ui.ItemSlotImageFactory.CELLCOLOUR_WHITE
import net.torvald.terrarum.ui.*
import net.torvald.terrarum.ui.UIItemTextButtonList.Companion.DEFAULT_BACKGROUNDCOL
import net.torvald.terrarum.worlddrawer.CreateTileAtlas.Companion.WALL_OVERLAY_COLOUR
import kotlin.math.roundToInt

/**
 * Created by minjaesong on 2019-02-14.
 */
class UIBuildingMakerBlockChooser(val parent: BuildingMaker): UICanvas() {

    companion object {
        const val TILES_X = 8
        const val TILES_Y = 10

        const val TILESREGION_SIZE = 24
        const val MENUBAR_SIZE = 72
        const val SCROLLBAR_SIZE = 24

        const val WIDTH = TILES_X * TILESREGION_SIZE + SCROLLBAR_SIZE + MENUBAR_SIZE
        const val HEIGHT = TILES_Y * TILESREGION_SIZE
    }

    override var width = WIDTH
    override var height = HEIGHT
    override var openCloseTime = 0f

    val paletteTerrain = ArrayList<UIItemImageButton>()
    val paletteWall = ArrayList<UIItemImageButton>()

    private val palettes = listOf(
        paletteTerrain, paletteWall
    )

    var currentPaletteSelection = 0
    val currentPalette: ArrayList<UIItemImageButton>
        get() = palettes[currentPaletteSelection]

    // TODO scrolling of the palette, as the old method flat out won't work with The Flattening

    private val tabs = UIItemTextButtonList(
        this, 36, arrayOf("Terrain", "Wall"), // TODO use inventory icons
        0, 0, textAreaWidth = MENUBAR_SIZE, width = MENUBAR_SIZE,
        defaultSelection = 0,
    ).also {
        it.selectionChangeListener = { _, new ->
            resetScroll()
            currentPaletteSelection = new
        }
    }
    private val closeButton = UIItemTextButtonList(
        this, 36, arrayOf("Close"),
        0, this.height - UIItemTextButtonList.DEFAULT_LINE_HEIGHT,
        width = MENUBAR_SIZE, textAreaWidth = MENUBAR_SIZE,
    )

    private var scroll = 0

    private fun paletteScroll(delta: Int) {
        currentPalette.forEach {
            it.posY -= delta * TILESREGION_SIZE
        }
    }

    private fun resetScroll() {
        scrollBar.scrolledForce(0f, scroll * -1f)
        scroll = 0
    }

    private val <E> List<E>.visible: List<E>
        get()  = this.subList(
            TILES_X * scroll,
            (TILES_X * scroll + TILES_X * TILES_Y).coerceAtMost(this.size)
        )

    init {

        BlockCodex.getAll().filter {
            (try {
                ItemCodex.getItemImage(it.id)
            }
            catch (e: NullPointerException) {
                null
            }) != null
        }.filter { !it.hasTag("INTERNAL") }.sortedBy { it.id }.forEachIndexed { index, prop -> // FIXME no lexicographic sorting
            val paletteItem = UIItemImageButton(
                this, ItemCodex.getItemImage(prop.id)!!,
                initialX = MENUBAR_SIZE + (index % TILES_X) * TILESREGION_SIZE,
                initialY = (index / TILES_X) * TILESREGION_SIZE,
                highlightable = false,
                width = TILESREGION_SIZE,
                height = TILESREGION_SIZE,
                highlightCol = Color.WHITE,
                activeCol = Color.WHITE,
                backgroundCol = Color(0)
            )
            val paletteItem2 = UIItemImageButton(
                this, ItemCodex.getItemImage(prop.id)!!,
                initialX = MENUBAR_SIZE + (index % TILES_X) * TILESREGION_SIZE,
                initialY = (index / TILES_X) * TILESREGION_SIZE,
                highlightable = false,
                width = TILESREGION_SIZE,
                height = TILESREGION_SIZE,
                highlightCol = Color.WHITE,
                activeCol = WALL_OVERLAY_COLOUR,
                inactiveCol = WALL_OVERLAY_COLOUR,
                backgroundCol = Color(0)
            )

            paletteItem.clickOnceListener = { _, _ -> parent.setPencilColour(prop.id) }
            paletteItem2.clickOnceListener = { _, _ -> parent.setPencilColour("wall@"+prop.id) }

            paletteTerrain.add(paletteItem)
            if (prop.isWallable) paletteWall.add(paletteItem2)

        }
    }

    private val scrollOverflowSize: Double
        get() = ((currentPalette.size - TILES_X * TILES_Y).toDouble() / TILES_X).ceilToDouble().coerceAtLeast(0.0)

    private val scrollBar = UIItemVertSlider(
        this, WIDTH - UIItemVertSlider.WIDTH, 0, 0.0, 0.0, scrollOverflowSize, height, (height * TILES_Y / (scrollOverflowSize + TILES_Y)).ceilToInt()
    ).also {
        println("scrollOverflowSize = $scrollOverflowSize")

        addUIitem(it)
        it.selectionChangeListener = { value ->
            val oldScroll = scroll
            scroll = value.roundToInt()
            paletteScroll(scroll - oldScroll)
        }
    }


    override fun updateUI(delta: Float) {
        if (!scrollBar.mouseUp) currentPalette.forEach { it.update(delta) }
        tabs.update(delta)
        closeButton.update(delta)
        if (closeButton.mousePushed) {
            closeButton.deselect()
            closeGracefully()
        }

        currentPalette.visible.forEach {
            it.update(delta)
        }

        uiItems.forEach { it.update(delta) }
    }

    private fun closeGracefully() {
        this.isVisible = false
        parent.tappedOnUI = true
    }

    private fun rebuildPalette() {

    }

    override fun renderUI(frameDelta: Float, batch: SpriteBatch, camera: OrthographicCamera) {
        blendNormalStraightAlpha(batch)

        // border
        batch.color = Toolkit.Theme.COL_INACTIVE
        Toolkit.drawBoxBorder(batch, -1, -1, width + 2, height + 2)

        // palette background
        batch.color = DEFAULT_BACKGROUNDCOL
        Toolkit.fillArea(batch, MENUBAR_SIZE, 0, width - MENUBAR_SIZE, height)

        // titlebar (sideways)
        batch.color = UINSMenu.DEFAULT_TITLEBACKCOL
        Toolkit.fillArea(batch, 0, 0, MENUBAR_SIZE, height)

        // the actual buttons
        tabs.render(frameDelta, batch, camera)
        closeButton.render(frameDelta, batch, camera)

        currentPalette.visible.forEach {
            it.render(frameDelta, batch, camera)
        }

        uiItems.forEach { it.render(frameDelta, batch, camera) }
    }

    private var dragOriginX = 0 // relative mousepos
    private var dragOriginY = 0 // relative mousepos
    private var dragForReal = false

    private fun mouseOnDragHandle() = relativeMouseX in 0 until MENUBAR_SIZE && relativeMouseY in tabs.height until height - closeButton.height

    override fun touchDragged(screenX: Int, screenY: Int, pointer: Int): Boolean {
        if (mouseOnDragHandle()) {
            if (dragForReal) {
                handler.setPosition(
                    (screenX / App.scr.magn - dragOriginX).roundToInt(),
                    (screenY / App.scr.magn - dragOriginY).roundToInt()
                )
            }
        }

        return super.touchDragged(screenX, screenY, pointer)
    }

    override fun touchDown(screenX: Int, screenY: Int, pointer: Int, button: Int): Boolean {
        if (mouseOnDragHandle()) {
            dragOriginX = relativeMouseX
            dragOriginY = relativeMouseY
            dragForReal = true
            parent.tappedOnUI = true
        }
        else {
            dragForReal = false
        }

        currentPalette.visible.forEach {
            it.touchDown(screenX, screenY, pointer, button)
        }

        return super.touchDown(screenX, screenY, pointer, button)
    }

    override fun touchUp(screenX: Int, screenY: Int, pointer: Int, button: Int): Boolean {
        return super.touchUp(screenX, screenY, pointer, button)
    }

    override fun scrolled(amountX: Float, amountY: Float): Boolean {
        scrollBar.scrolledForce(amountX, amountY)

        return super.scrolled(amountX, amountY)
    }

    override fun doOpening(delta: Float) {
    }

    override fun doClosing(delta: Float) {
    }

    override fun endOpening(delta: Float) {
    }

    override fun endClosing(delta: Float) {
    }

    override fun dispose() {
        // nothing to dispose; you can't dispose the palette as its image is dynamically assigned
    }
}