Add item iota renderer for splicing table

Author: object-Object

Common/src/main/java/gay/object/hexdebug/api/client/splicing/SplicingTableIotaRendererProvider.java

@@ -7,6 +7,7 @@
 import net.minecraft.client.gui.components.Tooltip;
 import net.minecraft.network.chat.Component;
 import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
 
 /**
  * A factory for {@link SplicingTableIotaRenderer} instances.
@@ -15,8 +16,13 @@
  * currently visible, so don't do anything too laggy in here.
  */
 public interface SplicingTableIotaRendererProvider {
-    /** Creates and returns a new {@link SplicingTableIotaRenderer} for the provided iota. */
-    @NotNull
+    /**
+     * Creates and returns a new {@link SplicingTableIotaRenderer} for the provided iota.
+     * <br>
+     * May return null if unable to create a renderer for the given iota; in that case, the default
+     * renderer will be used instead.
+     */
+    @Nullable
     SplicingTableIotaRenderer createRenderer(
         @NotNull IotaType<?> type,
         @NotNull SplicingTableIotaClientView iota,

Common/src/main/kotlin/gay/object/hexdebug/HexDebugClient.kt

@@ -11,6 +11,7 @@ import gay.`object`.hexdebug.api.client.splicing.SplicingTableIotaRenderers
 import gay.`object`.hexdebug.blocks.focusholder.FocusHolderBlock
 import gay.`object`.hexdebug.config.HexDebugClientConfig
 import gay.`object`.hexdebug.config.HexDebugServerConfig
+import gay.`object`.hexdebug.gui.splicing.renderers.ItemRendererProvider
 import gay.`object`.hexdebug.gui.splicing.renderers.ListRendererProvider
 import gay.`object`.hexdebug.gui.splicing.renderers.PatternRendererProvider
 import gay.`object`.hexdebug.gui.splicing.renderers.TextureRendererProvider
@@ -95,8 +96,13 @@ object HexDebugClient {
     }
 
     private fun registerSplicingTableIotaRenderers() {
-        SplicingTableIotaRenderers.register(HexDebug.id("list"), ListRendererProvider.PARSER)
-        SplicingTableIotaRenderers.register(HexDebug.id("pattern"), PatternRendererProvider.PARSER)
-        SplicingTableIotaRenderers.register(HexDebug.id("texture"), TextureRendererProvider.PARSER)
+        for ((name, parser) in arrayOf(
+            "item" to ItemRendererProvider.PARSER,
+            "list" to ListRendererProvider.PARSER,
+            "pattern" to PatternRendererProvider.PARSER,
+            "texture" to TextureRendererProvider.PARSER,
+        )) {
+            SplicingTableIotaRenderers.register(HexDebug.id(name), parser)
+        }
     }
 }

Common/src/main/kotlin/gay/object/hexdebug/gui/splicing/renderers/ItemRendererProvider.kt

@@ -0,0 +1,71 @@
+package gay.`object`.hexdebug.gui.splicing.renderers
+
+import at.petrak.hexcasting.api.casting.iota.IotaType
+import gay.`object`.hexdebug.api.client.splicing.SplicingTableIotaRenderer
+import gay.`object`.hexdebug.api.client.splicing.SplicingTableIotaRendererParser
+import gay.`object`.hexdebug.api.client.splicing.SplicingTableIotaRendererProvider
+import gay.`object`.hexdebug.api.splicing.SplicingTableIotaClientView
+import gay.`object`.hexdebug.utils.*
+import net.minecraft.client.Minecraft
+import net.minecraft.commands.arguments.NbtPathArgument.NbtPath
+import net.minecraft.core.registries.BuiltInRegistries
+import net.minecraft.nbt.CompoundTag
+import net.minecraft.util.GsonHelper
+import net.minecraft.world.item.ItemStack
+
+class ItemRendererProvider(
+    val itemPath: NbtPath,
+    val blockPath: NbtPath?,
+    val countPath: NbtPath?,
+    val tagPath: NbtPath?,
+    val xOffset: Float,
+    val yOffset: Float,
+    val scale: Float,
+) : SplicingTableIotaRendererProvider {
+    override fun createRenderer(
+        type: IotaType<*>,
+        iota: SplicingTableIotaClientView,
+        x: Int,
+        y: Int
+    ): SplicingTableIotaRenderer? {
+        val data = iota.data ?: return null
+
+        val item = itemPath.getResourceLocationOrNull(data)?.let(BuiltInRegistries.ITEM::getOrNull)
+            ?: blockPath?.getResourceLocationOrNull(data)?.let(BuiltInRegistries.BLOCK::getOrNull)
+            ?: return null
+
+        val count = countPath?.getIntOrNull(data) ?: 1
+
+        val tag = tagPath?.getOrNull(data) as? CompoundTag
+
+        val stack = ItemStack(item, count)
+        stack.tag = tag
+
+        return SplicingTableIotaRenderer { guiGraphics, _, _, _ ->
+            val ps = guiGraphics.pose()
+            ps.pushPose {
+                // align to center of iota display
+                ps.translate(x + (18f / 2f) + xOffset, y + (21f / 2f) + yOffset, 0f)
+                ps.scale(scale)
+                // renderItem wants the top left corner, but it'll calculate it after the scale has been applied
+                ps.translate(-8f, -8f, 0f)
+                guiGraphics.renderItem(stack, 0, 0)
+                guiGraphics.renderItemDecorations(Minecraft.getInstance().font, stack, 0, 0)
+            }
+        }
+    }
+
+    companion object {
+        val PARSER = SplicingTableIotaRendererParser<ItemRendererProvider> { _, json, parent ->
+            ItemRendererProvider(
+                itemPath = json.getAsNbtPath("itemPath", parent?.itemPath),
+                blockPath = json.getAsNbtPathOrNull("blockPath", parent?.blockPath),
+                countPath = json.getAsNbtPathOrNull("countPath", parent?.countPath),
+                tagPath = json.getAsNbtPathOrNull("tagPath", parent?.tagPath),
+                xOffset = GsonHelper.getAsFloat(json, "xOffset", 0f),
+                yOffset = GsonHelper.getAsFloat(json, "yOffset", 0f),
+                scale = GsonHelper.getAsFloat(json, "scale", 0.75f),
+            )
+        }
+    }
+}

Common/src/main/kotlin/gay/object/hexdebug/gui/splicing/widgets/BaseIotaButton.kt

@@ -61,6 +61,7 @@ abstract class BaseIotaButton(x: Int, y: Int) : HexagonButton(
             ?: return
 
         renderer = provider.createRenderer(iotaType, iotaView, x, y)
+            ?: SplicingTableIotasResourceReloadListener.FALLBACK?.createRenderer(iotaType, iotaView, x, y)
         backgroundType = provider.getBackgroundType(iotaType, iotaView)
         tooltip = provider.createTooltip(iotaType, iotaView)
     }

Common/src/main/kotlin/gay/object/hexdebug/utils/ClientExtensions.kt

@@ -0,0 +1,13 @@
+package gay.`object`.hexdebug.utils
+
+import com.mojang.blaze3d.vertex.PoseStack
+
+fun PoseStack.pushPose(f: () -> Unit) {
+    pushPose()
+    f()
+    popPose()
+}
+
+fun PoseStack.scale(scale: Float) {
+    scale(scale, scale, scale) // scale
+}

Common/src/main/kotlin/gay/object/hexdebug/utils/Extensions.kt

@@ -12,7 +12,13 @@ import at.petrak.hexcasting.api.utils.*
 import at.petrak.hexcasting.common.casting.PatternRegistryManifest
 import at.petrak.hexcasting.xplat.IXplatAbstractions
 import gay.`object`.hexdebug.api.splicing.SplicingTableIotaClientView
+import net.minecraft.commands.arguments.NbtPathArgument.NbtPath
+import net.minecraft.core.Registry
+import net.minecraft.nbt.NumericTag
+import net.minecraft.nbt.StringTag
+import net.minecraft.nbt.Tag
 import net.minecraft.network.chat.*
+import net.minecraft.resources.ResourceLocation
 import net.minecraft.world.Container
 import net.minecraft.world.InteractionHand
 import net.minecraft.world.entity.LivingEntity
@@ -236,3 +242,34 @@ fun <T : Comparable<T>, V : T> BlockEntity.setPropertyIfChanged(property: Proper
         level?.setBlockAndUpdate(blockPos, blockState.setValue(property, value))
     }
 }
+
+// registries
+
+fun <T> Registry<T>.getOrNull(name: ResourceLocation): T? {
+    if (containsKey(name)) {
+        return get(name)
+    }
+    return null
+}
+
+// nbt paths
+
+fun NbtPath.getOrNull(tag: Tag): List<Tag>? {
+    return try {
+        get(tag)
+    } catch (e: Exception) {
+        null
+    }
+}
+
+fun NbtPath.getIntOrNull(tag: Tag): Int? {
+    return (getOrNull(tag)?.first() as? NumericTag)?.asInt
+}
+
+fun NbtPath.getStringOrNull(tag: Tag): String? {
+    return (getOrNull(tag)?.first() as? StringTag)?.asString
+}
+
+fun NbtPath.getResourceLocationOrNull(tag: Tag): ResourceLocation? {
+    return getStringOrNull(tag)?.let(ResourceLocation::tryParse)
+}

Common/src/main/kotlin/gay/object/hexdebug/utils/GsonHelper.kt

@@ -3,13 +3,36 @@
 package gay.`object`.hexdebug.utils
 
 import com.google.gson.JsonObject
+import com.google.gson.JsonSyntaxException
+import com.mojang.brigadier.StringReader
+import com.mojang.brigadier.exceptions.CommandSyntaxException
+import net.minecraft.commands.arguments.NbtPathArgument
+import net.minecraft.commands.arguments.NbtPathArgument.NbtPath
 import net.minecraft.resources.ResourceLocation
 import net.minecraft.util.GsonHelper
 
-fun JsonObject.getAsResourceLocation(memberName: String, fallback: ResourceLocation?): ResourceLocation {
+operator fun JsonObject.contains(memberName: String) = has(memberName)
+
+fun JsonObject.getAsResourceLocation(memberName: String, fallback: ResourceLocation? = null): ResourceLocation {
     return try {
         ResourceLocation(GsonHelper.getAsString(this, memberName))
     } catch (e: Exception) {
         fallback ?: throw e
     }
 }
+
+fun JsonObject.getAsNbtPathOrNull(memberName: String, fallback: NbtPath? = null): NbtPath? {
+    if (memberName in this) {
+        return getAsNbtPath(memberName)
+    }
+    return fallback
+}
+
+fun JsonObject.getAsNbtPath(memberName: String, fallback: NbtPath? = null): NbtPath {
+    val rawPath = GsonHelper.getAsString(this, memberName)
+    return try {
+        NbtPathArgument().parse(StringReader(rawPath))
+    } catch (e: CommandSyntaxException) {
+        fallback ?: throw JsonSyntaxException("Invalid $memberName, expected a valid NBT path", e)
+    }
+}

Common/src/main/resources/assets/moreiotas/hexdebug_splicing_iotas/item_stack.json

@@ -0,0 +1,6 @@
+{
+  "type": "hexdebug:item",
+  "itemPath": "moreiotas:stack_id",
+  "countPath": "moreiotas:stack_count",
+  "tagPath": "moreiotas:stack_tag"
+}

Common/src/main/resources/assets/moreiotas/hexdebug_splicing_iotas/item_type.json

@@ -1,3 +1,5 @@
 {
-  "parent": "hexdebug:builtin/type"
+  "type": "hexdebug:item",
+  "itemPath": "item",
+  "blockPath": "block"
 }