From 0547cfbfafefa8c48667d026a2fc9ad2ab132b51 Mon Sep 17 00:00:00 2001
From: Sakura Ryoko <sakura@ryoko.soy>
Date: Tue, 10 Dec 2024 19:17:40 -0500
Subject: [PATCH] Update NbtUtils with Vec3i Int Array tags

---
 .../fi/dy/masa/malilib/util/nbt/NbtUtils.java | 42 +++++++++++++++++++
 1 file changed, 42 insertions(+)

diff --git a/src/main/java/fi/dy/masa/malilib/util/nbt/NbtUtils.java b/src/main/java/fi/dy/masa/malilib/util/nbt/NbtUtils.java
index 68a904d42b..5cdf25c208 100644
--- a/src/main/java/fi/dy/masa/malilib/util/nbt/NbtUtils.java
+++ b/src/main/java/fi/dy/masa/malilib/util/nbt/NbtUtils.java
@@ -100,6 +100,16 @@ public static <T> NbtList asListTag(Collection<T> values, Function<T, NbtElement
         return putVec3i(new NbtCompound(), pos);
     }
 
+    public static @NotNull NbtCompound createVec3iToArray(@Nonnull Vec3i pos, String tagName)
+    {
+        return writeBlockPosToArrayTag(pos, new NbtCompound(), tagName);
+    }
+
+    public static @NotNull NbtCompound createVec3iToArrayTag(@Nonnull Vec3i pos, String tagName)
+    {
+        return writeBlockPosToArrayTag(pos, new NbtCompound(), tagName);
+    }
+
     public static @NotNull NbtCompound createEntityPosition(@Nonnull Vec3d pos)
     {
         return createEntityPositionToTag(pos);
@@ -144,6 +154,16 @@ public static <T> NbtList asListTag(Collection<T> values, Function<T, NbtElement
         return tag;
     }
 
+    public static @NotNull NbtCompound writeVec3iToArray(@Nonnull Vec3i pos, @Nonnull NbtCompound tag, String tagName)
+    {
+        return writeBlockPosToArrayTag(pos, tag, tagName);
+    }
+
+    public static @NotNull NbtCompound writeVec3iToArrayTag(@Nonnull Vec3i pos, @Nonnull NbtCompound tag, String tagName)
+    {
+        return writeBlockPosToArrayTag(pos, tag, tagName);
+    }
+
     public static @NotNull NbtCompound writeBlockPosToArrayTag(@Nonnull Vec3i pos, @Nonnull NbtCompound tag, String tagName)
     {
         int[] arr = new int[] { pos.getX(), pos.getY(), pos.getZ() };
@@ -205,6 +225,28 @@ public static BlockPos readBlockPosFromArrayTag(@Nonnull NbtCompound tag, String
         return null;
     }
 
+    @Nullable
+    public static Vec3i readVec3iFromIntArray(@Nonnull NbtCompound nbt, String key)
+    {
+        return readVec3iFromIntArrayTag(nbt, key);
+    }
+
+    @Nullable
+    public static Vec3i readVec3iFromIntArrayTag(@Nonnull NbtCompound tag, String tagName)
+    {
+        if (NbtWrap.containsIntArray(tag, tagName))
+        {
+            int[] pos = NbtWrap.getIntArray(tag, tagName);
+
+            if (pos.length == 3)
+            {
+                return new Vec3i(pos[0], pos[1], pos[2]);
+            }
+        }
+
+        return null;
+    }
+
     public static @NotNull NbtCompound removeBlockPos(@Nonnull NbtCompound tag)
     {
         return removeBlockPosFromTag(tag);