package me.shedaniel.rei.plugin.smithing; import com.google.common.collect.Lists; import me.shedaniel.rei.api.EntryStack; import me.shedaniel.rei.api.RecipeDisplay; import me.shedaniel.rei.plugin.DefaultPlugin; import me.shedaniel.rei.utils.CollectionUtils; import net.minecraft.recipe.SmithingRecipe; import net.minecraft.util.Identifier; import org.jetbrains.annotations.NotNull; import org.jetbrains.annotations.Nullable; import java.util.Collections; import java.util.List; import java.util.Optional; public class DefaultSmithingDisplay implements RecipeDisplay { @NotNull private List> input; @NotNull private List output; @Nullable private Identifier location; public DefaultSmithingDisplay(@NotNull SmithingRecipe recipe) { this( Lists.newArrayList( CollectionUtils.map(recipe.base.getMatchingStacksClient(), EntryStack::create), CollectionUtils.map(recipe.addition.getMatchingStacksClient(), EntryStack::create) ), Collections.singletonList(EntryStack.create(recipe.getOutput())), recipe.getId() ); } public DefaultSmithingDisplay(@NotNull List> input, @NotNull List output, @Nullable Identifier location) { this.input = input; this.output = output; if (this.input.size() != 2) throw new IllegalArgumentException("input must have 2 entries."); this.location = location; } @Override public List> getInputEntries() { return input; } @Override public List getOutputEntries() { return output; } @Override public Identifier getRecipeCategory() { return DefaultPlugin.SMITHING; } @Override public Optional getRecipeLocation() { return Optional.ofNullable(location); } }