1
package net.wssz.tutorialmod.datagen;
2

3
import net.fabricmc.fabric.api.datagen.v1.FabricDataOutput;
4
import net.fabricmc.fabric.api.datagen.v1.provider.FabricModelProvider;
5
import net.minecraft.data.client.BlockStateModelGenerator;
6
import net.minecraft.data.client.ItemModelGenerator;
7
import net.minecraft.data.client.Models;
8
import net.minecraft.item.Item;
9
import net.wssz.tutorialmod.block.ModBlocks;
10
import net.wssz.tutorialmod.item.ModItems;
11

12
import java.lang.reflect.Field;
13

14
public class ModModelsProvider extends FabricModelProvider {
15
    public ModModelsProvider(FabricDataOutput output) {
16
        super(output);
17
    }
18

19
    @Override
20
    public void generateBlockStateModels(BlockStateModelGenerator blockStateModelGenerator) {
21
        blockStateModelGenerator.registerSimpleCubeAll(ModBlocks.RED_IRON_ORE);
22
    }
23

24
    @Override
25
    public void generateItemModels(ItemModelGenerator itemModelGenerator) {
26
        Field[] fields = ModItems.class.getDeclaredFields();
27
        try {
28
            for (Field field : fields) {
29
                if (field.getType().equals(Item.class)) {
30
                    itemModelGenerator.register((Item) field.get(null), Models.GENERATED);
31
                }
32
            }
33
        } catch (IllegalAccessException e) {
34
            throw new RuntimeException(e);
35
        }
36
    }
37
}