Fix goomba data flags not saving

WenXin20 · Jan 7, 2025 · 7fbcc87 · 7fbcc87
1 parent 0e57b32
commit 7fbcc87
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/src/main/java/com/wenxin2/marioverse/entities/GoombaEntity.java b/src/main/java/com/wenxin2/marioverse/entities/GoombaEntity.java
@@ -22,6 +22,7 @@
 import net.minecraft.core.particles.BlockParticleOption;
 import net.minecraft.core.particles.ParticleTypes;
 import net.minecraft.core.registries.Registries;
+import net.minecraft.nbt.CompoundTag;
 import net.minecraft.network.syncher.EntityDataAccessor;
 import net.minecraft.network.syncher.EntityDataSerializers;
 import net.minecraft.network.syncher.SynchedEntityData;
@@ -232,6 +233,24 @@ public AnimatableInstanceCache getAnimatableInstanceCache() {
         return cache;
     }
 
+    @Override
+    public void readAdditionalSaveData(CompoundTag tag) {
+        super.readAdditionalSaveData(tag);
+        this.entityData.set(DATA_ID_RIDE_FLAGS, tag.getByte("RideFlags"));
+        this.entityData.set(DATA_ID_SCARE_FLAGS, tag.getByte("ScareFlags"));
+        this.entityData.set(DATA_ID_SIT_FLAGS, tag.getByte("SitFlags"));
+        this.entityData.set(DATA_ID_SLEEP_FLAGS, tag.getByte("SleepFlags"));
+    }
+
+    @Override
+    public void addAdditionalSaveData(CompoundTag tag) {
+        super.addAdditionalSaveData(tag);
+        tag.putByte("RideFlags", this.entityData.get(DATA_ID_RIDE_FLAGS));
+        tag.putByte("ScareFlags", this.entityData.get(DATA_ID_SCARE_FLAGS));
+        tag.putByte("SitFlags", this.entityData.get(DATA_ID_SIT_FLAGS));
+        tag.putByte("SleepFlags", this.entityData.get(DATA_ID_SLEEP_FLAGS));
+    }
+
     public boolean isSitting() {
         return this.getSitFlag(8);
     }