Toposort recipes so it's actually usably fast

5 files changed, 213 insertions, 0 deletions

diff --git a/src/main/java/lv/enes/mc/eris_alchemy/utils/ConsList.java b/src/main/java/lv/enes/mc/eris_alchemy/utils/ConsList.java
new file mode 100644
index 0000000..bbab8a6
--- /dev/null
+++ b/src/main/java/lv/enes/mc/eris_alchemy/utils/ConsList.java
@@ -0,0 +1,103 @@
+package lv.enes.mc.eris_alchemy.utils;
+
+import jakarta.annotation.Nonnull;
+
+import java.util.Objects;
+import java.util.Optional;
+import java.util.function.Supplier;
+import java.util.stream.Stream;
+
+public sealed abstract class ConsList<E> {
+        public static <E> ConsList<E> cons(E car, ConsList<E> cdr) {
+                return new Cons<>(car, cdr);
+        }
+
+        public static <E> ConsList<E> nil() {
+                return Nil.of();
+        }
+
+        @Nonnull
+        public abstract Optional<E> car();
+
+        @Nonnull
+        public abstract ConsList<E> cdr();
+
+        public boolean contains(E elem) {
+                return stream().anyMatch(x -> Objects.equals(x, elem));
+        }
+
+        public Stream<E> stream() {
+                var car = car();
+                if (car.isEmpty()) {
+                        return Stream.empty();
+                }
+
+                return Stream.<Supplier<Stream<E>>>of(
+                                                () -> Stream.of(car.get()),
+                                                () -> cdr().stream()
+                                )
+                                .flatMap(Supplier::get);
+        }
+
+        @Override
+        public String toString() {
+                if (car().isEmpty()) {
+                        return "[]";
+                }
+
+                var sb = new StringBuilder("[");
+                sb.append(car().get());
+                stream().skip(1)
+                                .forEach(x -> {
+                                        sb.append(" -> ");
+                                        sb.append(x);
+                                });
+                sb.append("]");
+                return sb.toString();
+        }
+
+        private static final class Cons<E> extends ConsList<E> {
+                private final E car;
+                private final ConsList<E> cdr;
+
+                public Cons(E car, ConsList<E> cdr) {
+                        this.car = car;
+                        this.cdr = cdr;
+                }
+
+                @Nonnull
+                @Override
+                public Optional<E> car() {
+                        return Optional.of(car);
+                }
+
+                @Nonnull
+                @Override
+                public ConsList<E> cdr() {
+                        return cdr;
+                }
+        }
+
+        private static final class Nil<E> extends ConsList<E> {
+                private static final Nil<?> INSTANCE = new Nil<>();
+
+                @SuppressWarnings("unchecked")
+                public static <E> Nil<E> of() {
+                        return (Nil<E>)INSTANCE;
+                }
+
+                private Nil(){}
+
+                @Nonnull
+                @Override
+                public Optional<E> car() {
+                        return Optional.empty();
+                }
+
+                @Nonnull
+                @Override
+                public ConsList<E> cdr() {
+                        return Nil.of();
+                }
+        }
+}
diff --git a/src/main/java/lv/enes/mc/eris_alchemy/utils/IngredientProvider.java b/src/main/java/lv/enes/mc/eris_alchemy/utils/IngredientProvider.java
new file mode 100644
index 0000000..e859114
--- /dev/null
+++ b/src/main/java/lv/enes/mc/eris_alchemy/utils/IngredientProvider.java
@@ -0,0 +1,44 @@
+package lv.enes.mc.eris_alchemy.utils;
+
+import com.google.gson.JsonDeserializationContext;
+import com.google.gson.JsonDeserializer;
+import com.google.gson.JsonElement;
+import com.google.gson.JsonParseException;
+import net.minecraft.core.registries.Registries;
+import net.minecraft.resources.ResourceLocation;
+import net.minecraft.tags.TagKey;
+import net.minecraft.world.item.crafting.Ingredient;
+import net.minecraft.world.item.crafting.ShapedRecipe;
+
+import java.lang.reflect.Type;
+import java.util.function.Supplier;
+
+public class IngredientProvider implements JsonDeserializer<Supplier<Ingredient>> {
+        public static Supplier<Ingredient> deserialize(JsonElement el) {
+                if (el.isJsonObject()) {
+                        return () -> Ingredient.of(ShapedRecipe.itemStackFromJson(el.getAsJsonObject()));
+                } else if (el.isJsonPrimitive() && el.getAsJsonPrimitive().isString()) {
+                        return deserialize(el.getAsString().strip());
+                } else {
+                        throw new JsonParseException("Every ingredient should be an object or a string");
+                }
+        }
+
+        public static Supplier<Ingredient> deserialize(String str) {
+                if (str.startsWith("#")) {
+                        var tag = TagKey.create(Registries.ITEM, new ResourceLocation(str.substring(1).strip()));
+                        return () -> Ingredient.of(tag);
+                } else {
+                        return () -> Ingredient.of(ItemUtils.get(new ResourceLocation(str)));
+                }
+        }
+
+        @Override
+        public Supplier<Ingredient> deserialize(
+                        JsonElement el,
+                        Type t,
+                        JsonDeserializationContext c
+        ) {
+                return IngredientProvider.deserialize(el);
+        }
+}
diff --git a/src/main/java/lv/enes/mc/eris_alchemy/utils/ItemUtils.java b/src/main/java/lv/enes/mc/eris_alchemy/utils/ItemUtils.java
index 4414d06..7ac7358 100644
--- a/src/main/java/lv/enes/mc/eris_alchemy/utils/ItemUtils.java
+++ b/src/main/java/lv/enes/mc/eris_alchemy/utils/ItemUtils.java
@@ -3,10 +3,15 @@ package lv.enes.mc.eris_alchemy.utils;
 import net.minecraft.core.Holder;
 import net.minecraft.core.registries.BuiltInRegistries;
 import net.minecraft.resources.ResourceLocation;
+import net.minecraft.world.item.Item;
 import net.minecraft.world.item.ItemStack;
 import net.minecraft.world.level.ItemLike;
 
 public final class ItemUtils {
+        public static Item get(ResourceLocation id) {
+                return BuiltInRegistries.ITEM.get(id);
+        }
+
         public static <I extends ItemLike> ResourceLocation getId(Holder<I> holder) {
                 return getId(holder.value());
         }
diff --git a/src/main/java/lv/enes/mc/eris_alchemy/utils/OptionalDoubleSummer.java b/src/main/java/lv/enes/mc/eris_alchemy/utils/OptionalDoubleSummer.java
new file mode 100644
index 0000000..04b6ca8
--- /dev/null
+++ b/src/main/java/lv/enes/mc/eris_alchemy/utils/OptionalDoubleSummer.java
@@ -0,0 +1,47 @@
+package lv.enes.mc.eris_alchemy.utils;
+
+import java.util.OptionalDouble;
+import java.util.Set;
+import java.util.function.BiConsumer;
+import java.util.function.BinaryOperator;
+import java.util.function.Function;
+import java.util.function.Supplier;
+import java.util.stream.Collector;
+
+public class OptionalDoubleSummer implements Collector<OptionalDouble, Ref<OptionalDouble>, OptionalDouble> {
+        @Override
+        public Supplier<Ref<OptionalDouble>> supplier() {
+                return () -> new Ref<>(OptionalDouble.of(0));
+        }
+
+        @Override
+        public BiConsumer<Ref<OptionalDouble>, OptionalDouble> accumulator() {
+                return (ref, od) -> {
+                        if (ref.getValue().isPresent()) {
+                                if (od.isPresent()) {
+                                        ref.setValue(OptionalDouble.of(ref.getValue().getAsDouble() + od.getAsDouble()));
+                                } else {
+                                        ref.setValue(OptionalDouble.empty());
+                                }
+                        }
+                };
+        }
+
+        @Override
+        public BinaryOperator<Ref<OptionalDouble>> combiner() {
+                return (ref1, ref2) -> {
+                        accumulator().accept(ref1, ref2.getValue());
+                        return ref1;
+                };
+        }
+
+        @Override
+        public Function<Ref<OptionalDouble>, OptionalDouble> finisher() {
+                return Ref::getValue;
+        }
+
+        @Override
+        public Set<Characteristics> characteristics() {
+                return Set.of(Characteristics.UNORDERED);
+        }
+}
diff --git a/src/main/java/lv/enes/mc/eris_alchemy/utils/Ref.java b/src/main/java/lv/enes/mc/eris_alchemy/utils/Ref.java
new file mode 100644
index 0000000..dc2624a
--- /dev/null
+++ b/src/main/java/lv/enes/mc/eris_alchemy/utils/Ref.java
@@ -0,0 +1,14 @@
+package lv.enes.mc.eris_alchemy.utils;
+
+import lombok.Getter;
+import lombok.Setter;
+
+@Setter
+@Getter
+public class Ref<T> {
+        private T value;
+
+        public Ref(T value) {
+                this.value = value;
+        }
+}