133 lines
4.9 KiB
Java
133 lines
4.9 KiB
Java
package net.minecraft.world.item.crafting;
|
|
|
|
import com.mojang.serialization.MapCodec;
|
|
import com.mojang.serialization.codecs.RecordCodecBuilder;
|
|
import java.util.List;
|
|
import java.util.Optional;
|
|
import net.minecraft.core.Holder;
|
|
import net.minecraft.core.HolderLookup;
|
|
import net.minecraft.core.Holder.Reference;
|
|
import net.minecraft.core.component.DataComponents;
|
|
import net.minecraft.network.RegistryFriendlyByteBuf;
|
|
import net.minecraft.network.codec.StreamCodec;
|
|
import net.minecraft.world.item.ItemStack;
|
|
import net.minecraft.world.item.Items;
|
|
import net.minecraft.world.item.crafting.display.RecipeDisplay;
|
|
import net.minecraft.world.item.crafting.display.SlotDisplay;
|
|
import net.minecraft.world.item.crafting.display.SmithingRecipeDisplay;
|
|
import net.minecraft.world.item.equipment.trim.ArmorTrim;
|
|
import net.minecraft.world.item.equipment.trim.TrimMaterial;
|
|
import net.minecraft.world.item.equipment.trim.TrimMaterials;
|
|
import net.minecraft.world.item.equipment.trim.TrimPattern;
|
|
import net.minecraft.world.item.equipment.trim.TrimPatterns;
|
|
import org.jetbrains.annotations.Nullable;
|
|
|
|
public class SmithingTrimRecipe implements SmithingRecipe {
|
|
final Optional<Ingredient> template;
|
|
final Optional<Ingredient> base;
|
|
final Optional<Ingredient> addition;
|
|
@Nullable
|
|
private PlacementInfo placementInfo;
|
|
|
|
public SmithingTrimRecipe(Optional<Ingredient> template, Optional<Ingredient> base, Optional<Ingredient> addition) {
|
|
this.template = template;
|
|
this.base = base;
|
|
this.addition = addition;
|
|
}
|
|
|
|
public ItemStack assemble(SmithingRecipeInput input, HolderLookup.Provider registries) {
|
|
return applyTrim(registries, input.base(), input.addition(), input.template());
|
|
}
|
|
|
|
public static ItemStack applyTrim(HolderLookup.Provider registries, ItemStack base, ItemStack addition, ItemStack template) {
|
|
Optional<Reference<TrimMaterial>> optional = TrimMaterials.getFromIngredient(registries, addition);
|
|
Optional<Reference<TrimPattern>> optional2 = TrimPatterns.getFromTemplate(registries, template);
|
|
if (optional.isPresent() && optional2.isPresent()) {
|
|
ArmorTrim armorTrim = base.get(DataComponents.TRIM);
|
|
if (armorTrim != null && armorTrim.hasPatternAndMaterial((Holder<TrimPattern>)optional2.get(), (Holder<TrimMaterial>)optional.get())) {
|
|
return ItemStack.EMPTY;
|
|
} else {
|
|
ItemStack itemStack = base.copyWithCount(1);
|
|
itemStack.set(DataComponents.TRIM, new ArmorTrim((Holder<TrimMaterial>)optional.get(), (Holder<TrimPattern>)optional2.get()));
|
|
return itemStack;
|
|
}
|
|
} else {
|
|
return ItemStack.EMPTY;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public Optional<Ingredient> templateIngredient() {
|
|
return this.template;
|
|
}
|
|
|
|
@Override
|
|
public Optional<Ingredient> baseIngredient() {
|
|
return this.base;
|
|
}
|
|
|
|
@Override
|
|
public Optional<Ingredient> additionIngredient() {
|
|
return this.addition;
|
|
}
|
|
|
|
@Override
|
|
public RecipeSerializer<SmithingTrimRecipe> getSerializer() {
|
|
return RecipeSerializer.SMITHING_TRIM;
|
|
}
|
|
|
|
@Override
|
|
public PlacementInfo placementInfo() {
|
|
if (this.placementInfo == null) {
|
|
this.placementInfo = PlacementInfo.createFromOptionals(List.of(this.template, this.base, this.addition));
|
|
}
|
|
|
|
return this.placementInfo;
|
|
}
|
|
|
|
@Override
|
|
public List<RecipeDisplay> display() {
|
|
SlotDisplay slotDisplay = Ingredient.optionalIngredientToDisplay(this.base);
|
|
SlotDisplay slotDisplay2 = Ingredient.optionalIngredientToDisplay(this.addition);
|
|
SlotDisplay slotDisplay3 = Ingredient.optionalIngredientToDisplay(this.template);
|
|
return List.of(
|
|
new SmithingRecipeDisplay(
|
|
slotDisplay3,
|
|
slotDisplay,
|
|
slotDisplay2,
|
|
new SlotDisplay.SmithingTrimDemoSlotDisplay(slotDisplay, slotDisplay2, slotDisplay3),
|
|
new SlotDisplay.ItemSlotDisplay(Items.SMITHING_TABLE)
|
|
)
|
|
);
|
|
}
|
|
|
|
public static class Serializer implements RecipeSerializer<SmithingTrimRecipe> {
|
|
private static final MapCodec<SmithingTrimRecipe> CODEC = RecordCodecBuilder.mapCodec(
|
|
instance -> instance.group(
|
|
Ingredient.CODEC.optionalFieldOf("template").forGetter(smithingTrimRecipe -> smithingTrimRecipe.template),
|
|
Ingredient.CODEC.optionalFieldOf("base").forGetter(smithingTrimRecipe -> smithingTrimRecipe.base),
|
|
Ingredient.CODEC.optionalFieldOf("addition").forGetter(smithingTrimRecipe -> smithingTrimRecipe.addition)
|
|
)
|
|
.apply(instance, SmithingTrimRecipe::new)
|
|
);
|
|
public static final StreamCodec<RegistryFriendlyByteBuf, SmithingTrimRecipe> STREAM_CODEC = StreamCodec.composite(
|
|
Ingredient.OPTIONAL_CONTENTS_STREAM_CODEC,
|
|
smithingTrimRecipe -> smithingTrimRecipe.template,
|
|
Ingredient.OPTIONAL_CONTENTS_STREAM_CODEC,
|
|
smithingTrimRecipe -> smithingTrimRecipe.base,
|
|
Ingredient.OPTIONAL_CONTENTS_STREAM_CODEC,
|
|
smithingTrimRecipe -> smithingTrimRecipe.addition,
|
|
SmithingTrimRecipe::new
|
|
);
|
|
|
|
@Override
|
|
public MapCodec<SmithingTrimRecipe> codec() {
|
|
return CODEC;
|
|
}
|
|
|
|
@Override
|
|
public StreamCodec<RegistryFriendlyByteBuf, SmithingTrimRecipe> streamCodec() {
|
|
return STREAM_CODEC;
|
|
}
|
|
}
|
|
}
|