1 files changed, 59 insertions, 0 deletions
diff --git a/BJC-Utils2/src/main/java/bjc/utils/gen/RandomGrammar.java b/BJC-Utils2/src/main/java/bjc/utils/gen/RandomGrammar.java
new file mode 100644
index 0000000..0245b17
--- /dev/null
+++ b/BJC-Utils2/src/main/java/bjc/utils/gen/RandomGrammar.java
@@ -0,0 +1,59 @@
+package bjc.utils.gen;
+
+import java.util.HashMap;
+
+import bjc.utils.funcdata.FunctionalList;
+
+/**
+ * A weighted grammar where all the rules have a equal chance of occuring.
+ * @author ben
+ *
+ * @param <E> The type of grammar elements to use.
+ */
+public class RandomGrammar<E> extends WeightedGrammar<E> {
+
+	/**
+	 * Create a new random grammar.
+	 */
+	public RandomGrammar() {
+		rules = new HashMap<>();
+	}
+
+	/**
+	 * Add cases to a specified rule.
+	 * @param rule The name of the rule to add cases to.
+	 * @param cases The cases to add for this rule.
+	 */
+	@SafeVarargs
+	public final void addCases(E rule, FunctionalList<E>... cases) {		
+		for (FunctionalList<E> cse : cases) {
+			super.addCase(rule, 1, cse);
+		}
+	}
+
+	/**
+	 * Create a rule with the specified name and cases.
+	 * @param rule The name of the rule to add.
+	 * @param cases The cases to add for this rule.
+	 */
+	@SafeVarargs
+	public final void makeRule(E rule, FunctionalList<E>... cases) {
+		super.addRule(rule);
+		
+		for (FunctionalList<E> cse : cases) {
+			super.addCase(rule, 1, cse);
+		}
+	}
+
+	/**
+	 * Create a rule with the specified name and cases.
+	 * @param rule The name of the rule to add.
+	 * @param cases The cases to add for this rule.
+	 */
+	public void makeRule(E rule,
+			FunctionalList<FunctionalList<E>> cases) {
+		super.addRule(rule);
+		
+		cases.forEach(cse -> super.addCase(rule, 1, cse));
+	}
+}