HELP ME

src/main/java/com/volmit/iris/Iris.java

@@ -129,7 +129,7 @@ public class Iris extends VolmitPlugin implements Listener {
     private void testmca() {
         try
         {
-            int forceBits = 6;
+            int forceBits = 5;
             int possibilities = (int) (Math.pow(2, forceBits) - 1);
             KList<BlockData> bp = new KList<>();
             Set<BlockData> bf = new KSet<>();

src/main/java/com/volmit/iris/core/nms/INMSBinding.java

@@ -18,6 +18,7 @@
 
 package com.volmit.iris.core.nms;
 
+import com.volmit.iris.util.nbt.mca.nmspalettes.PaletteAccess;
 import com.volmit.iris.util.nbt.mca.palettes.RegistryBlockID;
 import com.volmit.iris.util.nbt.tag.CompoundTag;
 import org.bukkit.Location;
@@ -76,5 +77,5 @@ public interface INMSBinding {
         return false;
     }
 
-    RegistryBlockID computeBlockIDRegistry() throws NoSuchFieldException, IllegalAccessException;
+    PaletteAccess createPalette();
 }

src/main/java/com/volmit/iris/core/nms/v17_1/NMSBinding17_1.java

@@ -20,9 +20,11 @@ package com.volmit.iris.core.nms.v17_1;
 
 import com.volmit.iris.Iris;
 import com.volmit.iris.core.nms.INMSBinding;
+import com.volmit.iris.engine.data.cache.AtomicCache;
 import com.volmit.iris.util.collection.KMap;
 import com.volmit.iris.util.nbt.io.NBTUtil;
 import com.volmit.iris.util.nbt.mca.NBTWorld;
+import com.volmit.iris.util.nbt.mca.nmspalettes.*;
 import com.volmit.iris.util.nbt.mca.palettes.RegistryBlockID;
 import com.volmit.iris.util.nbt.tag.CompoundTag;
 import net.minecraft.core.BlockPosition;
@@ -46,6 +48,7 @@ import net.minecraft.world.level.chunk.Chunk;
 import net.minecraft.world.level.chunk.ChunkSection;
 import org.bukkit.Bukkit;
 import org.bukkit.Location;
+import org.bukkit.Material;
 import org.bukkit.World;
 import org.bukkit.block.Biome;
 import org.bukkit.block.data.BlockData;
@@ -66,7 +69,10 @@ import java.util.List;
 import java.util.concurrent.atomic.AtomicInteger;
 
 public class NMSBinding17_1 implements INMSBinding {
+    private final BlockData AIR = Material.AIR.createBlockData();
     private final KMap<Biome, Object> baseBiomeCache = new KMap<>();
+    private final AtomicCache<IdMapper<IBlockData>> registryCache = new AtomicCache<>();
+    private final AtomicCache<Palette<IBlockData>> globalCache = new AtomicCache<>();
     private Field biomeStorageCache = null;
 
     public boolean supportsDataPacks() {
@@ -74,21 +80,28 @@ public class NMSBinding17_1 implements INMSBinding {
     }
 
     @Override
-    public RegistryBlockID computeBlockIDRegistry() throws NoSuchFieldException, IllegalAccessException {
+    public PaletteAccess createPalette() {
+        IdMapper<IBlockData> registry = registryCache.aquireNasty(() -> {
             Field cf = net.minecraft.core.RegistryBlockID.class.getDeclaredField("c");
             Field df = net.minecraft.core.RegistryBlockID.class.getDeclaredField("d");
+            Field bf = net.minecraft.core.RegistryBlockID.class.getDeclaredField("b");
             cf.setAccessible(true);
             df.setAccessible(true);
+            bf.setAccessible(true);
             net.minecraft.core.RegistryBlockID<IBlockData> blockData = Block.p;
+            int b = bf.getInt(blockData);
             IdentityHashMap<IBlockData, Integer> c = (IdentityHashMap<IBlockData, Integer>) cf.get(blockData);
             List<IBlockData> d = (List<IBlockData>) df.get(blockData);
-        List<CompoundTag> realTags = new ArrayList<>();
+            return new IdMapper<>(c, d, b);
-        HashMap<CompoundTag, Integer> realMap = new HashMap<>(512);
+        });
-        d.forEach((i) -> realTags.add(NBTWorld.getCompound(CraftBlockData.fromData(i))));
+        Palette<IBlockData> global = globalCache.aquireNasty(() -> new GlobalPalette<>(registry, ((CraftBlockData)AIR).getState()));
-        c.forEach((k,v) -> realMap.put(NBTWorld.getCompound(CraftBlockData.fromData(k)), v));
+        PalettedContainer<IBlockData> container = new PalettedContainer<>(global, registry,
-        RegistryBlockID registry = new RegistryBlockID(realMap, realTags);
+                i -> ((CraftBlockData)NBTWorld.getBlockData(i)).getState(),
-        Iris.info("INMS: Stole Global Palette: " + realTags.size() + " Tags, " + realMap.size() + " Mapped");
+                i -> NBTWorld.getCompound(CraftBlockData.fromData(i)),
-        return registry;
+                ((CraftBlockData) AIR).getState());
+        return new WrappedPalettedContainer<>(container,
+                i -> NBTWorld.getCompound(CraftBlockData.fromData(i)),
+                i -> ((CraftBlockData)NBTWorld.getBlockData(i)).getState());
     }
 
     private Object getBiomeStorage(ChunkGenerator.BiomeGrid g) {

src/main/java/com/volmit/iris/core/nms/v1X/NMSBinding1X.java

@@ -20,6 +20,7 @@ package com.volmit.iris.core.nms.v1X;
 
 import com.volmit.iris.Iris;
 import com.volmit.iris.core.nms.INMSBinding;
+import com.volmit.iris.util.nbt.mca.nmspalettes.PaletteAccess;
 import com.volmit.iris.util.nbt.mca.palettes.RegistryBlockID;
 import com.volmit.iris.util.nbt.tag.CompoundTag;
 import org.bukkit.Location;
@@ -146,7 +147,7 @@ public class NMSBinding1X implements INMSBinding {
     }
 
     @Override
-    public RegistryBlockID computeBlockIDRegistry() throws NoSuchFieldException, IllegalAccessException {
+    public PaletteAccess createPalette() {
         Iris.error("Cannot use the global data palette! Iris is incapable of using MCA generation on this version of minecraft!");
         return null;
     }

src/main/java/com/volmit/iris/engine/data/cache/AtomicCache.java

@@ -19,6 +19,7 @@
 package com.volmit.iris.engine.data.cache;
 
 import com.volmit.iris.Iris;
+import com.volmit.iris.util.function.NastySupplier;
 
 import java.util.concurrent.atomic.AtomicBoolean;
 import java.util.concurrent.atomic.AtomicReference;
@@ -50,6 +51,21 @@ public class AtomicCache<T> {
         }
     }
 
+    public T aquireNasty(NastySupplier<T> t)
+    {
+        return aquire(() -> {
+            try
+            {
+                return t.get();
+            }
+
+            catch(Throwable e)
+            {
+                return null;
+            }
+        });
+    }
+
     public T aquire(Supplier<T> t) {
         if (this.t.get() != null) {
             return this.t.get();