rewrite model and texture detection

expose in mod configuration

src/main/kotlin/mods/betterfoliage/client/texture/GrassRegistry.kt

@@ -1,16 +1,27 @@
 package mods.betterfoliage.client.texture
 
-import mods.betterfoliage.client.Client
+import mods.betterfoliage.BetterFoliageMod
 import mods.betterfoliage.client.config.Config
+import mods.betterfoliage.client.integration.OptifineCTM
+import mods.octarinecore.client.render.BlockContext
 import mods.octarinecore.client.render.HSB
-import mods.octarinecore.client.resource.BlockTextureInspector
+import mods.octarinecore.client.resource.TextureListModelProcessor
+import mods.octarinecore.client.resource.TextureMediatedRegistry
 import mods.octarinecore.client.resource.averageColor
+import mods.octarinecore.client.resource.get
+import mods.octarinecore.common.Int3
+import mods.octarinecore.common.config.BlockMatcher
+import mods.octarinecore.common.config.ModelTextureList
+import mods.octarinecore.findFirst
 import net.minecraft.block.state.IBlockState
 import net.minecraft.client.renderer.texture.TextureAtlasSprite
 import net.minecraft.client.renderer.texture.TextureMap
+import net.minecraft.util.EnumFacing
+import net.minecraft.util.math.BlockPos
+import net.minecraft.world.IBlockAccess
+import net.minecraftforge.common.MinecraftForge
 import net.minecraftforge.fml.relauncher.Side
 import net.minecraftforge.fml.relauncher.SideOnly
-import org.apache.logging.log4j.Level.INFO
 
 const val defaultGrassColor = 0
 
@@ -23,28 +34,66 @@ class GrassInfo(
      * Color to use for Short Grass rendering instead of the biome color.
      *
      * Value is null if the texture is mostly grey (the saturation of its average color is under a configurable limit),
-     * the average color of the texture (significantly brightened) otherwise.
+     * the average color of the texture (significantly ) otherwise.
      */
     val overrideColor: Int?
 )
 
+interface IGrassRegistry {
+    fun get(state: IBlockState): GrassInfo?
+    fun get(state: IBlockState, world: IBlockAccess, pos: BlockPos, face: EnumFacing): GrassInfo?
+}
+
 /** Collects and manages rendering-related information for grass blocks. */
 @SideOnly(Side.CLIENT)
-object GrassRegistry : BlockTextureInspector<GrassInfo>() {
+object GrassRegistry : IGrassRegistry {
+    val subRegistries = mutableListOf(StandardGrassSupport)
 
-    init {
-        matchClassAndModel(Config.blocks.grass, "block/grass", listOf("top"))
-        matchClassAndModel(Config.blocks.grass, "block/cube_bottom_top", listOf("top"))
+    override fun get(state: IBlockState, world: IBlockAccess, pos: BlockPos, face: EnumFacing) =
+        subRegistries.findFirst { it.get(state, world, pos, face) }
+
+    operator fun get(ctx: BlockContext, face: EnumFacing) = get(ctx.blockState(Int3.zero), ctx.world!!, ctx.pos, face)
+
+    override fun get(state: IBlockState) = subRegistries.findFirst { it.get(state) }
+}
+
+object StandardGrassSupport :
+    TextureListModelProcessor<TextureAtlasSprite>,
+    TextureMediatedRegistry<List<String>, GrassInfo>,
+    IGrassRegistry
+{
+    init { MinecraftForge.EVENT_BUS.register(this) }
+
+    override var stateToKey = mutableMapOf<IBlockState, List<String>>()
+    override var stateToValue = mapOf<IBlockState, TextureAtlasSprite>()
+    override var textureToValue = mutableMapOf<TextureAtlasSprite, GrassInfo>()
+
+    override val logger = BetterFoliageMod.logDetail
+    override val logName = "StandardGrassSupport"
+    override val matchClasses: BlockMatcher get() = Config.blocks.grassClasses
+    override val modelTextures: List<ModelTextureList> get() = Config.blocks.grassModels.list
+
+    override fun get(state: IBlockState, world: IBlockAccess, pos: BlockPos, face: EnumFacing): GrassInfo? {
+        val baseTexture = stateToValue[state] ?: return null
+        return textureToValue[OptifineCTM.override(baseTexture, world, pos, face)] ?: textureToValue[baseTexture]
     }
 
-    override fun onAfterModelLoad() {
-        super.onAfterModelLoad()
-        Client.log(INFO, "Inspecting grass textures")
+    override fun get(state: IBlockState) = StandardLeafSupport.stateToValue[state].let {
+        if (it == null) null else textureToValue[it]
     }
 
-    override fun processTextures(state: IBlockState, textures: List<TextureAtlasSprite>, atlas: TextureMap): GrassInfo {
-        val hsb = HSB.fromColor(textures[0].averageColor ?: defaultGrassColor)
+    override fun processStitch(state: IBlockState, key: List<String>, atlas: TextureMap) = atlas[key[0]]
+
+    override fun processTexture(states: List<IBlockState>, texture: TextureAtlasSprite, atlas: TextureMap) {
+        registerGrass(texture, atlas)
+        OptifineCTM.getAllCTM(states, texture).forEach {
+            registerGrass(it, atlas)
+        }
+    }
+
+    fun registerGrass(texture: TextureAtlasSprite, atlas: TextureMap) {
+        val hsb = HSB.fromColor(texture.averageColor ?: defaultGrassColor)
         val overrideColor = if (hsb.saturation > Config.shortGrass.saturationThreshold) hsb.copy(brightness = 0.8f).asColor else null
-        return GrassInfo(textures[0], overrideColor)
+        textureToValue[texture] =  GrassInfo(texture, overrideColor)
     }
 }