Loot datagen

Author: wdog5734

src/main/java/org/polaris_bear/wild_wind/common/entity/Firefly.java

@@ -37,8 +37,8 @@
 import software.bernie.geckolib.util.GeckoLibUtil;
 
 public class Firefly extends PathfinderMob implements FlyingAnimal, GeoEntity {
-    private final AnimatableInstanceCache cache = GeckoLibUtil.createInstanceCache(this);
-    private static final EntityDataAccessor<Boolean> roost = SynchedEntityData.defineId(Firefly.class, EntityDataSerializers.BOOLEAN);
+    private final AnimatableInstanceCache CACHE = GeckoLibUtil.createInstanceCache(this);
+    private static final EntityDataAccessor<Boolean> ROOST = SynchedEntityData.defineId(Firefly.class, EntityDataSerializers.BOOLEAN);
     public Firefly(EntityType<? extends PathfinderMob> type, Level level) {
         super(type, level);
         this.moveControl = new FlyingMoveControl(this, 20, true);
@@ -47,7 +47,7 @@ public Firefly(EntityType<? extends PathfinderMob> type, Level level) {
     @Override
     protected void defineSynchedData(SynchedEntityData.Builder builder) {
         super.defineSynchedData(builder);
-        builder.define(roost, false);
+        builder.define(ROOST, false);
     }
 
     @Nullable
@@ -58,11 +58,11 @@ public SpawnGroupData finalizeSpawn(ServerLevelAccessor level, DifficultyInstanc
     }
 
     public void setRoost(boolean r) {
-        this.entityData.set(roost, r);
+        this.entityData.set(ROOST, r);
     }
 
     public boolean isRoost() {
-        return this.entityData.get(roost);
+        return this.entityData.get(ROOST);
     }
 
 
@@ -173,6 +173,6 @@ private PlayState handle(AnimationState<Firefly> fireflyAnimationState) {
      */
     @Override
     public AnimatableInstanceCache getAnimatableInstanceCache() {
-        return cache;
+        return CACHE;
     }
 }

src/main/java/org/polaris_bear/wild_wind/datagen/ModDataGenerator.java

@@ -6,6 +6,7 @@
 import org.polaris_bear.wild_wind.WildWindMod;
 import org.polaris_bear.wild_wind.datagen.lang.ModLangProviderEn;
 import org.polaris_bear.wild_wind.datagen.lang.ModLangProviderZh;
+import org.polaris_bear.wild_wind.datagen.loot.ModLootTableProvider;
 
 @EventBusSubscriber(bus = EventBusSubscriber.Bus.MOD, modid = WildWindMod.MOD_ID)
 public class ModDataGenerator {
@@ -21,5 +22,6 @@ public static void gatherData(GatherDataEvent event) {
         gen.addProvider(event.includeClient(), new ModItemModelProvider(pack, helper));
         gen.addProvider(event.includeServer(), new ModEntityTypeTagsProvider(pack, provider, helper));
         gen.addProvider(event.includeServer(), new ModBlockTagsProvider(pack, provider, helper));
+        gen.addProvider(event.includeServer(), new ModLootTableProvider(pack, provider));
     }
 }

src/main/java/org/polaris_bear/wild_wind/datagen/loot/ModEntityLoot.java

@@ -0,0 +1,48 @@
+package org.polaris_bear.wild_wind.datagen.loot;
+
+import net.minecraft.core.HolderLookup;
+import net.minecraft.data.loot.EntityLootSubProvider;
+import net.minecraft.world.entity.EntityType;
+import net.minecraft.world.flag.FeatureFlags;
+import net.minecraft.world.item.Items;
+import net.minecraft.world.level.storage.loot.LootPool;
+import net.minecraft.world.level.storage.loot.LootTable;
+import net.minecraft.world.level.storage.loot.entries.LootItem;
+import net.minecraft.world.level.storage.loot.functions.SmeltItemFunction;
+import net.minecraft.world.level.storage.loot.predicates.LootItemRandomChanceCondition;
+import net.minecraft.world.level.storage.loot.providers.number.ConstantValue;
+import net.neoforged.neoforge.registries.DeferredHolder;
+import org.polaris_bear.wild_wind.common.init.ModEntities;
+
+import java.util.stream.Stream;
+
+public class ModEntityLoot extends EntityLootSubProvider {
+
+    public ModEntityLoot(HolderLookup.Provider registries) {
+        super(FeatureFlags.REGISTRY.allFlags(), registries);
+    }
+
+    @Override
+    public void generate() {
+        this.add(
+                ModEntities.FIREFLY.get(),
+                LootTable.lootTable()
+                        .withPool(
+                                LootPool.lootPool()
+                                        .setRolls(ConstantValue.exactly(1.0F))
+                                        .add(LootItem.lootTableItem(Items.COD).apply(SmeltItemFunction.smelted().when(this.shouldSmeltLoot())))
+                        )
+                        .withPool(
+                                LootPool.lootPool()
+                                        .setRolls(ConstantValue.exactly(1.0F))
+                                        .add(LootItem.lootTableItem(Items.BONE_MEAL))
+                                        .when(LootItemRandomChanceCondition.randomChance(0.05F))
+                        )
+        );
+    }
+
+    @Override
+    protected Stream<EntityType<?>> getKnownEntityTypes() {
+        return ModEntities.ENTITIES.getEntries().stream().map(DeferredHolder::get);
+    }
+}

src/main/java/org/polaris_bear/wild_wind/datagen/loot/ModLootTableProvider.java

@@ -0,0 +1,31 @@
+package org.polaris_bear.wild_wind.datagen.loot;
+
+import net.minecraft.core.HolderLookup;
+import net.minecraft.core.WritableRegistry;
+import net.minecraft.data.PackOutput;
+import net.minecraft.data.loot.LootTableProvider;
+import net.minecraft.util.ProblemReporter;
+import net.minecraft.world.level.storage.loot.LootTable;
+import net.minecraft.world.level.storage.loot.ValidationContext;
+import net.minecraft.world.level.storage.loot.parameters.LootContextParamSets;
+
+import java.util.List;
+import java.util.Set;
+import java.util.concurrent.CompletableFuture;
+
+public class ModLootTableProvider extends LootTableProvider {
+
+    public ModLootTableProvider(PackOutput output, CompletableFuture<HolderLookup.Provider> registries) {
+        super(output, Set.of(), subProviderEntries(), registries);
+    }
+
+    static List<SubProviderEntry> subProviderEntries() {
+        return List.of(
+                new SubProviderEntry(ModEntityLoot::new, LootContextParamSets.ENTITY)
+        );
+    }
+
+    @Override
+    protected void validate(WritableRegistry<LootTable> writableregistry, ValidationContext validationcontext, ProblemReporter.Collector problemreporter$collector) {
+    }
+}