123 lines
5.3 KiB
Java
123 lines
5.3 KiB
Java
package net.minecraft.client.data.models;
|
|
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
import java.util.Optional;
|
|
import java.util.Map.Entry;
|
|
import java.util.concurrent.CompletableFuture;
|
|
import java.util.function.BiConsumer;
|
|
import net.fabricmc.api.EnvType;
|
|
import net.fabricmc.api.Environment;
|
|
import net.minecraft.client.resources.model.EquipmentClientInfo;
|
|
import net.minecraft.data.CachedOutput;
|
|
import net.minecraft.data.DataProvider;
|
|
import net.minecraft.data.PackOutput;
|
|
import net.minecraft.data.PackOutput.PathProvider;
|
|
import net.minecraft.data.PackOutput.Target;
|
|
import net.minecraft.resources.ResourceKey;
|
|
import net.minecraft.resources.ResourceLocation;
|
|
import net.minecraft.world.item.DyeColor;
|
|
import net.minecraft.world.item.equipment.EquipmentAsset;
|
|
import net.minecraft.world.item.equipment.EquipmentAssets;
|
|
|
|
@Environment(EnvType.CLIENT)
|
|
public class EquipmentAssetProvider implements DataProvider {
|
|
private final PathProvider pathProvider;
|
|
|
|
public EquipmentAssetProvider(PackOutput output) {
|
|
this.pathProvider = output.createPathProvider(Target.RESOURCE_PACK, "equipment");
|
|
}
|
|
|
|
private static void bootstrap(BiConsumer<ResourceKey<EquipmentAsset>, EquipmentClientInfo> output) {
|
|
output.accept(
|
|
EquipmentAssets.LEATHER,
|
|
EquipmentClientInfo.builder()
|
|
.addHumanoidLayers(ResourceLocation.withDefaultNamespace("leather"), true)
|
|
.addHumanoidLayers(ResourceLocation.withDefaultNamespace("leather_overlay"), false)
|
|
.addLayers(EquipmentClientInfo.LayerType.HORSE_BODY, EquipmentClientInfo.Layer.leatherDyeable(ResourceLocation.withDefaultNamespace("leather"), true))
|
|
.build()
|
|
);
|
|
output.accept(EquipmentAssets.CHAINMAIL, onlyHumanoid("chainmail"));
|
|
output.accept(EquipmentAssets.IRON, humanoidAndHorse("iron"));
|
|
output.accept(EquipmentAssets.GOLD, humanoidAndHorse("gold"));
|
|
output.accept(EquipmentAssets.DIAMOND, humanoidAndHorse("diamond"));
|
|
output.accept(
|
|
EquipmentAssets.TURTLE_SCUTE, EquipmentClientInfo.builder().addMainHumanoidLayer(ResourceLocation.withDefaultNamespace("turtle_scute"), false).build()
|
|
);
|
|
output.accept(EquipmentAssets.NETHERITE, onlyHumanoid("netherite"));
|
|
output.accept(
|
|
EquipmentAssets.ARMADILLO_SCUTE,
|
|
EquipmentClientInfo.builder()
|
|
.addLayers(EquipmentClientInfo.LayerType.WOLF_BODY, EquipmentClientInfo.Layer.onlyIfDyed(ResourceLocation.withDefaultNamespace("armadillo_scute"), false))
|
|
.addLayers(
|
|
EquipmentClientInfo.LayerType.WOLF_BODY, EquipmentClientInfo.Layer.onlyIfDyed(ResourceLocation.withDefaultNamespace("armadillo_scute_overlay"), true)
|
|
)
|
|
.build()
|
|
);
|
|
output.accept(
|
|
EquipmentAssets.ELYTRA,
|
|
EquipmentClientInfo.builder()
|
|
.addLayers(EquipmentClientInfo.LayerType.WINGS, new EquipmentClientInfo.Layer(ResourceLocation.withDefaultNamespace("elytra"), Optional.empty(), true))
|
|
.build()
|
|
);
|
|
EquipmentClientInfo.Layer layer = new EquipmentClientInfo.Layer(ResourceLocation.withDefaultNamespace("saddle"));
|
|
output.accept(
|
|
EquipmentAssets.SADDLE,
|
|
EquipmentClientInfo.builder()
|
|
.addLayers(EquipmentClientInfo.LayerType.PIG_SADDLE, layer)
|
|
.addLayers(EquipmentClientInfo.LayerType.STRIDER_SADDLE, layer)
|
|
.addLayers(EquipmentClientInfo.LayerType.CAMEL_SADDLE, layer)
|
|
.addLayers(EquipmentClientInfo.LayerType.HORSE_SADDLE, layer)
|
|
.addLayers(EquipmentClientInfo.LayerType.DONKEY_SADDLE, layer)
|
|
.addLayers(EquipmentClientInfo.LayerType.MULE_SADDLE, layer)
|
|
.addLayers(EquipmentClientInfo.LayerType.SKELETON_HORSE_SADDLE, layer)
|
|
.addLayers(EquipmentClientInfo.LayerType.ZOMBIE_HORSE_SADDLE, layer)
|
|
.build()
|
|
);
|
|
|
|
for (Entry<DyeColor, ResourceKey<EquipmentAsset>> entry : EquipmentAssets.CARPETS.entrySet()) {
|
|
DyeColor dyeColor = (DyeColor)entry.getKey();
|
|
ResourceKey<EquipmentAsset> resourceKey = (ResourceKey<EquipmentAsset>)entry.getValue();
|
|
output.accept(
|
|
resourceKey,
|
|
EquipmentClientInfo.builder()
|
|
.addLayers(EquipmentClientInfo.LayerType.LLAMA_BODY, new EquipmentClientInfo.Layer(ResourceLocation.withDefaultNamespace(dyeColor.getSerializedName())))
|
|
.build()
|
|
);
|
|
}
|
|
|
|
output.accept(
|
|
EquipmentAssets.TRADER_LLAMA,
|
|
EquipmentClientInfo.builder()
|
|
.addLayers(EquipmentClientInfo.LayerType.LLAMA_BODY, new EquipmentClientInfo.Layer(ResourceLocation.withDefaultNamespace("trader_llama")))
|
|
.build()
|
|
);
|
|
}
|
|
|
|
private static EquipmentClientInfo onlyHumanoid(String name) {
|
|
return EquipmentClientInfo.builder().addHumanoidLayers(ResourceLocation.withDefaultNamespace(name)).build();
|
|
}
|
|
|
|
private static EquipmentClientInfo humanoidAndHorse(String name) {
|
|
return EquipmentClientInfo.builder()
|
|
.addHumanoidLayers(ResourceLocation.withDefaultNamespace(name))
|
|
.addLayers(EquipmentClientInfo.LayerType.HORSE_BODY, EquipmentClientInfo.Layer.leatherDyeable(ResourceLocation.withDefaultNamespace(name), false))
|
|
.build();
|
|
}
|
|
|
|
@Override
|
|
public CompletableFuture<?> run(CachedOutput output) {
|
|
Map<ResourceKey<EquipmentAsset>, EquipmentClientInfo> map = new HashMap();
|
|
bootstrap((resourceKey, equipmentClientInfo) -> {
|
|
if (map.putIfAbsent(resourceKey, equipmentClientInfo) != null) {
|
|
throw new IllegalStateException("Tried to register equipment asset twice for id: " + resourceKey);
|
|
}
|
|
});
|
|
return DataProvider.saveAll(output, EquipmentClientInfo.CODEC, this.pathProvider::json, map);
|
|
}
|
|
|
|
@Override
|
|
public String getName() {
|
|
return "Equipment Asset Definitions";
|
|
}
|
|
}
|