package mods.betterfoliage.render.block.vanilla

import mods.betterfoliage.BetterFoliage
import mods.betterfoliage.BetterFoliageMod
import mods.betterfoliage.config.Config
import mods.betterfoliage.config.isSnow
import mods.betterfoliage.integration.ShadersModIntegration
import mods.betterfoliage.model.Color
import mods.betterfoliage.model.HalfBakedSpecialWrapper
import mods.betterfoliage.model.HalfBakedWrapperKey
import mods.betterfoliage.model.SpecialRenderModel
import mods.betterfoliage.model.SpriteSetDelegate
import mods.betterfoliage.model.crossModelsRaw
import mods.betterfoliage.model.crossModelsTextured
import mods.betterfoliage.render.lighting.RoundLeafLightingPreferUp
import mods.betterfoliage.render.particle.LeafBlockModel
import mods.betterfoliage.render.particle.LeafParticleKey
import mods.betterfoliage.render.particle.LeafParticleRegistry
import mods.betterfoliage.render.pipeline.RenderCtxBase
import mods.betterfoliage.resource.discovery.ModelBakingContext
import mods.betterfoliage.resource.discovery.ModelDiscoveryContext
import mods.betterfoliage.resource.discovery.ParametrizedModelDiscovery
import mods.betterfoliage.resource.generated.GeneratedLeafSprite
import mods.betterfoliage.util.Atlas
import mods.betterfoliage.util.averageHSB
import mods.betterfoliage.util.lazy
import mods.betterfoliage.util.lazyMap
import mods.betterfoliage.util.brighten
import mods.betterfoliage.util.logTextureColor
import mods.betterfoliage.util.saturate
import net.minecraft.client.renderer.RenderType
import net.minecraft.util.Direction.UP
import net.minecraft.util.ResourceLocation
import org.apache.logging.log4j.Level.INFO

object StandardLeafDiscovery : ParametrizedModelDiscovery() {
    override fun processModel(ctx: ModelDiscoveryContext, params: Map<String, String>) {
        val texture = params.location("texture") ?: return
        val tint = params.int("tint") ?: -1
        val color = Atlas.BLOCKS.file(texture).averageHSB.let {
            detailLogger.logTextureColor(INFO, "leaf texture \"$texture\"", it)
            it.brighten().asColor
        }
        val leafType = LeafParticleRegistry.typeMappings.getType(texture) ?: "default"
        val generated = GeneratedLeafSprite(texture, leafType)
            .register(BetterFoliage.generatedPack)
            .apply { ctx.sprites.add(this) }

        detailLogger.log(INFO, "     particle $leafType")
        ctx.addReplacement(StandardLeafKey(generated, leafType, tint, color))
        BetterFoliage.blockTypes.leaf.add(ctx.blockState)
    }
}

data class StandardLeafKey(
    val roundLeafTexture: ResourceLocation,
    override val leafType: String,
    override val tintIndex: Int,
    override val avgColor: Color
) : HalfBakedWrapperKey(), LeafParticleKey {
    override fun bake(ctx: ModelBakingContext, wrapped: SpecialRenderModel): SpecialRenderModel {
        return StandardLeafModel(wrapped, this)
    }
}

class StandardLeafModel(
    model: SpecialRenderModel,
    override val key: StandardLeafKey
) : HalfBakedSpecialWrapper(model), LeafBlockModel {
    val leafNormal by leafModelsNormal.delegate(key)
    val leafSnowed by leafModelsSnowed.delegate(key)

    override fun renderLayer(ctx: RenderCtxBase, data: Any, layer: RenderType) {
        ShadersModIntegration.leaves(ctx, true) {
            super.renderLayer(ctx, data, layer)
            if (!Config.enabled || !Config.leaves.enabled) return

            ctx.vertexLighter = RoundLeafLightingPreferUp
            val leafIdx = ctx.random.nextInt(64)
            ctx.renderQuads(leafNormal[leafIdx])
            if (ctx.state(UP).isSnow) ctx.renderQuads(leafSnowed[leafIdx])
        }
    }

    companion object {
        val leafSpritesSnowed by SpriteSetDelegate(Atlas.BLOCKS) { idx ->
            ResourceLocation(BetterFoliageMod.MOD_ID, "blocks/better_leaves_snowed_$idx")
        }
        val leafModelsBase by BetterFoliage.modelManager.lazy {
            Config.leaves.let { crossModelsRaw(64, it.size, it.hOffset, it.vOffset) }
        }
        val leafModelsNormal = BetterFoliage.modelManager.lazyMap { key: StandardLeafKey ->
            // generated leaf textures naturally carry the color of their source textures
            // no need to color the quad a second time
            crossModelsTextured(leafModelsBase, Color.white, key.tintIndex, true) { key.roundLeafTexture }
        }
        val leafModelsSnowed = BetterFoliage.modelManager.lazyMap { key: StandardLeafKey ->
            crossModelsTextured(leafModelsBase, Color.white, -1, false) { leafSpritesSnowed[it].name }
        }
    }
}