package net.minecraft.util.parsing.packrat;

import java.util.HashMap;
import java.util.Map;
import org.jetbrains.annotations.Nullable;

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

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

	public <T> void put(Atom<T> atom, Term<S> term, Rule.RuleAction<S, T> ruleAction) {
		this.put(atom, Rule.fromTerm(term, ruleAction));
	}

	public <T> void put(Atom<T> atom, Term<S> term, Rule.SimpleRuleAction<T> simpleRuleAction) {
		this.put(atom, Rule.fromTerm(term, simpleRuleAction));
	}

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