99 lines
2.5 KiB
Java
99 lines
2.5 KiB
Java
package com.extendedae_plus.util;
|
||
|
||
import appeng.api.stacks.AEFluidKey;
|
||
import appeng.api.stacks.AEItemKey;
|
||
import appeng.api.stacks.GenericStack;
|
||
import net.minecraft.resources.ResourceLocation;
|
||
import net.minecraft.world.item.Item;
|
||
import net.minecraft.world.item.ItemStack;
|
||
|
||
import java.util.ArrayList;
|
||
import java.util.List;
|
||
import java.util.Map;
|
||
|
||
/**
|
||
* 配方完整信息（包含配方ID、输入槽位候选项、输出）
|
||
*/
|
||
public class RecipeInfo {
|
||
private final Object recipeBase;
|
||
private final ResourceLocation recipeId;
|
||
private final boolean craftingRecipe;
|
||
private final List<List<GenericStack>> inputs;
|
||
private final List<GenericStack> outputs;
|
||
|
||
public RecipeInfo(
|
||
Object recipeBase,
|
||
ResourceLocation recipeId,
|
||
boolean craftingRecipe,
|
||
List<List<GenericStack>> inputs,
|
||
List<GenericStack> outputs
|
||
) {
|
||
this.recipeBase = recipeBase;
|
||
this.recipeId = recipeId;
|
||
this.craftingRecipe = craftingRecipe;
|
||
this.inputs = inputs;
|
||
this.outputs = outputs;
|
||
}
|
||
|
||
public Object getRecipeBase() {
|
||
return recipeBase;
|
||
}
|
||
|
||
public ResourceLocation getRecipeId() {
|
||
return recipeId;
|
||
}
|
||
|
||
public boolean isCraftingRecipe() {
|
||
return craftingRecipe;
|
||
}
|
||
|
||
public List<List<GenericStack>> getInputs() {
|
||
return inputs;
|
||
}
|
||
|
||
public List<GenericStack> getOutputs() {
|
||
return outputs;
|
||
}
|
||
|
||
public List<ItemStack> selectBestInputs(Map<Item, Integer> bookmarkPriorities) {
|
||
List<ItemStack> selected = new ArrayList<>();
|
||
for (List<GenericStack> slotOptions : inputs) {
|
||
if (slotOptions.isEmpty()) {
|
||
selected.add(ItemStack.EMPTY);
|
||
continue;
|
||
}
|
||
|
||
GenericStack best = slotOptions.get(0);
|
||
int bestPriority = getPriority(best, bookmarkPriorities);
|
||
for (int i = 1; i < slotOptions.size(); i++) {
|
||
GenericStack option = slotOptions.get(i);
|
||
int priority = getPriority(option, bookmarkPriorities);
|
||
if (priority < bestPriority) {
|
||
bestPriority = priority;
|
||
best = option;
|
||
}
|
||
}
|
||
selected.add(toItemStack(best));
|
||
}
|
||
return selected;
|
||
}
|
||
|
||
private int getPriority(GenericStack stack, Map<Item, Integer> priorities) {
|
||
if (stack.what() instanceof AEItemKey itemKey) {
|
||
return priorities.getOrDefault(itemKey.getItem(), Integer.MAX_VALUE);
|
||
}
|
||
return Integer.MAX_VALUE;
|
||
}
|
||
|
||
private ItemStack toItemStack(GenericStack stack) {
|
||
if (stack.what() instanceof AEItemKey itemKey) {
|
||
return itemKey.toStack((int) stack.amount());
|
||
}
|
||
if (stack.what() instanceof AEFluidKey) {
|
||
return GenericStack.wrapInItemStack(stack);
|
||
}
|
||
return ItemStack.EMPTY;
|
||
}
|
||
}
|
||
|