From d2f240ff0ca0d27f417f837e706c781a98c31311 Mon Sep 17 00:00:00 2001
From: Linnea Gräf <nea@nea.moe>
Date: Wed, 28 Aug 2024 19:04:24 +0200
Subject: Refactor source layout

Introduce compat source sets and move all kotlin sources to the main directory

[no changelog]
---
 src/main/kotlin/events/CustomItemModelEvent.kt | 43 ++++++++++++++++++++++++++
 1 file changed, 43 insertions(+)
 create mode 100644 src/main/kotlin/events/CustomItemModelEvent.kt

(limited to 'src/main/kotlin/events/CustomItemModelEvent.kt')

diff --git a/src/main/kotlin/events/CustomItemModelEvent.kt b/src/main/kotlin/events/CustomItemModelEvent.kt
new file mode 100644
index 0000000..27524a9
--- /dev/null
+++ b/src/main/kotlin/events/CustomItemModelEvent.kt
@@ -0,0 +1,43 @@
+
+
+package moe.nea.firmament.events
+
+import java.util.*
+import net.minecraft.client.render.model.BakedModel
+import net.minecraft.client.render.model.BakedModelManager
+import net.minecraft.client.util.ModelIdentifier
+import net.minecraft.item.ItemStack
+
+data class CustomItemModelEvent(
+    val itemStack: ItemStack,
+    var overrideModel: ModelIdentifier? = null,
+) : FirmamentEvent() {
+    companion object : FirmamentEventBus<CustomItemModelEvent>() {
+        private val cache = IdentityHashMap<ItemStack?, Any>()
+        private val sentinelNull = Object()
+
+        fun clearCache() {
+            cache.clear()
+        }
+
+        @JvmStatic
+        fun getModelIdentifier(itemStack: ItemStack?): ModelIdentifier? {
+            if (itemStack == null) return null
+            return publish(CustomItemModelEvent(itemStack)).overrideModel
+        }
+
+        @JvmStatic
+        fun getModel(itemStack: ItemStack?, thing: BakedModelManager): BakedModel? {
+            if (itemStack == null) return null
+            val cachedValue = cache.getOrPut(itemStack) {
+                val modelId = getModelIdentifier(itemStack) ?: return@getOrPut sentinelNull
+                val bakedModel = thing.getModel(modelId)
+                if (bakedModel === thing.missingModel) return@getOrPut sentinelNull
+                bakedModel
+            }
+            if (cachedValue === sentinelNull)
+                return null
+            return cachedValue as BakedModel
+        }
+    }
+}
-- 
cgit