1 files changed, 104 insertions, 0 deletions
diff --git a/src/main/java/bjc/rgens/parser/GenerationState.java b/src/main/java/bjc/rgens/parser/GenerationState.java
new file mode 100644
index 0000000..f5cbc60
--- /dev/null
+++ b/src/main/java/bjc/rgens/parser/GenerationState.java
@@ -0,0 +1,104 @@
+package bjc.rgens.parser;
+
+import bjc.utils.data.IPair;
+import bjc.utils.data.Pair;
+
+import java.util.HashMap;
+import java.util.Map;
+import java.util.Random;
+
+/* 
+ * The current state during generation.
+ *
+ */
+public class GenerationState {
+	/** The current string. */
+	public StringBuilder contents;
+	/** The RNG. */
+	public Random rnd;
+
+	/** The current grammar. */
+	public RGrammar gram;
+	/** The rules of the grammar. */
+	public Map<String, Rule> rules;
+	/** The rules imported from other grammars. */
+	public Map<String, Rule> importRules;
+
+	/** The current set of variables. */
+	public Map<String, String> vars;
+	public Map<String, Rule> rlVars;
+
+	private static final Random BASE = new Random();
+
+	/**
+	 * Create a new generation state.
+	 * 
+	 * @param cont
+	 *            The string being generated.
+	 *
+	 * @param rand
+	 *            The RNG to use.
+	 *
+	 * @param vs
+	 *            The variables to use.
+	 */
+	public GenerationState(StringBuilder cont, Random rand, Map<String, String> vs,
+			Map<String, Rule> rvs, RGrammar gram) {
+		contents = cont;
+		rnd      = rand;
+		vars     = vs;
+		rlVars   = rvs;
+
+		this.gram = gram;
+	
+		this.rules = gram.getRules();
+		this.importRules = gram.getImportRules();
+	}
+
+	public static GenerationState fromGrammar(RGrammar gram) {
+		return fromGrammar(BASE, gram);
+	}
+
+	public static GenerationState fromGrammar(Random rand, RGrammar gram) {
+		return new GenerationState(new StringBuilder(), rand, new HashMap<>(), new HashMap<>(), gram);
+	}
+
+	public void swapGrammar(RGrammar gram) {
+		if(this.gram == gram) return;
+
+		this.gram = gram;
+
+		rules = gram.getRules();
+		
+		importRules = gram.getImportRules();
+	}
+
+	public GenerationState newBuf() {
+		return new GenerationState(new StringBuilder(), rnd, vars, rlVars, gram);
+	}
+
+	/*
+	 * @TODO 6/5/18 Ben Culkin :ImportRefactor
+	 * 
+	 * Change this so that imports in almost all cases have to specify where
+	 * they are importing the rule from, so as to make it clear which rules
+	 * are imported, and which aren't
+	 */
+	public Rule findRule(String ruleName, boolean allowImports) {
+		if(rules.containsKey(ruleName)) {
+			return rules.get(ruleName);
+		}
+
+		if(allowImports) return findImport(ruleName);
+
+		return null;
+	}
+
+	public Rule findImport(String ruleName) {
+		if(importRules.containsKey(ruleName)) {
+			return  importRules.get(ruleName);
+		}
+
+		return null;
+	}
+}