package com.selfreplay.data; import net.minecraft.item.ItemStack; import net.minecraft.nbt.NbtCompound; import net.minecraft.nbt.NbtList; import net.minecraft.registry.RegistryWrapper; import java.util.ArrayList; import java.util.List; public class ReplayFrame { public long tick; public double posX, posY, posZ; public float yaw, pitch; public double velX, velY, velZ; public boolean sneaking, sprinting, jumping, flying; public int heldSlot; public ItemStack[] inventory; public ItemStack[] armor; public List actions = new ArrayList<>(); public List particles = new ArrayList<>(); public List sounds = new ArrayList<>(); public String chatMessage; public float health; public int food; public NbtCompound toNbt(RegistryWrapper.WrapperLookup registries) { NbtCompound nbt = new NbtCompound(); nbt.putLong("tick", tick); nbt.putDouble("x", posX); nbt.putDouble("y", posY); nbt.putDouble("z", posZ); nbt.putFloat("yaw", yaw); nbt.putFloat("pitch", pitch); nbt.putDouble("vx", velX); nbt.putDouble("vy", velY); nbt.putDouble("vz", velZ); nbt.putBoolean("sneak", sneaking); nbt.putBoolean("sprint", sprinting); nbt.putBoolean("jump", jumping); nbt.putBoolean("fly", flying); nbt.putInt("slot", heldSlot); nbt.putFloat("health", health); nbt.putInt("food", food); if (inventory != null) { NbtList invList = new NbtList(); for (int i = 0; i < inventory.length; i++) { if (inventory[i] != null && !inventory[i].isEmpty()) { NbtCompound slotNbt = inventory[i].toNbtAllowEmpty(registries); slotNbt.putInt("Slot", i); invList.add(slotNbt); } } if (!invList.isEmpty()) { nbt.put("inv", invList); } } if (armor != null) { NbtList armorList = new NbtList(); for (int i = 0; i < armor.length; i++) { if (armor[i] != null && !armor[i].isEmpty()) { NbtCompound slotNbt = armor[i].toNbtAllowEmpty(registries); slotNbt.putInt("Slot", i); armorList.add(slotNbt); } } if (!armorList.isEmpty()) { nbt.put("armor", armorList); } } if (!actions.isEmpty()) { NbtList actionList = new NbtList(); for (ActionData action : actions) { actionList.add(action.toNbt()); } nbt.put("actions", actionList); } if (!particles.isEmpty()) { NbtList particleList = new NbtList(); for (ParticleData particle : particles) { particleList.add(particle.toNbt()); } nbt.put("particles", particleList); } if (!sounds.isEmpty()) { NbtList soundList = new NbtList(); for (SoundData sound : sounds) { soundList.add(sound.toNbt()); } nbt.put("sounds", soundList); } if (chatMessage != null && !chatMessage.isEmpty()) { nbt.putString("chat", chatMessage); } return nbt; } public static ReplayFrame fromNbt(NbtCompound nbt, RegistryWrapper.WrapperLookup registries) { ReplayFrame frame = new ReplayFrame(); frame.tick = nbt.getLong("tick", 0); frame.posX = nbt.getDouble("x", 0); frame.posY = nbt.getDouble("y", 0); frame.posZ = nbt.getDouble("z", 0); frame.yaw = nbt.getFloat("yaw", 0); frame.pitch = nbt.getFloat("pitch", 0); frame.velX = nbt.getDouble("vx", 0); frame.velY = nbt.getDouble("vy", 0); frame.velZ = nbt.getDouble("vz", 0); frame.sneaking = nbt.getBoolean("sneak"); frame.sprinting = nbt.getBoolean("sprint"); frame.jumping = nbt.getBoolean("jump"); frame.flying = nbt.getBoolean("fly"); frame.heldSlot = nbt.getInt("slot", 0); frame.health = nbt.getFloat("health", 20); frame.food = nbt.getInt("food", 20); if (nbt.contains("inv")) { NbtList invList = nbt.getList("inv", 10); frame.inventory = new ItemStack[41]; for (int i = 0; i < invList.size(); i++) { NbtCompound slotNbt = invList.getCompound(i); int slot = slotNbt.getInt("Slot", 0); if (slot >= 0 && slot < 41) { frame.inventory[slot] = ItemStack.fromNbtOrEmpty(registries, slotNbt).orElse(ItemStack.EMPTY); } } } if (nbt.contains("armor")) { NbtList armorList = nbt.getList("armor", 10); frame.armor = new ItemStack[4]; for (int i = 0; i < armorList.size(); i++) { NbtCompound slotNbt = armorList.getCompound(i); int slot = slotNbt.getInt("Slot", 0); if (slot >= 0 && slot < 4) { frame.armor[slot] = ItemStack.fromNbtOrEmpty(registries, slotNbt).orElse(ItemStack.EMPTY); } } } if (nbt.contains("actions")) { NbtList actionList = nbt.getList("actions", 10); for (int i = 0; i < actionList.size(); i++) { frame.actions.add(ActionData.fromNbt(actionList.getCompound(i))); } } if (nbt.contains("particles")) { NbtList particleList = nbt.getList("particles", 10); for (int i = 0; i < particleList.size(); i++) { frame.particles.add(ParticleData.fromNbt(particleList.getCompound(i))); } } if (nbt.contains("sounds")) { NbtList soundList = nbt.getList("sounds", 10); for (int i = 0; i < soundList.size(); i++) { frame.sounds.add(SoundData.fromNbt(soundList.getCompound(i))); } } if (nbt.contains("chat")) { frame.chatMessage = nbt.getString("chat"); } return frame; } }