src/game/java/net/minecraft/util/parsing/packrat/Dictionary.java

package net.minecraft.util.parsing.packrat;

import java.util.HashMap;
import java.util.Map;
import javax.annotation.Nullable;

public class Dictionary<S> {
    private final Map<Atom<?>, Rule<S, ?>> terms = new HashMap<>();

    public <T> void put(Atom<T> p_329080_, Rule<S, T> p_331073_) {
        Rule<S, ?> rule = this.terms.putIfAbsent(p_329080_, p_331073_);
        if (rule != null) {
            throw new IllegalArgumentException("Trying to override rule: " + p_329080_);
        }
    }

    public <T> void put(Atom<T> p_335785_, Term<S> p_327787_, Rule.RuleAction<S, T> p_333483_) {
        this.put(p_335785_, Rule.fromTerm(p_327787_, p_333483_));
    }

    public <T> void put(Atom<T> p_333993_, Term<S> p_331531_, Rule.SimpleRuleAction<T> p_336076_) {
        this.put(p_333993_, Rule.fromTerm(p_331531_, p_336076_));
    }

    @Nullable
    public <T> Rule<S, T> get(Atom<T> p_335131_) {
        return (Rule<S, T>)this.terms.get(p_335131_);
    }
}